otbTensorflowStreamerFilter.hxx 3 KB
Newer Older
remi cresson's avatar
remi cresson committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
/*=========================================================================

  Copyright (c) Remi Cresson (IRSTEA). All rights reserved.


     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notices for more information.

=========================================================================*/
#ifndef otbTensorflowStreamerFilter_txx
#define otbTensorflowStreamerFilter_txx

#include "otbTensorflowStreamerFilter.h"
#include "itkImageAlgorithm.h"

namespace otb
{

template <class TInputImage, class TOutputImage>
TensorflowStreamerFilter<TInputImage, TOutputImage>
::TensorflowStreamerFilter()
 {
  m_OutputGridSize.Fill(1);
 }

/**
 * Compute the output image
 */
template <class TInputImage, class TOutputImage>
void
TensorflowStreamerFilter<TInputImage, TOutputImage>
::GenerateData()
 {
  // Output pointer and requested region
  OutputImageType * outputPtr = this->GetOutput();
  const RegionType outputReqRegion = outputPtr->GetRequestedRegion();
38
  outputPtr->SetBufferedRegion(outputReqRegion);
remi cresson's avatar
remi cresson committed
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
  outputPtr->Allocate();

  // Compute the aligned region
  RegionType region;
  for(unsigned int dim = 0; dim<OutputImageType::ImageDimension; ++dim)
    {
    // Get corners
    IndexValueType lower = outputReqRegion.GetIndex(dim);
    IndexValueType upper = lower + outputReqRegion.GetSize(dim);

    // Compute deltas between corners and the grid
    const IndexValueType deltaLo = lower % m_OutputGridSize[dim];
    const IndexValueType deltaUp = upper % m_OutputGridSize[dim];

    // Move corners to aligned positions
    lower -= deltaLo;
    if (deltaUp > 0)
      {
      upper += m_OutputGridSize[dim] - deltaUp;
      }

    // Update region
    region.SetIndex(dim, lower);
    region.SetSize(dim, upper - lower);

    }

  // Compute the number of subregions to process
  const unsigned int nbTilesX = region.GetSize(0) / m_OutputGridSize[0];
  const unsigned int nbTilesY = region.GetSize(1) / m_OutputGridSize[1];

  // Progress
  itk::ProgressReporter progress(this, 0, nbTilesX*nbTilesY);

  // For each tile, propagate the input region and recopy the output
  ImageType * inputImage = static_cast<ImageType * >(  Superclass::ProcessObject::GetInput(0) );
  unsigned int tx, ty;
  RegionType subRegion;
  subRegion.SetSize(m_OutputGridSize);
  for (ty = 0; ty < nbTilesY; ty++)
  {
    subRegion.SetIndex(1, ty*m_OutputGridSize[1] + region.GetIndex(1));
    for (tx = 0; tx < nbTilesX; tx++)
    {
      // Update the input subregion
      subRegion.SetIndex(0, tx*m_OutputGridSize[0] + region.GetIndex(0));
      inputImage->SetRequestedRegion(subRegion);
      inputImage->PropagateRequestedRegion();
      inputImage->UpdateOutputData();

      // Copy the subregion to output
      RegionType cpyRegion(subRegion);
      cpyRegion.Crop(outputReqRegion);
      itk::ImageAlgorithm::Copy( inputImage, outputPtr, cpyRegion, cpyRegion );

      progress.CompletedPixel();
    }
  }
 }


} // end namespace otb


#endif