Use underlying image for drawing certainty and add draw grids

Author: Zomis

src/test/groovy/net/zomis/machlearn/images/ImagePainter.java

@@ -3,6 +3,7 @@
 import java.awt.*;
 import java.awt.image.BufferedImage;
 import java.io.File;
+import java.util.Random;
 import java.util.function.ToDoubleFunction;
 
 public class ImagePainter {
@@ -16,6 +17,12 @@ public ImagePainter(int width, int height) {
         graphics.fillRect(0, 0, width, width);
     }
 
+    public ImagePainter(BufferedImage image) {
+        img = new BufferedImage(image.getWidth(), image.getHeight(), BufferedImage.TYPE_INT_ARGB);
+        graphics = img.createGraphics();
+        graphics.drawImage(image, 0, 0, null);
+    }
+
     public void drawGrayscale(int x, int y, double value) {
         int grayscaleValue = (int) (value * 255);
         int rgb = 0xff << 24 | grayscaleValue << 16 | grayscaleValue << 8 | grayscaleValue;
@@ -96,4 +103,19 @@ public static ImagePainter[] visualizeNetworks(ImageNetwork network,
         return painters;
     }
 
+    public void drawGrids(ZRect[][] gridLocations) {
+        Color[] colors = { new Color(1, 0, 0, 0.6f), new Color(0, 1, 0, 0.6f), new Color(0, 0, 1, 0.6f),
+                new Color(1, 1, 0, 0.6f), new Color(1, 0, 1, 0.6f), new Color(0, 1, 1, 0.6f)};
+        for (int y = 0; y < gridLocations.length; y++) {
+            for (int x = 0; x < gridLocations[y].length; x++) {
+                ZRect rect = gridLocations[y][x];
+                if (rect == null) {
+                    continue;
+                }
+                Color color = colors[(y * 2 + x) % colors.length];
+                graphics.setColor(color);
+                graphics.fillRect(rect.left, rect.top, rect.width(), rect.height());
+            }
+        }
+    }
 }

src/test/groovy/net/zomis/machlearn/images/MinesweeperScan.java

@@ -148,6 +148,10 @@ public void scan() {
         // also try find separations by scanning lines and finding the line with the lowest delta diff
 
         ZRect[][] gridLocations = findGrid(board.getImage(), rect);
+        ImagePainter painter = new ImagePainter(board.getImage());
+        painter.drawGrids(gridLocations);
+        painter.save("grids");
+
         char[][] gridValues = scanGrid(board.getImage(), gridLocations, board.getExpected());
         for (int y = 0; y < gridValues.length; y++) {
             for (int x = 0; x < gridValues[y].length; x++) {
@@ -159,7 +163,7 @@ public void scan() {
 
     private char[][] scanGrid(BufferedImage runImage, ZRect[][] gridLocations, String expected) {
         char[][] result = new char[gridLocations.length][gridLocations[0].length];
-        ImagePainter painter = new ImagePainter(runImage.getWidth(), runImage.getHeight());
+        ImagePainter painter = new ImagePainter(runImage);
 
         // MinesweeperScan.runOnImage(analyze, network, runImage, m -> m.values().stream().mapToDouble(d -> d).max().getAsDouble());
 //        ImagePainter[] painters = ImagePainter.visualizeNetworks(network, gridLocations[0][0].width(), gridLocations[0][0].height(), runImage,