<br />
<b>Warning</b>:  session_start(): open(/var/lib/php/sessions/sess_g1ra8dmtgs6m01fgb9enm2c5ks, O_RDWR) failed: No space left on device (28) in <b>/var/www/tb-usercake/models/config.php</b> on line <b>51</b><br />
<br />
<b>Warning</b>:  session_start(): Failed to read session data: files (path: /var/lib/php/sessions) in <b>/var/www/tb-usercake/models/config.php</b> on line <b>51</b><br />
!7

cmodule ChessBoardRecognizer {
  S segmenterPreset;
  ParameterizedSegmenter segmenter;
  S status;
  S selectedSquare;
  S piece;
  LPair<S, Double> recognized;

  transient int recreatedSquareSize = 32;
  transient ReliableSingleThread rst = dm_rst(module(), r recognize);
  transient BufferedImage img, board, square, recreated;
  transient LL<BufferedImage> squareImages;
  transient ImageSurface isInput, isBoard, isSquare, isRecreated;
  
  transient LS pieces = ai_chessPieces();
  transient Pair<Map<S, BufferedImage>> recreateStock; // dark, light
  
  visualize {
    isBoard = jImageSurface();
    imageSurfaceOnHover(isBoard, voidfunc(Pt p) {
      if (board == null || p == null) ret with noToolTip(isBoard);
      S square = isPointToSquare(p);
      Pair<S, Double> rec = recognitionForSquare(square);
      setToolTip(isBoard, square + " - " + (rec == null ? "?" : rec.a + " [" + iround(rec.b*100) + "%]"));
    });
    imageSurfaceOnLeftClick(isBoard, voidfunc(Pt p) {
      setField(selectedSquare := isPointToSquare(p));
      S rec = pairA(recognitionForSquare(selectedSquare));
      if (rec != null) setField(piece := rec);
      int x, y = unpair chess_squareToPos(selectedSquare);
      square = _get(_get(squareImages, y), x);
      //printVars_str("leftClick", +p, +selectedSquare, +x, +y, +square);
      isSquare.setImage(square);
    });
    
    ret withCalc(rst, jvsplit(
      jCenteredSection("Input image", jscroll_center(isInput = jImageSurface(img))),
      northCenterAndSouthWithMargins(
        centerAndEastWithMargin(withLabel("Segmenter preset: ", main.onChange(rst, dm_comboBox segmenterPreset(agiBlue_segmenterPresetNames()))), jbutton("Recognize", rst)),
        hgridWithSpacing(
          jCenteredSection("Recreated", jscroll_center(isRecreated = jImageSurface(recreated))),
          jCenteredSection("Chess board", jscroll_center(isBoard)),
          jCenteredLiveValueSection(dm_calculatedLiveValue(S, () -> "Selected square" + (empty(selectedSquare) ? "" : " (" + selectedSquare + ")")), centerAndSouthWithMargin(
            jscroll_center(isSquare = jImageSurface(square)),
            rightAlignedLine(
              withLabel("Piece: ", dm_comboBox('piece, pieces)),
              jbutton("Save & upload", rThread saveAndUploadPiece))
        ))),
        dm_label('status)
      )));
  }
    
  void recognize {
    segmenter = parameterizedSegmenterFromAGIBlue(segmenterPreset);
    if (segmenter == null) ret with infoBox(status("No segmenter"));
    //img = dm_shootScreenHidingOS();
    if (isInput != null) img = isInput.getImage();
    L<Rect> rects = segmenter.get(img);
    if (empty(rects)) ret with status("No chess board found");
    Rect r = first(rects);
    if (l(rects) > 1) status("More than one chess board found, choosing one");
    else status("Chess board found at: " + r);
    board = clipBufferedImage(img, r);
    
    squareImages = bufferedImageMNGrid(board, 8, 8);
    
    if (isBoard != null)
      isBoard.setImageAndZoomToDisplay(
        mergeBufferedImagesVertically(
          map(bufferedImageNVerticalSlices(8, board), i ->
            mergeBufferedImagesHorizontally(bufferedImageNHorizontalSlices(8, i)))));
            
    new ChessPieceRecognizer pieceRecognizer;
    pieceRecognizer.load();
    recognized = new L;
    for (BufferedImage img : concatLists(squareImages))
      recognized.add(pieceRecognizer.recognize(img));
    change();
    print(recognized);
    recreate();
  }
  
  S status(S status) { setField(+status); ret status; }
  
  O _getReloadData() { ret img; }
  void _setReloadData(BufferedImage img) { this.img = img; }
 
  void setImage(BufferedImage img) {
    this.img = img;
    if (isInput != null) isInput.setImageAndZoomToDisplay(img);
    rst.trigger();
  }

  S isPointToSquare(Pt p) {
    double space = imageMergeSpacing();
    double squareW = board.getWidth()/8.0+space;
    double squareH = board.getHeight()/8.0+space;
    int x = clamp(iratio_floor(p.x+space/2, squareW), 0, 7);
    int y = clamp(iratio_floor(p.y+space/2, squareH), 0, 7);
    ret strCharPlus('A', x) + (8-y);
  }
  
  void saveAndUploadPiece enter {
    if (square == null || empty(piece)) ret;
    temp tempInfoBox_noHide("Uploading...");
    S imageURL = uploadToImageServer("Chess square: " + piece, square);
    agiBlue_postInSlice(agiBlue_chessPieceImagesSlice(), imageURL, "is", piece);
    agiBlue_postInSlice(agiBlue_chessPieceImagesSlice(), imageURL, "was on square", selectedSquare);
    infoBox("Uploaded!");
    rst.trigger();
  }
  
  void recreate {
    if (recognized == null) ret;
    if (recreateStock == null)
      recreateStock = mapValues(pairsToMap(reversePairs(chessPieceImagesFromAGIBlue())), imageID -> resizeImage(loadImage2(imageID), recreatedSquareSize, recreatedSquareSize));
      agiBlue_pagesAndValuesForKey(agiBlue_chessPieceImagesSlice(), "is")));

    new L<BufferedImage> out;
    for (S piece : pairsA(recognized))
      out.add(or_func(recreateStock.get(piece), () -> whiteImage(recreatedSquareSize, recreatedSquareSize)));
      
    new L<BufferedImage> rows;
    for (L<BufferedImage> row : listToChunks(out, 8))
      rows.add(mergeBufferedImagesHorizontally(row, spacing := 0));
    recreated = mergeBufferedImagesVertically(rows, spacing := 0);
    if (isRecreated != null) isRecreated.setImageAndZoomToDisplay(recreated);
  }
  
  Pair<S, Double> recognitionForSquare(S square) {
    int x, y = unpair chess_squareToPos(square);
    ret _get(recognized, y*8+x);
  }
}