ImageInfo.hh

   1 #ifndef _GOIR_IMAGEINFO_HH
   2 #define _GOIR_IMAGEINFO_HH
   3
   4 #include "sImLib/Image.hh"
   5
   6 #include "histogram.hh"
   7 #include "util.hh"
   8 #include "uvector.hh"
   9
  10 #include <cstdio>
  11 #include <vector>
  12
  13 namespace goir {
  14
  15   class ImageInfo {
  16
  17     class LineData;
  18
  19   public:
  20     enum { ALGO_1PIX_PERCOMP,
  21            ALGO_1PIX_COMPSUM,
  22            ALGO_1PIX_COMPSUM_FILTERED,
  23            //    ALGO_1PIX_COMBINED,
  24            ALGO_2PIX_PERCOMP,
  25            ALGO_2PIX_COMPSUM,
  26            //    ALGO_2PIX_COMBINED,
  27
  28            NB_ALGOS
  29     };
  30
  31   private:
  32
  33     class AlgoResults {
  34     public:
  35       struct MatchData {
  36         unsigned x,y;
  37         unsigned delta;
  38         bool false_match;
  39
  40         MatchData(unsigned _x, unsigned _y, unsigned _delta, bool _is_known = true)
  41           : x(_x), y(_y), delta(_delta), false_match(!_is_known) {};
  42         MatchData(const std::pair<unsigned,unsigned> p)
  43           : x(p.first), y(p.second), delta(0), false_match(false) {};
  44       };
  45
  46       // results of algorithm
  47       std::vector<MatchData> found_edges;
  48
  49       /*** everything below is for analysis only ***/
  50
  51       const LineData* linedata;
  52
  53       // watermarks for calibration
  54       unsigned min_positive, max_negative;
  55
  56       std::vector<MatchData> missed;
  57
  58       AlgoResults(unsigned maxvalue, const LineData* _linedata)
  59         : linedata(_linedata)
  60         , min_positive(maxvalue), max_negative(0)
  61       {}
  62
  63       void update_watermarks(bool is_edge, unsigned value);
  64       void fprint(FILE* outf);
  65       void draw_on(sImLib::Image & img);
  66     };
  67
  68     class LineData {
  69     private:
  70       ImageInfo* ii; // line lives in this image
  71       bool _edges_are_known;
  72
  73     public:
  74       unsigned x0, y0, x1, y1;
  75       UVector known_edges;
  76       std::vector<AlgoResults> results;
  77
  78       LineData(ImageInfo* _ii, unsigned _x0, unsigned _y0, unsigned _x1, unsigned _y1)
  79         : ii(_ii)
  80         , _edges_are_known(false)
  81         , x0(_x0), y0(_y0), x1(_x1), y1(_y1)
  82       {
  83         // FIXME: I hate that way of initializing a vector...
  84         results.push_back(AlgoResults(255, this)); // 1PIX_PERCOMP
  85         results.push_back(AlgoResults(3*255, this)); // 1PIX_COMPSUM
  86         results.push_back(AlgoResults(3*255, this)); // 1PIX_COMPSUM_FILTER
  87         results.push_back(AlgoResults(255, this)); // 2PIX_PERCOMP
  88         results.push_back(AlgoResults(3*255, this)); // 2PIX_COMPSUM
  89         assert(NB_ALGOS == 5);
  90       }
  91
  92       // Read a list of unsigned values as coords of edges along the line
  93       void read_edges(FILE* f);
  94
  95       bool edges_are_known() const {
  96         return _edges_are_known;
  97       }
  98
  99       void add_to_histo() {
 100         ii->add_to_histo(x0, y0, x1, y1, &known_edges);
 101       }
 102
 103       void find_edges();
 104       void fprint(FILE* outf);
 105
 106     private:
 107       void maybe_update_watermarks(int algo_id, unsigned x,
 108                                    bool is_edge, unsigned value);
 109     };
 110
 111   private:
 112     const sImLib::Image * const imgp;
 113
 114     std::vector<Histogram> histos;
 115     std::vector< std::vector<Histogram> > histos_c;
 116     std::vector<Histogram> histos_allc;
 117
 118   public:
 119     std::vector<LineData*> lines;
 120     enum { NONEDGES, EDGES };
 121
 122     ImageInfo(const sImLib::Image* const _imgp)
 123       : imgp(_imgp)
 124       , histos(2, Histogram(3*256, 3*256))
 125       , histos_c (2, std::vector<Histogram>(3, Histogram(256,256)))
 126       , histos_allc(2, Histogram(256, 256))
 127     {}
 128
 129     void maybe_read_edge_data();
 130
 131     void add_line(unsigned x0, unsigned y0, unsigned x1, unsigned y1) {
 132       lines.push_back(new LineData(this, x0, y0, x1, y1));
 133     }
 134
 135     void collect_histo_from_lines() {
 136       const_foreach(i, lines)
 137         (*i)->add_to_histo();
 138     }
 139
 140     void display_histo();
 141
 142     void look_for_edges() const {
 143       const_foreach(i, lines)
 144         (*i)->find_edges();
 145     }
 146
 147     void fprint(FILE* outf) const {
 148       fprintf(outf, "# GoIR data\n");
 149       const_foreach(i, lines)
 150         (*i)->fprint(outf);
 151     }
 152
 153   private:
 154     void add_to_histo(unsigned x0, unsigned y0, unsigned x1, unsigned y1,
 155                       UVector *known_edges = NULL);
 156   };
 157
 158 }
 159
 160 #endif // _GOIR_IMAGEINFO_HH