FirearmActionAttachMagazine.c

Go to the documentation of this file.

class AttachMagazineActionReciveData : ActionReciveData
{
  ref InventoryLocation m_ilOldMagazine;
}
class AttachMagazineActionData : ActionData
{
  ref InventoryLocation m_ilOldMagazine;
  Magazine m_oldMagazine;
}
 
class FirearmActionAttachMagazine : FirearmActionBase
{
  //-----------------------------------------------------
  // Action events and methods
  //-----------------------------------------------------
  void FirearmActionAttachMagazine()
  {
    m_Text = "#attach";
  }
 
  override int GetActionCategory()
  {
    return AC_SINGLE_USE;
  }
 
  override bool ActionCondition(PlayerBase player, ActionTarget target, ItemBase item) //condition for action
  {
    if (!super.ActionCondition(player, target, item))
      return false;
 
    HumanCommandWeapons hcw = player.GetCommandModifier_Weapons();
    Magazine mag = Magazine.Cast(target.GetObject());
    Weapon_Base wpn = Weapon_Base.Cast(item);
    return mag && (player.GetWeaponManager().CanAttachMagazine(wpn, mag) || player.GetWeaponManager().CanSwapMagazine(wpn, mag)) && (!hcw || hcw.GetRunningAction() != WeaponActions.RELOAD);
  }
 
  override bool ActionConditionContinue(ActionData action_data)
  {
    return true;
  }
 
  override ActionData CreateActionData()
  {
    AttachMagazineActionData action_data = new AttachMagazineActionData;
    return action_data;
  }
 
 
  override void WriteToContext(ParamsWriteContext ctx, ActionData action_data)
  {
    super.WriteToContext(ctx, action_data);
 
    AttachMagazineActionData action_data_am = AttachMagazineActionData.Cast(action_data);
 
    action_data_am.m_ilOldMagazine.WriteToContext(ctx);
  }
 
  override bool ReadFromContext(ParamsReadContext ctx, out ActionReciveData action_recive_data)
  {
    if (!action_recive_data)
      action_recive_data = new AttachMagazineActionReciveData;
 
    super.ReadFromContext(ctx, action_recive_data);
 
    InventoryLocation il = new InventoryLocation;
    if (!il.ReadFromContext(ctx))
      return false;
 
    AttachMagazineActionReciveData recive_data_am = AttachMagazineActionReciveData.Cast(action_recive_data);
    recive_data_am.m_ilOldMagazine = il;
 
    return true;
  }
 
  override void HandleReciveData(ActionReciveData action_recive_data, ActionData action_data)
  {
    AttachMagazineActionReciveData recive_data_am = AttachMagazineActionReciveData.Cast(action_recive_data);
    AttachMagazineActionData action_data_am = AttachMagazineActionData.Cast(action_data);
 
    action_data.m_MainItem = action_recive_data.m_MainItem;
    if (!action_recive_data.m_Target)
      action_data.m_Target = new ActionTarget(NULL, NULL, -1, vector.Zero, 0);
    else
      action_data.m_Target = action_recive_data.m_Target;
    action_data_am.m_ilOldMagazine = recive_data_am.m_ilOldMagazine;
  }
 
  override bool Post_SetupAction(ActionData action_data)
  {
    if (!GetGame().IsDedicatedServer())
    {
      Weapon_Base wpn = Weapon_Base.Cast(action_data.m_MainItem);
      int muzzle_index = wpn.GetCurrentMuzzle();
 
      AttachMagazineActionData am_action_data = AttachMagazineActionData.Cast(action_data);
      am_action_data.m_oldMagazine = wpn.GetMagazine(muzzle_index);
 
      InventoryLocation new_il = new InventoryLocation;
 
      if (!action_data.m_Player.GetWeaponManager().PrepareInventoryLocationForMagazineSwap(wpn, Magazine.Cast(action_data.m_Target.GetObject()), new_il))
        return false;
      am_action_data.m_ilOldMagazine = new_il;
    }
    return true;
  }
 
