Jigsaw Mantle Components

src/main/java/com/volmit/iris/engine/mantle/components/MantleJigsawComponent.java

@@ -0,0 +1,132 @@
+/*
+ * Iris is a World Generator for Minecraft Bukkit Servers
+ * Copyright (c) 2021 Arcane Arts (Volmit Software)
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <https://www.gnu.org/licenses/>.
+ */
+
+package com.volmit.iris.engine.mantle.components;
+
+import com.volmit.iris.engine.data.cache.Cache;
+import com.volmit.iris.engine.jigsaw.PlannedStructure;
+import com.volmit.iris.engine.mantle.EngineMantle;
+import com.volmit.iris.engine.mantle.IrisMantleComponent;
+import com.volmit.iris.engine.object.basic.IrisPosition;
+import com.volmit.iris.engine.object.biome.IrisBiome;
+import com.volmit.iris.engine.object.feature.IrisFeaturePositional;
+import com.volmit.iris.engine.object.jigsaw.IrisJigsawStructure;
+import com.volmit.iris.engine.object.jigsaw.IrisJigsawStructurePlacement;
+import com.volmit.iris.engine.object.regional.IrisRegion;
+import com.volmit.iris.util.collection.KList;
+import com.volmit.iris.util.mantle.MantleFlag;
+import com.volmit.iris.util.math.Position2;
+import com.volmit.iris.util.math.RNG;
+import com.volmit.iris.util.parallel.BurstExecutor;
+
+import java.util.List;
+import java.util.function.Consumer;
+
+public class MantleJigsawComponent extends IrisMantleComponent
+{
+    public MantleJigsawComponent(EngineMantle engineMantle) {
+        super(engineMantle, MantleFlag.JIGSAW);
+    }
+
+    @Override
+    public void generateLayer(int x, int z, Consumer<Runnable> post) {
+        int s = getRadius();
+        BurstExecutor burst = burst();
+
+        for (int i = -s; i <= s; i++) {
+            int xx = i+x;
+            int xxx = 8 + (xx << 4);
+            for (int j = -s; j <= s; j++) {
+                int zz = j + z;
+                int zzz = 8 + (zz << 4);
+                burst.queue(() -> {
+                    RNG rng = new RNG(Cache.key(xx, zz) + seed());
+                    IrisRegion region = getComplex().getRegionStream().get(xxx, zzz);
+                    IrisBiome biome = getComplex().getTrueBiomeStreamNoFeatures().get(xxx, zzz);
+                    generateParallaxJigsaw(rng, xx, zz, biome, region, post);
+                });
+            }
+        }
+
+        burst.complete();
+    }
+
+    private void generateParallaxJigsaw(RNG rng, int x, int z, IrisBiome biome, IrisRegion region, Consumer<Runnable> post) {
+        boolean placed = false;
+
+        if (getDimension().getStronghold() != null) {
+            List<Position2> poss = getDimension().getStrongholds(seed());
+
+            if (poss != null) {
+                for (Position2 pos : poss) {
+                    if (x == pos.getX() >> 4 && z == pos.getZ() >> 4) {
+                        IrisJigsawStructure structure = getData().getJigsawStructureLoader().load(getDimension().getStronghold());
+                        placeStructure(pos.toIris(), structure, rng, post);
+                        placed = true;
+                    }
+                }
+            }
+        }
+
+        if (!placed) {
+            for (IrisJigsawStructurePlacement i : biome.getJigsawStructures()) {
+                if (rng.nextInt(i.getRarity()) == 0) {
+                    IrisPosition position = new IrisPosition((x << 4) + rng.nextInt(15), 0, (z << 4) + rng.nextInt(15));
+                    IrisJigsawStructure structure = getData().getJigsawStructureLoader().load(i.getStructure());
+                    placeStructure(position, structure, rng, post);
+                    placed = true;
+                }
+            }
+        }
+
+        if (!placed) {
+            for (IrisJigsawStructurePlacement i : region.getJigsawStructures()) {
+                if (rng.nextInt(i.getRarity()) == 0) {
+                    IrisPosition position = new IrisPosition((x << 4) + rng.nextInt(15), 0, (z << 4) + rng.nextInt(15));
+                    IrisJigsawStructure structure = getData().getJigsawStructureLoader().load(i.getStructure());
+                    placeStructure(position, structure, rng, post);
+                    placed = true;
+                }
+            }
+        }
+
+        if (!placed) {
+            for (IrisJigsawStructurePlacement i : getDimension().getJigsawStructures()) {
+                if (rng.nextInt(i.getRarity()) == 0) {
+                    IrisPosition position = new IrisPosition((x << 4) + rng.nextInt(15), 0, (z << 4) + rng.nextInt(15));
+                    IrisJigsawStructure structure = getData().getJigsawStructureLoader().load(i.getStructure());
+                    placeStructure(position, structure, rng, post);
+                }
+            }
+        }
+    }
+
+
+    private void placeStructure(IrisPosition position, IrisJigsawStructure structure, RNG rng, Consumer<Runnable> post) {
+        if (structure.getFeature() != null) {
+            if (structure.getFeature().getBlockRadius() == 32) {
+                structure.getFeature().setBlockRadius((double) structure.getMaxDimension() / 3);
+            }
+
+            getMantle().set(position.getX(), 0, position.getZ(),
+                    new IrisFeaturePositional(position.getX(), position.getZ(), structure.getFeature()));
+        }
+
+        post.accept(() -> new PlannedStructure(structure, position, rng).place(getEngineMantle(), getMantle(), post));
+    }
+}