Skip to content
GitLab
Select Git revision
  • master default
  • otbtf_4.3.1
  • todel_otbtf330
  • 1.4
  • 1.3
  • v1.2
  • v1.0
Find file BlameHistoryPermalink
otbDecloudTimeSeriesPreProcessor.cxx 23.39 KiB
1 
/*=========================================================================
2
3 
     Copyright (c) INRAE 2020-2022. 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 
#include "itkFixedArray.h"
12 
#include "itkObjectFactory.h"
13 
#include "otbWrapperApplicationFactory.h"
14
15 
// Application engine
16 
#include "otbStandardFilterWatcher.h"
17 
#include "itkFixedArray.h"
18
19 
// Stack
20 
#include "otbTensorflowSource.h"
21 
#include "otbTensorflowCommon.h"
22 
#include <vector>
23 
#include <algorithm>
24 
#include <numeric>
25
26 
// Functor
27 
#include "otbFunctorImageFilter.h"
28
29 
// Channels slices
30 
#include "otbMultiChannelExtractROI.h"
31
32 
namespace otb
33 
{
34
35 
namespace Wrapper
36 
{
37
38 
// Name of the environment variable for the number of outputs
39 
const std::string ENV_VAR_NOUTPUTS = "DECLOUD_PREPROCESSING_NOUTPUTS";
40
41 
// Structure to store one timestamp and one index
42 
template <class TimestampType>
43 
struct TimestampWithIndex
44 
{
45 
  TimestampType timestamp;
46 
  unsigned int  index;
47 
};
48
49 
// Sorting modes
50 
enum SortMode
51 
{
52 
  ASC, // Ascending order
53 
  DES, // Descending order
54 
  ABS  // Ascending order computed from the abs(t-tref)
55 
};
56
57 
/*
58 
 * \class PixelFunction
59 
 *
60 
 * \brief Computes the output pixel from
61 
 * - SAR pixels (stacked in channels)
62 
 * - Optical pixels (stacked in channels)
63 
 * - pairs (list of pair of SAR and Optical input images indices)
64 
 * - Nodata (of SAR, and Optical)
65 
 * - Number of channels (in SAR, and Optical)
66 
 *
67 
 * The function that computes the output pixel, outputs the stacked [SAR, Optical] pixels in channels
68 
 *
69 
 * \ingroup OTBDecloud
70 
 */
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
template <class TPixel, class TPairList> class PixelFunction { public: PixelFunction() {} ~PixelFunction() {} bool operator!=(const PixelFunction &) const { return false; } bool operator==(const PixelFunction & other) const { return !(*this != other); } // Returns the output pixel size std::size_t OutputSize(std::array<size_t, 2ul> Inputs) const { return m_NbOutputImages * (m_SARNbBands + m_OptNbBands); } // Set parameters void SetParameters(TPairList pairs, unsigned int sarNbBands, unsigned int optNbBands, float sarNdVal, float optNdVal, unsigned int nbOutputImages) { m_Pairs = pairs; m_SARNbBands = sarNbBands; m_OptNbBands = optNbBands; m_SARNoDataValue = sarNdVal; m_OptNoDataValue = optNdVal; m_NbOutputImages = nbOutputImages; } /* * Get the pixel of n input image */ TPixel GetPixel(const TPixel & inPix, unsigned int idx, unsigned int nbBands) const { TPixel pix; pix.SetSize(nbBands); unsigned int outBand = 0; unsigned int start = idx * nbBands; for (unsigned int i = start; i < start + nbBands; i++) { pix[outBand] = inPix[i]; outBand++; } return pix; } /* * Tell if the pixel is full on no-data values */ bool IsNoData(const TPixel & pix, const float noDataValue) const { for (unsigned int i = 0; i < pix.GetSize(); i++) if (pix[i] != noDataValue) return false;