DeleteShortLines.cpp

Go to the documentation of this file.

#include <face_contour_detector/filters/DeleteShortLines.h>

#include <face_contour_detector/ImageForest.h>
#include <opencv/cv.h>
#include <string>

namespace face_contour_detector {
        namespace filters {

                DeleteShortLines::DeleteShortLines() {
                        ResetParameters();
                }

                std::vector<Parameter> DeleteShortLines::GetParameters() {
                        std::vector<Parameter> re;
                        std::string name("minNumPixels");
                        re.push_back(Parameter(name, &m_minNumPixels, 1, 50));
                        return re;
                }

                void DeleteShortLines::Apply(const cv::Mat& input, cv::Mat& result) {
                        assert(input.channels() == 1);
                        result = input;
                        ImageForest forest(input, 200);
                        std::map<int, ImageForest::Graph> graphs = forest.GetGraphs();
                        bool* toDelete = new bool[forest.NumGraphs()+1];
                        toDelete[0] = false;
                        for (int i = 1; i <= forest.NumGraphs(); i++) {
                                if (graphs[i].numPixels >= m_minNumPixels) {
                                        toDelete[i] = false;
                                } else {
                                        toDelete[i] = true;
                                }
                        }
                        for (int y = 0; y < input.rows; y++) {
                                for (int x = 0; x < input.cols; x++) {
                                        if (forest.GetGraphId(x,y) != 0 && toDelete[forest.GetGraphId(x,y)]) {
                                                result.at<unsigned char>(y,x) = 0;
                                        }
                                }
                        }
                        delete toDelete;
                }

                void DeleteShortLines::ResetParameters() {
                        m_minNumPixels = 3;
                }

                const std::string& DeleteShortLines::GetFilterName() {
                        return m_filterName;
                }

                //static
                std::string DeleteShortLines::m_filterName = std::string("DeleteShortLines");
        } //namepspace filters
} //namespace face_contour_detector