SCR_InventoryStorageManagerComponent.c

Go to the documentation of this file.

[EntityEditorProps(category: "GameScripted/UI/Inventory", description: "[MOD] Inventory Slot UI class")]
class SCR_InventoryStorageManagerComponentClass : ScriptedInventoryStorageManagerComponentClass
{
}
 
enum EInventoryRetCode
{
    RETCODE_OK = 0,
    RETCODE_ITEM_TOO_BIG = 2,
    RETCODE_ITEM_TOO_HEAVY = 4,
    RETCODE_DEFAULT_STATE = 0xFFFF
}
 
enum ECallbackState
{
    DROP = 0,
    INSERT = 1,
    MOVE = 2,
    DELETE = 3,
    FINAL = 4
}
 
enum EResupplyUnavailableReason
{
    //~ If multiple reasons for Resupply Unavailable than the Highst enum will be used
    NONE,
    NO_VALID_WEAPON = 10,
    ENOUGH_ITEMS = 20,
    NOT_IN_GIVEN_STORAGE = 30,
    INVENTORY_FULL = 40,
 
    //~ Resupply was valid. Add invalid reasons above
    RESUPPLY_VALID = 99999,
}
 
class SCR_HoldableItemPredicate : InventorySearchPredicate
{
    ECommonItemType wanted;
 
    //------------------------------------------------------------------------------------------------
    // constructor
    void SCR_HoldableItemPredicate()
    {
        QueryAttributeTypes.Insert(SCR_ItemOfInterestAttribute);
    }
    
    //------------------------------------------------------------------------------------------------
    override protected bool IsMatch(BaseInventoryStorageComponent storage, IEntity item, array<GenericComponent> queriedComponents, array<BaseItemAttributeData> queriedAttributes)
    {
        SCR_ItemOfInterestAttribute optionalAttribute = SCR_ItemOfInterestAttribute.Cast(queriedAttributes[0]);
        return optionalAttribute.GetInterestType() == wanted;
    }
}
 
class SCR_BandagePredicate : InventorySearchPredicate
{
    //------------------------------------------------------------------------------------------------
    // constructor
    void SCR_BandagePredicate()
    {
        QueryComponentTypes.Insert(SCR_ConsumableItemComponent);
    }
 
    //------------------------------------------------------------------------------------------------
    override protected bool IsMatch(BaseInventoryStorageComponent storage, IEntity item, array<GenericComponent> queriedComponents, array<BaseItemAttributeData> queriedAttributes)
    {       
        return (SCR_ConsumableItemComponent.Cast(queriedComponents[0])).GetConsumableType() == SCR_EConsumableType.BANDAGE;
    }
}
 
class SCR_ApplicableMedicalItemPredicate : InventorySearchPredicate
{
    IEntity characterEntity;
    ECharacterHitZoneGroup hitZoneGroup;
 
    //------------------------------------------------------------------------------------------------
    // constructor
    void SCR_ApplicableMedicalItemPredicate()
    {
        QueryComponentTypes.Insert(SCR_ConsumableItemComponent);
    }
 
    //------------------------------------------------------------------------------------------------
    override protected bool IsMatch(BaseInventoryStorageComponent storage, IEntity item, array<GenericComponent> queriedComponents, array<BaseItemAttributeData> queriedAttributes)
    {
        SCR_EConsumableType type = SCR_ConsumableItemComponent.Cast(queriedComponents[0]).GetConsumableType();
        bool isMatch = (type == SCR_EConsumableType.BANDAGE)
            || (type == SCR_EConsumableType.HEALTH)
            || (type == SCR_EConsumableType.TOURNIQUET)
            || (type == SCR_EConsumableType.SALINE)
            || (type == SCR_EConsumableType.MORPHINE);
 
        if (!isMatch)
            return false;
 
        SCR_ConsumableItemComponent medicalItem = SCR_ConsumableItemComponent.Cast(item.FindComponent(SCR_ConsumableItemComponent));
        SCR_ConsumableEffectHealthItems effect = SCR_ConsumableEffectHealthItems.Cast(medicalItem.GetConsumableEffect());
        if (!effect)
            return false;
 
        return effect.CanApplyEffectToHZ(characterEntity, characterEntity, hitZoneGroup);
    }
}
 
class SCR_ItemTypeSearchPredicate : InventorySearchPredicate
{
    int m_iItemType = -1;
    IEntity m_iOriginalItem;
    
    //------------------------------------------------------------------------------------------------
    // constructor
    void SCR_ItemTypeSearchPredicate(typename type, int wantedItemType, IEntity originalItem)
    {
        QueryComponentTypes.Insert(type);
        m_iItemType = wantedItemType;
    }
    
    //------------------------------------------------------------------------------------------------
    override protected bool IsMatch(BaseInventoryStorageComponent storage, IEntity item, array<GenericComponent> queriedComponents, array<BaseItemAttributeData> queriedAttributes)
    {
        return (item != m_iOriginalItem) && (SCR_CharacterInventoryStorageComponent.GetItemType(item) == m_iItemType);
    }
}
 
class SCR_CompatibleAttachmentPredicate : InventorySearchPredicate
{
    typename attachmentType;
 
    //------------------------------------------------------------------------------------------------
    // constructor
    void SCR_CompatibleAttachmentPredicate()
    {
        QueryComponentTypes.Insert(InventoryItemComponent);
    }
 
    //------------------------------------------------------------------------------------------------
    override protected bool IsMatch(BaseInventoryStorageComponent storage, IEntity item, array<GenericComponent> queriedComponents, array<BaseItemAttributeData> queriedAttributes)
    {
        InventoryItemComponent itemComp = InventoryItemComponent.Cast(queriedComponents[0]);
 
        if (!itemComp)
            return false;
 
        ItemAttributeCollection itemAttributes = itemComp.GetAttributes();
        if (!itemAttributes)
            return false;
 
        WeaponAttachmentAttributes itemAttribute = WeaponAttachmentAttributes.Cast(itemAttributes.FindAttribute(WeaponAttachmentAttributes));
        if (!itemAttribute)
            return false;
 
        BaseAttachmentType itemAttachmentType = itemAttribute.GetAttachmentType();
        if (!itemAttachmentType)
            return false;
 
        typename itemAttachmentTypename = itemAttachmentType.Type();
        if (!itemAttachmentTypename)
            return false;
 
        return itemAttachmentTypename.IsInherited(attachmentType); // Check if attachment types match
    }
}
 
class SCR_SalinePredicate : InventorySearchPredicate
{
    //------------------------------------------------------------------------------------------------
    // constructor
    void SCR_SalinePredicate()
    {
        QueryComponentTypes.Insert(SCR_ConsumableItemComponent);
    }
    
    //------------------------------------------------------------------------------------------------
    override protected bool IsMatch(BaseInventoryStorageComponent storage, IEntity item, array<GenericComponent> queriedComponents, array<BaseItemAttributeData> queriedAttributes)
    {
        if (storage.Type().IsInherited(EquipmentStorageComponent))
            return true;
        
        return false;
    }
}
 
class SCR_MagazinePredicate : InventorySearchPredicate
{
    typename magWellType;
    
    //------------------------------------------------------------------------------------------------
    // constructor
    void SCR_MagazinePredicate()
    {
        QueryComponentTypes.Insert(BaseMagazineComponent);
    }
    
    //------------------------------------------------------------------------------------------------
    override protected bool IsMatch(BaseInventoryStorageComponent storage, IEntity item, array<GenericComponent> queriedComponents, array<BaseItemAttributeData> queriedAttributes)
    {
        BaseMagazineComponent iMag = BaseMagazineComponent.Cast(queriedComponents[0]);
        if (!iMag)
            return false;
        
        BaseMagazineWell iMagWell = iMag.GetMagazineWell();
        if (!iMagWell)
            return false;
        
        return (iMagWell.IsInherited(magWellType)); // Check if magwells match
    }
}
 
class SCR_PrefabNamePredicate : InventorySearchPredicate
{
    string prefabName;
    
    //------------------------------------------------------------------------------------------------
    override protected bool IsMatch(BaseInventoryStorageComponent storage, IEntity item, array<GenericComponent> queriedComponents, array<BaseItemAttributeData> queriedAttributes)
    {
        EntityPrefabData pd = item.GetPrefabData();
        return pd.GetPrefabName() == this.prefabName;
    }
}
 
