1 files changed, 39 insertions, 4 deletions
diff --git a/Mundus/Service/Tiles/Mobs/LandMobs/Player.cs b/Mundus/Service/Tiles/Mobs/LandMobs/Player.cs
index bf102e4..ddec45a 100644
--- a/Mundus/Service/Tiles/Mobs/LandMobs/Player.cs
+++ b/Mundus/Service/Tiles/Mobs/LandMobs/Player.cs
@@ -1,15 +1,50 @@
-using Mundus.Data.SuperLayers;
+using System;
+using Mundus.Data.SuperLayers;
+using Mundus.Data.Windows;
 using Mundus.Service.Tiles.Items;
 
 namespace Mundus.Service.Tiles.Mobs.LandMobs {
-    public class Player : MobTile {
+    public class Player : MobTile {
+        public double Energy { get; private set; }
+
         /// <summary>
         /// Note: player has an rndMovementQualifier of -1 and drops first item in the hotbar
         /// </summary>
-        public Player(string stock_id, int health, int defence, ISuperLayer currentSuperLayer, int inventorySize)
-               : base(stock_id, health, defence, currentSuperLayer, inventorySize, null, -1) 
+        public Player(string stock_id, int defence, ISuperLayer currentSuperLayer)
+               : base(stock_id, WI.SelWin.Size * 4, defence, currentSuperLayer, WI.SelWin.Size, null, -1) 
         {
+            this.Energy = WI.SelWin.Size * 6;
             this.DroppedUponDeath = (Material)this.Inventory.Hotbar[0];
         }
+
+        /// <summary>
+        /// Removes energy from player. If energy gets below 0 it will start taking health
+        /// </summary>
+        /// <param name="value">Energy points to drain from player (will do nothing if value less than 0</param>
+        public void DrainEnergy(double value) {
+            if (value > 0) {
+                this.Energy -= value;
+
+                if (this.Energy < 0) {
+                    this.TakeDamage((int)Math.Ceiling(Math.Abs(this.Energy)));
+                    this.Energy = 0;
+                }
+            }
+        }
+
+        /// <summary>
+        /// Restores energy from player. If energy is maxed out (WI.SelWin.Size * 6) it starts healing the player
+        /// </summary>
+        /// <param name="value">Energy points to restore energy (will do nothing if value less than 0</param>
+        public void RestoreEnergy(double value) {
+            if (value > 0) {
+                this.Energy += value;
+
+                if (this.Energy > WI.SelWin.Size * 6) {
+                    this.Heal((int)Math.Ceiling(Energy - WI.SelWin.Size * 6));
+                    this.Energy = WI.SelWin.Size * 6;
+                }
+            }
+        }
     }
 }