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.
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.
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
16 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
19 #include "HYDROGUI_StricklerTableOp.h"
21 #include "HYDROGUI_StricklerTableDlg.h"
22 #include "HYDROGUI_DataObject.h"
23 #include "HYDROGUI_Module.h"
24 #include "HYDROGUI_OCCDisplayer.h"
25 #include "HYDROGUI_Operations.h"
26 #include "HYDROGUI_Tool.h"
27 #include "HYDROGUI_UpdateFlags.h"
29 #include <HYDROData_Document.h>
31 #include <LightApp_UpdateFlags.h>
35 HYDROGUI_StricklerTableOp::HYDROGUI_StricklerTableOp( HYDROGUI_Module* theModule, int theType )
36 : HYDROGUI_Operation( theModule ),
39 setName( isEdit() ? tr( "EDIT_STRICKLER_TABLE" ) :
40 ( isImport() ? tr( "IMPORT_STRICKLER_TABLE" ) : tr( "EXPORT_STRICKLER_TABLE" ) ) );
43 HYDROGUI_StricklerTableOp::~HYDROGUI_StricklerTableOp()
47 void HYDROGUI_StricklerTableOp::startOperation()
49 HYDROGUI_Operation::startOperation();
51 HYDROGUI_StricklerTableDlg* aPanel = (HYDROGUI_StricklerTableDlg*)inputPanel();
54 if ( !isImport() && isApplyAndClose() )
55 myObject = Handle(HYDROData_StricklerTable)::DownCast( HYDROGUI_Tool::GetSelectedObject( module() ) );
59 if ( !myObject.IsNull() )
61 // Edit selected Strickler table
62 aPanel->setTableName( myObject->GetName() );
64 // Get Strickler table data from the data model
65 HYDROGUI_StricklerTableDlg::StricklerCoefficientList aData;
66 QStringList aTypes = myObject->GetTypes();
67 for ( QStringList::iterator it = aTypes.begin(); it != aTypes.end(); ++it )
68 aData.append( HYDROGUI_StricklerTableDlg::StricklerCoefficient( *it, myObject->Get( *it, 0 ) ) );
70 aPanel->setData( aData );
73 else if ( isExport() )
75 if ( !myObject.IsNull() )
76 aPanel->setTableName( myObject->GetName() );
78 aPanel->setTableNameReadOnly( isExport() );
81 void HYDROGUI_StricklerTableOp::abortOperation()
83 HYDROGUI_Operation::abortOperation();
86 void HYDROGUI_StricklerTableOp::commitOperation()
88 HYDROGUI_Operation::commitOperation();
91 HYDROGUI_InputPanel* HYDROGUI_StricklerTableOp::createInputPanel() const
93 int type = isEdit() ? HYDROGUI_StricklerTableDlg::Edit : ( isImport() ? HYDROGUI_StricklerTableDlg::Import : HYDROGUI_StricklerTableDlg::Export );
94 HYDROGUI_StricklerTableDlg* aPanel = new HYDROGUI_StricklerTableDlg( module(), getName(), type );
95 connect( aPanel, SIGNAL( fileSelected( const QString& ) ), SLOT( onFileSelected() ) );
99 bool HYDROGUI_StricklerTableOp::processApply( int& theUpdateFlags, QString& theErrorMsg,
100 QStringList& theBrowseObjectsEntries )
102 HYDROGUI_StricklerTableDlg* aPanel = ::qobject_cast<HYDROGUI_StricklerTableDlg*>( inputPanel() );
109 aFilePath = aPanel->getFileName();
110 if ( aFilePath.isEmpty() )
112 theErrorMsg = tr( "SELECT_STRICKLER_TABLE_FILE" ).arg( aFilePath );
117 QString aStricklerTableName = aPanel->getTableName().simplified();
118 if ( aStricklerTableName.isEmpty() )
120 theErrorMsg = tr( "INCORRECT_OBJECT_NAME" );
124 if ( isImport() || ( isEdit() && !myObject.IsNull() && myObject->GetName() != aStricklerTableName ) )
126 // check that there are no other objects with the same name in the document
127 Handle(HYDROData_Entity) anObject = HYDROGUI_Tool::FindObjectByName( module(), aStricklerTableName );
128 if( !anObject.IsNull() )
130 theErrorMsg = tr( "OBJECT_EXISTS_IN_DOCUMENT" ).arg( aStricklerTableName );
135 Handle(HYDROData_StricklerTable) aStricklerTableObj;
138 aStricklerTableObj = Handle(HYDROData_StricklerTable)::DownCast( doc()->CreateObject( KIND_STRICKLER_TABLE ) );
139 QString anEntry = HYDROGUI_DataObject::dataObjectEntry( aStricklerTableObj );
140 theBrowseObjectsEntries.append( anEntry );
143 aStricklerTableObj = myObject;
145 if ( aStricklerTableObj.IsNull() )
151 QString aFilePath = aPanel->getFileName().simplified();
152 if ( !aFilePath.isEmpty() )
153 res = aStricklerTableObj->Export( aFilePath );
159 aStricklerTableObj->SetName( aStricklerTableName );
163 // Get data from input panel's table and save it into data model object
164 aStricklerTableObj->Clear();
165 HYDROGUI_StricklerTableDlg::StricklerCoefficientList aData = aPanel->getData();
166 for ( HYDROGUI_StricklerTableDlg::StricklerCoefficientList::iterator it = aData.begin(); it != aData.end(); ++it )
168 const HYDROGUI_StricklerTableDlg::StricklerCoefficient& anInfo = *it;
169 aStricklerTableObj->Set( anInfo.myType, anInfo.myCoefficient );
174 // Import data from Strickler table file into data model object
175 aStricklerTableObj->Import( aFilePath );
178 aStricklerTableObj->Update();
180 commitDocOperation();
184 QString anEntry = HYDROGUI_DataObject::dataObjectEntry( aStricklerTableObj );
185 theBrowseObjectsEntries.append( anEntry );
187 module()->getOCCDisplayer()->SetToUpdateColorScale();
190 theUpdateFlags |= UF_ObjBrowser | UF_OCCViewer;
195 void HYDROGUI_StricklerTableOp::onFileSelected()
197 HYDROGUI_StricklerTableDlg* aPanel =
198 ::qobject_cast<HYDROGUI_StricklerTableDlg*>( inputPanel() );
202 QString anStricklerTableName = aPanel->getTableName().simplified();
203 if ( anStricklerTableName.isEmpty() )
205 anStricklerTableName = aPanel->getFileName();
206 if ( !anStricklerTableName.isEmpty() ) {
207 anStricklerTableName = QFileInfo( anStricklerTableName ).baseName();
210 if ( anStricklerTableName.isEmpty() ) {
211 anStricklerTableName = HYDROGUI_Tool::GenerateObjectName( module(), tr( "DEFAULT_STRICKLER_TABLE_NAME" ) );
213 aPanel->setTableName( anStricklerTableName );
217 bool HYDROGUI_StricklerTableOp::isEdit() const
219 return myType == EditStricklerTableId;
222 bool HYDROGUI_StricklerTableOp::isImport() const
224 return myType == ImportStricklerTableFromFileId;
227 bool HYDROGUI_StricklerTableOp::isExport() const
229 return myType == ExportStricklerTableFromFileId;