CInventoryItemPtr CInventory::ValidateWeapon(idEntity* ent, const bool gotFromShop) // grayman (#2376)
{
// Sanity check
if (ent == NULL) return CInventoryItemPtr();
idStr weaponName = ent->spawnArgs.GetString("inv_weapon_name");
if (weaponName.IsEmpty())
{
// Not a weapon item
return CInventoryItemPtr();
}
// Entity has a weapon name set, check for match in our inventory
// Find the weapon category
CInventoryCategoryPtr weaponCategory = GetCategory(TDM_PLAYER_WEAPON_CATEGORY);
if (weaponCategory == NULL)
{
DM_LOG(LC_INVENTORY, LT_ERROR)LOGSTRING("Could not find weapon category in inventory.\r");
return CInventoryItemPtr();
}
// Look for the weapon with the given name
for (int i = 0; i < weaponCategory->GetNumItems(); i++)
{
CInventoryWeaponItemPtr weaponItem =
boost::dynamic_pointer_cast<CInventoryWeaponItem>(weaponCategory->GetItem(i));
// Is this the right weapon? (must be a melee weapon, allowed empty)
if (weaponItem != NULL && weaponItem->IsAllowedEmpty() && weaponItem->GetWeaponName() == weaponName)
{
DM_LOG(LC_INVENTORY, LT_DEBUG)LOGSTRING("Entity %s is matching the melee weapon %s.\r", ent->name.c_str(), weaponName.c_str());
// Enable this weapon
if (!gotFromShop) // grayman (#2376)
{
weaponItem->SetEnabled(true);
}
if (!ent->spawnArgs.GetBool("inv_map_start", "0") && !ent->spawnArgs.GetBool("inv_no_pickup_message", "0"))
{
NotifyOwnerAboutPickup( common->Translate( ent->spawnArgs.GetString("inv_name") ), weaponItem);
}
// We're done
return weaponItem;
}
}
DM_LOG(LC_INVENTORY, LT_DEBUG)LOGSTRING("Couldn't find a match for weapon name: %s.\r", weaponName.c_str());
return CInventoryItemPtr();
}
void CInventory::CopyPersistentItemsFrom(const CInventory& sourceInventory, idEntity* newOwner)
{
// Obtain the weapon category for this inventory
CInventoryCategoryPtr weaponCategory = GetCategory(TDM_PLAYER_WEAPON_CATEGORY);
// Cycle through all categories to add them
for ( int c = 0 ; c < sourceInventory.GetNumCategories() ; ++c )
{
const CInventoryCategoryPtr& category = sourceInventory.GetCategory(c);
for ( int itemIdx = 0 ; itemIdx < category->GetNumItems() ; ++itemIdx )
{
const CInventoryItemPtr& item = category->GetItem(itemIdx);
if (item->GetPersistentCount() <= 0)
{
DM_LOG(LC_INVENTORY, LT_DEBUG)LOGSTRING(
"Item %s is not marked as persistent, won't add to player inventory.\r",
common->Translate(item->GetName().c_str()));
continue; // not marked as persistent
}
// Check if the shop handled that item already
const idDict* itemDict = item->GetSavedItemEntityDict();
if (itemDict != NULL)
{
CShopItemPtr shopItem = gameLocal.m_Shop->FindShopItemDefByClassName(itemDict->GetString("classname"));
if ( (shopItem != NULL) && (CShop::GetQuantityForItem(item) > 0) )
{
// grayman #3723 - if there's no shop in this mission,
// then we can't rely on it to process this inventory item.
if ( gameLocal.m_Shop->ShopExists() )
{
DM_LOG(LC_INVENTORY, LT_DEBUG)LOGSTRING(
"Item %s would be handled by the shop, won't add that to player inventory.\r",
common->Translate(item->GetName().c_str()));
continue;
}
}
}
// Is set to true if we should add this item.
// For weapon items with ammo this will be set to false to prevent double-additions
bool addItem = true;
// Handle weapons separately, otherwise we might end up with duplicate weapon items
CInventoryWeaponItemPtr weaponItem = boost::dynamic_pointer_cast<CInventoryWeaponItem>(item);
if (weaponItem && weaponCategory)
{
// Weapon items need special consideration. For arrow-based weapons try to merge the ammo.
for ( int w = 0 ; w < weaponCategory->GetNumItems() ; ++w )
{
CInventoryWeaponItemPtr thisWeapon = boost::dynamic_pointer_cast<CInventoryWeaponItem>(weaponCategory->GetItem(w));
if (!thisWeapon)
{
continue;
}
if (thisWeapon->GetWeaponName() == weaponItem->GetWeaponName())
{
// Prevent adding this item, we already have one
addItem = false;
// Found a matching weapon, does it use ammo?
if (thisWeapon->NeedsAmmo())
{
DM_LOG(LC_INVENTORY, LT_DEBUG)LOGSTRING(
"Adding persistent ammo %d to player weapon %s.\r",
weaponItem->GetAmmo(), thisWeapon->GetWeaponName().c_str());
// Add the persistent ammo count to this item
thisWeapon->SetAmmo(thisWeapon->GetAmmo() + weaponItem->GetAmmo());
}
else
{
// Doesn't need ammo, check enabled state
if (weaponItem->IsEnabled() && !thisWeapon->IsEnabled())
{
DM_LOG(LC_INVENTORY, LT_DEBUG)LOGSTRING(
"Enabling weapon item %s as the persistent inventory contains an enabled one.\r",
thisWeapon->GetWeaponName().c_str());
thisWeapon->SetEnabled(true);
}
}
break;
}
}
}
if (addItem)
{
DM_LOG(LC_INVENTORY, LT_DEBUG)LOGSTRING(
"Adding persistent item %s to player inventory, quantity: %d.\r",
common->Translate(item->GetName().c_str()), item->GetPersistentCount());
item->SetOwner(newOwner);
// Add this item to our inventory
PutItem(item, item->Category()->GetName());
// If we didn't have a weapon category at this point, we should be able to get one now
if (weaponItem && !weaponCategory)
{
weaponCategory = GetCategory(TDM_PLAYER_WEAPON_CATEGORY);
}
}
}
}
}