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
|
/******************************************************************************
*
* Project: GDAL
* Purpose: "gdal vector buffer"
* Author: Even Rouault <even dot rouault at spatialys.com>
*
******************************************************************************
* Copyright (c) 2025, Even Rouault <even dot rouault at spatialys.com>
*
* SPDX-License-Identifier: MIT
****************************************************************************/
#include "gdalalg_vector_buffer.h"
#include "gdal_priv.h"
#include "ogrsf_frmts.h"
//! @cond Doxygen_Suppress
#ifndef _
#define _(x) (x)
#endif
/************************************************************************/
/* GDALVectorBufferAlgorithm() */
/************************************************************************/
GDALVectorBufferAlgorithm::GDALVectorBufferAlgorithm(bool standaloneStep)
: GDALVectorGeomAbstractAlgorithm(NAME, DESCRIPTION, HELP_URL,
standaloneStep, m_opts)
{
AddArg("distance", 0, _("Distance to which to extend the geometry."),
&m_opts.m_distance)
.SetPositional()
.SetRequired();
AddArg("endcap-style", 0, _("Endcap style."), &m_opts.m_endCapStyle)
.SetChoices("round", "flat", "square")
.SetDefault(m_opts.m_endCapStyle);
AddArg("join-style", 0, _("Join style."), &m_opts.m_joinStyle)
.SetChoices("round", "mitre", "bevel")
.SetDefault(m_opts.m_joinStyle);
AddArg("mitre-limit", 0,
_("Mitre ratio limit (only affects mitered join style)."),
&m_opts.m_mitreLimit)
.SetDefault(m_opts.m_mitreLimit)
.SetMinValueIncluded(0);
AddArg("quadrant-segments", 0,
_("Number of line segments used to approximate a quarter circle."),
&m_opts.m_quadrantSegments)
.SetDefault(m_opts.m_quadrantSegments)
.SetMinValueIncluded(1);
AddArg("side", 0,
_("Sets whether the computed buffer should be single-sided or not."),
&m_opts.m_side)
.SetChoices("both", "left", "right")
.SetDefault(m_opts.m_side);
}
#ifdef HAVE_GEOS
namespace
{
/************************************************************************/
/* GDALVectorBufferAlgorithmLayer */
/************************************************************************/
class GDALVectorBufferAlgorithmLayer final
: public GDALVectorGeomOneToOneAlgorithmLayer<GDALVectorBufferAlgorithm>
{
public:
GDALVectorBufferAlgorithmLayer(
OGRLayer &oSrcLayer, const GDALVectorBufferAlgorithm::Options &opts)
: GDALVectorGeomOneToOneAlgorithmLayer<GDALVectorBufferAlgorithm>(
oSrcLayer, opts)
{
m_aosBufferOptions.SetNameValue("ENDCAP_STYLE",
opts.m_endCapStyle.c_str());
m_aosBufferOptions.SetNameValue("JOIN_STYLE", opts.m_joinStyle.c_str());
m_aosBufferOptions.SetNameValue("MITRE_LIMIT",
CPLSPrintf("%.17g", opts.m_mitreLimit));
m_aosBufferOptions.SetNameValue(
"QUADRANT_SEGMENTS", CPLSPrintf("%d", opts.m_quadrantSegments));
m_aosBufferOptions.SetNameValue("SINGLE_SIDED",
m_opts.m_side != "both" ? "YES" : "NO");
}
protected:
using GDALVectorGeomOneToOneAlgorithmLayer::TranslateFeature;
std::unique_ptr<OGRFeature>
TranslateFeature(std::unique_ptr<OGRFeature> poSrcFeature) const override;
private:
CPLStringList m_aosBufferOptions{};
};
/************************************************************************/
/* TranslateFeature() */
/************************************************************************/
std::unique_ptr<OGRFeature> GDALVectorBufferAlgorithmLayer::TranslateFeature(
std::unique_ptr<OGRFeature> poSrcFeature) const
{
const int nGeomFieldCount = poSrcFeature->GetGeomFieldCount();
for (int i = 0; i < nGeomFieldCount; ++i)
{
if (IsSelectedGeomField(i))
{
if (auto poGeom = std::unique_ptr<OGRGeometry>(
poSrcFeature->StealGeometry(i)))
{
poGeom.reset(poGeom->BufferEx(m_opts.m_distance,
m_aosBufferOptions.List()));
if (poGeom)
{
poGeom->assignSpatialReference(m_srcLayer.GetLayerDefn()
->GetGeomFieldDefn(i)
->GetSpatialRef());
poSrcFeature->SetGeomField(i, std::move(poGeom));
}
}
}
}
return poSrcFeature;
}
} // namespace
#endif // HAVE_GEOS
/************************************************************************/
/* GDALVectorBufferAlgorithm::CreateAlgLayer() */
/************************************************************************/
std::unique_ptr<OGRLayerWithTranslateFeature>
GDALVectorBufferAlgorithm::CreateAlgLayer([[maybe_unused]] OGRLayer &srcLayer)
{
#ifdef HAVE_GEOS
return std::make_unique<GDALVectorBufferAlgorithmLayer>(srcLayer, m_opts);
#else
CPLAssert(false);
return nullptr;
#endif
}
/************************************************************************/
/* GDALVectorBufferAlgorithm::RunStep() */
/************************************************************************/
bool GDALVectorBufferAlgorithm::RunStep(GDALPipelineStepRunContext &ctxt)
{
#ifdef HAVE_GEOS
if (m_opts.m_side == "right")
m_opts.m_distance = -m_opts.m_distance;
return GDALVectorGeomAbstractAlgorithm::RunStep(ctxt);
#else
(void)ctxt;
ReportError(CE_Failure, CPLE_NotSupported,
"This algorithm is only supported for builds against GEOS");
return false;
#endif
}
GDALVectorBufferAlgorithmStandalone::~GDALVectorBufferAlgorithmStandalone() =
default;
//! @endcond
|
