E:/Projetos_Primeiro_Semestre_2012/TerraView/terralib/src/terralib/image_processing/TePDIParaSegRegGrowStrategy.hpp

Go to the documentation of this file.

/*
TerraLib - a library for developing GIS applications.
Copyright  2001, 2002, 2003 INPE and Tecgraf/PUC-Rio.

This code is part of the TerraLib library.
This library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License as published by the Free Software Foundation; either
version 2.1 of the License, or (at your option) any later version.

You should have received a copy of the GNU Lesser General Public
License along with this library.

The authors reassure the license terms regarding the warranties.
They specifically disclaim any warranties, including, but not limited to,
the implied warranties of merchantability and fitness for a particular
purpose. The library provided hereunder is on an "as is" basis, and the
authors have no obligation to provide maintenance, support, updates,
enhancements, or modifications.
In no event shall INPE be held liable to any party
for direct, indirect, special, incidental, or consequential damages arising
out of the use of this library and its documentation.
*/

#ifndef TEPDIPARASEGREGGROWSTRATEGY_HPP
  #define TEPDIPARASEGREGGROWSTRATEGY_HPP

  #include "TePDIParaSegStrategy.hpp"
  #include "TePDIParaSegSegment.hpp"
  #include "TePDIRegGrowSeg.hpp"
  
  #include <vector>

  class PDI_DLL TePDIParaSegRegGrowStrategy : public TePDIParaSegStrategy
  {
    public :
    
      class PDI_DLL Segment : public TePDIParaSegSegment
      {
        public :
        
          std::vector< double > bandsMeansVec_;
          
          double pixelsNumber_;       
             
          bool wasMerged_;             
        
          Segment();
          
          ~Segment();
      };    
    
      // This method has the same parameters as the 
      // TePDIParaSegStrategy constructor
      TePDIParaSegRegGrowStrategy( const TePDIParaSegStrategyParams& params );

      ~TePDIParaSegRegGrowStrategy();

      // overloaded
      bool execute( const RasterDataVecT& rasterDataVector, 
        TePDIParaSegSegmentsBlock::SegmentsPointersMatrixT& outSegsMatrix );
             
      //overloaded
      bool setParameters( const TePDIParameters& params );        
        
     //overloaded
      bool mergeSegments( 
       TePDIParaSegSegmentsBlock::SegmentsPointersMatrixT& centerMatrix,
       TePDIParaSegSegmentsBlock::SegmentsPointersMatrixT* topMatrixPtr,
       TePDIParaSegSegmentsBlock::SegmentsPointersMatrixT* leftMatrixPtr );  
          
      // just as the mergeSegments above
      static bool staticMergeSegments( 
        TePDIParaSegSegmentsBlock::SegmentsPointersMatrixT& centerMatrix,
        TePDIParaSegSegmentsBlock::SegmentsPointersMatrixT* topMatrixPtr,
        TePDIParaSegSegmentsBlock::SegmentsPointersMatrixT* leftMatrixPtr,
        double eucTreshold );  
                
      template< class StrategyTemplateT >
      static bool createSegsBlkFromLImg( const RasterDataVecT& rasterDataVector,
        TeRaster& labelRaster, StrategyTemplateT* stratPtr,
        TePDIParaSegSegmentsBlock::SegmentsPointersMatrixT& outSegsMatrix );
        
      //overloaded
      double getMemUsageFactor() const { return 1.0; };

    protected :
      
      class PDI_DLL MergingSegmentInfo
      {
        public:
        
        TePDIParaSegSegment const* segPtr_;
        
        TePDIParaSegSegmentsBlock::SegmentsPointersMatrixT const* mtxPtr_;
        
        unsigned int jointBorderLenght_;
        
        MergingSegmentInfo();
        
        ~MergingSegmentInfo();
      
      };      
      
      typedef std::vector< MergingSegmentInfo >  MergingSegmentInfoVecT;       
      
      TePDITypes::TePDIRasterPtrType inputRasterPtr_;
      
      TePDITypes::TePDIRasterPtrType outputRasterPtr_;
      
      TePDIParameters segParams_;
      
      TePDIRegGrowSeg segmenter_;
      
      double eucTreshold_;
      
      static bool staticLocateMergingSegments( Segment const* inputSegPtr,
        TePDIParaSegSegmentsBlock::SegmentsPointersMatrixT const* inputSegMtxPtr,
        const MergingSegmentInfoVecT& candidateSegmentsVec,
        std::vector< unsigned int >& locatedSegsIndexes, double eucTreshold );      

  };
    
