ColorSegmentationOp.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   : ICLFilter/src/ICLFilter/ColorSegmentationOp.h          **
** Module : ICLFilter                                              **
** Authors: Christof Elbrechter                                    **
**                                                                 **
**                                                                 **
** 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.GPL  **
** 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 <ICLFilter/UnaryOp.h>
#include <ICLCore/Img.h>
#include <ICLCore/Color.h>

namespace icl{
  namespace filter{

    class ColorSegmentationOp : public UnaryOp{
      public:
      class LUT3D; 
  
      private:
      core::format m_segFormat;       
      core::Img8u m_inputBuffer;      
      core::Img8u m_outputBuffer;     
      core::Img8u m_segPreview;       
      core::Img8u m_lastDst;          
      icl8u m_bitShifts[3];     
      LUT3D *m_lut;             
  
      public:
      

      ColorSegmentationOp(icl8u c0shift=2, icl8u c1shift=2, icl8u c2shift=2,core::format fmt=core::formatYUV) throw (utils::ICLException);
  
      ~ColorSegmentationOp();
      
      virtual void apply(const core::ImgBase *src, core::ImgBase **dst);
      
      using UnaryOp::apply;
      
      icl8u classifyPixel(icl8u r, icl8u g, icl8u b);
          
      void clearLUT(icl8u value=0);
      
      void setSegmentationShifts(icl8u c0shift, icl8u c1shift, icl8u c2shift);
      
      void setSegmentationFormat(core::format fmt) throw (utils::ICLException);
  
      const icl8u *getSegmentationShifts() const { return m_bitShifts; }
      
      core::format getSegmentationFormat() const { return m_segFormat; }
      

      const core::Img8u &getSegmentationPreview();
      
  
      void lutEntry(icl8u a, icl8u b, icl8u c, icl8u rA, icl8u rB, icl8u rC, icl8u value);
      
      void lutEntry(core::format fmt, int a, int b, int c, int rA, int rB, int rC, icl8u value) throw (utils::ICLException);
      
      void load(const std::string &filename);
      
      void save(const std::string &filename);
      

      const core::Img8u &getLUTPreview(int xDim, int yDim, icl8u zValue);
  

      const core::Img8u &getColoredLUTPreview(int xDim, int yDim, icl8u zValue);
      
      const std::vector<core::Color> &getClassMeanColors();
      

      const icl8u *getLUT() const;
  

      icl8u *getLUT();
  

      void getLUTDims(int &w, int &h, int &t) const;
  
    };
  
  } // namespace filter
}