class SCR_PrefabDataPredicate : InventorySearchPredicate
{
    EntityPrefabData prefabData;
    
    //------------------------------------------------------------------------------------------------
    override protected bool IsMatch(BaseInventoryStorageComponent storage, IEntity item, array<GenericComponent> queriedComponents, array<BaseItemAttributeData> queriedAttributes)
    {
        return item.GetPrefabData() == this.prefabData;
    }
}
 
class DropAndMoveOperationCallback : ScriptedInventoryOperationCallback
{
    InventoryItemComponent m_ItemBefore;
    InventoryItemComponent m_ItemAfter;
    InventoryStorageSlot m_TargetSlot;
    SCR_InventoryStorageManagerComponent m_Manager;
    SCR_InvCallBack m_FinalCB;
    bool m_bIstakenFromArsenal;
    bool m_bDeleteItemIfEmpty;
    ref array<IEntity> m_aItemsToMove = {};
    ECallbackState m_ECurrentState = 0; // 0 - drop, 1 - insert, 2 - move, 3 - delete, 4 - final
    
    //------------------------------------------------------------------------------------------------
    override protected void OnComplete()
    {
        switch (m_ECurrentState)
        {
            case ECallbackState.DROP:
            {
                OnDropComplete();
            }
            break;
 
            case ECallbackState.INSERT:
            {
                OnInsertComplete();
            }
            break;
 
            case ECallbackState.MOVE:
            {
                OnMoveComplete();
            }
            break;
 
            case ECallbackState.DELETE:
            {
                OnDeleteComplete();
            }
            break;
 
            case ECallbackState.FINAL:
            {
                OnFinalState();
            }
            break;
        }
    } 
 
    //------------------------------------------------------------------------------------------------
    protected void OnDropComplete()
    {
        m_ECurrentState++;
        m_Manager.TryMoveItemToStorage(m_ItemAfter.GetOwner(), m_TargetSlot.GetStorage(), m_TargetSlot.GetID(), this);
    }
    
    //------------------------------------------------------------------------------------------------
    protected void OnInsertComplete()
    {
        if (!m_bIstakenFromArsenal)
        {
            m_ECurrentState = ECallbackState.FINAL;
            OnFinalState();
            return;
        } 
        
        m_ECurrentState++;
        //BaseInventoryStorageComponent itemIsStorage = BaseInventoryStorageComponent.Cast(m_ItemBefore);
        //m_Manager.GetAllItems(m_aItemsToMove, itemIsStorage);
        OnMoveComplete();
    }
    
    //------------------------------------------------------------------------------------------------
    protected void OnMoveComplete()
    {
        // Temporarily disabled for performance reasons. Once it gets fixed the transfer of items can be reenabled.
//      if (m_aItemsToMove.Count() == 0)
//      {
//          m_ECurrentState++;
//          OnDeleteComplete();
//          return;
//      }
//      IEntity item = m_aItemsToMove[m_aItemsToMove.Count() - 1];
//      m_aItemsToMove.Resize(m_aItemsToMove.Count() - 1);
//      BaseInventoryStorageComponent storage = m_Manager.FindStorageForItem(item, EStoragePurpose.PURPOSE_ANY);
//      if (!m_Manager.TryMoveItemToStorage(item, storage, -1, this))
//          OnMoveComplete();
        
        if (m_bDeleteItemIfEmpty)
        {
            m_ECurrentState++;
            OnDeleteComplete();
        }
        else
        {       
            m_ECurrentState = ECallbackState.FINAL;
            OnFinalState();
        }
            
        return;
    }
    
    //------------------------------------------------------------------------------------------------
    protected void OnDeleteComplete()
    {           
        m_aItemsToMove.Clear();
        BaseInventoryStorageComponent itemIsStorage = BaseInventoryStorageComponent.Cast(m_ItemBefore);
        m_Manager.GetAllItems(m_aItemsToMove, itemIsStorage);
            
        if (m_aItemsToMove.IsEmpty())
            m_Manager.AskServerToDeleteEntity(m_ItemBefore.GetOwner());
        
        OnFinalState();
    }
    
    //------------------------------------------------------------------------------------------------
    protected void OnFinalState()   
    {
        if (m_FinalCB)
            m_FinalCB.InternalComplete();
    }
}
 
class SCR_ResupplyMagazinesCallback : ScriptedInventoryOperationCallback
{
    protected SCR_InventoryStorageManagerComponent m_Manager;
    protected ref map<ResourceName, int> m_MagazinesToSpawn = new map<ResourceName, int>();
    
    //------------------------------------------------------------------------------------------------
    void Insert(ResourceName prefab, int count)
    {
        int currentCount;
        m_MagazinesToSpawn.Find(prefab, currentCount);
        m_MagazinesToSpawn.Insert(prefab, currentCount + count);
    }
 
    //------------------------------------------------------------------------------------------------
    void Start()
    {
        OnComplete();
    }
    
    //------------------------------------------------------------------------------------------------
    override protected void OnComplete()
    {
        if (!m_Manager)
            return;
        
        if (!m_MagazinesToSpawn.IsEmpty())
        {
            //--- Next item to process
            ResourceName prefab = m_MagazinesToSpawn.GetKey(0);
            int count = m_MagazinesToSpawn.GetElement(0);
            count--;
            
            if (count == 0)
                m_MagazinesToSpawn.Remove(prefab);
            else
                m_MagazinesToSpawn.Set(prefab, count);
            
            m_Manager.TrySpawnPrefabToStorage(prefab, cb: this);
        }
        else
        {
            //--- All processed
            OnFailed();
        }
    }
 
    //------------------------------------------------------------------------------------------------
    override protected void OnFailed()
    {
        //--- Delete itself (after a delay - can't delete itself at this frame)
        GetGame().GetCallqueue().CallLater(m_Manager.EndResupplyMagazines, 1, false);
    }
 
    //------------------------------------------------------------------------------------------------
    // constructor
    void SCR_ResupplyMagazinesCallback(SCR_InventoryStorageManagerComponent manager)
    {
        m_Manager = manager;
    }
}
 
class SCR_InventoryStorageManagerComponent : ScriptedInventoryStorageManagerComponent
{
    private SCR_CharacterInventoryStorageComponent              m_Storage;
    private SCR_CharacterControllerComponent                    m_CharacterController;
    private ref SCR_BandagePredicate                            m_BandagePredicate = new SCR_BandagePredicate();
    protected EInventoryRetCode                                 m_ERetCode;
    protected int                                               m_iHealthEquipment  =   0;
    protected bool                                              m_bIsInventoryLocked = false;
    protected ref SCR_WeaponSwitchingBaseUI                     m_pWeaponSwitchingUI;
    private bool                                                m_bWasRaised;
    private IEntity                                             m_StorageToOpen;
    protected ref SCR_ResupplyMagazinesCallback                     m_ResupplyMagazineCallback;
    
    ref ScriptInvokerBool                                   m_OnInventoryOpenInvoker    = new ScriptInvokerBool();
    ref ScriptInvokerBool                                   m_OnQuickBarOpenInvoker     = new ScriptInvokerBool();
 
    //------------------------------------------------------------------------------------------------
    int GetAllRootItems(out notnull array<IEntity> rootItems)
    {
        rootItems.Clear();
        array<BaseInventoryStorageComponent> storages = {};
        
        //~ Get deposits like backpacks and jackets as well as any held weapons
        GetStorages(storages, EStoragePurpose.PURPOSE_DEPOSIT);
        GetStorages(storages, EStoragePurpose.PURPOSE_WEAPON_PROXY);
        
        array<IEntity> items = {};
        array<BaseInventoryStorageComponent> clothStorages;
        foreach (BaseInventoryStorageComponent storage : storages)
        {
            //~ If backpack or jacked or any other cloth storage only get what is inside the storage
            if (ClothNodeStorageComponent.Cast(storage))
            {
                if (!storage)
                    continue;
                
                clothStorages = {};
                storage.GetOwnedStorages(clothStorages, 1, false);
            
                foreach (BaseInventoryStorageComponent clothStorage : clothStorages)
                {
                    if (!clothStorage)
                        continue;
                    
                    clothStorage.GetAll(items);
                    rootItems.Copy(items);
                }
                
                continue;
            }
            else 
            {
                storage.GetAll(items);
                rootItems.Copy(items);
            }
        }
        
        return rootItems.Count();
    }
 
