package net.minecraft.client.gui.components.debugchart;

import java.text.DecimalFormat;
import java.text.DecimalFormatSymbols;
import java.util.ArrayList;
import java.util.List;
import java.util.Locale;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.gui.Font;
import net.minecraft.client.gui.GuiGraphicsExtractor;
import net.minecraft.util.profiling.ProfileResults;
import net.minecraft.util.profiling.ResultField;
import org.jspecify.annotations.Nullable;

@Environment(EnvType.CLIENT)
public class ProfilerPieChart {
	public static final int RADIUS = 105;
	public static final int PIE_CHART_THICKNESS = 10;
	private static final DecimalFormat PERCENTAGE_FORMAT = new DecimalFormat("##0.00", DecimalFormatSymbols.getInstance(Locale.ROOT));
	private static final int MARGIN = 5;
	private static final int WIDTH = 260;
	private static final int SUBSEQUENT_LINES_INDENT = 10;
	private final Font font;
	@Nullable
	private ProfileResults profilerPieChartResults;
	private String profilerTreePath = "root";
	private int bottomOffset = 0;

	public ProfilerPieChart(final Font font) {
		this.font = font;
	}

	public void setPieChartResults(@Nullable final ProfileResults results) {
		this.profilerPieChartResults = results;
	}

	public void setBottomOffset(final int bottomOffset) {
		this.bottomOffset = bottomOffset;
	}

	public void extractRenderState(final GuiGraphicsExtractor graphics) {
		if (this.profilerPieChartResults != null) {
			List<ResultField> list = this.profilerPieChartResults.getTimes(this.profilerTreePath);
			ResultField currentNode = (ResultField)list.removeFirst();
			int chartCenterX = graphics.guiWidth() - 130 - 10;
			int left = chartCenterX - 130;
			int right = chartCenterX + 130;
			int textUnderChartHeight = list.size() * 9;
			int bottom = graphics.guiHeight() - this.bottomOffset - 5;
			int textStartY = bottom - textUnderChartHeight;
			int chartHalfSizeY = 62;
			int chartCenterY = textStartY - 62 - 5;
			String globalPercentage = PERCENTAGE_FORMAT.format(currentNode.globalPercentage) + "%";
			int globalPercentageWidth = this.font.width(globalPercentage);
			int zeroPrefixWidth = this.font.width("[0] ");
			int topTextMaxWidth = right - globalPercentageWidth - 5 - left - zeroPrefixWidth;
			String currentNodeName = ProfileResults.demanglePath(currentNode.name);
			List<String> currentNodeNameLines = this.splitNodeName(currentNodeName, topTextMaxWidth, topTextMaxWidth - 10);
			int currentNodeNameTop = chartCenterY - 62 - (currentNodeNameLines.size() - 1) * 9;
			graphics.fill(left - 5, currentNodeNameTop - 5, right + 5, bottom + 5, -1873784752);
			graphics.profilerChart(list, left, chartCenterY - 62 + 10, right, chartCenterY + 62);
			String firstLineText = "";
			if (!"unspecified".equals(currentNodeName) && !"root".equals(currentNodeName)) {
				firstLineText = firstLineText + "[0] ";
			}

			firstLineText = firstLineText + (String)currentNodeNameLines.getFirst();
			int col = -1;
			graphics.text(this.font, firstLineText, left, currentNodeNameTop, -1);

			for (int i = 1; i < currentNodeNameLines.size(); i++) {
				graphics.text(this.font, (String)currentNodeNameLines.get(i), left + 10 + zeroPrefixWidth, currentNodeNameTop + i * 9, -1);
			}

			graphics.text(this.font, globalPercentage, right - globalPercentageWidth, currentNodeNameTop, -1);

			for (int i = 0; i < list.size(); i++) {
				ResultField result = (ResultField)list.get(i);
				StringBuilder string = new StringBuilder();
				if ("unspecified".equals(result.name)) {
					string.append("[?] ");
				} else {
					string.append("[").append(i + 1).append("] ");
				}

				String msg = string.append(result.name).toString();
				int textY = textStartY + i * 9;
				graphics.text(this.font, msg, left, textY, result.getColor());
				msg = PERCENTAGE_FORMAT.format(result.percentage) + "%";
				graphics.text(this.font, msg, right - 50 - this.font.width(msg), textY, result.getColor());
				msg = PERCENTAGE_FORMAT.format(result.globalPercentage) + "%";
				graphics.text(this.font, msg, right - this.font.width(msg), textY, result.getColor());
			}
		}
	}

	private List<String> splitNodeName(final String nodeName, final int firstLineMaxWidth, final int maxWidth) {
		String[] nodeNameSplit = nodeName.split("\\.");
		List<String> lines = new ArrayList();
		String currentLine = "";
		int nameIndex = 0;

		while (nameIndex < nodeNameSplit.length) {
			String currentName = nodeNameSplit[nameIndex];
			String currentNameWithPeriod = (nameIndex != 0 ? "." : "") + currentName;
			String newLine = currentLine + currentNameWithPeriod;
			int newWidth = this.font.width(newLine);
			if (newWidth > (!lines.isEmpty() ? maxWidth : firstLineMaxWidth)) {
				if (currentLine.isEmpty()) {
					lines.add(currentNameWithPeriod);
					nameIndex++;
				} else {
					lines.add(currentLine);
					currentLine = "";
				}
			} else {
				currentLine = newLine;
				nameIndex++;
			}
		}

		if (!currentLine.isEmpty()) {
			lines.add(currentLine);
		}

		return lines;
	}

	public void profilerPieChartKeyPress(int key) {
		if (this.profilerPieChartResults != null) {
			List<ResultField> list = this.profilerPieChartResults.getTimes(this.profilerTreePath);
			if (!list.isEmpty()) {
				ResultField node = (ResultField)list.remove(0);
				if (key == 0) {
					if (!node.name.isEmpty()) {
						int pos = this.profilerTreePath.lastIndexOf(30);
						if (pos >= 0) {
							this.profilerTreePath = this.profilerTreePath.substring(0, pos);
						}
					}
				} else {
					key--;
					if (key < list.size() && !"unspecified".equals(((ResultField)list.get(key)).name)) {
						if (!this.profilerTreePath.isEmpty()) {
							this.profilerTreePath = this.profilerTreePath + "\u001e";
						}

						this.profilerTreePath = this.profilerTreePath + ((ResultField)list.get(key)).name;
					}
				}
			}
		}
	}
}