107 lines
4.1 KiB
Java
107 lines
4.1 KiB
Java
package WayofTime.bloodmagic.recipe.alchemyTable;
|
|
|
|
import java.util.*;
|
|
|
|
import net.minecraft.item.ItemStack;
|
|
import net.minecraft.potion.Potion;
|
|
import net.minecraft.potion.PotionEffect;
|
|
import net.minecraft.potion.PotionUtils;
|
|
import WayofTime.bloodmagic.potion.BMPotionUtils;
|
|
import WayofTime.bloodmagic.registry.ModItems;
|
|
|
|
public class AlchemyTablePotionAugmentRecipe extends AlchemyTablePotionRecipe
|
|
{
|
|
protected double lengthAugment = 0;
|
|
protected int powerAugment = 0;
|
|
protected Potion wantedPotion;
|
|
|
|
public AlchemyTablePotionAugmentRecipe(int lpDrained, int ticksRequired, int tierRequired, List<ItemStack> inputItems, PotionEffect baseEffect, double lengthAugment, int powerAugment)
|
|
{
|
|
super(lpDrained, ticksRequired, tierRequired, inputItems, baseEffect);
|
|
|
|
ArrayList<Object> recipe = new ArrayList<Object>();
|
|
for (ItemStack stack : inputItems)
|
|
{
|
|
recipe.add(stack);
|
|
}
|
|
recipe.add(getAugmentedPotionFlask(baseEffect));
|
|
|
|
this.input = recipe;
|
|
|
|
this.wantedPotion = baseEffect.getPotion();
|
|
this.lengthAugment = lengthAugment;
|
|
this.powerAugment = powerAugment;
|
|
}
|
|
|
|
public AlchemyTablePotionAugmentRecipe(int lpDrained, int ticksRequired, int tierRequired, ItemStack inputItem, PotionEffect baseEffect, double lengthAugment, int powerAugment)
|
|
{
|
|
this(lpDrained, ticksRequired, tierRequired, Collections.singletonList(inputItem), baseEffect, lengthAugment, powerAugment);
|
|
}
|
|
|
|
@Override
|
|
public boolean isPotionFlaskValidInput(ItemStack stack)
|
|
{
|
|
List<PotionEffect> effectList = PotionUtils.getEffectsFromStack(stack);
|
|
for (PotionEffect eff : effectList)
|
|
{
|
|
if (eff.getPotion() == wantedPotion)
|
|
{
|
|
double currentAugment = BMPotionUtils.getLengthAugment(stack, wantedPotion);
|
|
|
|
return currentAugment < lengthAugment || eff.getAmplifier() < powerAugment;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
@Override
|
|
public ItemStack getModifiedFlaskForInput(ItemStack inputStack)
|
|
{
|
|
if (inputStack == null)
|
|
{
|
|
ItemStack outputStack = new ItemStack(ModItems.POTION_FLASK);
|
|
|
|
List<PotionEffect> effectList = new ArrayList<PotionEffect>();
|
|
int potionLength = wantedPotion.isInstant() ? 1 : BMPotionUtils.getAugmentedLength(baseEffect.getDuration(), lengthAugment, powerAugment - baseEffect.getAmplifier());
|
|
effectList.add(new PotionEffect(wantedPotion, potionLength, powerAugment - baseEffect.getAmplifier()));
|
|
|
|
BMPotionUtils.setEffects(outputStack, effectList);
|
|
|
|
return outputStack;
|
|
}
|
|
|
|
ItemStack outputStack = inputStack.copy();
|
|
|
|
List<PotionEffect> effectList = PotionUtils.getEffectsFromStack(outputStack);
|
|
List<PotionEffect> newEffectList = new ArrayList<PotionEffect>();
|
|
|
|
for (PotionEffect effect : effectList) {
|
|
if (effect.getPotion() == wantedPotion) {
|
|
double currentLengthAugment = Math.max(lengthAugment, BMPotionUtils.getLengthAugment(outputStack, wantedPotion));
|
|
int currentPowerAugment = Math.max(powerAugment, effect.getAmplifier());
|
|
int potionLength = wantedPotion.isInstant() ? 1 : BMPotionUtils.getAugmentedLength(baseEffect.getDuration(), currentLengthAugment, currentPowerAugment);
|
|
newEffectList.add(new PotionEffect(wantedPotion, potionLength, currentPowerAugment));
|
|
BMPotionUtils.setLengthAugment(outputStack, wantedPotion, currentLengthAugment);
|
|
} else {
|
|
newEffectList.add(effect);
|
|
}
|
|
}
|
|
|
|
BMPotionUtils.setEffects(outputStack, newEffectList);
|
|
|
|
return outputStack;
|
|
}
|
|
|
|
public static ItemStack getAugmentedPotionFlask(PotionEffect baseEffect)
|
|
{
|
|
ItemStack outputStack = new ItemStack(ModItems.POTION_FLASK);
|
|
|
|
List<PotionEffect> effectList = new ArrayList<PotionEffect>();
|
|
effectList.add(baseEffect);
|
|
|
|
BMPotionUtils.setEffects(outputStack, effectList);
|
|
|
|
return outputStack;
|
|
}
|
|
}
|