]> SALOME platform Git repositories - modules/hydro.git/blob - src/HYDROGUI/HYDROGUI_CalculationDlg.cxx
Salome HOME
refs #573, #574: implementation of ordered list widget and a table to define prioriti...
[modules/hydro.git] / src / HYDROGUI / HYDROGUI_CalculationDlg.cxx
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.
6 //
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.
11 //
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
15 //
16 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
17 //
18
19 #include "HYDROGUI_CalculationDlg.h"
20
21 #include "HYDROGUI_ObjSelector.h"
22 #include "HYDROGUI_Tool.h"
23 #include "HYDROGUI_DataBrowser.h"
24 #include "HYDROGUI_DataModel.h"
25 #include "HYDROGUI_ListSelector.h"
26 #include "HYDROGUI_Module.h"
27 #include "HYDROGUI_DataObject.h"
28 #include "HYDROGUI_NameValidator.h"
29 #include "HYDROGUI_Region.h"
30 #include "HYDROGUI_Zone.h"
31 #include "HYDROGUI_OrderedListWidget.h"
32 #include "HYDROGUI_PriorityWidget.h"
33 #include "HYDROGUI_PriorityTableModel.h"
34
35 #include <HYDROData_Document.h>
36 #include <HYDROData_Entity.h>
37
38 #include <CAM_Application.h>
39 #include <LightApp_DataObject.h>
40
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>
46
47 #include <LightApp_Application.h>
48 #include <LightApp_SelectionMgr.h>
49 #include <SUIT_Desktop.h>
50 #include <SUIT_MessageBox.h>
51
52 #include <QButtonGroup>
53 #include <QComboBox>
54 #include <QGroupBox>
55 #include <QLabel>
56 #include <QLayout>
57 #include <QLineEdit>
58 #include <QListWidget>
59 #include <QPicture>
60 #include <QPushButton>
61 #include <QRadioButton>
62 #include <QSplitter>
63 #include <QTableWidget>
64 #include <QToolButton>
65 #include <QTreeView>
66 #include <QWizardPage>
67
68
69 HYDROGUI_CalculationDlg::HYDROGUI_CalculationDlg( HYDROGUI_Module* theModule, const QString& theTitle )
70 : HYDROGUI_Wizard( theModule, theTitle )
71 {
72   addPage( createObjectsPage() );
73   addPage( createGroupsPage() );
74   addPage( createLandCoversPage() );
75   addPage( createZonesPage() );
76   addPage( createLandCoversPartitionPage() );
77 }
78
79 HYDROGUI_CalculationDlg::~HYDROGUI_CalculationDlg()
80 {
81 }
82
83 void HYDROGUI_CalculationDlg::reset()
84 {
85   myObjectName->clear();
86   HYDROGUI_ListModel::Object2VisibleList anObject2VisibleList;
87   myGeomObjects->setObjects(anObject2VisibleList);
88   myLandCovers->setObjects(anObject2VisibleList);
89   myPolylineName->clear();
90   myStricklerTableName->clear();
91   myAvailableGeomObjects->clear();
92   myAvailableLandCovers->clear();
93
94   // Activate the automatic mode
95   setMode( HYDROData_CalculationCase::AUTOMATIC );
96   setLandCoverMode( HYDROData_CalculationCase::AUTOMATIC );
97
98   // Reset the priority widget state
99   QList<Handle(HYDROData_Entity)> anObjects;
100   myPriorityWidget->setObjects( anObjects );
101   myLandCoverPriorityWidget->setObjects( anObjects );
102 }
103
104 QWizardPage* HYDROGUI_CalculationDlg::createObjectsPage() {
105   QWizardPage* aPage = new QWizardPage( mainFrame() );
106   QFrame* aFrame = new QFrame( aPage );
107
108   // Splitter
109   mySplitter = new QSplitter(Qt::Vertical);
110
111   // Top of the page
112   QWidget* aTopContainer = new QWidget;
113    
114   // calculation name
115   myObjectName = new QLineEdit( aPage );
116   myValidator = new HYDROGUI_NameValidator(module(), myObjectName);
117   myObjectName->setValidator( myValidator );
118
119   connect( myValidator, SIGNAL( emptyName() ), SLOT( onEmptyName() ) );
120   connect( myValidator, SIGNAL( alreadyExists( QString ) ), SLOT( onAlreadyExists( QString ) ) );
121
122   // polyline name
123   myPolylineName = new QComboBox( aPage );
124   connect( myPolylineName, SIGNAL( activated( const QString & ) ), 
125     SIGNAL( boundarySelected( const QString & ) ) );
126
127   // names labels
128   QLabel* aNameLabel = new QLabel( tr( "NAME" ), aPage );
129   QLabel* aLimitsLabel = new QLabel( tr( "LIMITS" ), aPage );
130
131   // mode selector (auto/manual)
132   QGroupBox* aModeGroup = new QGroupBox( tr( "MODE" ) );
133
134   QRadioButton* aManualRB = new QRadioButton( tr( "MANUAL" ), mainFrame() );
135   QRadioButton* anAutoRB = new QRadioButton( tr( "AUTO" ), mainFrame() );
136
137   myModeButtons = new QButtonGroup( mainFrame() );
138   myModeButtons->addButton( anAutoRB, HYDROData_CalculationCase::AUTOMATIC );
139   myModeButtons->addButton( aManualRB, HYDROData_CalculationCase::MANUAL );
140   
141   QBoxLayout* aModeSelectorLayout = new QHBoxLayout;
142   aModeSelectorLayout->setMargin( 5 );
143   aModeSelectorLayout->setSpacing( 5 );
144   aModeSelectorLayout->addWidget( anAutoRB );
145   aModeSelectorLayout->addWidget( aManualRB );
146   aModeGroup->setLayout( aModeSelectorLayout );
147
148   // geometry objects
149   QLabel* anObjectsLabel = new QLabel( tr( "CALCULATION_REFERENCE_OBJECTS" ) );
150   myGeomObjects = new HYDROGUI_OrderedListWidget( aPage, 16 );
151   myGeomObjects->setHiddenObjectsShown(true);
152   myGeomObjects->setVisibilityIconShown(false);
153   myGeomObjects->setContentsMargins(QMargins());
154  
155   // included geometry objects
156   QLabel* anIncludedLabel = new QLabel( tr( "INCLUDED_OBJECTS" ) );
157   myAvailableGeomObjects = new QListWidget( aPage );
158   myAvailableGeomObjects->setSelectionMode( QListWidget::ExtendedSelection );
159   myAvailableGeomObjects->setEditTriggers( QListWidget::NoEditTriggers );
160   myAvailableGeomObjects->setViewMode( QListWidget::ListMode );
161   myAvailableGeomObjects->setSortingEnabled( true );
162
163   // buttons
164   QFrame* aBtnsFrame = new QFrame;
165   QVBoxLayout* aBtnsLayout = new QVBoxLayout( aBtnsFrame );
166   aBtnsLayout->setMargin( 5 );
167   aBtnsLayout->setSpacing( 5 );
168   aBtnsFrame->setLayout( aBtnsLayout );
169   QPushButton* anAddBtn = new QPushButton( tr("INCLUDE"), aBtnsFrame );
170   QPushButton* aRemoveBtn = new QPushButton( tr("EXCLUDE"), aBtnsFrame );
171
172   // fill the butons frame with two buttons
173   aBtnsLayout->addWidget( anAddBtn );
174   aBtnsLayout->addWidget( aRemoveBtn );
175   aBtnsLayout->addStretch( 1 );
176   
177   // top of the page layout
178   
179   // objects frame
180   QFrame* anObjectsFrame = new QFrame( aPage );
181   anObjectsFrame->setFrameStyle( QFrame::Panel | QFrame::Raised );
182   QGridLayout* anObjsLayout = new QGridLayout( anObjectsFrame );
183   anObjsLayout->setMargin( 5 );
184   anObjsLayout->setSpacing( 5 );
185   anObjectsFrame->setLayout( anObjsLayout );
186   
187   // fill the objects frame with two lists, two labels and with buttons frame
188   anObjsLayout->addWidget( anObjectsLabel, 0, 0, Qt::AlignHCenter );
189   anObjsLayout->addWidget( anIncludedLabel, 0, 2, Qt::AlignHCenter );
190   anObjsLayout->addWidget( myAvailableGeomObjects, 1, 0 );
191   anObjsLayout->addWidget( aBtnsFrame, 1, 1, Qt::AlignHCenter );
192   anObjsLayout->addWidget( myGeomObjects, 1, 2 );
193   
194   // fill the top of the page
195   QGridLayout* aTopLayout = new QGridLayout;
196   aTopLayout->setMargin( 5 );
197   aTopLayout->setSpacing( 5 );
198   aTopLayout->setVerticalSpacing( 10 );
199   aTopLayout->addWidget( aNameLabel,     0, 0, Qt::AlignHCenter );
200   aTopLayout->addWidget( myObjectName,   0, 1 );
201   aTopLayout->addWidget( aLimitsLabel,   1, 0, Qt::AlignHCenter );
202   aTopLayout->addWidget( myPolylineName, 1, 1 );
203   aTopLayout->addWidget( aModeGroup, 2, 0, 1, 2 );
204   aTopLayout->addWidget( anObjectsFrame, 3, 0, 1, 2 );
205
206   aTopContainer->setLayout( aTopLayout );
207
208   // add the top of the page to the splitter
209   mySplitter->insertWidget(0, aTopContainer);
210   mySplitter->setStretchFactor(0, 2);
211
212   // Bottom of the page
213   myPriorityWidget = new HYDROGUI_PriorityWidget( mainFrame() );
214
215   QGroupBox* aPriorityGroup = new QGroupBox( tr( "PRIORITY" ) );
216   QBoxLayout* aPriorityLayout = new QHBoxLayout;
217   aPriorityLayout->setMargin( 5 );
218   aPriorityLayout->setSpacing( 5 );
219   aPriorityLayout->addWidget( myPriorityWidget );
220   aPriorityGroup->setLayout( aPriorityLayout );
221
222   // add the bottom of the page to the splitter
223   mySplitter->insertWidget(1, aPriorityGroup);
224   mySplitter->setStretchFactor(1, 1);
225
226   // Page layout
227   QVBoxLayout* aPageLayout = new QVBoxLayout;
228   aPageLayout->setMargin( 5 );
229   aPageLayout->setSpacing( 5 );
230   aPageLayout->addWidget( mySplitter );
231
232   aPage->setLayout( aPageLayout );
233
234   // Create selector
235   if ( module() ) {
236     HYDROGUI_ListSelector* aListSelector = 
237       new HYDROGUI_ListSelector( myGeomObjects, module()->getApp()->selectionMgr() );
238     aListSelector->setAutoBlock( true );
239   }
240
241   // Connections
242   connect( myModeButtons, SIGNAL( buttonClicked( int ) ), SIGNAL( changeMode( int ) ) );
243   connect( anAddBtn, SIGNAL( clicked() ), SIGNAL( addObjects() ) );
244   connect( aRemoveBtn, SIGNAL( clicked() ), SIGNAL( removeObjects() ) );
245
246   connect( myGeomObjects, SIGNAL( orderChanged() ), SLOT( onOrderChanged() ) );
247
248   return aPage;
249 }
250
251 QWizardPage* HYDROGUI_CalculationDlg::createGroupsPage() {
252   QWizardPage* aPage = new QWizardPage( mainFrame() );
253   QFrame* aFrame = new QFrame( aPage );
254
255   myGroups = new QListWidget( aPage );
256   myGroups->setSelectionMode( QListWidget::ExtendedSelection );
257   myGroups->setEditTriggers( QListWidget::NoEditTriggers );
258   myGroups->setViewMode( QListWidget::ListMode );
259   myGroups->setSortingEnabled( true );
260
261   myAvailableGroups = new QListWidget( aPage );
262   myAvailableGroups->setSelectionMode( QListWidget::ExtendedSelection );
263   myAvailableGroups->setEditTriggers( QListWidget::NoEditTriggers );
264   myAvailableGroups->setViewMode( QListWidget::ListMode );
265   myAvailableGroups->setSortingEnabled( true );
266
267   connect( myGroups, SIGNAL( itemSelectionChanged() ), 
268     SIGNAL( groupsSelected() ) );
269
270   QFrame* aGroupsFrame = new QFrame( aPage );
271   QGridLayout* aGroupsLayout = new QGridLayout( aGroupsFrame );
272   aGroupsLayout->setMargin( 5 );
273   aGroupsLayout->setSpacing( 5 );
274   aGroupsFrame->setLayout( aGroupsLayout );
275
276   QFrame* aBtnsFrame = new QFrame( aGroupsFrame );
277   QVBoxLayout* aBtnsLayout = new QVBoxLayout( aBtnsFrame );
278   aBtnsLayout->setMargin( 5 );
279   aBtnsLayout->setSpacing( 5 );
280   aBtnsFrame->setLayout( aBtnsLayout );
281   QPushButton* anAddBtn = new QPushButton( tr("INCLUDE"), aBtnsFrame );
282   QPushButton* aRemoveBtn = new QPushButton( tr("EXCLUDE"), aBtnsFrame );
283
284   // Fill the butons frame with two buttons
285   aBtnsLayout->addWidget( anAddBtn );
286   aBtnsLayout->addWidget( aRemoveBtn );
287   aBtnsLayout->addStretch( 1 );
288
289   QLabel* anIncludedLabel = new QLabel( tr( "INCLUDED_GROUPS" ), aGroupsFrame );
290   QLabel* anAvailableLabel = new QLabel( tr( "AVAILABLE_GROUPS" ), aGroupsFrame );
291
292   // Fill the objects frame with two lists, two labels and with buttons frame
293   aGroupsLayout->addWidget( anAvailableLabel, 0, 0, Qt::AlignHCenter );
294   aGroupsLayout->addWidget( anIncludedLabel, 0, 2, Qt::AlignHCenter );
295   aGroupsLayout->addWidget( myAvailableGroups, 1, 0 );
296   aGroupsLayout->addWidget( aBtnsFrame, 1, 1, Qt::AlignHCenter );
297   aGroupsLayout->addWidget( myGroups, 1, 2 );
298
299   // Fill the page
300   QGridLayout* aPageLayout = new QGridLayout( aPage );
301   aPageLayout->setMargin( 5 );
302   aPageLayout->setSpacing( 5 );
303   aPageLayout->setVerticalSpacing( 10 );
304   aPageLayout->addWidget( aGroupsFrame, 0, 0 );
305
306   aPage->setLayout( aPageLayout );
307
308   connect( anAddBtn, SIGNAL( clicked() ), SIGNAL( addGroups() ) );
309   connect( aRemoveBtn, SIGNAL( clicked() ), SIGNAL( removeGroups() ) );
310
311   return aPage;
312 }
313
314 QWizardPage* HYDROGUI_CalculationDlg::createLandCoversPage() {
315   QWizardPage* aPage = new QWizardPage( mainFrame() );
316   QFrame* aFrame = new QFrame( aPage );
317
318   // Splitter
319   myLandCoverSplitter = new QSplitter(Qt::Vertical);
320
321   // Top of the page
322   QWidget* aTopContainer = new QWidget;
323    
324   // Combo-box to choose Strickler table name
325   QLabel* aStricklerTableLabel = new QLabel( tr( "STRICKLER_TABLE" ), aPage );
326   myStricklerTableName = new QComboBox( aPage );
327   myStricklerTableName->setSizePolicy( QSizePolicy( QSizePolicy::Expanding, QSizePolicy::Fixed ) );
328   connect( myStricklerTableName, SIGNAL( activated( const QString & ) ), 
329                                  SIGNAL( StricklerTableSelected( const QString & ) ) );
330
331   // Mode selector (auto/manual)
332   QGroupBox* aModeGroup = new QGroupBox( tr( "MODE" ) );
333
334   QRadioButton* aManualRB = new QRadioButton( tr( "MANUAL" ), mainFrame() );
335   QRadioButton* anAutoRB = new QRadioButton( tr( "AUTO" ), mainFrame() );
336
337   myLandCoverModeButtons = new QButtonGroup( mainFrame() );
338   myLandCoverModeButtons->addButton( anAutoRB, HYDROData_CalculationCase::AUTOMATIC );
339   myLandCoverModeButtons->addButton( aManualRB, HYDROData_CalculationCase::MANUAL );
340   
341   QBoxLayout* aModeSelectorLayout = new QHBoxLayout;
342   aModeSelectorLayout->setMargin( 5 );
343   aModeSelectorLayout->setSpacing( 5 );
344   aModeSelectorLayout->addWidget( anAutoRB );
345   aModeSelectorLayout->addWidget( aManualRB );
346   aModeGroup->setLayout( aModeSelectorLayout );
347
348   // Available land covers
349   QLabel* aLandCoversLabel = new QLabel( tr( "CALCULATION_REFERENCE_LAND_COVERS" ) );
350   myAvailableLandCovers = new QListWidget( aPage );
351   myAvailableLandCovers->setSelectionMode( QListWidget::ExtendedSelection );
352   myAvailableLandCovers->setEditTriggers( QListWidget::NoEditTriggers );
353   myAvailableLandCovers->setViewMode( QListWidget::ListMode );
354   myAvailableLandCovers->setSortingEnabled( true );
355  
356   // Included land covers
357   QLabel* anIncludedLabel = new QLabel( tr( "INCLUDED_LAND_COVERS" ) );
358   myLandCovers = new HYDROGUI_OrderedListWidget( aPage, 16 );
359   myLandCovers->setHiddenObjectsShown(true);
360   myLandCovers->setVisibilityIconShown(false);
361   myLandCovers->setContentsMargins(QMargins()); 
362
363   // Include/Exclude buttons
364   QFrame* aBtnsFrame = new QFrame;
365   QVBoxLayout* aBtnsLayout = new QVBoxLayout( aBtnsFrame );
366   aBtnsLayout->setMargin( 5 );
367   aBtnsLayout->setSpacing( 5 );
368   aBtnsFrame->setLayout( aBtnsLayout );
369   QPushButton* anAddBtn = new QPushButton( tr("INCLUDE"), aBtnsFrame );
370   QPushButton* aRemoveBtn = new QPushButton( tr("EXCLUDE"), aBtnsFrame );
371
372   // Fill the butons frame with two buttons
373   aBtnsLayout->addWidget( anAddBtn );
374   aBtnsLayout->addWidget( aRemoveBtn );
375   aBtnsLayout->addStretch( 1 );
376   
377   // Land covers frame
378   QFrame* aLandCoversFrame = new QFrame( aPage );
379   aLandCoversFrame->setFrameStyle( QFrame::Panel | QFrame::Raised );
380   QGridLayout* aLandCoversLayout = new QGridLayout( aLandCoversFrame );
381   aLandCoversLayout->setMargin( 5 );
382   aLandCoversLayout->setSpacing( 5 );
383   aLandCoversFrame->setLayout( aLandCoversLayout );
384   
385   // Fill the land covers frame with two lists, two labels and with buttons frame
386   aLandCoversLayout->addWidget( aLandCoversLabel, 0, 0, Qt::AlignHCenter );
387   aLandCoversLayout->addWidget( anIncludedLabel, 0, 2, Qt::AlignHCenter );
388   aLandCoversLayout->addWidget( myAvailableLandCovers, 1, 0 );
389   aLandCoversLayout->addWidget( aBtnsFrame, 1, 1, Qt::AlignHCenter );
390   aLandCoversLayout->addWidget( myLandCovers, 1, 2 );
391   
392   // Fill the top layout of the page
393   QGridLayout* aTopLayout = new QGridLayout;
394   aTopLayout->setMargin( 5 );
395   aTopLayout->setSpacing( 5 );
396   aTopLayout->setVerticalSpacing( 10 );
397   aTopLayout->addWidget( aStricklerTableLabel, 0, 0, Qt::AlignHCenter );
398   aTopLayout->addWidget( myStricklerTableName, 0, 1 );
399   aTopLayout->addWidget( aModeGroup, 1, 0, 1, 2 );
400   aTopLayout->addWidget( aLandCoversFrame, 2, 0, 1, 2 );
401
402   aTopContainer->setLayout( aTopLayout );
403
404   // Add the top of the page to the splitter
405   myLandCoverSplitter->insertWidget(0, aTopContainer);
406   myLandCoverSplitter->setStretchFactor(0, 2);
407
408   // Bottom of the page
409   myLandCoverPriorityWidget = new HYDROGUI_PriorityWidget( mainFrame() );
410   HYDROGUI_PriorityTableModel* aModel = 
411     dynamic_cast<HYDROGUI_PriorityTableModel*>( myLandCoverPriorityWidget->getTable()->model() );
412   if ( aModel )
413     aModel->setColumnCount( 3 );
414
415   QGroupBox* aPriorityGroup = new QGroupBox( tr( "PRIORITY" ) );
416   QBoxLayout* aPriorityLayout = new QHBoxLayout;
417   aPriorityLayout->setMargin( 5 );
418   aPriorityLayout->setSpacing( 5 );
419   aPriorityLayout->addWidget( myLandCoverPriorityWidget );
420   aPriorityGroup->setLayout( aPriorityLayout );
421
422   // Add the bottom of the page to the splitter
423   myLandCoverSplitter->insertWidget(1, aPriorityGroup);
424   myLandCoverSplitter->setStretchFactor(1, 1);
425
426   // Page layout
427   QVBoxLayout* aPageLayout = new QVBoxLayout;
428   aPageLayout->setMargin( 5 );
429   aPageLayout->setSpacing( 5 );
430   aPageLayout->addWidget( myLandCoverSplitter );
431
432   aPage->setLayout( aPageLayout );
433
434   // Create selector
435   if ( module() ) {
436     HYDROGUI_ListSelector* aListSelector = 
437       new HYDROGUI_ListSelector( myLandCovers, module()->getApp()->selectionMgr() );
438     aListSelector->setAutoBlock( true );
439   }
440
441   // Connections
442   connect( myLandCoverModeButtons, SIGNAL( buttonClicked( int ) ), SIGNAL( changeLandCoverMode( int ) ) );
443   connect( anAddBtn, SIGNAL( clicked() ), SIGNAL( addLandCovers() ) );
444   connect( aRemoveBtn, SIGNAL( clicked() ), SIGNAL( removeLandCovers() ) );
445
446   connect( myLandCovers, SIGNAL( orderLandCoverChanged() ), SLOT( onOrderLandCoverChanged() ) );
447
448   return aPage;
449 }
450
451 QWizardPage* HYDROGUI_CalculationDlg::createZonesPage() {
452   QWizardPage* aPage = new QWizardPage( mainFrame() );
453   QFrame* aFrame = new QFrame( aPage );
454
455   QGridLayout* aLayout = new QGridLayout( aPage );
456   
457   myBrowser = new HYDROGUI_DataBrowser( module(), NULL, aPage );
458   myBrowser->setAutoOpenLevel( 3 );
459   aLayout->setMargin( 5 );
460   aLayout->setSpacing( 5 );
461
462   aLayout->addWidget( myBrowser, 0, 0, 1, 2 );
463
464   myBathymetryLabel = new QLabel( tr( "BATHYMETRY" ), aFrame );
465   myBathymetryChoice = new QComboBox( aFrame );
466
467   myBathymetryChoice->setVisible( false );
468   myBathymetryLabel->setVisible( false );
469
470   aLayout->addWidget( myBathymetryLabel, 1, 0 );
471   aLayout->addWidget( myBathymetryChoice, 1, 1 );
472
473   aPage->setLayout( aLayout );
474
475   connect( myBrowser, SIGNAL( dataChanged() ), SLOT( onDataChanged() ) );
476   connect( myBrowser, SIGNAL( clicked( SUIT_DataObject* ) ), SIGNAL( clickedInZonesBrowser( SUIT_DataObject* ) ) );
477   connect( myBrowser, SIGNAL( clicked( SUIT_DataObject* ) ), SLOT( onSelected( SUIT_DataObject* ) ) );
478   connect( myBathymetryChoice, SIGNAL( activated( int ) ), SLOT( onMergeTypeSelected( int ) ) );
479   connect( myBrowser, 
480       SIGNAL( dropped( const QList<SUIT_DataObject*>&, SUIT_DataObject*, int, Qt::DropAction ) ),
481       SLOT( onZonesDropped( const QList<SUIT_DataObject*>&, SUIT_DataObject*, int, Qt::DropAction ) ) );
482   connect( myBrowser, SIGNAL( newRegion() ), this, SLOT( OnNewRegion() ) );
483
484   return aPage;
485 }
486
487 QWizardPage* HYDROGUI_CalculationDlg::createLandCoversPartitionPage() {
488   QWizardPage* aPage = new QWizardPage( mainFrame() );
489   QFrame* aFrame = new QFrame( aPage );
490
491   QGridLayout* aLayout = new QGridLayout( aPage );
492   
493   myLandCoverBrowser = new HYDROGUI_DataBrowser( module(), NULL, aPage );
494   myLandCoverBrowser->setAutoOpenLevel( 3 );
495   aLayout->setMargin( 5 );
496   aLayout->setSpacing( 5 );
497
498   aLayout->addWidget( myLandCoverBrowser, 0, 0, 1, 2 );
499
500   aPage->setLayout( aLayout );
501
502   // Connections
503   //...
504
505   return aPage;
506 }
507
508
509 bool HYDROGUI_CalculationDlg::acceptCurrent() const
510 {
511   QString anErrorMsg;
512
513   if ( false /*myGeomObjects->count() == 0*/ )
514   {
515     anErrorMsg = tr( "EMPTY_GEOMETRY_OBJECTS" );
516   }
517
518   if ( !anErrorMsg.isEmpty() )
519   {
520     anErrorMsg += "\n" + tr( "INPUT_VALID_DATA" );
521     
522     QString aTitle = tr( "INSUFFICIENT_INPUT_DATA" );
523     SUIT_MessageBox::critical( module()->getApp()->desktop(), aTitle, anErrorMsg );
524   }
525
526   return anErrorMsg.isEmpty();
527 }
528
529 void HYDROGUI_CalculationDlg::onEmptyName()
530 {
531   QString aTitle = tr( "INSUFFICIENT_INPUT_DATA" );
532   QString aMessage = tr( "INCORRECT_OBJECT_NAME" ) + "\n" + tr( "INPUT_VALID_DATA" );
533   SUIT_MessageBox::critical( module()->getApp()->desktop(), aTitle, aMessage );
534 }
535
536 void HYDROGUI_CalculationDlg::onAlreadyExists( QString theName )
537 {
538   QString aTitle = tr( "INSUFFICIENT_INPUT_DATA" );
539   QString aMessage = QObject::tr( "OBJECT_EXISTS_IN_DOCUMENT" ).arg( theName ) +
540                      "\n" + tr( "INPUT_VALID_DATA" );
541   SUIT_MessageBox::critical( module()->getApp()->desktop(), aTitle, aMessage );
542 }
543
544 void HYDROGUI_CalculationDlg::onZonesDropped( const QList<SUIT_DataObject*>& theList, 
545     SUIT_DataObject* theTargetParent, int theTargetRow, Qt::DropAction theDropAction )
546 {
547   QList<SUIT_DataObject*> aZonesList;
548   HYDROGUI_Zone* aZone;
549   // Get a list of dropped zones
550   for ( int i = 0; i < theList.length(); i++ )
551   {
552     aZone = dynamic_cast<HYDROGUI_Zone*>( theList.at( i ) );
553     if ( aZone )
554     {
555       aZonesList.append( aZone );
556     }
557   }
558   if ( aZonesList.length() > 0 )
559   {
560     // Get the target region
561     HYDROGUI_NamedObject* aRegionsRoot = dynamic_cast<HYDROGUI_NamedObject*>(theTargetParent);
562     if ( aRegionsRoot )
563     {
564       // Create a new region
565       emit createRegion( aZonesList );
566     }
567     else
568     {
569       HYDROGUI_Region* aRegion = dynamic_cast<HYDROGUI_Region*>(theTargetParent);
570       if ( aRegion )
571       {
572         emit moveZones( theTargetParent, aZonesList );
573       }
574     }
575   }
576 }
577
578 void HYDROGUI_CalculationDlg::OnNewRegion()
579 {
580   emit createRegion( myBrowser->getSelected() );
581 }
582
583 void HYDROGUI_CalculationDlg::onMergeTypeSelected( int theIndex )
584 {
585   int aType = myBathymetryChoice->itemData( theIndex ).toInt();
586   QString aText = myBathymetryChoice->itemText( theIndex );
587   emit setMergeType( aType, aText );
588 }
589
590 void HYDROGUI_CalculationDlg::onSelected( SUIT_DataObject* theObject )
591 {
592   bool doShow = false;
593   HYDROGUI_Zone* aZone = dynamic_cast<HYDROGUI_Zone*>( theObject );
594   if ( aZone )
595   {
596     doShow = aZone->isMergingNeed();
597   }
598
599   if ( doShow )
600   {
601     // Fill the merge type combo box
602     bool prevBlock = myBathymetryChoice->blockSignals( true );
603     myCurrentZone = aZone;
604     myBathymetryChoice->clear();
605     myBathymetryChoice->addItem( tr("MERGE_UNKNOWN"), HYDROData_Zone::Merge_UNKNOWN );
606     myBathymetryChoice->addItem( tr("MERGE_ZMIN"), HYDROData_Zone::Merge_ZMIN );
607     myBathymetryChoice->addItem( tr("MERGE_ZMAX"), HYDROData_Zone::Merge_ZMAX );
608     QStringList aList = aZone->getAltitudes();
609     for ( int i = 0; i < aList.length(); i++ )
610     {
611       myBathymetryChoice->addItem( aList.at( i ), HYDROData_Zone::Merge_Object );
612     }
613     // Select the current choice if any
614     int aCurIndex = 0;
615     switch ( aZone->getMergeType() )
616     {
617       case HYDROData_Zone::Merge_ZMIN:
618         aCurIndex = 1;
619         break;
620       case HYDROData_Zone::Merge_ZMAX:
621         aCurIndex = 2;
622         break;
623       case HYDROData_Zone::Merge_Object:
624         aCurIndex = 3 + aList.indexOf( aZone->text( HYDROGUI_DataObject::AltitudeObjId ) );
625         break;
626       default:
627         aCurIndex = 0; // Select unknown by default
628     }
629     myBathymetryChoice->setCurrentIndex( aCurIndex );
630     myBathymetryChoice->blockSignals( prevBlock );
631   }
632
633   myBathymetryChoice->setVisible( doShow );
634   myBathymetryChoice->setEnabled( getMode() == HYDROData_CalculationCase::MANUAL );
635   myBathymetryLabel->setVisible( doShow );
636 }
637
638 void HYDROGUI_CalculationDlg::setObjectName( const QString& theName )
639 {
640   myObjectName->setText( theName );
641 }
642
643 QString HYDROGUI_CalculationDlg::getObjectName() const
644 {
645   return myObjectName->text();
646 }
647
648 void moveItems( QListWidget* theSource, QListWidget* theDest, const QStringList& theObjects )
649 {
650   QList<QListWidgetItem*> aFoundItems;
651   int anIdx;
652   QListWidgetItem* anItem;
653
654   for ( int i = 0, n = theObjects.length(); i < n; ++i )
655   {
656     QString anObjName = theObjects.at( i );
657     aFoundItems = theSource->findItems( anObjName, Qt::MatchExactly );
658     for ( anIdx = 0; anIdx < aFoundItems.length(); anIdx++ )
659     {
660       anItem = aFoundItems.at( anIdx );
661       // Remove this object from available objects list
662       anItem = theSource->takeItem( theSource->row( anItem ) );
663       // Add the item to the included objects list
664       theDest->addItem( anItem );
665     }
666   }
667 }
668
669 void HYDROGUI_CalculationDlg::includeGeomObjects( const QStringList& theObjects )
670 {
671   HYDROGUI_ListModel::Object2VisibleList anObjectsToInclude;
672   QList<QListWidgetItem*> aFoundItems;
673   foreach ( const QString& anObjName, theObjects ) {
674     // Hide the object in the available objects list
675     aFoundItems = myAvailableGeomObjects->findItems( anObjName, Qt::MatchExactly );
676     foreach ( QListWidgetItem* anItem, aFoundItems ) {
677       anItem->setHidden( true );
678     }
679
680     // Add the object to the list of included objects
681     Handle(HYDROData_Entity) anObject = 
682       HYDROGUI_Tool::FindObjectByName( module(), anObjName );
683     myGeomObjects->addObject( HYDROGUI_ListModel::Object2Visible( anObject, true ) );
684   }
685
686   myPriorityWidget->setObjects( getGeometryObjects() );
687 }
688
689 void HYDROGUI_CalculationDlg::excludeGeomObjects( const QStringList& theObjects )
690 {
691   QList<QListWidgetItem*> aFoundItems;
692   foreach ( const QString& anObjName, theObjects ) {
693     // Set visible the object in the available objects list
694     aFoundItems = myAvailableGeomObjects->findItems( anObjName, Qt::MatchExactly );
695     foreach ( QListWidgetItem* anItem, aFoundItems ) {
696       anItem->setHidden( false );
697     }
698
699     // Remove the object from the list of included objects
700     myGeomObjects->removeObjectByName( anObjName );
701   }
702
703   myPriorityWidget->setObjects( getGeometryObjects() );
704 }
705
706 void HYDROGUI_CalculationDlg::setBoundary( const QString& theObjName )
707 {
708   bool isBlocked = myPolylineName->blockSignals( true );
709   myPolylineName->setCurrentIndex( myPolylineName->findText( theObjName ) );
710   myPolylineName->blockSignals( isBlocked );
711 }
712
713 void HYDROGUI_CalculationDlg::setPolylineNames( const QStringList& theObjects, const QStringList& theObjectsEntries )
714 {
715   myPolylineName->clear();
716   myPolylineName->addItem( "", "" ); // No boundary item
717
718   for ( int i = 0, n = theObjects.length(); i < n; ++i )
719   {
720     myPolylineName->addItem( theObjects.at( i ), theObjectsEntries.at( i ) );
721   }
722 }
723
724 void HYDROGUI_CalculationDlg::setStricklerTableNames( const QStringList& theObjects, const QStringList& theObjectsEntries )
725 {
726   myStricklerTableName->clear();
727
728   for ( int i = 0, n = theObjects.length(); i < n; ++i )
729   {
730     myStricklerTableName->addItem( theObjects.at( i ), theObjectsEntries.at( i ) );
731   }
732 }
733
734 void HYDROGUI_CalculationDlg::setAllGeomObjects( const QStringList& theObjects, const QStringList& theObjectsEntries )
735 {
736   myAvailableGeomObjects->clear();
737
738   for ( int i = 0, n = theObjects.length(); i < n; ++i )
739   {
740     QString anObjName = theObjects.at( i );
741
742     QListWidgetItem* aListItem = new QListWidgetItem( anObjName, myAvailableGeomObjects );
743     aListItem->setFlags( Qt::ItemIsEnabled | Qt::ItemIsSelectable );
744     aListItem->setData( Qt::UserRole, theObjectsEntries.at( i ) );
745   }
746 }
747
748 void HYDROGUI_CalculationDlg::setAllLandCovers( const QStringList& theObjects, const QStringList& theObjectsEntries )
749 {
750   myAvailableLandCovers->clear();
751
752   for ( int i = 0, n = theObjects.length(); i < n; ++i )
753   {
754     QString anObjName = theObjects.at( i );
755
756     QListWidgetItem* aListItem = new QListWidgetItem( anObjName, myAvailableLandCovers );
757     aListItem->setFlags( Qt::ItemIsEnabled | Qt::ItemIsSelectable );
758     aListItem->setData( Qt::UserRole, theObjectsEntries.at( i ) );
759   }
760 }
761
762 QStringList getSelected( QListWidget* theWidget )
763 {
764   QStringList aResList;
765   QList<QListWidgetItem*> aList = theWidget->selectedItems();
766   for ( int i = 0, n = aList.length(); i < n; ++i )
767   {
768     aResList.append( aList.at( i )->text() );
769   }
770   return aResList;
771 }
772
773 QStringList HYDROGUI_CalculationDlg::getSelectedGeomObjects() const
774 {
775   return myGeomObjects->getSelectedNames();
776 }
777
778 QStringList HYDROGUI_CalculationDlg::getSelectedLandCovers() const
779 {
780   return myLandCovers->getSelectedNames();
781 }
782
783 QStringList HYDROGUI_CalculationDlg::getAllGeomObjects() const
784 {
785   return myGeomObjects->getAllNames();
786 }
787
788 QStringList HYDROGUI_CalculationDlg::getAllLandCovers() const
789 {
790   return myLandCovers->getAllNames();
791 }
792
793 QStringList HYDROGUI_CalculationDlg::getSelectedAvailableGeomObjects() const
794 {
795   return getSelected( myAvailableGeomObjects );
796 }
797
798 QStringList HYDROGUI_CalculationDlg::getSelectedAvailableLandCovers() const
799 {
800   return getSelected( myAvailableLandCovers );
801 }
802
803 void HYDROGUI_CalculationDlg::setEditedObject( const Handle(HYDROData_CalculationCase) theCase )
804 {
805   myEditedObject = theCase;
806   myValidator->setEditedObject( theCase );
807
808   // Build the calculation case subtree
809   module()->getDataModel()->buildCaseTree( myBrowser->root(), myEditedObject );
810
811   myBrowser->updateTree();
812   myBrowser->openLevels();
813   myBrowser->adjustColumnsWidth();
814   myBrowser->setAutoUpdate( true );
815   myBrowser->setUpdateModified( true );
816 }
817
818 HYDROGUI_Zone* HYDROGUI_CalculationDlg::getCurrentZone() const
819 {
820   return myCurrentZone;
821 }
822
823 void HYDROGUI_CalculationDlg::refreshZonesBrowser()
824 {
825   SUIT_DataObject* aRoot = myBrowser->root();
826   module()->getDataModel()->updateObjectTree( myEditedObject );
827   module()->getDataModel()->buildCaseTree( aRoot, myEditedObject );
828   myBrowser->updateTree( aRoot );
829 }
830
831 void HYDROGUI_CalculationDlg::onDataChanged()
832 {
833   SUIT_DataObject* aRoot = myBrowser->root();
834   module()->getDataModel()->buildCaseTree( aRoot, myEditedObject );
835   myBrowser->updateTree( aRoot );
836 }
837
838 void HYDROGUI_CalculationDlg::setAvailableGroups( const QStringList& theGroups )
839 {
840   myAvailableGroups->clear();
841   myGroups->clear();
842   foreach( QString aGroup, theGroups )
843     myAvailableGroups->addItem( aGroup );
844 }
845
846 QStringList HYDROGUI_CalculationDlg::getSelectedGroups() const
847 {
848   return getSelected( myGroups );
849 }
850
851 QStringList HYDROGUI_CalculationDlg::getSelectedAvailableGroups() const
852 {
853   return getSelected( myAvailableGroups );
854 }
855
856 void HYDROGUI_CalculationDlg::includeGroups( const QStringList& theObjects )
857 {
858   moveItems( myAvailableGroups, myGroups, theObjects );
859 }
860
861 void HYDROGUI_CalculationDlg::excludeGroups( const QStringList& theObjects )
862 {
863   moveItems( myGroups, myAvailableGroups, theObjects );
864 }
865
866 /**
867   Get creation mode.
868   @param theMode the mode
869 */
870 int HYDROGUI_CalculationDlg::getMode() const
871 {
872   return myModeButtons->checkedId();
873 }
874
875 /**
876   Set creation mode.
877   @param theMode the mode
878 */
879 void HYDROGUI_CalculationDlg::setMode( int theMode )
880 {
881   bool isBlocked = myModeButtons->blockSignals( true );
882   myModeButtons->button( theMode )->setChecked( true );
883   myModeButtons->blockSignals( isBlocked );
884
885   bool isAuto = ( theMode == HYDROData_CalculationCase::AUTOMATIC );
886
887   myGeomObjects->setOrderingEnabled( isAuto );
888   QWidget* aWidget = mySplitter->widget( 1 );
889   if ( aWidget ) {
890     aWidget->setVisible( isAuto );
891   }
892 }
893
894 /**
895   Get creation mode for land covers panel.
896   @param theMode the mode
897 */
898 int HYDROGUI_CalculationDlg::getLandCoverMode() const
899 {
900   return myLandCoverModeButtons->checkedId();
901 }
902
903 /**
904   Set creation mode for land cover panel.
905   @param theMode the mode
906 */
907 void HYDROGUI_CalculationDlg::setLandCoverMode( int theMode )
908 {
909   bool isBlocked = myLandCoverModeButtons->blockSignals( true );
910   myLandCoverModeButtons->button( theMode )->setChecked( true );
911   myLandCoverModeButtons->blockSignals( isBlocked );
912
913   bool isAuto = ( theMode == HYDROData_CalculationCase::AUTOMATIC );
914
915   myLandCovers->setOrderingEnabled( isAuto );
916   QWidget* aWidget = myLandCoverSplitter->widget( 1 );
917   if ( aWidget ) {
918     aWidget->setVisible( isAuto );
919   }
920 }
921
922 /**
923   Enable/disable zones drag'n'drop and renaming.
924   @param theIsEnabled if true - zones drag'n'drop and renaming will be enabled
925 */
926 void HYDROGUI_CalculationDlg::setEditZonesEnabled( const bool theIsEnabled )
927 {
928   myBrowser->setReadOnly( !theIsEnabled );
929 }
930
931 /**
932   Enable/disable land covers drag'n'drop and renaming.
933   @param theIsEnabled if true - land covers drag'n'drop and renaming will be enabled
934 */
935 void HYDROGUI_CalculationDlg::setEditLandCoversEnabled( const bool theIsEnabled )
936 {
937   myLandCoverBrowser->setReadOnly( !theIsEnabled );
938 }
939
940 /**
941   Get included geometry objects.
942   @return the list of geometry objects
943  */
944 QList<Handle(HYDROData_Entity)> HYDROGUI_CalculationDlg::getGeometryObjects()
945 {
946   QList<Handle(HYDROData_Entity)> anEntities = myGeomObjects->getObjects();
947   QList<Handle(HYDROData_Entity)> anObjects;
948
949   foreach ( Handle(HYDROData_Entity) anEntity, anEntities ) {
950     Handle(HYDROData_Object) anObj = Handle(HYDROData_Object)::DownCast( anEntity );
951     if ( anObj.IsNull() ) {
952       continue;
953     }
954
955     anObjects << anObj;
956   }
957
958   return anObjects;
959 }
960
961 /**
962   Get included land covers.
963   @return the list of land covers
964  */
965 QList<Handle(HYDROData_Entity)> HYDROGUI_CalculationDlg::getLandCovers()
966 {
967   QList<Handle(HYDROData_Entity)> anEntities = myLandCovers->getObjects();
968   QList<Handle(HYDROData_Entity)> aLandCovers;
969
970   foreach ( Handle(HYDROData_Entity) anEntity, anEntities ) {
971     Handle(HYDROData_LandCover) aLandCover = Handle(HYDROData_LandCover)::DownCast( anEntity );
972     if ( aLandCover.IsNull() ) {
973       continue;
974     }
975
976     aLandCovers << aLandCover;
977   }
978
979   return aLandCovers;
980 }
981
982 /**
983   Get rules.
984   @return the list of rules
985  */
986 HYDROData_ListOfRules HYDROGUI_CalculationDlg::getRules() const
987 {
988   return myPriorityWidget->getRules();
989 }
990
991 /**
992   Set rules.
993   @param theRules the list of rules
994  */
995 void  HYDROGUI_CalculationDlg::setRules( const HYDROData_ListOfRules& theRules ) const
996 {
997   myPriorityWidget->setRules( theRules );
998 }
999
1000 /**
1001   Get rules defined for land covers.
1002   @return the list of rules
1003  */
1004 HYDROData_ListOfRules HYDROGUI_CalculationDlg::getLandCoverRules() const
1005 {
1006   return myLandCoverPriorityWidget->getRules();
1007 }
1008
1009 /**
1010   Set rules for land covers.
1011   @param theRules the list of rules
1012  */
1013 void  HYDROGUI_CalculationDlg::setLandCoverRules( const HYDROData_ListOfRules& theRules ) const
1014 {
1015   myLandCoverPriorityWidget->setRules( theRules );
1016 }
1017
1018 /**
1019   Slot called when objects order is changed.
1020  */
1021 void HYDROGUI_CalculationDlg::onOrderChanged()
1022 {
1023   bool isConfirmed = true;
1024   emit orderChanged( isConfirmed );
1025   if( isConfirmed )
1026     myPriorityWidget->setObjects( getGeometryObjects() );
1027   else
1028     myGeomObjects->undoLastMove();
1029 }
1030
1031 void HYDROGUI_CalculationDlg::setStricklerTable( const QString& theStricklerTableName, bool theBlockSignals )
1032 {
1033   bool isBlocked;
1034   if ( theBlockSignals )
1035     isBlocked = myStricklerTableName->blockSignals( true );
1036   
1037   myStricklerTableName->setCurrentIndex( myStricklerTableName->findText( theStricklerTableName ) );
1038
1039   if ( theBlockSignals )
1040     myStricklerTableName->blockSignals( isBlocked );
1041   else
1042     emit StricklerTableSelected( theStricklerTableName );
1043 }
1044
1045 void HYDROGUI_CalculationDlg::includeLandCovers( const QStringList& theLandCovers, bool theReset )
1046 {
1047   if ( theReset )
1048   {
1049     HYDROGUI_ListModel::Object2VisibleList anObject2VisibleList;
1050     myLandCovers->setObjects(anObject2VisibleList);
1051   }
1052         
1053   HYDROGUI_ListModel::Object2VisibleList anObjectsToInclude;
1054   QList<QListWidgetItem*> aFoundItems;
1055   foreach ( const QString& anObjName, theLandCovers ) {
1056     // Hide the land cover in the available land covers list
1057     aFoundItems = myAvailableLandCovers->findItems( anObjName, Qt::MatchExactly );
1058     foreach ( QListWidgetItem* anItem, aFoundItems ) {
1059       anItem->setHidden( true );
1060     }
1061
1062     // Add the land cover to the list of included objects
1063     Handle(HYDROData_Entity) anObject = 
1064       HYDROGUI_Tool::FindObjectByName( module(), anObjName );
1065     myLandCovers->addObject( HYDROGUI_ListModel::Object2Visible( anObject, true ) );
1066   }
1067
1068   myLandCoverPriorityWidget->setObjects( getLandCovers() );
1069 }
1070
1071 void HYDROGUI_CalculationDlg::excludeLandCovers( const QStringList& theLandCovers )
1072 {
1073   QList<QListWidgetItem*> aFoundItems;
1074   foreach ( const QString& anObjName, theLandCovers ) {
1075     // Set visible the land cover in the available objects list
1076     aFoundItems = myAvailableLandCovers->findItems( anObjName, Qt::MatchExactly );
1077     foreach ( QListWidgetItem* anItem, aFoundItems ) {
1078       anItem->setHidden( false );
1079     }
1080
1081     // Remove the land cover from the list of included objects
1082     myLandCovers->removeObjectByName( anObjName );
1083   }
1084
1085   myLandCoverPriorityWidget->setObjects( getLandCovers() );
1086 }
1087
1088 /**
1089   Slot called when land covers order is changed.
1090  */
1091 void HYDROGUI_CalculationDlg::onOrderLandCoverChanged()
1092 {
1093   bool isConfirmed = true;
1094   emit orderLandCoverChanged( isConfirmed );
1095   if( isConfirmed )
1096     myLandCoverPriorityWidget->setObjects( getLandCovers() );
1097   else
1098     myLandCovers->undoLastMove();  
1099 }