ResourceManager.cs 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105
  1. using UnityEngine;
  2. using System.Collections.Generic;
  3. using System;
  4. [System.Serializable]
  5. public enum ResourceType
  6. {
  7. Wood,
  8. Stone,
  9. Food,
  10. None
  11. }
  12. [System.Serializable]
  13. public class ResourceAmount
  14. {
  15. public ResourceType type;
  16. public int amount;
  17. public ResourceAmount(ResourceType type, int amount)
  18. {
  19. this.type = type;
  20. this.amount = amount;
  21. }
  22. }
  23. public class ResourceManager : MonoBehaviour
  24. {
  25. [Header("Resources")]
  26. [SerializeField] private Dictionary<ResourceType, int> resources = new Dictionary<ResourceType, int>();
  27. public static event Action<ResourceType, int> OnResourceChanged;
  28. void Awake()
  29. {
  30. // Initialize resources
  31. foreach (ResourceType resourceType in Enum.GetValues(typeof(ResourceType)))
  32. {
  33. resources[resourceType] = 0;
  34. }
  35. }
  36. public void AddResource(ResourceType type, int amount)
  37. {
  38. resources[type] += amount;
  39. OnResourceChanged?.Invoke(type, resources[type]);
  40. Debug.Log($"Added {amount} {type}. Total: {resources[type]}");
  41. }
  42. public bool SpendResource(ResourceType type, int amount)
  43. {
  44. if (resources[type] >= amount)
  45. {
  46. resources[type] -= amount;
  47. OnResourceChanged?.Invoke(type, resources[type]);
  48. Debug.Log($"Spent {amount} {type}. Remaining: {resources[type]}");
  49. return true;
  50. }
  51. return false;
  52. }
  53. public int GetResource(ResourceType type)
  54. {
  55. return resources.ContainsKey(type) ? resources[type] : 0;
  56. }
  57. public void RemoveResource(ResourceType type, int amount)
  58. {
  59. resources[type] = Mathf.Max(0, resources[type] - amount);
  60. OnResourceChanged?.Invoke(type, resources[type]);
  61. Debug.Log($"Removed {amount} {type}. Remaining: {resources[type]}");
  62. }
  63. public bool HasResource(ResourceType type, int amount)
  64. {
  65. return GetResource(type) >= amount;
  66. }
  67. public bool HasResources(List<ResourceAmount> costs)
  68. {
  69. foreach (var cost in costs)
  70. {
  71. if (GetResource(cost.type) < cost.amount)
  72. return false;
  73. }
  74. return true;
  75. }
  76. public bool SpendResources(List<ResourceAmount> costs)
  77. {
  78. if (!HasResources(costs))
  79. return false;
  80. foreach (var cost in costs)
  81. {
  82. SpendResource(cost.type, cost.amount);
  83. }
  84. return true;
  85. }
  86. public Dictionary<ResourceType, int> GetAllResources()
  87. {
  88. return new Dictionary<ResourceType, int>(resources);
  89. }
  90. }