    //------------------------------------------------------------------------------------------------
    // Callback when item is added (will be performed locally after server completed the Insert/Move operation)
    override protected void OnItemAdded(BaseInventoryStorageComponent storageOwner, IEntity item)
    {       
        super.OnItemAdded(storageOwner, item);
 
        SCR_ConsumableItemComponent consumable = SCR_ConsumableItemComponent.Cast(item.FindComponent(SCR_ConsumableItemComponent));
        if ( consumable && consumable.GetConsumableType() == SCR_EConsumableType.BANDAGE )
            m_iHealthEquipment++;   //store count of the health components
    }
 
    //------------------------------------------------------------------------------------------------
    // Callback when item is removed (will be performed locally after server completed the Remove/Move operation)
    override protected void OnItemRemoved(BaseInventoryStorageComponent storageOwner, IEntity item)
    {
        super.OnItemRemoved(storageOwner, item);
 
        SCR_ConsumableItemComponent consumable = SCR_ConsumableItemComponent.Cast(item.FindComponent(SCR_ConsumableItemComponent));
        if ( consumable && consumable.GetConsumableType() == SCR_EConsumableType.BANDAGE )
            m_iHealthEquipment--;   //store count of the health components
    }
 
    //------------------------------------------------------------------------------------------------
    override protected bool ShouldForbidRemoveByInstigator(InventoryStorageManagerComponent instigatorManager, BaseInventoryStorageComponent fromStorage, IEntity item)
    {
        //in case of health items, permit medics to donate healing items to targets
        SCR_ConsumableItemComponent consumableItemComp = SCR_ConsumableItemComponent.Cast(item.FindComponent(SCR_ConsumableItemComponent));
        if (consumableItemComp)
            return false;
        
        if (m_CharacterController && m_CharacterController.GetLifeState() == ECharacterLifeState.ALIVE)
            return true;
        
        return false;
    }
    
    //------------------------------------------------------------------------------------------------
    void PlayItemSound(IEntity entity, string soundEvent)
    {   
        if (!entity)
            return;
 
        RplComponent rplComp = RplComponent.Cast(entity.FindComponent(RplComponent));
        if (rplComp)
        {
            Rpc(RpcAsk_PlaySound, rplComp.Id(), soundEvent);
        }
        else
        {
            SoundComponent soundComp = SoundComponent.Cast(entity.FindComponent(SoundComponent));
            if (soundComp)
            {
                soundComp.SoundEvent(soundEvent);
            }
            else
            {
                SCR_SoundManagerEntity soundManagerEntity = GetGame().GetSoundManagerEntity();
                if (!soundManagerEntity)
                    return;
            
                soundManagerEntity.CreateAndPlayAudioSource(entity, soundEvent);    
            }
        }
    }
    
    //------------------------------------------------------------------------------------------------
    [RplRpc(RplChannel.Reliable, RplRcver.Server)]
    void RpcAsk_PlaySound(RplId targetRplId, string soundAction)
    {
        Rpc(RpcDo_PlaySound, targetRplId, soundAction);
        RpcDo_PlaySound(targetRplId, soundAction);
    }
    
    //------------------------------------------------------------------------------------------------
    [RplRpc(RplChannel.Reliable, RplRcver.Broadcast)]
    void RpcDo_PlaySound(RplId targetRplId, string soundAction)
    {
        RplComponent rplComp = RplComponent.Cast(Replication.FindItem(targetRplId));
        if (!rplComp)
            return;
        
        IEntity entity = rplComp.GetEntity();
        if (!entity)
            return;
        
        SoundComponent soundComp = SoundComponent.Cast(entity.FindComponent(SoundComponent));
        if (soundComp)
        {
            soundComp.SoundEvent(soundAction);
        }
        else
        {
            SCR_SoundManagerEntity soundManagerEntity = GetGame().GetSoundManagerEntity();
            if (!soundManagerEntity)
                return;
            
            soundManagerEntity.CreateAndPlayAudioSource(entity, soundAction);   
        }
    }
    
#ifndef DISABLE_INVENTORY
 
    //------------------------------------------------------------------------------------------------
    SCR_CharacterInventoryStorageComponent GetCharacterStorage()
    {
        return m_Storage;
    }
    
    //------------------------------------------------------------------------------------------------
    // TODO: make this method as native (cannot override the proto native CanMoveItemToStorage)
    bool CanInsertItemInActualStorage(IEntity item, BaseInventoryStorageComponent storage, int slotID = -1)
    {
        if (!IsAnimationReady() || IsInventoryLocked())
            return false;
 
        array<BaseInventoryStorageComponent> pStorages = {};
        storage.GetOwnedStorages( pStorages, 1, false );
        pStorages.Insert( storage );
 
        foreach ( BaseInventoryStorageComponent pStorage : pStorages )
        {   
            bool bCanInsert = CanInsertItemInStorage( item, pStorage, -1 ); //split because of debug purposes
            bool bCanMove = CanMoveItemToStorage( item, pStorage, -1 );
            if ( bCanInsert || bCanMove )
                return true;
        }
        return false;
    }
    
    //------------------------------------------------------------------------------------------------
    // ! The return code informs about the state of the operation ( i.e. cannot insert item, since it is too large )
    // ! it clear the flag
    void SetReturnCode( EInventoryRetCode ERetCode ) 
    { 
        m_ERetCode &= ~ERetCode; 
    }
    
    //------------------------------------------------------------------------------------------------
    // ! The return code informs about the state of the operation ( i.e. cannot insert item, since it is too large )
    // ! it clear the flag
    void SetReturnCodeDefault() 
    { 
        m_ERetCode = EInventoryRetCode.RETCODE_DEFAULT_STATE;
    }
    
    //------------------------------------------------------------------------------------------------
    EInventoryRetCode GetReturnCode()
    {
        return m_ERetCode;
    }
    
    //------------------------------------------------------------------------------------------------
    float GetTotalWeightOfAllStorages()
    {
        array<BaseInventoryStorageComponent> storages = {};
        float fTotalWeight = 0.0;
 
        //TODO: actually not a very good way how to get storages, but using the GetStorages() method causes the weight being doubled. We need to get just the "parent" storages
        storages.Insert(m_Storage.GetWeaponStorage());
        storages.Insert(m_Storage);
                
        foreach (BaseInventoryStorageComponent storage : storages)
        {
            fTotalWeight += storage.GetTotalWeight();
        }
 
        return fTotalWeight;
    }
    
    //------------------------------------------------------------------------------------------------
    void SetLootStorage( IEntity pOwner )
    {
        if (m_Storage)
            m_Storage.SetLootStorage(pOwner);
    }
    
