weaponreplacingmagandchambernext.c

Go to the documentation of this file.

 
class DetachOldMagazine extends WeaponStateBase
{
    Magazine m_oldMagazine; 
    ref InventoryLocation m_newDst;
 
    void DetachOldMagazine (Weapon_Base w = NULL, WeaponStateBase parent = NULL)
    {
        m_oldMagazine = NULL;
        m_newDst = NULL;
    }
    
    override void OnEntry (WeaponEventBase e)
    {
        super.OnEntry(e);
    }
 
    override void OnAbort (WeaponEventBase e)
    {
        super.OnAbort(e);
        m_oldMagazine = NULL;
        m_newDst = NULL;
    }
 
    override void OnExit (WeaponEventBase e)
    {
        if (m_oldMagazine)
        {
            InventoryLocation il = new InventoryLocation();
            e.m_player.GetInventory().FindFreeLocationFor( m_oldMagazine , FindInventoryLocationType.CARGO, il );           
            
            if (!m_newDst || !m_newDst.IsValid() || m_newDst.GetType() == InventoryLocationType.GROUND)
            {
                if (DayZPlayerUtils.HandleDropMagazine(e.m_player, m_oldMagazine))
                {
                    if (LogManager.IsWeaponLogEnable()) { wpnDebugPrint("[wpnfsm] " + Object.GetDebugName(m_weapon) + " DetachOldMagazine, ok - no inventory space for old magazine - dropped to ground"); }
                }
                else
                    Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " DetachOldMagazine, error - cannot drop magazine from left hand after not found inventory space for old magazine");
                
            }
            else
            {
                InventoryLocation oldSrc = new InventoryLocation();
                m_oldMagazine.GetInventory().GetCurrentInventoryLocation(oldSrc);
                
                if (GameInventory.LocationSyncMoveEntity(oldSrc, m_newDst))
                {
                    if (LogManager.IsWeaponLogEnable()) { wpnDebugPrint("[wpnfsm] " + Object.GetDebugName(m_weapon) + " DetachOldMagazine, ok - old magazine removed from wpn (LHand->inv)"); }
                }
                else
                    Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " DetachOldMagazine, error - cannot remove old mag from wpn");
            }
        }
 
        m_weapon.HideMagazine();
        m_oldMagazine = NULL;
        m_newDst = null;
        super.OnExit(e);
    }
    
    override bool SaveCurrentFSMState (ParamsWriteContext ctx)
    {
        if (!super.SaveCurrentFSMState(ctx))
            return false;
 
        if (!ctx.Write(m_oldMagazine))
        {
            Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " DetachOldMagazine.SaveCurrentFSMState: cannot write m_oldMagazine for weapon=" + m_weapon);
            return false;
        }
        
                
        if (!OptionalLocationWriteToContext(m_newDst, ctx))
        {
            Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " DetachOldMagazine.SaveCurrentFSMState: cannot write m_newDst for weapon=" + m_weapon);
            return false;
        }
 
        return true;
    }
 
    override bool LoadCurrentFSMState (ParamsReadContext ctx, int version)
    {
        if (!super.LoadCurrentFSMState(ctx, version))
            return false;
 
        if (!ctx.Read(m_oldMagazine))
        {
            Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " DetachOldMagazine.LoadCurrentFSMState: cannot read m_oldMagazine for weapon=" + m_weapon);
            return false;
        }
        
        if (!OptionalLocationReadFromContext(m_newDst, ctx))
        {
            Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " DetachOldMagazine.LoadCurrentFSMState: cannot read m_newDst for weapon=" + m_weapon);
            return false;
        }
        return true;
    }
};
 
class OldMagazineHide : MagazineHide
{
};
 
class SwapOldAndNewMagazine extends WeaponStateBase
{
    Magazine m_newMagazine; 
    ref InventoryLocation m_newDst;
 
