kalle/superb-warfare
1
0
Fork 0
Code Issues Pull requests Projects Releases 2 Packages Wiki Activity Actions

优化容器载具提取物品能量实现

Browse source
This commit is contained in:
Light_Quanta 2025-01-30 12:06:29 +08:00
parent 222a9707a7
commit 6980e692d0
No known key found for this signature in database
GPG key ID: 11A39A1B8C890959
1 changed files with 15 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
src/main/java/com/atsuishio/superbwarfare/entity/vehicle/ContainerMobileEntity.java
View file

@ -22,7 +22,6 @@ import net.minecraft.world.level.gameevent.GameEvent;
import net.minecraftforge.common.capabilities.Capability; import net.minecraftforge.common.capabilities.Capability;
import net.minecraftforge.common.capabilities.ForgeCapabilities; import net.minecraftforge.common.capabilities.ForgeCapabilities;
import net.minecraftforge.common.util.LazyOptional; import net.minecraftforge.common.util.LazyOptional;
import net.minecraftforge.energy.IEnergyStorage;
import net.minecraftforge.items.wrapper.InvWrapper; import net.minecraftforge.items.wrapper.InvWrapper;
import org.jetbrains.annotations.Nullable; import org.jetbrains.annotations.Nullable;
import org.joml.Math; import org.joml.Math;
@ -82,17 +81,21 @@ public class ContainerMobileEntity extends MobileVehicleEntity implements HasCus
super.baseTick(); super.baseTick();
// pickUpItem(); // pickUpItem();
this.getItemStacks().stream().filter(stack -> stack.is(ModItems.CELL.get()) && stack.getCapability(ForgeCapabilities.ENERGY).map(IEnergyStorage::getEnergyStored).orElse(0) > 0) for (var stack : this.getItemStacks()) {
.forEach(stack -> int neededEnergy = this.getMaxEnergy() - this.getEnergy();
stack.getCapability(ForgeCapabilities.ENERGY).ifPresent(energyStorage -> { if (neededEnergy <= 0) break;
if (this.getEnergy() < this.getMaxEnergy()) {
int energy = this.getMaxEnergy() - this.getEnergy(); var energyCap = stack.getCapability(ForgeCapabilities.ENERGY).resolve();
int stackEnergyNeed = Math.min(energyStorage.getEnergyStored(), energy); if (energyCap.isEmpty()) continue;
energyStorage.extractEnergy(stackEnergyNeed, false);
this.setEnergy(this.getEnergy() + stackEnergyNeed); var energyStorage = energyCap.get();
} var stored = energyStorage.getEnergyStored();
}) if (stored <= 0) continue;
);
int energyToExtract = Math.min(stored, neededEnergy);
energyStorage.extractEnergy(energyToExtract, false);
this.setEnergy(this.getEnergy() + energyToExtract);
}
this.refreshDimensions(); this.refreshDimensions();
} }