    //------------------------------------------------------------------------------------------------
    void InsertItem( IEntity pItem, BaseInventoryStorageComponent pStorageTo = null, BaseInventoryStorageComponent pStorageFrom = null, SCR_InvCallBack cb = null  )
    {
        if (!pItem || !IsAnimationReady() || IsInventoryLocked())
            return;
        
        SetInventoryLocked(true);
 
        bool canInsert = true;
        if ( !pStorageTo ) // no storage selected, put it into best fitting storage
        {
            string soundEvent = SCR_SoundEvent.SOUND_EQUIP;
            //TryInsertItem( pItem, EStoragePurpose.PURPOSE_DEPOSIT);   // works for the owned storages ( not for the vicinity storages )
            if ( !TryInsertItem( pItem, EStoragePurpose.PURPOSE_WEAPON_PROXY, cb ) )
            {
                if ( !TryInsertItem( pItem, EStoragePurpose.PURPOSE_DEPOSIT, cb ) )
                {
                    if ( !TryMoveItemToStorage( pItem, FindStorageForItem( pItem, EStoragePurpose.PURPOSE_ANY ), -1, cb ) )
                        canInsert = TryMoveItemToStorage(pItem, m_Storage, -1, cb);                 // clothes from storage in vicinity
                    else
                        soundEvent = SCR_SoundEvent.SOUND_PICK_UP;  // play pick up sound for everything else
                
                    SCR_UISoundEntity.SoundEvent(SCR_SoundEvent.SOUND_INV_HOTKEY_CONFIRM);
                }
                else
                    soundEvent = SCR_SoundEvent.SOUND_PICK_UP;
            }
            
            if (canInsert)
                PlayItemSound(pItem, soundEvent);
        }
        else
        {
            if (pStorageTo == m_Storage)
            {
                canInsert = TryReplaceItem( pStorageTo, pItem, 0, cb );
                if (canInsert)
                {
                    SetInventoryLocked(false);
                    return;
                }
            }
            
            //~ Find a valid storage to insert item in
            BaseInventoryStorageComponent validStorage = FindStorageForInsert( pItem, pStorageTo, EStoragePurpose.PURPOSE_ANY );
            if (validStorage)
            {
                pStorageTo = validStorage;
            }
            //~ Check if item can be inserted in linked storages
            else 
            {
                //~ Find valid storage in linked storages
                SCR_UniversalInventoryStorageComponent universalStorage = SCR_UniversalInventoryStorageComponent.Cast(pStorageTo);
                if (universalStorage)
                {
                    array<BaseInventoryStorageComponent> linkedStorages = {};
                    universalStorage.GetLinkedStorages(linkedStorages);
                    
                    foreach(BaseInventoryStorageComponent linkedStorage : linkedStorages)
                    {
                        //~ Valid linked storage found
                        if (FindStorageForInsert(pItem, linkedStorage, EStoragePurpose.PURPOSE_ANY))
                        {
                            pStorageTo = linkedStorage;
                            break;
                        }
                    }
                }
            }
            
            if ( !pStorageFrom )
                canInsert = TryInsertItemInStorage( pItem, pStorageTo, -1, cb );    // if we move item from ground to opened storage
            else
                canInsert = TryMoveItemToStorage( pItem, pStorageTo, -1, cb );      // if we move item between storages
        }
        
        if (!canInsert)
            SCR_UISoundEntity.SoundEvent(SCR_SoundEvent.SOUND_INV_DROP_ERROR);
        else
            SCR_UISoundEntity.SoundEvent(SCR_SoundEvent.SOUND_INV_CONTAINER_DIFR_DROP);
        
        if (m_CharacterController && canInsert && !pStorageFrom)
            m_CharacterController.TryPlayItemGesture(EItemGesture.EItemGesturePickUp);
 
        SetInventoryLocked(false);
    }
    
    //------------------------------------------------------------------------------------------------
    bool TryRemoveItemFromInventory(IEntity pItem, BaseInventoryStorageComponent storage = null, InventoryOperationCallback cb = null)
    {
        if (!CanMoveItem(pItem))
            return false;
 
        if (!storage)
        {
            InventoryItemComponent itemComp = InventoryItemComponent.Cast(pItem.FindComponent(InventoryItemComponent));
            if (!itemComp)
                return false;
            
            InventoryStorageSlot parentSlot = itemComp.GetParentSlot();
            if (parentSlot)
                storage = parentSlot.GetStorage();
        }
        
        SetInventoryLocked(true);
        bool result = TryRemoveItemFromStorage(pItem, storage, cb);
        SetInventoryLocked(false);
 
        return result;
    }
    
    //------------------------------------------------------------------------------------------------
    bool CanMoveItem(IEntity item)
    {
        if (!item || item.IsDeleted() || !IsAnimationReady() || IsInventoryLocked())
            return false;
        
        return true;
    }
    
//  //------------------------------------------------------------------------------------------------
//  override bool TryInsertItemInStorageScr( IEntity pItem, BaseInventoryStorageComponent pStorageTo, int slotID = -1, InventoryOperationCallback cb = null )
//  {
//
//      if ( !pStorageTo )
//          return;
//
//      array<BaseInventoryStorageComponent> pStorages = {};
//      pStorage.GetOwnedStorages( pStorages, 1, false );   // get all the storages, the storage has attached to it
//      pStorages.Insert( pStorage );                       // and put there also the storage
//      foreach ( BaseInventoryStorageComponent tmpStorage : pStorages )
//      {
//          if ( MoveOperation( itemComponent, tmpStorage ) )
//          {
//              bRet = true;
//              break;
//          }
//      }
//  }
    
    //------------------------------------------------------------------------------------------------
    bool TrySwapItems( IEntity pOwnerEntity, BaseInventoryStorageComponent pStorageTo, SCR_InvCallBack cb = null )
    {
        if ( !pStorageTo )
            return false;
                        
        InventoryStorageSlot slot =  pStorageTo.FindSuitableSlotForItem( pOwnerEntity );
        if ( !slot )
            return false;
 
        if ( slot.GetAttachedEntity() )
        {
            if (!TrySwapItemStorages( pOwnerEntity, slot.GetAttachedEntity(), cb ))
            {
                CharacterHandWeaponSlotComponent handWeaponSlot = CharacterHandWeaponSlotComponent.Cast(slot.GetParentContainer());
                //Move 
                if (handWeaponSlot)
                {
                    if (TryRemoveItemFromInventory(slot.GetAttachedEntity()))
                    {
                        TryInsertItem(pOwnerEntity, EStoragePurpose.PURPOSE_ANY, cb);
                        return true;
                    }
                }
                
                return false;
            }
            
            return true;
        }
        else
        {
            return TryMoveItemToStorage( pOwnerEntity, pStorageTo, slot.GetID(), cb );
        }
 
        return false;
    }
    
    //------------------------------------------------------------------------------------------------
    void EquipWeapon( IEntity pOwnerEntity, SCR_InvCallBack cb = null, bool bFromVicinity = true )
    {
        if ( !bFromVicinity )
        {
            if (!TrySwapItems(pOwnerEntity, m_Storage.GetWeaponStorage(), cb))
                SCR_UISoundEntity.SoundEvent(SCR_SoundEvent.SOUND_INV_DROP_ERROR);
            else
                SCR_UISoundEntity.SoundEvent(SCR_SoundEvent.SOUND_INV_CONTAINER_DIFR_DROP);
            
            return;
        }
        
        IEntity user = GetOwner();
        if (!user)
            return;
        
        BaseWeaponManagerComponent weaponManager = BaseWeaponManagerComponent.Cast(user.FindComponent(BaseWeaponManagerComponent));
        if (!weaponManager)
            return;
    
        WeaponSlotComponent slot = weaponManager.GetCurrentSlot();
        int preferred = 0;
        if ( slot )
            preferred = slot.GetWeaponSlotIndex();
        
        if (!EquipAny(m_Storage.GetWeaponStorage(), pOwnerEntity, preferred, cb))
            SCR_UISoundEntity.SoundEvent(SCR_SoundEvent.SOUND_INV_DROP_ERROR);
        
        if (cb && cb.m_pStorageFrom != cb.m_pStorageTo)
                SCR_UISoundEntity.SoundEvent(SCR_SoundEvent.SOUND_INV_CONTAINER_DIFR_DROP);
            else
                SCR_UISoundEntity.SoundEvent(SCR_SoundEvent.SOUND_INV_CONTAINER_SAME_DROP);
        
        // Play sound
        PlayItemSound(pOwnerEntity, SCR_SoundEvent.SOUND_EQUIP);
    }
 
    //------------------------------------------------------------------------------------------------
    void EquipWeaponAttachment( IEntity pOwnerEntity, IEntity pUserEntity, SCR_InvCallBack cb = null )
    {
        BaseWeaponManagerComponent weaponManager = BaseWeaponManagerComponent.Cast(pUserEntity.FindComponent(BaseWeaponManagerComponent));
        if (!weaponManager)
            return;
 
        WeaponSlotComponent slot = weaponManager.GetCurrentSlot();
        if (!slot)
            return;
 
        IEntity weaponEntity = slot.GetWeaponEntity();
        if (!weaponEntity)
            return;
 
        BaseInventoryStorageComponent storage = BaseInventoryStorageComponent.Cast(weaponEntity.FindComponent(BaseInventoryStorageComponent));
        if (!storage)
            return;
        
        EquipAny(storage, pOwnerEntity, -1, cb );
        
        // Play sound
        PlayItemSound(pOwnerEntity, SCR_SoundEvent.SOUND_EQUIP);
    }
    
    //------------------------------------------------------------------------------------------------
    void EquipGadget( IEntity pOwnerEntity, SCR_InvCallBack cb = null )
    {
        //(kamil) the gadget slots are now present directly on individual clothing items - will have to revise logic here if swapping is wanted
        BaseInventoryStorageComponent storageComp = FindStorageForItem(pOwnerEntity, EStoragePurpose.PURPOSE_EQUIPMENT_ATTACHMENT);
        if (storageComp)
        {
            EquipAny( storageComp, pOwnerEntity, -1, cb );
            
            // Play sound
            PlayItemSound(pOwnerEntity, SCR_SoundEvent.SOUND_EQUIP);
        }
    }
    
