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
38
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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
|
/*
* Copyright (C) 2005-2017 Centre National d'Etudes Spatiales (CNES)
*
* This file is part of Orfeo Toolbox
*
* https://www.orfeo-toolbox.org/
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef otbOGRDataSourceToLabelImageFilter_h
#define otbOGRDataSourceToLabelImageFilter_h
#include "itkImageToImageFilter.h"
#include "otbMacro.h"
#include "gdal.h"
#include "gdal_alg.h"
#include "otbOGRDataSourceWrapper.h"
namespace otb {
/** \class OGRDataSourceToLabelImageFilter
* \brief Burn geometries from the specified VectorData into raster
*
* This class handles burning several input \c OGRDataSource into the
* output raster. It has two different modes:
* - either the burn values are extracted from a field set by
* the user (default field is DN). This is the default behaviour,
* - or the burn values are made of foreground (when inside
* geometries) and background (outside geometries) values, which can
* be set by the user. One can set this behaviour by disable the
* BurnAttributeMode using the BurnAttributeModeOff() method.
*
* Please note that the background value is also used in
* BurnAttributeModeOn() to fill areas where there are no geometries.
*
* Setting the output raster information can be done in two ways by:
* - Setting the Origin/Size/Spacing of the output image
* - Using an existing image as support via SetOutputParametersFromImage(ImageBase)
*
*
* \ingroup OTBConversion
*/
template < class TOutputImage >
class ITK_EXPORT OGRDataSourceToLabelImageFilter :
public itk::ImageSource<TOutputImage>
{
public:
/** Standard class typedefs */
typedef OGRDataSourceToLabelImageFilter Self;
typedef itk::ImageSource<TOutputImage> Superclass;
typedef itk::SmartPointer< Self > Pointer;
typedef itk::SmartPointer<const Self> ConstPointer;
/** Run-time type information (and related methods). */
itkTypeMacro(OGRDataSourceToLabelImageFilter, itk::ImageSource);
/** Method for creation through the object factory. */
itkNewMacro(Self);
typedef TOutputImage OutputImageType;
typedef typename OutputImageType::Pointer OutputImagePointer;
typedef typename OutputImageType::SizeType OutputSizeType;
typedef typename OutputImageType::IndexType OutputIndexType;
typedef typename OutputImageType::SpacingType OutputSpacingType;
typedef typename OutputImageType::PointType OutputOriginType;
typedef typename OutputImageType::RegionType OutputImageRegionType;
typedef typename OutputImageType::PixelType OutputImagePixelType;
typedef typename OutputImageType::InternalPixelType OutputImageInternalPixelType;
/** VectorData typedefs*/
typedef ogr::DataSource OGRDataSourceType;
typedef typename OGRDataSourceType::Pointer OGRDataSourcePointerType;
typedef ogr::Layer OGRLayerType;
typedef itk::ImageBase<OutputImageType::ImageDimension> ImageBaseType;
/** Get Nth input \c OGRDataSource */
const OGRDataSourceType* GetInput(unsigned int idx);
/** Method for adding a \c OGRDataSource to rasterize */
virtual void AddOGRDataSource(const OGRDataSourceType* ds);
/** Set the size of the output image. */
itkSetMacro(OutputSize, OutputSizeType);
/** Get the size of the output image. */
itkGetConstReferenceMacro(OutputSize, OutputSizeType);
/** Set the origin of the output image.
* \sa GetOrigin()
*/
itkSetMacro(OutputOrigin, OutputOriginType);
virtual void SetOutputOrigin(const double origin[2]);
virtual void SetOutputOrigin(const float origin[2]);
itkGetConstReferenceMacro(OutputOrigin, OutputOriginType);
/** Set the spacing (size of a pixel) of the output image.
* \sa GetSignedSpacing()
*/
virtual void SetOutputSpacing(const OutputSpacingType& spacing);
virtual void SetOutputSpacing(const double spacing[2]);
virtual void SetOutputSpacing(const float spacing[2]);
/** Set/Get Output Projection Ref */
itkSetStringMacro(OutputProjectionRef);
itkGetStringMacro(OutputProjectionRef);
/** Set the attribute field on the \c OGRFeature to be used as burn value in the output image*/
itkSetStringMacro(BurnAttribute);
itkGetStringMacro(BurnAttribute);
/** Set/Get the background value */
itkSetMacro(BackgroundValue,OutputImageInternalPixelType);
itkGetConstReferenceMacro(BackgroundValue,OutputImageInternalPixelType);
/** Set/Get the foreground value */
itkSetMacro(ForegroundValue,OutputImageInternalPixelType);
itkGetConstReferenceMacro(ForegroundValue,OutputImageInternalPixelType);
/** Set/Get the BurnAttributeMode flag */
itkSetMacro(BurnAttributeMode,bool);
itkGetConstReferenceMacro(BurnAttributeMode,bool);
itkBooleanMacro(BurnAttributeMode);
/** Useful to set the output parameters from an existing image*/
void SetOutputParametersFromImage(const ImageBaseType * image);
protected:
void GenerateData() override;
OGRDataSourceToLabelImageFilter();
~OGRDataSourceToLabelImageFilter() override {}
void GenerateOutputInformation() override;
void PrintSelf(std::ostream& os, itk::Indent indent) const override;
private:
OGRDataSourceToLabelImageFilter(const Self&); //purposely not implemented
void operator=(const Self&); //purposely not implemented
std::vector< OGRLayerH > m_SrcDataSetLayers;
std::vector<int> m_BandsToBurn;
// Field used to extract the burn value
std::string m_BurnAttribute;
// Output params
std::string m_OutputProjectionRef;
OutputSpacingType m_OutputSpacing;
OutputOriginType m_OutputOrigin;
OutputSizeType m_OutputSize;
OutputIndexType m_OutputStartIndex;
OutputImageInternalPixelType m_BackgroundValue;
OutputImageInternalPixelType m_ForegroundValue;
bool m_BurnAttributeMode;
}; // end of class VectorDataToLabelImageFilter
} // end of namespace otb
#ifndef OTB_MANUAL_INSTANTIATION
#include "otbOGRDataSourceToLabelImageFilter.txx"
#endif
#endif
|
