InventoryQuickbar.c

Go to the documentation of this file.

class InventoryQuickbar extends InventoryGridController
{
 
  protected ref TItemsMap m_Items;
  protected InventoryGrid m_Grid;
  protected int m_DraggingIndex;
 
  void InventoryQuickbar(Widget quickbarGridWidget)
  {
    m_DraggingIndex = INDEX_NOT_FOUND;
    m_Items = new TItemsMap;
    UpdateItems(quickbarGridWidget);
  }
 
  void ~InventoryQuickbar()
  {
  }
 
  void UpdateItems(Widget quickbarGridWidget)
  {
    if (!quickbarGridWidget)
      return;
 
    PlayerBase player = PlayerBase.Cast(GetGame().GetPlayer());
    if (!player)
      return;
 
    int i;
 
    // create grid and align it to center
    if (!m_Grid)
    {
      quickbarGridWidget.GetScript(m_Grid);
      m_Grid.SetController(this);
      m_Grid.GenerateQuickBarBackgroundTiles(10);
    }
 
    m_Items.Clear();
    InventoryItem item;
 
    if (m_Grid)
    {
      if (m_Grid.GetGridSize() != player.GetQuickBarSize())
        m_Grid.SetGridSize(player.GetQuickBarSize());
    }
 
    for (i = 0; i < m_Grid.GetGridSize(); i++)
    {
      item = InventoryItem.Cast(player.GetQuickBarEntity(i));
      if (item)
        m_Items.Set(item, Vector(i, 1, 1));
    }
 
    if (m_Grid)
      m_Grid.UpdateQuickbarItems(m_Items);
  }
 
  void Remove(InventoryItem itemToRemove)
  {
    PlayerBase player = PlayerBase.Cast(GetGame().GetPlayer());
 
    for (int i = 0; i < m_Grid.GetGridSize(); i++)
    {
      InventoryItem item = InventoryItem.Cast(player.GetQuickBarEntity(i));
      if (item == itemToRemove)
        player.RemoveQuickBarEntityShortcut(item);
    }
  }
 
  // InventoryGridController override
  override void OnItemEnter(InventoryGrid grid, Widget w, int row, int col)
  {
    Widget quickbar = grid.GetRoot();
    if (quickbar)
    {
      // quickbar.SetColor( 0xFFFFFFFF );
    }
  }
 
  override void OnItemLeave(InventoryGrid grid, Widget w)
  {
    Widget quickbar = grid.GetRoot();
    if (quickbar)
    {
      // quickbar.SetColor( 0x7FFFFFFF );
    }
  }
 
  override void OnItemDraggingOver(InventoryGrid grid, Widget w, int row, int col)
  {
    ItemPreviewWidget iw;
    InventoryItem dragged_item;
    iw = ItemPreviewWidget.Cast(w.FindAnyWidget("Render"));
    if (!iw)
    {
      string name = w.GetName();
      name.Replace("PanelWidget", "Render");
      iw = ItemPreviewWidget.Cast(w.FindAnyWidget(name));
    }
 
    if (!iw || !iw.GetItem())
      return;
 
    if (!dragged_item && iw)
      dragged_item = InventoryItem.Cast(iw.GetItem());
 
    if (dragged_item == NULL)
      return;
 
    if (dragged_item && dragged_item.GetInventory().CanRemoveEntity() == false)
    {
      int color;
 
      color = ARGB(150, 255, 1, 1);
 
      if (w.FindAnyWidget("Cursor"))
        w.FindAnyWidget("Cursor").SetColor(color);
      else
      {
        name = w.GetName();
        name.Replace("PanelWidget", "Cursor");
        if (w.FindAnyWidget(name))
          w.FindAnyWidget(name).SetColor(color);
      }
    }
    else
    {
      color = ARGB(150, 1, 255, 1);
 
      if (w.FindAnyWidget("Cursor"))
        w.FindAnyWidget("Cursor").SetColor(color);
      else
      {
        name = w.GetName();
        name.Replace("PanelWidget", "Cursor");
        if (w.FindAnyWidget(name))
          w.FindAnyWidget(name).SetColor(color);
      }
    }
  }
 
