Add inputExtent parameter and separate in/outputTransform for deformRegion

Author: TomyLobo

worldedit-core/src/main/java/com/sk89q/worldedit/EditSession.java

@@ -28,6 +28,7 @@
 import com.sk89q.worldedit.extension.platform.Watchdog;
 import com.sk89q.worldedit.extent.ChangeSetExtent;
 import com.sk89q.worldedit.extent.Extent;
+import com.sk89q.worldedit.extent.InputExtent;
 import com.sk89q.worldedit.extent.MaskingExtent;
 import com.sk89q.worldedit.extent.NullExtent;
 import com.sk89q.worldedit.extent.TracingExtent;
@@ -2437,6 +2438,28 @@ public int deformRegion(final Region region, final Vector3 zero, final Vector3 u
         return deformRegion(region, transform, expressionString, timeout);
     }
 
+    /**
+     * Deforms the region by a given expression. A deform provides a block's x, y, and z coordinates (possibly scaled)
+     * to an expression, and then sets the block to the block given by the resulting values of the variables, if they
+     * have changed.
+     *
+     * @param region           the region to deform
+     * @param outputTransform  the output coordinate system
+     * @param expressionString the expression to evaluate for each block
+     * @param timeout          maximum time for the expression to evaluate for each block. -1 for unlimited.
+     * @param inputExtent      the InputExtent to fetch blocks from, for instance a World or a Clipboard
+     * @param inputTransform   the input coordinate system
+     * @return number of blocks changed
+     * @throws ExpressionException       thrown on invalid expression input
+     * @throws MaxChangedBlocksException thrown if too many blocks are changed
+     */
+    public int deformRegion(final Region region, final Transform outputTransform, final String expressionString,
+                            final int timeout, InputExtent inputExtent, Transform inputTransform) throws ExpressionException, MaxChangedBlocksException {
+        final Expression expression = Expression.compile(expressionString, "x", "y", "z");
+        expression.optimize();
+        return deformRegion(region, outputTransform, expression, timeout, inputExtent, inputTransform);
+    }
+
     /**
      * Deforms the region by a given expression. A deform provides a block's x, y, and z coordinates (possibly scaled)
      * to an expression, and then sets the block to the block given by the resulting values of the variables, if they
@@ -2450,11 +2473,10 @@ public int deformRegion(final Region region, final Vector3 zero, final Vector3 u
      * @throws ExpressionException       thrown on invalid expression input
      * @throws MaxChangedBlocksException thrown if too many blocks are changed
      */
+    @Deprecated
     public int deformRegion(final Region region, final Transform transform, final String expressionString,
                             final int timeout) throws ExpressionException, MaxChangedBlocksException {
-        final Expression expression = Expression.compile(expressionString, "x", "y", "z");
-        expression.optimize();
-        return deformRegion(region, transform, expression, timeout);
+        return deformRegion(region, transform, expressionString, timeout, world, transform);
     }
 
     /**
@@ -2467,7 +2489,8 @@ public int deformRegion(final Region region, final Transform transform, final St
     @Deprecated
     public int deformRegion(final Region region, final Vector3 zero, final Vector3 unit, final Expression expression,
                             final int timeout) throws ExpressionException, MaxChangedBlocksException {
-        return deformRegion(region, new SimpleTransform(zero, unit), expression, timeout);
+        final Transform transform = new SimpleTransform(zero, unit);
+        return deformRegion(region, transform, expression, timeout, world, transform);
     }
 
     /**
@@ -2477,33 +2500,33 @@ public int deformRegion(final Region region, final Vector3 zero, final Vector3 u
      * The Expression class is subject to change. Expressions should be provided via the string overload.
      * </p>
      */
