Initial commit

1 files changed, 77 insertions, 0 deletions

diff --git a/Minecraft.World/ThornsEnchantment.cpp b/Minecraft.World/ThornsEnchantment.cpp
new file mode 100644
index 00000000..e2d9f1fa
--- /dev/null
+++ b/Minecraft.World/ThornsEnchantment.cpp
@@ -0,0 +1,77 @@
+#include "stdafx.h"
+#include "net.minecraft.world.item.h"
+#include "net.minecraft.world.item.enchantment.h"
+#include "net.minecraft.world.entity.h"
+#include "net.minecraft.world.damagesource.h"
+#include "ThornsEnchantment.h"
+
+const float ThornsEnchantment::CHANCE_PER_LEVEl = 0.15f;
+
+ThornsEnchantment::ThornsEnchantment(int id, int freq) : Enchantment(id, freq, EnchantmentCategory::armor_torso)
+{
+	setDescriptionId(IDS_ENCHANTMENT_THORNS);
+}
+
+int ThornsEnchantment::getMinCost(int level)
+{
+	return 10 + 20 * (level - 1);
+}
+
+int ThornsEnchantment::getMaxCost(int level)
+{
+	return Enchantment::getMinCost(level) + 50;
+}
+
+int ThornsEnchantment::getMaxLevel()
+{
+	return 3;
+}
+
+bool ThornsEnchantment::canEnchant(shared_ptr<ItemInstance> item)
+{
+	ArmorItem *armor = dynamic_cast<ArmorItem *>(item->getItem());
+	if (armor) return true;
+	return Enchantment::canEnchant(item);
+}
+
+bool ThornsEnchantment::shouldHit(int level, Random *random)
+{
+	if (level <= 0) return false;
+	return random->nextFloat() < CHANCE_PER_LEVEl * level;
+}
+
+int ThornsEnchantment::getDamage(int level, Random *random)
+{
+	if (level > 10)
+	{
+		return level - 10;
+	}
+	else
+	{
+		return 1 + random->nextInt(4);
+	}
+}
+
+void ThornsEnchantment::doThornsAfterAttack(shared_ptr<Entity> source, shared_ptr<Mob> target, Random *random)
+{
+	int level = EnchantmentHelper::getArmorThorns(target);
+	shared_ptr<ItemInstance> item = EnchantmentHelper::getRandomItemWith(Enchantment::thorns, target);
+
+	if (shouldHit(level, random))
+	{
+		source->hurt(DamageSource::thorns(target), getDamage(level, random));
+		source->playSound(eSoundType_DAMAGE_THORNS, .5f, 1.0f);
+
+		if (item != NULL)
+		{
+			item->hurt(3, target);
+		}
+	}
+	else
+	{
+		if (item != NULL)
+		{
+			item->hurt(1, target);
+		}
+	}
+}
\ No newline at end of file