package net.minecraft.client.renderer.debug;

import com.google.common.collect.Sets;
import com.mojang.blaze3d.vertex.PoseStack;
import java.util.Set;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.renderer.MultiBufferSource;
import net.minecraft.core.BlockPos;
import net.minecraft.core.SectionPos;

@Environment(EnvType.CLIENT)
public class VillageSectionsDebugRenderer implements DebugRenderer.SimpleDebugRenderer {
	private static final int MAX_RENDER_DIST_FOR_VILLAGE_SECTIONS = 60;
	private final Set<SectionPos> villageSections = Sets.<SectionPos>newHashSet();

	VillageSectionsDebugRenderer() {
	}

	@Override
	public void clear() {
		this.villageSections.clear();
	}

	public void setVillageSection(SectionPos pos) {
		this.villageSections.add(pos);
	}

	public void setNotVillageSection(SectionPos pos) {
		this.villageSections.remove(pos);
	}

	@Override
	public void render(PoseStack poseStack, MultiBufferSource bufferSource, double camX, double camY, double camZ) {
		BlockPos blockPos = BlockPos.containing(camX, camY, camZ);
		this.villageSections.forEach(sectionPos -> {
			if (blockPos.closerThan(sectionPos.center(), 60.0)) {
				highlightVillageSection(poseStack, bufferSource, sectionPos);
			}
		});
	}

	private static void highlightVillageSection(PoseStack poseStack, MultiBufferSource buffer, SectionPos pos) {
		DebugRenderer.renderFilledUnitCube(poseStack, buffer, pos.center(), 0.2F, 1.0F, 0.2F, 0.15F);
	}
}