-    public int deformRegion(final Region region, final Transform transform, final Expression expression,
-                            final int timeout) throws ExpressionException, MaxChangedBlocksException {
+    public int deformRegion(final Region region, final Transform outputTransform, final Expression expression,
+                            final int timeout, InputExtent inputExtent, final Transform inputTransform) throws ExpressionException, MaxChangedBlocksException {
         final Variable x = expression.getSlots().getVariable("x")
             .orElseThrow(IllegalStateException::new);
         final Variable y = expression.getSlots().getVariable("y")
             .orElseThrow(IllegalStateException::new);
         final Variable z = expression.getSlots().getVariable("z")
             .orElseThrow(IllegalStateException::new);
 
-        final WorldEditExpressionEnvironment environment = new WorldEditExpressionEnvironment(this, transform);
+        final WorldEditExpressionEnvironment environment = new WorldEditExpressionEnvironment(this, outputTransform);
         expression.setEnvironment(environment);
 
         final DoubleArrayList<BlockVector3, BaseBlock> queue = new DoubleArrayList<>(false);
 
-        final Transform transformInverse = transform.inverse();
+        final Transform outputTransformInverse = outputTransform.inverse();
         for (BlockVector3 position : region) {
             // transform
-            final Vector3 scaled = transformInverse.apply(position.toVector3());
+            final Vector3 scaled = outputTransformInverse.apply(position.toVector3());
 
             // deform
             expression.evaluate(new double[]{ scaled.x(), scaled.y(), scaled.z() }, timeout);
 
             // untransform, round-nearest
-            final BlockVector3 sourcePosition = transform.apply(Vector3.at(x.value(), y.value(), z.value())).add(0.5, 0.5, 0.5).toBlockPoint();
+            final BlockVector3 sourcePosition = inputTransform.apply(Vector3.at(x.value(), y.value(), z.value())).add(0.5, 0.5, 0.5).toBlockPoint();
 
-            // read block from world
-            final BaseBlock material = world.getFullBlock(sourcePosition);
+            // read block from world/clipboard
+            final BaseBlock material = inputExtent.getFullBlock(sourcePosition);
 
             // queue operation
             queue.put(position, material);

worldedit-core/src/main/java/com/sk89q/worldedit/command/RegionCommands.java

@@ -31,6 +31,7 @@
 import com.sk89q.worldedit.entity.Player;
 import com.sk89q.worldedit.extension.platform.Actor;
 import com.sk89q.worldedit.extent.Extent;
+import com.sk89q.worldedit.extent.InputExtent;
 import com.sk89q.worldedit.extent.clipboard.BlockArrayClipboard;
 import com.sk89q.worldedit.function.GroundFunction;
 import com.sk89q.worldedit.function.RegionFunction;
@@ -501,9 +502,10 @@ public int deform(Actor actor, LocalSession session, EditSession editSession,
                           boolean offsetCenter) throws WorldEditException {
 
         final Transform transform = TransformUtil.createTransformForExpressionCommand(actor, session, region, useRawCoords, offsetPlacement, offsetCenter);
+        final InputExtent inputExtent = editSession.getWorld();
 
         try {
-            final int affected = editSession.deformRegion(region, transform, String.join(" ", expression), session.getTimeout());
+            final int affected = editSession.deformRegion(region, transform, String.join(" ", expression), session.getTimeout(), inputExtent, transform);
             if (actor instanceof Player) {
                 ((Player) actor).findFreePosition();
             }

worldedit-core/src/main/java/com/sk89q/worldedit/function/factory/Deform.java

@@ -170,7 +170,8 @@ private record DeformOperation(
         public Operation resume(RunContext run) throws WorldEditException {
             try {
                 // TODO: Move deformation code
-                ((EditSession) destination).deformRegion(region, transform, expression, timeout);
+                final EditSession editSession = (EditSession) destination;
+                editSession.deformRegion(region, transform, expression, timeout, editSession.getWorld(), transform);
                 return null;
             } catch (ExpressionException e) {
                 throw new RuntimeException("Failed to execute expression", e); // TODO: Better exception to throw here?