propShapefileCreateHandler.h

Go to the documentation of this file.

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

#pragma once

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

#include <map>
#include <string>

namespace makVrv
{
   class DtDe;
   class DtUnicode;
}

namespace makArchives
{
   class DtPropRecord;
   class DtTerrainRecord;
}

class OGRLayer;
class OGRSFDriver;
class OGRDataSource;

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

      virtual ~DtPropShapefileCreateHandler();

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

      virtual const DtFilename& exportTo() const;
      virtual void setExportToDirectory(const DtFilename&);

      virtual const DtFilename& propInformationFilename() const;
      virtual void setPropInformationFilename(const DtFilename&);

      virtual bool exportPropsToShapefile(const std::string&, makVrv::DtUnicode& sError, DtString& filesCreated) const;

   protected:
      typedef struct PropInformation
      {
         double length;
         double width;
         double height;
         int fidCode;
      } PropInformation;

      typedef std::map<std::string, PropInformation> PropInformationMap;

      virtual PropInformationMap readPropInformation(const DtFilename&) const;

      virtual void addPropRecords(const std::vector<makArchives::DtPropRecord>& propRecords, OGRLayer* poLayer,
         const PropInformationMap& propInformationMap, const DtFilename& dataRoot, bool forSpeedTree) const;

      virtual DtFilename modelDefinitionFilename(const std::string& modelDefinition, const DtFilename& dataRoot,
         bool& isSpeedtree) const;

      virtual bool createBuildingsShapefile(const makArchives::DtTerrainRecord&, const PropInformationMap&, const DtFilename& relativePath,
         OGRSFDriver*, const DtFilename&, makVrv::DtUnicode& sError, bool& haveSome) const;
      virtual bool createSpeedTreeShapefile(const makArchives::DtTerrainRecord&, const PropInformationMap&, const DtFilename& relativePath,
         OGRSFDriver*, const DtFilename&, makVrv::DtUnicode& sError, bool& haveSome) const;
      virtual void removeSourceFiles(const DtFilename&, const std::string& type) const;

      virtual OGRDataSource* createDataSource(OGRSFDriver*, const DtFilename&, const std::string& type, makVrv::DtUnicode& sError) const;
      virtual OGRLayer* createLayer(OGRDataSource*, bool forSpeedTree, makVrv::DtUnicode& sError) const;

      virtual bool hasBuildings(const makArchives::DtTerrainRecord& terrainRecord, const DtFilename&) const;
      virtual bool hasSpeedTrees(const makArchives::DtTerrainRecord& terrainRecord, const DtFilename&) const;

   protected:
      DtFilename myExportToDirectory;
      DtFilename myPropInformationFilename;
      makVrv::DtDe& myDe;
   };
}