84 lines
4.0 KiB
Java
84 lines
4.0 KiB
Java
package org.KaiFlo.SolarCell.Systems;
|
|
|
|
import com.hypixel.hytale.component.*;
|
|
import com.hypixel.hytale.component.query.Query;
|
|
import com.hypixel.hytale.component.system.tick.EntityTickingSystem;
|
|
import com.hypixel.hytale.math.util.ChunkUtil;
|
|
import com.hypixel.hytale.protocol.Vector3i;
|
|
import com.hypixel.hytale.server.core.asset.type.blocktick.BlockTickStrategy;
|
|
import com.hypixel.hytale.server.core.universe.world.chunk.BlockComponentChunk;
|
|
import com.hypixel.hytale.server.core.universe.world.chunk.WorldChunk;
|
|
import com.hypixel.hytale.server.core.universe.world.chunk.section.BlockSection;
|
|
import com.hypixel.hytale.server.core.universe.world.chunk.section.ChunkSection;
|
|
import com.hypixel.hytale.server.core.universe.world.storage.ChunkStore;
|
|
import org.checkerframework.checker.nullness.compatqual.NonNullDecl;
|
|
import org.checkerframework.checker.nullness.compatqual.NullableDecl;
|
|
|
|
import java.util.*;
|
|
import java.util.concurrent.atomic.AtomicBoolean;
|
|
|
|
public class EnergyTickingSystem extends EntityTickingSystem<ChunkStore> {
|
|
|
|
private final Map<List<ComponentType<ChunkStore, ?>>, ITickingSystem> componentsToTickingSystem = new HashMap<>();
|
|
|
|
|
|
@Override
|
|
public void tick(float v, int archetypeIndex, @NonNullDecl ArchetypeChunk<ChunkStore> archetypeChunk, @NonNullDecl Store<ChunkStore> store, @NonNullDecl CommandBuffer<ChunkStore> commandBuffer) {
|
|
var blockSection = archetypeChunk.getComponent(archetypeIndex, BlockSection.getComponentType());
|
|
if (blockSection == null) return;
|
|
|
|
var chunkSection = archetypeChunk.getComponent(archetypeIndex, ChunkSection.getComponentType());
|
|
if (chunkSection == null) return;
|
|
|
|
var blockComponentChunk = commandBuffer.getComponent(chunkSection.getChunkColumnReference(), BlockComponentChunk.getComponentType());
|
|
var worldChunk = commandBuffer.getComponent(chunkSection.getChunkColumnReference(), WorldChunk.getComponentType());
|
|
if (blockComponentChunk == null || worldChunk == null) return;
|
|
|
|
var entrySet = componentsToTickingSystem.entrySet();
|
|
var foundComponentTypes = new ArrayList<ComponentType<ChunkStore, ?>>();
|
|
|
|
blockSection.forEachTicking(null, null, chunkSection.getY(), (_, _, localX, localY, localZ, _) -> {
|
|
var blockRef = blockComponentChunk.getEntityReference(ChunkUtil.indexBlockInColumn(localX, localY, localZ));
|
|
if (blockRef == null) {
|
|
return BlockTickStrategy.IGNORED;
|
|
}
|
|
|
|
int globalX = localX + (worldChunk.getX() * 32);
|
|
int globalZ = localZ + (worldChunk.getZ() * 32);
|
|
var globalPosition = new Vector3i(globalX, localY, globalZ);
|
|
|
|
var archetype = commandBuffer.getArchetype(blockRef);
|
|
|
|
foundComponentTypes.clear();
|
|
var foundComponents = new ArrayList<Component<ChunkStore>>();
|
|
for (int i = 0; i < archetype.length(); i++) {
|
|
var type = archetype.get(i);
|
|
if (type == null) continue;
|
|
foundComponentTypes.add(type);
|
|
foundComponents.add(commandBuffer.getComponent(blockRef, type));
|
|
}
|
|
|
|
AtomicBoolean hasAny = new AtomicBoolean(false);
|
|
entrySet.stream()
|
|
.filter(entry -> foundComponentTypes.containsAll(entry.getKey())).map(Map.Entry::getValue)
|
|
.forEach(tickingSystem -> {
|
|
hasAny.set(true);
|
|
tickingSystem.accept(blockRef, foundComponents, archetype, globalPosition, blockComponentChunk, commandBuffer, worldChunk.getWorld());
|
|
});
|
|
return hasAny.get() ? BlockTickStrategy.CONTINUE : BlockTickStrategy.IGNORED;
|
|
});
|
|
|
|
}
|
|
|
|
@NullableDecl
|
|
@Override
|
|
public Query<ChunkStore> getQuery() {
|
|
return Query.and(BlockSection.getComponentType(), ChunkSection.getComponentType());
|
|
}
|
|
|
|
public EnergyTickingSystem withTickingSystemForComponentTypes(List<ComponentType<ChunkStore, ?>> componentTypes, ITickingSystem tickingSystem) {
|
|
componentsToTickingSystem.put(componentTypes, tickingSystem);
|
|
return this;
|
|
}
|
|
}
|