    //------------------------------------------------------------------------------------------------
    void EquipCloth( IEntity pOwnerEntity )
    {
        // m_pStorage because character storage is inherited from SCR_EquipedLoadoutStorageComponent
        EquipAny(m_Storage, pOwnerEntity);
        
        // Play sound
        PlayItemSound(pOwnerEntity, SCR_SoundEvent.SOUND_EQUIP);
    }
    
    //------------------------------------------------------------------------------------------------
    bool EquipAny(BaseInventoryStorageComponent storage, IEntity item, int preferred = -1, SCR_InvCallBack cb = null)
    {
        if (!storage || !item)
            return false;
        InventoryItemComponent itemComp = InventoryItemComponent.Cast(item.FindComponent(InventoryItemComponent));
        if (!itemComp)
            return false;
        
        InventoryStorageSlot m_TargetSlot = storage.FindSuitableSlotForItem(item);
        // Storage doesn't have suitable slot for item (therefore any future opearation would fail)
        if (!m_TargetSlot)
            return false;
 
        if (preferred > 0 && preferred < storage.GetSlotsCount())
            m_TargetSlot = storage.GetSlot(preferred);
        else
            preferred = m_TargetSlot.GetID();
 
        InventoryStorageSlot sourceSlot = itemComp.GetParentSlot();
        // Item is on the ground as it does not belong to any storage (eg is not in the slot)
        if (!sourceSlot || !sourceSlot.GetStorage())
        {
            // we are picking up item from ground
            // if target slot is not empty return the result of replace operation
            if (m_TargetSlot.GetAttachedEntity())
                return TryReplaceItem(item, storage, preferred, cb);
 
            // we are picking up item from ground into empty slot, simply return result of the insert operation
            return TryInsertItemInStorage(item, storage, preferred, cb);
        }
 
        // our target slot is empty and we moving item from another storage
        // simply return the result of move operation
        if (!m_TargetSlot.GetAttachedEntity())
            return TryMoveItemToStorage(item, storage, preferred, cb);
 
        BaseInventoryStorageComponent sourceStorage = sourceSlot.GetStorage();
        bool isTakenFromArsenal = sourceStorage.GetOwner().FindComponent(SCR_ArsenalComponent);
        bool isTakenFromBody = false;
        
        ChimeraCharacter lootedBodyCharacter = ChimeraCharacter.Cast(sourceStorage.GetOwner());
        DamageManagerComponent lootedBodyDamageManager;
        
        if (lootedBodyCharacter)
            lootedBodyDamageManager = lootedBodyCharacter.GetDamageManager();
        
        if (lootedBodyDamageManager)
            isTakenFromBody = lootedBodyDamageManager.GetState() == EDamageState.DESTROYED;
        
        bool performDropOfOriginalItem = isTakenFromBody || isTakenFromArsenal || sourceStorage.GetOwner().FindComponent(SCR_CampaignArmoryStorageComponent) ;
        
        // If we don't want to drop item
        if (performDropOfOriginalItem)
        {
            // if we want to drop originally equipped item
            // here sequence would be as follows:
            // 1 - drop original item
            // 2 - insert item to target storage
            // 3 - try move as many items as possible from dropped item back to inventory
            // 4 - delete dropped item
    
            // At first - let's validate if this is even possible
            if (!CanSwapItemStorages(item, m_TargetSlot.GetAttachedEntity()))
                return false;
    
            DropAndMoveOperationCallback chainedCallback = new DropAndMoveOperationCallback();
            chainedCallback.m_Manager = this;
            chainedCallback.m_ItemAfter = itemComp;
            chainedCallback.m_ItemBefore = InventoryItemComponent.Cast(m_TargetSlot.GetAttachedEntity().FindComponent(InventoryItemComponent));
            chainedCallback.m_TargetSlot = m_TargetSlot;
            chainedCallback.m_FinalCB = cb;
            chainedCallback.m_bDeleteItemIfEmpty = true;
            chainedCallback.m_bIstakenFromArsenal = isTakenFromArsenal;
    
            return TryRemoveItemFromStorage(m_TargetSlot.GetAttachedEntity(), m_TargetSlot.GetStorage(), chainedCallback);
        }
        
        // we return the result of swap opoeration were item from slotA will be transfered to slotB and item from slotB to slotA
        return TrySwapItemStorages(item, m_TargetSlot.GetAttachedEntity(), cb);
    } 
    
    //------------------------------------------------------------------------------------------------
    bool TryReplaceAndDropItemAtSlot(BaseInventoryStorageComponent storage, IEntity item, int slotID, SCR_InvCallBack cb = null, bool isTakenFromArsenal = false, bool deleteOriginalItemIfEmpty = false)
    {
        if (!storage || !item)
            return false;
        InventoryItemComponent itemComp = InventoryItemComponent.Cast(item.FindComponent(InventoryItemComponent));
        if (!itemComp)
            return false;
        
        InventoryStorageSlot m_TargetSlot = storage.GetSlot(slotID);
        // Storage doesn't have suitable slot for item (therefore any future opearation would fail)
        if (!m_TargetSlot)
            return false;       
 
        InventoryStorageSlot sourceSlot = itemComp.GetParentSlot();
        // Item is on the ground as it does not belong to any storage (eg is not in the slot)
        if (!sourceSlot || !sourceSlot.GetStorage())
        {
            // we are picking up item from ground
            // if target slot is not empty return the result of replace operation
            if (m_TargetSlot.GetAttachedEntity())
                return TryReplaceItem(item, storage, slotID, cb);
            // we are picking up item from ground into empty slot, simply return result of the insert operation
            return TryInsertItemInStorage(item, storage, slotID, cb);
        }
 
        // our target slot is empty and we moving item from another storage
        // simply return the result of move operation
        if (!m_TargetSlot.GetAttachedEntity())
            return TryMoveItemToStorage(item, storage, slotID, cb);
 
        // if we want to drop originally equipped item
        // here sequence would be as follows:
        // 1 - drop original item
        // 2 - insert item to target storage
        // 3 - try move as many items as possible from dropped item back to inventory
        // 4 - delete dropped item
 
        // At first - let's validate if this is even possible
        if (!CanSwapItemStorages(item, m_TargetSlot.GetAttachedEntity()))
            return false;
 
        DropAndMoveOperationCallback chainedCallback = new DropAndMoveOperationCallback();
        chainedCallback.m_Manager = this;
        chainedCallback.m_ItemAfter = itemComp;
        chainedCallback.m_ItemBefore = InventoryItemComponent.Cast(m_TargetSlot.GetAttachedEntity().FindComponent(InventoryItemComponent));
        chainedCallback.m_TargetSlot = m_TargetSlot;
        chainedCallback.m_bDeleteItemIfEmpty = deleteOriginalItemIfEmpty; 
        chainedCallback.m_bIstakenFromArsenal = isTakenFromArsenal;
        chainedCallback.m_FinalCB = cb;
        
        return TryRemoveItemFromStorage(m_TargetSlot.GetAttachedEntity(), m_TargetSlot.GetStorage(), chainedCallback);
    }
    
    //------------------------------------------------------------------------------------------------
    bool TryReplaceItem( BaseInventoryStorageComponent storage, IEntity item, int preferred, SCR_InvCallBack cb )
    {
        int slotCount = storage.GetSlotsCount();
        
        for ( int i = 0; i < slotCount; i++ )
        {
            int j = ( i + preferred ) % slotCount;
            if ( CanReplaceItem( item, storage, j ) )
            {
                if ( TryReplaceItem( item, storage, j, cb ) )
                    return true;
            }
        }
 
        return false;
    } 
    
    //------------------------------------------------------------------------------------------------
    void EquipItem( EquipedWeaponStorageComponent weaponStorage, IEntity weapon )
    {
        // There are empty suitable slots at weapon storage
        if ( CanInsertItemInStorage( weapon, weaponStorage, -1) )
        {
            TryInsertItemInStorage( weapon, weaponStorage, -1 );
            return;
        }
 
        // Otherwise try replace weapon at suitable slot
        BaseWeaponManagerComponent weaponManager = BaseWeaponManagerComponent.Cast( GetOwner().FindComponent(BaseWeaponManagerComponent) );
        if (!weaponManager)
            return;
 
        WeaponSlotComponent slot = weaponManager.GetCurrentSlot();
        int slotCount = weaponStorage.GetSlotsCount();
        int preferred = 0;
        if ( slot )
            preferred = slot.GetWeaponSlotIndex();
 
        for ( int i = 0; i < slotCount; i++ )
        {
            int j = ( i + preferred ) % slotCount;
            if ( CanReplaceItem( weapon, weaponStorage, j ) )
            {
                TryReplaceItem( weapon, weaponStorage, j );
                return;
            }
        }
    }
    
