/src/gdal/ogr/ogrsf_frmts/geojson/ogrgeojsonwritelayer.cpp

Source (jump to first uncovered line)
/******************************************************************************
 *
 * Project:  OpenGIS Simple Features Reference Implementation
 * Purpose:  Implementation of OGRGeoJSONWriteLayer class (OGR GeoJSON Driver).
 * Author:   Mateusz Loskot, mateusz@loskot.net
 *
 ******************************************************************************
 * Copyright (c) 2011, Even Rouault <even dot rouault at spatialys.com>
 * Copyright (c) 2007, Mateusz Loskot
 *
 * SPDX-License-Identifier: MIT
 ****************************************************************************/

#include "ogr_geojson.h"
#include "ogrgeojsonwriter.h"

#include "cpl_vsi_virtual.h"

#include <algorithm>

/************************************************************************/
/*                         OGRGeoJSONWriteLayer()                       */
/************************************************************************/

OGRGeoJSONWriteLayer::OGRGeoJSONWriteLayer(const char *pszName,
                                           OGRwkbGeometryType eGType,
                                           CSLConstList papszOptions,
                                           bool bWriteFC_BBOXIn,
                                           OGRCoordinateTransformation *poCT,
                                           OGRGeoJSONDataSource *poDS)
    : poDS_(poDS), poFeatureDefn_(new OGRFeatureDefn(pszName)), nOutCounter_(0),
      bWriteBBOX(CPLTestBool(
          CSLFetchNameValueDef(papszOptions, "WRITE_BBOX", "FALSE"))),
      bBBOX3D(false), bWriteFC_BBOX(bWriteFC_BBOXIn),
      nSignificantFigures_(atoi(
          CSLFetchNameValueDef(papszOptions, "SIGNIFICANT_FIGURES", "-1"))),
      bRFC7946_(
          CPLTestBool(CSLFetchNameValueDef(papszOptions, "RFC7946", "FALSE"))),
      bWrapDateLine_(CPLTestBool(
          CSLFetchNameValueDef(papszOptions, "WRAPDATELINE", "YES"))),
      osForeignMembers_(
          CSLFetchNameValueDef(papszOptions, "FOREIGN_MEMBERS_FEATURE", "")),
      poCT_(poCT)
{
    if (!osForeignMembers_.empty())
    {
        // Already checked in OGRGeoJSONDataSource::ICreateLayer()
        CPLAssert(osForeignMembers_.front() == '{');
        CPLAssert(osForeignMembers_.back() == '}');
        osForeignMembers_ =
            osForeignMembers_.substr(1, osForeignMembers_.size() - 2);
    }
    poFeatureDefn_->Reference();
    poFeatureDefn_->SetGeomType(eGType);
    SetDescription(poFeatureDefn_->GetName());
    const char *pszCoordPrecision =
        CSLFetchNameValue(papszOptions, "COORDINATE_PRECISION");
    if (pszCoordPrecision)
    {
        oWriteOptions_.nXYCoordPrecision = atoi(pszCoordPrecision);
        oWriteOptions_.nZCoordPrecision = atoi(pszCoordPrecision);
    }
    else
    {
        oWriteOptions_.nXYCoordPrecision = atoi(CSLFetchNameValueDef(
            papszOptions, "XY_COORD_PRECISION", bRFC7946_ ? "7" : "-1"));
        oWriteOptions_.nZCoordPrecision = atoi(CSLFetchNameValueDef(
            papszOptions, "Z_COORD_PRECISION", bRFC7946_ ? "3" : "-1"));
    }
    oWriteOptions_.bWriteBBOX = bWriteBBOX;
    oWriteOptions_.nSignificantFigures = nSignificantFigures_;
    if (bRFC7946_)
    {
        oWriteOptions_.SetRFC7946Settings();
    }
    oWriteOptions_.SetIDOptions(papszOptions);
    oWriteOptions_.bAllowNonFiniteValues = CPLTestBool(
        CSLFetchNameValueDef(papszOptions, "WRITE_NON_FINITE_VALUES", "FALSE"));
    oWriteOptions_.bAutodetectJsonStrings = CPLTestBool(
        CSLFetchNameValueDef(papszOptions, "AUTODETECT_JSON_STRINGS", "TRUE"));
}

/************************************************************************/
/*                        ~OGRGeoJSONWriteLayer()                       */
/************************************************************************/

OGRGeoJSONWriteLayer::~OGRGeoJSONWriteLayer()
{
    FinishWriting();

    if (nullptr != poFeatureDefn_)
    {
        poFeatureDefn_->Release();
    }

    delete poCT_;
}

/************************************************************************/
/*                           FinishWriting()                            */
/************************************************************************/

