package net.minecraft.world.level.block.grower;

import java.util.Random;
import javax.annotation.Nullable;
import net.minecraft.core.BlockPosition;
import net.minecraft.server.level.WorldServer;
import net.minecraft.world.level.IBlockAccess;
import net.minecraft.world.level.block.Block;
import net.minecraft.world.level.block.Blocks;
import net.minecraft.world.level.block.state.IBlockData;
import net.minecraft.world.level.chunk.ChunkGenerator;
import net.minecraft.world.level.levelgen.feature.WorldGenFeatureConfigured;
import net.minecraft.world.level.levelgen.feature.configurations.WorldGenFeatureTreeConfiguration;

public abstract class WorldGenMegaTreeProvider extends WorldGenTreeProvider {

    public WorldGenMegaTreeProvider() {}

    @Override
    public boolean a(WorldServer worldserver, ChunkGenerator chunkgenerator, BlockPosition blockposition, IBlockData iblockdata, Random random) {
        for (int i = 0; i >= -1; --i) {
            for (int j = 0; j >= -1; --j) {
                if (a(iblockdata, worldserver, blockposition, i, j)) {
                    return this.a(worldserver, chunkgenerator, blockposition, iblockdata, random, i, j);
                }
            }
        }

        return super.a(worldserver, chunkgenerator, blockposition, iblockdata, random);
    }

    @Nullable
    protected abstract WorldGenFeatureConfigured<WorldGenFeatureTreeConfiguration, ?> a(Random random);

    public boolean a(WorldServer worldserver, ChunkGenerator chunkgenerator, BlockPosition blockposition, IBlockData iblockdata, Random random, int i, int j) {
        WorldGenFeatureConfigured<WorldGenFeatureTreeConfiguration, ?> worldgenfeatureconfigured = this.a(random);

        if (worldgenfeatureconfigured == null) {
            return false;
        } else {
            IBlockData iblockdata1 = Blocks.AIR.getBlockData();

            worldserver.setTypeAndData(blockposition.c(i, 0, j), iblockdata1, 4);
            worldserver.setTypeAndData(blockposition.c(i + 1, 0, j), iblockdata1, 4);
            worldserver.setTypeAndData(blockposition.c(i, 0, j + 1), iblockdata1, 4);
            worldserver.setTypeAndData(blockposition.c(i + 1, 0, j + 1), iblockdata1, 4);
            if (worldgenfeatureconfigured.a(worldserver, chunkgenerator, random, blockposition.c(i, 0, j))) {
                return true;
            } else {
                worldserver.setTypeAndData(blockposition.c(i, 0, j), iblockdata, 4);
                worldserver.setTypeAndData(blockposition.c(i + 1, 0, j), iblockdata, 4);
                worldserver.setTypeAndData(blockposition.c(i, 0, j + 1), iblockdata, 4);
                worldserver.setTypeAndData(blockposition.c(i + 1, 0, j + 1), iblockdata, 4);
                return false;
            }
        }
    }

    public static boolean a(IBlockData iblockdata, IBlockAccess iblockaccess, BlockPosition blockposition, int i, int j) {
        Block block = iblockdata.getBlock();

        return iblockaccess.getType(blockposition.c(i, 0, j)).a(block) && iblockaccess.getType(blockposition.c(i + 1, 0, j)).a(block) && iblockaccess.getType(blockposition.c(i, 0, j + 1)).a(block) && iblockaccess.getType(blockposition.c(i + 1, 0, j + 1)).a(block);
    }
}