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.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 <ICLCore/Img.h>
#include <ICLCore/Color.h>
#include <ICLFilter/UnaryOp.h>

namespace icl{
  namespace filter{

    class ICLFilter_API 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) ;

      ~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) ;

      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) ;

      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
}