From def4e193596507611955a35dd2b5a1bdb7e1bd82 Mon Sep 17 00:00:00 2001
From: Daniel Mills <danielmillst@gmail.com>
Date: Mon, 19 Jul 2021 04:21:33 -0400
Subject: [PATCH] Tweaks to headless gens

---
 .../com/volmit/iris/engine/headless/HeadlessGenerator.java  | 6 ++++++
 .../java/com/volmit/iris/engine/headless/HeadlessWorld.java | 6 ++++++
 2 files changed, 12 insertions(+)

diff --git a/src/main/java/com/volmit/iris/engine/headless/HeadlessGenerator.java b/src/main/java/com/volmit/iris/engine/headless/HeadlessGenerator.java
index 68a2dcbc9..95b4e7251 100644
--- a/src/main/java/com/volmit/iris/engine/headless/HeadlessGenerator.java
+++ b/src/main/java/com/volmit/iris/engine/headless/HeadlessGenerator.java
@@ -18,6 +18,7 @@
 
 package com.volmit.iris.engine.headless;
 
+import com.volmit.iris.core.pregenerator.PregenListener;
 import com.volmit.iris.engine.data.mca.NBTWorld;
 import com.volmit.iris.engine.framework.EngineCompositeGenerator;
 import com.volmit.iris.engine.parallel.MultiBurst;
@@ -51,6 +52,11 @@ public class HeadlessGenerator {
         generator.directWriteMCA(world.getWorld(), x, z, writer, burst);
     }
 
+    public void generateRegion(int x, int z, PregenListener listener)
+    {
+        generator.directWriteMCA(world.getWorld(), x, z, writer, burst, listener);
+    }
+
     public File generateRegionToFile(int x, int z)
     {
         generateRegionToFile(x, z);
diff --git a/src/main/java/com/volmit/iris/engine/headless/HeadlessWorld.java b/src/main/java/com/volmit/iris/engine/headless/HeadlessWorld.java
index c0c09565a..4397274f9 100644
--- a/src/main/java/com/volmit/iris/engine/headless/HeadlessWorld.java
+++ b/src/main/java/com/volmit/iris/engine/headless/HeadlessWorld.java
@@ -20,7 +20,9 @@ package com.volmit.iris.engine.headless;
 
 import com.volmit.iris.Iris;
 import com.volmit.iris.core.IrisDataManager;
+import com.volmit.iris.engine.IrisWorlds;
 import com.volmit.iris.engine.framework.EngineCompositeGenerator;
+import com.volmit.iris.engine.framework.IrisAccess;
 import com.volmit.iris.engine.object.IrisDimension;
 import com.volmit.iris.engine.object.common.IrisWorld;
 import com.volmit.iris.util.plugin.VolmitSender;
@@ -73,6 +75,10 @@ public class HeadlessWorld {
                 .createWorld();
     }
 
+    public static HeadlessWorld from(World world) {
+        return new HeadlessWorld(world.getName(), IrisWorlds.access(world).getTarget().getDimension(), world.getSeed());
+    }
+
     public static HeadlessWorld from(String name, String dimension, long seed)
     {
         return new HeadlessWorld(name, IrisDataManager.loadAnyDimension(dimension), seed);