]> SALOME platform Git repositories - modules/hydro.git/blob - src/HYDROGUI/HYDROGUI_CalculationDlg.cxx
Salome HOME
refs #568: use ordered list view with selection synchronized with object browser...
[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   QList<QListWidgetItem*> aFoundItems;
672   foreach ( const QString& anObjName, theObjects ) {
673     // Hide the object in the available objects list
674     aFoundItems = myAvailableGeomObjects->findItems( anObjName, Qt::MatchExactly );
675     foreach ( QListWidgetItem* anItem, aFoundItems ) {
676       anItem->setHidden( true );
677     }
678
679     // Add the object to the list of included objects
680     Handle(HYDROData_Entity) anObject = 
681       HYDROGUI_Tool::FindObjectByName( module(), anObjName );
682     myGeomObjects->addObject( HYDROGUI_ListModel::Object2Visible( anObject, true ) );
683   }
684
685   myPriorityWidget->setObjects( getGeometryObjects() );
686 }
687
688 void HYDROGUI_CalculationDlg::excludeGeomObjects( const QStringList& theObjects )
689 {
690   QList<QListWidgetItem*> aFoundItems;
691   foreach ( const QString& anObjName, theObjects ) {
692     // Set visible the object in the available objects list
693     aFoundItems = myAvailableGeomObjects->findItems( anObjName, Qt::MatchExactly );
694     foreach ( QListWidgetItem* anItem, aFoundItems ) {
695       anItem->setHidden( false );
696     }
697
698     // Remove the object from the list of included objects
699     myGeomObjects->removeObjectByName( anObjName );
700   }
701
702   myPriorityWidget->setObjects( getGeometryObjects() );
703 }
704
705 void HYDROGUI_CalculationDlg::setBoundary( const QString& theObjName )
706 {
707   bool isBlocked = myPolylineName->blockSignals( true );
708   myPolylineName->setCurrentIndex( myPolylineName->findText( theObjName ) );
709   myPolylineName->blockSignals( isBlocked );
710 }
711
712 void HYDROGUI_CalculationDlg::setPolylineNames( const QStringList& theObjects, const QStringList& theObjectsEntries )
713 {
714   myPolylineName->clear();
715   myPolylineName->addItem( "", "" ); // No boundary item
716
717   for ( int i = 0, n = theObjects.length(); i < n; ++i )
718   {
719     myPolylineName->addItem( theObjects.at( i ), theObjectsEntries.at( i ) );
720   }
721 }
722
723 void HYDROGUI_CalculationDlg::setStricklerTableNames( const QStringList& theObjects, const QStringList& theObjectsEntries )
724 {
725   myStricklerTableName->clear();
726
727   for ( int i = 0, n = theObjects.length(); i < n; ++i )
728   {
729     myStricklerTableName->addItem( theObjects.at( i ), theObjectsEntries.at( i ) );
730   }
731 }
732
733 void HYDROGUI_CalculationDlg::setAllGeomObjects( const QStringList& theObjects, const QStringList& theObjectsEntries )
734 {
735   myAvailableGeomObjects->clear();
736
737   for ( int i = 0, n = theObjects.length(); i < n; ++i )
738   {
739     QString anObjName = theObjects.at( i );
740
741     QListWidgetItem* aListItem = new QListWidgetItem( anObjName, myAvailableGeomObjects );
742     aListItem->setFlags( Qt::ItemIsEnabled | Qt::ItemIsSelectable );
743     aListItem->setData( Qt::UserRole, theObjectsEntries.at( i ) );
744   }
745 }
746
747 void HYDROGUI_CalculationDlg::setAllLandCovers( const QStringList& theObjects, const QStringList& theObjectsEntries )
748 {
749   myAvailableLandCovers->clear();
750
751   for ( int i = 0, n = theObjects.length(); i < n; ++i )
752   {
753     QString anObjName = theObjects.at( i );
754
755     QListWidgetItem* aListItem = new QListWidgetItem( anObjName, myAvailableLandCovers );
756     aListItem->setFlags( Qt::ItemIsEnabled | Qt::ItemIsSelectable );
757     aListItem->setData( Qt::UserRole, theObjectsEntries.at( i ) );
758   }
759 }
760
761 QStringList getSelected( QListWidget* theWidget )
762 {
763   QStringList aResList;
764   QList<QListWidgetItem*> aList = theWidget->selectedItems();
765   for ( int i = 0, n = aList.length(); i < n; ++i )
766   {
767     aResList.append( aList.at( i )->text() );
768   }
769   return aResList;
770 }
771
772 QStringList HYDROGUI_CalculationDlg::getSelectedGeomObjects() const
773 {
774   return myGeomObjects->getSelectedNames();
775 }
776
777 QStringList HYDROGUI_CalculationDlg::getSelectedLandCovers() const
778 {
779   return myLandCovers->getSelectedNames();
780 }
781
782 QStringList HYDROGUI_CalculationDlg::getAllGeomObjects() const
783 {
784   return myGeomObjects->getAllNames();
785 }
786
787 QStringList HYDROGUI_CalculationDlg::getAllLandCovers() const
788 {
789   return myLandCovers->getAllNames();
790 }
791
792 QStringList HYDROGUI_CalculationDlg::getSelectedAvailableGeomObjects() const
793 {
794   return getSelected( myAvailableGeomObjects );
795 }
796
797 QStringList HYDROGUI_CalculationDlg::getSelectedAvailableLandCovers() const
798 {
799   return getSelected( myAvailableLandCovers );
800 }
801
802 void HYDROGUI_CalculationDlg::setEditedObject( const Handle(HYDROData_CalculationCase) theCase )
803 {
804   myEditedObject = theCase;
805   myValidator->setEditedObject( theCase );
806
807   // Build the calculation case subtree
808   module()->getDataModel()->buildCaseTree( myBrowser->root(), myEditedObject );
809
810   myBrowser->updateTree();
811   myBrowser->openLevels();
812   myBrowser->adjustColumnsWidth();
813   myBrowser->setAutoUpdate( true );
814   myBrowser->setUpdateModified( true );
815 }
816
817 HYDROGUI_Zone* HYDROGUI_CalculationDlg::getCurrentZone() const
818 {
819   return myCurrentZone;
820 }
821
822 void HYDROGUI_CalculationDlg::refreshZonesBrowser()
823 {
824   SUIT_DataObject* aRoot = myBrowser->root();
825   module()->getDataModel()->updateObjectTree( myEditedObject );
826   module()->getDataModel()->buildCaseTree( aRoot, myEditedObject );
827   myBrowser->updateTree( aRoot );
828 }
829
830 void HYDROGUI_CalculationDlg::onDataChanged()
831 {
832   SUIT_DataObject* aRoot = myBrowser->root();
833   module()->getDataModel()->buildCaseTree( aRoot, myEditedObject );
834   myBrowser->updateTree( aRoot );
835 }
836
837 void HYDROGUI_CalculationDlg::setAvailableGroups( const QStringList& theGroups )
838 {
839   myAvailableGroups->clear();
840   myGroups->clear();
841   foreach( QString aGroup, theGroups )
842     myAvailableGroups->addItem( aGroup );
843 }
844
845 QStringList HYDROGUI_CalculationDlg::getSelectedGroups() const
846 {
847   return getSelected( myGroups );
848 }
849
850 QStringList HYDROGUI_CalculationDlg::getSelectedAvailableGroups() const
851 {
852   return getSelected( myAvailableGroups );
853 }
854
855 void HYDROGUI_CalculationDlg::includeGroups( const QStringList& theObjects )
856 {
857   moveItems( myAvailableGroups, myGroups, theObjects );
858 }
859
860 void HYDROGUI_CalculationDlg::excludeGroups( const QStringList& theObjects )
861 {
862   moveItems( myGroups, myAvailableGroups, theObjects );
863 }
864
865 /**
866   Get creation mode.
867   @param theMode the mode
868 */
869 int HYDROGUI_CalculationDlg::getMode() const
870 {
871   return myModeButtons->checkedId();
872 }
873
874 /**
875   Set creation mode.
876   @param theMode the mode
877 */
878 void HYDROGUI_CalculationDlg::setMode( int theMode )
879 {
880   bool isBlocked = myModeButtons->blockSignals( true );
881   myModeButtons->button( theMode )->setChecked( true );
882   myModeButtons->blockSignals( isBlocked );
883
884   bool isAuto = ( theMode == HYDROData_CalculationCase::AUTOMATIC );
885
886   myGeomObjects->setOrderingEnabled( isAuto );
887   QWidget* aWidget = mySplitter->widget( 1 );
888   if ( aWidget ) {
889     aWidget->setVisible( isAuto );
890   }
891 }
892
893 /**
894   Get creation mode for land covers panel.
895   @param theMode the mode
896 */
897 int HYDROGUI_CalculationDlg::getLandCoverMode() const
898 {
899   return myLandCoverModeButtons->checkedId();
900 }
901
902 /**
903   Set creation mode for land cover panel.
904   @param theMode the mode
905 */
906 void HYDROGUI_CalculationDlg::setLandCoverMode( int theMode )
907 {
908   bool isBlocked = myLandCoverModeButtons->blockSignals( true );
909   myLandCoverModeButtons->button( theMode )->setChecked( true );
910   myLandCoverModeButtons->blockSignals( isBlocked );
911
912   bool isAuto = ( theMode == HYDROData_CalculationCase::AUTOMATIC );
913
914   myLandCovers->setOrderingEnabled( isAuto );
915   QWidget* aWidget = myLandCoverSplitter->widget( 1 );
916   if ( aWidget ) {
917     aWidget->setVisible( isAuto );
918   }
919 }
920
921 /**
922   Enable/disable zones drag'n'drop and renaming.
923   @param theIsEnabled if true - zones drag'n'drop and renaming will be enabled
924 */
925 void HYDROGUI_CalculationDlg::setEditZonesEnabled( const bool theIsEnabled )
926 {
927   myBrowser->setReadOnly( !theIsEnabled );
928 }
929
930 /**
931   Enable/disable land covers drag'n'drop and renaming.
932   @param theIsEnabled if true - land covers drag'n'drop and renaming will be enabled
933 */
934 void HYDROGUI_CalculationDlg::setEditLandCoversEnabled( const bool theIsEnabled )
935 {
936   myLandCoverBrowser->setReadOnly( !theIsEnabled );
937 }
938
939 /**
940   Get included geometry objects.
941   @return the list of geometry objects
942  */
943 QList<Handle(HYDROData_Entity)> HYDROGUI_CalculationDlg::getGeometryObjects()
944 {
945   QList<Handle(HYDROData_Entity)> anEntities = myGeomObjects->getObjects();
946   QList<Handle(HYDROData_Entity)> anObjects;
947
948   foreach ( Handle(HYDROData_Entity) anEntity, anEntities ) {
949     Handle(HYDROData_Object) anObj = Handle(HYDROData_Object)::DownCast( anEntity );
950     if ( anObj.IsNull() ) {
951       continue;
952     }
953
954     anObjects << anObj;
955   }
956
957   return anObjects;
958 }
959
960 /**
961   Get included land covers.
962   @return the list of land covers
963  */
964 QList<Handle(HYDROData_Entity)> HYDROGUI_CalculationDlg::getLandCovers()
965 {
966   QList<Handle(HYDROData_Entity)> anEntities = myLandCovers->getObjects();
967   QList<Handle(HYDROData_Entity)> aLandCovers;
968
969   foreach ( Handle(HYDROData_Entity) anEntity, anEntities ) {
970     Handle(HYDROData_LandCover) aLandCover = Handle(HYDROData_LandCover)::DownCast( anEntity );
971     if ( aLandCover.IsNull() ) {
972       continue;
973     }
974
975     aLandCovers << aLandCover;
976   }
977
978   return aLandCovers;
979 }
980
981 /**
982   Get rules.
983   @return the list of rules
984  */
985 HYDROData_ListOfRules HYDROGUI_CalculationDlg::getRules() const
986 {
987   return myPriorityWidget->getRules();
988 }
989
990 /**
991   Set rules.
992   @param theRules the list of rules
993  */
994 void  HYDROGUI_CalculationDlg::setRules( const HYDROData_ListOfRules& theRules ) const
995 {
996   myPriorityWidget->setRules( theRules );
997 }
998
999 /**
1000   Get rules defined for land covers.
1001   @return the list of rules
1002  */
1003 HYDROData_ListOfRules HYDROGUI_CalculationDlg::getLandCoverRules() const
1004 {
1005   return myLandCoverPriorityWidget->getRules();
1006 }
1007
1008 /**
1009   Set rules for land covers.
1010   @param theRules the list of rules
1011  */
1012 void  HYDROGUI_CalculationDlg::setLandCoverRules( const HYDROData_ListOfRules& theRules ) const
1013 {
1014   myLandCoverPriorityWidget->setRules( theRules );
1015 }
1016
1017 /**
1018   Slot called when objects order is changed.
1019  */
1020 void HYDROGUI_CalculationDlg::onOrderChanged()
1021 {
1022   bool isConfirmed = true;
1023   emit orderChanged( isConfirmed );
1024   if( isConfirmed )
1025     myPriorityWidget->setObjects( getGeometryObjects() );
1026   else
1027     myGeomObjects->undoLastMove();
1028 }
1029
1030 void HYDROGUI_CalculationDlg::setStricklerTable( const QString& theStricklerTableName, bool theBlockSignals )
1031 {
1032   bool isBlocked;
1033   if ( theBlockSignals )
1034     isBlocked = myStricklerTableName->blockSignals( true );
1035   
1036   myStricklerTableName->setCurrentIndex( myStricklerTableName->findText( theStricklerTableName ) );
1037
1038   if ( theBlockSignals )
1039     myStricklerTableName->blockSignals( isBlocked );
1040   else
1041     emit StricklerTableSelected( theStricklerTableName );
1042 }
1043
1044 void HYDROGUI_CalculationDlg::includeLandCovers( const QStringList& theLandCovers, bool theReset )
1045 {
1046   if ( theReset )
1047   {
1048     HYDROGUI_ListModel::Object2VisibleList anObject2VisibleList;
1049     myLandCovers->setObjects(anObject2VisibleList);
1050   }
1051         
1052   QList<QListWidgetItem*> aFoundItems;
1053   foreach ( const QString& anObjName, theLandCovers ) {
1054     // Hide the land cover in the available land covers list
1055     aFoundItems = myAvailableLandCovers->findItems( anObjName, Qt::MatchExactly );
1056     foreach ( QListWidgetItem* anItem, aFoundItems ) {
1057       anItem->setHidden( true );
1058     }
1059
1060     // Add the land cover to the list of included objects
1061     Handle(HYDROData_Entity) anObject = 
1062       HYDROGUI_Tool::FindObjectByName( module(), anObjName );
1063     myLandCovers->addObject( HYDROGUI_ListModel::Object2Visible( anObject, true ) );
1064   }
1065
1066   myLandCoverPriorityWidget->setObjects( getLandCovers() );
1067 }
1068
1069 void HYDROGUI_CalculationDlg::excludeLandCovers( const QStringList& theLandCovers )
1070 {
1071   QList<QListWidgetItem*> aFoundItems;
1072   foreach ( const QString& anObjName, theLandCovers ) {
1073     // Set visible the land cover in the available objects list
1074     aFoundItems = myAvailableLandCovers->findItems( anObjName, Qt::MatchExactly );
1075     foreach ( QListWidgetItem* anItem, aFoundItems ) {
1076       anItem->setHidden( false );
1077     }
1078
1079     // Remove the land cover from the list of included objects
1080     myLandCovers->removeObjectByName( anObjName );
1081   }
1082
1083   myLandCoverPriorityWidget->setObjects( getLandCovers() );
1084 }
1085
1086 /**
1087   Slot called when land covers order is changed.
1088  */
1089 void HYDROGUI_CalculationDlg::onOrderLandCoverChanged()
1090 {
1091   bool isConfirmed = true;
1092   emit orderLandCoverChanged( isConfirmed );
1093   if( isConfirmed )
1094     myLandCoverPriorityWidget->setObjects( getLandCovers() );
1095   else
1096     myLandCovers->undoLastMove();  
1097 }