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

 *

 * Project:  OpenGIS Simple Features Reference Implementation

 * Purpose:  Implements OGRGmtDataSource class.

 * Author:   Frank Warmerdam, warmerdam@pobox.com

 *

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

 * Copyright (c) 2007, Frank Warmerdam <warmerdam@pobox.com>

 *

 * SPDX-License-Identifier: MIT

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

#include "cpl_conv.h"

#include "cpl_string.h"

#include "ogr_gmt.h"

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

/*                          OGRGmtDataSource()                          */

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

OGRGmtDataSource::OGRGmtDataSource()

    : papoLayers(nullptr), nLayers(0), bUpdate(false)

{

}

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

/*                         ~OGRGmtDataSource()                          */

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

OGRGmtDataSource::~OGRGmtDataSource()

{

    for (int i = 0; i < nLayers; i++)

        delete papoLayers[i];

    CPLFree(papoLayers);

}

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

/*                                Open()                                */

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

int OGRGmtDataSource::Open(const char *pszFilename, VSILFILE *fp,

                           const OGRSpatialReference *poSRS, int bUpdateIn)

{

    bUpdate = CPL_TO_BOOL(bUpdateIn);

    OGRGmtLayer *poLayer =

        new OGRGmtLayer(this, pszFilename, fp, poSRS, bUpdate);

    if (!poLayer->bValidFile)

    {

        delete poLayer;

        return FALSE;

    }

    papoLayers = static_cast<OGRGmtLayer **>(

        CPLRealloc(papoLayers, (nLayers + 1) * sizeof(OGRGmtLayer *)));

    papoLayers[nLayers] = poLayer;

    nLayers++;

    return TRUE;

}

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

/*                            ICreateLayer()                            */

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

OGRLayer *

OGRGmtDataSource::ICreateLayer(const char *pszLayerName,

                               const OGRGeomFieldDefn *poGeomFieldDefn,

                               CSLConstList /*papszOptions*/)

{

    if (nLayers != 0)

        return nullptr;

    const auto eType = poGeomFieldDefn ? poGeomFieldDefn->GetType() : wkbNone;

    const auto poSRS =

        poGeomFieldDefn ? poGeomFieldDefn->GetSpatialRef() : nullptr;

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

    /*      Establish the geometry type.  Note this logic                   */

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

    const char *pszGeom = nullptr;

    switch (wkbFlatten(eType))

    {

        case wkbPoint:

            pszGeom = " @GPOINT";

            break;

        case wkbLineString:

            pszGeom = " @GLINESTRING";

            break;

        case wkbPolygon:

            pszGeom = " @GPOLYGON";

            break;

        case wkbMultiPoint:

            pszGeom = " @GMULTIPOINT";

            break;

        case wkbMultiLineString:

            pszGeom = " @GMULTILINESTRING";

            break;

        case wkbMultiPolygon:

            pszGeom = " @GMULTIPOLYGON";

            break;

        default:

            pszGeom = "";

            break;

    }

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

    /*      If this is the first layer for this datasource, and if the      */

    /*      datasource name ends in .gmt we will override the provided      */

    /*      layer name with the name from the gmt.                          */

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

    CPLString osPath = CPLGetPathSafe(GetDescription());

    CPLString osFilename(GetDescription());

    const char *pszFlags = "wb+";

    if (osFilename == "/dev/stdout")

        osFilename = "/vsistdout";

    if (STARTS_WITH(osFilename, "/vsistdout"))

        pszFlags = "wb";

    else if (!EQUAL(CPLGetExtensionSafe(GetDescription()).c_str(), "gmt"))

        osFilename = CPLFormFilenameSafe(osPath, pszLayerName, "gmt");

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

    /*      Open the file.                                                  */

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

    VSILFILE *fp = VSIFOpenL(osFilename, pszFlags);

    if (fp == nullptr)

    {

        CPLError(CE_Failure, CPLE_OpenFailed, "open(%s) failed: %s",

                 osFilename.c_str(), VSIStrerror(errno));

        return nullptr;

    }

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

    /*      Write out header.                                               */

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

    VSIFPrintfL(fp, "# @VGMT1.0%s\n", pszGeom);

    if (!STARTS_WITH(osFilename, "/vsistdout"))

    {

        VSIFPrintfL(fp, "# REGION_STUB                                      "

                        "                       \n");

    }

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

    /*      Write the projection, if possible.                              */

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

    if (poSRS != nullptr)

    {

        if (poSRS->GetAuthorityName(nullptr) &&

            EQUAL(poSRS->GetAuthorityName(nullptr), "EPSG"))

        {

            VSIFPrintfL(fp, "# @Je%s\n", poSRS->GetAuthorityCode(nullptr));

        }

        char *pszValue = nullptr;

        if (poSRS->exportToProj4(&pszValue) == OGRERR_NONE)

        {

            VSIFPrintfL(fp, "# @Jp\"%s\"\n", pszValue);

        }

        CPLFree(pszValue);

        pszValue = nullptr;

        if (poSRS->exportToWkt(&pszValue) == OGRERR_NONE)

        {

            char *pszEscapedWkt =

                CPLEscapeString(pszValue, -1, CPLES_BackslashQuotable);

            VSIFPrintfL(fp, "# @Jw\"%s\"\n", pszEscapedWkt);

            CPLFree(pszEscapedWkt);

        }

        CPLFree(pszValue);

    }

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

    /*      Return open layer handle.                                       */

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

    if (Open(osFilename, fp, poSRS, TRUE))

    {

        OGRLayer *poLayer = papoLayers[nLayers - 1];

        if (strcmp(pszGeom, "") != 0)

        {

            poLayer->GetLayerDefn()->SetGeomType(wkbFlatten(eType));

        }

        return poLayer;

    }

    VSIFCloseL(fp);

    return nullptr;

}

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

/*                           TestCapability()                           */

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

int OGRGmtDataSource::TestCapability(const char *pszCap) const

{

    if (EQUAL(pszCap, ODsCCreateLayer))

        return TRUE;

    else if (EQUAL(pszCap, ODsCZGeometries))

        return TRUE;

    return FALSE;

}

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

/*                              GetLayer()                              */

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

const OGRLayer *OGRGmtDataSource::GetLayer(int iLayer) const

{

    if (iLayer < 0 || iLayer >= nLayers)

        return nullptr;

    return papoLayers[iLayer];

}