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_Operations.h"
24 #include "HYDROGUI_Tool.h"
26 #include <HYDROData_Document.h>
28 #include <LightApp_UpdateFlags.h>
32 HYDROGUI_StricklerTableOp::HYDROGUI_StricklerTableOp( HYDROGUI_Module* theModule, int theType )
33 : HYDROGUI_Operation( theModule ),
36 setName( isEdit() ? tr( "EDIT_STRICKLER_TABLE" ) :
37 ( isImport() ? tr( "IMPORT_STRICKLER_TABLE" ) : tr( "EXPORT_STRICKLER_TABLE" ) ) );
40 HYDROGUI_StricklerTableOp::~HYDROGUI_StricklerTableOp()
44 void HYDROGUI_StricklerTableOp::startOperation()
46 HYDROGUI_Operation::startOperation();
48 HYDROGUI_StricklerTableDlg* aPanel = (HYDROGUI_StricklerTableDlg*)inputPanel();
51 if ( !isImport() && isApplyAndClose() )
52 myObject = Handle(HYDROData_StricklerTable)::DownCast( HYDROGUI_Tool::GetSelectedObject( module() ) );
56 if ( !myObject.IsNull() )
58 // Edit selected Strickler table
59 aPanel->setTableName( myObject->GetName() );
61 // Get Strickler table data from the data model
62 HYDROGUI_StricklerTableDlg::StricklerCoefficientList aData;
63 TColStd_SequenceOfExtendedString aTypes = myObject->GetTypes();
64 for ( int i = 1; i <= aTypes.Length(); i++ )
65 aData.append( HYDROGUI_StricklerTableDlg::StricklerCoefficient( HYDROGUI_Tool::ToQString( aTypes.Value( i ) ),
66 myObject->Get( aTypes.Value( i ), 0 ) ) );
68 aPanel->setData( aData );
71 else if ( isExport() )
73 if ( !myObject.IsNull() )
74 aPanel->setTableName( myObject->GetName() );
76 aPanel->setTableNameReadOnly( isExport() );
79 void HYDROGUI_StricklerTableOp::abortOperation()
81 HYDROGUI_Operation::abortOperation();
84 void HYDROGUI_StricklerTableOp::commitOperation()
86 HYDROGUI_Operation::commitOperation();
89 HYDROGUI_InputPanel* HYDROGUI_StricklerTableOp::createInputPanel() const
91 int type = isEdit() ? HYDROGUI_StricklerTableDlg::Edit : ( isImport() ? HYDROGUI_StricklerTableDlg::Import : HYDROGUI_StricklerTableDlg::Export );
92 HYDROGUI_StricklerTableDlg* aPanel = new HYDROGUI_StricklerTableDlg( module(), getName(), type );
93 connect( aPanel, SIGNAL( fileSelected( const QString& ) ), SLOT( onFileSelected() ) );
97 bool HYDROGUI_StricklerTableOp::processApply( int& theUpdateFlags, QString& theErrorMsg,
98 QStringList& theBrowseObjectsEntries )
100 HYDROGUI_StricklerTableDlg* aPanel = ::qobject_cast<HYDROGUI_StricklerTableDlg*>( inputPanel() );
107 aFilePath = aPanel->getFileName();
108 if ( aFilePath.isEmpty() )
110 theErrorMsg = tr( "SELECT_STRICKLER_TABLE_FILE" ).arg( aFilePath );
115 QString aStricklerTableName = aPanel->getTableName().simplified();
116 if ( aStricklerTableName.isEmpty() )
118 theErrorMsg = tr( "INCORRECT_OBJECT_NAME" );
122 if ( isImport() || ( isEdit() && !myObject.IsNull() && myObject->GetName() != aStricklerTableName ) )
124 // check that there are no other objects with the same name in the document
125 Handle(HYDROData_Entity) anObject = HYDROGUI_Tool::FindObjectByName( module(), aStricklerTableName );
126 if( !anObject.IsNull() )
128 theErrorMsg = tr( "OBJECT_EXISTS_IN_DOCUMENT" ).arg( aStricklerTableName );
133 Handle(HYDROData_StricklerTable) aStricklerTableObj;
136 aStricklerTableObj = Handle(HYDROData_StricklerTable)::DownCast( doc()->CreateObject( KIND_STRICKLER_TABLE ) );
137 QString anEntry = HYDROGUI_DataObject::dataObjectEntry( aStricklerTableObj );
138 theBrowseObjectsEntries.append( anEntry );
141 aStricklerTableObj = myObject;
143 if ( aStricklerTableObj.IsNull() )
149 QString aFilePath = aPanel->getFileName().simplified();
150 if ( !aFilePath.isEmpty() )
151 res = aStricklerTableObj->Export( HYDROGUI_Tool::ToAsciiString( aFilePath ) );
157 aStricklerTableObj->SetName( aStricklerTableName );
161 // Get data from input panel's table and save it into data model object
162 aStricklerTableObj->Clear();
163 HYDROGUI_StricklerTableDlg::StricklerCoefficientList aData = aPanel->getData();
164 for ( HYDROGUI_StricklerTableDlg::StricklerCoefficientList::iterator it = aData.begin(); it != aData.end(); ++it )
166 const HYDROGUI_StricklerTableDlg::StricklerCoefficient& anInfo = *it;
167 aStricklerTableObj->Set( HYDROGUI_Tool::ToExtString( anInfo.myType ), anInfo.myCoefficient );
172 // Import data from Strickler table file into data model object
173 aStricklerTableObj->Import( HYDROGUI_Tool::ToAsciiString( aFilePath ) );
176 aStricklerTableObj->Update();
178 commitDocOperation();
182 QString anEntry = HYDROGUI_DataObject::dataObjectEntry( aStricklerTableObj );
183 theBrowseObjectsEntries.append( anEntry );
186 theUpdateFlags |= UF_ObjBrowser;
191 void HYDROGUI_StricklerTableOp::onFileSelected()
193 HYDROGUI_StricklerTableDlg* aPanel =
194 ::qobject_cast<HYDROGUI_StricklerTableDlg*>( inputPanel() );
198 QString anStricklerTableName = aPanel->getTableName().simplified();
199 if ( anStricklerTableName.isEmpty() )
201 anStricklerTableName = aPanel->getFileName();
202 if ( !anStricklerTableName.isEmpty() ) {
203 anStricklerTableName = QFileInfo( anStricklerTableName ).baseName();
206 if ( anStricklerTableName.isEmpty() ) {
207 anStricklerTableName = HYDROGUI_Tool::GenerateObjectName( module(), tr( "DEFAULT_STRICKLER_TABLE_NAME" ) );
209 aPanel->setTableName( anStricklerTableName );
213 bool HYDROGUI_StricklerTableOp::isEdit() const
215 return myType == EditStricklerTableId;
218 bool HYDROGUI_StricklerTableOp::isImport() const
220 return myType == ImportStricklerTableFromFileId;
223 bool HYDROGUI_StricklerTableOp::isExport() const
225 return myType == ExportStricklerTableFromFileId;