battlearray_client/Assets/Scripts/UI/EquipPanel/EquipManager.cs

using UnityEngine;
using UnityEngine.Events;
using System.Collections;
using System.Collections.Generic;
using System.Xml;
using LitJson;

public class EquipManager
{

	public UnityEvent InventoryUpdate;

	private Dictionary<int, EquipData> dataDict;
	private Dictionary<int, EquipedData> equipedDict;

	private List<Equipment> inventoryList;
	private List<Equipment> hpList;
	private List<Equipment> dmgList;
	private List<Equipment> moveList;

	private bool m_Initialized;
	public bool initialized
	{
		get{ 
			return m_Initialized;
		}
	}

	public EquipManager()
	{
		InventoryUpdate = new UnityEvent ();

		dataDict = new Dictionary<int, EquipData>();
		InitByXML("XML/Config/All/equip_config");

		equipedDict = new Dictionary<int, EquipedData> ();

		inventoryList = new List<Equipment> ();
		hpList = new List<Equipment> ();
		dmgList = new List<Equipment> ();
		moveList = new List<Equipment> ();
	}

	public void InitByXML(string path)
	{
		TextAsset textAsset = (TextAsset)Resources.Load(path);

		XmlDocument xml = new XmlDocument();
		xml.LoadXml(textAsset.text);

		XmlNode mapNode = xml.SelectSingleNode("data");

		XmlNodeList nodeList = mapNode.SelectNodes("item");
		for(int i=0; i<nodeList.Count; i++)
		{
			EquipData data = new EquipData(nodeList[i] as XmlElement);
			if(!dataDict.ContainsKey(data.id))
				dataDict.Add(data.id, data);
			else
				Debuger.LogError("Equipment " + data.id + " has duplicate");
		}
	}

	public EquipData GetData(int id)
	{
		EquipData data = null;
		dataDict.TryGetValue (id, out data);
		return data;
	}









	public void ClearInventory()
	{
		m_Initialized = false;
		UAVManager.GetInstance ().ClearInventory ();
	}

	public void RequestInventory()
	{
		if (m_Initialized) {
			InventoryUpdate.Invoke ();
			return;
		}

		EquipRequest.Inventory ().ResultEvent.AddListener ((bool success, JsonData data)=>{
			if(success)
			{
				InitEquipInventory(data);
			}
		});
	}

	private void InitEquipInventory(JsonData jsonArr)
	{
		inventoryList.Clear ();
		hpList.Clear ();
		dmgList.Clear ();
		moveList.Clear ();

		for(int i=0; i<jsonArr.Count; i++)
		{
			JsonData data = jsonArr [i];
			int id = JsonUtil.ToInt (data["id"]);
			int itemId = JsonUtil.ToInt (data["item_id"]);
			int level = JsonUtil.ToInt (data["level"]);
			int count = JsonUtil.ToInt (data["count"]);

			Equipment equipment = new Equipment (id, GetData (itemId));
			equipment.SetCount (count);
			equipment.SetLevel (level);

			if (equipment.GetData ().equipType == EquipData.EquipType.Equip)
				AddEquipment (equipment);
			else
				UAVManager.GetInstance ().AddUAVItem (equipment);
		}

		SortRecommand ();
		InitEquiped ();

		InventoryUpdate.Invoke ();
	}

	public Equipment GetEquipment(int id)
	{
		for(int i=0; i<inventoryList.Count; i++)
		{
			Equipment equipment = inventoryList [i];
			if(equipment.GetId() == id)
			{
				return equipment;
			}
		}
		return null;
	}

	public Equipment GetEquipmentByItemId(int itemId)
	{
		for(int i=0; i<inventoryList.Count; i++)
		{
			Equipment equipment = inventoryList [i];
			if(equipment.GetData().id == itemId)
			{
				return equipment;
			}
		}
		return null;
	}

	public void RewardEquipment(int id, int itemId, int count)
	{
		bool isNew = false;
		EquipData data = GetData (itemId);

		if (data.equipType == EquipData.EquipType.Equip) {
			Equipment equipment = GetEquipment (id);
			if (equipment == null) {
				equipment = new Equipment (id, GetData (itemId));
				equipment.SetLevel (1);
				isNew = true;
				AddEquipment (equipment);
			}
			equipment.SetCount (count);

			if (isNew) {
				RefreshAllRecommandEquiped ();
			}
		}
		else if(data.equipType == EquipData.EquipType.UAV)
		{
			UAVItem uavItem = UAVManager.GetInstance ().GetUAVItem (itemId);
			if(uavItem == null)
			{
				Equipment equipment = new Equipment (id, GetData (itemId));
				equipment.SetLevel (1);
				isNew = true;
				uavItem = UAVManager.GetInstance ().AddUAVItem (equipment);
			}
			uavItem.GetEquipment ().SetCount (count);
		}
	}

