ReplaceUndefImage.cpp

Go to the documentation of this file.

 
#include "SEImplementation/Background/SE/ReplaceUndefImage.h"
 
namespace SourceXtractor {
 
template<typename T>
static T getMaskedValue(int x, int y, const VectorImage<T>& img, T invalid) {
  auto v = img.getValue(x, y);
  if (v != invalid)
    return v;
 
  auto min_distance = std::numeric_limits<T>::max();
 
  T acc = 0;
  size_t count = 0;
 
  for (int iy = 0; iy < img.getHeight(); ++iy) {
    for (int ix = 0; ix < img.getWidth(); ++ix) {
      v = img.getValue(ix, iy);
      if (v != invalid) {
        auto dx = x - ix;
        auto dy = y - iy;
        auto distance = dx * dx + dy * dy;
        // This pixel is closer than the last close one, so restart
        if (distance < min_distance) {
          acc = v;
          count = 1;
          min_distance = distance;
        }
          // This pixel is as close as the closest one, so take it into account
        else if (distance - min_distance <= std::numeric_limits<T>::epsilon()) {
          acc += v;
          ++count;
        }
      }
    }
  }
 
  // Take the average
  if (count > 0)
    acc /= count;
 
  return acc;
}
 
template<typename T>
std::shared_ptr<VectorImage<T>> ReplaceUndef(const VectorImage<T>& original, T mask) {
  auto output = VectorImage<T>::create(original.getWidth(), original.getHeight());
  for (int y = 0; y < original.getHeight(); ++y) {
    for (int x = 0; x < original.getWidth(); ++x) {
      output->at(x, y) = getMaskedValue(x, y, original, mask);
    }
  }
  return output;
}
 
// Instantiation
template std::shared_ptr<VectorImage<SeFloat>> ReplaceUndef(const VectorImage<SeFloat>&, SeFloat);
 
} // end of namespace SourceXtractor
 
#include "SEFramework/Image/ImageAccessor.h"