void OGRGeoJSONWriteLayer::FinishWriting()
{
    if (m_nPositionBeforeFCClosed == 0)
    {
        VSILFILE *fp = poDS_->GetOutputFile();

        m_nPositionBeforeFCClosed = fp->Tell();

        VSIFPrintfL(fp, "\n]");

        if (bWriteFC_BBOX && sEnvelopeLayer.IsInit())
        {
            CPLString osBBOX = "[ ";
            char szFormat[32];
            if (oWriteOptions_.nXYCoordPrecision >= 0)
                snprintf(szFormat, sizeof(szFormat), "%%.%df",
                         oWriteOptions_.nXYCoordPrecision);
            else
                snprintf(szFormat, sizeof(szFormat), "%s", "%.15g");

            osBBOX += CPLSPrintf(szFormat, sEnvelopeLayer.MinX);
            osBBOX += ", ";
            osBBOX += CPLSPrintf(szFormat, sEnvelopeLayer.MinY);
            osBBOX += ", ";
            if (bBBOX3D)
            {
                osBBOX += CPLSPrintf(szFormat, sEnvelopeLayer.MinZ);
                osBBOX += ", ";
            }
            osBBOX += CPLSPrintf(szFormat, sEnvelopeLayer.MaxX);
            osBBOX += ", ";
            osBBOX += CPLSPrintf(szFormat, sEnvelopeLayer.MaxY);
            if (bBBOX3D)
            {
                osBBOX += ", ";
                osBBOX += CPLSPrintf(szFormat, sEnvelopeLayer.MaxZ);
            }
            osBBOX += " ]";

            if (poDS_->GetFpOutputIsSeekable() &&
                osBBOX.size() + 9 < OGRGeoJSONDataSource::SPACE_FOR_BBOX)
            {
                VSIFSeekL(fp, poDS_->GetBBOXInsertLocation(), SEEK_SET);
                VSIFPrintfL(fp, "\"bbox\": %s,", osBBOX.c_str());
                VSIFSeekL(fp, 0, SEEK_END);
            }
            else
            {
                VSIFPrintfL(fp, ",\n\"bbox\": %s", osBBOX.c_str());
            }
        }

        VSIFPrintfL(fp, "\n}\n");
        fp->Flush();
    }
}

/************************************************************************/
/*                           SyncToDisk()                               */
/************************************************************************/

OGRErr OGRGeoJSONWriteLayer::SyncToDisk()
{
    if (m_nPositionBeforeFCClosed == 0 && poDS_->GetFpOutputIsSeekable())
    {
        FinishWriting();
    }

    return OGRERR_NONE;
}

/************************************************************************/
/*                           ICreateFeature()                            */
/************************************************************************/

