/******************************************************************************
 *
 * Project:  OpenGIS Simple Features Reference Implementation
 * Purpose:  OGR Driver for DGNv8
 * Author:   Even Rouault <even.rouault at spatialys.com>
 *
 ******************************************************************************
 * Copyright (c) 2017, Even Rouault <even.rouault at spatialys.com>
 *
 * SPDX-License-Identifier: MIT
 ****************************************************************************/

#ifndef OGR_DGNV8_H_INCLUDED
#define OGR_DGNV8_H_INCLUDED

#include "ogrsf_frmts.h"

#include <set>
#include <utility>
#include <vector>

#include "dgnv8_headers.h"

/************************************************************************/
/*                         OGRDGNV8Services                             */
/*                                                                      */
/*      Services implementation for OGR.  Eventually we should          */
/*      override the OdExDgnSystemServices IO to use VSI*L.             */
/************************************************************************/

class OGRDGNV8Services : public OdExDgnSystemServices,
                         public OdExDgnHostAppServices
{
  protected:
    ODRX_USING_HEAP_OPERATORS(OdExDgnSystemServices);
};

/************************************************************************/
/*                           OGRDGNV8Layer                              */
/************************************************************************/

class OGRDGNV8DataSource;
typedef std::pair<OGRFeature *, bool> tPairFeatureHoleFlag;

class OGRDGNV8Layer final : public OGRLayer
{
    friend class OGRDGNV8DataSource;

    OGRDGNV8DataSource *m_poDS;
    OGRFeatureDefn *m_poFeatureDefn;
    OdDgModelPtr m_pModel;
    OdDgElementIteratorPtr m_pIterator;
    std::vector<tPairFeatureHoleFlag> m_aoPendingFeatures;
    size_t m_nIdxInPendingFeatures;
    std::set<CPLString> m_aoSetIgnoredFeatureClasses;

    void CleanPendingFeatures();
    std::vector<tPairFeatureHoleFlag>
    CollectSubElements(OdDgElementIteratorPtr iterator, int level);
    std::vector<tPairFeatureHoleFlag>
    ProcessElement(OdDgGraphicsElementPtr element, int level = 0);
    OGRFeature *GetNextUnfilteredFeature();

    void AddToComplexCurve(OGRFeature *poFeature, OGRCircularString *poCS,
                           OdDgComplexCurvePtr complexCurve);
    void AddToComplexCurve(OGRFeature *poFeature, OGRCompoundCurve *poCC,
                           OdDgComplexCurvePtr complexCurve);
    OdDgGraphicsElementPtr CreateShape(OGRFeature *poFeature, OGRCurve *poCurve,
                                       bool bIsHole = false);
    OdDgGraphicsElementPtr CreateGraphicsElement(OGRFeature *poFeature,
                                                 OGRGeometry *poGeom);
    OdDgGraphicsElementPtr TranslateLabel(OGRFeature *poFeature,
                                          OGRPoint *poPoint);
    void AttachFillLinkage(OGRFeature *poFeature,
                           OdDgGraphicsElementPtr element);
    void AttachCommonAttributes(OGRFeature *poFeature,
                                OdDgGraphicsElementPtr element);
    int GetColorFromString(const char *pszColor);
    OdDgGraphicsElementPtr GetFeatureInternal(GIntBig nFID,
                                              OdDg::OpenMode openMode);

  public:
    OGRDGNV8Layer(OGRDGNV8DataSource *poDS, OdDgModelPtr pModel);
    ~OGRDGNV8Layer() override;

    void ResetReading() override;
    OGRFeature *GetNextFeature() override;
    OGRFeature *GetFeature(GIntBig nFID) override;

    OGRErr IGetExtent(int iGeomField, OGREnvelope *psExtent,
                      bool bForce) override;

    const OGRFeatureDefn *GetLayerDefn() const override
    {
        return m_poFeatureDefn;
    }

    int TestCapability(const char *) const override;

    OGRErr ICreateFeature(OGRFeature *poFeature) override;
    OGRErr DeleteFeature(GIntBig nFID) override;

    GDALDataset *GetDataset() override;
};

/************************************************************************/
/*                         OGRDGNV8DataSource                           */
/************************************************************************/

class OGRDGNV8DataSource final : public GDALDataset
{
    OGRDGNV8Services *m_poServices;
    OGRDGNV8Layer **m_papoLayers;
    int m_nLayers;
    char **m_papszOptions;
    OdDgDatabasePtr m_poDb;
    bool m_bUpdate;
    bool m_bModified;
    CPLStringList m_osDGNMD;

    void InitWithSeed();

  public:
    explicit OGRDGNV8DataSource(OGRDGNV8Services *poServices);
    ~OGRDGNV8DataSource() override;

    int Open(const char *, bool bUpdate);
    bool PreCreate(const char *, char **);

    OGRLayer *ICreateLayer(const char *pszName,
                           const OGRGeomFieldDefn *poGeomFieldDefn,
                           CSLConstList papszOptions) override;

    int GetLayerCount() const override
    {
        return m_nLayers;
    }

    const OGRLayer *GetLayer(int) const override;

    int TestCapability(const char *) const override;
    CPLErr FlushCache(bool bAtClosing) override;

    char **GetMetadataDomainList() override;
    char **GetMetadata(const char *pszDomain = "") override;
    virtual const char *GetMetadataItem(const char *pszName,
                                        const char *pszDomain = "") override;

    OdDgDatabasePtr GetDb() const
    {
        return m_poDb;
    }

    bool GetUpdate() const
    {
        return m_bUpdate;
    }

    void SetModified()
    {
        m_bModified = true;
    }

    static OdString FromUTF8(const CPLString &str);
    static CPLString ToUTF8(const OdString &str);
};

#endif /* ndef OGR_DGNV8_H_INCLUDED */