/*** Ammo equipment for pawn using FireTypeManager with ammo.** Copyright (c) 2016 The Farm 51** Authors: sadamczyk, draczka*/#include "WW3.h"#include "WW3AmmoEquipment.h"#include "WW3WAPreview_AmmoType.h"UWW3AmmoEquipment::UWW3AmmoEquipment(){DefaultEquipment = TArray<UWW3AmmoInfo*>();SetIsReplicated(true);PrimaryComponentTick.bCanEverTick = false;bAnyAmmoInLowLevelState = false;}void UWW3AmmoEquipment::OnConstruction(const FTransform& Transform){}void UWW3AmmoEquipment::OnComponentDestroyed(bool bDestroyingHierarchy){UWW3AssetManager* const AssetManager = UWW3FunctionLibrary::GetWW3AssetManager();if (AssetManager != nullptr){for (int32 i = AdditionalAmmoTypesToAddLoadingHandles.Num() - 1; i >= 0; --i){AssetManager->ReleaseHandle(AdditionalAmmoTypesToAddLoadingHandles[i]);}}Super::OnComponentDestroyed(bDestroyingHierarchy);}void UWW3AmmoEquipment::AddAmmoTypeFromReplication(UWW3AmmoInfo* AmmoToAdd){if (AmmoToAdd != nullptr){DefaultEquipment.Add(AmmoToAdd);SubscribeToEventsFromAmmo(AmmoToAdd);}}void UWW3AmmoEquipment::RemoveAmmoTypeFromReplication(UWW3AmmoInfo* AmmoToAdd){if (AmmoToAdd != nullptr){DefaultEquipment.Remove(AmmoToAdd);UnSubscribeToEventsFromAmmo(AmmoToAdd);}}int32 UWW3AmmoEquipment::AddAmmoType(UWW3AmmoInfo* AmmoToAdd){if (AmmoToAdd != nullptr && AmmoToAdd->IsValid()){DefaultEquipment.Add(AmmoToAdd);SubscribeToEventsFromAmmo(AmmoToAdd);return AmmoToAdd->Amount;}return 0;}void UWW3AmmoEquipment::AddAmmoType(TSubclassOf<UWW3AmmoType> AmmoType, int32 Amount, int32 MaximumAmountExpand, EWW3VehicleAmmoSlot AmmoSlot /*= EWW3VehicleAmmoSlot::None*/, uint16 AdditionalAmmoID /*= INDEX_NONE*/){if (AmmoType == nullptr){UE_LOG(LogWW3Components, Error, TEXT("UWW3AmmoEquipment::AddAmmoType - Cannot add ammo of no type! "));return;}uint8 AmmoPackRemoveMultiplier = 1;UWW3AmmoType* AmmoTypeDefObj = AmmoType->GetDefaultObject<UWW3AmmoType>();if (IsValid(AmmoTypeDefObj)){AmmoPackRemoveMultiplier = AmmoTypeDefObj->AmmoPackRemoveMultiplier;}const int32 IndexInEquipment = GetIndexInEquipment(AmmoType);if (IndexInEquipment < 0){/** Add ammo to equipment only when we do not have this ammo type already */UWW3AmmoInfo* AmmoInfo = NewObject<UWW3AmmoInfo>(this);if (AmmoInfo != nullptr){AmmoInfo->AmmoPackRemoveMultiplier = AmmoPackRemoveMultiplier;AmmoInfo->SetAmmoType(AmmoType);AmmoInfo->SetMaximumAmount(MaximumAmountExpand);AmmoInfo->Add(Amount);AmmoInfo->VehicleAmmoSlot = AmmoSlot;AmmoInfo->SetAdditionalAmmoID(AdditionalAmmoID);AddAmmoType(AmmoInfo);}}else{if (AmmoPackRemoveMultiplier > 1){DefaultEquipment[IndexInEquipment]->AmmoPackRemoveMultiplier = AmmoPackRemoveMultiplier;}/** When we have this ammo type in equipment then just add ammo to it */DefaultEquipment[IndexInEquipment]->SetMaximumAmount(DefaultEquipment[IndexInEquipment]->MaximumAmount + MaximumAmountExpand); //maxiumum ammo amount may changeDefaultEquipment[IndexInEquipment]->Add(Amount);}}void UWW3AmmoEquipment::AddAmmoType(uint16 AdditionalAmmoID, int32 Amount, int32 MaximumAmountExpand, EWW3VehicleAmmoSlot AmmoSlot /*= EWW3VehicleAmmoSlot::None*/){UWW3ItemDatabase* const Database = UWW3FunctionLibrary::GetItemDatabase();if (Database == nullptr){return;}const UWW3DatabaseItemData* const ItemData = Database->GetItemById(AdditionalAmmoID);if (ItemData == nullptr || ItemData->ItemClass.IsNull()){return;}UWW3AssetManager* const AssetManager = UWW3FunctionLibrary::GetWW3AssetManager();if (AssetManager != nullptr){TSharedPtr<FWW3LoadingAssetHandle> Handle;AssetManager->RequestAsyncLoad(Handle, ItemData->ItemClass.ToSoftObjectPath(), FWW3AssetLoadedDelegate::CreateUObject(this, &UWW3AmmoEquipment::OnAdditionalAmmoLoadedForSetup, AdditionalAmmoID, Amount, MaximumAmountExpand, AmmoSlot));AdditionalAmmoTypesToAddLoadingHandles.Add(Handle);}}void UWW3AmmoEquipment::RemoveAdditionalAmmo(uint16 AdditionalAmmoID){for (int i = 0; i < DefaultEquipment.Num(); ++i){if (DefaultEquipment[i] != nullptr && DefaultEquipment[i]->GetAdditionalAmmoID() == AdditionalAmmoID){DefaultEquipment[i]->SetAdditionalAmmoID(INDEX_NONE);DefaultEquipment[i]->FlushDormancy();DefaultEquipment[i]->SetCleanUpAfterReplicate();break;}}}void UWW3AmmoEquipment::AddAmmoTypeAndInitReload(uint16 InWeaponSpecialAmmoItemId, uint16 NewAdditionalAmmoID, int32 Amount){UWW3ItemDatabase* const Database = UWW3FunctionLibrary::GetItemDatabase();if (Database == nullptr){return;}const UWW3DatabaseItemData* const ItemData = Database->GetItemById(NewAdditionalAmmoID);if (ItemData == nullptr || ItemData->ItemClass.IsNull()){return;}UWW3AssetManager* const AssetManager = UWW3FunctionLibrary::GetWW3AssetManager();if (AssetManager != nullptr){TSharedPtr<FWW3LoadingAssetHandle> Handle;AssetManager->RequestAsyncLoad(Handle, ItemData->ItemClass.ToSoftObjectPath(), FWW3AssetLoadedDelegate::CreateUObject(this, &UWW3AmmoEquipment::OnAddAmmoTypeAndInitReload, InWeaponSpecialAmmoItemId, NewAdditionalAmmoID, Amount));AdditionalAmmoTypesToAddLoadingHandles.Add(Handle);}}void UWW3AmmoEquipment::ReplaceAmmoType(uint16 AdditionalAmmoID, uint16 NewAdditionalAmmoID, int32 Amount){UWW3ItemDatabase* const Database = UWW3FunctionLibrary::GetItemDatabase();if (Database == nullptr){return;}const UWW3DatabaseItemData* const ItemData = Database->GetItemById(NewAdditionalAmmoID);if (ItemData == nullptr || ItemData->ItemClass.IsNull()){return;}UWW3AssetManager* const AssetManager = UWW3FunctionLibrary::GetWW3AssetManager();if (AssetManager != nullptr){TSharedPtr<FWW3LoadingAssetHandle> Handle;AssetManager->RequestAsyncLoad(Handle, ItemData->ItemClass.ToSoftObjectPath(), FWW3AssetLoadedDelegate::CreateUObject(this, &UWW3AmmoEquipment::OnReplaceAdditionalAmmoLoadedForSetup, AdditionalAmmoID, NewAdditionalAmmoID, Amount));AdditionalAmmoTypesToAddLoadingHandles.Add(Handle);}}int32 UWW3AmmoEquipment::RestoreAmmoForIndex(int32 Index, int32 Amount){if (Index >= 0 && DefaultEquipment.IsValidIndex(Index) && DefaultEquipment[Index] != nullptr){return DefaultEquipment[Index]->Add(Amount);}return 0;}void UWW3AmmoEquipment::ClearAmmoEquipment(){UnsubscribeEventsFromAllAmmo();DefaultEquipment.Empty();OnAmmoChanged();}void UWW3AmmoEquipment::OnAdditionalAmmoLoadedForSetup(uint16 AdditionalAmmoID, int32 Amount, int32 MaximumAmountExpand, EWW3VehicleAmmoSlot AmmoSlot){UWW3ItemDatabase* const Database = UWW3FunctionLibrary::GetItemDatabase();if (Database == nullptr){return;}const UWW3DatabaseItemData* const ItemData = Database->GetItemById(AdditionalAmmoID);if (ItemData == nullptr || ItemData->ItemClass.IsNull()){return;}if (ItemData->ItemClass.IsValid()){UWW3WAPreview_AmmoType* AmmoType = Cast<UWW3WAPreview_AmmoType>(ItemData->ItemClass.Get()->GetDefaultObject());if (AmmoType != nullptr){AddAmmoType(AmmoType->AmmoTypeToAdd, Amount, MaximumAmountExpand, AmmoSlot, AdditionalAmmoID);}}}void UWW3AmmoEquipment::OnReplaceAdditionalAmmoLoadedForSetup(uint16 AdditionalAmmoID, uint16 NewAdditionalAmmoID, int32 Amount){UWW3ItemDatabase* const Database = UWW3FunctionLibrary::GetItemDatabase();if (Database == nullptr){return;}const UWW3DatabaseItemData* const ItemData = Database->GetItemById(NewAdditionalAmmoID);if (ItemData == nullptr || ItemData->ItemClass.IsNull()){return;}if (ItemData->ItemClass.IsValid()){UWW3WAPreview_AmmoType* AmmoType = Cast<UWW3WAPreview_AmmoType>(ItemData->ItemClass.Get()->GetDefaultObject());if (AmmoType != nullptr){RemoveAdditionalAmmo(AdditionalAmmoID);AddAmmoType(AmmoType->AmmoTypeToAdd, Amount, Amount, EWW3VehicleAmmoSlot::None, NewAdditionalAmmoID);}}}void UWW3AmmoEquipment::OnAddAmmoTypeAndInitReload(uint16 InWeaponSpecialAmmoItemId, uint16 NewAdditionalAmmoID, int32 Amount){UWW3ItemDatabase* const Database = UWW3FunctionLibrary::GetItemDatabase();if (Database == nullptr){return;}const UWW3DatabaseItemData* const ItemData = Database->GetItemById(NewAdditionalAmmoID);if (ItemData == nullptr || ItemData->ItemClass.IsNull()){return;}if (ItemData->ItemClass.IsValid()){UWW3WAPreview_AmmoType* AmmoType = Cast<UWW3WAPreview_AmmoType>(ItemData->ItemClass.Get()->GetDefaultObject());if (AmmoType != nullptr){AddAmmoType(AmmoType->AmmoTypeToAdd, Amount, Amount, EWW3VehicleAmmoSlot::None, NewAdditionalAmmoID);auto AmmoInfoPtr = DefaultEquipment.FindByPredicate([NewAdditionalAmmoID](auto AmmoInfo) -> bool {if (AmmoInfo->GetAdditionalAmmoID() == NewAdditionalAmmoID){return true;}return false;});if (AmmoInfoPtr != nullptr){(*AmmoInfoPtr)->SetStartChangeAmmoOnClient(InWeaponSpecialAmmoItemId);(*AmmoInfoPtr)->FlushDormancy();}}}}void UWW3AmmoEquipment::SetupAndClearAmmoEquipment(const TArray<UWW3AmmoInfo*>& NewEquipment){ClearAmmoEquipment();DefaultEquipment = NewEquipment;OnAmmoChanged();}int32 UWW3AmmoEquipment::WithdrawAmmo(TSubclassOf<UWW3AmmoType> AmmoToWithdraw, int32 AmmoAmount){int32 IndexInEquipment = GetIndexInEquipment(AmmoToWithdraw);return WithdrawAmmo(IndexInEquipment, AmmoAmount);}int32 UWW3AmmoEquipment::WithdrawAmmo(int32 AmmmoToWithdrawIndex, int32 AmmoAmount){/** We shouldnt physicly withdraw ammo on clients. But for avoiding desynhronization we simulate withdrawing but do not subtract it from equipment. *//** Equipment replicate back to client form server. */int32 SubtractedAmmo = 0;if (AmmmoToWithdrawIndex >= 0 && DefaultEquipment[AmmmoToWithdrawIndex] != nullptr){if (GetNetMode() < ENetMode::NM_Client){SubtractedAmmo = DefaultEquipment[AmmmoToWithdrawIndex]->Subtract(AmmoAmount);OnAmmoChanged();}else{SubtractedAmmo = DefaultEquipment[AmmmoToWithdrawIndex]->CheckAmmo(AmmoAmount);}}return SubtractedAmmo;}bool UWW3AmmoEquipment::HaveAlternativeAmmoInEquipment(TSubclassOf<UWW3AmmoType> AllowedAmmoType, TSubclassOf<UWW3AmmoType> CurrentAmmoTypeInClip){TArray<TSubclassOf<UWW3AmmoType>> AmmoTypes;GetAlternativeAmmoTypes(AllowedAmmoType, CurrentAmmoTypeInClip, false, AmmoTypes);return AmmoTypes.Num() > 0;}bool UWW3AmmoEquipment::CanRestoreAnyAmmo(const TArray<WW3RestoreAmmoInfo>& RestoreAmmoInfo, const TArray<WW3RestoreAmmoInfo>& OtherWeaponRestoreAmmoInfo){for (int32 i = 0; i < DefaultEquipment.Num(); i++){if (DefaultEquipment[i] != nullptr && DefaultEquipment[i]->IsValid()){int IndexOfAllowedAmmoType = INDEX_NONE;for (int j = 0; j < RestoreAmmoInfo.Num(); ++j){if (DefaultEquipment[i]->IsTypeOf(RestoreAmmoInfo[j].GetAllowedAmmoType())){IndexOfAllowedAmmoType = j;break;}}if (IndexOfAllowedAmmoType != INDEX_NONE){int AmountInClip = 0;if (RestoreAmmoInfo[IndexOfAllowedAmmoType].GetFireType()->GetCurrentAmmoTypeInClip() == DefaultEquipment[i]->AmmoType){AmountInClip = RestoreAmmoInfo[IndexOfAllowedAmmoType].GetFireType()->GetCurrentAmmoAmountInClip();}for (const WW3RestoreAmmoInfo& AmmoInfo : OtherWeaponRestoreAmmoInfo){if (DefaultEquipment[i]->IsTypeOf(AmmoInfo.GetAllowedAmmoType())){AmountInClip += AmmoInfo.GetFireType()->GetCurrentAmmoAmountInClip();}}if (DefaultEquipment[i]->CanBeRestore(AmountInClip)){return true;}}}}return false;}bool UWW3AmmoEquipment::CanRestoreAmmoFromEquipment(const TArray<AWW3InventoryWeapon*> & Weapons) const{return CanRestoreAmmoFromEquipment(DefaultEquipment, Weapons);}bool UWW3AmmoEquipment::CanRestoreAmmoFromEquipment(TSubclassOf<UWW3AmmoType> AllowedAmmoType, const TArray<AWW3InventoryWeapon*> & Weapons) const{auto ValidAmmoTypes = DefaultEquipment.FilterByPredicate([&](auto AmmoInfo) -> bool {if (AmmoInfo->IsTypeOf(AllowedAmmoType))return true;return false;});return CanRestoreAmmoFromEquipment(ValidAmmoTypes, Weapons);}bool UWW3AmmoEquipment::CanRestoreAmmoFromEquipment(const TArray<UWW3AmmoInfo*> &DefEquipment, const TArray<AWW3InventoryWeapon*> & Weapons) const{for (int32 i = 0; i < DefEquipment.Num(); ++i){if (DefEquipment[i] != nullptr && DefEquipment[i]->IsValid()){if (DefEquipment[i]->GetRefillFromEquipmentPack()){int AmmoInClip = 0;for (AWW3InventoryWeapon* InventoryWeapon : Weapons){if (InventoryWeapon != nullptr && InventoryWeapon->FireTypeManager != nullptr &&InventoryWeapon->FireTypeManager->GetCurrentFireType() != nullptr){if (DefEquipment[i]->AmmoType == InventoryWeapon->FireTypeManager->GetCurrentFireType()->GetCurrentAmmoTypeInClip() ){AmmoInClip = InventoryWeapon->FireTypeManager->GetCurrentFireType()->GetCurrentAmmoAmountInClip();break;}}}if (DefEquipment[i]->CanAddAmmo(AmmoInClip)){return true;}}}}return false;}void UWW3AmmoEquipment::GetAlternativeAmmoTypes(TSubclassOf<UWW3AmmoType> AllowedAmmoType, TSubclassOf<UWW3AmmoType> CurrentAmmoTypeInClip, bool bAddCurrent, TArray<TSubclassOf<UWW3AmmoType>>& OutTypes){TArray<UWW3AmmoInfo*> AmmoInfos;GetAmmoInfosOfType(AllowedAmmoType, AmmoInfos);for (int32 i = 0; i < AmmoInfos.Num(); i++){if (AmmoInfos[i] != nullptr && AmmoInfos[i]->IsValid() && (AmmoInfos[i]->AmmoType != CurrentAmmoTypeInClip || bAddCurrent)){OutTypes.Add(AmmoInfos[i]->AmmoType);}}}TSubclassOf<UWW3AmmoType> UWW3AmmoEquipment::GetAlternativeAmmoTypeOfIndex(TSubclassOf<UWW3AmmoType> AllowedAmmoType, TSubclassOf<UWW3AmmoType> CurrentAmmoTypeInClip, int32 Index){TArray<TSubclassOf<UWW3AmmoType>> AlternativeAmmos;GetAlternativeAmmoTypes(AllowedAmmoType, CurrentAmmoTypeInClip, true, AlternativeAmmos);if (AlternativeAmmos.IsValidIndex(Index)){return AlternativeAmmos[Index];}return nullptr;}TSubclassOf<UWW3AmmoType> UWW3AmmoEquipment::GetNextAlternativeAmmoType(TSubclassOf<UWW3AmmoType> AllowedAmmoType, TSubclassOf<UWW3AmmoType> CurrentAmmoTypeInClip){TArray<UWW3AmmoInfo*> AmmoInfos;GetAmmoInfosOfType(AllowedAmmoType, AmmoInfos);int32 Index = -1;for (int32 i = 0; i < AmmoInfos.Num(); i++){if (AmmoInfos[i] != nullptr && AmmoInfos[i]->IsValid() && AmmoInfos[i]->AmmoType == CurrentAmmoTypeInClip){if (i + 1 < AmmoInfos.Num()){Index = i + 1;}else if (AmmoInfos.Num() > 1){Index = 0;}break;}}return AmmoInfos.IsValidIndex(Index) && AmmoInfos[Index] != nullptr ? AmmoInfos[Index]->AmmoType : nullptr;}UWW3AmmoInfo * UWW3AmmoEquipment::GetNextAlternativeAmmoInfo(TSubclassOf<UWW3AmmoType> AllowedAmmoType, TSubclassOf<UWW3AmmoType> CurrentAmmoTypeInClip){TArray<UWW3AmmoInfo*> AmmoInfos;GetAmmoInfosOfType(AllowedAmmoType, AmmoInfos);int32 Index = -1;for (int32 i = 0; i < AmmoInfos.Num(); i++){if (AmmoInfos[i] != nullptr && AmmoInfos[i]->IsValid() && AmmoInfos[i]->AmmoType == CurrentAmmoTypeInClip){if (i + 1 < AmmoInfos.Num()){Index = i + 1;}else if (AmmoInfos.Num() > 1){Index = 0;}break;}}return AmmoInfos.IsValidIndex(Index) && AmmoInfos[Index] != nullptr ? AmmoInfos[Index] : nullptr;}int32 UWW3AmmoEquipment::GetAmountOfAmmoInfosOfType(TSubclassOf<UWW3AmmoType> AmmoAmount){TArray<UWW3AmmoInfo*> AmmoInfo;GetAmmoInfosOfType(AmmoAmount, AmmoInfo);return AmmoInfo.Num();}int32 UWW3AmmoEquipment::GetAmmoAmount(TSubclassOf<UWW3AmmoType> AmmoType) const{int32 IndexInEquipment = GetIndexInEquipment(AmmoType);return GetAmmoAmount(IndexInEquipment);}int32 UWW3AmmoEquipment::GetAmmoAmount(int32 AmmoTypeIndex) const{int32 AmmoAmount = 555;if (AmmoTypeIndex >= 0 && DefaultEquipment.IsValidIndex(AmmoTypeIndex)){AmmoAmount = DefaultEquipment[AmmoTypeIndex] != nullptr ? DefaultEquipment[AmmoTypeIndex]->Amount : 444;}return AmmoAmount;}void UWW3AmmoEquipment::SubtractMaxAmmoUsingAmmoType(TSubclassOf<UWW3AmmoType> AmmoType, int32 AmountToSubtract){int32 IndexInEquipment = GetIndexInEquipment(AmmoType);if (IndexInEquipment >= 0 && DefaultEquipment[IndexInEquipment] != nullptr){if (GetNetMode() < ENetMode::NM_Client){DefaultEquipment[IndexInEquipment]->SubtractMaxAmmo(AmountToSubtract);OnAmmoChanged();}}}void UWW3AmmoEquipment::GetAmmoInfosOfType(TSubclassOf<UWW3AmmoType> TypeOf, TArray<UWW3AmmoInfo*>& OutInfos){if (TypeOf != nullptr){for (int32 i = 0; i < DefaultEquipment.Num(); i++){if (DefaultEquipment[i] != nullptr && DefaultEquipment[i]->IsValid() && DefaultEquipment[i]->IsTypeOf(TypeOf)){OutInfos.Add(DefaultEquipment[i]);}}}}int32 UWW3AmmoEquipment::GetNumAmmoInfoElements() const{return DefaultEquipment.Num();}UWW3AmmoInfo* UWW3AmmoEquipment::GetAmmoInfoElementAt(int32 index){return DefaultEquipment[index];}void UWW3AmmoEquipment::GetDisplayNameOfAmmo(int32 Index, FString& OutName){OutName = TEXT("Invalid");if (DefaultEquipment.IsValidIndex(Index)){GetDisplayNameOfAmmo(DefaultEquipment[Index], OutName);}}void UWW3AmmoEquipment::GetDisplayNameOfAmmo(UWW3AmmoInfo* AmmoInfo, FString& OutName){OutName = TEXT("Invalid");if (AmmoInfo != nullptr && AmmoInfo->IsValid()){OutName = AmmoInfo->AmmoType->GetDefaultObject<UWW3AmmoType>()->GetAmmoTypeDisplayName();}}TArray<UWW3AmmoInfo*>& UWW3AmmoEquipment::GetCurrentEquipment(){ return DefaultEquipment;}void UWW3AmmoEquipment::Restore(const TArray<UWW3AmmoInfo*>& NewEquipment){ DefaultEquipment = NewEquipment;OnAmmoChanged();}float UWW3AmmoEquipment::RestoreAmmo(float PercentToRestore){float RestoredAmount = 0;for (int32 i = 0; i < DefaultEquipment.Num(); i++){if (DefaultEquipment[i] != nullptr && DefaultEquipment[i]->CanBeRestore(0)){int32 AmountToAdd = DefaultEquipment[i]->MaximumAmount * (PercentToRestore / 100);AmountToAdd = FMath::Max(AmountToAdd, 1);int32 AmmoAddedToBackpack = RestoreAmmoForIndex(i, AmountToAdd);if (AmmoAddedToBackpack > 0){RestoredAmount += 1.0f;break;}}}if (RestoredAmount > 0.0f){OnAmmoChanged();}return RestoredAmount;}float UWW3AmmoEquipment::RestoreAllAmmoAtOnce(float PercentToRestore){float RestoredAmount = 0;for (int32 i = 0; i < DefaultEquipment.Num(); i++){if (DefaultEquipment[i] != nullptr && DefaultEquipment[i]->CanBeRestore(0)){int32 AmountToAdd = DefaultEquipment[i]->MaximumAmount * (PercentToRestore / 100);AmountToAdd = FMath::Max(AmountToAdd, 1);int32 AmmoAddedToBackpack = RestoreAmmoForIndex(i, AmountToAdd);if (AmmoAddedToBackpack > 0){RestoredAmount += 1.0f;}}}if (RestoredAmount > 0.0f){OnAmmoChanged();}return RestoredAmount;}float UWW3AmmoEquipment::RestoreAllAmmoAtOnce(float PercentToRestore, const TArray<UWW3FireTypeManager*>& FireTypeManagers){float RestoredAmount = 0;for (int32 i = 0; i < DefaultEquipment.Num(); i++){if (DefaultEquipment[i] != nullptr){int32 AmmoInClip = 0;for (UWW3FireTypeManager* FTM : FireTypeManagers){if (FTM != nullptr){AmmoInClip = FTM->GetAmmoInClip(DefaultEquipment[i]->AmmoType);if (AmmoInClip >= 0){break;}}}if (DefaultEquipment[i]->CanBeRestore(AmmoInClip)){int32 AmountToAdd = DefaultEquipment[i]->MaximumAmount * (PercentToRestore / 100);AmountToAdd = FMath::Max(AmountToAdd, 1);AmountToAdd = FMath::Min(AmountToAdd, DefaultEquipment[i]->MaximumAmount - (DefaultEquipment[i]->Amount + AmmoInClip));int32 AmmoAddedToBackpack = RestoreAmmoForIndex(i, AmountToAdd);if (AmmoAddedToBackpack > 0){RestoredAmount += 1.0f;}}}}if (RestoredAmount > 0.0f){OnAmmoChanged();}return RestoredAmount;}float UWW3AmmoEquipment::RestoreAmmoForAllAmmoOfType(TSubclassOf<UWW3AmmoType> AllowedAmmoType, float PercentToRestore, float Amount){float RestoredAmount = 0;if (PercentToRestore <= 0.0f){return RestoredAmount;}for (int32 i = 0; i < DefaultEquipment.Num(); i++){if (RestoredAmount < Amount && DefaultEquipment[i] != nullptr){if (DefaultEquipment[i]->IsValid() && DefaultEquipment[i]->IsTypeOf(AllowedAmmoType)){int32 AmountToAdd = FMath::Max((int32)(DefaultEquipment[i]->MaximumAmount * (PercentToRestore / 100)) , 1 ) ;int32 AmmoAddedToBackpack = RestoreAmmoForIndex(i, AmountToAdd);if (AmmoAddedToBackpack > 0){RestoredAmount += 1.0f;break;}}}}if (RestoredAmount > 0.0f){OnAmmoChanged();}return RestoredAmount;}float UWW3AmmoEquipment::RestoreAmmoClipsForAllAmmoOfType(const TArray<WW3RestoreAmmoInfo>& AllowedAmmoType, int ClipsToRestore, UWW3FireType* CurrentFireType, const TArray<WW3RestoreAmmoInfo>& OtherWeaponRestoreInfo){float RestoredAmount = 0;if (ClipsToRestore <= 0.0f){return RestoredAmount;}TArray< TTuple< int32, UWW3AmmoInfo*, UWW3FireType* >> DefaultAmmoEqu;for (int32 i = 0; i < DefaultEquipment.Num(); ++i){if (DefaultEquipment[i] != nullptr && DefaultEquipment[i]->IsValid()){for (int j = 0; j < AllowedAmmoType.Num(); ++j){if (DefaultEquipment[i]->IsTypeOf(AllowedAmmoType[j].GetAllowedAmmoType())){DefaultAmmoEqu.Add(MakeTuple(i, DefaultEquipment[i], AllowedAmmoType[j].GetFireType()));break;}}}}int CurrentAmmoIndex = 0;for (int32 i = 0; i < DefaultAmmoEqu.Num(); ++i){if (DefaultAmmoEqu[i].Get<1>()->IsTypeOf(CurrentFireType->GetCurrentAmmoTypeInClip())){CurrentAmmoIndex = i;break;}}if (CurrentAmmoIndex != 0){DefaultAmmoEqu.Swap(0, CurrentAmmoIndex);}for (int32 i = 0; i < DefaultAmmoEqu.Num(); ++i){int32 AmountToAdd = DefaultAmmoEqu[i].Get<2>()->GetClipSize() * ClipsToRestore;if(AmountToAdd <= 0)continue;int AmountInClip = 0;if (DefaultAmmoEqu[i].Get<1>()->AmmoType == CurrentFireType->GetCurrentAmmoTypeInClip()){AmountInClip = CurrentFireType->GetCurrentAmmoAmountInClip();}for (const WW3RestoreAmmoInfo& AmmoInfo : OtherWeaponRestoreInfo){if (DefaultAmmoEqu[i].Get<1>()->IsTypeOf(AmmoInfo.GetAllowedAmmoType())){AmountInClip += AmmoInfo.GetFireType()->GetCurrentAmmoAmountInClip();}}if ((DefaultEquipment[DefaultAmmoEqu[i].Get<0>()]->Amount + AmountInClip + AmountToAdd) > DefaultEquipment[DefaultAmmoEqu[i].Get<0>()]->MaximumAmount){AmountToAdd = (DefaultEquipment[DefaultAmmoEqu[i].Get<0>()]->MaximumAmount - DefaultEquipment[DefaultAmmoEqu[i].Get<0>()]->Amount - AmountInClip);}int32 AmmoAddedToBackpack = RestoreAmmoForIndex( DefaultAmmoEqu[i].Get<0>(), AmountToAdd);if (AmmoAddedToBackpack > 0){RestoredAmount += AmmoAddedToBackpack;break;}}if (RestoredAmount > 0.0f){OnAmmoChanged();}return RestoredAmount;}float UWW3AmmoEquipment::RestoreAllAmmoFromEquipmentPack(const TArray<AWW3InventoryWeapon*> & Weapons){float RestoredAmount = 0;for (int32 i = 0; i < DefaultEquipment.Num(); i++){if (DefaultEquipment[i] != nullptr && DefaultEquipment[i]->GetRefillFromEquipmentPack()){int AmountInClip = 0;for (AWW3InventoryWeapon* InventoryWeapon : Weapons){if (InventoryWeapon != nullptr && InventoryWeapon->FireTypeManager != nullptr && InventoryWeapon->FireTypeManager->GetCurrentFireType() != nullptr){if (DefaultEquipment[i]->AmmoType == InventoryWeapon->FireTypeManager->GetCurrentFireType()->GetCurrentAmmoTypeInClip() ){AmountInClip = InventoryWeapon->FireTypeManager->GetCurrentFireType()->GetCurrentAmmoAmountInClip();break;}}}if (DefaultEquipment[i]->CanAddAmmo(AmountInClip)){int32 AmountToAdd = DefaultEquipment[i]->MaximumAmount - DefaultEquipment[i]->Amount - AmountInClip;if (AmountToAdd > 0){int32 AmmoAddedToBackpack = RestoreAmmoForIndex(i, AmountToAdd);if (AmmoAddedToBackpack > 0){RestoredAmount += 1.0f;}}}}}if (RestoredAmount > 0.0f){OnAmmoChanged();}return RestoredAmount;}float UWW3AmmoEquipment::TryRestoreGadgetsAmount(float PercentToRestore, float Amount){return 0.0f;}void UWW3AmmoEquipment::SubscribeToEventsFromAmmo(UWW3AmmoInfo* AmmoInfoEnteringBackpack){if (AmmoInfoEnteringBackpack != nullptr){AmmoInfoEnteringBackpack->EventOnAmmoOnLowAmmoState.AddUObject(this, &UWW3AmmoEquipment::OnAmmoGetsToLowLevelState);AmmoInfoEnteringBackpack->EventOnAmmoOnOutOfLowAmmoState.AddUObject(this, &UWW3AmmoEquipment::OnAmmoGetsOutOfLowLevelState);}}void UWW3AmmoEquipment::UnSubscribeToEventsFromAmmo(UWW3AmmoInfo* AmmoInfoEnteringBackpack){if (AmmoInfoEnteringBackpack != nullptr){AmmoInfoEnteringBackpack->EventOnAmmoOnLowAmmoState.RemoveAll(this);AmmoInfoEnteringBackpack->EventOnAmmoOnOutOfLowAmmoState.RemoveAll(this);}}void UWW3AmmoEquipment::UnsubscribeToEventsFromAmmo(UWW3AmmoInfo* AmmoLeavingBackpack){if (AmmoLeavingBackpack != nullptr){AmmoLeavingBackpack->EventOnAmmoOnLowAmmoState.RemoveAll(this);AmmoLeavingBackpack->EventOnAmmoOnOutOfLowAmmoState.RemoveAll(this);}}void UWW3AmmoEquipment::UnsubscribeEventsFromAllAmmo(){for (int32 i = 0; i < DefaultEquipment.Num(); i++){UnsubscribeToEventsFromAmmo(DefaultEquipment[i]);}}void UWW3AmmoEquipment::OnAmmoGetsToLowLevelState(){if (bAnyAmmoInLowLevelState == false){bAnyAmmoInLowLevelState = true;if (EventOnAnyAmmoInLowLevelState.IsBound()){EventOnAnyAmmoInLowLevelState.Broadcast();}}}void UWW3AmmoEquipment::OnAmmoGetsOutOfLowLevelState(){if (!IsAnyAmmoInLowLevelState()){bAnyAmmoInLowLevelState = false;if (EventOnAllAmmoOutOfLowLevelState.IsBound()){EventOnAllAmmoOutOfLowLevelState.Broadcast();}}}bool UWW3AmmoEquipment::IsAnyAmmoInLowLevelState(){for (int32 i = 0; i < DefaultEquipment.Num(); i++){if (DefaultEquipment[i] != nullptr && DefaultEquipment[i]->IsInLowLevelState()){return true;}}return false;}int32 UWW3AmmoEquipment::GetIndexInEquipment(UWW3AmmoInfo* AmmoInfo) const{int32 IndexInEquipment = -1;if (AmmoInfo != nullptr && AmmoInfo->IsValid()){for (int32 i = 0; i < DefaultEquipment.Num(); i++){if (DefaultEquipment[i] != nullptr && DefaultEquipment[i]->IsValid() && DefaultEquipment[i]->IsEqual(AmmoInfo) ){IndexInEquipment = i;break;}}}return IndexInEquipment;}void UWW3AmmoEquipment::SetNoAmmo(){for (int i = 0; i < DefaultEquipment.Num(); ++i){if (DefaultEquipment[i] != nullptr){DefaultEquipment[i]->SetAmmoAndMax(0,0);}}OnAmmoChanged();}int32 UWW3AmmoEquipment::GetIndexInEquipment(TSubclassOf<UWW3AmmoType> AmmoClass) const{int32 IndexInEquipment = -1;if (AmmoClass != nullptr){for (int32 i = 0; i < DefaultEquipment.Num(); i++){if (DefaultEquipment[i] != nullptr && DefaultEquipment[i]->IsValid() && DefaultEquipment[i]->AmmoType == AmmoClass){IndexInEquipment = i;break;}}}return IndexInEquipment;}bool UWW3AmmoEquipment::IsAsyncLoadingAnyAmmoType() const{if (AdditionalAmmoTypesToAddLoadingHandles.Num() == 0){return false;}for (int32 i = 0; i < AdditionalAmmoTypesToAddLoadingHandles.Num(); ++i){TSharedPtr<FWW3LoadingAssetHandle> SharedHandle = AdditionalAmmoTypesToAddLoadingHandles[i];if (!SharedHandle.IsValid()){continue;}if (!SharedHandle->HasLoadCompleted()){return true;}}return false;}void UWW3AmmoEquipment::Server_Cheat_Ammo_Implementation(){#if CHEAT_ENABLEDfor (int i = 0; i < DefaultEquipment.Num(); ++i){if (DefaultEquipment[i] != nullptr){DefaultEquipment[i]->CheatAmmo();}}OnAmmoChanged();#endif // #if CHEAT_ENABLED}bool UWW3AmmoEquipment::Server_Cheat_Ammo_Validate(){#if CHEAT_ENABLEDreturn true;#else // #if CHEAT_ENABLEDreturn false;#endif // #if CHEAT_ENABLED}float UWW3AmmoEquipment::ValidateAmmoAmmount(float AmountToAdd, float Amount){if (AmountToAdd > Amount){AmountToAdd = Amount;}if (AmountToAdd < 0){return 0;}else if (AmountToAdd < 1){return 1;}else{return AmountToAdd;}}void UWW3AmmoEquipment::OnAmmoChanged(){if (EventOnAmmoChanged.IsBound()){EventOnAmmoChanged.Broadcast(&DefaultEquipment);}}void UWW3AmmoEquipment::OnRep_DefaultEquipment(){OnAmmoChanged();}Mateusz SzlechtMateusz Szlecht22023-12-15T08:29:00Z2023-12-15T08:29:00Z023411424685Microsoft Office Word020557falsefalse28742falsefalse16.0000