]> SALOME platform Git repositories - modules/hydro.git/blob - src/HYDROGUI/HYDROGUI_ExportFileOp.cxx
Salome HOME
Merge branch 'BR_H2018_2' of https://codev-tuleap.cea.fr/plugins/git/salome/hydro...
[modules/hydro.git] / src / HYDROGUI / HYDROGUI_ExportFileOp.cxx
1 // Copyright (C) 2014-2015  EDF-R&D
2 // This library is free software; you can redistribute it and/or
3 // modify it under the terms of the GNU Lesser General Public
4 // License as published by the Free Software Foundation; either
5 // version 2.1 of the License, or (at your option) any later version.
6 //
7 // This library is distributed in the hope that it will be useful,
8 // but WITHOUT ANY WARRANTY; without even the implied warranty of
9 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
10 // Lesser General Public License for more details.
11 //
12 // You should have received a copy of the GNU Lesser General Public
13 // License along with this library; if not, write to the Free Software
14 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
15 //
16 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
17 //
18
19 #include "HYDROGUI_ExportFileOp.h"
20
21 #include "HYDROGUI_DataModel.h"
22 #include "HYDROGUI_Module.h"
23 #include "HYDROGUI_UpdateFlags.h"
24 #include "HYDROGUI_Tool2.h"
25 #include "HYDROGUI_ExportLandCoverMapDlg.h"
26 #include <HYDROData_PolylineXY.h>
27 #include <HYDROData_Polyline3D.h>
28 #include <HYDROGUI_DataObject.h>
29 #include <HYDROData_Bathymetry.h>
30 #include <HYDROData_LandCoverMap.h>
31
32 #include <HYDROData_StricklerTable.h>
33
34 #include <HYDROData_Profile.h>
35
36 #include <SUIT_Desktop.h>
37 #include <SUIT_FileDlg.h>
38 #include <LightApp_Application.h>
39
40 #include <QApplication>
41 #include <QFile>
42 #include <QFileInfo>
43 #include <SUIT_MessageBox.h>
44 #include <TopoDS.hxx>
45 #include <TopExp_Explorer.hxx>
46 #include <TopoDS_Wire.hxx>
47 #include <TopoDS_Vertex.hxx>
48 #include <TopoDS_Edge.hxx>
49 #include <BRep_Tool.hxx>
50 #include <Precision.hxx>
51 #include <Geom_Curve.hxx>
52 #include <Geom_Line.hxx>
53 #include <Geom_TrimmedCurve.hxx>
54 #include <Geom_TrimmedCurve.hxx>
55 #include <HYDROData_ShapeFile.h>
56 #include <SUIT_Desktop.h>
57 #include <SalomeApp_Study.h>
58 #include <HYDROData_Iterator.h>
59
60 #include <LightApp_Application.h>
61 #include <QSet>
62
63 HYDROGUI_ExportFileOp::HYDROGUI_ExportFileOp( HYDROGUI_Module* theModule )
64 : HYDROGUI_Operation( theModule )
65 {
66   setName( tr( "EXPORT_TO_SHAPE_FILE" ) );
67 }
68
69 HYDROGUI_ExportFileOp::~HYDROGUI_ExportFileOp()
70 {
71 }
72
73 void HYDROGUI_ExportFileOp::startOperation()
74 {
75   HYDROGUI_Operation::startOperation();
76   
77   QString aFilter( tr("SHP_FILTER") ); 
78   
79   Handle(HYDROData_PolylineXY) aPolyXY;
80   Handle(HYDROData_Polyline3D) aPoly3D;
81   NCollection_Sequence<Handle(HYDROData_PolylineXY)> aPolyXYSeq;
82   NCollection_Sequence<Handle(HYDROData_Polyline3D)> aPoly3DSeq;
83   //
84   HYDROData_SequenceOfObjects aSeq = HYDROGUI_Tool::GetSelectedObjects( module() );
85   for( int anIndex = 1, aLength = aSeq.Length(); anIndex <= aLength; anIndex++ )
86   {
87     aPolyXY = Handle(HYDROData_PolylineXY)::DownCast( aSeq.Value( anIndex ));
88     if (!aPolyXY.IsNull())
89       aPolyXYSeq.Append(aPolyXY);
90
91     aPoly3D = Handle(HYDROData_Polyline3D)::DownCast( aSeq.Value( anIndex ));
92     if (!aPoly3D.IsNull())
93       aPoly3DSeq.Append(aPoly3D);
94   }
95
96   if (!aPolyXYSeq.IsEmpty() && !aPoly3DSeq.IsEmpty())
97     SUIT_MessageBox::warning( module()->getApp()->desktop(), tr( "EXPORT_TO_SHAPE_FILE" ), tr("ERROR_POLYLINES_OF_DIFF_KIND"));
98   else
99   {
100     QString aName = "";
101     if (aPolyXYSeq.Size() == 1 && aPoly3DSeq.IsEmpty())
102       aName = aPolyXYSeq(1)->GetName();
103     if (aPoly3DSeq.Size() == 1 && aPolyXYSeq.IsEmpty())
104       aName = aPoly3DSeq(1)->GetName();
105     QString aFileName = SUIT_FileDlg::getFileName( module()->getApp()->desktop(), aName, aFilter, tr( "EXPORT_TO_SHAPE_FILE" ), false );
106     if (!aFileName.isEmpty())
107     {
108       QStringList aNonExpList;
109       HYDROData_ShapeFile anExporter;
110       Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( application()->activeStudy()->id() );
111       if (!aPolyXYSeq.IsEmpty() || !aPolyXYSeq.IsEmpty())
112         //Export polylines
113         anExporter.Export(aDoc, aFileName, aPolyXYSeq, aPoly3DSeq, aNonExpList);
114       else
115       {
116         //Export polygons
117         //Extract all attribute names from all strickler tables
118         QSet<QString> anAttrNames;
119         //use QSet to store attribute names. Yet it's not so good if the document contains two strickler types
120         //with the same name
121         if (aDoc)
122         {
123           HYDROData_Iterator anIt( aDoc, KIND_STRICKLER_TABLE );
124           for( ; anIt.More(); anIt.Next() )
125           {
126             Handle(HYDROData_StricklerTable) aStricklerTableObj = Handle(HYDROData_StricklerTable)::DownCast( anIt.Current() );
127             if ( !aStricklerTableObj.IsNull())
128               anAttrNames << aStricklerTableObj->GetAttrName();
129           }
130         }
131         else
132           return; 
133
134         QStringList SortedListOfAttr = anAttrNames.toList();
135         SortedListOfAttr.sort();
136         //
137         Handle(HYDROData_LandCoverMap) aLCM = Handle(HYDROData_LandCoverMap)::DownCast( aSeq(1) );
138         bool IsLinear = aLCM->CheckLinear();
139         HYDROGUI_ExportLandCoverMapDlg aDlg( module()->getApp()->desktop(), IsLinear, SortedListOfAttr);
140         if ( aDlg.exec() == HYDROGUI_ExportLandCoverMapDlg::Accepted )
141         { 
142           //In our case :  aSeq.Size() == 1
143           //Export of multiple landcover maps into the one shp-file is disallowed.
144           QString aCItem = aDlg.getCurrentItem();
145           Handle(HYDROData_StricklerTable) aStricklerTableObj;
146           {
147             HYDROData_Iterator anIt( aDoc, KIND_STRICKLER_TABLE );
148             for( ; anIt.More(); anIt.Next() )
149             {
150               aStricklerTableObj = Handle(HYDROData_StricklerTable)::DownCast( anIt.Current() );
151               if ( !aStricklerTableObj.IsNull() && aStricklerTableObj->GetAttrName() == aCItem)
152                 break;
153             }
154           }
155
156           //export shape-data
157           anExporter.Export(aDoc, aFileName, aLCM, aNonExpList, false, !IsLinear, aDlg.getDeflValue());
158           QString aDBFFileName = aFileName.replace( ".shp", ".dbf", Qt::CaseInsensitive);
159           //Even if attribute-checkbox is unchecked, the .dbf-file should be removed. 
160           //otherwise it may be used with wrong .shp-file. This is an incorrect behaivor.
161           remove (aDBFFileName.toStdString().c_str());
162           bool bToSaveAttrInfo = aDlg.getAttrCheckBoxState() && !aDlg.getCurrentItem().isEmpty();
163           if (bToSaveAttrInfo)
164           {
165             //export attribute info
166             QStringList anAttrValues;
167             QStringList aStricklerTypes;
168             if (aNonExpList.empty())
169             {
170               HYDROData_LandCoverMap::Explorer aLCMIt( aLCM );
171               for( ; aLCMIt.More(); aLCMIt.Next() )
172               {
173                 QString aST = aLCMIt.StricklerType();
174                 anAttrValues << aStricklerTableObj->GetAttrValue(aST);
175                 aStricklerTypes << aST;
176               }
177             }
178             aLCM->ExportDBF(aDBFFileName, aCItem, anAttrValues, aStricklerTypes);
179           }
180         }
181         else
182         {
183           abort();
184           return;
185         }
186
187       }
188
189       if (!aNonExpList.empty())
190       {
191         QString aMessage = tr("CANNOT_BE_EXPORTED") + "\n";
192         foreach (QString anNonExpEntName, aNonExpList)
193           aMessage += anNonExpEntName + "\n"; 
194         SUIT_MessageBox::warning( module()->getApp()->desktop(), tr( "EXPORT_TO_SHAPE_FILE" ), aMessage);
195       }
196       commit();
197     }
198     else
199       abort();
200   }
201
202 }
203