minecraft-kajetans-mod/src/main/java/me/km/blocks/ModTree.java

package me.km.blocks;

import java.util.Random;
import javax.annotation.Nullable;
import net.minecraft.block.BlockState;
import net.minecraft.block.trees.Tree;
import net.minecraft.world.gen.blockstateprovider.SimpleBlockStateProvider;
import net.minecraft.world.gen.feature.ConfiguredFeature;
import net.minecraft.world.gen.feature.Feature;
import net.minecraft.world.gen.feature.TreeFeatureConfig;
import net.minecraft.world.gen.foliageplacer.BlobFoliagePlacer;
import net.minecraftforge.common.IPlantable;

public class ModTree extends Tree
{
    private TreeFeatureConfig treeConfig = null;
    
    @Nullable
    @Override
    protected ConfiguredFeature<TreeFeatureConfig, ?> getTreeFeature(Random r, boolean p_225546_2_)
    {
        return Feature.NORMAL_TREE.withConfiguration(treeConfig);
    }
    
    public void setConfig(BlockState wood, BlockState leaves, IPlantable sapling)
    {
        treeConfig = (new TreeFeatureConfig.Builder(new SimpleBlockStateProvider(wood), new SimpleBlockStateProvider(leaves), new BlobFoliagePlacer(2, 0)))
                    .baseHeight(5)
                    .heightRandA(2)
                    .foliageHeight(3)
                    .ignoreVines()
                    .setSapling(sapling)
                    .build();
    }
}