1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
/**
* @file BestPatchList.h
*
* Sorted list evaluated non-overlaping patches
*/

#ifndef _BestPatchList_H_
#define _BestPatchList_H_

#include <Tools/Math/Vector2.h>
#include <list>

class BestPatchList
{

public:
  class Patch 
  {
  public:
    Patch() : value(-1) {}
    Patch(const Vector2i& min, const Vector2i& max, double value) 
      : min(min), max(max), value(value)
    {}
    Patch(int minX, int minY, int maxX, int maxY, double value) 
      : min(minX, minY), max(maxX, maxY), value(value)
    {}
    Vector2i min;
    Vector2i max;
    double value;
  };

  // make it cons iterable
  typedef std::list<Patch> PatchList;
  typedef std::list<Patch>::const_iterator iterator;
  typedef std::list<Patch>::const_reverse_iterator reverse_iterator;
  const iterator begin() const { return patches.cbegin(); }
  const iterator end() const { return patches.cend(); }
  const reverse_iterator rbegin() const { return patches.crbegin(); }
  const reverse_iterator rend() const { return patches.crend(); }


public:

  void add(int minX, int minY, int maxX, int maxY, double value) {
    add(Patch(minX, minY, maxX, maxY, value));
  }


  void add(const Patch& patch)
  {
    // add the first patch
    if(patches.empty()) {
      patches.push_back(patch);
      return;
    }

    // check if there are overlaping patches
    bool stop = false;
    for (std::list<Patch>::iterator i = patches.begin(); i != patches.end(); /*nothing*/)
    {
      if ( overlap(patch,(*i)) ) {

        // if the new patch is better, remove the old one
        if(patch.value > (*i).value) { 
          i = patches.erase(i);
        } else {
          stop = true;
          ++i;
        }
      } else {
        ++i;
      }
    }

    // there is an overlaping patch in the list which is better than the new
    if(stop) {
      return;
    }

    if(patches.empty()) {
      patches.push_back(patch);
      return;
    }

    // insertion sort - inset the new patch accoring to it's value
    for(std::list<Patch>::iterator i = patches.begin(); i != patches.end(); ++i)
    {
      // insert
      if(patch.value < (*i).value) {
        i = patches.insert(i,patch);
        break;
      } else if(nextIter(i) == patches.end()) {
        patches.push_back(patch);
        break;
      }
    }

    /*
    // TODO: do we need a limited size feature?
    if(patches.size() > 30) {
      patches.pop_front();
    }
    */
  }

  void addMean(int minX, int minY, int maxX, int maxY, double value)
  {
    addMean(Patch(minX, minY, maxX, maxY, value));
  }

  void addMean(const Patch& patch)
  {
    // add the first patch
    if(patches.empty()) {
      patches.push_back(patch);
      return;
    }

    // check if there are overlaping patches
    bool newPatch = true;
    for (std::list<Patch>::iterator i = patches.begin(); i != patches.end(); /*nothing*/)
    {
      if ( overlap(patch,(*i)) ) {
          (*i).max = ((*i).max + patch.max)/2;
          (*i).min = ((*i).min + patch.min)/2;
          newPatch = false;
          ++i;
      } else {
        ++i;
      }
    }

    // there is an overlaping patch in the list which is better than the new
    if(newPatch) {
      patches.push_back(patch);
    }

    /*
    // TODO: do we need a limited size feature?
    if(patches.size() > 30) {
      patches.pop_front();
    }
    */
  }

  void addPlain(int minX, int minY, int maxX, int maxY, double value)
  {
    patches.emplace_back(minX, minY, maxX, maxY, value);
  }

  void clear() {
    patches.clear();
  }

  size_t size() const {
    return patches.size();
  }

private:
  PatchList patches;

private:

  template <typename Iter>
  Iter nextIter(Iter iter) {
      return ++iter;
  }

  /*
  //OLD
  bool overlap(const Patch& one, const Patch& two) {
    return std::max(std::abs(two.center.x - one.center.x), std::abs(two.center.y - one.center.y)) < (two.radius + one.radius);
  }
  */

  bool overlap(const Patch& one, const Patch& two) {
    return 
      one.min.x < two.max.x && 
      one.max.x > two.min.x &&
      one.min.y < two.max.y && 
      one.max.y > two.min.y;
  }

  bool overlapTest(const Patch& one, const Patch& two) {<--- Unused private function: 'BestPatchList::overlapTest'
    int w1 = (one.max.x - one.min.x)/4;
    int w2 = (two.max.x - two.min.x)/4;
    
    return 
      one.min.x+w1 < two.max.x-w2 && 
      one.max.x-w1 > two.min.x+w2 &&
      one.min.y+w1 < two.max.y-w2 && 
      one.max.y-w1 > two.min.y+w2;
  }

};

#endif // _BestPatchList_H_