WeaponAttachMagazine.c

Go to the documentation of this file.

class RemoveNewMagazineFromInventory extends WeaponStateBase
{
  Magazine m_newMagazine; 
  ref InventoryLocation m_newSrc;
 
  void RemoveNewMagazineFromInventory(Weapon_Base w = NULL, WeaponStateBase parent = NULL)
  {
    m_newMagazine = NULL;
    m_newSrc = NULL;
  }
 
  override void OnEntry(WeaponEventBase e)
  {
    if (e)
    {
      if (!m_newSrc.IsValid())
        Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " RemoveNewMagazineFromInventory m_newSrc=invalid, item not in bubble?");
 
      if (m_newMagazine && m_newSrc && m_newSrc.IsValid())
      {
        InventoryLocation curr = new InventoryLocation;
        m_newMagazine.GetInventory().GetCurrentInventoryLocation(curr);
 
        if (m_newSrc.GetType() == InventoryLocationType.GROUND && curr.GetType() == InventoryLocationType.ATTACHMENT && curr.GetSlot() == InventorySlots.LEFTHAND)
        {
          // already in LH
        }
        else
        {
          InventoryLocation lhand = new InventoryLocation;
          lhand.SetAttachment(e.m_player, m_newMagazine, InventorySlots.LEFTHAND);
          if (GameInventory.LocationSyncMoveEntity(m_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");
        }
      }
      else
        Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " RemoveNewMagazineFromInventory, error - no magazines configured for replace (m_old=m_new=NULL)");
    }
    super.OnEntry(e);
  }
 
  override void OnAbort(WeaponEventBase e)
  {
    m_newMagazine = NULL;
    m_newSrc = NULL;
 
    super.OnAbort(e);
  }
 
  override void OnExit(WeaponEventBase e)
  {
    m_weapon.ShowMagazine();
 
    m_newMagazine = NULL;
    m_newSrc = 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) + " RemoveNewMagazineFromInventory.SaveCurrentFSMState: cannot write m_newMagazine for weapon=" + m_weapon);
      return false;
    }
 
    if (!OptionalLocationWriteToContext(m_newSrc, ctx))
    {
      Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " RemoveNewMagazineFromInventory.SaveCurrentFSMState: cannot write m_newSrc 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) + " RemoveNewMagazineFromInventory.LoadCurrentFSMState: cannot read m_newMagazine for weapon=" + m_weapon);
      return false;
    }
 
    if (!OptionalLocationReadFromContext(m_newSrc, ctx))
    {
      Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " RemoveNewMagazineFromInventory.LoadCurrentFSMState: cannot read m_newSrc for weapon=" + m_weapon);
      return false;
    }
    return true;
  }
};
 
 
class RemoveNewMagazineFromInventory_OnEntryShowMag extends RemoveNewMagazineFromInventory
{
  override void OnEntry(WeaponEventBase e)
  {
    if (e)
      m_weapon.ShowMagazine();
    super.OnEntry(e);
  }
};
 
 
class WeaponAttachMagazine extends WeaponStateBase
{
  WeaponActions m_action;
  int m_actionType;
 
  ref WeaponStartAction m_start;
  ref AttachNewMagazine m_attach;
  ref WeaponChamberFromAttMag_W4T m_chamber;
  ref WeaponCharging_CK m_onCK;
  ref WeaponEjectCasing m_eject;
 
