register_forward(FM_SetModel, "fwSetModel");

new szModel_P[] = "models/warcraft3/p_heartbomb.mdl"
new szModel_V[] = "models/warcraft3/v_heartbomb.mdl"
new szModel_W[] = "models/warcraft3/w_heartbomb.mdl"

new bool:bIsHasHE[33];

public fwSetModel(entModel, const model[])
{
       if ( !pev_valid ( entModel ) )
        return FMRES_IGNORED;

       if (strlen(model) < 8)
        return FMRES_IGNORED;

       new idUser, model_w[64];
       idUser = pev(entModel, pev_owner);

       if(p_data[idUser][P_RACE] == RACE_ORC)
       {
        format(model_w, charsmax(model_w), "%s", model[9]);

        new idwpn = get_idwpn_from_wmodel(model_w);
        if(idwpn == CSW_HEGRENADE)
        {
         engfunc(EngFunc_SetModel, entModel, szModel_W);
                  
         return FMRES_SUPERCEDE;
        }
       }
             
       return FMRES_IGNORED;
}

get_idwpn_from_wmodel(model_w[])
{
       new wpnname[64], idwpn;
       format(wpnname, charsmax(wpnname), "weapon_%s", model_w);
       replace(wpnname, charsmax(wpnname), ".mdl", "");
       replace(wpnname, charsmax(wpnname), "_mp5", "_mp5navy");
       strtolower(wpnname);
       idwpn = get_weaponid(wpnname);
             
       return idwpn;
}

precache_model(szModel_P);
precache_model(szModel_V);
precache_model(szModel_W);

       
       if(p_data_b[idUser][PB_HAS_SPAWNED] && p_data[idUser][P_RACE] == RACE_ORC)
       {
        if(iCurWeapon == CSW_HEGRENADE)
        {
         if(!bIsHasHE[idUser])
         {
          entity_set_string(idUser,EV_SZ_viewmodel,szModel_V);
          entity_set_string(idUser,EV_SZ_weaponmodel,szModel_P);
          bIsHasHE[idUser] = true;
         }
        }       
              
        if(iCurWeapon != CSW_HEGRENADE)       
         bIsHasHE[idUser] = false;
}