l2j_mobius/trunk/dist/game/data/scripts/handlers/effecthandlers/RebalanceHP.java

/*
 * This file is part of the L2J Mobius project.
 * 
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */
package handlers.effecthandlers;

import com.l2jmobius.gameserver.model.L2Party;
import com.l2jmobius.gameserver.model.StatsSet;
import com.l2jmobius.gameserver.model.actor.L2Character;
import com.l2jmobius.gameserver.model.actor.L2Summon;
import com.l2jmobius.gameserver.model.actor.instance.L2PcInstance;
import com.l2jmobius.gameserver.model.effects.AbstractEffect;
import com.l2jmobius.gameserver.model.effects.L2EffectType;
import com.l2jmobius.gameserver.model.items.instance.L2ItemInstance;
import com.l2jmobius.gameserver.model.skills.Skill;
import com.l2jmobius.gameserver.util.Util;

/**
 * Rebalance HP effect implementation.
 * @author Adry_85, earendil
 */
public final class RebalanceHP extends AbstractEffect
{
	public RebalanceHP(StatsSet params)
	{
	}
	
	@Override
	public L2EffectType getEffectType()
	{
		return L2EffectType.REBALANCE_HP;
	}
	
	@Override
	public boolean isInstant()
	{
		return true;
	}
	
	@Override
	public void instant(L2Character effector, L2Character effected, Skill skill, L2ItemInstance item)
	{
		if (!effector.isPlayer())
		{
			return;
		}
		
		double fullHP = 0;
		double currentHPs = 0;
		final L2Party party = effector.getParty();
		
		switch (skill.getTargetType())
		{
			case MY_PARTY:
			{
				if (party != null)
				{
					for (L2PcInstance member : party.getMembers())
					{
						if (!member.isDead() && Util.checkIfInRange(skill.getAffectRange(), effector, member, true))
						{
							fullHP += member.getMaxHp();
							currentHPs += member.getCurrentHp();
						}
						
						final L2Summon summon = member.getPet();
						if ((summon != null) && (!summon.isDead() && Util.checkIfInRange(skill.getAffectRange(), effector, summon, true)))
						{
							fullHP += summon.getMaxHp();
							currentHPs += summon.getCurrentHp();
						}
						
						for (L2Summon servitors : member.getServitors().values())
						{
							fullHP += servitors.getMaxHp();
							currentHPs += servitors.getCurrentHp();
						}
					}
					
					final double percentHP = currentHPs / fullHP;
					for (L2PcInstance member : party.getMembers())
					{
						if (!member.isDead() && Util.checkIfInRange(skill.getAffectRange(), effector, member, true))
						{
							double newHP = member.getMaxHp() * percentHP;
							if (newHP > member.getCurrentHp()) // The target gets healed
							{
								// The heal will be blocked if the current hp passes the limit
								if (member.getCurrentHp() > member.getMaxRecoverableHp())
								{
									newHP = member.getCurrentHp();
								}
								else if (newHP > member.getMaxRecoverableHp())
								{
									newHP = member.getMaxRecoverableHp();
								}
							}
							
							member.setCurrentHp(newHP);
						}
						
						final L2Summon summon = member.getPet();
						if ((summon != null) && (!summon.isDead() && Util.checkIfInRange(skill.getAffectRange(), effector, summon, true)))
						{
							double newHP = summon.getMaxHp() * percentHP;
							if (newHP > summon.getCurrentHp()) // The target gets healed
							{
								// The heal will be blocked if the current hp passes the limit
								if (summon.getCurrentHp() > summon.getMaxRecoverableHp())
								{
									newHP = summon.getCurrentHp();
								}
								else if (newHP > summon.getMaxRecoverableHp())
								{
									newHP = summon.getMaxRecoverableHp();
								}
							}
							summon.setCurrentHp(newHP);
						}
						
						for (L2Summon servitors : member.getServitors().values())
						{
							if (!servitors.isDead() && Util.checkIfInRange(skill.getAffectRange(), effector, summon, true))
							{
								double newHP = servitors.getMaxHp() * percentHP;
								if (newHP > servitors.getCurrentHp()) // The target gets healed
								{
									// The heal will be blocked if the current hp passes the limit
									if (servitors.getCurrentHp() > servitors.getMaxRecoverableHp())
									{
										newHP = servitors.getCurrentHp();
									}
									else if (newHP > servitors.getMaxRecoverableHp())
									{
										newHP = servitors.getMaxRecoverableHp();
									}
								}
								servitors.setCurrentHp(newHP);
							}
						}
					}
				}
				break;
			}
			case SUMMON:
			{
				for (L2Summon summon : effector.getServitors().values())
				{
					if (!summon.isDead() && Util.checkIfInRange(skill.getAffectRange(), effector, summon, true))
					{
						fullHP += summon.getMaxHp();
						currentHPs += summon.getCurrentHp();
					}
				}
				
				final double percentHP = currentHPs / fullHP;
				for (L2Summon summon : effector.getServitors().values())
				{
					if (!summon.isDead() && Util.checkIfInRange(skill.getAffectRange(), effector, summon, true))
					{
						double newHP = summon.getMaxHp() * percentHP;
						if (newHP > summon.getCurrentHp()) // The target gets healed
						{
							// The heal will be blocked if the current hp passes the limit
							if (summon.getCurrentHp() > summon.getMaxRecoverableHp())
							{
								newHP = summon.getCurrentHp();
							}
							else if (newHP > summon.getMaxRecoverableHp())
							{
								newHP = summon.getMaxRecoverableHp();
							}
						}
						
						summon.setCurrentHp(newHP);
					}
				}
			}
		}
		
	}
}