!7

cmodule VectorLiveRecogSpike1 > DynImageSurfaceAndEnabled {
  transient ReliableSingleThread rstRecog = dm_rst(this, r recognize);
  
  start {
    rstRecog.trigger();
    dm_vmBus_onMessage haveVectorImage((mod, _img) -> rstRecog.trigger());
  }
  
  void recognize {
    if (!enabled) ret;
    BufferedImage inputImage = dm_getLatestVectorImage();
    CheesyRecognizer rec = new(inputImage);
    CheesyRecognizer.Prototype proto = rec.addPrototype(loadBWImage(#1102893));
    int w = rec.getWidth(); // input image width
    int w1 = 16; // coarse resolution to scan
    int w2 = 128; // fine resolution to scan
    print(+w);
    L<Int> assumedProtoSizes = sqrt2progression_int_dropFirst(5, w);
    print(+assumedProtoSizes);
    time "Recog" {
      // Look for center-of-object candidates in coarse resolution
    
      new LPair<Rect, Double> allCandidates; // for all proto sizes in coarse image
      L<Int> pwScaled = uniquify(map(pw -> scaleDouble_iround(w, w1, pw), assumedProtoSizes));
      for (int pw : pwScaled) {
        // candidates are in full image resolution
        LPair<Rect, Double> candidates = rec.fullScanToNBest(proto, w1, pw, 0f, 3);
    
        printVars(+pwScaled, +candidates);
        allCandidates.addAll(candidates);
      }
      L<Rect> candidates = pairsA(takeFirst(5, sortBySecondOfPairsDesc_inPlace(allCandidates)));
      L<Pt> centers = centerOfRects(candidates);
      print(+candidates);
      print(+centers);
    
      // Stage 2: scan around candidates in higher resolution
    
      int h2 = rec.scaledInputImage(w2).getHeight();
      int spiralSize = w2/w1*2;
      int pixelsPerSpiral = sqr(spiralSize);
      print("Spiral size: " + spiralSize + " / " + pixelsPerSpiral);
    
      new Best<Rect> best;
      CheesyRecognizer.ScaledInputImage scaledInputImage = rec.scaledInputImage(w2);
      L<Int> pwScaled2 = uniquify(map(pw -> scaleDouble_iround(w, w2, pw), assumedProtoSizes));
      print(+pwScaled2);
      for (int pw : pwScaled2) {
        CheesyRecognizer.ScaledPrototype scaledProto = proto.scaledPrototype(pw);
        new Set<Pt> pointsTried;
        for (Pt p : centers) {
          // scaled center point, moved to top left of prototype
          Pt pScaled = translatePt(-pw/2, -scaledProto.getHeight()/2,
            scalePt(doubleRatio(w2, rec.getWidth()), p));
          //print(+pScaled);
          
          for (Pt p2 : pixelSpiral(pScaled.x, pScaled.y, w2, h2, pixelsPerSpiral)) {
            if (!pointsTried.add(p2)) continue;
      //print("Trying point " + p2);
            Pair<Rect, Double> result = rec.singleCheck(scaledInputImage, scaledProto, p2.x, p2.y, 0f);
            best.put(result);
          }
        }
      }
      Scored<Rect> bestInInputSpace = scored(scaleRect(doubleRatio(rec.getWidth(), w2), best!), best.score);
      print(bestInInputSpace);
    }
    
    setImage(inputImage);
    setSelection(bestInInputSpace!);
  }
}