  void WeaponAttachMagazine(Weapon_Base w = NULL, WeaponStateBase parent = NULL, WeaponActions action = WeaponActions.NONE, int actionType = -1)
  {
    m_action = action;
    m_actionType = actionType;
 
    // setup nested state machine
    m_start = new WeaponStartAction(m_weapon, this, m_action, m_actionType);
    m_eject = new WeaponEjectCasing(m_weapon, this);
    m_attach = new AttachNewMagazine(m_weapon, this);
    m_chamber = new WeaponChamberFromAttMag_W4T(m_weapon, this);
    m_onCK = new WeaponCharging_CK(m_weapon, this);
 
    // events: MS, MA, BE, CK
    WeaponEventBase _fin_ = new WeaponEventHumanCommandActionFinished;
    WeaponEventBase __ms_ = new WeaponEventAnimMagazineShow;
    WeaponEventBase __so_ = new WeaponEventAnimSliderOpen;
    WeaponEventBase __ma_ = new WeaponEventAnimMagazineAttached;
    WeaponEventBase __ck_ = new WeaponEventAnimCocked;
 
    m_fsm = new WeaponFSM(this); // @NOTE: set owner of the submachine fsm
 
    m_fsm.AddTransition(new WeaponTransition(m_start, __ms_, m_attach));
    m_fsm.AddTransition(new WeaponTransition(m_start, __so_, m_eject));
    m_fsm.AddTransition(new WeaponTransition(m_eject, __ms_, m_attach));
    m_fsm.AddTransition(new WeaponTransition(m_attach, __ck_, m_chamber, NULL, new GuardAnd(new WeaponGuardCurrentChamberEmpty(m_weapon), new WeaponGuardHasAmmo(m_weapon)))); // when opened, there is no __be_ event
    m_fsm.AddTransition(new WeaponTransition(m_attach, __ck_, m_onCK, NULL, new GuardAnd(new WeaponGuardCurrentChamberEmpty(m_weapon), new GuardNot(new WeaponGuardHasAmmo(m_weapon)))));
    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_eject, _fin_, null));
    m_fsm.AddTransition(new WeaponTransition(m_start, _fin_, null));
 
 
    m_fsm.SetInitialState(m_start);
  }
 
  override void OnEntry(WeaponEventBase e)
  {
    if (e)
    {
      Magazine mag = e.m_magazine;
 
      InventoryLocation newSrc = new InventoryLocation;
      mag.GetInventory().GetCurrentInventoryLocation(newSrc);
 
      // move to LH
      InventoryLocation lhand = new InventoryLocation;
      lhand.SetAttachment(e.m_player, mag, InventorySlots.LEFTHAND);
      if (GameInventory.LocationSyncMoveEntity(newSrc, lhand))
      {
        if (LogManager.IsWeaponLogEnable()) wpnDebugPrint("[wpnfsm] " + Object.GetDebugName(m_weapon) + " WeaponAttachMagazine, ok - new magazine removed from inv (inv->LHand)");
      }
      else
        Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " WeaponAttachMagazine, error - cannot new remove mag from inv");
 
      InventoryLocation il = new InventoryLocation;
      il.SetAttachment(m_weapon, mag, InventorySlots.MAGAZINE);
      m_attach.m_newMagazine = mag;
      m_attach.m_newDst = il;
    }
    super.OnEntry(e); // @NOTE: super at the end (prevent override from submachine start)
  }
 
  override void OnAbort(WeaponEventBase e)
  {
    EntityAI leftHandItem = e.m_player.GetInventory().FindAttachment(InventorySlots.LEFTHAND);
    Magazine mag = Magazine.Cast(leftHandItem);
 
    if (mag)
    {
      e.m_player.GetInventory().ClearInventoryReservationEx(mag, null);
      InventoryLocation il = new InventoryLocation;
      e.m_player.GetInventory().FindFreeLocationFor(mag, FindInventoryLocationType.CARGO, il);
 
      if (!il || !il.IsValid())
      {
        if (DayZPlayerUtils.HandleDropMagazine(e.m_player, mag))
        {
          if (LogManager.IsWeaponLogEnable()) wpnDebugPrint("[wpnfsm] " + Object.GetDebugName(m_weapon) + " WeaponAttachMagazine, ok - no inventory space for old magazine - dropped to ground");
        }
        else
          Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " WeaponAttachMagazine, error - cannot drop magazine from left hand after not found inventory space for old magazine");
 
      }
      else
      {
        InventoryLocation oldSrc = new InventoryLocation;
        mag.GetInventory().GetCurrentInventoryLocation(oldSrc);
 
        if (GameInventory.LocationSyncMoveEntity(oldSrc, il))
        {
          if (LogManager.IsWeaponLogEnable()) wpnDebugPrint("[wpnfsm] " + Object.GetDebugName(m_weapon) + " WeaponAttachMagazine, ok - old magazine removed from wpn (LHand->inv)");
        }
        else
          Error("[wpnfsm] " + Object.GetDebugName(m_weapon) + " WeaponAttachMagazine, error - cannot remove old mag from wpn");
      }
    }
    super.OnAbort(e);
  }
};