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_Tool2.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() );
63 aPanel->setGuiData( myObject );
66 else if ( isExport() )
68 if ( !myObject.IsNull() )
69 aPanel->setTableName( myObject->GetName() );
71 aPanel->setTableNameReadOnly( isExport() );
74 void HYDROGUI_StricklerTableOp::abortOperation()
76 HYDROGUI_Operation::abortOperation();
79 void HYDROGUI_StricklerTableOp::commitOperation()
81 HYDROGUI_Operation::commitOperation();
84 HYDROGUI_InputPanel* HYDROGUI_StricklerTableOp::createInputPanel() const
86 int type = isEdit() ? HYDROGUI_StricklerTableDlg::Edit : ( isImport() ? HYDROGUI_StricklerTableDlg::Import : HYDROGUI_StricklerTableDlg::Export );
87 HYDROGUI_StricklerTableDlg* aPanel = new HYDROGUI_StricklerTableDlg( module(), getName(), type );
88 connect( aPanel, SIGNAL( fileSelected( const QString& ) ), SLOT( onFileSelected() ) );
92 bool HYDROGUI_StricklerTableOp::processApply( int& theUpdateFlags, QString& theErrorMsg,
93 QStringList& theBrowseObjectsEntries )
95 HYDROGUI_StricklerTableDlg* aPanel = ::qobject_cast<HYDROGUI_StricklerTableDlg*>( inputPanel() );
102 aFilePath = aPanel->getFileName();
103 if ( aFilePath.isEmpty() )
105 theErrorMsg = tr( "SELECT_STRICKLER_TABLE_FILE" ).arg( aFilePath );
110 QString aStricklerTableName = aPanel->getTableName().simplified();
111 if ( aStricklerTableName.isEmpty() )
113 theErrorMsg = tr( "INCORRECT_OBJECT_NAME" );
117 if ( isImport() || ( isEdit() && !myObject.IsNull() && myObject->GetName() != aStricklerTableName ) )
119 // check that there are no other objects with the same name in the document
120 Handle(HYDROData_Entity) anObject = HYDROGUI_Tool::FindObjectByName( module(), aStricklerTableName );
121 if( !anObject.IsNull() )
123 theErrorMsg = tr( "OBJECT_EXISTS_IN_DOCUMENT" ).arg( aStricklerTableName );
128 Handle(HYDROData_StricklerTable) aStricklerTableObj;
131 aStricklerTableObj = Handle(HYDROData_StricklerTable)::DownCast( doc()->CreateObject( KIND_STRICKLER_TABLE ) );
132 QString anEntry = HYDROGUI_DataObject::dataObjectEntry( aStricklerTableObj );
133 theBrowseObjectsEntries.append( anEntry );
136 aStricklerTableObj = myObject;
138 if ( aStricklerTableObj.IsNull() )
144 QString aFilePath = aPanel->getFileName().simplified();
145 if ( !aFilePath.isEmpty() )
146 res = aStricklerTableObj->Export( aFilePath );
152 aStricklerTableObj->SetName( aStricklerTableName );
156 // Get data from input panel's table and save it into data model object
157 aPanel->getGuiData( aStricklerTableObj );
161 // Import data from Strickler table file into data model object
162 aStricklerTableObj->Import( aFilePath );
165 aStricklerTableObj->Update();
167 commitDocOperation();
171 QString anEntry = HYDROGUI_DataObject::dataObjectEntry( aStricklerTableObj );
172 theBrowseObjectsEntries.append( anEntry );
174 module()->getOCCDisplayer()->SetToUpdateColorScale();
177 theUpdateFlags |= UF_ObjBrowser | UF_OCCViewer;
182 void HYDROGUI_StricklerTableOp::onFileSelected()
184 HYDROGUI_StricklerTableDlg* aPanel =
185 ::qobject_cast<HYDROGUI_StricklerTableDlg*>( inputPanel() );
189 QString anStricklerTableName = aPanel->getTableName().simplified();
190 if ( anStricklerTableName.isEmpty() )
192 anStricklerTableName = aPanel->getFileName();
193 if ( !anStricklerTableName.isEmpty() ) {
194 anStricklerTableName = QFileInfo( anStricklerTableName ).baseName();
197 if ( anStricklerTableName.isEmpty() ) {
198 anStricklerTableName = HYDROGUI_Tool::GenerateObjectName( module(), tr( "DEFAULT_STRICKLER_TABLE_NAME" ) );
200 aPanel->setTableName( anStricklerTableName );
204 bool HYDROGUI_StricklerTableOp::isEdit() const
206 return myType == EditStricklerTableId;
209 bool HYDROGUI_StricklerTableOp::isImport() const
211 return myType == ImportStricklerTableFromFileId;
214 bool HYDROGUI_StricklerTableOp::isExport() const
216 return myType == ExportStricklerTableFromFileId;