overlayShapeFileSaveHandler.h

Go to the documentation of this file.

/*********************************************************************
** Copyright (c) 2016 MaK Technologies, Inc.
** All rights reserved.
*********************************************************************/

#pragma once

#include <vrfGuiCore/vrfGuiCore.h>
#include <vrvUtil/DtVirtualBaseClass.h>
#include <vrvCore/DtUniqueID.h>
#include <vlutil/vlFilename.h>

#include <set>

namespace makVrv
{
   class DtDe;
   class DtModel;
   class DtSceneObject;
}

class GDALDriver;
class OGRLayer;
class GDALDataset;
class OGRGeometry;

namespace makVrf
{
   class DT_DLL_VRFGUICORE DtOverlayShapeFileSaveHandler : public makVrv::DtVirtualBaseClass
   {
   public:
      DtOverlayShapeFileSaveHandler(makVrv::DtDe& de);

      virtual ~DtOverlayShapeFileSaveHandler();

      static DtOverlayShapeFileSaveHandler& instance(makVrv::DtDe& de);
      static void registerInstance(makVrv::DtDe& de, DtOverlayShapeFileSaveHandler*);

      virtual bool overlayLayerCanBeExported(const std::string& overlayLayer) const;

      virtual bool exportOverlay(const std::string& overlay, const std::string& rootname);

      virtual const DtFilename& exportTo() const;

      virtual bool isExportable(makVrv::DtUniqueID) const;

   protected:
      virtual bool exportPointModels(const std::set<makVrv::DtUniqueID>& points, const std::string& root,
         const std::string& type = "P");
      virtual bool exportAreaModels(const std::set<makVrv::DtUniqueID>& areas, const std::string& root,
         const std::string& type = "A");
      virtual bool exportLinearModels(const std::set<makVrv::DtUniqueID>& lines, const std::string& root,
         const std::string& type = "L");

      virtual void removeExistingFiles(const std::string&) const;

   protected:
      virtual bool isExportableModel(makVrv::DtModel*) const;

      virtual bool isPointModel(makVrv::DtModel*) const;
      virtual bool isAreaModel(makVrv::DtModel*) const;
      virtual bool isLinearModel(makVrv::DtModel*) const;

      virtual bool isPointModel(makVrv::DtUniqueID) const;
      virtual bool isAreaModel(makVrv::DtUniqueID) const;
      virtual bool isLinearModel(makVrv::DtUniqueID) const;

      virtual makVrv::DtModel* modelFor(makVrv::DtUniqueID) const;

      virtual makVrv::DtSceneObject* findSceneObjectFor(makVrv::DtUniqueID) const;

      virtual GDALDriver* createOGRDriver(const char* driverName = "ESRI Shapefile");

      virtual OGRLayer* initializeLayerBasedOnType(const std::string& root, const std::string& fileExtension, int ogrType);

      virtual void addFeaturesToLayer(const std::set<makVrv::DtUniqueID>& items, OGRLayer*, int type) const;

      virtual void cleanup();

      virtual OGRGeometry* convertObjectToGeometry(makVrv::DtUniqueID, int type) const;
      virtual std::string objectName(makVrv::DtUniqueID) const;
      virtual std::string objectPaletteName(makVrv::DtUniqueID) const;
      virtual std::string objectEntityType(makVrv::DtUniqueID) const;

   protected:
      makVrv::DtDe& myDe;
      GDALDriver *myCurrentDriver;
      GDALDataset *myCurrentDataSource;
      DtFilename myExportToDirectory;
   };
}