Skip to content
GitLab
Select Git revision
  • master default
  • enhancements
Find file BlameHistoryPermalink
Forked from Cresson Remi / SimpleExtractionTools
Source project has a limited visibility.
otbMosaicFromDirectoryHandler.txx 4.66 KiB
1 
/*
2 
 * otbMosaicFromDirectoryHandler.hxx
3 
 *
4 
 *  Created on: 24 mars 2016
5 
 *      Author: cresson
6 
 */
7
8 
#ifndef MODULES_REMOTE_OTB_MosaicFromDirectoryHandler_INCLUDE_OTBMosaicFromDirectoryHandler_HXX_
9 
#define MODULES_REMOTE_OTB_MosaicFromDirectoryHandler_INCLUDE_OTBMosaicFromDirectoryHandler_HXX_
10
11 
#include "otbMosaicFromDirectoryHandler.h"
12 
#include "otbImageFileWriter.h"
13
14 
namespace otb
15 
{
16
17 
template <class TOutputImage, class TReferenceImage>
18 
MosaicFromDirectoryHandler<TOutputImage, TReferenceImage>
19 
::MosaicFromDirectoryHandler()
20 
 {
21 
  mosaicFilter = MosaicFilterType::New();
22 
  castFilter = CastFilterType::New();
23 
  m_UseReferenceImage = false;
24 
  m_RefImagePtr = 0;
25 
 }
26
27 
template <class TOutputImage, class TReferenceImage>
28 
MosaicFromDirectoryHandler<TOutputImage, TReferenceImage>
29 
::~MosaicFromDirectoryHandler()
30 
{
31 
}
32
33 
template <class TOutputImage, class TReferenceImage>
34 
void
35 
MosaicFromDirectoryHandler<TOutputImage, TReferenceImage>
36 
::GenerateOutputInformation()
37 
 {
38 
  if (m_Directory[m_Directory.size()-1] != '/')
39 
    {
40 
    // If not, we add the separator
41 
    m_Directory.append("/");
42 
    }
43
44 
  // Get the list of files in the directory
45 
  itk::Directory::Pointer dir = itk::Directory::New();
46 
  if (!dir->Load(m_Directory.c_str()))
47 
    {
48 
    itkExceptionMacro(<< "Unable to browse directory " << m_Directory);
49 
    }
50
51 
  // Instanciate a new mosaic filter
52 
  mosaicFilter = MosaicFilterType::New();
53 
  mosaicFilter->SetGlobalWarningDisplay(false);
54 
  readers.clear();
55 
  resamplers.clear();
56
57 
  // Browse the directory
58 
  for (unsigned int i = 0; i < dir->GetNumberOfFiles(); i++)
59 
    {
60 
    const char *filename = dir->GetFile(i);
61 
    std::string sfilename(filename);
62 
    sfilename = m_Directory + sfilename;
63
64 
    // Try to read the file
65 
    otb::ImageIOBase::Pointer imageIO =
66 
        otb::ImageIOFactory::CreateImageIO(sfilename.c_str(),otb::ImageIOFactory::ReadMode);
67 
    if( imageIO.IsNotNull() )
68 
      {
69 
        // create reader
70 
        ReaderPointerType reader = ReaderType::New();
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
reader->SetFileName(sfilename); reader->UpdateOutputInformation(); readers.push_back(reader); if (m_UseReferenceImage) { ResamplerPointerType resampler = ResamplerType::New(); resampler->SetInput(reader->GetOutput()); // Setup transform through projRef and Keywordlist SpacingType defSpacing = m_RefImagePtr->GetSignedSpacing(); defSpacing[0] *= 10; defSpacing[1] *= 10; resampler->SetDisplacementFieldSpacing(defSpacing); resampler->SetInputKeywordList(reader->GetOutput()->GetImageKeywordlist()); resampler->SetInputProjectionRef(reader->GetOutput()->GetProjectionRef()); resampler->SetOutputKeywordList(m_RefImagePtr->GetImageKeywordlist()); resampler->SetOutputProjectionRef(m_RefImagePtr->GetProjectionRef()); resampler->SetOutputOrigin(m_RefImagePtr->GetOrigin()); resampler->SetOutputSpacing(m_RefImagePtr->GetSignedSpacing()); resampler->SetOutputSize(m_RefImagePtr->GetLargestPossibleRegion().GetSize()); resampler->SetOutputStartIndex(m_RefImagePtr->GetLargestPossibleRegion().GetIndex()); typename NNInterpolatorType::Pointer interpolator = NNInterpolatorType::New(); resampler->SetInterpolator(interpolator); resamplers.push_back(resampler); mosaicFilter->PushBackInput(resampler->GetOutput()); } else { mosaicFilter->PushBackInput(reader->GetOutput()); } } else { // itkWarningMacro(<<"Unable to read file " << sfilename); } } if (m_UseReferenceImage) { mosaicFilter->SetOutputOrigin(m_RefImagePtr->GetOrigin()); mosaicFilter->SetOutputSpacing(m_RefImagePtr->GetSignedSpacing()); mosaicFilter->SetOutputSize(m_RefImagePtr->GetLargestPossibleRegion().GetSize()); } else { mosaicFilter->SetOutputOrigin(m_OutputOrigin); mosaicFilter->SetOutputSpacing(m_OutputSpacing); mosaicFilter->SetOutputSize(m_OutputSize); } mosaicFilter->SetAutomaticOutputParametersComputation(false); castFilter->SetInput(mosaicFilter->GetOutput()); castFilter->GraftOutput( this->GetOutput() ); castFilter->UpdateOutputInformation(); this->GraftOutput( castFilter->GetOutput() ); } template <class TOutputImage, class TReferenceImage> void MosaicFromDirectoryHandler<TOutputImage, TReferenceImage> ::GenerateData() { castFilter->GraftOutput( this->GetOutput() );
141142143144145146147148
castFilter->Update(); this->GraftOutput( castFilter->GetOutput() ); } } // end namespace otb #endif /* MODULES_REMOTE_OTB_MosaicFromDirectoryHandler_INCLUDE_OTBMosaicFromDirectoryHandler_HXX_ */