  override bool InventoryReservation(ActionData action_data)
  {
    if ((IsLocal() || !UseAcknowledgment()) && IsInstant())
      return true;
 
    bool success = true;
    InventoryLocation oldMagTargetInventoryLocation = NULL;
    InventoryLocation targetInventoryLocation = NULL;
    InventoryLocation handInventoryLocation = NULL;
 
    PlayerBase player = action_data.m_Player;
    AttachMagazineActionData am_action_data = AttachMagazineActionData.Cast(action_data);
 
    Weapon_Base wpn = Weapon_Base.Cast(action_data.m_MainItem);
    int muzzle_index = wpn.GetCurrentMuzzle();
 
    Magazine new_mag = Magazine.Cast(action_data.m_Target.GetObject());
 
    if (am_action_data.m_oldMagazine)
    {
      oldMagTargetInventoryLocation = new InventoryLocation;
      if (action_data.m_Player.GetInventory().HasInventoryReservation(am_action_data.m_oldMagazine, am_action_data.m_ilOldMagazine))
        success = false;
      else
        player.GetInventory().AddInventoryReservationEx(am_action_data.m_oldMagazine, am_action_data.m_ilOldMagazine, GameInventory.c_InventoryReservationTimeoutMS);
    }
 
    if (success)
    {
      targetInventoryLocation = new InventoryLocation;
      targetInventoryLocation.SetAttachment(wpn, new_mag, InventorySlots.MAGAZINE);
      if (action_data.m_Player.GetInventory().HasInventoryReservation(new_mag, targetInventoryLocation))
      {
        success = false;
        if (am_action_data.m_oldMagazine)
          player.GetInventory().ClearInventoryReservation(am_action_data.m_oldMagazine, am_action_data.m_ilOldMagazine);
      }
      else
        action_data.m_Player.GetInventory().AddInventoryReservationEx(new_mag, targetInventoryLocation, GameInventory.c_InventoryReservationTimeoutMS);
    }
 
    if (success)
    {
      handInventoryLocation = new InventoryLocation;
      handInventoryLocation.SetHands(action_data.m_Player, wpn);
 
      if (action_data.m_Player.GetInventory().HasInventoryReservation(wpn, handInventoryLocation))
      {
        if (am_action_data.m_oldMagazine)
          player.GetInventory().ClearInventoryReservation(am_action_data.m_oldMagazine, am_action_data.m_ilOldMagazine);
        player.GetInventory().ClearInventoryReservation(new_mag, targetInventoryLocation);
        success = false;
      }
      else
        action_data.m_Player.GetInventory().AddInventoryReservationEx(wpn, handInventoryLocation, GameInventory.c_InventoryReservationTimeoutMS);
    }
 
    if (success)
    {
      if (am_action_data.m_ilOldMagazine)
        action_data.m_ReservedInventoryLocations.Insert(am_action_data.m_ilOldMagazine);
 
      if (targetInventoryLocation)
        action_data.m_ReservedInventoryLocations.Insert(targetInventoryLocation);
 
      if (handInventoryLocation)
        action_data.m_ReservedInventoryLocations.Insert(handInventoryLocation);
    }
 
    return success;
  }
 
  override void Start(ActionData action_data)
  {
    super.Start(action_data);
    AttachMagazineActionData action_data_am = AttachMagazineActionData.Cast(action_data);
    Weapon_Base wpn = Weapon_Base.Cast(action_data.m_MainItem);
    Magazine mag = Magazine.Cast(action_data.m_Target.GetObject());
    ClearInventoryReservationEx(action_data);
    if (action_data.m_Player.GetWeaponManager().CanAttachMagazine(wpn, mag, false))
      action_data.m_Player.GetWeaponManager().AttachMagazine(mag, this);
    else
      action_data.m_Player.GetWeaponManager().SwapMagazineEx(mag, action_data_am.m_ilOldMagazine, this);
    InventoryReservation(action_data);
  }
 
  override bool CanBePerformedFromInventory()
  {
    return true;
  }
 