    override void OnEntry (WeaponEventBase e)
    {
        super.OnEntry(e);
        
        if( e )
        {
            if (!m_newMagazine || !m_newDst || !m_newDst.IsValid())
            {
                Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " SwapOldAndNewMagazine, error - m_newMagazine(" + m_newMagazine + ") or destination(" + InventoryLocation.DumpToStringNullSafe(m_newDst) + ") is not set ");
            }
            else
            {
                e.m_player.GetInventory().ClearInventoryReservationEx( m_newMagazine , m_newDst );
                m_weapon.ShowMagazine();
                InventoryLocation lhand = new InventoryLocation();
                lhand.SetAttachment(e.m_player, m_newMagazine, InventorySlots.LEFTHAND);
            
                if (GameInventory.LocationSyncMoveEntity(lhand, m_newDst))
                {
                    if (LogManager.IsWeaponLogEnable()) { wpnDebugPrint("[wpnfsm] " + Object.GetDebugName(m_weapon) + " SwapOldAndNewMagazine, ok - new magazine removed from inv (LHand->Att)"); }
                }
                else
                    Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " SwapOldAndNewMagazine, error - cannot remove new mag from LHand");
            }
        }
    }
    
    override void OnAbort (WeaponEventBase e)
    {
        //m_weapon.HideMagazine();
 
        m_newMagazine = NULL;
        m_newDst = NULL;
 
        super.OnAbort(e);
    }
 
    override void OnExit (WeaponEventBase e)
    {
        m_newMagazine = NULL;
        m_newDst = NULL;
 
        super.OnExit(e);
    }
    
    override bool SaveCurrentFSMState (ParamsWriteContext ctx)
    {
        if (!super.SaveCurrentFSMState(ctx))
            return false;
 
        if (!ctx.Write(m_newMagazine))
        {
            Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " SwapOldAndNewMagazine.SaveCurrentFSMState: cannot write m_newMagazine for weapon=" + m_weapon);
            return false;
        }
        
        if (!OptionalLocationWriteToContext(m_newDst, ctx))
        {
            Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " SwapOldAndNewMagazine.SaveCurrentFSMState: cannot write m_newDst for weapon=" + m_weapon);
            return false;
        }
 
        return true;
    }
 
    override bool LoadCurrentFSMState (ParamsReadContext ctx, int version)
    {
        if (!super.LoadCurrentFSMState(ctx, version))
            return false;
 
        if (!ctx.Read(m_newMagazine))
        {
            Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " SwapOldAndNewMagazine.LoadCurrentFSMState: cannot read m_newMagazine for weapon=" + m_weapon);
            return false;
        }
        if (!OptionalLocationReadFromContext(m_newDst, ctx))
        {
            Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " SwapOldAndNewMagazine.LoadCurrentFSMState: cannot read m_newDst for weapon=" + m_weapon);
            return false;
        }
 
        return true;
    }
};
 
class AttachNewMagazine extends WeaponStateBase
{
    Magazine m_newMagazine; 
    ref InventoryLocation m_newDst;
 
    void AttachNewMagazine (Weapon_Base w = NULL, WeaponStateBase parent = NULL)
    {
        m_newMagazine = NULL;
        m_newDst = NULL;
    }
    
    override bool IsWaitingForActionFinish ()
    {
        return true;
    }
 
    override void OnEntry (WeaponEventBase e)
    {
        super.OnEntry(e);
        if (e)
        {
            if (m_newMagazine && m_newDst)
            {
                InventoryLocation il = new InventoryLocation();
                if (m_newMagazine.GetInventory().GetCurrentInventoryLocation(il))
                {
                    InventoryLocation lhand = new InventoryLocation();
                    lhand.SetAttachment(e.m_player, m_newMagazine, InventorySlots.LEFTHAND);
                    if (GameInventory.LocationSyncMoveEntity(il, m_newDst))
                    {
                        m_weapon.ShowMagazine();
                        if (LogManager.IsWeaponLogEnable()) { wpnDebugPrint("[wpnfsm] " + Object.GetDebugName(m_weapon) + " AttachNewMagazine, ok - attached new magazine (LHand->dst)"); }
                    }
                    else
                    {
                        if (LogManager.IsWeaponLogEnable()) { wpnDebugPrint("[wpnfsm] " + Object.GetDebugName(m_weapon) + " AttachNewMagazine, error - cannot attach new magazine!"); }
                    }
                }
                else
                    Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " AttachNewMagazine, error - cannot get curr location");
            }
            else
                Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " AttachNewMagazine, error - no magazines configured for replace (m_new=NULL)");
        }
    }
 
    override void OnAbort (WeaponEventBase e)
    {
        if (m_newMagazine && m_newDst)
        {
            if (DayZPlayerUtils.HandleDropMagazine(e.m_player, m_newMagazine))
            {
                if (LogManager.IsWeaponLogEnable()) { wpnDebugPrint("[wpnfsm] " + Object.GetDebugName(m_weapon) + " WeaponDetachingMag_Store, ok - aborting, detached magazine dropped to ground"); }
            }
            else
                Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " WeaponDetachingMag_Store, error - cannot abort detaching of magazine");
 
            m_weapon.HideMagazine(); // force hide on abort
        }
        
        if (LogManager.IsWeaponLogEnable()) { wpnDebugPrint("[wpnfsm] " + Object.GetDebugName(m_weapon) + " AttachNewMagazine, aborting, but attached new magazine already in place"); }
        m_newMagazine = NULL;
        m_newDst = NULL;
 
        super.OnAbort(e);
    }
 
    override void OnExit (WeaponEventBase e)
    {
        m_newMagazine = NULL;
        m_newDst = NULL;
        super.OnExit(e);
    }
 
    override bool SaveCurrentFSMState (ParamsWriteContext ctx)
    {
        if (!super.SaveCurrentFSMState(ctx))
            return false;
 
        if (!ctx.Write(m_newMagazine))
        {
            Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " AttachNewMagazine.SaveCurrentFSMState: cannot write m_newMagazine for weapon=" + m_weapon);
            return false;
        }
 
        if (!OptionalLocationWriteToContext(m_newDst, ctx))
        {
            Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " AttachNewMagazine.SaveCurrentFSMState: cannot write m_newDst for weapon=" + m_weapon);
            return false;
        }
        return true;
    }
 
    override bool LoadCurrentFSMState (ParamsReadContext ctx, int version)
    {
        if (!super.LoadCurrentFSMState(ctx, version))
            return false;
 
        if (!ctx.Read(m_newMagazine))
        {
            Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " AttachNewMagazine.LoadCurrentFSMState: cannot read m_newMagazine for weapon=" + m_weapon);
            return false;
        }
        if (!OptionalLocationReadFromContext(m_newDst, ctx))
        {
            Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " AttachNewMagazine.LoadCurrentFSMState: cannot read m_newDst for weapon=" + m_weapon);
            return false;
        }
        return true;
    }
};
 
