src/HYDROData/HYDROData_ShapeFile.h

   1 // Copyright (C) 2007-2015  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   5 //
   6 // This library is free software; you can redistribute it and/or
   7 // modify it under the terms of the GNU Lesser General Public
   8 // License as published by the Free Software Foundation; either
   9 // version 2.1 of the License, or (at your option) any later version.
  10 //
  11 // This library is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 // Lesser General Public License for more details.
  15 //
  16 // You should have received a copy of the GNU Lesser General Public
  17 // License along with this library; if not, write to the Free Software
  18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  19 //
  20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21 //
  22
  23 #ifndef HYDRODATA_SHAPEFILE_H
  24 #define HYDRODATA_SHAPEFILE_H
  25
  26 #include <vector>
  27 #include <NCollection_Sequence.hxx>
  28 #include <QStringList>
  29 #include "HYDROData.h"
  30
  31 //extern "C" {
  32 #include <shapefil.h>
  33 //};
  34
  35 class gp_XYZ;
  36 class Handle_HYDROData_PolylineXY;
  37 class Handle_HYDROData_Polyline3D;
  38 class Handle(HYDROData_Document);
  39 class TopTools_SequenceOfShape;
  40 class TopoDS_Face;
  41 class Handle_HYDROData_Entity;
  42 class Handle_HYDROData_LandCoverMap;
  43
  44 class HYDROData_ShapeFile
  45 {
  46
  47 public:
  48
  49 enum ShapeType
  50 {
  51   ShapeType_Polyline,
  52   ShapeType_Polygon
  53 };
  54
  55 enum DBF_FieldType
  56 {
  57   DBF_FieldType_None,
  58   DBF_FieldType_String,
  59   DBF_FieldType_Integer,
  60   DBF_FieldType_Double,
  61   DBF_FieldType_Invalid
  62 };
  63
  64
  65 struct DBF_AttrValue
  66 {
  67   DBF_AttrValue()
  68   {
  69     myIntVal = -1;
  70     myDoubleVal = -1.0;
  71     myIsNull = true;
  72     myFieldType = DBF_FieldType_None;
  73   }
  74   QString myStrVal;
  75   int myIntVal;
  76   double myDoubleVal;
  77   std::string myRawValue;
  78   DBF_FieldType myFieldType;
  79   bool myIsNull;
  80 };
  81
  82 public:
  83   HYDRODATA_EXPORT HYDROData_ShapeFile(  );
  84   virtual HYDRODATA_EXPORT ~HYDROData_ShapeFile();
  85
  86   //Export operation
  87   HYDRODATA_EXPORT void Export(const QString& aFileName,
  88                                NCollection_Sequence<Handle_HYDROData_PolylineXY> aPolyXYSeq,
  89                                NCollection_Sequence<Handle_HYDROData_Polyline3D> aPoly3DSeq,
  90                                const Handle_HYDROData_LandCoverMap& aLCSeq,
  91                                QStringList& aNonExpList);
  92   int WriteObjectPolyXY(SHPHandle theShpHandle, Handle_HYDROData_PolylineXY thePoly );
  93   int WriteObjectPoly3D(SHPHandle theShpHandle, Handle_HYDROData_Polyline3D thePoly );
  94   //TODO:int WriteObjectLC(SHPHandle theShpHandle, Handle_HYDROData_LandCover theLC );
  95   //Import
  96   bool Parse(SHPHandle theHandle, ShapeType theType, int& theShapeTypeOfFile);
  97   //Import Landcover
  98   void ReadSHPPolygon(SHPObject* anObj, int i, TopoDS_Face& F);
  99   HYDRODATA_EXPORT int ImportLandCovers(const QString theFileName, QStringList& thePolygonsList, TopTools_SequenceOfShape& theFaces, int& theShapeTypeOfFile);
 100   HYDRODATA_EXPORT void Free();
 101   //Import Polyline
 102   void ReadSHPPolyXY(Handle(HYDROData_Document) theDocument, SHPObject* anObj, QString theFileName,
 103                         int theInd, NCollection_Sequence<Handle_HYDROData_Entity>& theEntities);
 104   void ReadSHPPoly3D(Handle(HYDROData_Document) theDocument, SHPObject* anObj, QString theFileName,
 105                         int theInd, NCollection_Sequence<Handle_HYDROData_Entity>& theEntities);
 106   HYDRODATA_EXPORT int ImportPolylines(Handle(HYDROData_Document) theDocument, const QString& theFileName,
 107                        NCollection_Sequence<Handle_HYDROData_Entity>& theEntities, int& theShapeTypeOfFile);
 108   HYDRODATA_EXPORT QString GetShapeTypeName(int theType);
 109
 110   //DBF I/O Methods
 111   bool CheckDBFFileExisting(const QString& theSHPFilePath, QString& thePathToDBFFile);
 112   bool DBF_OpenDBF(const QString& thePathToDBFFile);
 113   int DBF_GetNbFields();
 114   QStringList DBF_GetFieldList();
 115   void DBF_GetFieldTypeList(std::vector<DBF_FieldType>& FTVect);
 116   int DBF_GetNbRecords();
 117   void DBF_CloseDBF();
 118   void DBF_GetAttributeList(int theIndexOfField, std::vector<DBF_AttrValue>& theAttrV);
 119   bool DBF_WriteFieldAndValues(const QString& theFileName, const QString& theFieldName, DBF_FieldType theType, const std::vector<DBF_AttrValue>& theAttrV, bool bUseStrValue);
 120
 121 private:
 122   void ProcessFace(TopoDS_Face theFace, SHPHandle theShpHandle);
 123   int TryOpenShapeFile(QString theFileName);
 124 private:
 125   std::vector<SHPObject*> mySHPObjects;
 126   SHPHandle myHSHP;
 127   DBFHandle     myHDBF;
 128
 129   friend class test_HYDROData_ShapeFile;
 130 };
 131
 132 #endif