ObjectEdgeDetector.h

Go to the documentation of this file.

/********************************************************************
**                Image Component Library (ICL)                    **
**                                                                 **
** Copyright (C) 2006-2013 CITEC, University of Bielefeld          **
**                         Neuroinformatics Group                  **
** Website: www.iclcv.org and                                      **
**          http://opensource.cit-ec.de/projects/icl               **
**                                                                 **
** File   : ICLGeom/src/ICLGeom/ObjectEdgeDetector.h               **
** Module : ICLGeom                                                **
** Authors: Andre Ueckermann                                       **
**                                                                 **
**                                                                 **
** GNU LESSER GENERAL PUBLIC LICENSE                               **
** This file may be used under the terms of the GNU Lesser General **
** Public License version 3.0 as published by the                  **
**                                                                 **
** Free Software Foundation and appearing in the file LICENSE.LGPL **
** included in the packaging of this file.  Please review the      **
** following information to ensure the license requirements will   **
** be met: http://www.gnu.org/licenses/lgpl-3.0.txt                **
**                                                                 **
** The development of this software was supported by the           **
** Excellence Cluster EXC 277 Cognitive Interaction Technology.    **
** The Excellence Cluster EXC 277 is a grant of the Deutsche       **
** Forschungsgemeinschaft (DFG) in the context of the German       **
** Excellence Initiative.                                          **
**                                                                 **
********************************************************************/

#pragma once

#include <ICLUtils/CompatMacros.h>
#include <ICLGeom/GeomDefs.h>
#include <ICLCore/Img.h>
#include <ICLGeom/Camera.h>
#include <ICLUtils/Uncopyable.h>
#include <ICLCore/DataSegment.h>

#include <ICLGeom/ObjectEdgeDetectorPlugin.h>

namespace icl{
  namespace geom{

    class ICLGeom_API ObjectEdgeDetector{

     struct Data;  
     Data *m_data; 

     public:

      enum Mode {BEST, GPU, CPU};


      ObjectEdgeDetector(Mode mode=BEST);

      virtual ~ObjectEdgeDetector();


      const core::Img8u &calculate(const core::Img32f &depthImage, bool filter, bool average, bool gauss);


      void setDepthImage(const core::Img32f &depthImg);


      void applyMedianFilter();


      const core::Img32f &getFilteredDepthImage();


      void setFilteredDepthImage(const core::Img32f &filteredImg);


      void applyNormalCalculation();


      void applyLinearNormalAveraging();


      void applyGaussianNormalSmoothing();


      const core::DataSegment<float,4> getNormals();


      void applyWorldNormalCalculation(const Camera &cam);


      const core::DataSegment<float,4> getWorldNormals();


      const core::Img8u &getRGBNormalImage();


      void setNormals(core::DataSegment<float,4> pNormals);


      void applyAngleImageCalculation();


      const core::Img32f &getAngleImage();


      void setAngleImage(const core::Img32f &angleImg);


      void applyImageBinarization();


      const core::Img8u &getBinarizedAngleImage();


      void setMedianFilterSize(int size);


      void setNormalCalculationRange(int range);


      void setNormalAveragingRange(int range);


      void setAngleNeighborhoodMode(int mode);


      void setAngleNeighborhoodRange(int range);


      void setBinarizationThreshold(float threshold);


      void setUseNormalAveraging(bool use);


      void setUseGaussSmoothing(bool use);

     private:
      ObjectEdgeDetectorPlugin* objectEdgeDetector;

      void initialize(utils::Size size);

    };
  }
}