Skip to content
GitLab
Select Git revision
  • master default
Find file BlameHistoryPermalink
otbClassificationMosaicArgmaxFilter.hxx 4.25 KiB
1 
/*=========================================================================
2
3 
  Copyright (c) Remi Cresson (IRSTEA). All rights reserved.
4
5
6 
     This software is distributed WITHOUT ANY WARRANTY; without even
7 
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
8 
     PURPOSE.  See the above copyright notices for more information.
9
10 
=========================================================================*/
11 
#ifndef __ClassificationMosaicArgmaxFilter_txx
12 
#define __ClassificationMosaicArgmaxFilter_txx
13
14 
#include "otbClassificationMosaicArgmaxFilter.h"
15
16 
namespace otb {
17
18 
template <class TInputImage, class TOutputImage, class TInternalValueType>
19 
void
20 
ClassificationMosaicArgmaxFilter<TInputImage, TOutputImage, TInternalValueType>
21 
::GenerateOutputInformation(void)
22 
{
23 
  Superclass::GenerateOutputInformation();
24
25 
  // Output pixel has 1 channel
26 
  unsigned int outNbBands = 1;
27 
  OutputImageType * outputPtr = this->GetOutput();
28 
  outputPtr->SetNumberOfComponentsPerPixel(outNbBands);
29
30 
  // No data value is 255
31 
  OutputImagePixelType noDataPix;
32 
  noDataPix.SetSize(outNbBands);
33 
  noDataPix.Fill(255);
34 
  this->SetNoDataOutputPixel(noDataPix);
35 
}
36
37 
/**
38 
 * Processing
39 
 */
40 
template <class TInputImage, class TOutputImage, class TInternalValueType>
41 
void
42 
ClassificationMosaicArgmaxFilter<TInputImage, TOutputImage, TInternalValueType>
43 
::ThreadedGenerateData(const OutputImageRegionType& outputRegionForThread, itk::ThreadIdType threadId)
44 
{
45
46 
  // Debug info
47 
  itkDebugMacro(<<"Actually executing thread " << threadId << " in region " << outputRegionForThread);
48
49 
  // Support progress methods/callbacks
50 
  itk::ProgressReporter progress(this, threadId, outputRegionForThread.GetNumberOfPixels() );
51
52 
  // Get output pointer
53 
  OutputImageType * mosaicImage = this->GetOutput();
54
55 
  // Get number of used inputs
56 
  const unsigned int nbOfUsedInputImages = Superclass::GetNumberOfUsedInputImages();
57
58 
  // Get number of bands
59 
  const unsigned int nBands = Superclass::GetNumberOfBands();
60
61 
  // Iterate through the thread region
62 
  IteratorType outputIt(mosaicImage, outputRegionForThread);
63
64 
  // Prepare interpolated pixel
65 
  InternalPixelType interpolatedMathPixel;
66 
  interpolatedMathPixel.SetSize(nBands);
67
68 
  // Prepare input pointers, interpolators, and valid regions (input images)
69 
  typename std::vector<InputImageType *>        currentImage;
70 
  typename std::vector<InterpolatorPointerType> interp;
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139
Superclass::PrepareImageAccessors(currentImage, interp); // Container for geo coordinates OutputImagePointType geoPoint; InputImagePixelType tmpPixel(Superclass::GetNoDataInputPixel() ); bool isData; for ( outputIt.GoToBegin(); !outputIt.IsAtEnd(); ++outputIt ) { // Prepare output pixel tmpPixel.Fill(0); isData = false; OutputImagePixelType outPixel(Superclass::GetNoDataOutputPixel() ); // Current pixel --> Geographical point mosaicImage->TransformIndexToPhysicalPoint (outputIt.GetIndex(), geoPoint) ; // Loop on used input images for (unsigned int i = 0 ; i < nbOfUsedInputImages ; i++) { // Check if the point is inside the transformed thread region // (i.e. the region in the current input image which match the thread // region) if (interp[i]->IsInsideBuffer(geoPoint) ) { // Compute the interpolated pixel value InputImagePixelType interpolatedPixel = interp[i]->Evaluate(geoPoint); // Check that interpolated pixel is not empty if (Superclass::IsPixelNotEmpty(interpolatedPixel) ) { // Update the output pixel isData = true; for (unsigned int band = 0 ; band < nBands ; band++) { tmpPixel[band] += interpolatedPixel[band]; } } // Interpolated pixel is not empty } // point inside buffer } // next image // Update output pixel value if (isData) { OutputImageInternalPixelType argmax = 0; float max = -1.0; for (unsigned int band = 0 ; band < nBands ; band++) { if (tmpPixel[band] > max) { argmax = band; max = tmpPixel[band]; } } outPixel[0] = argmax; } outputIt.Set(outPixel); // Update progress progress.CompletedPixel(); } // next output pixel } } // end namespace gtb #endif