Not logged in.  Login/Logout/Register | List snippets | | Create snippet | Upload image | Upload data

84
LINES

< > BotCompany Repo | #1034419 // G22BWWalker [dev.]

JavaX fragment (include)

// has a rectangular window (visionW*visionH)
// walking range per frame is a circle with radius "range"
sclass G22Walker {
  // configuration (set)
  int range = 1; // how far we can walk in each frame
  double roundingPolicy = .9; // for making the range circle (between 0 and 1)
  int visionW = 3, visionH = 3; // window rectangle size (both should be odd)
  
  // current position (set/get)
  Pt position; // current position (center), not null
  
  // the ideal image it is looking for (set/get)
  byte[] rememberedImage; // size visionW*visionH
  
  // status variables (read)
  bool stuck; // can neither stay nor move anywhere for some reason
  int guessWork; // how many "optimal" new positions we chose from
  
  void move(IBWImage image) {
    stuck = false;
    guessWork = 0;
    
    // How far can I move until I hit the image's borders
    Rect movement1 = rectFromPoints(visionW/2, visionH/2,
      image.getWidth()-(visionW+1)/2, (visionH+1)/2);
    
    // How far can I walk in one frame
    Rect movement2 = rectAroundPt(position, range*2+1);
    
    Rect movement = intersectRects(movement1, movement2);
    if (empty(movement)) ret with set stuck;
    
    printVars(+movement, +movement1, +movement2);
    
    if (rememberedImage == null)
      ret with rememberImage(image);
    
    Rect r = window_uncropped();
    
    new MultiBest<Pt> bestPositions;
    
    int mx1 = movement.x1(), mx2 = movement.x2();
    int my1 = movement.y1(), my2 = movement.y2();
    for (int y = my1; y < my2; y++)
      for (int x = mx1; x < mx2; x++) {
        var newPos = pt(x, y);
        bestPositions.put(newPos, scorePosition(newPos, image));
      }
    var best = bestPositions!;
    guessWork = l(best);
    position = random(best);
    
    // It's actually a question how often we grab a fresh image.
    // For now, every time.
    rememberImage(image);
  }
  
  // only works when we already have a rememberedImage
  double scorePosition(Pt pos, IBWImage image) {
    int vx = pos.x-visionW, vy = pos.y-visionH, i = 0;
    int diff = 0;
    for y to visionH:
      for x to visionW:
        diff += abs(ubyteToInt(rememberedImage[i++])-image.getInt(vx+x, vy+y));
    ret -diff;
  }
  
  void rememberImage(IBWImage image) {
    if (rememberedImage == null)
      rememberedImage = new byte[visionW*visionH];
      
    int vx = visionX1(), vy = visionY1(), i = 0;
    for y to visionH:
      for x to visionW:
        rememberedImage[i++] = clampToUByte(image.getInt(vx+x, vy+y));
  }
  
  int visionX1() { ret position.x-visionW; }
  int visionY1() { ret position.y-visionH; }
  
  Rect window_uncropped(Pt center default position) {
    ret rectAroundPt(center, visionW, visionH);
  }
}

Author comment

Began life as a copy of #1034119

download  show line numbers  debug dex  old transpilations   

Travelled to 3 computer(s): bhatertpkbcr, mowyntqkapby, mqqgnosmbjvj

No comments. add comment

Snippet ID: #1034419
Snippet name: G22BWWalker [dev.]
Eternal ID of this version: #1034419/1
Text MD5: 01a26ef3ca8f8d06918024ba3a062814
Author: stefan
Category: javax / imaging
Type: JavaX fragment (include)
Public (visible to everyone): Yes
Archived (hidden from active list): No
Created/modified: 2022-02-06 03:56:08
Source code size: 2794 bytes / 84 lines
Pitched / IR pitched: No / No
Views / Downloads: 71 / 85
Referenced in: [show references]