OGRErr OGRGeoJSONWriteLayer::ICreateFeature(OGRFeature *poFeature)
{
    VSILFILE *fp = poDS_->GetOutputFile();

    OGRFeature *poFeatureToWrite;
    if (poCT_ != nullptr || bRFC7946_)
    {
        poFeatureToWrite = new OGRFeature(poFeatureDefn_);
        poFeatureToWrite->SetFrom(poFeature);
        poFeatureToWrite->SetFID(poFeature->GetFID());
        OGRGeometry *poGeometry = poFeatureToWrite->GetGeometryRef();
        if (poGeometry)
        {
            const char *const apszOptions[] = {
                bWrapDateLine_ ? "WRAPDATELINE=YES" : nullptr, nullptr};
            OGRGeometry *poNewGeom = OGRGeometryFactory::transformWithOptions(
                poGeometry, poCT_, const_cast<char **>(apszOptions),
                oTransformCache_);
            if (poNewGeom == nullptr)
            {
                delete poFeatureToWrite;
                return OGRERR_FAILURE;
            }

            OGREnvelope sEnvelope;
            poNewGeom->getEnvelope(&sEnvelope);
            if (sEnvelope.MinX < -180.0 || sEnvelope.MaxX > 180.0 ||
                sEnvelope.MinY < -90.0 || sEnvelope.MaxY > 90.0)
            {
                CPLError(CE_Failure, CPLE_AppDefined,
                         "Geometry extent outside of "
                         "[-180.0,180.0]x[-90.0,90.0] bounds");
                delete poFeatureToWrite;
                return OGRERR_FAILURE;
            }

            poFeatureToWrite->SetGeometryDirectly(poNewGeom);
        }
    }
    else
    {
        poFeatureToWrite = poFeature;
    }

    const auto IsValid = [](const OGRGeometry *poGeom)
    {
        CPLErrorHandlerPusher oErrorHandler(CPLQuietErrorHandler);
        return poGeom->IsValid();
    };

    // Special processing to detect and repair invalid geometries due to
    // coordinate precision.
    // Normally drivers shouldn't do that as similar code is triggered by
    // setting the OGR_APPLY_GEOM_SET_PRECISION=YES configuration option by
    // the generic OGRLayer::CreateFeature() code path. But this code predates
    // its introduction and RFC99, and can be useful in RFC7946 mode due to
    // coordinate reprojection.
    OGRGeometry *poOrigGeom = poFeature->GetGeometryRef();
    if (OGRGeometryFactory::haveGEOS() &&
        oWriteOptions_.nXYCoordPrecision >= 0 && poOrigGeom &&
        wkbFlatten(poOrigGeom->getGeometryType()) != wkbPoint &&
        IsValid(poOrigGeom))
    {
        const double dfXYResolution =
            std::pow(10.0, double(-oWriteOptions_.nXYCoordPrecision));
        auto poNewGeom = std::unique_ptr<OGRGeometry>(
            poFeatureToWrite->GetGeometryRef()->clone());
        OGRGeomCoordinatePrecision sPrecision;
        sPrecision.dfXYResolution = dfXYResolution;
        poNewGeom->roundCoordinates(sPrecision);
        if (!IsValid(poNewGeom.get()))
        {
            std::unique_ptr<OGRGeometry> poValidGeom;
            if (poFeature == poFeatureToWrite)
            {
                CPLDebug("GeoJSON",
                         "Running SetPrecision() to correct an invalid "
                         "geometry due to reduced precision output");
                poValidGeom.reset(
                    poOrigGeom->SetPrecision(dfXYResolution, /* nFlags = */ 0));
            }
            else
            {
                CPLDebug("GeoJSON", "Running MakeValid() to correct an invalid "
                                    "geometry due to reduced precision output");
                poValidGeom.reset(poNewGeom->MakeValid());
                if (poValidGeom)
                {
                    auto poValidGeomRoundCoordinates =
                        std::unique_ptr<OGRGeometry>(poValidGeom->clone());
                    poValidGeomRoundCoordinates->roundCoordinates(sPrecision);
                    if (!IsValid(poValidGeomRoundCoordinates.get()))
                    {
                        CPLDebug("GeoJSON",
                                 "Running SetPrecision() to correct an invalid "
                                 "geometry due to reduced precision output");
                        auto poValidGeom2 = std::unique_ptr<OGRGeometry>(
                            poValidGeom->SetPrecision(dfXYResolution,
                                                      /* nFlags = */ 0));
                        if (poValidGeom2)
                            poValidGeom = std::move(poValidGeom2);
                    }
                }
            }
            if (poValidGeom)
            {
                if (poFeature == poFeatureToWrite)
                {
                    poFeatureToWrite = new OGRFeature(poFeatureDefn_);
                    poFeatureToWrite->SetFrom(poFeature);
                    poFeatureToWrite->SetFID(poFeature->GetFID());
                }
                poFeatureToWrite->SetGeometryDirectly(poValidGeom.release());
            }
        }
    }

    if (oWriteOptions_.bGenerateID && poFeatureToWrite->GetFID() == OGRNullFID)
    {
        poFeatureToWrite->SetFID(nOutCounter_);
    }
    json_object *poObj =
        OGRGeoJSONWriteFeature(poFeatureToWrite, oWriteOptions_);
    CPLAssert(nullptr != poObj);

    if (m_nPositionBeforeFCClosed)
    {
        // If we had called SyncToDisk() previously, undo its effects
        fp->Seek(m_nPositionBeforeFCClosed, SEEK_SET);
        m_nPositionBeforeFCClosed = 0;
    }

    if (nOutCounter_ > 0)
    {
        /* Separate "Feature" entries in "FeatureCollection" object. */
        VSIFPrintfL(fp, ",\n");
    }
    const char *pszJson = json_object_to_json_string_ext(
        poObj, JSON_C_TO_STRING_SPACED
#ifdef JSON_C_TO_STRING_NOSLASHESCAPE
                   | JSON_C_TO_STRING_NOSLASHESCAPE
#endif
    );

    OGRErr eErr = OGRERR_NONE;
    size_t nLen = strlen(pszJson);
    if (!osForeignMembers_.empty())
    {
        if (nLen > 2 && pszJson[nLen - 2] == ' ' && pszJson[nLen - 1] == '}')
        {
            nLen -= 2;
        }
        else
        {
            // should not happen
            CPLError(CE_Failure, CPLE_AppDefined,
                     "Unexpected JSON output for feature. Cannot write foreign "
                     "member");
            osForeignMembers_.clear();
        }
    }
    if (VSIFWriteL(pszJson, nLen, 1, fp) != 1)
    {
        CPLError(CE_Failure, CPLE_FileIO, "Cannot write feature");
        eErr = OGRERR_FAILURE;
    }
    else if (!osForeignMembers_.empty() &&
             (VSIFWriteL(", ", 2, 1, fp) != 1 ||
              VSIFWriteL(osForeignMembers_.c_str(), osForeignMembers_.size(), 1,
                         fp) != 1 ||
              VSIFWriteL("}", 1, 1, fp) != 1))
    {
        CPLError(CE_Failure, CPLE_FileIO, "Cannot write feature");
        eErr = OGRERR_FAILURE;
    }

    json_object_put(poObj);

    ++nOutCounter_;

    OGRGeometry *poGeometry = poFeatureToWrite->GetGeometryRef();
    if (poGeometry != nullptr && !poGeometry->IsEmpty())
    {
        OGREnvelope3D sEnvelope = OGRGeoJSONGetBBox(poGeometry, oWriteOptions_);
        if (poGeometry->getCoordinateDimension() == 3)
            bBBOX3D = true;

        if (!sEnvelopeLayer.IsInit())
        {
            sEnvelopeLayer = sEnvelope;
        }
        else if (oWriteOptions_.bBBOXRFC7946)
        {
            const bool bEnvelopeCrossAM = (sEnvelope.MinX > sEnvelope.MaxX);
            const bool bEnvelopeLayerCrossAM =
                (sEnvelopeLayer.MinX > sEnvelopeLayer.MaxX);
            if (bEnvelopeCrossAM)
            {
                if (bEnvelopeLayerCrossAM)
                {
                    sEnvelopeLayer.MinX =
                        std::min(sEnvelopeLayer.MinX, sEnvelope.MinX);
                    sEnvelopeLayer.MaxX =
                        std::max(sEnvelopeLayer.MaxX, sEnvelope.MaxX);
                }
                else
                {
                    if (sEnvelopeLayer.MinX > 0)
                    {
                        sEnvelopeLayer.MinX =
                            std::min(sEnvelopeLayer.MinX, sEnvelope.MinX);
                        sEnvelopeLayer.MaxX = sEnvelope.MaxX;
                    }
                    else if (sEnvelopeLayer.MaxX < 0)
                    {
                        sEnvelopeLayer.MaxX =
                            std::max(sEnvelopeLayer.MaxX, sEnvelope.MaxX);
                        sEnvelopeLayer.MinX = sEnvelope.MinX;
                    }
                    else
                    {
                        sEnvelopeLayer.MinX = -180.0;
                        sEnvelopeLayer.MaxX = 180.0;
                    }
                }
            }
            else if (bEnvelopeLayerCrossAM)
            {
                if (sEnvelope.MinX > 0)
                {
                    sEnvelopeLayer.MinX =
                        std::min(sEnvelopeLayer.MinX, sEnvelope.MinX);
                }
                else if (sEnvelope.MaxX < 0)
                {
                    sEnvelopeLayer.MaxX =
                        std::max(sEnvelopeLayer.MaxX, sEnvelope.MaxX);
                }
                else
                {
                    sEnvelopeLayer.MinX = -180.0;
                    sEnvelopeLayer.MaxX = 180.0;
                }
            }
            else
            {
                sEnvelopeLayer.MinX =
                    std::min(sEnvelopeLayer.MinX, sEnvelope.MinX);
                sEnvelopeLayer.MaxX =
                    std::max(sEnvelopeLayer.MaxX, sEnvelope.MaxX);
            }

            sEnvelopeLayer.MinY = std::min(sEnvelopeLayer.MinY, sEnvelope.MinY);
            sEnvelopeLayer.MaxY = std::max(sEnvelopeLayer.MaxY, sEnvelope.MaxY);
        }
        else
        {
            sEnvelopeLayer.Merge(sEnvelope);
        }
    }

    if (poFeatureToWrite != poFeature)
        delete poFeatureToWrite;

    return eErr;
}

