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