Block state processor

gradle.properties

@@ -9,7 +9,7 @@ yarn_build=7
 loader_version=0.15.11
 
 # Mod Properties
-mod_version=1.5.4
+mod_version=1.5.5
 maven_group=net.mcbrawls
 mod_id=blueprint
 

src/main/kotlin/net/mcbrawls/blueprint/structure/BlockStateProcessor.kt

@@ -0,0 +1,19 @@
+package net.mcbrawls.blueprint.structure
+
+import net.minecraft.block.BlockState
+
+/**
+ * Processes a block state.
+ */
+fun interface BlockStateProcessor {
+    /**
+     * Processes a block state.
+     * @return the output block state
+     */
+    fun process(
+        /**
+         * The input block state.
+         */
+        state: BlockState
+    ): BlockState
+}

src/main/kotlin/net/mcbrawls/blueprint/structure/Blueprint.kt

@@ -49,9 +49,10 @@ data class Blueprint(
      * Places this blueprint in the world at the given position.
      * @return a placed blueprint
      */
-    fun place(world: ServerWorld, position: BlockPos): PlacedBlueprint {
+    fun place(world: ServerWorld, position: BlockPos, processor: BlockStateProcessor? = null): PlacedBlueprint {
         forEach { offset, state ->
-            world.setBlockState(position.add(offset), state)
+            val trueState = processor?.process(state) ?: state
+            world.setBlockState(position.add(offset), trueState)
         }
 
         return PlacedBlueprint(this, position)
@@ -61,14 +62,15 @@ data class Blueprint(
      * Launches a completable future placing this blueprint in the world at the given position.
      * @return a placed blueprint future and a progress provider
      */
-    fun placeWithProgress(world: ServerWorld, position: BlockPos): Pair<CompletableFuture<PlacedBlueprint>, ProgressProvider> {
+    fun placeWithProgress(world: ServerWorld, position: BlockPos, processor: BlockStateProcessor? = null): Pair<CompletableFuture<PlacedBlueprint>, ProgressProvider> {
         val progress = AtomicReference(0.0f)
 
         val future: CompletableFuture<PlacedBlueprint> = CompletableFuture.supplyAsync {
             synchronized(world) {
                 var i = 0
                 forEach { offset, state ->
-                    world.setBlockState(position.add(offset), state)
+                    val trueState = processor?.process(state) ?: state
+                    world.setBlockState(position.add(offset), trueState)
                     progress.set(++i / totalBlocks.toFloat())
                 }
             }