/************************************************************************/
/*                           CreateField()                              */
/************************************************************************/

OGRErr OGRGeoJSONWriteLayer::CreateField(const OGRFieldDefn *poField,
                                         int /* bApproxOK */)
{
    if (poFeatureDefn_->GetFieldIndexCaseSensitive(poField->GetNameRef()) >= 0)
    {
        CPLDebug("GeoJSON", "Field '%s' already present in schema",
                 poField->GetNameRef());

        // TODO - mloskot: Is this return code correct?
        return OGRERR_NONE;
    }

    poFeatureDefn_->AddFieldDefn(poField);

    return OGRERR_NONE;
}

/************************************************************************/
/*                           TestCapability()                           */
/************************************************************************/

int OGRGeoJSONWriteLayer::TestCapability(const char *pszCap)
{
    if (EQUAL(pszCap, OLCCreateField))
        return TRUE;
    else if (EQUAL(pszCap, OLCSequentialWrite))
        return TRUE;
    else if (EQUAL(pszCap, OLCStringsAsUTF8))
        return TRUE;
    return FALSE;
}

/************************************************************************/
/*                           IGetExtent()                               */
/************************************************************************/

OGRErr OGRGeoJSONWriteLayer::IGetExtent(int /*iGeomField*/,
                                        OGREnvelope *psExtent, bool)
{
    if (sEnvelopeLayer.IsInit())
    {
        *psExtent = sEnvelopeLayer;
        return OGRERR_NONE;
    }
    return OGRERR_FAILURE;
}

/************************************************************************/
/*                             GetDataset()                             */
/************************************************************************/

GDALDataset *OGRGeoJSONWriteLayer::GetDataset()
{
    return poDS_;
}

Coverage Report

Created: 2025-06-13 06:29