    //------------------------------------------------------------------------------------------------
    bool IsResupplyMagazinesAvailable(int resupplyMagazineCount = 4, out EResupplyUnavailableReason resupplyUnavailableReason = EResupplyUnavailableReason.NONE, EMuzzleType muzzleType = -1, InventoryStorageManagerComponent mustBeInStorage = null)
    {
        BaseWeaponManagerComponent weaponsManager = BaseWeaponManagerComponent.Cast(GetOwner().FindComponent(BaseWeaponManagerComponent));
        if (!weaponsManager)
            return false;
 
        array<IEntity> weaponList = {};
        weaponsManager.GetWeaponsList(weaponList);
 
        bool foundValidWeapon = false;
        ResourceName magazineOrProjectilePrefab;
        IEntity spawnedMagazine;
 
        BaseWeaponComponent comp;
        array<BaseMuzzleComponent> muzzles;
        SCR_MuzzleInMagComponent inMagMuzzle;
        SCR_ArsenalInventoryStorageManagerComponent arsenalStorage;
        foreach (IEntity weapon : weaponList)
        {
            comp = BaseWeaponComponent.Cast(weapon.FindComponent(BaseWeaponComponent));
            if (!comp)
                continue;
 
            string weaponSlotType = comp.GetWeaponSlotType();
 
            // Only refill primary and secondary weapons
            if ((weaponSlotType != "primary" && weaponSlotType != "secondary"))
                continue;
 
            muzzles = {};
 
            //~ Get base muzzle to only supply magazines
            comp.GetMuzzlesList(muzzles);
            foreach (BaseMuzzleComponent muzzle : muzzles)
            {
                if (muzzleType != -1 && muzzle.GetMuzzleType() != muzzleType)
                    continue;
 
                inMagMuzzle = SCR_MuzzleInMagComponent.Cast(muzzle);
                if (inMagMuzzle && !inMagMuzzle.CanBeReloaded())
                    continue;
 
                magazineOrProjectilePrefab = muzzle.GetDefaultMagazineOrProjectileName();
                if (SCR_StringHelper.IsEmptyOrWhiteSpace(magazineOrProjectilePrefab))
                    continue;
 
                //~ At least one valid weapon was found
                foundValidWeapon = true;
 
                //~ If storage is given check if magazine or projectile is in storage and only allow resupply if it is (Does not care for amount and intended use is with Arsenal)
                if (mustBeInStorage)
                {
                    arsenalStorage = SCR_ArsenalInventoryStorageManagerComponent.Cast(mustBeInStorage);
                    if ((arsenalStorage && !arsenalStorage.IsPrefabInArsenalStorage(magazineOrProjectilePrefab)) || (!arsenalStorage && mustBeInStorage.GetDepositItemCountByResource(magazineOrProjectilePrefab) < 1))
                    {
                        if (resupplyUnavailableReason < EResupplyUnavailableReason.NOT_IN_GIVEN_STORAGE)
                            resupplyUnavailableReason = EResupplyUnavailableReason.NOT_IN_GIVEN_STORAGE;
 
                        continue;
                    }
                }
 
                //~ Check if there are already enough magazines
                if (resupplyMagazineCount - GetMagazineCountByMuzzle(muzzle) <= 0)
                {
                    if (resupplyUnavailableReason < EResupplyUnavailableReason.ENOUGH_ITEMS)
                        resupplyUnavailableReason = EResupplyUnavailableReason.ENOUGH_ITEMS;
 
                    continue;
                }
                
                //~ If it can be stored
                if (!FindStorageForResourceInsert(magazineOrProjectilePrefab, m_Storage))
                {
                    if (resupplyUnavailableReason < EResupplyUnavailableReason.INVENTORY_FULL)
                        resupplyUnavailableReason = EResupplyUnavailableReason.INVENTORY_FULL;
                    
                    continue;
                }
 
                //~ Passes all the checks
                //~ Set reason none just in case as resupply is available
                resupplyUnavailableReason = EResupplyUnavailableReason.NONE;
 
                //~ All checks passed and at least one magazine can be added so function returns true
                return true;
            }
        }
 
        //~ Did not have any valid weapons so set unavailable reason to no valid weapon
        if (!foundValidWeapon)
            resupplyUnavailableReason = EResupplyUnavailableReason.NO_VALID_WEAPON;
 
        return false;
    }
    
    //------------------------------------------------------------------------------------------------
    void GetValidResupplyItemsAndCount(out notnull map<ResourceName, int> validResupplyItems, int maxMagazineCount = 4, EMuzzleType muzzleType = -1, InventoryStorageManagerComponent mustBeInStorage = null)
    {
        validResupplyItems.Clear();
 
        BaseWeaponManagerComponent weaponsManager = BaseWeaponManagerComponent.Cast(GetOwner().FindComponent(BaseWeaponManagerComponent));
        if (!weaponsManager)
            return;
        
        array<IEntity> weaponList = {};
        weaponsManager.GetWeaponsList(weaponList);
 
        BaseWeaponComponent comp;
        array<BaseMuzzleComponent> muzzles;
        SCR_MuzzleInMagComponent inMagMuzzle;
        SCR_ArsenalInventoryStorageManagerComponent arsenalStorage;
        foreach (IEntity weapon : weaponList)
        {
            comp = BaseWeaponComponent.Cast(weapon.FindComponent(BaseWeaponComponent));
            string weaponSlotType = comp.GetWeaponSlotType();
 
            // Only refill primary and secondary weapons
            if (!(weaponSlotType == "primary" || weaponSlotType == "secondary"))
                continue;
 
            muzzles = {};
 
            //~ Get base muzzle to only supply magazines
            comp.GetMuzzlesList(muzzles);
            foreach (BaseMuzzleComponent muzzle : muzzles)
            {
                if (muzzleType != -1 && muzzle.GetMuzzleType() != muzzleType)
                    continue;
 
                inMagMuzzle = SCR_MuzzleInMagComponent.Cast(muzzle);
                if (inMagMuzzle && !inMagMuzzle.CanBeReloaded())
                    continue;
 
                ResourceName magazineOrProjectilePrefab = muzzle.GetDefaultMagazineOrProjectileName();
                if (SCR_StringHelper.IsEmptyOrWhiteSpace(magazineOrProjectilePrefab))
                    continue;
 
                //~ Get current magazine count and see if it needs to be increased
                int resupplyCount = maxMagazineCount - GetMagazineCountByMuzzle(muzzle);
                if (resupplyCount <= 0)
                    continue;
 
                //~ If storage is given check if magazine or projectile is in storage and only allow resupply if it is (Does not care for amount and intended use is with Arsenal)
                if (mustBeInStorage)
                {
                    arsenalStorage = SCR_ArsenalInventoryStorageManagerComponent.Cast(mustBeInStorage);
                    if ((arsenalStorage && !arsenalStorage.IsPrefabInArsenalStorage(magazineOrProjectilePrefab)) || (!arsenalStorage && mustBeInStorage.GetDepositItemCountByResource(magazineOrProjectilePrefab) < 1))
                        continue;
                }
 
                //~ Add magazine to be resupplied
                if (!validResupplyItems.Contains(magazineOrProjectilePrefab))
                    validResupplyItems.Insert(magazineOrProjectilePrefab, resupplyCount);
                //~ Update count till max
                else
                    validResupplyItems[magazineOrProjectilePrefab] = Math.Clamp(validResupplyItems[magazineOrProjectilePrefab] + resupplyCount, 0, maxMagazineCount);
            }
        }
    }
    
    //------------------------------------------------------------------------------------------------
    void ResupplyMagazines(notnull map<ResourceName, int> validResupplyItems)
    {
        //~ Nothing to resupply
        if (validResupplyItems.IsEmpty())
            return;
        
        if (!m_ResupplyMagazineCallback)
            m_ResupplyMagazineCallback = new SCR_ResupplyMagazinesCallback(this);
 
        //~ Resupply each given entry
        foreach (ResourceName itemPrefab, int count : validResupplyItems)
        {
            m_ResupplyMagazineCallback.Insert(itemPrefab, count);
        }
 
        m_ResupplyMagazineCallback.Start();
    }
    
