comparison maths/MedianFilter.h @ 392:e7010b8eb0c6

Median filter standalone method, and tests

 
 #include <algorithm>
 #include <cassert>
 #include <cmath>
 #include <iostream>
+#include <vector>
 
 template <typename T>
 class MedianFilter
 {
 public:

     void reset() {
 	for (int i = 0; i < m_size; ++i) m_frame[i] = 0;
 	for (int i = 0; i < m_size; ++i) m_sorted[i] = 0;
     }
 
+    static std::vector<T> filter(int size, const std::vector<T> &in) {
+        std::vector<T> out;
+        MedianFilter<T> f(size);
+        for (int i = 0; i < int(in.size()); ++i) {
+            f.push(in[i]);
+            T median = f.get();
+            if (i >= size/2) out.push_back(median);
+        }
+        while (out.size() < in.size()) {
+            f.push(T());
+            out.push_back(f.get());
+        }
+        return out;
+    }
+
 private:
     const int m_size;
     T *const m_frame;
     T *const m_sorted;
     T *const m_sortend;