Line	Count	Source (jump to first uncovered line)
1		/******************************************************************************
2		*
3		* Project: OpenGIS Simple Features Reference Implementation
4		* Purpose: Implementation of OGRGeoJSONWriteLayer class (OGR GeoJSON Driver).
5		* Author: Mateusz Loskot, mateusz@loskot.net
6		*
7		******************************************************************************
8		* Copyright (c) 2011, Even Rouault <even dot rouault at spatialys.com>
9		* Copyright (c) 2007, Mateusz Loskot
10		*
11		* SPDX-License-Identifier: MIT
12		****************************************************************************/
13
14		#include "ogr_geojson.h"
15		#include "ogrgeojsonwriter.h"
16
17		#include "cpl_vsi_virtual.h"
18
19		#include <algorithm>
20
21		/************************************************************************/
22		/* OGRGeoJSONWriteLayer() */
23		/************************************************************************/
24
25		OGRGeoJSONWriteLayer::OGRGeoJSONWriteLayer(const char *pszName,
26		OGRwkbGeometryType eGType,
27		CSLConstList papszOptions,
28		bool bWriteFC_BBOXIn,
29		OGRCoordinateTransformation *poCT,
30		OGRGeoJSONDataSource *poDS)
31	0	: poDS_(poDS), poFeatureDefn_(new OGRFeatureDefn(pszName)), nOutCounter_(0),
32	0	bWriteBBOX(CPLTestBool(
33	0	CSLFetchNameValueDef(papszOptions, "WRITE_BBOX", "FALSE"))),
34	0	bBBOX3D(false), bWriteFC_BBOX(bWriteFC_BBOXIn),
35	0	nSignificantFigures_(atoi(
36	0	CSLFetchNameValueDef(papszOptions, "SIGNIFICANT_FIGURES", "-1"))),
37		bRFC7946_(
38	0	CPLTestBool(CSLFetchNameValueDef(papszOptions, "RFC7946", "FALSE"))),
39	0	bWrapDateLine_(CPLTestBool(
40	0	CSLFetchNameValueDef(papszOptions, "WRAPDATELINE", "YES"))),
41	0	osForeignMembers_(
42	0	CSLFetchNameValueDef(papszOptions, "FOREIGN_MEMBERS_FEATURE", "")),
43	0	poCT_(poCT)
44	0	{
45	0	if (!osForeignMembers_.empty())
46	0	{
47		// Already checked in OGRGeoJSONDataSource::ICreateLayer()
48	0	CPLAssert(osForeignMembers_.front() == '{');
49	0	CPLAssert(osForeignMembers_.back() == '}');
50	0	osForeignMembers_ =
51	0	osForeignMembers_.substr(1, osForeignMembers_.size() - 2);
52	0	}
53	0	poFeatureDefn_->Reference();
54	0	poFeatureDefn_->SetGeomType(eGType);
55	0	SetDescription(poFeatureDefn_->GetName());
56	0	const char *pszCoordPrecision =
57	0	CSLFetchNameValue(papszOptions, "COORDINATE_PRECISION");
58	0	if (pszCoordPrecision)
59	0	{
60	0	oWriteOptions_.nXYCoordPrecision = atoi(pszCoordPrecision);
61	0	oWriteOptions_.nZCoordPrecision = atoi(pszCoordPrecision);
62	0	}
63	0	else
64	0	{
65	0	oWriteOptions_.nXYCoordPrecision = atoi(CSLFetchNameValueDef(
66	0	papszOptions, "XY_COORD_PRECISION", bRFC7946_ ? "7" : "-1"));
67	0	oWriteOptions_.nZCoordPrecision = atoi(CSLFetchNameValueDef(
68	0	papszOptions, "Z_COORD_PRECISION", bRFC7946_ ? "3" : "-1"));
69	0	}
70	0	oWriteOptions_.bWriteBBOX = bWriteBBOX;
71	0	oWriteOptions_.nSignificantFigures = nSignificantFigures_;
72	0	if (bRFC7946_)
73	0	{
74	0	oWriteOptions_.SetRFC7946Settings();
75	0	}
76	0	oWriteOptions_.SetIDOptions(papszOptions);
77	0	oWriteOptions_.bAllowNonFiniteValues = CPLTestBool(
78	0	CSLFetchNameValueDef(papszOptions, "WRITE_NON_FINITE_VALUES", "FALSE"));
79	0	oWriteOptions_.bAutodetectJsonStrings = CPLTestBool(
80	0	CSLFetchNameValueDef(papszOptions, "AUTODETECT_JSON_STRINGS", "TRUE"));
81	0	}
82
83		/************************************************************************/
84		/* ~OGRGeoJSONWriteLayer() */
85		/************************************************************************/
86
87		OGRGeoJSONWriteLayer::~OGRGeoJSONWriteLayer()
88	0	{
89	0	FinishWriting();
90
91	0	if (nullptr != poFeatureDefn_)
92	0	{
93	0	poFeatureDefn_->Release();
94	0	}
95
96	0	delete poCT_;
97	0	}
98
99		/************************************************************************/
100		/* FinishWriting() */
101		/************************************************************************/
102
103		void OGRGeoJSONWriteLayer::FinishWriting()
104	0	{
105	0	if (m_nPositionBeforeFCClosed == 0)
106	0	{
107	0	VSILFILE *fp = poDS_->GetOutputFile();
108
109	0	m_nPositionBeforeFCClosed = fp->Tell();
110
111	0	VSIFPrintfL(fp, "\n]");
112
113	0	if (bWriteFC_BBOX && sEnvelopeLayer.IsInit())
114	0	{
115	0	CPLString osBBOX = "[ ";
116	0	char szFormat[32];
117	0	if (oWriteOptions_.nXYCoordPrecision >= 0)
118	0	snprintf(szFormat, sizeof(szFormat), "%%.%df",
119	0	oWriteOptions_.nXYCoordPrecision);
120	0	else
121	0	snprintf(szFormat, sizeof(szFormat), "%s", "%.15g");
122
123	0	osBBOX += CPLSPrintf(szFormat, sEnvelopeLayer.MinX);
124	0	osBBOX += ", ";
125	0	osBBOX += CPLSPrintf(szFormat, sEnvelopeLayer.MinY);
126	0	osBBOX += ", ";
127	0	if (bBBOX3D)
128	0	{
129	0	osBBOX += CPLSPrintf(szFormat, sEnvelopeLayer.MinZ);
130	0	osBBOX += ", ";
131	0	}
132	0	osBBOX += CPLSPrintf(szFormat, sEnvelopeLayer.MaxX);
133	0	osBBOX += ", ";
134	0	osBBOX += CPLSPrintf(szFormat, sEnvelopeLayer.MaxY);
135	0	if (bBBOX3D)
136	0	{
137	0	osBBOX += ", ";
138	0	osBBOX += CPLSPrintf(szFormat, sEnvelopeLayer.MaxZ);
139	0	}
140	0	osBBOX += " ]";
141
142	0	if (poDS_->GetFpOutputIsSeekable() &&
143	0	osBBOX.size() + 9 < OGRGeoJSONDataSource::SPACE_FOR_BBOX)
144	0	{
145	0	VSIFSeekL(fp, poDS_->GetBBOXInsertLocation(), SEEK_SET);
146	0	VSIFPrintfL(fp, "\"bbox\": %s,", osBBOX.c_str());
147	0	VSIFSeekL(fp, 0, SEEK_END);
148	0	}
149	0	else
150	0	{
151	0	VSIFPrintfL(fp, ",\n\"bbox\": %s", osBBOX.c_str());
152	0	}
153	0	}
154
155	0	VSIFPrintfL(fp, "\n}\n");
156	0	fp->Flush();
157	0	}
158	0	}
159
160		/************************************************************************/
161		/* SyncToDisk() */
162		/************************************************************************/
163
164		OGRErr OGRGeoJSONWriteLayer::SyncToDisk()
165	0	{
166	0	if (m_nPositionBeforeFCClosed == 0 && poDS_->GetFpOutputIsSeekable())
167	0	{
168	0	FinishWriting();
169	0	}
170
171	0	return OGRERR_NONE;
172	0	}
173
174		/************************************************************************/
175		/* ICreateFeature() */
176		/************************************************************************/
177
178		OGRErr OGRGeoJSONWriteLayer::ICreateFeature(OGRFeature *poFeature)
179	0	{
180	0	VSILFILE *fp = poDS_->GetOutputFile();
181
182	0	OGRFeature *poFeatureToWrite;
183	0	if (poCT_ != nullptr \|\| bRFC7946_)
184	0	{
185	0	poFeatureToWrite = new OGRFeature(poFeatureDefn_);
186	0	poFeatureToWrite->SetFrom(poFeature);
187	0	poFeatureToWrite->SetFID(poFeature->GetFID());
188	0	OGRGeometry *poGeometry = poFeatureToWrite->GetGeometryRef();
189	0	if (poGeometry)
190	0	{
191	0	const char *const apszOptions[] = {
192	0	bWrapDateLine_ ? "WRAPDATELINE=YES" : nullptr, nullptr};
193	0	OGRGeometry *poNewGeom = OGRGeometryFactory::transformWithOptions(
194	0	poGeometry, poCT_, const_cast<char **>(apszOptions),
195	0	oTransformCache_);
196	0	if (poNewGeom == nullptr)
197	0	{
198	0	delete poFeatureToWrite;
199	0	return OGRERR_FAILURE;
200	0	}
201
202	0	OGREnvelope sEnvelope;
203	0	poNewGeom->getEnvelope(&sEnvelope);
204	0	if (sEnvelope.MinX < -180.0 \|\| sEnvelope.MaxX > 180.0 \|\|
205	0	sEnvelope.MinY < -90.0 \|\| sEnvelope.MaxY > 90.0)
206	0	{
207	0	CPLError(CE_Failure, CPLE_AppDefined,
208	0	"Geometry extent outside of "
209	0	"[-180.0,180.0]x[-90.0,90.0] bounds");
210	0	delete poFeatureToWrite;
211	0	return OGRERR_FAILURE;
212	0	}
213
214	0	poFeatureToWrite->SetGeometryDirectly(poNewGeom);
215	0	}
216	0	}
217	0	else
218	0	{
219	0	poFeatureToWrite = poFeature;
220	0	}
221
222	0	const auto IsValid = [](const OGRGeometry *poGeom)
223	0	{
224	0	CPLErrorHandlerPusher oErrorHandler(CPLQuietErrorHandler);
225	0	return poGeom->IsValid();
226	0	};
227
228		// Special processing to detect and repair invalid geometries due to
229		// coordinate precision.
230		// Normally drivers shouldn't do that as similar code is triggered by
231		// setting the OGR_APPLY_GEOM_SET_PRECISION=YES configuration option by
232		// the generic OGRLayer::CreateFeature() code path. But this code predates
233		// its introduction and RFC99, and can be useful in RFC7946 mode due to
234		// coordinate reprojection.
235	0	OGRGeometry *poOrigGeom = poFeature->GetGeometryRef();
236	0	if (OGRGeometryFactory::haveGEOS() &&
237	0	oWriteOptions_.nXYCoordPrecision >= 0 && poOrigGeom &&
238	0	wkbFlatten(poOrigGeom->getGeometryType()) != wkbPoint &&
239	0	IsValid(poOrigGeom))
240	0	{
241	0	const double dfXYResolution =
242	0	std::pow(10.0, double(-oWriteOptions_.nXYCoordPrecision));
243	0	auto poNewGeom = std::unique_ptr<OGRGeometry>(
244	0	poFeatureToWrite->GetGeometryRef()->clone());
245	0	OGRGeomCoordinatePrecision sPrecision;
246	0	sPrecision.dfXYResolution = dfXYResolution;
247	0	poNewGeom->roundCoordinates(sPrecision);
248	0	if (!IsValid(poNewGeom.get()))
249	0	{
250	0	std::unique_ptr<OGRGeometry> poValidGeom;
251	0	if (poFeature == poFeatureToWrite)
252	0	{
253	0	CPLDebug("GeoJSON",
254	0	"Running SetPrecision() to correct an invalid "
255	0	"geometry due to reduced precision output");
256	0	poValidGeom.reset(
257	0	poOrigGeom->SetPrecision(dfXYResolution, /* nFlags = */ 0));
258	0	}
259	0	else
260	0	{
261	0	CPLDebug("GeoJSON", "Running MakeValid() to correct an invalid "
262	0	"geometry due to reduced precision output");
263	0	poValidGeom.reset(poNewGeom->MakeValid());
264	0	if (poValidGeom)
265	0	{
266	0	auto poValidGeomRoundCoordinates =
267	0	std::unique_ptr<OGRGeometry>(poValidGeom->clone());
268	0	poValidGeomRoundCoordinates->roundCoordinates(sPrecision);
269	0	if (!IsValid(poValidGeomRoundCoordinates.get()))
270	0	{
271	0	CPLDebug("GeoJSON",
272	0	"Running SetPrecision() to correct an invalid "
273	0	"geometry due to reduced precision output");
274	0	auto poValidGeom2 = std::unique_ptr<OGRGeometry>(
275	0	poValidGeom->SetPrecision(dfXYResolution,
276	0	/* nFlags = */ 0));
277	0	if (poValidGeom2)
278	0	poValidGeom = std::move(poValidGeom2);
279	0	}
280	0	}
281	0	}
282	0	if (poValidGeom)
283	0	{
284	0	if (poFeature == poFeatureToWrite)
285	0	{
286	0	poFeatureToWrite = new OGRFeature(poFeatureDefn_);
287	0	poFeatureToWrite->SetFrom(poFeature);
288	0	poFeatureToWrite->SetFID(poFeature->GetFID());
289	0	}
290	0	poFeatureToWrite->SetGeometryDirectly(poValidGeom.release());
291	0	}
292	0	}
293	0	}
294
295	0	if (oWriteOptions_.bGenerateID && poFeatureToWrite->GetFID() == OGRNullFID)
296	0	{
297	0	poFeatureToWrite->SetFID(nOutCounter_);
298	0	}
299	0	json_object *poObj =
300	0	OGRGeoJSONWriteFeature(poFeatureToWrite, oWriteOptions_);
301	0	CPLAssert(nullptr != poObj);
302
303	0	if (m_nPositionBeforeFCClosed)
304	0	{
305		// If we had called SyncToDisk() previously, undo its effects
306	0	fp->Seek(m_nPositionBeforeFCClosed, SEEK_SET);
307	0	m_nPositionBeforeFCClosed = 0;
308	0	}
309
310	0	if (nOutCounter_ > 0)
311	0	{
312		/* Separate "Feature" entries in "FeatureCollection" object. */
313	0	VSIFPrintfL(fp, ",\n");
314	0	}
315	0	const char *pszJson = json_object_to_json_string_ext(
316	0	poObj, JSON_C_TO_STRING_SPACED
317	0	#ifdef JSON_C_TO_STRING_NOSLASHESCAPE
318	0	\| JSON_C_TO_STRING_NOSLASHESCAPE
319	0	#endif
320	0	);
321
322	0	OGRErr eErr = OGRERR_NONE;
323	0	size_t nLen = strlen(pszJson);
324	0	if (!osForeignMembers_.empty())
325	0	{
326	0	if (nLen > 2 && pszJson[nLen - 2] == ' ' && pszJson[nLen - 1] == '}')
327	0	{
328	0	nLen -= 2;
329	0	}
330	0	else
331	0	{
332		// should not happen
333	0	CPLError(CE_Failure, CPLE_AppDefined,
334	0	"Unexpected JSON output for feature. Cannot write foreign "
335	0	"member");
336	0	osForeignMembers_.clear();
337	0	}
338	0	}
339	0	if (VSIFWriteL(pszJson, nLen, 1, fp) != 1)
340	0	{
341	0	CPLError(CE_Failure, CPLE_FileIO, "Cannot write feature");
342	0	eErr = OGRERR_FAILURE;
343	0	}
344	0	else if (!osForeignMembers_.empty() &&
345	0	(VSIFWriteL(", ", 2, 1, fp) != 1 \|\|
346	0	VSIFWriteL(osForeignMembers_.c_str(), osForeignMembers_.size(), 1,
347	0	fp) != 1 \|\|
348	0	VSIFWriteL("}", 1, 1, fp) != 1))
349	0	{
350	0	CPLError(CE_Failure, CPLE_FileIO, "Cannot write feature");
351	0	eErr = OGRERR_FAILURE;
352	0	}
353
354	0	json_object_put(poObj);
355
356	0	++nOutCounter_;
357
358	0	OGRGeometry *poGeometry = poFeatureToWrite->GetGeometryRef();
359	0	if (poGeometry != nullptr && !poGeometry->IsEmpty())
360	0	{
361	0	OGREnvelope3D sEnvelope = OGRGeoJSONGetBBox(poGeometry, oWriteOptions_);
362	0	if (poGeometry->getCoordinateDimension() == 3)
363	0	bBBOX3D = true;
364
365	0	if (!sEnvelopeLayer.IsInit())
366	0	{
367	0	sEnvelopeLayer = sEnvelope;
368	0	}
369	0	else if (oWriteOptions_.bBBOXRFC7946)
370	0	{
371	0	const bool bEnvelopeCrossAM = (sEnvelope.MinX > sEnvelope.MaxX);
372	0	const bool bEnvelopeLayerCrossAM =
373	0	(sEnvelopeLayer.MinX > sEnvelopeLayer.MaxX);
374	0	if (bEnvelopeCrossAM)
375	0	{
376	0	if (bEnvelopeLayerCrossAM)
377	0	{
378	0	sEnvelopeLayer.MinX =
379	0	std::min(sEnvelopeLayer.MinX, sEnvelope.MinX);
380	0	sEnvelopeLayer.MaxX =
381	0	std::max(sEnvelopeLayer.MaxX, sEnvelope.MaxX);
382	0	}
383	0	else
384	0	{
385	0	if (sEnvelopeLayer.MinX > 0)
386	0	{
387	0	sEnvelopeLayer.MinX =
388	0	std::min(sEnvelopeLayer.MinX, sEnvelope.MinX);
389	0	sEnvelopeLayer.MaxX = sEnvelope.MaxX;
390	0	}
391	0	else if (sEnvelopeLayer.MaxX < 0)
392	0	{
393	0	sEnvelopeLayer.MaxX =
394	0	std::max(sEnvelopeLayer.MaxX, sEnvelope.MaxX);
395	0	sEnvelopeLayer.MinX = sEnvelope.MinX;
396	0	}
397	0	else
398	0	{
399	0	sEnvelopeLayer.MinX = -180.0;
400	0	sEnvelopeLayer.MaxX = 180.0;
401	0	}
402	0	}
403	0	}
404	0	else if (bEnvelopeLayerCrossAM)
405	0	{
406	0	if (sEnvelope.MinX > 0)
407	0	{
408	0	sEnvelopeLayer.MinX =
409	0	std::min(sEnvelopeLayer.MinX, sEnvelope.MinX);
410	0	}
411	0	else if (sEnvelope.MaxX < 0)
412	0	{
413	0	sEnvelopeLayer.MaxX =
414	0	std::max(sEnvelopeLayer.MaxX, sEnvelope.MaxX);
415	0	}
416	0	else
417	0	{
418	0	sEnvelopeLayer.MinX = -180.0;
419	0	sEnvelopeLayer.MaxX = 180.0;
420	0	}
421	0	}
422	0	else
423	0	{
424	0	sEnvelopeLayer.MinX =
425	0	std::min(sEnvelopeLayer.MinX, sEnvelope.MinX);
426	0	sEnvelopeLayer.MaxX =
427	0	std::max(sEnvelopeLayer.MaxX, sEnvelope.MaxX);
428	0	}
429
430	0	sEnvelopeLayer.MinY = std::min(sEnvelopeLayer.MinY, sEnvelope.MinY);
431	0	sEnvelopeLayer.MaxY = std::max(sEnvelopeLayer.MaxY, sEnvelope.MaxY);
432	0	}
433	0	else
434	0	{
435	0	sEnvelopeLayer.Merge(sEnvelope);
436	0	}
437	0	}
438
439	0	if (poFeatureToWrite != poFeature)
440	0	delete poFeatureToWrite;
441
442	0	return eErr;
443	0	}
444
445		/************************************************************************/
446		/* CreateField() */
447		/************************************************************************/
448
449		OGRErr OGRGeoJSONWriteLayer::CreateField(const OGRFieldDefn *poField,
450		int /* bApproxOK */)
451	0	{
452	0	if (poFeatureDefn_->GetFieldIndexCaseSensitive(poField->GetNameRef()) >= 0)
453	0	{
454	0	CPLDebug("GeoJSON", "Field '%s' already present in schema",
455	0	poField->GetNameRef());
456
457		// TODO - mloskot: Is this return code correct?
458	0	return OGRERR_NONE;
459	0	}
460
461	0	poFeatureDefn_->AddFieldDefn(poField);
462
463	0	return OGRERR_NONE;
464	0	}
465
466		/************************************************************************/
467		/* TestCapability() */
468		/************************************************************************/
469
470		int OGRGeoJSONWriteLayer::TestCapability(const char *pszCap)
471	0	{
472	0	if (EQUAL(pszCap, OLCCreateField))
473	0	return TRUE;
474	0	else if (EQUAL(pszCap, OLCSequentialWrite))
475	0	return TRUE;
476	0	else if (EQUAL(pszCap, OLCStringsAsUTF8))
477	0	return TRUE;
478	0	return FALSE;
479	0	}
480
481		/************************************************************************/
482		/* IGetExtent() */
483		/************************************************************************/
484
485		OGRErr OGRGeoJSONWriteLayer::IGetExtent(int /iGeomField/,
486		OGREnvelope *psExtent, bool)
487	0	{
488	0	if (sEnvelopeLayer.IsInit())
489	0	{
490	0	*psExtent = sEnvelopeLayer;
491	0	return OGRERR_NONE;
492	0	}
493	0	return OGRERR_FAILURE;
494	0	}
495
496		/************************************************************************/
497		/* GetDataset() */
498		/************************************************************************/
499
500		GDALDataset *OGRGeoJSONWriteLayer::GetDataset()
501	0	{
502	0	return poDS_;
503	0	}