/******************************************************************************

 *

 * Project:  OpenGIS Simple Features Reference Implementation

 * Purpose:  The OGRMultiSurface class.

 * Author:   Even Rouault <even dot rouault at spatialys dot com>

 *

 ******************************************************************************

 * Copyright (c) 2014, Even Rouault <even dot rouault at spatialys dot com>

 *

 * SPDX-License-Identifier: MIT

 ****************************************************************************/

#include "cpl_port.h"

#include "ogr_geometry.h"

#include <cstddef>

#include "cpl_conv.h"

#include "cpl_error.h"

#include "ogr_api.h"

#include "ogr_core.h"

#include "ogr_p.h"

/************************************************************************/

/*              OGRMultiSurface( const OGRMultiSurface& )               */

/************************************************************************/

/**

 * \brief Copy constructor.

 */

OGRMultiSurface::OGRMultiSurface(const OGRMultiSurface &) = default;

/************************************************************************/

/*                  operator=( const OGRMultiCurve&)                    */

/************************************************************************/

/**

 * \brief Assignment operator.

 */

OGRMultiSurface &OGRMultiSurface::operator=(const OGRMultiSurface &other)

{

    if (this != &other)

    {

        OGRGeometryCollection::operator=(other);

    }

    return *this;

}

/************************************************************************/

/*                               clone()                                */

/************************************************************************/

OGRMultiSurface *OGRMultiSurface::clone() const

{

    auto ret = new (std::nothrow) OGRMultiSurface(*this);

    if (ret)

    {

        if (ret->WkbSize() != WkbSize())

        {

            delete ret;

            ret = nullptr;

        }

    }

    return ret;

}

/************************************************************************/

/*                          getGeometryType()                           */

/************************************************************************/

OGRwkbGeometryType OGRMultiSurface::getGeometryType() const

{

    if ((flags & OGR_G_3D) && (flags & OGR_G_MEASURED))

        return wkbMultiSurfaceZM;

    else if (flags & OGR_G_MEASURED)

        return wkbMultiSurfaceM;

    else if (flags & OGR_G_3D)

        return wkbMultiSurfaceZ;

    else

        return wkbMultiSurface;

}

/************************************************************************/

/*                            getDimension()                            */

/************************************************************************/

int OGRMultiSurface::getDimension() const

{

    return 2;

}

/************************************************************************/

/*                          getGeometryName()                           */

/************************************************************************/

const char *OGRMultiSurface::getGeometryName() const

{

    return "MULTISURFACE";

}

/************************************************************************/

/*                          isCompatibleSubType()                       */

/************************************************************************/

OGRBoolean

OGRMultiSurface::isCompatibleSubType(OGRwkbGeometryType eGeomType) const

{

    OGRwkbGeometryType eFlattenGeomType = wkbFlatten(eGeomType);

    return eFlattenGeomType == wkbPolygon ||

           eFlattenGeomType == wkbCurvePolygon;

}

/************************************************************************/

/*                           importFromWkt()                            */

/*                                                                      */

/*      Instantiate from well known text format.                        */

/************************************************************************/

OGRErr OGRMultiSurface::importFromWkt(const char **ppszInput)

