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
|
/*
* Copyright (C) 1999-2011 Insight Software Consortium
* 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 otbPersistentImageToOGRDataFilter_h
#define otbPersistentImageToOGRDataFilter_h
#include "otbPersistentImageFilter.h"
#include "otbLabelImageToOGRDataSourceFilter.h"
#include "itkMacro.h"
#include <string>
namespace otb
{
/** \class PersistentImageToOGRDataFilter
* \brief Perform vectorization in a persistent way.
*
* This filter is a generic filter. It is the base class of the Large scale segmentation framework.
* The \c Initialize() method is used to create the new layer (LayerName) in the input \c OGRDataSource,
* and the field (FieldName) in the layer. This field will contain the label of the polygon in the input image.
* The \c ProcessTile() method is pure virtual and is implemented in sub class
* (@see \c PersistentStreamingLabelImageToOGRDataFilter). It returns a "memory" DataSource.
* This filter only copy each feature of the layer in the "memory" DataSource into the LayerName layer in the
* input \c OGRDataSource set by \c SetOGRDataSource() method.
*
* \sa PersistentImageFilter
*
*
* \ingroup OTBConversion
*/
template<class TImage>
class ITK_EXPORT PersistentImageToOGRDataFilter :
public PersistentImageFilter<TImage, TImage>
{
public:
/** Standard Self typedef */
typedef PersistentImageToOGRDataFilter Self;
typedef PersistentImageFilter<TImage, TImage> Superclass;
typedef itk::SmartPointer<Self> Pointer;
typedef itk::SmartPointer<const Self> ConstPointer;
/** Runtime information support. */
itkTypeMacro(PersistentImageToOGRDataFilter, PersistentImageFilter);
typedef TImage InputImageType;
typedef typename InputImageType::Pointer InputImagePointer;
typedef typename InputImageType::RegionType RegionType;
typedef typename InputImageType::SizeType SizeType;
typedef typename InputImageType::IndexType IndexType;
typedef typename InputImageType::PixelType PixelType;
typedef typename InputImageType::InternalPixelType InternalPixelType;
typedef ogr::DataSource OGRDataSourceType;
typedef typename OGRDataSourceType::Pointer OGRDataSourcePointerType;
typedef ogr::Layer OGRLayerType;
typedef ogr::Feature OGRFeatureType;
void AllocateOutputs();
virtual void Reset(void);
virtual void Synthetize(void);
/** This method creates the output layer in the OGRDataSource set by the user.
* The name of the layer is set by \c SetLayerName .
* \note This methode must be called before the call of Update .
*/
virtual void Initialize(void);
/** Set/Get the name of the output layer to write in the \c ogr::DataSource. */
itkSetStringMacro(LayerName);
itkGetStringMacro(LayerName);
/** Set the Field Name in which labels will be written. (default is "DN")
* A field FieldName is created in the output layer LayerName. The Field type is Integer.
*/
itkSetMacro(FieldName, std::string);
/**
* Return the Field name in which labels have been written.
*/
itkGetMacro(FieldName, std::string);
/** Get the size of the tile used for streaming.
* This is useful if you use for example the \c OGRFusionTileFilter
* for fusioning streaming tiles.
*/
itkGetMacro(StreamSize, SizeType);
/** Set the geometry type */
itkSetMacro(GeometryType,OGRwkbGeometryType);
/** Get the geometry type */
itkGetMacro(GeometryType,OGRwkbGeometryType);
/** Set the \c ogr::DataSource in which the layer LayerName will be created. */
void SetOGRDataSource( OGRDataSourcePointerType ogrDS );
/** Get the \c ogr::DataSource output. */
OGRDataSourceType * GetOGRDataSource( void );
/** Add one option for OGR layer creation */
void AddOGRLayerCreationOption(const std::string& option);
/** Clear all OGR layer creation options */
void ClearOGRLayerCreationOptions();
/** Set the OGR layer creation options */
void SetOGRLayerCreationOptions(const std::vector<std::string> & options);
/** Get the OGR layer creation options */
const std::vector<std::string> & GetOGRLayerCreationOptions(void);
/** Set the field type for class label */
itkSetMacro(FieldType,OGRFieldType);
/** Get the field type for class label */
itkGetMacro(FieldType,OGRFieldType);
protected:
PersistentImageToOGRDataFilter();
virtual ~PersistentImageToOGRDataFilter();
void PrintSelf(std::ostream& os, itk::Indent indent) const;
virtual void GenerateData();
private:
PersistentImageToOGRDataFilter(const Self &); //purposely not implemented
void operator =(const Self&); //purposely not implemented
virtual OGRDataSourcePointerType ProcessTile() = 0;
std::string m_FieldName;
std::string m_LayerName;
OGRwkbGeometryType m_GeometryType;
SizeType m_StreamSize;
std::vector<std::string> m_OGRLayerCreationOptions;
OGRFieldType m_FieldType;
}; // end of class
} // end namespace otb
#ifndef OTB_MANUAL_INSTANTIATION
#include "otbPersistentImageToOGRDataFilter.txx"
#endif
#endif
|
