cuberite-2a/src/Items/ItemPotion.h

#pragma once

#include "../Entities/EntityEffects.h"
#include "../Entities/SplashPotionEntity.h"

class cItemPotionHandler:
	public cItemHandler
{
	typedef cItemHandler super;
	
	int GetPotionName(short a_ItemDamage)
	{
		return a_ItemDamage & 63;
	}
	
	cEntityEffect::eType GetEntityEffectType(short a_ItemDamage)
	{
		// Potion effect bits are different from entity effect values
		// For reference: http://minecraft.gamepedia.com/Data_values#.22Potion_effect.22_bits
		switch (a_ItemDamage & 15)
		{
			case 1:  return cEntityEffect::effRegeneration;
			case 2:  return cEntityEffect::effSpeed;
			case 3:  return cEntityEffect::effFireResistance;
			case 4:  return cEntityEffect::effPoison;
			case 5:  return cEntityEffect::effInstantHealth;
			case 6:  return cEntityEffect::effNightVision;
			case 8:  return cEntityEffect::effWeakness;
			case 9:  return cEntityEffect::effStrength;
			case 10: return cEntityEffect::effSlowness;
			case 12: return cEntityEffect::effInstantDamage;
			case 13: return cEntityEffect::effWaterBreathing;
			case 14: return cEntityEffect::effInvisibility;
			
			// No effect potions
			case 0:
			case 7:
			case 11:
			case 15:
			{
				break;
			}
		}
		
		return cEntityEffect::effNoEffect;
	}
	
	short GetEntityEffectIntensity(short a_ItemDamage)
	{
		// Level II potion if fifth bit is set
		if (a_ItemDamage & 32) return 1;
		else                   return 0;
	}
	
	int GetEntityEffectDuration(short a_ItemDamage)
	{
		// Base duration in ticks
		int base = 0;
		double TierCoeff = 1, ExtCoeff = 1, SplashCoeff = 1;
		
		switch (GetEntityEffectType(a_ItemDamage))
		{
			case cEntityEffect::effRegeneration:
			case cEntityEffect::effPoison:
			{
				base = 900;
				break;
			}
				
			case cEntityEffect::effSpeed:
			case cEntityEffect::effFireResistance:
			case cEntityEffect::effNightVision:
			case cEntityEffect::effStrength:
			case cEntityEffect::effWaterBreathing:
			case cEntityEffect::effInvisibility:
			{
				base = 3600;
				break;
			}
				
			case cEntityEffect::effWeakness:
			case cEntityEffect::effSlowness:
			{
				base = 1800;
				break;
			}
		}
		
		// If potion is level 2, half the duration. If not, stays the same
		TierCoeff = (GetEntityEffectIntensity(a_ItemDamage) > 0) ? 0.5 : 1;
		
		// If potion is extended, multiply duration by 8/3. If not, stays the same
		// Extended potion if sixth bit is set
		ExtCoeff = (a_ItemDamage & 64) ? (8.0/3.0) : 1;
		
		// If potion is splash potion, multiply duration by 3/4. If not, stays the same
		SplashCoeff = IsDrinkable(a_ItemDamage) ? 1 : 0.75;
		
		// For reference: http://minecraft.gamepedia.com/Data_values#.22Tier.22_bit
		//                http://minecraft.gamepedia.com/Data_values#.22Extended_duration.22_bit
		//                http://minecraft.gamepedia.com/Data_values#.22Splash_potion.22_bit
		
		return (int)(base * TierCoeff * ExtCoeff * SplashCoeff);
	}
	
public:
	cItemPotionHandler():
	super(E_ITEM_POTIONS)
	{
	}
	
	virtual bool IsDrinkable(short a_ItemDamage) override
	{
		// Drinkable potion if 13th bit is set
		// For reference: http://minecraft.gamepedia.com/Potions#Data_value_table
		return ((a_ItemDamage & 8192) != 0);
	}
	
	virtual bool OnItemUse(cWorld * a_World, cPlayer * a_Player, const cItem & a_Item, int a_BlockX, int a_BlockY, int a_BlockZ, eBlockFace a_Dir) override
	{
		Vector3d Pos = a_Player->GetThrowStartPos();
		Vector3d Speed = a_Player->GetLookVector() * 7;
		
		short potion_damage = a_Item.m_ItemDamage;
		cSplashPotionEntity *Projectile = new cSplashPotionEntity(a_Player, Pos.x, Pos.y, Pos.z, &Speed, GetEntityEffectType(potion_damage), cEntityEffect(GetEntityEffectDuration(potion_damage), GetEntityEffectIntensity(potion_damage), a_Player), GetPotionName(potion_damage));
		if (Projectile == NULL)
		{
			return false;
		}
		if (!Projectile->Initialize(*a_World))
		{
			delete Projectile;
			return false;
		}
		
		if (!a_Player->IsGameModeCreative())
		{
			a_Player->GetInventory().RemoveOneEquippedItem();
		}
		
		// Called when potion is a splash potion
		return true;
	}
	
	virtual bool EatItem(cPlayer * a_Player, cItem * a_Item) override
	{
		// Called when potion is a drinkable potion
		a_Player->AddEntityEffect(GetEntityEffectType(a_Item->m_ItemDamage), GetEntityEffectDuration(a_Item->m_ItemDamage), GetEntityEffectIntensity(a_Item->m_ItemDamage), a_Player);
		a_Player->GetInventory().RemoveOneEquippedItem();
		a_Player->GetInventory().AddItem(E_ITEM_GLASS_BOTTLE);
		return true;
	}
};