add block_facing mechanism

DenizenScript · Jan 16, 2019 · 2cfecbb · 2cfecbb
1 parent b2fa22e
commit 2cfecbb
Showing 1 changed file with 31 additions and 0 deletions.
diff --git a/plugin/src/main/java/net/aufdemrand/denizen/objects/dLocation.java b/plugin/src/main/java/net/aufdemrand/denizen/objects/dLocation.java
@@ -2167,6 +2167,15 @@ public void applyProperty(Mechanism mechanism) {
         dB.echoError("Cannot apply properties to a location!");
     }
 
+    public static BlockFace faceFor(Vector vec) {
+        for (BlockFace face : BlockFace.values()) {
+            if (face.getDirection().distanceSquared(vec) < 0.01) { // floating-point safe check
+                return face;
+            }
+        }
+        return null;
+    }
+
     @Override
     public void adjust(Mechanism mechanism) {
 
@@ -2178,6 +2187,28 @@ public void adjust(Mechanism mechanism) {
             blockData.setBlock(getBlock(), false);
         }
 
+        // <--[mechanism]
+        // @object dLocation
+        // @name block_facing
+        // @input dLocation
+        // @description
+        // Sets the facing direction of the block, as a vector.
+        // @tags
+        // <l@location.block_facing>
+        // -->
+        if (mechanism.matches("block_facing") && mechanism.requireObject(dLocation.class)) {
+            dLocation faceVec = value.asType(dLocation.class);
+            if (getBlock().getBlockData() instanceof Directional) {
+                Directional dir = (Directional) getBlock().getBlockData();
+                BlockFace newFace = faceFor(faceVec.toVector());
+                if (newFace == null) {
+                    dB.echoError("Direction '" + faceVec + "' does not appear to be a valid block face.");
+                }
+                dir.setFacing(newFace);
+                getBlock().setBlockData(dir);
+            }
+        }
+
         // <--[mechanism]
         // @object dLocation
         // @name block_type