Salome HOME
refs #669, #677: selection of land covers in extended mode without pressing Shift...
[modules/hydro.git] / src / HYDROGUI / HYDROGUI_ImportBathymetryOp.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_ImportBathymetryOp.h"
20
21 #include "HYDROGUI_DataModel.h"
22 #include "HYDROGUI_DataObject.h"
23 #include "HYDROGUI_ImportBathymetryDlg.h"
24 #include "HYDROGUI_Module.h"
25 #include "HYDROGUI_Tool.h"
26 #include "HYDROGUI_UpdateFlags.h"
27
28 #include <HYDROData_Bathymetry.h>
29
30 #include <LightApp_Application.h>
31 #include <LightApp_UpdateFlags.h>
32 #include <SUIT_ViewManager.h>
33 #include <SVTK_ViewModel.h>
34
35 #include <QFileInfo>
36
37 HYDROGUI_ImportBathymetryOp::HYDROGUI_ImportBathymetryOp( HYDROGUI_Module* theModule, 
38                                                          const bool theIsEdit  )
39 : HYDROGUI_Operation( theModule ),
40   myIsEdit( theIsEdit )
41 {
42   setName( theIsEdit ? tr( "EDIT_IMPORTED_BATHYMETRY" ) : tr( "IMPORT_BATHYMETRY" ) );
43 }
44
45 HYDROGUI_ImportBathymetryOp::~HYDROGUI_ImportBathymetryOp()
46 {
47 }
48
49 void HYDROGUI_ImportBathymetryOp::startOperation()
50 {
51   HYDROGUI_Operation::startOperation();
52
53   HYDROGUI_ImportBathymetryDlg* aPanel = 
54     ::qobject_cast<HYDROGUI_ImportBathymetryDlg*>( inputPanel() );
55   if ( !aPanel )
56     return;
57
58   aPanel->reset();
59
60   if( myIsEdit )
61   {
62     if ( isApplyAndClose() )
63       myEditedObject = Handle(HYDROData_Bathymetry)::DownCast( HYDROGUI_Tool::GetSelectedObject( module() ) );
64     if( !myEditedObject.IsNull() )
65     {
66       QString aName = myEditedObject->GetName();
67       QString aFileName = HYDROGUI_Tool::ToQString( myEditedObject->GetFilePath() );
68       bool anIsAltitudesInverted = myEditedObject->IsAltitudesInverted();
69
70       aPanel->setObjectName( aName );
71       aPanel->setFileName( aFileName );
72       aPanel->setInvertAltitudes( anIsAltitudesInverted );
73     }
74   }
75 }
76
77 void HYDROGUI_ImportBathymetryOp::abortOperation()
78 {
79   HYDROGUI_Operation::abortOperation();
80 }
81
82 void HYDROGUI_ImportBathymetryOp::commitOperation()
83 {
84   HYDROGUI_Operation::commitOperation();
85 }
86
87 HYDROGUI_InputPanel* HYDROGUI_ImportBathymetryOp::createInputPanel() const
88 {
89   HYDROGUI_InputPanel* aPanel = new HYDROGUI_ImportBathymetryDlg( module(), getName() );
90   
91   connect ( aPanel, SIGNAL( FileSelected( const QString& ) ), SLOT( onFileSelected() ) );
92
93   return aPanel;
94 }
95
96 bool HYDROGUI_ImportBathymetryOp::processApply( int& theUpdateFlags,
97                                                 QString& theErrorMsg,
98                                                 QStringList& theBrowseObjectsEntries )
99 {
100   HYDROGUI_ImportBathymetryDlg* aPanel = 
101     ::qobject_cast<HYDROGUI_ImportBathymetryDlg*>( inputPanel() );
102   if ( !aPanel )
103     return false;
104
105   QString anObjectName = aPanel->getObjectName().simplified();
106   if ( anObjectName.isEmpty() )
107   {
108     theErrorMsg = tr( "INCORRECT_OBJECT_NAME" );
109     return false;
110   }
111
112   QString aFileName = aPanel->getFileName().simplified();
113   bool anIsInvertAltitudes = aPanel->isInvertAltitudes();
114
115   if ( aFileName.isEmpty() )
116   {
117     theErrorMsg = tr( "INCORRECT_FILE_NAME" );
118     return false;
119   }
120
121   QFileInfo aFileInfo( aFileName );
122   if ( !aFileInfo.exists() || !aFileInfo.isReadable() )
123   {
124     theErrorMsg = tr( "FILE_NOT_EXISTS_OR_CANT_BE_READ" ).arg( aFileName );
125     return false;
126   }
127
128   if( !myIsEdit || ( !myEditedObject.IsNull() && myEditedObject->GetName() != anObjectName ) )
129   {
130     // check that there are no other objects with the same name in the document
131     Handle(HYDROData_Entity) anObject = HYDROGUI_Tool::FindObjectByName( module(), anObjectName );
132     if( !anObject.IsNull() )
133     {
134       theErrorMsg = tr( "OBJECT_EXISTS_IN_DOCUMENT" ).arg( anObjectName );
135       return false;
136     }
137   }
138
139   Handle(HYDROData_Bathymetry) aBathymetryObj;
140   if ( myIsEdit )
141   {
142     aBathymetryObj = myEditedObject;
143   }
144   else
145   {
146     aBathymetryObj = 
147       Handle(HYDROData_Bathymetry)::DownCast( doc()->CreateObject( KIND_BATHYMETRY ) );
148   }
149   if ( aBathymetryObj.IsNull() )
150     return false;
151
152   QString anOldFileName = HYDROGUI_Tool::ToQString( aBathymetryObj->GetFilePath() );
153   if ( aFileName != anOldFileName )
154   {
155     aBathymetryObj->SetAltitudesInverted( anIsInvertAltitudes, false );
156     if ( !aBathymetryObj->ImportFromFile( HYDROGUI_Tool::ToAsciiString( aFileName ) ) )
157     {
158       theErrorMsg = tr( "BAD_IMPORTED_BATHYMETRY_FILE" ).arg( aFileName );
159       return false;
160     }
161   }
162   else if ( anIsInvertAltitudes != aBathymetryObj->IsAltitudesInverted() )
163   {
164     aBathymetryObj->SetAltitudesInverted( anIsInvertAltitudes );
165   }
166
167   aBathymetryObj->SetName( anObjectName );
168
169   aBathymetryObj->Update();
170
171   // Activate VTK viewer and show the bathymetry
172   SUIT_ViewManager* aVTKMgr = 0;
173   SUIT_ViewManager* aViewMgr = module()->getApp()->activeViewManager();
174   // Try to get a VTK viewer as an active or existing one
175   if ( aViewMgr )
176   {
177     if ( aViewMgr->getType() == SVTK_Viewer::Type() )
178     {
179       aVTKMgr = aViewMgr;
180     }
181     else
182     {
183       aVTKMgr = module()->getApp()->viewManager( SVTK_Viewer::Type() );
184     }
185   }
186
187   /*
188   // If there is no VTK viewer yet then create a new one
189   if ( !aVTKMgr )
190   {
191     aVTKMgr = module()->getApp()->createViewManager( SVTK_Viewer::Type() );
192   }
193   // Set the bathymetry visible in the VTK viewer
194   if ( aVTKMgr )
195   {
196     module()->setObjectVisible( (size_t)aVTKMgr->getViewModel(), aBathymetryObj, true );
197   }*/
198
199   theUpdateFlags = UF_Model | UF_OCCViewer | UF_OCC_Forced |
200                               UF_VTKViewer | UF_VTK_Forced;
201
202   if( !myIsEdit )
203   {
204     QString anEntry = HYDROGUI_DataObject::dataObjectEntry( aBathymetryObj );
205     theBrowseObjectsEntries.append( anEntry );
206   }
207
208   return true;
209 }
210
211 void HYDROGUI_ImportBathymetryOp::onFileSelected()
212 {
213   HYDROGUI_ImportBathymetryDlg* aPanel = 
214     ::qobject_cast<HYDROGUI_ImportBathymetryDlg*>( inputPanel() );
215   if ( !aPanel )
216     return;
217
218   QString anObjectName = aPanel->getObjectName().simplified();
219   //if ( anObjectName.isEmpty() )
220   {
221     anObjectName = aPanel->getFileName();
222     if ( !anObjectName.isEmpty() ) {
223         anObjectName = QFileInfo( anObjectName ).baseName();
224     }
225
226     if ( anObjectName.isEmpty() ) {
227       anObjectName = HYDROGUI_Tool::GenerateObjectName( module(), tr( "DEFAULT_BATHYMETRY_NAME" ) );
228     }
229     aPanel->setObjectName( anObjectName );
230   }
231 }
232
233
234