class AttachNewMagazine_W4T extends WeaponStateBase
{
    override bool IsWaitingForActionFinish () { return true; }
};
 
 
class WeaponReplacingMagAndChamberNext extends WeaponStateBase
{
    WeaponActions m_action;
    int m_actionType;
 
    ref WeaponStartAction m_start;
    ref DetachOldMagazine m_detach;
    ref OldMagazineHide m_hideOld;
    ref SwapOldAndNewMagazine m_swapMags;
    ref AttachNewMagazine_W4T m_attach;
    ref WeaponChamberFromAttMag_W4T m_chamber;
    ref WeaponCharging_CK m_onCK;
    ref WeaponEjectCasing m_eject;
 
    // substates configuration
    Magazine m_oldMagazine; 
    Magazine m_newMagazine; 
    ref InventoryLocation m_newDst;
 
    void WeaponReplacingMagAndChamberNext (Weapon_Base w = NULL, WeaponStateBase parent = NULL, WeaponActions action = WeaponActions.NONE, int actionType = -1)
    {
        m_action = action;
        m_actionType = actionType;
        m_newMagazine = NULL;
 
        // setup nested state machine
        m_start = new WeaponStartAction(m_weapon, this, m_action, m_actionType);
        m_eject = new WeaponEjectCasing(m_weapon,this);
        m_detach = new DetachOldMagazine(m_weapon, this);
        m_hideOld = new OldMagazineHide(m_weapon, this);
        m_swapMags = new SwapOldAndNewMagazine(m_weapon, this);
        m_attach = new AttachNewMagazine_W4T(m_weapon, this);
        m_chamber = new WeaponChamberFromAttMag_W4T(m_weapon, this);
        m_onCK = new WeaponCharging_CK(m_weapon, this);
 
        // events
        WeaponEventBase __so_ = new WeaponEventAnimSliderOpen;
        WeaponEventBase __md_ = new WeaponEventAnimMagazineDetached;
        WeaponEventBase __mh_ = new WeaponEventAnimMagazineHide;
        WeaponEventBase __ms_ = new WeaponEventAnimMagazineShow;
        WeaponEventBase __ma_ = new WeaponEventAnimMagazineAttached;
        WeaponEventBase __ck_ = new WeaponEventAnimCocked;
        WeaponEventBase _fin_ = new WeaponEventHumanCommandActionFinished;
 
        m_fsm = new WeaponFSM(this); // @NOTE: set owner of the submachine fsm
        m_fsm.AddTransition(new WeaponTransition(   m_start, __md_, m_detach));
        m_fsm.AddTransition(new WeaponTransition(   m_start, __so_, m_eject));
        m_fsm.AddTransition(new WeaponTransition(   m_eject, __md_, m_detach));
        m_fsm.AddTransition(new WeaponTransition(  m_detach, __mh_, m_hideOld));
        m_fsm.AddTransition(new WeaponTransition( m_hideOld, __ms_, m_swapMags));
        m_fsm.AddTransition(new WeaponTransition(m_swapMags, __ma_, m_attach));
        m_fsm.AddTransition(new WeaponTransition(  m_attach, __ck_, m_chamber, NULL, new GuardAnd(new WeaponGuardCurrentChamberEmpty(m_weapon), new WeaponGuardHasAmmo(m_weapon))));
        m_fsm.AddTransition(new WeaponTransition(  m_attach, __ck_, m_onCK));
        
        m_fsm.AddTransition(new WeaponTransition(  m_attach, _fin_, NULL));
        m_fsm.AddTransition(new WeaponTransition( m_chamber, _fin_, NULL));
        m_fsm.AddTransition(new WeaponTransition(    m_onCK, _fin_, NULL));
        
        // Safety exits
        m_fsm.AddTransition(new WeaponTransition(m_swapMags, _fin_, null));
        m_fsm.AddTransition(new WeaponTransition(m_hideOld,  _fin_, null));
        m_fsm.AddTransition(new WeaponTransition(m_detach,   _fin_, null));
        m_fsm.AddTransition(new WeaponTransition(m_eject,    _fin_, null));
        m_fsm.AddTransition(new WeaponTransition(m_start,    _fin_, null)); 
 
        m_fsm.SetInitialState(m_start);
    }
    
