Add gravity toggle to models

AtlasEngineCa · May 15, 2023 · ee42101 · ee42101
1 parent 8fb0da8
commit ee42101
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 10 deletions.
diff --git a/src/main/java/ca/atlasengine/gui/windows/ModelModifyGUI.java b/src/main/java/ca/atlasengine/gui/windows/ModelModifyGUI.java
@@ -25,7 +25,7 @@ public AnimationSelect() {
         }
     }
 
-    public ModelModifyGUI(List<String> animations, UUID uuid, String modelName, int customModelData, boolean hideOnLoad) {
+    public ModelModifyGUI(List<String> animations, UUID uuid, String modelName, int customModelData, boolean hideOnLoad, boolean gravity) {
         WGridPanel root = new WGridPanel();
         setRootPanel(root);
 
@@ -65,6 +65,14 @@ public ModelModifyGUI(List<String> animations, UUID uuid, String modelName, int
         hideOnLoadButton.setToggle(hideOnLoad);
         root.add(hideOnLoadButton, 2, 4, 5, 1);
 
+        WToggleButton gravityButton = new WToggleButton(Text.of("Has Gravity"));
+        gravityButton.setOnToggle((b) -> {
+            ModelManager.setGravity(uuid, b);
+        });
+
+        gravityButton.setToggle(gravity);
+        root.add(gravityButton, 2, 5, 5, 1);
+
         var moveButton = new WButton(Text.of("Move Model"));
         moveButton.setOnClick(() -> {
             ModelManager.moveModel(uuid);
@@ -73,14 +81,6 @@ public ModelModifyGUI(List<String> animations, UUID uuid, String modelName, int
 
         root.add(moveButton, 1, 2, 5, 1);
 
-        var closeButton = new WButton(Text.of("Delete Model"));
-        closeButton.setOnClick(() -> {
-            ModelManager.deleteModel(uuid);
-            MinecraftClient.getInstance().setScreen(null);
-        });
-
-        root.add(closeButton, 1, root.getHeight() / 16 - 4, 5, 1);
-
         var stopAnimations = new WButton(Text.of("Stop Animation"));
         stopAnimations.setOnClick(() -> {
             ModelManager.setAnimation(uuid, "");

diff --git a/src/main/java/ca/atlasengine/models/ModelLoader.java b/src/main/java/ca/atlasengine/models/ModelLoader.java
@@ -48,14 +48,15 @@ public static void receive(PacketByteBuf packetByteBuf) {
                 String name = packetByteBuf.readString();
                 int customModelData = packetByteBuf.readInt();
                 boolean hideOnLoad = packetByteBuf.readBoolean();
+                boolean gravity = packetByteBuf.readBoolean();
 
                 int count = packetByteBuf.readInt();
                 List<String> animations = new ArrayList<>();
                 for (int i = 0; i < count; i++) {
                     animations.add(packetByteBuf.readString());
                 }
 
-                providedScreen = new GUIScreen(new ModelModifyGUI(animations, uuid, name, customModelData, hideOnLoad));
+                providedScreen = new GUIScreen(new ModelModifyGUI(animations, uuid, name, customModelData, gravity, hideOnLoad));
             }
         }
     }

diff --git a/src/main/java/ca/atlasengine/models/ModelManager.java b/src/main/java/ca/atlasengine/models/ModelManager.java
@@ -87,4 +87,13 @@ public static void setHideOnLoad(UUID uuid, Boolean b) {
 
         ClientPlayNetworking.send(animationIdentifier, buf);
     }
+
+    public static void setGravity(UUID uuid, Boolean b) {
+        PacketByteBuf buf = PacketByteBufs.create();
+        buf.writeByte(7);
+        buf.writeUuid(uuid);
+        buf.writeBoolean(b);
+
+        ClientPlayNetworking.send(animationIdentifier, buf);
+    }
 }