ActuallyAdditions/src/main/java/de/ellpeck/actuallyadditions/mod/tile/TileEntityFireworkBox.java

308 lines
10 KiB
Java

/*
* This file ("TileEntityFireworkBox.java") is part of the Actually Additions mod for Minecraft.
* It is created and owned by Ellpeck and distributed
* under the Actually Additions License to be found at
* http://ellpeck.de/actaddlicense
* View the source code at https://github.com/Ellpeck/ActuallyAdditions
*
* © 2015-2017 Ellpeck
*/
package de.ellpeck.actuallyadditions.mod.tile;
import de.ellpeck.actuallyadditions.mod.inventory.ContainerFireworkBox;
import de.ellpeck.actuallyadditions.mod.network.gui.INumberReactor;
import net.minecraft.entity.player.PlayerEntity;
import net.minecraft.entity.player.PlayerInventory;
import net.minecraft.entity.projectile.FireworkRocketEntity;
import net.minecraft.inventory.container.Container;
import net.minecraft.inventory.container.INamedContainerProvider;
import net.minecraft.item.DyeColor;
import net.minecraft.item.ItemStack;
import net.minecraft.item.Items;
import net.minecraft.nbt.CompoundNBT;
import net.minecraft.nbt.ListNBT;
import net.minecraft.util.Direction;
import net.minecraft.util.WeightedRandom;
import net.minecraft.util.math.BlockPos;
import net.minecraft.util.math.MathHelper;
import net.minecraft.util.text.ITextComponent;
import net.minecraft.util.text.StringTextComponent;
import net.minecraft.world.World;
import net.minecraftforge.common.util.LazyOptional;
import net.minecraftforge.energy.IEnergyStorage;
import javax.annotation.Nullable;
import java.util.ArrayList;
import java.util.List;
public class TileEntityFireworkBox extends TileEntityBase implements IEnergyDisplay, INumberReactor, INamedContainerProvider {
public static final int USE_PER_SHOT = 500;
public final CustomEnergyStorage storage = new CustomEnergyStorage(20000, 200, 0);
public final LazyOptional<IEnergyStorage> lazyEnergy = LazyOptional.of(() -> this.storage);
public int intValuePlay = 2;
public int chargeAmount = 2;
public int flightTime = 2;
public float trailOrFlickerChance = 0.65F;
public float flickerChance = 0.25F;
public int colorAmount = 3;
public float typeChance0 = 1F;
public float typeChance1 = 0F;
public float typeChance2 = 0F;
public float typeChance3 = 0F;
public float typeChance4 = 0F;
public int areaOfEffect = 2;
private int timeUntilNextFirework;
private int oldEnergy;
public TileEntityFireworkBox() {
super(ActuallyTiles.FIREWORKBOX_TILE.get());
}
@Override
public void writeSyncableNBT(CompoundNBT compound, NBTType type) {
super.writeSyncableNBT(compound, type);
this.storage.writeToNBT(compound);
if (type != NBTType.SAVE_BLOCK) {
compound.putInt("Play", this.intValuePlay);
compound.putInt("ChargeAmount", this.chargeAmount);
compound.putInt("FlightTime", this.flightTime);
compound.putFloat("TrailFlickerChance", this.trailOrFlickerChance);
compound.putFloat("FlickerChance", this.flickerChance);
compound.putInt("ColorAmount", this.colorAmount);
compound.putFloat("TypeChance0", this.typeChance0);
compound.putFloat("TypeChance1", this.typeChance1);
compound.putFloat("TypeChance2", this.typeChance2);
compound.putFloat("TypeChance3", this.typeChance3);
compound.putFloat("TypeChance4", this.typeChance4);
compound.putInt("Area", this.areaOfEffect);
}
}
@Override
public void readSyncableNBT(CompoundNBT compound, NBTType type) {
super.readSyncableNBT(compound, type);
this.storage.readFromNBT(compound);
if (type != NBTType.SAVE_BLOCK) {
this.intValuePlay = compound.getInt("Play");
this.chargeAmount = compound.getInt("ChargeAmount");
this.flightTime = compound.getInt("FlightTime");
this.trailOrFlickerChance = compound.getFloat("TrailFlickerChance");
this.flickerChance = compound.getFloat("FlickerChance");
this.colorAmount = compound.getInt("ColorAmount");
this.typeChance0 = compound.getFloat("TypeChance0");
this.typeChance1 = compound.getFloat("TypeChance1");
this.typeChance2 = compound.getFloat("TypeChance2");
this.typeChance3 = compound.getFloat("TypeChance3");
this.typeChance4 = compound.getFloat("TypeChance4");
this.areaOfEffect = compound.getInt("Area");
}
}
@Override
public void onNumberReceived(double number, int id, PlayerEntity player) {
switch (id) {
case 0:
this.intValuePlay = (int) number;
break;
case 1:
this.chargeAmount = (int) number;
break;
case 2:
this.flightTime = (int) number;
break;
case 3:
this.trailOrFlickerChance = (float) number;
break;
case 4:
this.flickerChance = (float) number;
break;
case 5:
this.colorAmount = (int) number;
break;
case 6:
this.typeChance0 = (float) number;
break;
case 7:
this.typeChance1 = (float) number;
break;
case 8:
this.typeChance2 = (float) number;
break;
case 9:
this.typeChance3 = (float) number;
break;
case 10:
this.typeChance4 = (float) number;
break;
case 11:
this.areaOfEffect = (int) number;
break;
}
this.sendUpdate();
}
public void spawnFireworks(World world, double x, double y, double z) {
ItemStack firework = this.makeFirework();
double newX = x + this.getRandomAoe();
double newZ = z + this.getRandomAoe();
if (world.isBlockLoaded(new BlockPos(newX, y, newZ))) {
FireworkRocketEntity rocket = new FireworkRocketEntity(world, newX, y + 1, newZ, firework);
world.addEntity(rocket);
}
}
private double getRandomAoe() {
if (this.areaOfEffect <= 0) {
return 0.5;
} else {
return MathHelper.nextDouble(this.world.rand, 0, this.areaOfEffect * 2) - this.areaOfEffect;
}
}
private ItemStack makeFirework() {
ListNBT list = new ListNBT();
for (int i = 0; i < this.getRandomWithPlay(this.chargeAmount); i++) {
list.add(this.makeFireworkCharge());
}
CompoundNBT compound1 = new CompoundNBT();
compound1.put("Explosions", list);
compound1.putByte("Flight", (byte) this.getRandomWithPlay(this.flightTime));
CompoundNBT compound = new CompoundNBT();
compound.put("Fireworks", compound1);
ItemStack firework = new ItemStack(Items.FIREWORK_ROCKET);
firework.setTag(compound);
return firework;
}
private CompoundNBT makeFireworkCharge() {
CompoundNBT compound = new CompoundNBT();
if (this.world.rand.nextFloat() <= this.trailOrFlickerChance) {
if (this.world.rand.nextFloat() <= this.flickerChance) {
compound.putBoolean("Flicker", true);
} else {
compound.putBoolean("Trail", true);
}
}
// TODO: [port] Validate this is the correct way to get colors
int[] colors = new int[this.getRandomWithPlay(this.colorAmount)];
for (int i = 0; i < colors.length; i++) {
colors[i] = DyeColor.values()[this.world.rand.nextInt(DyeColor.values().length)].getColorValue();
}
compound.putIntArray("Colors", colors);
compound.putByte("Type", (byte) this.getRandomType());
return compound;
}
private int getRandomWithPlay(int value) {
return MathHelper.clamp(MathHelper.nextInt(this.world.rand, value - this.intValuePlay, value + this.intValuePlay), 1, 6);
}
private int getRandomType() {
List<WeightedFireworkType> possible = new ArrayList<>();
possible.add(new WeightedFireworkType(0, this.typeChance0));
possible.add(new WeightedFireworkType(1, this.typeChance1));
possible.add(new WeightedFireworkType(2, this.typeChance2));
possible.add(new WeightedFireworkType(3, this.typeChance3));
possible.add(new WeightedFireworkType(4, this.typeChance4));
int weight = WeightedRandom.getTotalWeight(possible);
if (weight <= 0) {
return 0;
} else {
return WeightedRandom.getRandomItem(this.world.rand, possible, weight).type;
}
}
@Override
public void updateEntity() {
super.updateEntity();
if (!this.world.isRemote) {
if (!this.isRedstonePowered && !this.isPulseMode) {
if (this.timeUntilNextFirework > 0) {
this.timeUntilNextFirework--;
if (this.timeUntilNextFirework <= 0) {
this.doWork();
}
} else {
this.timeUntilNextFirework = 100;
}
}
if (this.oldEnergy != this.storage.getEnergyStored() && this.sendUpdateWithInterval()) {
this.oldEnergy = this.storage.getEnergyStored();
}
}
}
private void doWork() {
if (this.storage.getEnergyStored() >= USE_PER_SHOT) {
this.spawnFireworks(this.world, this.pos.getX(), this.pos.getY(), this.pos.getZ());
this.storage.extractEnergyInternal(USE_PER_SHOT, false);
}
}
@Override
public boolean isRedstoneToggle() {
return true;
}
@Override
public void activateOnPulse() {
this.doWork();
}
@Override
public CustomEnergyStorage getEnergyStorage() {
return this.storage;
}
@Override
public boolean needsHoldShift() {
return false;
}
@Override
public LazyOptional<IEnergyStorage> getEnergyStorage(Direction facing) {
return this.lazyEnergy;
}
@Override
public ITextComponent getDisplayName() {
return StringTextComponent.EMPTY;
}
@Nullable
@Override
public Container createMenu(int windowId, PlayerInventory playerInventory, PlayerEntity p_createMenu_3_) {
return new ContainerFireworkBox(windowId, playerInventory);
}
private static class WeightedFireworkType extends WeightedRandom.Item {
public final int type;
public WeightedFireworkType(int type, float chance) {
super((int) (chance * 100F));
this.type = type;
}
}
}