WTP, SourceTree?

src/main/java/WayofTime/bloodmagic/livingArmour/tracker/StatTrackerStepAssist.java

@@ -0,0 +1,111 @@
+package WayofTime.bloodmagic.livingArmour.tracker;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+import net.minecraft.entity.player.EntityPlayer;
+import net.minecraft.nbt.NBTTagCompound;
+import net.minecraft.world.World;
+import WayofTime.bloodmagic.api.Constants;
+import WayofTime.bloodmagic.api.livingArmour.LivingArmourUpgrade;
+import WayofTime.bloodmagic.api.livingArmour.StatTracker;
+import WayofTime.bloodmagic.livingArmour.LivingArmour;
+import WayofTime.bloodmagic.livingArmour.upgrade.LivingArmourUpgradeStepAssist;
+
+public class StatTrackerStepAssist extends StatTracker
+{
+    public static Map<EntityPlayer, Double> lastPosX = new HashMap<EntityPlayer, Double>();
+    public static Map<EntityPlayer, Double> lastPosZ = new HashMap<EntityPlayer, Double>();
+
+    public static int blocksRequired = 1000;
+
+    public double totalMovement = 0;
+
+    @Override
+    public String getUniqueIdentifier()
+    {
+        return Constants.Mod.MODID + ".tracker.stepAssist";
+    }
+
+    @Override
+    public void resetTracker()
+    {
+        this.totalMovement = 0;
+    }
+
+    @Override
+    public void readFromNBT(NBTTagCompound tag)
+    {
+        totalMovement = tag.getDouble(Constants.Mod.MODID + ".tracker.stepAssist");
+    }
+
+    @Override
+    public void writeToNBT(NBTTagCompound tag)
+    {
+        tag.setDouble(Constants.Mod.MODID + ".tracker.stepAssist", totalMovement);
+
+    }
+
+    @Override
+    public boolean onTick(World world, EntityPlayer player, LivingArmour livingArmour)
+    {
+        if (!lastPosX.containsKey(player))
+        {
+            lastPosX.put(player, player.posX);
+            lastPosZ.put(player, player.posZ);
+            return false;
+        }
+
+        if (!player.onGround || player.stepHeight < 1)
+        {
+            return false;
+        }
+
+        double distanceTravelled = Math.sqrt(Math.pow(lastPosX.get(player) - player.posX, 2) + Math.pow(lastPosZ.get(player) - player.posZ, 2));
+
+        if (distanceTravelled > 0.0001 && distanceTravelled < 2)
+        {
+            double previousMovement = totalMovement;
+            totalMovement += distanceTravelled;
+
+            lastPosX.put(player, player.posX);
+            lastPosZ.put(player, player.posZ);
+
+            markDirty();
+
+            return previousMovement < blocksRequired && totalMovement >= blocksRequired;
+        }
+
+        lastPosX.put(player, player.posX);
+        lastPosZ.put(player, player.posZ);
+
+        return false;
+    }
+
+    @Override
+    public void onDeactivatedTick(World world, EntityPlayer player, LivingArmour livingArmour)
+    {
+
+    }
+
+    @Override
+    public List<LivingArmourUpgrade> getUpgrades()
+    {
+        List<LivingArmourUpgrade> upgradeList = new ArrayList<LivingArmourUpgrade>();
+
+        if (totalMovement >= blocksRequired)
+        {
+            upgradeList.add(new LivingArmourUpgradeStepAssist(0));
+        }
+
+        return upgradeList;
+    }
+
+    @Override
+    public boolean providesUpgrade(String key)
+    {
+        return key.equals(Constants.Mod.MODID + ".upgrade.stepAssist");
+    }
+}