template< class StrategyTemplateT >    
bool TePDIParaSegRegGrowStrategy::createSegsBlkFromLImg( 
  const RasterDataVecT& rasterDataVector, TeRaster& labelRaster,
  StrategyTemplateT* stratPtr,
  TePDIParaSegSegmentsBlock::SegmentsPointersMatrixT& outSegsMatrix )
{

  const unsigned int nLines = (unsigned int)labelRaster.params().nlines_;
  const unsigned int nCols = (unsigned int)labelRaster.params().ncols_;      
  const unsigned int nBands = (unsigned int)labelRaster.params().nBands(); 
  
  // Initiating the output matrix
  
  TEAGN_TRUE_OR_RETURN( outSegsMatrix.Reset( rasterDataVector[ 0 ].GetLines(),
    rasterDataVector[ 0 ].GetColumns() ),
    "output segments matrix reset error " );  
  
  unsigned int band = 0;
  unsigned int line = 0;
  unsigned int col = 0;
  TePDIParaSegSegment** segsMtxLinePtr = 0;
  TeDecoder& labelRasterDecoder = *(labelRaster.decoder());
  double value = 0;
  Segment* segPtr = 0;
  Segment* borderSegPtr = 0;
  const unsigned int lastLineIdx = nLines - 1;
  const unsigned int lastColIdx = nCols - 1;
  double oldSegPixelsNmb = 0;
  double newBandMeanValue = 0;
  std::map< double, Segment* > allSegsMap;
  std::map< double, Segment* > borderSegsMap;
  bool blockMergingEnabled = stratPtr->isBlockMergingEnabled();

  for( line = 0 ; line < nLines ; ++line )
  {
    segsMtxLinePtr = outSegsMatrix[ line ];
  
    for( col = 0 ; col < nCols ; ++col )
    {
      TEAGN_TRUE_OR_THROW( labelRasterDecoder.getElement( col, line, 
        value, 0 ), "Error writing input raster" )
      
      // Does this segment already was created ??
        
      segPtr = allSegsMap[ value ];
      if( segPtr == 0 )
      {
        segPtr = new Segment;
        segPtr->id_ = stratPtr->acquireNewSegID();
        
        allSegsMap[ value ] = segPtr;
      }
      
      segsMtxLinePtr[ col ] = segPtr;
      
      // Is this a border segment ??
      
      if( blockMergingEnabled && ( ( line == 0 ) || ( line == lastLineIdx ) || ( col == 0 ) || 
        ( col == lastColIdx ) ) )
      {
        borderSegPtr = borderSegsMap[ value ];
        
        if( borderSegPtr == 0 )
        {
          borderSegsMap[ value ] = segPtr;
          borderSegPtr = segPtr;
          
          borderSegPtr->pixelsNumber_ = 1.0;
          
          borderSegPtr->bandsMeansVec_.resize( nBands );
          for( band = 0 ; band < nBands ; ++band )
          {
            borderSegPtr->bandsMeansVec_[ band ] = rasterDataVector[ 
              band ][ line ][ col ];
          }
        }
        else
        {
          oldSegPixelsNmb = borderSegPtr->pixelsNumber_;
          ++(borderSegPtr->pixelsNumber_);
          
          for( band = 0 ; band < nBands ; ++band )
          {
            newBandMeanValue = ( borderSegPtr->bandsMeansVec_[ band ] * 
              oldSegPixelsNmb );
            newBandMeanValue += rasterDataVector[ band ][ line ][ col ];
            newBandMeanValue /= borderSegPtr->pixelsNumber_;
          
            borderSegPtr->bandsMeansVec_[ band ] = newBandMeanValue;
          }   
        }
      }
    }
  }
  
  return true;
}    
    
#endif