  override bool CanBePerformedFromQuickbar()
  {
    return true;
  }
};
 
 
class FirearmActionAttachMagazineQuick : FirearmActionBase
{
  void FirearmActionAttachMagazineQuick()
  {
  }
 
  override bool ActionCondition(PlayerBase player, ActionTarget target, ItemBase item) //condition for action
  {
    if (!super.ActionCondition(player, target, item))
      return false;
 
    HumanCommandWeapons hcw = player.GetCommandModifier_Weapons();
    if (hcw && hcw.GetRunningAction() == WeaponActions.RELOAD)
      return false;
 
    MagazineStorage mag = MagazineStorage.Cast(player.GetWeaponManager().GetPreparedMagazine());
 
    if (!mag)
      return false;
 
    Weapon weapon = Weapon.Cast(item);
    bool isLoadedMag = false;
 
    for (int i = 0, count = weapon.GetMuzzleCount(); i < count; ++i)
      isLoadedMag |= (mag == weapon.GetMagazine(i));
 
    return !isLoadedMag;
  }
 
  override ActionData CreateActionData()
  {
    AttachMagazineActionData action_data = new AttachMagazineActionData;
    return action_data;
  }
 
 
  override void WriteToContext(ParamsWriteContext ctx, ActionData action_data)
  {
    super.WriteToContext(ctx, action_data);
 
    AttachMagazineActionData action_data_am = AttachMagazineActionData.Cast(action_data);
 
    action_data_am.m_ilOldMagazine.WriteToContext(ctx);
  }
 
  override bool ReadFromContext(ParamsReadContext ctx, out ActionReciveData action_recive_data)
  {
    if (!action_recive_data)
      action_recive_data = new AttachMagazineActionReciveData;
 
    super.ReadFromContext(ctx, action_recive_data);
 
    InventoryLocation il = new InventoryLocation;
    if (!il.ReadFromContext(ctx))
      return false;
 
    AttachMagazineActionReciveData recive_data_am = AttachMagazineActionReciveData.Cast(action_recive_data);
    recive_data_am.m_ilOldMagazine = il;
 
    return true;
  }
 
  override void HandleReciveData(ActionReciveData action_recive_data, ActionData action_data)
  {
    AttachMagazineActionReciveData recive_data_am = AttachMagazineActionReciveData.Cast(action_recive_data);
    AttachMagazineActionData action_data_am = AttachMagazineActionData.Cast(action_data);
 
    action_data.m_MainItem = action_recive_data.m_MainItem;
    if (!action_recive_data.m_Target)
      action_data.m_Target = new ActionTarget(NULL, NULL, -1, vector.Zero, 0);
    else
      action_data.m_Target = action_recive_data.m_Target;
    action_data_am.m_ilOldMagazine = recive_data_am.m_ilOldMagazine;
  }
 
  override bool Post_SetupAction(ActionData action_data)
  {
    if (!GetGame().IsDedicatedServer())
    {
      Weapon_Base wpn = Weapon_Base.Cast(action_data.m_MainItem);
      int muzzle_index = wpn.GetCurrentMuzzle();
 
      AttachMagazineActionData am_action_data = AttachMagazineActionData.Cast(action_data);
      am_action_data.m_oldMagazine = wpn.GetMagazine(muzzle_index);
 
      ActionTarget newTarget = new ActionTarget(action_data.m_Player.GetWeaponManager().GetPreparedMagazine(), null, -1, vector.Zero, -1);
      action_data.m_Target = newTarget;
 
      InventoryLocation new_il = new InventoryLocation;
 
      if (!action_data.m_Player.GetWeaponManager().PrepareInventoryLocationForMagazineSwap(wpn, Magazine.Cast(action_data.m_Target.GetObject()), new_il))
        return false;
      am_action_data.m_ilOldMagazine = new_il;
    }
    return true;
  }
 
