NaturesAura/src/main/java/de/ellpeck/naturesaura/gen/LevelGenNetherWartMushroom.java

65 lines
2.6 KiB
Java
Raw Normal View History

package de.ellpeck.naturesaura.gen;
import com.mojang.serialization.Codec;
import de.ellpeck.naturesaura.blocks.ModBlocks;
import net.minecraft.core.BlockPos;
import net.minecraft.world.level.WorldGenLevel;
import net.minecraft.world.level.block.Block;
import net.minecraft.world.level.block.Blocks;
import net.minecraft.world.level.levelgen.feature.Feature;
import net.minecraft.world.level.levelgen.feature.FeaturePlaceContext;
import net.minecraft.world.level.levelgen.feature.TreeFeature;
import net.minecraft.world.level.levelgen.feature.configurations.FeatureConfiguration;
import net.minecraft.world.level.levelgen.feature.configurations.NoneFeatureConfiguration;
public class LevelGenNetherWartMushroom extends Feature<NoneFeatureConfiguration> {
public LevelGenNetherWartMushroom() {
super(Codec.unit(FeatureConfiguration.NONE));
}
@Override
public boolean place(FeaturePlaceContext<NoneFeatureConfiguration> ctx) {
var levelIn = ctx.level();
var pos = ctx.origin();
var rand = ctx.random();
2021-12-15 16:30:22 +01:00
var height = rand.nextInt(5) + 4;
if (rand.nextInt(10) == 0)
height += 5;
// Check if the stem has space
2021-12-15 16:30:22 +01:00
for (var i = 1; i < height; i++) {
var offset = pos.above(i);
if (levelIn.isStateAtPosition(offset, s -> !TreeFeature.validTreePos(levelIn, offset)))
return false;
}
// Place stem
this.setBlock(levelIn, pos, Blocks.AIR.defaultBlockState());
2021-12-15 16:30:22 +01:00
for (var i = 0; i < height; i++)
this.placeIfPossible(levelIn, pos.above(i), Blocks.NETHER_WART_BLOCK);
// Place hat
2021-12-15 16:30:22 +01:00
var rad = 3;
for (var x = -rad; x <= rad; x++) {
for (var z = -rad; z <= rad; z++) {
var absX = Math.abs(x);
var absZ = Math.abs(z);
if (absX <= 1 && absZ <= 1) {
this.placeIfPossible(levelIn, pos.offset(x, height, z), ModBlocks.NETHER_WART_MUSHROOM);
} else if (absX <= 2 && absZ <= 2 && absX != absZ) {
this.placeIfPossible(levelIn, pos.offset(x, height - 1, z), ModBlocks.NETHER_WART_MUSHROOM);
} else if (absX < rad - 1 || absZ < rad - 1 || absX == rad - 1 && absZ == rad - 1) {
this.placeIfPossible(levelIn, pos.offset(x, height - 2, z), ModBlocks.NETHER_WART_MUSHROOM);
}
}
}
return true;
}
private void placeIfPossible(WorldGenLevel level, BlockPos pos, Block block) {
if (level.isStateAtPosition(pos, s -> TreeFeature.validTreePos(level, pos)))
level.setBlock(pos, block.defaultBlockState(), 19);
}
}