116 lines
3.4 KiB
Java
116 lines
3.4 KiB
Java
package carpet.mixins;
|
|
|
|
import java.util.Arrays;
|
|
|
|
import org.spongepowered.asm.mixin.Final;
|
|
import org.spongepowered.asm.mixin.Mixin;
|
|
import org.spongepowered.asm.mixin.Shadow;
|
|
import org.spongepowered.asm.mixin.Unique;
|
|
import org.spongepowered.asm.mixin.injection.At;
|
|
import org.spongepowered.asm.mixin.injection.Inject;
|
|
import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
|
|
|
|
import carpet.fakes.LightStorageInterface;
|
|
import it.unimi.dsi.fastutil.longs.Long2ObjectMap;
|
|
import it.unimi.dsi.fastutil.longs.LongOpenHashSet;
|
|
import it.unimi.dsi.fastutil.longs.LongSet;
|
|
import net.minecraft.util.math.ChunkSectionPos;
|
|
import net.minecraft.world.chunk.ChunkNibbleArray;
|
|
import net.minecraft.world.chunk.ChunkToNibbleArrayMap;
|
|
import net.minecraft.world.chunk.light.ChunkLightProvider;
|
|
import net.minecraft.world.chunk.light.LightStorage;
|
|
|
|
@Mixin(LightStorage.class)
|
|
public abstract class LightStorage_scarpetChunkCreationMixin implements LightStorageInterface
|
|
{
|
|
@Shadow
|
|
protected abstract ChunkNibbleArray getLightSection(final long sectionPos, final boolean cached);
|
|
|
|
@Shadow
|
|
protected abstract void removeSection(final ChunkLightProvider<?, ?> storage, final long blockChunkPos);
|
|
|
|
@Shadow
|
|
@Final
|
|
protected LongSet dirtySections;
|
|
|
|
@Shadow
|
|
@Final
|
|
protected ChunkToNibbleArrayMap<?> storage;
|
|
|
|
@Shadow protected abstract boolean hasSection(final long sectionPos);
|
|
|
|
@Shadow
|
|
@Final
|
|
protected Long2ObjectMap<ChunkNibbleArray> queuedSections;
|
|
|
|
@Shadow protected abstract void setColumnEnabled(final long l, final boolean bl);
|
|
|
|
@Unique
|
|
private final LongSet removedChunks = new LongOpenHashSet();
|
|
|
|
@Unique
|
|
private final LongSet relightChunks = new LongOpenHashSet();
|
|
|
|
@Override
|
|
public void removeLightData(final long pos)
|
|
{
|
|
this.removedChunks.add(pos);
|
|
}
|
|
|
|
@Override
|
|
public void relight(final long pos)
|
|
{
|
|
this.relightChunks.add(pos);
|
|
}
|
|
|
|
@Inject(
|
|
method = "updateLight(Lnet/minecraft/world/chunk/light/ChunkLightProvider;ZZ)V",
|
|
at = @At("HEAD")
|
|
)
|
|
private void processData(final ChunkLightProvider<?, ?> lightProvider, final boolean doSkylight, final boolean skipEdgeLightPropagation, final CallbackInfo ci)
|
|
{
|
|
// Process light removal
|
|
|
|
for (final long cPos : this.removedChunks)
|
|
{
|
|
for (int y = -1; y < 17; ++y)
|
|
{
|
|
final long sectionPos = ChunkSectionPos.asLong(ChunkSectionPos.unpackX(cPos), y, ChunkSectionPos.unpackZ(cPos));
|
|
|
|
this.queuedSections.remove(sectionPos);
|
|
|
|
if (this.hasSection(sectionPos))
|
|
{
|
|
this.removeSection(lightProvider, sectionPos);
|
|
|
|
if (this.dirtySections.add(sectionPos))
|
|
this.storage.replaceWithCopy(sectionPos);
|
|
|
|
Arrays.fill(this.getLightSection(sectionPos, true).asByteArray(), (byte) 0);
|
|
}
|
|
}
|
|
|
|
this.processRemoveLightData(cPos);
|
|
}
|
|
|
|
this.removedChunks.clear();
|
|
|
|
// Process relighting
|
|
|
|
for (final long cPos : this.relightChunks)
|
|
this.processRelight(lightProvider, cPos);
|
|
|
|
this.relightChunks.clear();
|
|
}
|
|
|
|
@Override
|
|
public void processRemoveLightData(final long pos)
|
|
{
|
|
}
|
|
|
|
@Override
|
|
public void processRelight(final ChunkLightProvider<?, ?> lightProvider, final long pos)
|
|
{
|
|
}
|
|
}
|