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 TColStd_SequenceOfExtendedString aTypes = myObject->GetTypes();
67 for ( int i = 1; i <= aTypes.Length(); i++ )
68 aData.append( HYDROGUI_StricklerTableDlg::StricklerCoefficient( HYDROGUI_Tool::ToQString( aTypes.Value( i ) ),
69 myObject->Get( aTypes.Value( i ), 0 ) ) );
71 aPanel->setData( aData );
74 else if ( isExport() )
76 if ( !myObject.IsNull() )
77 aPanel->setTableName( myObject->GetName() );
79 aPanel->setTableNameReadOnly( isExport() );
82 void HYDROGUI_StricklerTableOp::abortOperation()
84 HYDROGUI_Operation::abortOperation();
87 void HYDROGUI_StricklerTableOp::commitOperation()
89 HYDROGUI_Operation::commitOperation();
92 HYDROGUI_InputPanel* HYDROGUI_StricklerTableOp::createInputPanel() const
94 int type = isEdit() ? HYDROGUI_StricklerTableDlg::Edit : ( isImport() ? HYDROGUI_StricklerTableDlg::Import : HYDROGUI_StricklerTableDlg::Export );
95 HYDROGUI_StricklerTableDlg* aPanel = new HYDROGUI_StricklerTableDlg( module(), getName(), type );
96 connect( aPanel, SIGNAL( fileSelected( const QString& ) ), SLOT( onFileSelected() ) );
100 bool HYDROGUI_StricklerTableOp::processApply( int& theUpdateFlags, QString& theErrorMsg,
101 QStringList& theBrowseObjectsEntries )
103 HYDROGUI_StricklerTableDlg* aPanel = ::qobject_cast<HYDROGUI_StricklerTableDlg*>( inputPanel() );
110 aFilePath = aPanel->getFileName();
111 if ( aFilePath.isEmpty() )
113 theErrorMsg = tr( "SELECT_STRICKLER_TABLE_FILE" ).arg( aFilePath );
118 QString aStricklerTableName = aPanel->getTableName().simplified();
119 if ( aStricklerTableName.isEmpty() )
121 theErrorMsg = tr( "INCORRECT_OBJECT_NAME" );
125 if ( isImport() || ( isEdit() && !myObject.IsNull() && myObject->GetName() != aStricklerTableName ) )
127 // check that there are no other objects with the same name in the document
128 Handle(HYDROData_Entity) anObject = HYDROGUI_Tool::FindObjectByName( module(), aStricklerTableName );
129 if( !anObject.IsNull() )
131 theErrorMsg = tr( "OBJECT_EXISTS_IN_DOCUMENT" ).arg( aStricklerTableName );
136 Handle(HYDROData_StricklerTable) aStricklerTableObj;
139 aStricklerTableObj = Handle(HYDROData_StricklerTable)::DownCast( doc()->CreateObject( KIND_STRICKLER_TABLE ) );
140 QString anEntry = HYDROGUI_DataObject::dataObjectEntry( aStricklerTableObj );
141 theBrowseObjectsEntries.append( anEntry );
144 aStricklerTableObj = myObject;
146 if ( aStricklerTableObj.IsNull() )
152 QString aFilePath = aPanel->getFileName().simplified();
153 if ( !aFilePath.isEmpty() )
154 res = aStricklerTableObj->Export( HYDROGUI_Tool::ToAsciiString( aFilePath ) );
160 aStricklerTableObj->SetName( aStricklerTableName );
164 // Get data from input panel's table and save it into data model object
165 aStricklerTableObj->Clear();
166 HYDROGUI_StricklerTableDlg::StricklerCoefficientList aData = aPanel->getData();
167 for ( HYDROGUI_StricklerTableDlg::StricklerCoefficientList::iterator it = aData.begin(); it != aData.end(); ++it )
169 const HYDROGUI_StricklerTableDlg::StricklerCoefficient& anInfo = *it;
170 aStricklerTableObj->Set( HYDROGUI_Tool::ToExtString( anInfo.myType ), anInfo.myCoefficient );
175 // Import data from Strickler table file into data model object
176 aStricklerTableObj->Import( HYDROGUI_Tool::ToAsciiString( aFilePath ) );
179 aStricklerTableObj->Update();
181 commitDocOperation();
185 QString anEntry = HYDROGUI_DataObject::dataObjectEntry( aStricklerTableObj );
186 theBrowseObjectsEntries.append( anEntry );
188 module()->getOCCDisplayer()->SetToUpdateColorScale();
191 theUpdateFlags |= UF_ObjBrowser | UF_OCCViewer;
196 void HYDROGUI_StricklerTableOp::onFileSelected()
198 HYDROGUI_StricklerTableDlg* aPanel =
199 ::qobject_cast<HYDROGUI_StricklerTableDlg*>( inputPanel() );
203 QString anStricklerTableName = aPanel->getTableName().simplified();
204 if ( anStricklerTableName.isEmpty() )
206 anStricklerTableName = aPanel->getFileName();
207 if ( !anStricklerTableName.isEmpty() ) {
208 anStricklerTableName = QFileInfo( anStricklerTableName ).baseName();
211 if ( anStricklerTableName.isEmpty() ) {
212 anStricklerTableName = HYDROGUI_Tool::GenerateObjectName( module(), tr( "DEFAULT_STRICKLER_TABLE_NAME" ) );
214 aPanel->setTableName( anStricklerTableName );
218 bool HYDROGUI_StricklerTableOp::isEdit() const
220 return myType == EditStricklerTableId;
223 bool HYDROGUI_StricklerTableOp::isImport() const
225 return myType == ImportStricklerTableFromFileId;
228 bool HYDROGUI_StricklerTableOp::isExport() const
230 return myType == ExportStricklerTableFromFileId;