    override void OnEntry (WeaponEventBase e)
    {
        if (e != NULL)
        {
            WeaponEventSwapMagazine se;
            if (Class.CastTo(se, e))
            {
                int mi = m_weapon.GetCurrentMuzzle();
                m_oldMagazine = m_weapon.GetMagazine(mi);
                m_newMagazine = se.m_magazine;
                m_newDst = se.m_dst;
                
                if (LogManager.IsWeaponLogEnable()) { wpnDebugPrint("[wpnfsm] " + Object.GetDebugName(m_weapon) + " WeaponReplacingMagAndChamberNext, m_oldMagazine= " + m_oldMagazine + " m_newMagazine= " + m_newMagazine + " m_oldMagazineDst= " + typename.EnumToString(InventoryLocationType, se.m_dst.GetType())); }
                
                InventoryLocation oldSrc = new InventoryLocation();
                InventoryLocation newSrc = new InventoryLocation();
 
                if (!m_newMagazine.GetInventory().GetCurrentInventoryLocation(newSrc))
                {
                    Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " WeaponReplacingMagAndChamberNext cannot get curr inv loc of NEW mag=" + Object.GetDebugName(m_newMagazine));
                }
                if (!m_oldMagazine.GetInventory().GetCurrentInventoryLocation(oldSrc))
                {
                    Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " WeaponReplacingMagAndChamberNext cannot get curr inv loc of OLD mag=" + Object.GetDebugName(m_oldMagazine));
                }
                
                // move to LH
                InventoryLocation lhand = new InventoryLocation();
                lhand.SetAttachment(e.m_player, m_newMagazine, InventorySlots.LEFTHAND);
                
                if (GameInventory.LocationSyncMoveEntity(newSrc, lhand))
                {
                    if (LogManager.IsWeaponLogEnable()) { wpnDebugPrint("[wpnfsm] " + Object.GetDebugName(m_weapon) + " RemoveNewMagazineFromInventory, ok - new magazine removed from inv (inv->LHand)"); }
                }
                else
                    Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " RemoveNewMagazineFromInventory, error - cannot new remove mag from inv");
 
                m_detach.m_oldMagazine = m_oldMagazine;
                m_detach.m_newDst = m_newDst;
 
                m_swapMags.m_newMagazine = m_newMagazine;
                