    //------------------------------------------------------------------------------------------------
    void ResupplyMagazines(int maxMagazineCount = 4, EMuzzleType muzzleType = -1, InventoryStorageManagerComponent mustBeInStorage = null)
    {
        //~ Get resupply prefabs
        map<ResourceName, int> validResupplyItems = new map<ResourceName, int>();
        GetValidResupplyItemsAndCount(validResupplyItems, maxMagazineCount, muzzleType, mustBeInStorage);
        
        //~ Resupply
        ResupplyMagazines(validResupplyItems);
    }
 
    //------------------------------------------------------------------------------------------------
    void EndResupplyMagazines()
    {
        delete m_ResupplyMagazineCallback;
    }
    
    //------------------------------------------------------------------------------------------------
    EResupplyUnavailableReason CanResupplyMuzzle(notnull BaseMuzzleComponent muzzle, int maxMagazineCount = -1, InventoryStorageManagerComponent mustBeInStorage = null, out int currentMagazineAmount = -1)
    {
        //~ Cannot resupply weapons that cannot be ressupplied like the US rocket Launcer
        SCR_MuzzleInMagComponent inMagMuzzle = SCR_MuzzleInMagComponent.Cast(muzzle);
        if (inMagMuzzle && !inMagMuzzle.CanBeReloaded())
            return EResupplyUnavailableReason.NO_VALID_WEAPON;
 
        //~ Get default magazine to resupply
        ResourceName magazineToResupply = muzzle.GetDefaultMagazineOrProjectileName();
        if (SCR_StringHelper.IsEmptyOrWhiteSpace(magazineToResupply))
            return EResupplyUnavailableReason.NO_VALID_WEAPON;
        
        //~ Check if it is in arsenal or in the storage
        if (mustBeInStorage)
        {
            SCR_ArsenalInventoryStorageManagerComponent arsenalStorage = SCR_ArsenalInventoryStorageManagerComponent.Cast(mustBeInStorage);
            if ((arsenalStorage && !arsenalStorage.IsPrefabInArsenalStorage(magazineToResupply)) || (!arsenalStorage && mustBeInStorage.GetDepositItemCountByResource(magazineToResupply) < 1))
                return EResupplyUnavailableReason.NOT_IN_GIVEN_STORAGE;
        }
        
        //~ Already has enough magazines
        currentMagazineAmount = GetMagazineCountByMuzzle(muzzle);
        if (maxMagazineCount > 0 && currentMagazineAmount >= maxMagazineCount)
            return EResupplyUnavailableReason.ENOUGH_ITEMS;
        
        //~ Check if there is space in the inventory
        if (!FindStorageForResourceInsert(magazineToResupply, m_Storage))
            return EResupplyUnavailableReason.INVENTORY_FULL;
                
        return EResupplyUnavailableReason.RESUPPLY_VALID;   
    }
    
    //------------------------------------------------------------------------------------------------
    EResupplyUnavailableReason CanResupplyItem(ResourceName itemToResupply, int maxItemCount = -1, InventoryStorageManagerComponent mustBeInStorage = null, out int currentItemAmount = -1)
    {
        //~ Check if it is in arsenal or in the storage
        if (mustBeInStorage)
        {
            SCR_ArsenalInventoryStorageManagerComponent arsenalStorage = SCR_ArsenalInventoryStorageManagerComponent.Cast(mustBeInStorage);
            if ((arsenalStorage && !arsenalStorage.IsPrefabInArsenalStorage(itemToResupply)) || (!arsenalStorage && mustBeInStorage.GetDepositItemCountByResource(itemToResupply) < 1))
                return EResupplyUnavailableReason.NOT_IN_GIVEN_STORAGE;
        }
        
        //~ Already has enough of the item in storage
        currentItemAmount = GetDepositItemCountByResource(itemToResupply);
        if (maxItemCount > 0 && currentItemAmount >= maxItemCount)
            return EResupplyUnavailableReason.ENOUGH_ITEMS;
            
        //~ Check if there is space in the inventory
        if (!FindStorageForResourceInsert(itemToResupply, m_Storage))
            return EResupplyUnavailableReason.INVENTORY_FULL;
    
        return EResupplyUnavailableReason.RESUPPLY_VALID;
    }
    
    //------------------------------------------------------------------------------------------------
    void MoveItemToVicinity(IEntity pItem, BaseInventoryStorageComponent pStorageTo = null);
    
    //------------------------------------------------------------------------------------------------
    void OpenInventory()
    {
        if (m_CharacterController && m_CharacterController.GetLifeState() != ECharacterLifeState.ALIVE)
            return;
 
        MenuManager menuManager = GetGame().GetMenuManager();
        ChimeraMenuPreset menu = ChimeraMenuPreset.Inventory20Menu;
        
        MenuBase inventoryMenu = menuManager.FindMenuByPreset(menu);
        if (inventoryMenu)
            return;
        
        menuManager.OpenMenu( menu );
        m_OnInventoryOpenInvoker.Invoke(true);
        
        if (!m_CharacterController)
            return;
        
        // Quit ADS
        m_CharacterController.SetWeaponADS(false);
        m_CharacterController.SetGadgetRaisedModeWanted(false);
        
        // Pin grenade
        if (m_CharacterController.GetInputContext() && m_CharacterController.GetInputContext().GetThrow())
            m_CharacterController.SetThrow(false, true);
                
        // Inspection or lowered weapon stance
        m_bWasRaised = m_CharacterController.IsWeaponRaised();
        m_CharacterController.SetWeaponRaised(false);
    }
 
    //------------------------------------------------------------------------------------------------
    void CloseInventory()
    {
        MenuManager menuManager = GetGame().GetMenuManager();
        if (!menuManager)
            return;
        
        menuManager.CloseMenuByPreset(ChimeraMenuPreset.Inventory20Menu);
    }
    
    //------------------------------------------------------------------------------------------------
    IEntity GetStorageToOpen()
    {
        IEntity result = m_StorageToOpen;
        m_StorageToOpen = null;
        
        return result;
    }
 
    //------------------------------------------------------------------------------------------------
    void SetStorageToOpen(IEntity storage)
    {
        m_StorageToOpen = storage;
    }
 
    //------------------------------------------------------------------------------------------------
    void Action_OpenInventory()
    {
        CompartmentAccessComponent cac = m_CharacterController.GetCharacter().GetCompartmentAccessComponent();
        if (cac && cac.IsInCompartment())
        {
            IEntity owner = cac.GetCompartment().GetOwner();
            while (owner)
            {
                UniversalInventoryStorageComponent comp = UniversalInventoryStorageComponent.Cast(owner.FindComponent(UniversalInventoryStorageComponent));
                if (comp)
                {
                    SetStorageToOpen(owner);
                    break;
                }
 
                owner = owner.GetParent();
            }
        }
 
        OpenInventory();
    }
    
    //------------------------------------------------------------------------------------------------
    void OnInventoryMenuClosed()
    {
        m_OnInventoryOpenInvoker.Invoke(false);
        
        // Revert inspection or lowered weapon stance
        if (m_CharacterController)
            m_CharacterController.SetWeaponRaised(m_bWasRaised);
    }
    
    //------------------------------------------------------------------------------------------------
    override void OnStorageAdded(BaseInventoryStorageComponent storage) 
    { 
        // do nothing
    }
    
//  //------------------------------------------------------------------------------------------------
//  //! Even after physics update
//  //! \param[in] owner The owner entity
//  //! \param[in] frameNumber Time passed since last frame
//  override void EOnPostFrame(IEntity owner, float timeSlice)
//  {
//      m_OnPostFrameInvoker.Invoke(timeSlice);
//  }
    
    //------------------------------------------------------------------------------------------------
    void DebugListAllItemsInInventory()
    {
        array<IEntity> items = {};
        GetItems(items);
        Print("INV: no item", LogLevel.NORMAL);
        InventoryItemComponent pInvComp;
        ItemAttributeCollection attribs;
        foreach (IEntity item : items)
        {
            pInvComp = InventoryItemComponent.Cast(  item .FindComponent( InventoryItemComponent ) );
            if( pInvComp )
            {
                attribs = pInvComp.GetAttributes();
                if( !attribs )
                    break;
 
                string sName = attribs.GetUIInfo().GetName();
                Print("INV: " + sName, LogLevel.NORMAL);
 
            }
        }
    }
    
