ShowEndpoints.cpp

Go to the documentation of this file.

#include <face_contour_detector/filters/ShowEndpoints.h>
#include <face_contour_detector/ImageForest.h>
#include <string>
#include <face_contour_detector/Bgr8.h>

namespace face_contour_detector {
        namespace filters {

                ShowEndpoints::ShowEndpoints() {
                        ResetParameters();
                }

                std::vector<Parameter> ShowEndpoints::GetParameters() {
                        std::vector<Parameter> re;
                        return re;
                }

                void ShowEndpoints::Apply(const cv::Mat& input, cv::Mat& result) {
                        ImageForest forest = ImageForest(input, 200);
                        std::map<int, ImageForest::Graph> graphs = forest.GetGraphs();
                        result = cv::Mat(input.rows, input.cols, CV_8UC3);
                        for (int y = 0; y < input.rows; y++) {
                                for (int x = 0; x < input.cols; x++) {
                                        result.at<Bgr8>(y,x).b = input.at<unsigned char>(y,x);
                                        result.at<Bgr8>(y,x).g = input.at<unsigned char>(y,x);
                                        result.at<Bgr8>(y,x).r = input.at<unsigned char>(y,x);
                                }
                        }
                        for (int i = 0; i < forest.NumGraphs(); i++) {
                                std::vector<ImageForest::Point>::iterator it;
                                for  (it = graphs[i].endPoints.begin(); it != graphs[i].endPoints.end(); it++) {
                                        result.at<Bgr8>(it->y, it->x).b = 0;
                                        result.at<Bgr8>(it->y, it->x).g = 255;
                                        result.at<Bgr8>(it->y, it->x).r = 0;
                                }
                        }
                }

                void ShowEndpoints::ResetParameters() {

                }

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

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