                oldSrc.SetItem(m_newMagazine);
                m_swapMags.m_newDst = oldSrc;
            }
        }
        super.OnEntry(e);
    }
    
    override void OnExit (WeaponEventBase e)
    {
        super.OnExit(e);
        
        EntityAI leftHandItem = e.m_player.GetInventory().FindAttachment(InventorySlots.LEFTHAND);
        Magazine mag = Magazine.Cast(leftHandItem);
        
        if(mag)
        {
            if (m_newMagazine)
                e.m_player.GetInventory().ClearInventoryReservationEx( m_newMagazine , null );
            if (m_oldMagazine)
                e.m_player.GetInventory().ClearInventoryReservationEx( m_oldMagazine , null );
            
            InventoryLocation il = new InventoryLocation();
            
            e.m_player.GetInventory().FindFreeLocationFor(mag, FindInventoryLocationType.CARGO, il);
            
            if (!il.IsValid())
            {
                if (DayZPlayerUtils.HandleDropMagazine(e.m_player, mag))
                {
                    if (LogManager.IsWeaponLogEnable()) { wpnDebugPrint("[wpnfsm] " + Object.GetDebugName(m_weapon) + " DetachOldMagazine, ok - no inventory space for old magazine - dropped to ground - exit"); }
                }
                else
                    Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " DetachOldMagazine, error - cannot drop magazine from left hand after not found inventory space for old magazine - exit");
                
            }
            else
            {
                InventoryLocation oldSrc = new InventoryLocation();
                mag.GetInventory().GetCurrentInventoryLocation(oldSrc);
                
                if (GameInventory.LocationSyncMoveEntity(oldSrc, il))
                {
                    if (LogManager.IsWeaponLogEnable()) { wpnDebugPrint("[wpnfsm] " + Object.GetDebugName(m_weapon) + " DetachOldMagazine, ok - old magazine removed from wpn (LHand->inv) - exit"); }
                }
                else
                    Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " DetachOldMagazine, error - cannot remove old mag from wpn - exit");
            }
        }
 
        m_oldMagazine = NULL;
        m_newMagazine = NULL;
        m_newDst = NULL;
    }
    
    override void OnAbort(WeaponEventBase e)
    {
        super.OnAbort(e);
        
        EntityAI leftHandItem = e.m_player.GetInventory().FindAttachment(InventorySlots.LEFTHAND);
        Magazine mag = Magazine.Cast(leftHandItem);
        
        if (mag)
        {
            if (m_newMagazine)
                e.m_player.GetInventory().ClearInventoryReservationEx( m_newMagazine , null );
            if (m_oldMagazine)
                e.m_player.GetInventory().ClearInventoryReservationEx( m_oldMagazine , null );
            
            InventoryLocation il = new InventoryLocation();
            
            e.m_player.GetInventory().FindFreeLocationFor(mag, FindInventoryLocationType.CARGO, il);
        
            if (!il.IsValid())
            {
                if (DayZPlayerUtils.HandleDropMagazine(e.m_player, mag))
                {
                    if (LogManager.IsWeaponLogEnable()) { wpnDebugPrint("[wpnfsm] " + Object.GetDebugName(m_weapon) + " DetachOldMagazine, ok - no inventory space for old magazine - dropped to ground - abort"); }
                }
                else
                    Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " DetachOldMagazine, error - cannot drop magazine from left hand after not found inventory space for old magazine - abort");
                
            }
            else
            {
                InventoryLocation oldSrc = new InventoryLocation();
                mag.GetInventory().GetCurrentInventoryLocation(oldSrc);
                
                if (GameInventory.LocationSyncMoveEntity(oldSrc, il))
                {
                    if (LogManager.IsWeaponLogEnable()) { wpnDebugPrint("[wpnfsm] " + Object.GetDebugName(m_weapon) + " DetachOldMagazine, ok - old magazine removed from wpn (LHand->inv) - abort"); }
                }
                else
                    Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " DetachOldMagazine, error - cannot remove old mag from wpn - abort");
            }
        }
    }
    
    override bool SaveCurrentFSMState (ParamsWriteContext ctx)
    {
        if (!super.SaveCurrentFSMState(ctx))
            return false;
 
        if (!ctx.Write(m_newMagazine))
        {
            Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " DetachOldMagazine.SaveCurrentFSMState: cannot write m_newMagazine for weapon=" + m_weapon);
            return false;
        }
 
        if (!ctx.Write(m_oldMagazine))
        {
            Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " DetachOldMagazine.SaveCurrentFSMState: cannot write m_oldMagazine for weapon=" + m_weapon);
            return false;
        }
        
        if (!m_newDst.WriteToContext(ctx))
        {
            Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " DetachOldMagazine.SaveCurrentFSMState: cannot write m_newDst for weapon=" + m_weapon);
            return false;
        }
        
        return true;
    }
 
    override bool LoadCurrentFSMState (ParamsReadContext ctx, int version)
    {
        if (!super.LoadCurrentFSMState(ctx, version))
            return false;
 
        if (!ctx.Read(m_newMagazine))
        {
            Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " DetachOldMagazine.LoadCurrentFSMState: cannot read m_newMagazine for weapon=" + m_weapon);
            return false;
        }
        if (!ctx.Read(m_oldMagazine))
        {
            Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " DetachOldMagazine.LoadCurrentFSMState: cannot read m_oldMagazine for weapon=" + m_weapon);
            return false;
        }
        m_newDst = new InventoryLocation;
        
        if (!m_newDst.ReadFromContext(ctx))
        {
            Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " DetachOldMagazine.LoadCurrentFSMState: cannot read m_newDst for weapon=" + m_weapon);
            return false;
        }
        
        return true;
    }
};