  override void OnItemDropReceived(InventoryGrid grid, Widget w, int row, int col)
  {
    PlayerBase player = PlayerBase.Cast(GetGame().GetPlayer());
 
    ItemBase dragged_item = ItemBase.Cast(ItemManager.GetInstance().GetDraggedItem());
 
    ItemPreviewWidget iw = ItemPreviewWidget.Cast(w.FindAnyWidget("Render"));
    if (!iw)
    {
      string name = w.GetName();
      name.Replace("PanelWidget", "Render");
      iw = ItemPreviewWidget.Cast(w.FindAnyWidget(name));
    }
 
    if (!dragged_item && iw)
      dragged_item = ItemBase.Cast(iw.GetItem());
 
    if (dragged_item && dragged_item.GetHierarchyRootPlayer() != GetGame().GetPlayer())
      return;
 
    ItemBase quickbar_item = ItemBase.Cast(player.GetQuickBarEntity(col));
 
    if (m_DraggingIndex != INDEX_NOT_FOUND)
      dragged_item = ItemBase.Cast(player.GetQuickBarEntity(m_DraggingIndex));
 
    if (dragged_item && dragged_item.GetInventory().CanRemoveEntity())
    {
      player.SetQuickBarEntityShortcut(dragged_item, col);
 
      if (quickbar_item && quickbar_item != dragged_item && m_DraggingIndex != INDEX_NOT_FOUND)
        player.SetQuickBarEntityShortcut(quickbar_item, m_DraggingIndex);
    }
    m_DraggingIndex = INDEX_NOT_FOUND;
    InventoryMenu menu = InventoryMenu.Cast(GetGame().GetUIManager().FindMenu(MENU_INVENTORY));
    if (menu)
      menu.RefreshQuickbar();
  }
 
  override void OnItemDrop(InventoryGrid grid, Widget w, int row, int col)
  {
    if (m_DraggingIndex != INDEX_NOT_FOUND)
    {
      PlayerBase player = PlayerBase.Cast(GetGame().GetPlayer());
      InventoryItem dragged_item = InventoryItem.Cast(player.GetQuickBarEntity(m_DraggingIndex));
      Remove(dragged_item);
      m_DraggingIndex = INDEX_NOT_FOUND;
    }
 
    InventoryMenu menu = InventoryMenu.Cast(GetGame().GetUIManager().FindMenu(MENU_INVENTORY));
    if (menu)
      menu.RefreshQuickbar();
    ItemManager.GetInstance().HideDropzones();
  }
 
  override void OnItemDrag(InventoryGrid grid, Widget w, int row, int col)
  {
    PlayerBase player = PlayerBase.Cast(GetGame().GetPlayer());
    InventoryItem item = InventoryItem.Cast(player.GetQuickBarEntity(col));
    if (item)
    {
      m_DraggingIndex = col;
      m_Grid.SetItemColor(item, InventoryGrid.ITEM_COLOR_DRAG);
    }
  }
 
  override int GetQuickbarItemColor(InventoryGrid grid, InventoryItem item)
  {
    int color = 0x0AFFFFFF;
    PlayerBase player = PlayerBase.Cast(GetGame().GetPlayer());
    EntityAI itemInHand = player.GetHumanInventory().GetEntityInHands();
    if (player && itemInHand == item)
    {
      if (player.GetHumanInventory().FindUserReservedLocationIndex(item) != -1 || grid.CanAddItemInHandToInventory())
        color = InventoryGrid.ITEM_COLOR_QUICKBAR_H_GOOD;
      else
        color = InventoryGrid.ITEM_COLOR_QUICKBAR_H_BAD;
    }
    else if (itemInHand)
    {
      if (!grid.CanAddItemInHandToInventory() && !GameInventory.CanSwapEntitiesEx(item, itemInHand))
        color = InventoryGrid.ITEM_COLOR_QUICKBAR_I_BAD;
    }
    return color;
  }
 
 
  override int HasItemQuantity(InventoryItem item)
  {
    return QuantityConversions.HasItemQuantity(item);
  }
 
  override string GetItemQuantityText(InventoryItem item)
  {
    return QuantityConversions.GetItemQuantityText(item);
  }
 
  override float GetItemQuantity(InventoryItem item)
  {
    return QuantityConversions.GetItemQuantity(item);
  }
 
  override int GetItemQuantityMax(InventoryItem item)
  {
    ItemBase ib = ItemBase.Cast(item);
    return ib.m_VarQuantityMax;
  }
 
  override int GetItemCount(InventoryItem item)
  {
    ItemBase ib = ItemBase.Cast(item);
    return ib.m_Count;
  }
 
  override bool CanAddItemInHandToInventory()
  {
    PlayerBase player = PlayerBase.Cast(GetGame().GetPlayer());
    EntityAI itemInHand = player.GetHumanInventory().GetEntityInHands();
    return itemInHand && player.GetInventory().CanAddEntityToInventory(itemInHand);
  }
}