	private void AddEquipment(Equipment equipment)
	{
		inventoryList.Add (equipment);

		if (equipment.HasHpProperty ())
			hpList.Add (equipment);
		if (equipment.HasDamageProperty ())
			dmgList.Add (equipment);

		moveList.Add (equipment);
	}

	private void SortRecommand()
	{
		hpList.Sort (HpCompare);
		dmgList.Sort (DamageCompare);
		moveList.Sort (MoveCompare);

//		Debuger.LogWarning ("");
//		for(int i=0; i<moveList.Count; i++)
//		{
//			Equipment equip = moveList [i];
//			Debuger.LogError ("["+equip.GetData().id+"]:"+equip.GetMoveProperty()+"        "+(equip.GetHpProperty()+equip.GetDamageProperty()));
//		}
//
//		Debuger.LogWarning ("");
//		for(int i=0; i<hpList.Count; i++)
//		{
//			Equipment equip = hpList [i];
//			Debuger.LogError ("["+equip.GetData().id+"]:"+equip.GetHpProperty());
//		}
//
//		Debuger.LogWarning ("");
//		for(int i=0; i<dmgList.Count; i++)
//		{
//			Equipment equip = dmgList [i];
//			Debuger.LogError ("["+equip.GetData().id+"]:"+equip.GetDamageProperty());
//		}
	}

	private static int HpCompare(Equipment a, Equipment b)
	{
		if (a.GetHpProperty () > b.GetHpProperty ())
			return -1;
		else if (a.GetHpProperty () < b.GetHpProperty ())
			return 1;
		return 0;
	}

	private static int DamageCompare(Equipment a, Equipment b)
	{
		if (a.GetDamageProperty () > b.GetDamageProperty ())
			return -1;
		else if (a.GetDamageProperty () < b.GetDamageProperty ())
			return 1;
		return 0;
	}

	private static int MoveCompare(Equipment a, Equipment b)
	{
		if (a.GetMoveProperty () > b.GetMoveProperty ()) {
			return -1;
		}
		else if (a.GetMoveProperty () < b.GetMoveProperty ()) {
			return 1;
		}

		float ap = a.GetHpProperty () + a.GetDamageProperty ();
		float bp = b.GetHpProperty () + b.GetDamageProperty ();
		if (ap > bp)
			return -1;
		else if (ap < bp)
			return 1;
		return 0;
	}





	public List<Equipment> GetInventoryList()
	{
		return inventoryList;
	}













	private void InitEquiped()
	{
		equipedDict.Clear ();

		List<CraftConfigData> list = CraftManager.GetInstance ().GetDataList ();
		for(int i=0; i<list.Count; i++)
		{
			CraftConfigData data = list [i];
			EquipedData equipedData = new EquipedData (data.id);
			equipedDict.Add (data.id, equipedData);
		}

		m_Initialized = true;
	}

	public EquipedData GetEquipedData(int craftId)
	{
		if(equipedDict.ContainsKey(craftId))
			return equipedDict [craftId];
		return null;
	}

	public void RefreshAllRecommandEquiped()
	{
		SortRecommand ();

		List<CraftConfigData> list = CraftManager.GetInstance ().GetDataList ();
		for(int i=0; i<list.Count; i++)
		{
			EquipedData equipedData = GetEquipedData (list[i].id);
			if(equipedData != null)
			{
				equipedData.UpdateRecommandList ();
			}
		}
	}

	public Equipment GetHpPriority(Dictionary<int, Equipment> rcmdDict)
	{
		for(int i=0; i<hpList.Count; i++)
		{
			Equipment equipment = hpList [i];
			if(!rcmdDict.ContainsKey(equipment.GetData().id))
			{
				return equipment;
			}
		}
		return null;
	}

	public Equipment GetDamagePriority(Dictionary<int, Equipment> rcmdDict)
	{
		for(int i=0; i<dmgList.Count; i++)
		{
			Equipment equipment = dmgList [i];
			if(!rcmdDict.ContainsKey(equipment.GetData().id))
			{
				return equipment;
			}
		}
		return null;
	}

	public Equipment GetMovePriority(Dictionary<int, Equipment> rcmdDict)
	{
		for(int i=0; i<moveList.Count; i++)
		{
			Equipment equipment = moveList [i];
			if(!rcmdDict.ContainsKey(equipment.GetData().id))
			{
				return equipment;
			}
		}
		return null;
	}







	private static EquipManager instance;
	public static EquipManager GetInstance()
	{
		if(instance == null)
		{
			instance = new EquipManager ();
		}
		return instance;
	}
}