80 lines
2.1 KiB
Java
80 lines
2.1 KiB
Java
package gui;
|
|
|
|
import java.util.HashMap;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
|
|
import gui.ColorGenerator.Color;
|
|
import imgui.ImGui;
|
|
import imgui.ImVec2;
|
|
import imgui.ImVec4;
|
|
import imgui.flag.ImGuiCol;
|
|
import imgui.flag.ImGuiStyleVar;
|
|
import sudoku.Block;
|
|
import sudoku.Cell;
|
|
import sudoku.Sudoku;
|
|
|
|
public class SudokuRenderer {
|
|
|
|
private final Sudoku sudoku;
|
|
private int currentIndex = -1;
|
|
private final Map<Block, Color> colorPalette;
|
|
|
|
public SudokuRenderer(Sudoku sudoku) {
|
|
this.sudoku = sudoku;
|
|
this.colorPalette = new HashMap<>();
|
|
initColors();
|
|
}
|
|
|
|
private void initColors() {
|
|
List<Color> colors = ColorGenerator.greatPalette(sudoku.getSize());
|
|
int index = 0;
|
|
for (Block block : sudoku.getBlocks()) {
|
|
colorPalette.put(block, colors.get(index));
|
|
index++;
|
|
}
|
|
}
|
|
|
|
private void renderPopup() {
|
|
if (ImGui.beginPopup("editPopup")) {
|
|
for (int i = 1; i < sudoku.getSize() + 1; i++) {
|
|
if (i % (int) (Math.sqrt(sudoku.getSize())) != 1)
|
|
ImGui.sameLine();
|
|
if (ImGui.button(Integer.toString(i), new ImVec2(50, 50))) {
|
|
this.sudoku.setCellSymbol(currentIndex % sudoku.getSize(), currentIndex / sudoku.getSize(), i - 1);
|
|
ImGui.closeCurrentPopup();
|
|
}
|
|
}
|
|
ImGui.endPopup();
|
|
}
|
|
}
|
|
|
|
public void render() {
|
|
ImGui.begin("Sudoku Window");
|
|
for (int y = 0; y < sudoku.getSize(); y++) {
|
|
for (int x = 0; x < sudoku.getSize(); x++) {
|
|
if (x > 0)
|
|
ImGui.sameLine();
|
|
int index = y * sudoku.getSize() + x;
|
|
Cell cell = sudoku.getCell(x, y);
|
|
int symbol = cell.getSymbolIndex();
|
|
Color blockColor = colorPalette.get(cell.getBlock());
|
|
ImGui.pushStyleVar(ImGuiStyleVar.SelectableTextAlign, new ImVec2(0.5f, 0.5f));
|
|
ImGui.pushStyleColor(ImGuiCol.Header, new ImVec4(blockColor.r, blockColor.g, blockColor.b, 1.0f));
|
|
String cellText = "";
|
|
if (symbol != -1)
|
|
cellText += Integer.toString(symbol + 1);
|
|
if (ImGui.selectable(cellText + "##" + index, true, 0, new ImVec2(50, 50))) {
|
|
ImGui.openPopup("editPopup");
|
|
currentIndex = index;
|
|
}
|
|
ImGui.popStyleVar();
|
|
ImGui.popStyleColor();
|
|
}
|
|
}
|
|
renderPopup();
|
|
ImGui.end();
|
|
}
|
|
|
|
}
|