aos/vision/blob/range_image.h

#ifndef _AOS_VISION_BLOB_RANGE_IMAGE_H_
#define _AOS_VISION_BLOB_RANGE_IMAGE_H_

#include <vector>

#include "aos/vision/image/image_types.h"

namespace aos::vision {

struct Point {
  int x;
  int y;
};

struct ImageRange {
  ImageRange(int a, int b) : st(a), ed(b) {}
  int st;
  int ed;
  int last() const { return ed - 1; }
  int calc_width() const { return ed - st; }

  bool operator<(const ImageRange &o) const { return st < o.st; }
  bool operator==(const ImageRange &other) const {
    return st == other.st && ed == other.ed;
  }
  bool operator!=(const ImageRange &other) const { return !(*this == other); }
};

void PrintTo(const ImageRange &range, std::ostream *os);

// Image in pre-thresholded run-length encoded format.
class RangeImage {
 public:
  RangeImage(int min_y, std::vector<std::vector<ImageRange>> &&ranges)
      : min_y_(min_y), ranges_(std::move(ranges)) {}
  explicit RangeImage(int l) { ranges_.reserve(l); }
  RangeImage() {}

  bool operator==(const RangeImage &other) const {
    if (min_y_ != other.min_y_) {
      return false;
    }
    if (ranges_ != other.ranges_) {
      return false;
    }
    return true;
  }
  bool operator!=(const RangeImage &other) const { return !(*this == other); }

  int size() const { return ranges_.size(); }

  // Returns the total number of included pixels.
  int npixels();
  // Calculates the total number of included pixels.
  //
  // TODO(Brian): Present a nicer API than the current duality between this and
  // npixels(), which is annoying because npixels() has to modify the cached
  // data so it can't be const.
  int calc_area() const;

  void Flip(ImageFormat fmt) { Flip(fmt.w, fmt.h); }
  void Flip(int image_width, int image_height);

  std::vector<std::vector<ImageRange>>::const_iterator begin() const {
    return ranges_.begin();
  }

  std::vector<std::vector<ImageRange>>::const_iterator end() const {
    return ranges_.end();
  }

  const std::vector<std::vector<ImageRange>> &ranges() const { return ranges_; }

  int min_y() const { return min_y_; }

  int mini() const { return min_y_; }

  int height() const { return min_y_ + ranges_.size(); }

 private:
  // minimum index in y where the blob starts
  int min_y_ = 0;

  // Each vector<ImageRange> represents all the matched ranges in a given row.
  // Each ImageRange within that row represents a run of pixels which matches.
  std::vector<std::vector<ImageRange>> ranges_;

  // Cached pixel count.
  int npixelsc_ = -1;
};

void PrintTo(const RangeImage &range, std::ostream *os);

typedef std::vector<RangeImage> BlobList;
typedef std::vector<const RangeImage *> BlobLRef;

// Debug print range image as ranges.
std::string ShortDebugPrint(const BlobList &blobl);
// Debug print range image as ### for the ranges.
void DebugPrint(const BlobList &blobl);

}  // namespace aos::vision

#endif  // _AOS_VISION_BLOB_RANGE_IMAGE_H_