overlayShapeFileLoadHandler.h

Go to the documentation of this file.

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

#pragma once

#include <vrvUtil/signalslib.h>
#include <vrfGuiCore/vrfGuiCore.h>
#include <vrvUtil/DtVirtualBaseClass.h>
#include <vrvUtil/DtUnicode.h>
#include <vrvCore/DtUniqueID.h>
#include <vlpi/entityType.h>
#include <vrfGuiCore/environmentalCreationData.h>

class OGRSFDriver;
class OGRLayer;
class OGRFeature;
class OGRPoint;
class OGRGeometry;
class OGRDataSource;

namespace makVrv
{
   class DtDe;
}

namespace makVrf
{
   struct DtDataFileCreationData
   {
      DtEnvironmentalCreationData data;
      DtEntityType entityType;
      int force;
      std::vector<DtVector> localLocations;
   };

   class DT_DLL_VRFGUICORE DtDataFileReader
   {
   public:
      DtDataFileReader(makVrv::DtDe&);

      virtual std::vector<DtDataFileCreationData> loadDataFile(const std::string& file, const std::string& defaultOverlay = "") const = 0;

   protected:
      makVrv::DtDe& myDe;
   };

   class DT_DLL_VRFGUICORE DtOGRReader : public DtDataFileReader
   {
   public:
      DtOGRReader(makVrv::DtDe& de, const std::string& gdalDriver);

      virtual ~DtOGRReader();

      virtual std::vector<DtDataFileCreationData> loadDataFile(const std::string& file, const std::string& defaultOverlay = "") const;

   protected:
      virtual OGRSFDriver* createOGRDriver(const char* driverName = "ESRI Shapefile") const;

      virtual bool importLayer(OGRDataSource* dataSource, OGRLayer*, const std::string&, const std::string& shapeFileName,
         std::vector<DtDataFileCreationData>& importedData) const;

      virtual DtDataFileCreationData importFeature(OGRDataSource* dataSource, OGRLayer*, OGRFeature*, const std::string&, const std::string& shapeFileName) const;

      virtual int getForceFromFeature(OGRDataSource* dataSource, OGRLayer*, OGRFeature*, const std::string&) const;

      virtual DtEntityType getEntityTypeToCreateFromFeature(OGRDataSource* dataSource, OGRLayer* layer, OGRFeature*, const std::string& shapeFileName) const;

      virtual int geometryType(OGRGeometry*) const;

      virtual std::vector<DtVector> getFeatureGeometry(OGRFeature*) const;

      virtual DtVector convertPointToLocal(OGRPoint*) const;

      virtual std::string nameFromFeature(OGRFeature*, const std::string& shapeFileName) const;

      struct FeatureStyle
      {
         int penColor;
         int penWidth;
         int brushColor;
      };

      virtual FeatureStyle getStyleForFeature(OGRDataSource* dataSource, OGRLayer*, OGRFeature*) const;

      static void GDALErrorHandler(int eclass, int code, const char *msg );

   protected:
      std::string myGDALDriver;
   };

   class DT_DLL_VRFGUICORE DtOGRReaderCreator : public makVrv::DtVirtualBaseClass
   {
   public:
      DtOGRReaderCreator(makVrv::DtDe&);
      virtual ~DtOGRReaderCreator();

      virtual DtOGRReader* create(const std::string& gdalDriver);

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

      virtual int getForceFromFeature(OGRDataSource* dataSource, OGRLayer*, OGRFeature*, const std::string&) const;

      virtual DtEntityType lineType(OGRDataSource* dataSource, OGRLayer* layer, OGRFeature*, const std::string& shapeFileName) const;
      virtual DtEntityType pointType(OGRDataSource* dataSource, OGRLayer* layer, OGRFeature*, const std::string& shapeFileName) const;
      virtual DtEntityType areaType(OGRDataSource* dataSource, OGRLayer* layer, OGRFeature*, const std::string& shapeFileName) const;

   protected:
      makVrv::DtDe& myDe;
   };

   class DT_DLL_VRFGUICORE DtOverlayShapeFileLoadHandler : public makVrv::DtVirtualBaseClass
   {
   public:
      typedef std::map<std::string, std::pair<DtDataFileReader*, makVrv::DtUnicode> > Drivers;

      DtOverlayShapeFileLoadHandler(makVrv::DtDe& de);

      virtual ~DtOverlayShapeFileLoadHandler();

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

      virtual void setDefaultDriver(const std::string&);
      virtual const std::string& defaultDriver() const;

      virtual bool importShapeFileOntoOverlay(const std::string& shapefile, const std::string& overlay, bool keepLocal = false);

      virtual const DtFilename& importFrom() const;

      virtual DtDataFileReader* readerBasedOnFilename(const std::string&) const;

      virtual void addDriver(const std::string& extension, DtDataFileReader*, const makVrv::DtUnicode& displayString);

      const Drivers& drivers() const
      {
         return myDrivers;
      }

   protected:
      makVrv::DtDe& myDe;
      DtFilename myImportFromDirectory;
      Drivers myDrivers;
      std::string myDefaultDriver;
   };
}