package com.example.treelabyrinth.worldgen; import com.example.treelabyrinth.ModBlocks; import com.mojang.serialization.Codec; import net.minecraft.block.Block; import net.minecraft.block.BlockState; import net.minecraft.block.Blocks; import net.minecraft.block.LeavesBlock; import net.minecraft.util.math.BlockPos; import net.minecraft.util.math.Direction; import net.minecraft.util.math.random.Random; import net.minecraft.world.StructureWorldAccess; import net.minecraft.world.gen.feature.DefaultFeatureConfig; import net.minecraft.world.gen.feature.Feature; import net.minecraft.world.gen.feature.util.FeatureContext; public class StairTreeFeature extends Feature { private static final Direction[] DIRECTIONS = { Direction.NORTH, Direction.EAST, Direction.SOUTH, Direction.WEST }; public StairTreeFeature(Codec codec) { super(codec); } @Override public boolean generate(FeatureContext context) { StructureWorldAccess world = context.getWorld(); BlockPos origin = context.getOrigin(); Random random = context.getRandom(); if (!world.getBlockState(origin.down()).isSolidBlock(world, origin.down())) { return false; } Direction spiralDir = DIRECTIONS[random.nextInt(DIRECTIONS.length)]; BlockPos current = origin; int steps = 6 + random.nextInt(5); for (int step = 0; step < steps; step++) { int pillarHeight = 3 + random.nextInt(3); for (int h = 0; h < pillarHeight; h++) { BlockPos pillarPos = current.up(h); BlockState existing = world.getBlockState(pillarPos); if (existing.isAir() || existing.isReplaceable()) { world.setBlockState(pillarPos, ModBlocks.DYNAMIC_TREE_BLOCK.getDefaultState(), Block.NOTIFY_ALL); } } BlockPos stepTop = current.up(pillarHeight); for (int s = 0; s <= 2; s++) { BlockPos stepPos = stepTop.offset(spiralDir, s); BlockState existing = world.getBlockState(stepPos); if (existing.isAir() || existing.isReplaceable()) { world.setBlockState(stepPos, Blocks.OAK_LOG.getDefaultState(), Block.NOTIFY_ALL); } } placeLeafCluster(world, stepTop.up(), random); Direction crossDir = spiralDir.rotateYClockwise(); for (int c = -1; c <= 1; c += 2) { BlockPos sidePillarBase = current.offset(crossDir, c * 2); int sideHeight = 2 + random.nextInt(3); for (int h = 0; h < sideHeight; h++) { BlockPos sidePos = sidePillarBase.up(h); BlockState existing = world.getBlockState(sidePos); if (existing.isAir() || existing.isReplaceable()) { world.setBlockState(sidePos, ModBlocks.DYNAMIC_TREE_BLOCK.getDefaultState(), Block.NOTIFY_ALL); } } } current = current.offset(spiralDir, 3).up(pillarHeight - 1); if (step % 2 == 1) { spiralDir = spiralDir.rotateYClockwise(); } } return true; } private void placeLeafCluster(StructureWorldAccess world, BlockPos center, Random random) { BlockState leafState = Blocks.OAK_LEAVES.getDefaultState() .with(LeavesBlock.PERSISTENT, true) .with(LeavesBlock.DISTANCE, 1); for (int dx = -2; dx <= 2; dx++) { for (int dy = -1; dy <= 1; dy++) { for (int dz = -2; dz <= 2; dz++) { if (Math.abs(dx) == 2 && Math.abs(dz) == 2) continue; if (random.nextInt(5) == 0) continue; BlockPos leafPos = center.add(dx, dy, dz); if (world.getBlockState(leafPos).isAir()) { world.setBlockState(leafPos, leafState, Block.NOTIFY_ALL); } } } } } }