Code cleanup (upgrade types)

This commit is contained in:
LemADEC 2015-12-06 20:09:06 +01:00
parent fd0074bf76
commit eb1243720e

View file

@ -18,7 +18,7 @@ import cr0s.warpdrive.WarpDrive;
import cr0s.warpdrive.api.IUpgradable; import cr0s.warpdrive.api.IUpgradable;
import cr0s.warpdrive.block.TileEntityAbstractEnergy; import cr0s.warpdrive.block.TileEntityAbstractEnergy;
import cr0s.warpdrive.config.WarpDriveConfig; import cr0s.warpdrive.config.WarpDriveConfig;
import cr0s.warpdrive.data.EnumUpgradeTypes; import cr0s.warpdrive.data.UpgradeType;
import cr0s.warpdrive.data.Vector3; import cr0s.warpdrive.data.Vector3;
import dan200.computercraft.api.lua.ILuaContext; import dan200.computercraft.api.lua.ILuaContext;
import dan200.computercraft.api.peripheral.IComputerAccess; import dan200.computercraft.api.peripheral.IComputerAccess;
@ -341,8 +341,8 @@ public class TileEntityTransporter extends TileEntityAbstractEnergy implements I
private double getLockStrength() { private double getLockStrength() {
if (isLocked) { if (isLocked) {
double upgradeBoost = 1; double upgradeBoost = 1;
if (upgrades.containsKey(EnumUpgradeTypes.Range)) if (upgrades.containsKey(UpgradeType.Range))
upgradeBoost = Math.pow(1.2, upgrades.get(EnumUpgradeTypes.Range)); upgradeBoost = Math.pow(1.2, upgrades.get(UpgradeType.Range));
return clamp(0, 1, baseLockStrength * lockStrengthMul * Math.pow(2, powerBoost - 1) * upgradeBoost * (1 + beaconEffect)); return clamp(0, 1, baseLockStrength * lockStrengthMul * Math.pow(2, powerBoost - 1) * upgradeBoost * (1 + beaconEffect));
} }
return -1; return -1;
@ -412,8 +412,8 @@ public class TileEntityTransporter extends TileEntityAbstractEnergy implements I
@Override @Override
public int getMaxEnergyStored() { public int getMaxEnergyStored() {
int max = WarpDriveConfig.TRANSPORTER_MAX_ENERGY; int max = WarpDriveConfig.TRANSPORTER_MAX_ENERGY;
if (upgrades.containsKey(EnumUpgradeTypes.Energy)) { if (upgrades.containsKey(UpgradeType.Energy)) {
max = (int) Math.floor(max * Math.pow(1.2, upgrades.get(EnumUpgradeTypes.Energy))); max = (int) Math.floor(max * Math.pow(1.2, upgrades.get(UpgradeType.Energy)));
} }
return max; return max;
} }
@ -460,13 +460,13 @@ public class TileEntityTransporter extends TileEntityAbstractEnergy implements I
} }
@Override @Override
public boolean takeUpgrade(EnumUpgradeTypes upgradeType, boolean simulate) { public boolean takeUpgrade(UpgradeType upgradeType, boolean simulate) {
int max = 0; int max = 0;
if (upgradeType == EnumUpgradeTypes.Energy) if (upgradeType == UpgradeType.Energy)
max = 2; max = 2;
else if (upgradeType == EnumUpgradeTypes.Power) else if (upgradeType == UpgradeType.Power)
max = 4; max = 4;
else if (upgradeType == EnumUpgradeTypes.Range) else if (upgradeType == UpgradeType.Range)
max = 4; max = 4;
if (max == 0) if (max == 0)
@ -486,7 +486,7 @@ public class TileEntityTransporter extends TileEntityAbstractEnergy implements I
} }
@Override @Override
public Map<EnumUpgradeTypes, Integer> getInstalledUpgrades() { public Map<UpgradeType, Integer> getInstalledUpgrades() {
return upgrades; return upgrades;
} }
} }