  override bool InventoryReservation(ActionData action_data)
  {
    if ((IsLocal() || !UseAcknowledgment()) && IsInstant())
      return true;
 
    bool success = true;
    InventoryLocation oldMagTargetInventoryLocation = NULL;
    InventoryLocation targetInventoryLocation = NULL;
    InventoryLocation handInventoryLocation = NULL;
 
    PlayerBase player = action_data.m_Player;
    AttachMagazineActionData am_action_data = AttachMagazineActionData.Cast(action_data);
 
    Weapon_Base wpn = Weapon_Base.Cast(action_data.m_MainItem);
    int muzzle_index = wpn.GetCurrentMuzzle();
 
    Magazine new_mag = Magazine.Cast(action_data.m_Target.GetObject());
 
    if (am_action_data.m_oldMagazine)
    {
      oldMagTargetInventoryLocation = new InventoryLocation;
      if (action_data.m_Player.GetInventory().HasInventoryReservation(am_action_data.m_oldMagazine, am_action_data.m_ilOldMagazine))
        success = false;
      else
        player.GetInventory().AddInventoryReservationEx(am_action_data.m_oldMagazine, am_action_data.m_ilOldMagazine, GameInventory.c_InventoryReservationTimeoutMS);
    }
 
    if (success)
    {
      targetInventoryLocation = new InventoryLocation;
      targetInventoryLocation.SetAttachment(wpn, new_mag, InventorySlots.MAGAZINE);
      if (action_data.m_Player.GetInventory().HasInventoryReservation(new_mag, targetInventoryLocation))
      {
        success = false;
        if (am_action_data.m_oldMagazine)
          player.GetInventory().ClearInventoryReservation(am_action_data.m_oldMagazine, am_action_data.m_ilOldMagazine);
      }
      else
        action_data.m_Player.GetInventory().AddInventoryReservationEx(new_mag, targetInventoryLocation, GameInventory.c_InventoryReservationTimeoutMS);
    }
 
    if (success)
    {
      handInventoryLocation = new InventoryLocation;
      handInventoryLocation.SetHands(action_data.m_Player, wpn);
 
      if (action_data.m_Player.GetInventory().HasInventoryReservation(wpn, handInventoryLocation))
      {
        if (am_action_data.m_oldMagazine)
          player.GetInventory().ClearInventoryReservation(am_action_data.m_oldMagazine, am_action_data.m_ilOldMagazine);
        player.GetInventory().ClearInventoryReservation(new_mag, targetInventoryLocation);
        success = false;
      }
      else
        action_data.m_Player.GetInventory().AddInventoryReservationEx(wpn, handInventoryLocation, GameInventory.c_InventoryReservationTimeoutMS);
    }
 
    if (success)
    {
      if (am_action_data.m_ilOldMagazine)
        action_data.m_ReservedInventoryLocations.Insert(am_action_data.m_ilOldMagazine);
 
      if (targetInventoryLocation)
        action_data.m_ReservedInventoryLocations.Insert(targetInventoryLocation);
 
      if (handInventoryLocation)
        action_data.m_ReservedInventoryLocations.Insert(handInventoryLocation);
    }
 
    return success;
  }
 
  override void Start(ActionData action_data)
  {
    super.Start(action_data);
    AttachMagazineActionData action_data_am = AttachMagazineActionData.Cast(action_data);
    Weapon_Base wpn = Weapon_Base.Cast(action_data.m_MainItem);
    Magazine mag = Magazine.Cast(action_data.m_Target.GetObject());
    ClearInventoryReservationEx(action_data);
    if (action_data.m_Player.GetWeaponManager().CanAttachMagazine(wpn, mag, false))
      action_data.m_Player.GetWeaponManager().AttachMagazine(mag, this);
    else
      action_data.m_Player.GetWeaponManager().SwapMagazineEx(mag, action_data_am.m_ilOldMagazine, this);
    InventoryReservation(action_data);
  }
 
  override bool HasTarget()
  {
    return true;
  }
 
  override bool HasProgress()
  {
    return false;
  }
 
  override typename GetInputType()
  {
    return ContinuousWeaponManipulationActionInput;
  }
 
  override void CreateConditionComponents()
  {
    m_ConditionItem = new CCINonRuined();
    m_ConditionTarget = new CCTSelf;
  }
};