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