1 // Copyright (C) 2007-2013 CEA/DEN, EDF R&D, OPEN CASCADE
3 // Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
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.
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.
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
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
23 #include "HYDROGUI_CalculationDlg.h"
25 #include "HYDROGUI_ObjSelector.h"
26 #include "HYDROGUI_Tool.h"
27 #include "HYDROGUI_DataBrowser.h"
28 #include "HYDROGUI_DataModel.h"
29 #include "HYDROGUI_Module.h"
30 #include "HYDROGUI_DataObject.h"
31 #include "HYDROGUI_NameValidator.h"
32 #include "HYDROGUI_Region.h"
33 #include "HYDROGUI_Zone.h"
35 #include <HYDROData_Document.h>
36 #include <HYDROData_Entity.h>
38 #include <CAM_Application.h>
39 #include <LightApp_DataObject.h>
41 #include <SUIT_DataObject.h>
42 #include <SUIT_FileDlg.h>
43 #include <SUIT_ResourceMgr.h>
44 #include <SUIT_Session.h>
45 #include <SUIT_Study.h>
47 #include <LightApp_Application.h>
48 #include <SUIT_Desktop.h>
49 #include <SUIT_MessageBox.h>
55 #include <QListWidget>
57 #include <QPushButton>
58 #include <QToolButton>
59 #include <QWizardPage>
61 #include <QStackedWidget>
63 HYDROGUI_CalculationDlg::HYDROGUI_CalculationDlg( HYDROGUI_Module* theModule, const QString& theTitle )
64 : HYDROGUI_Wizard( theModule, theTitle )
66 addPage( createObjectsPage() );
67 addPage( createZonesPage() );
68 connect( button( QWizard::NextButton ), SIGNAL( clicked() ), SIGNAL( splitZones() ) );
69 connect( button( QWizard::BackButton ), SIGNAL( clicked() ), SIGNAL( hideZones() ) );
72 HYDROGUI_CalculationDlg::~HYDROGUI_CalculationDlg()
76 void HYDROGUI_CalculationDlg::reset()
78 myObjectName->clear();
79 myGeomObjects->clear();
82 QWizardPage* HYDROGUI_CalculationDlg::createObjectsPage() {
83 QWizardPage* aPage = new QWizardPage( mainFrame() );
84 QFrame* aFrame = new QFrame( aPage );
87 myObjectName = new QLineEdit( aPage );
88 myValidator = new HYDROGUI_NameValidator(module(), myObjectName);
89 myObjectName->setValidator( myValidator );
91 connect( myValidator, SIGNAL( emptyName() ), SLOT( onEmptyName() ) );
92 connect( myValidator, SIGNAL( alreadyExists( QString ) ), SLOT( onAlreadyExists( QString ) ) );
94 myPolylineName = new QComboBox( aPage );
96 myGeomObjects = new QListWidget( aPage );
97 myGeomObjects->setSelectionMode( QListWidget::SingleSelection );
98 myGeomObjects->setEditTriggers( QListWidget::NoEditTriggers );
99 myGeomObjects->setViewMode( QListWidget::ListMode );
100 myGeomObjects->setSortingEnabled( true );
102 myAvailableGeomObjects = new QListWidget( aPage );
103 myAvailableGeomObjects->setSelectionMode( QListWidget::SingleSelection );
104 myAvailableGeomObjects->setEditTriggers( QListWidget::NoEditTriggers );
105 myAvailableGeomObjects->setViewMode( QListWidget::ListMode );
106 myAvailableGeomObjects->setSortingEnabled( true );
108 connect( myGeomObjects, SIGNAL( currentTextChanged( const QString & ) ),
109 SIGNAL( objectSelected( const QString & ) ) );
111 QFrame* anObjectsFrame = new QFrame( aPage );
112 QGridLayout* anObjsLayout = new QGridLayout( anObjectsFrame );
113 anObjsLayout->setMargin( 5 );
114 anObjsLayout->setSpacing( 5 );
115 anObjectsFrame->setLayout( anObjsLayout );
117 QFrame* aBtnsFrame = new QFrame( anObjectsFrame );
118 QVBoxLayout* aBtnsLayout = new QVBoxLayout( aBtnsFrame );
119 aBtnsLayout->setMargin( 5 );
120 aBtnsLayout->setSpacing( 5 );
121 aBtnsFrame->setLayout( aBtnsLayout );
122 QPushButton* anAddBtn = new QPushButton( tr("INCLUDE"), aBtnsFrame );
123 QPushButton* aRemoveBtn = new QPushButton( tr("EXCLUDE"), aBtnsFrame );
125 // Fill the butons frame with two buttons
126 aBtnsLayout->addWidget( anAddBtn );
127 aBtnsLayout->addWidget( aRemoveBtn );
128 aBtnsLayout->addStretch( 1 );
130 QLabel* anIncludedLabel = new QLabel( tr( "INCLUDED_OBJECTS" ), anObjectsFrame );
131 QLabel* anObjectsLabel = new QLabel( tr( "CALCULATION_REFERENCE_OBJECTS" ), anObjectsFrame );
133 // Fill the objects frame with two lists, two labels and with buttons frame
134 anObjsLayout->addWidget( anObjectsLabel, 0, 0, Qt::AlignHCenter );
135 anObjsLayout->addWidget( anIncludedLabel, 0, 2, Qt::AlignHCenter );
136 anObjsLayout->addWidget( myAvailableGeomObjects, 1, 0, Qt::AlignHCenter );
137 anObjsLayout->addWidget( aBtnsFrame, 1, 1, Qt::AlignHCenter );
138 anObjsLayout->addWidget( myGeomObjects, 1, 2, Qt::AlignHCenter );
141 QLabel* aNameLabel = new QLabel( tr( "NAME" ), aPage );
142 QLabel* aLimitsLabel = new QLabel( tr( "LIMITS" ), aPage );
145 QGridLayout* aPageLayout = new QGridLayout( aPage );
146 aPageLayout->setMargin( 5 );
147 aPageLayout->setSpacing( 5 );
148 aPageLayout->setVerticalSpacing( 10 );
149 aPageLayout->addWidget( aNameLabel, 0, 0, Qt::AlignHCenter );
150 aPageLayout->addWidget( myObjectName, 0, 1 );
151 aPageLayout->addWidget( aLimitsLabel, 1, 0, Qt::AlignHCenter );
152 aPageLayout->addWidget( myPolylineName, 1, 1 );
153 aPageLayout->addWidget( anObjectsFrame, 2, 0, 1, 2, Qt::AlignHCenter );
155 aPage->setLayout( aPageLayout );
157 connect( anAddBtn, SIGNAL( clicked() ), SIGNAL( addObjects() ) );
158 connect( aRemoveBtn, SIGNAL( clicked() ), SIGNAL( removeObjects() ) );
163 void HYDROGUI_CalculationDlg::onEmptyName()
165 QString aTitle = QObject::tr( "INSUFFICIENT_INPUT_DATA" );
166 QString aMessage = QObject::tr( "INCORRECT_OBJECT_NAME" );
167 SUIT_MessageBox::critical( module()->getApp()->desktop(), aTitle, aMessage );
170 void HYDROGUI_CalculationDlg::onAlreadyExists( QString theName )
172 QString aTitle = QObject::tr( "INSUFFICIENT_INPUT_DATA" );
173 QString aMessage = QObject::tr( "OBJECT_EXISTS_IN_DOCUMENT" ).arg( theName );
174 SUIT_MessageBox::critical( module()->getApp()->desktop(), aTitle, aMessage );
177 QWizardPage* HYDROGUI_CalculationDlg::createZonesPage() {
178 QWizardPage* aPage = new QWizardPage( mainFrame() );
179 QFrame* aFrame = new QFrame( aPage );
181 QGridLayout* aLayout = new QGridLayout( aPage );
183 myBrowser = new HYDROGUI_DataBrowser( module(), 0, aPage );
184 myBrowser->setAutoOpenLevel( 3 );
185 aLayout->setMargin( 5 );
186 aLayout->setSpacing( 5 );
188 aLayout->addWidget( myBrowser, 0, 0, 1, 2 );
190 myBatimetryLabel = new QLabel( tr( "BATHYMETRY" ), aFrame );
191 myBathymetryChoice = new QComboBox( aFrame );
193 myBathymetryChoice->setVisible( false );
194 myBatimetryLabel->setVisible( false );
196 aLayout->addWidget( myBatimetryLabel, 1, 0 );
197 aLayout->addWidget( myBathymetryChoice, 1, 1 );
199 aPage->setLayout( aLayout );
201 connect( myBrowser, SIGNAL( clicked( SUIT_DataObject* ) ), SIGNAL( clickedInZonesBrowser( SUIT_DataObject* ) ) );
202 connect( myBrowser, SIGNAL( clicked( SUIT_DataObject* ) ), SLOT( onSelected( SUIT_DataObject* ) ) );
203 connect( myBathymetryChoice, SIGNAL( activated( int ) ), SLOT( onMergeTypeSelected( int ) ) );
205 SIGNAL( dropped( const QList<SUIT_DataObject*>&, SUIT_DataObject*, int, Qt::DropAction ) ),
206 SLOT( onZonesDropped( const QList<SUIT_DataObject*>&, SUIT_DataObject*, int, Qt::DropAction ) ) );
211 void HYDROGUI_CalculationDlg::onZonesDropped( const QList<SUIT_DataObject*>& theList,
212 SUIT_DataObject* theTargetParent, int theTargetRow, Qt::DropAction theDropAction )
214 QList<SUIT_DataObject*> aZonesList;
215 HYDROGUI_Zone* aZone;
216 // Get a list of dropped zones
217 for ( int i = 0; i < theList.length(); i++ )
219 aZone = dynamic_cast<HYDROGUI_Zone*>( theList.at( i ) );
222 aZonesList.append( aZone );
225 if ( aZonesList.length() > 0 )
227 // Get the target region
228 HYDROGUI_NamedObject* aRegionsRoot = dynamic_cast<HYDROGUI_NamedObject*>(theTargetParent);
231 // Create a new region
232 emit createRegion( aZonesList );
236 HYDROGUI_Region* aRegion = dynamic_cast<HYDROGUI_Region*>(theTargetParent);
239 emit moveZones( theTargetParent, aZonesList );
245 void HYDROGUI_CalculationDlg::onMergeTypeSelected( int theIndex )
247 int aType = myBathymetryChoice->itemData( theIndex ).toInt();
248 QString aText = myBathymetryChoice->itemText( theIndex );
249 emit setMergeType( aType, aText );
252 void HYDROGUI_CalculationDlg::onSelected( SUIT_DataObject* theObject )
255 HYDROGUI_Zone* aZone = dynamic_cast<HYDROGUI_Zone*>( theObject );
258 doShow = aZone->isMergingNeed();
263 // Fill the merge type combo box
264 bool prevBlock = myBathymetryChoice->blockSignals( true );
265 myCurrentZone = aZone;
266 myBathymetryChoice->clear();
267 myBathymetryChoice->addItem( tr("MERGE_UNKNOWN"), HYDROData_Zone::Merge_UNKNOWN );
268 myBathymetryChoice->addItem( tr("MERGE_ZMIN"), HYDROData_Zone::Merge_ZMIN );
269 myBathymetryChoice->addItem( tr("MERGE_ZMAX"), HYDROData_Zone::Merge_ZMAX );
270 QStringList aList = aZone->getBathymetries();
271 for ( int i = 0; i < aList.length(); i++ )
273 myBathymetryChoice->addItem( aList.at( i ), HYDROData_Zone::Merge_Object );
275 // Select the current choice if any
277 switch ( aZone->getMergeType() )
279 case HYDROData_Zone::Merge_ZMIN:
282 case HYDROData_Zone::Merge_ZMAX:
285 case HYDROData_Zone::Merge_Object:
286 aCurIndex = 3 + aList.indexOf( aZone->text( HYDROGUI_DataObject::BathymetryId ) );
289 aCurIndex = 0; // Select unknown by default
291 myBathymetryChoice->setCurrentIndex( aCurIndex );
292 myBathymetryChoice->blockSignals( prevBlock );
295 myBathymetryChoice->setVisible( doShow );
296 myBatimetryLabel->setVisible( doShow );
299 void HYDROGUI_CalculationDlg::setObjectName( const QString& theName )
301 myObjectName->setText( theName );
304 QString HYDROGUI_CalculationDlg::getObjectName() const
306 return myObjectName->text();
309 void HYDROGUI_CalculationDlg::includeGeomObjects( const QStringList& theObjects )
311 QList<QListWidgetItem*> aFoundItems;
313 QListWidgetItem* anItem;
315 for ( int i = 0, n = theObjects.length(); i < n; ++i )
317 QString anObjName = theObjects.at( i );
318 aFoundItems = myAvailableGeomObjects->findItems( anObjName, Qt::MatchExactly );
319 for ( anIdx = 0; anIdx < aFoundItems.length(); anIdx++ )
321 anItem = aFoundItems.at( anIdx );
322 // Remove this object from available objects list
323 anItem = myAvailableGeomObjects->takeItem( myAvailableGeomObjects->row( anItem ) );
324 // Add the item to the included objects list
325 myGeomObjects->addItem( anItem );
330 void HYDROGUI_CalculationDlg::excludeGeomObjects( const QStringList& theObjects )
332 QList<QListWidgetItem*> aFoundItems;
334 QListWidgetItem* anItem;
336 for ( int i = 0, n = theObjects.length(); i < n; ++i )
338 QString anObjName = theObjects.at( i );
339 aFoundItems = myGeomObjects->findItems( anObjName, Qt::MatchExactly );
340 for ( anIdx = 0; anIdx < aFoundItems.length(); anIdx++ )
342 anItem = aFoundItems.at( anIdx );
343 // Remove this object from included objects list
344 anItem = myGeomObjects->takeItem( myGeomObjects->row( anItem ) );
345 // Add the item to the excluded objects list
346 myAvailableGeomObjects->addItem( anItem );
351 void HYDROGUI_CalculationDlg::setAllGeomObjects( const QStringList& theObjects, const QStringList& theObjectsEntries )
353 myAvailableGeomObjects->clear();
355 for ( int i = 0, n = theObjects.length(); i < n; ++i )
357 QString anObjName = theObjects.at( i );
359 QListWidgetItem* aListItem = new QListWidgetItem( anObjName, myAvailableGeomObjects );
360 aListItem->setFlags( Qt::ItemIsEnabled | Qt::ItemIsSelectable );
361 aListItem->setData( Qt::UserRole, theObjectsEntries.at( i ) );
365 QStringList HYDROGUI_CalculationDlg::getSelectedGeomObjects() const
367 QStringList aResList;
368 QList<QListWidgetItem*> aList = myGeomObjects->selectedItems();
369 for ( int i = 0, n = aList.length(); i < n; ++i )
371 aResList.append( aList.at( i )->text() );
376 QStringList HYDROGUI_CalculationDlg::getSelectedAvailableGeomObjects() const
378 QStringList aResList;
379 QList<QListWidgetItem*> aList = myAvailableGeomObjects->selectedItems();
380 for ( int i = 0, n = aList.length(); i < n; ++i )
382 aResList.append( aList.at( i )->text() );
387 void HYDROGUI_CalculationDlg::setEditedObject( const Handle(HYDROData_CalculationCase) theCase )
389 myEditedObject = theCase;
390 myValidator->setEditedObject( theCase );
392 HYDROGUI_DataObject* anobj = new HYDROGUI_DataObject( 0, NULL, "" );
393 myBrowser->setRoot(anobj);
395 module()->getDataModel()->buildCaseTree( anobj, myEditedObject );
396 myBrowser->updateTree();
397 myBrowser->openLevels();
398 myBrowser->adjustColumnsWidth();
401 HYDROGUI_Zone* HYDROGUI_CalculationDlg::getCurrentZone() const
403 return myCurrentZone;
406 void HYDROGUI_CalculationDlg::refreshZonesBrowser()
408 myBrowser->updateTree();