    //------------------------------------------------------------------------------------------------
    [RplRpc(RplChannel.Reliable, RplRcver.Server)]
    protected void RpcAsk_ServerToDeleteEntity(RplId targetRplId)
    {
        RplComponent rplComp = RplComponent.Cast(Replication.FindItem(targetRplId));
        if (!rplComp)
            return;
 
        IEntity entity = rplComp.GetEntity();
        if (!entity)
            return;
 
        RplComponent.DeleteRplEntity(entity, false);    
    }
    
    //------------------------------------------------------------------------------------------------
    void AskServerToDeleteEntity(IEntity ent)
    {
        RplComponent rplComp = RplComponent.Cast(ent.FindComponent(RplComponent));
        if (!rplComp)
            return;
            
        RplId rplId = rplComp.Id();
        Rpc(RpcAsk_ServerToDeleteEntity, rplId);
    }
    
    //------------------------------------------------------------------------------------------------
    void EnablePostFrame(bool enable)
    {
        if (enable)
            SetEventMask(GetOwner(), EntityEvent.POSTFRAME);
        else
            ClearEventMask(GetOwner(), EntityEvent.POSTFRAME);
    }
    
    //------------------------------------------------------------------------------------------------
    int GetHealthComponentCount()
    {
        return m_iHealthEquipment;
    }
    
    //------------------------------------------------------------------------------------------------  
    IEntity GetBandageItem()
    {   
        return FindItem(m_BandagePredicate, EStoragePurpose.PURPOSE_DEPOSIT);
    }
    
    //------------------------------------------------------------------------------------------------
    // destructor
    void ~SCR_InventoryStorageManagerComponent()
    {
        m_ERetCode = EInventoryRetCode.RETCODE_DEFAULT_STATE;
    }
 
    //------------------------------------------------------------------------------------------------
    bool IsAnimationReady()
    {
        if (m_CharacterController)
            return m_CharacterController.CanPlayItemGesture() || m_CharacterController.IsPlayingItemGesture();
        return true;
    }
 
    //------------------------------------------------------------------------------------------------
    bool IsInventoryLocked()
    {
        return m_bIsInventoryLocked;    
    }
    
    //------------------------------------------------------------------------------------------------
    void SetInventoryLocked(bool isLocked)
    {
        m_bIsInventoryLocked = isLocked;
    }
    
    //------------------------------------------------------------------------------------------------
    int GetAllItems(inout array<IEntity> items, BaseInventoryStorageComponent storage)
    {
        if (!storage || !items)
            return 0;
            
        int count = 0;
        
        if (!ClothNodeStorageComponent.Cast(storage))
            count = storage.GetAll(items);
 
        array<BaseInventoryStorageComponent> itemToReplaceAttachedStorages = {};
        storage.GetOwnedStorages(itemToReplaceAttachedStorages, 1, false);
        foreach (BaseInventoryStorageComponent attachedStorage : itemToReplaceAttachedStorages)
        {           
            if (ClothNodeStorageComponent.Cast(attachedStorage))
                attachedStorage.GetOwnedStorages(itemToReplaceAttachedStorages, 1, false);
            else
                count += attachedStorage.GetAll(items);
        }
            
        return count;
    }
    
    //------------------------------------------------------------------------------------------------  
    void GetWeaponPrefabsOfType(notnull array<IEntity> weapons, EWeaponType weaponType, notnull out array<EntityPrefabData> prefabs)
    {
        BaseWeaponComponent weapon;
        EntityPrefabData prefabData;
        foreach (IEntity item : weapons)
        {
            weapon = BaseWeaponComponent.Cast(item.FindComponent(BaseWeaponComponent));
            if (weapon.GetWeaponType() != weaponType)
                continue;
            
            // Ignore currently selected items
            prefabData = item.GetPrefabData();
            if (prefabData && !prefabs.Contains(prefabData))
                prefabs.Insert(prefabData);
        }
    }
    
    //------------------------------------------------------------------------------------------------  
    IEntity FindNextWeaponOfType(EWeaponType weaponType, IEntity currentItem = null, bool allowCurrentPrefab = false)
    {
        array<EntityPrefabData> prefabs = {};
        
        // Currently selected weapon may be outside of inventory and it has to be considered for sorting
        EntityPrefabData currentPrefab;
        BaseWeaponComponent currentWeapon;
        if (currentItem)
        {
            currentPrefab = currentItem.GetPrefabData();
            currentWeapon = BaseWeaponComponent.Cast(currentItem.FindComponent(BaseWeaponComponent));
            
            if (currentPrefab && currentWeapon.GetWeaponType() == weaponType)
                prefabs.Insert(currentPrefab);
        }
        
        // Collect all the matching prefabs
        array<IEntity> items = {};
        FindItemsWithComponents(items, {BaseWeaponComponent});
        GetWeaponPrefabsOfType(items, weaponType, prefabs);
        
        // No valid prefabs
        if (prefabs.IsEmpty())
            return null;
        
        // TODO: better sorting, perhaps by name
        prefabs.Sort();
        
//      foreach (EntityPrefabData prefab : prefabs)
//      {
//          Print(prefab.GetPrefabName(), LogLevel.WARNING);
//      }
        
        // Select next prefab
        int nextPrefabID = (prefabs.Find(currentPrefab) + 1) % prefabs.Count();
        EntityPrefabData nextPrefab = prefabs[nextPrefabID];
        
        // Return nothing if prefab is unchanged
        if (!allowCurrentPrefab && nextPrefab == currentPrefab)
            return null;
        
        // Select the weapon that matches the type to be selected
        foreach (IEntity item : items)
        {
            if (item && item.GetPrefabData() == nextPrefab)
                return item;
        }
        
        return null;
    }
 
#else
 
    //------------------------------------------------------------------------------------------------
    void SetReturnCode( EInventoryRetCode ERetCode ) ;
 
    //------------------------------------------------------------------------------------------------
    void SetReturnCodeDefault() ;
 
    //------------------------------------------------------------------------------------------------
    EInventoryRetCode GetReturnCode()
    {
        return m_ERetCode;
    }
 
    //------------------------------------------------------------------------------------------------
    float GetTotalWeightOfAllStorages();
 
    //------------------------------------------------------------------------------------------------
    void SetLootStorage( IEntity pOwner );
 
    //------------------------------------------------------------------------------------------------
    void InsertItem( IEntity pItem );
 
    //------------------------------------------------------------------------------------------------
    void EquipWeapon( IEntity pOwnerEntity, SCR_InvCallBack cb = null, bool bFromVicinity = true );
 
    //------------------------------------------------------------------------------------------------
    void EquipGadget( IEntity pOwnerEntity );
 
    //------------------------------------------------------------------------------------------------
    void EquipCloth( IEntity pOwnerEntity );
 
    //------------------------------------------------------------------------------------------------
    void EquipAny(BaseInventoryStorageComponent storage, IEntity item, int preferred = 0);
 
    //------------------------------------------------------------------------------------------------
    void EquipItem( EquipedWeaponStorageComponent weaponStorage, IEntity weapon );
 
    //------------------------------------------------------------------------------------------------
    void OpenInventory();
 
    //------------------------------------------------------------------------------------------------
    void Action_OpenInventory();
 
    //------------------------------------------------------------------------------------------------
    override void OnStorageAdded(BaseInventoryStorageComponent storage);
 
    //------------------------------------------------------------------------------------------------
    void EnablePostFrame(bool enable);
 
    //------------------------------------------------------------------------------------------------
    int GetHealthComponentCount();
 
    //------------------------------------------------------------------------------------------------
    IEntity GetBandageItem();
    
#endif  
    
    //------------------------------------------------------------------------------------------------
    // constructor
    void SCR_InventoryStorageManagerComponent(IEntityComponentSource src, IEntity ent, IEntity parent)
    {
        #ifndef DISABLE_INVENTORY
        
        //ChimeraCharacter pChimeraChar = ChimeraCharacter.Cast( ent );
        //pChimeraChar.s_OnCharacterCreated.Insert( DebugListAllItemsInInventory );
        
        m_Storage = SCR_CharacterInventoryStorageComponent.Cast(ent.FindComponent(CharacterInventoryStorageComponent));
        m_CharacterController = SCR_CharacterControllerComponent.Cast(ent.FindComponent(SCR_CharacterControllerComponent));
        #endif
    }
}