{

    int bHasZ = FALSE;

    int bHasM = FALSE;

    bool bIsEmpty = false;

    OGRErr eErr = importPreambleFromWkt(ppszInput, &bHasZ, &bHasM, &bIsEmpty);

    flags = 0;

    if (eErr != OGRERR_NONE)

        return eErr;

    if (bHasZ)

        flags |= OGR_G_3D;

    if (bHasM)

        flags |= OGR_G_MEASURED;

    if (bIsEmpty)

        return OGRERR_NONE;

    char szToken[OGR_WKT_TOKEN_MAX] = {};

    const char *pszInput = *ppszInput;

    eErr = OGRERR_NONE;

    // Skip first '('.

    pszInput = OGRWktReadToken(pszInput, szToken);

    /* ==================================================================== */

    /*      Read each surface in turn.  Note that we try to reuse the same  */

    /*      point list buffer from ring to ring to cut down on              */

    /*      allocate/deallocate overhead.                                   */

    /* ==================================================================== */

    OGRRawPoint *paoPoints = nullptr;

    int nMaxPoints = 0;

    double *padfZ = nullptr;

    do

    {

        /* --------------------------------------------------------------------

         */

        /*      Get the first token, which should be the geometry type. */

        /* --------------------------------------------------------------------

         */

        const char *pszInputBefore = pszInput;

        pszInput = OGRWktReadToken(pszInput, szToken);

        OGRSurface *poSurface = nullptr;

        /* --------------------------------------------------------------------

         */

        /*      Do the import. */

        /* --------------------------------------------------------------------

         */

        if (EQUAL(szToken, "("))

        {

            OGRPolygon *poPolygon = new OGRPolygon();

            poSurface = poPolygon;

            pszInput = pszInputBefore;

            eErr = poPolygon->importFromWKTListOnly(

                &pszInput, bHasZ, bHasM, paoPoints, nMaxPoints, padfZ);

        }

        else if (EQUAL(szToken, "EMPTY"))

        {

            poSurface = new OGRPolygon();

        }

        // We accept POLYGON() but this is an extension to the BNF, also

        // accepted by PostGIS.

        else if (STARTS_WITH_CI(szToken, "POLYGON") ||

                 STARTS_WITH_CI(szToken, "CURVEPOLYGON"))

        {

            OGRGeometry *poGeom = nullptr;

            pszInput = pszInputBefore;

            eErr =

                OGRGeometryFactory::createFromWkt(&pszInput, nullptr, &poGeom);

            if (poGeom == nullptr)

            {

                eErr = OGRERR_CORRUPT_DATA;

                break;

            }

            poSurface = poGeom->toSurface();

        }

        else

        {

            CPLError(CE_Failure, CPLE_AppDefined, "Unexpected token : %s",

                     szToken);

            eErr = OGRERR_CORRUPT_DATA;

            break;

        }

        if (eErr == OGRERR_NONE)

            eErr = addGeometryDirectly(poSurface);

        if (eErr != OGRERR_NONE)

        {

            delete poSurface;

            break;

        }

        /* --------------------------------------------------------------------

         */

        /*      Read the delimiter following the surface. */

        /* --------------------------------------------------------------------

         */

        pszInput = OGRWktReadToken(pszInput, szToken);

    } while (szToken[0] == ',' && eErr == OGRERR_NONE);

    CPLFree(paoPoints);

    CPLFree(padfZ);

    /* -------------------------------------------------------------------- */

    /*      freak if we don't get a closing bracket.                        */

    /* -------------------------------------------------------------------- */

    if (eErr != OGRERR_NONE)

        return eErr;

    if (szToken[0] != ')')

        return OGRERR_CORRUPT_DATA;

    *ppszInput = pszInput;

    return OGRERR_NONE;

}

/************************************************************************/

/*                            exportToWkt()                             */

/************************************************************************/

std::string OGRMultiSurface::exportToWkt(const OGRWktOptions &opts,

                                         OGRErr *err) const

{

    OGRWktOptions optsModified(opts);

    optsModified.variant = wkbVariantIso;

    return exportToWktInternal(optsModified, err, "POLYGON");

}

/************************************************************************/

/*                         hasCurveGeometry()                           */

/************************************************************************/

OGRBoolean OGRMultiSurface::hasCurveGeometry(int bLookForNonLinear) const

{

    if (bLookForNonLinear)

        return OGRGeometryCollection::hasCurveGeometry(TRUE);

    return TRUE;

}

/************************************************************************/

/*                            PointOnSurface()                          */

/************************************************************************/

/** \brief This method relates to the SFCOM

 * IMultiSurface::get_PointOnSurface() method.

 *

 * NOTE: Only implemented when GEOS included in build.

 *

 * @param poPoint point to be set with an internal point.

 *

 * @return OGRERR_NONE if it succeeds or OGRERR_FAILURE otherwise.

 */

OGRErr OGRMultiSurface::PointOnSurface(OGRPoint *poPoint) const

{

    return PointOnSurfaceInternal(poPoint);

}

/************************************************************************/

/*                         CastToMultiPolygon()                         */

/************************************************************************/

/**

 * \brief Cast to multipolygon.

 *

 * This method should only be called if the multisurface actually only contains

 * instances of OGRPolygon. This can be verified if hasCurveGeometry(TRUE)

 * returns FALSE. It is not intended to approximate curve polygons. For that

 * use getLinearGeometry().

 *

 * The passed in geometry is consumed and a new one returned (or NULL in case

 * of failure).

 *

 * @param poMS the input geometry - ownership is passed to the method.

 * @return new geometry.

 */

OGRMultiPolygon *OGRMultiSurface::CastToMultiPolygon(OGRMultiSurface *poMS)

{

    for (auto &&poSubGeom : *poMS)

    {

        poSubGeom = OGRSurface::CastToPolygon(poSubGeom);

        if (poSubGeom == nullptr)

        {

            delete poMS;

            return nullptr;

        }

    }

    OGRMultiPolygon *poMP = new OGRMultiPolygon();

    TransferMembersAndDestroy(poMS, poMP);

    return poMP;

}