scan tool fixes (#719)

Fix slab/domum handling and water handling and grass/dirt path handling and stairs handling
ldtteam · Feb 8, 2025 · 98a9a16 · 98a9a16
1 parent 52cbdc7
commit 98a9a16
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 10 deletions.
diff --git a/src/main/java/com/ldtteam/structurize/operations/RemoveBlockOperation.java b/src/main/java/com/ldtteam/structurize/operations/RemoveBlockOperation.java
@@ -6,6 +6,7 @@
 import net.minecraft.server.level.ServerLevel;
 import net.minecraft.world.entity.player.Player;
 import net.minecraft.world.item.ItemStack;
+import net.minecraft.world.level.block.Blocks;
 
 /**
  * Operation for removing one type of block.
@@ -37,6 +38,10 @@ protected void apply(final ServerLevel world, final BlockPos position)
         {
             storage.addPreviousDataFor(position, world);
             world.removeBlock(position, false);
+            if (!world.getFluidState(position).isEmpty())
+            {
+                world.setBlock(position, Blocks.AIR.defaultBlockState(), 3);
+            }
             storage.addPostDataFor(position, world);
         }
     }

diff --git a/src/main/java/com/ldtteam/structurize/placement/handlers/placement/PlacementHandlers.java b/src/main/java/com/ldtteam/structurize/placement/handlers/placement/PlacementHandlers.java
@@ -23,10 +23,7 @@
 import net.minecraft.world.level.block.*;
 import net.minecraft.world.level.block.entity.BlockEntity;
 import net.minecraft.world.level.block.state.BlockState;
-import net.minecraft.world.level.block.state.properties.BedPart;
-import net.minecraft.world.level.block.state.properties.BlockStateProperties;
-import net.minecraft.world.level.block.state.properties.DoubleBlockHalf;
-import net.minecraft.world.level.block.state.properties.DripstoneThickness;
+import net.minecraft.world.level.block.state.properties.*;
 import net.minecraft.world.phys.AABB;
 import org.jetbrains.annotations.Nullable;
 
@@ -342,9 +339,11 @@ public List<ItemStack> getRequiredItems(
           @Nullable final CompoundTag tileEntityData,
           final boolean complete)
         {
-            final List<ItemStack> itemList = new ArrayList<>();
-            itemList.add(new ItemStack(Blocks.DIRT));
-            return itemList;
+            if (complete)
+            {
+                return Collections.singletonList(new ItemStack(blockState.getBlock()));
+            }
+            return Collections.singletonList(new ItemStack(Blocks.DIRT));
         }
     }
 
@@ -648,9 +647,11 @@ public List<ItemStack> getRequiredItems(
           @Nullable final CompoundTag tileEntityData,
           final boolean complete)
         {
-            final List<ItemStack> itemList = new ArrayList<>();
-            itemList.add(new ItemStack(Blocks.DIRT, 1));
-            return itemList;
+            if (complete)
+            {
+                return Collections.singletonList(new ItemStack(blockState.getBlock()));
+            }
+            return Collections.singletonList(new ItemStack(Blocks.DIRT));
         }
     }
 
@@ -687,6 +688,10 @@ public List<ItemStack> getRequiredItems(
           @Nullable final CompoundTag tileEntityData,
           final boolean complete)
         {
+            if (complete)
+            {
+                return Collections.singletonList(new ItemStack(blockState.getBlock()));
+            }
             return new ArrayList<>();
         }
     }

diff --git a/src/main/java/com/ldtteam/structurize/util/BlockUtils.java b/src/main/java/com/ldtteam/structurize/util/BlockUtils.java
@@ -553,6 +553,7 @@ else if (item instanceof BlockItem)
             // place
             world.setBlock(here, Blocks.COBBLESTONE.defaultBlockState(), Block.UPDATE_CLIENTS);
             world.setBlock(here, newState, Constants.UPDATE_FLAG);
+            world.getBlockEntity(here).applyComponentsFromItemStack(stackToPlace);
             targetBlock.setPlacedBy(world, here, newState, fakePlayer, stackToPlace);
         }
         else if (item instanceof BucketItem)