Ejemplo n.º 1
0
float CAttributeManager::ApplyAttributeFloat( float flValue, const CBaseEntity *pEntity, string_t strAttributeClass )
{
	if ( m_bParsingMyself || m_hOuter.Get() == NULL )
	{
		return flValue;
	}

	// Safeguard to prevent potential infinite loops.
	m_bParsingMyself = true;

	for ( int i = 0; i < m_AttributeProviders.Count(); i++ )
	{
		CBaseEntity *pProvider = m_AttributeProviders[i].Get();

		if ( !pProvider || pProvider == pEntity )
			continue;

		IHasAttributes *pAttributes = pProvider->GetHasAttributesInterfacePtr();

		if ( pAttributes )
		{
			flValue = pAttributes->GetAttributeManager()->ApplyAttributeFloat( flValue, pEntity, strAttributeClass );
		}
	}

	IHasAttributes *pAttributes = m_hOuter->GetHasAttributesInterfacePtr();
	CBaseEntity *pOwner = pAttributes->GetAttributeOwner();

	if ( pOwner )
	{
		IHasAttributes *pOwnerAttrib = pOwner->GetHasAttributesInterfacePtr();
		if ( pOwnerAttrib )
		{
			flValue = pOwnerAttrib->GetAttributeManager()->ApplyAttributeFloat( flValue, pEntity, strAttributeClass );
		}
	}

	m_bParsingMyself = false;

	return flValue;
}
void CTFPowerupBottle::ReapplyProvision()
{
	CTFWearable::ReapplyProvision();
	
	CBaseEntity *owner = this->GetOwnerEntity();
	if (owner == nullptr) {
		return;
	}
	
	IHasAttributes *owner_ihasattr = owner->GetHasAttributesInterfacePtr();
	if (owner_ihasattr == nullptr) {
		return;
	}
	
	if (this->m_bActive) {
		if (!owner_ihasattr->GetAttributeManager()->IsBeingProvidedToBy(this)) {
			this->GetAttributeManager()->ProvideTo(owner);
		}
	} else {
		this->GetAttributeManager()->StopProvidingTo(owner);
	}
	
	CTFPlayer *player = dynamic_cast<CTFPlayer *>(owner);
	if (player == nullptr) {
		return;
	}
	
	/* BUG: because "canteen_specialist" is an integer attribute, the call to
	 * AttribHookValue<int> will truncate any fractional part of the powerup
	 * duration */
	float duration = CAttributeManager::AttribHookValue<float>(0.0f,
		"powerup_duration", this, nullptr, true);
	duration = CAttributeManager::AttribHookValue<int>((int)duration,
		"canteen_specialist", player, nullptr, true);
	
	CTFPlayer *share_with = nullptr;
	int share_attr = 0;
	
	if (player->IsPlayerClass(TF_CLASS_MEDIC)) {
		CTFWeaponBase *weapon = player->GetActiveWeapon();
		if (weapon != nullptr) {
			CWeaponMedigun *medigun = dynamic_cast<CWeaponMedigun *>(weapon);
			if (medigun != nullptr) {
				share_with = ToTFPlayer(medigun->GetHealTarget());
				if (share_with != nullptr) {
					share_attr = CAttributeManager::AttribHookValue<int>(0,
						"canteen_specialist", player, nullptr, true);
				}
			}
		}
	}
	
	bool did_share = false;
	
	if (CAttributeManager::AttribHookValue<int>(0, "critboost",
		this, nullptr, true) != 0) {
		if (this->m_bActive) {
			player->m_Shared.AddCond(TF_COND_CRITBOOSTED_USER_BUFF, duration);
			
			if (share_with != nullptr && share_attr != 0) {
				share_with->m_Shared.AddCond(TF_COND_CRITBOOSTED_USER_BUFF,
					duration);
				did_share = true;
			}
		} else {
			player->m_Shared.RemoveCond(TF_COND_CRITBOOSTED_USER_BUFF, true);
		}
	}
	
	if (CAttributeManager::AttribHookValue<int>(0, "ubercharge",
		this, nullptr, true) != 0) {
		if (this->m_bActive) {
			player->m_Shared.AddCond(TF_COND_INVULNERABLE_USER_BUFF, duration);
			
			if (player->IsPlayerClass(TF_CLASS_ENGINEER)) {
				int obj_count = player->GetObjectCount();
				if (obj_count > 0) {
					for (int i = obj_count - 1; i != -1; --i) {
						CBaseObject *obj = player->GetObject(i);
						if (obj != nullptr) {
							CObjectSentrygun *sentry = dynamic_cast<CObjectSentrygun *>(obj);
							if (sentry != nullptr && !sentry->m_bCarried) {
								sentry->m_nShieldLevel = 2;
								// TODO: set float @ CObjectSentrygun+0xb14
								// to gpGlobals->curtime + duration
							}
						}
					}
				}
			} else if (share_with != nullptr && share_attr != 0) {
				share_with->m_Shared.AddCond(TF_COND_INVULNERABLE_USER_BUFF,
					duration);
				did_share = true;
			}
		} else {
			player->m_Shared.RemoveCond(TF_COND_INVULNERABLE_USER_BUFF, true);
		}
	}
	
	if (CAttributeManager::AttribHookValue<int>(0, "recall",
		this, nullptr, true) != 0) {
		if (this->m_bActive) {
			player->ForceRespawn();
			player->m_Shared.AddCond(TF_COND_SPEED_BOOST, 7.0f);
		}
	}
	
	if (CAttributeManager::AttribHookValue<int>(0, "refill_ammo",
		this, nullptr, true) != 0) {
		if (this->m_bActive) {
			for (int i = 0; i < MAX_WEAPONS; ++i) {
				CBaseCombatWeapon *weapon = player->GetWeapon(i);
				if (weapon == nullptr) {
					continue;
				}
				
				/* BUG: as soon as this loop hits a weapon which is completely
				 * devoid of ammo (clip 0, reserve 0), it will
				 * (a) fail to refill the clip for that weapon, and
				 * (b) fail to refill the clip for any weapons in later slots
				 * NOTE: for the purposes of this, energy weapons are treated as
				 * if their reserve ammo is always empty
				 */
				if (TFGameRules() != nullptr && TFGameRules()->IsMannVsMachineMode() &&
					((weapon->UsesPrimaryAmmo() && !weapon->HasPrimaryAmmo()) ||
					(weapon->UsesSecondaryAmmo() && !weapon->HasSecondaryAmmo()))) {
					player->AwardAchievement(TF_MVM_USE_AMMO_BOTTLE);
					break;
				}
				
				/* BUG: doesn't refill clip of energy weapons
				 * (should vcall weapon->IsEnergyWeapon,
				 * then call weapon->Energy_Recharge
				 * until weapon->Energy_FullyCharged) */
				
				
				// CBaseCombatWeapon::UsesPrimaryAmmo:
				// - if m_iPrimaryAmmoType < 0: return false
				// - else:                      return true
				// CBaseCombatWeapon::UsesSecondaryAmmo:
				// - if m_iSecondaryAmmoType < 0: return false
				// - else:                        return true
				
				// CTFWeaponBaseGun::HasPrimaryAmmo:
				// - there's a special case if m_iPrimaryAmmoType == TF_AMMO_METAL (e.g. Widowmaker)
				// - if UsesClipsForAmmo1(): return (m_iClip1 > 0) || (pOwner->GetAmmoCount(m_iPrimaryAmmoType) > 0)
				// - else:                   return (pOwner->GetAmmoCount(m_iPrimaryAmmoType) > 0)
				// CBaseCombatWeapon::HasSecondaryAmmo:
				// - if UsesClipsForAmmo1(): return (m_iClip2 > 0) || (pOwner->GetAmmoCount(m_iSecondaryAmmoType) > 0)
				// - else:                   return (pOwner->GetAmmoCount(m_iSecondaryAmmoType) > 0)
				
			
				weapon->GiveDefaultAmmo();
				
				if (share_with != nullptr && share_attr != 0) {
					CBaseCombatWeapon *share_weapon = share_with->GetWeapon(i);
					if (share_weapon != nullptr) {
						share_weapon->GiveDefaultAmmo();
						did_share = true;
					}
				}
			}
			
			/* NOTE: invented TF_AMMO_SOURCE_RESUPPLY for (EAmmoSource)1 */
			if (share_with != nullptr && share_attr != 0) {
				for (int i = 0; i < TF_AMMO_COUNT; ++i) {
					player->GiveAmmo(player->GetMaxAmmo(i), i, true,
						TF_AMMO_SOURCE_RESUPPLY);
					share_with->GiveAmmo(share_with->GetMaxAmmo(i), i, true,
						TF_AMMO_SOURCE_RESUPPLY);
				}
				did_share = true;
			} else {
				for (int i = 0; i < TF_AMMO_COUNT; ++i) {
					player->GiveAmmo(player->GetMaxAmmo(i), i, true,
						TF_AMMO_SOURCE_RESUPPLY);
				}
			}
		}
	}
	
	if (CAttributeManager::AttribHookValue<int>(0, "building_instant_upgrade",
		this, nullptr, true) != 0) {
		if (this->m_bActive) {
			int obj_count = player->GetObjectCount();
			if (obj_count > 0) {
				for (int i = obj_count - 1; i != -1; --i) {
					CBaseObject *obj = player->GetObject(i);
					if (obj != nullptr) {
						int max_level = obj->GetMaxUpgradeLevel();
						
						if (obj->m_bCarried) {
							/* BUG: this can't possibly be right */
							obj->m_iHighestUpgradeLevel = obj->GetMaxUpgradeLevel();
						} else if (obj->GetUpgradeLevel() == max_level) {
							obj->SetHealth(obj->GetMaxHealth());
						} else {
							if (TFGameRules() && TFGameRules()->IsMannVsMachineMode()) {
								// TODO: event mvm_quick_sentry_upgrade
							}
							obj->DoQuickBuild(true);
						}
					}
				}
			}
		}
	}
	
	if (did_share) {
		// TODO: event mvm_medic_powerup_shared
	}
}