Salome HOME
epaisseur des polylignes = 2
[modules/hydro.git] / src / HYDROGUI / HYDROGUI_CalculationDlg.cxx
index da21a508e0c96aa8c852785427bbde35c45bbb24..778ac4311666fb01038a44833c13200a089a1ba7 100644 (file)
@@ -1,12 +1,8 @@
-// Copyright (C) 2007-2013  CEA/DEN, EDF R&D, OPEN CASCADE
-//
-// Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
-// CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
-//
+// Copyright (C) 2014-2015  EDF-R&D
 // This library is free software; you can redistribute it and/or
 // modify it under the terms of the GNU Lesser General Public
 // License as published by the Free Software Foundation; either
-// version 2.1 of the License.
+// version 2.1 of the License, or (at your option) any later version.
 //
 // This library is distributed in the hope that it will be useful,
 // but WITHOUT ANY WARRANTY; without even the implied warranty of
 #include "HYDROGUI_CalculationDlg.h"
 
 #include "HYDROGUI_ObjSelector.h"
-#include "HYDROGUI_Tool.h"
+#include "HYDROGUI_Tool2.h"
 #include "HYDROGUI_DataBrowser.h"
 #include "HYDROGUI_DataModel.h"
+#include "HYDROGUI_ListSelector.h"
 #include "HYDROGUI_Module.h"
 #include "HYDROGUI_DataObject.h"
 #include "HYDROGUI_NameValidator.h"
 #include "HYDROGUI_Region.h"
 #include "HYDROGUI_Zone.h"
+#include "HYDROGUI_OrderedListWidget.h"
+#include "HYDROGUI_PriorityWidget.h"
+#include "HYDROGUI_PriorityTableModel.h"
 
 #include <HYDROData_Document.h>
 #include <HYDROData_Entity.h>
 #include <SUIT_Study.h>
 
 #include <LightApp_Application.h>
+#include <LightApp_SelectionMgr.h>
 #include <SUIT_Desktop.h>
 #include <SUIT_MessageBox.h>
 
+#include <QButtonGroup>
+#include <QComboBox>
 #include <QGroupBox>
 #include <QLabel>
 #include <QLayout>
 #include <QListWidget>
 #include <QPicture>
 #include <QPushButton>
+#include <QRadioButton>
+#include <QSplitter>
+#include <QTableWidget>
 #include <QToolButton>
+#include <QTreeView>
 #include <QWizardPage>
-#include <QComboBox>
-#include <QStackedWidget>
+
 
 HYDROGUI_CalculationDlg::HYDROGUI_CalculationDlg( HYDROGUI_Module* theModule, const QString& theTitle )
 : HYDROGUI_Wizard( theModule, theTitle )
 {
   addPage( createObjectsPage() );
   addPage( createGroupsPage() );
+  addPage( createLandCoverMapPage() );
   addPage( createZonesPage() );
 }
 
@@ -75,16 +82,32 @@ HYDROGUI_CalculationDlg::~HYDROGUI_CalculationDlg()
 void HYDROGUI_CalculationDlg::reset()
 {
   myObjectName->clear();
-  myGeomObjects->clear();
+  HYDROGUI_ListModel::Object2VisibleList anObject2VisibleList;
+  myGeomObjects->setObjects(anObject2VisibleList);
   myPolylineName->clear();
+  myLandCoverMapName->clear();
+  myStricklerTableName->clear();
   myAvailableGeomObjects->clear();
+
+  // Activate the automatic mode
+  setMode( HYDROData_CalculationCase::AUTOMATIC );
+
+  // Reset the priority widget state
+  QList<Handle(HYDROData_Entity)> anObjects;
+  myPriorityWidget->setObjects( anObjects );
 }
 
 QWizardPage* HYDROGUI_CalculationDlg::createObjectsPage() {
   QWizardPage* aPage = new QWizardPage( mainFrame() );
   QFrame* aFrame = new QFrame( aPage );
 
-  // Calculation name
+  // Splitter
+  mySplitter = new QSplitter(Qt::Vertical);
+
+  // Top of the page
+  QWidget* aTopContainer = new QWidget;
+   
+  // calculation name
   myObjectName = new QLineEdit( aPage );
   myValidator = new HYDROGUI_NameValidator(module(), myObjectName);
   myObjectName->setValidator( myValidator );
@@ -92,32 +115,49 @@ QWizardPage* HYDROGUI_CalculationDlg::createObjectsPage() {
   connect( myValidator, SIGNAL( emptyName() ), SLOT( onEmptyName() ) );
   connect( myValidator, SIGNAL( alreadyExists( QString ) ), SLOT( onAlreadyExists( QString ) ) );
 
+  // polyline name
   myPolylineName = new QComboBox( aPage );
   connect( myPolylineName, SIGNAL( activated( const QString & ) ), 
     SIGNAL( boundarySelected( const QString & ) ) );
 
-  myGeomObjects = new QListWidget( aPage );
-  myGeomObjects->setSelectionMode( QListWidget::ExtendedSelection );
-  myGeomObjects->setEditTriggers( QListWidget::NoEditTriggers );
-  myGeomObjects->setViewMode( QListWidget::ListMode );
-  myGeomObjects->setSortingEnabled( true );
+  // names labels
+  QLabel* aNameLabel = new QLabel( tr( "NAME" ), aPage );
+  QLabel* aLimitsLabel = new QLabel( tr( "LIMITS" ), aPage );
+
+  // mode selector (auto/manual)
+  QGroupBox* aModeGroup = new QGroupBox( tr( "MODE" ) );
 
+  QRadioButton* aManualRB = new QRadioButton( tr( "MANUAL" ), mainFrame() );
+  QRadioButton* anAutoRB = new QRadioButton( tr( "AUTO" ), mainFrame() );
+
+  myModeButtons = new QButtonGroup( mainFrame() );
+  myModeButtons->addButton( anAutoRB, HYDROData_CalculationCase::AUTOMATIC );
+  myModeButtons->addButton( aManualRB, HYDROData_CalculationCase::MANUAL );
+  
+  QBoxLayout* aModeSelectorLayout = new QHBoxLayout;
+  aModeSelectorLayout->setMargin( 5 );
+  aModeSelectorLayout->setSpacing( 5 );
+  aModeSelectorLayout->addWidget( anAutoRB );
+  aModeSelectorLayout->addWidget( aManualRB );
+  aModeGroup->setLayout( aModeSelectorLayout );
+
+  // geometry objects
+  QLabel* anObjectsLabel = new QLabel( tr( "CALCULATION_REFERENCE_OBJECTS" ) );
+  myGeomObjects = new HYDROGUI_OrderedListWidget( aPage, 16 );
+  myGeomObjects->setHiddenObjectsShown(true);
+  myGeomObjects->setVisibilityIconShown(false);
+  myGeomObjects->setContentsMargins(QMargins());
+  // included geometry objects
+  QLabel* anIncludedLabel = new QLabel( tr( "INCLUDED_OBJECTS" ) );
   myAvailableGeomObjects = new QListWidget( aPage );
   myAvailableGeomObjects->setSelectionMode( QListWidget::ExtendedSelection );
   myAvailableGeomObjects->setEditTriggers( QListWidget::NoEditTriggers );
   myAvailableGeomObjects->setViewMode( QListWidget::ListMode );
   myAvailableGeomObjects->setSortingEnabled( true );
 
-  connect( myGeomObjects, SIGNAL( itemSelectionChanged() ), 
-    SIGNAL( objectsSelected() ) );
-
-  QFrame* anObjectsFrame = new QFrame( aPage );
-  QGridLayout* anObjsLayout = new QGridLayout( anObjectsFrame );
-  anObjsLayout->setMargin( 5 );
-  anObjsLayout->setSpacing( 5 );
-  anObjectsFrame->setLayout( anObjsLayout );
-
-  QFrame* aBtnsFrame = new QFrame( anObjectsFrame );
+  // buttons
+  QFrame* aBtnsFrame = new QFrame;
   QVBoxLayout* aBtnsLayout = new QVBoxLayout( aBtnsFrame );
   aBtnsLayout->setMargin( 5 );
   aBtnsLayout->setSpacing( 5 );
@@ -125,41 +165,84 @@ QWizardPage* HYDROGUI_CalculationDlg::createObjectsPage() {
   QPushButton* anAddBtn = new QPushButton( tr("INCLUDE"), aBtnsFrame );
   QPushButton* aRemoveBtn = new QPushButton( tr("EXCLUDE"), aBtnsFrame );
 
-  // Fill the butons frame with two buttons
+  // fill the butons frame with two buttons
   aBtnsLayout->addWidget( anAddBtn );
   aBtnsLayout->addWidget( aRemoveBtn );
   aBtnsLayout->addStretch( 1 );
-
-  QLabel* anIncludedLabel = new QLabel( tr( "INCLUDED_OBJECTS" ), anObjectsFrame );
-  QLabel* anObjectsLabel = new QLabel( tr( "CALCULATION_REFERENCE_OBJECTS" ), anObjectsFrame );
-
-  // Fill the objects frame with two lists, two labels and with buttons frame
+  
+  // top of the page layout
+  
+  // objects frame
+  QFrame* anObjectsFrame = new QFrame( aPage );
+  anObjectsFrame->setFrameStyle( QFrame::Panel | QFrame::Raised );
+  QGridLayout* anObjsLayout = new QGridLayout( anObjectsFrame );
+  anObjsLayout->setMargin( 5 );
+  anObjsLayout->setSpacing( 5 );
+  anObjectsFrame->setLayout( anObjsLayout );
+  
+  // fill the objects frame with two lists, two labels and with buttons frame
   anObjsLayout->addWidget( anObjectsLabel, 0, 0, Qt::AlignHCenter );
   anObjsLayout->addWidget( anIncludedLabel, 0, 2, Qt::AlignHCenter );
-  anObjsLayout->addWidget( myAvailableGeomObjects, 1, 0, Qt::AlignHCenter );
+  anObjsLayout->addWidget( myAvailableGeomObjects, 1, 0 );
   anObjsLayout->addWidget( aBtnsFrame, 1, 1, Qt::AlignHCenter );
-  anObjsLayout->addWidget( myGeomObjects, 1, 2, Qt::AlignHCenter );
-
-
-  QLabel* aNameLabel = new QLabel( tr( "NAME" ), aPage );
-  QLabel* aLimitsLabel = new QLabel( tr( "LIMITS" ), aPage );
-
-  // Fill the page
-  QGridLayout* aPageLayout = new QGridLayout( aPage );
+  anObjsLayout->addWidget( myGeomObjects, 1, 2 );
+  
+  // fill the top of the page
+  QGridLayout* aTopLayout = new QGridLayout;
+  aTopLayout->setMargin( 5 );
+  aTopLayout->setSpacing( 5 );
+  aTopLayout->setVerticalSpacing( 10 );
+  aTopLayout->addWidget( aNameLabel,     0, 0, Qt::AlignHCenter );
+  aTopLayout->addWidget( myObjectName,   0, 1 );
+  aTopLayout->addWidget( aLimitsLabel,   1, 0, Qt::AlignHCenter );
+  aTopLayout->addWidget( myPolylineName, 1, 1 );
+  aTopLayout->addWidget( aModeGroup, 2, 0, 1, 2 );
+  aTopLayout->addWidget( anObjectsFrame, 3, 0, 1, 2 );
+
+  aTopContainer->setLayout( aTopLayout );
+
+  // add the top of the page to the splitter
+  mySplitter->insertWidget(0, aTopContainer);
+  mySplitter->setStretchFactor(0, 2);
+
+  // Bottom of the page
+  myPriorityWidget = new HYDROGUI_PriorityWidget( mainFrame() );
+
+  QGroupBox* aPriorityGroup = new QGroupBox( tr( "PRIORITY" ) );
+  QBoxLayout* aPriorityLayout = new QHBoxLayout;
+  aPriorityLayout->setMargin( 5 );
+  aPriorityLayout->setSpacing( 5 );
+  aPriorityLayout->addWidget( myPriorityWidget );
+  aPriorityGroup->setLayout( aPriorityLayout );
+
+  // add the bottom of the page to the splitter
+  mySplitter->insertWidget(1, aPriorityGroup);
+  mySplitter->setStretchFactor(1, 1);
+
+  // Page layout
+  QVBoxLayout* aPageLayout = new QVBoxLayout;
   aPageLayout->setMargin( 5 );
   aPageLayout->setSpacing( 5 );
-  aPageLayout->setVerticalSpacing( 10 );
-  aPageLayout->addWidget( aNameLabel,     0, 0, Qt::AlignHCenter );
-  aPageLayout->addWidget( myObjectName,   0, 1 );
-  aPageLayout->addWidget( aLimitsLabel,   1, 0, Qt::AlignHCenter );
-  aPageLayout->addWidget( myPolylineName, 1, 1 );
-  aPageLayout->addWidget( anObjectsFrame, 2, 0, 1, 2, Qt::AlignHCenter );
+  aPageLayout->addWidget( mySplitter );
 
   aPage->setLayout( aPageLayout );
 
+  // Create selector
+  if ( module() ) {
+    HYDROGUI_ListSelector* aListSelector = 
+      new HYDROGUI_ListSelector( myGeomObjects, module()->getApp()->selectionMgr() );
+    aListSelector->setAutoBlock( true );
+  }
+
+  // Connections
+  connect( myModeButtons, SIGNAL( buttonClicked( int ) ), SIGNAL( changeMode( int ) ) );
   connect( anAddBtn, SIGNAL( clicked() ), SIGNAL( addObjects() ) );
   connect( aRemoveBtn, SIGNAL( clicked() ), SIGNAL( removeObjects() ) );
 
+  connect( myGeomObjects, SIGNAL( orderChanged() ), SLOT( onOrderChanged() ) );
+
+  connect( myPriorityWidget, SIGNAL( ruleChanged() ), SLOT( onRuleChanged() ) );
+
   return aPage;
 }
 
@@ -207,16 +290,16 @@ QWizardPage* HYDROGUI_CalculationDlg::createGroupsPage() {
   // Fill the objects frame with two lists, two labels and with buttons frame
   aGroupsLayout->addWidget( anAvailableLabel, 0, 0, Qt::AlignHCenter );
   aGroupsLayout->addWidget( anIncludedLabel, 0, 2, Qt::AlignHCenter );
-  aGroupsLayout->addWidget( myAvailableGroups, 1, 0, Qt::AlignHCenter );
+  aGroupsLayout->addWidget( myAvailableGroups, 1, 0 );
   aGroupsLayout->addWidget( aBtnsFrame, 1, 1, Qt::AlignHCenter );
-  aGroupsLayout->addWidget( myGroups, 1, 2, Qt::AlignHCenter );
+  aGroupsLayout->addWidget( myGroups, 1, 2 );
 
   // Fill the page
   QGridLayout* aPageLayout = new QGridLayout( aPage );
   aPageLayout->setMargin( 5 );
   aPageLayout->setSpacing( 5 );
   aPageLayout->setVerticalSpacing( 10 );
-  aPageLayout->addWidget( aGroupsFrame, 0, 0, Qt::AlignHCenter );
+  aPageLayout->addWidget( aGroupsFrame, 0, 0 );
 
   aPage->setLayout( aPageLayout );
 
@@ -226,27 +309,76 @@ QWizardPage* HYDROGUI_CalculationDlg::createGroupsPage() {
   return aPage;
 }
 
+QWizardPage* HYDROGUI_CalculationDlg::createLandCoverMapPage() {
+  QWizardPage* aPage = new QWizardPage( mainFrame() );
+  QFrame* aFrame = new QFrame( aPage );
+
+  // Top of the page
+  QWidget* aTopContainer = new QWidget;
+
+  // Combo-box to choose land cover map object
+  QLabel* aLandCoverMapLabel = new QLabel( tr( "LAND_COVER_MAP" ), aPage );
+  myLandCoverMapName = new QComboBox( aPage );
+  myLandCoverMapName->setSizePolicy( QSizePolicy( QSizePolicy::Expanding, QSizePolicy::Fixed ) );
+  connect( myLandCoverMapName, SIGNAL( activated( const QString & ) ), 
+                               SIGNAL( landCoverMapSelected( const QString & ) ) );
+  // Combo-box to choose Strickler table name
+  QLabel* aStricklerTableLabel = new QLabel( tr( "STRICKLER_TABLE" ), aPage );
+  myStricklerTableName = new QComboBox( aPage );
+  myStricklerTableName->setSizePolicy( QSizePolicy( QSizePolicy::Expanding, QSizePolicy::Fixed ) );
+  connect( myStricklerTableName, SIGNAL( activated( const QString & ) ), 
+                                 SIGNAL( StricklerTableSelected( const QString & ) ) );
+
+  // Fill the top layout of the page
+  QGridLayout* aGridLayout = new QGridLayout;
+  aGridLayout->setMargin( 5 );
+  aGridLayout->setSpacing( 5 );
+  aGridLayout->setVerticalSpacing( 10 );
+  aGridLayout->addWidget( aLandCoverMapLabel,   0, 0 );
+  aGridLayout->addWidget( myLandCoverMapName,   0, 1 );
+  aGridLayout->addWidget( aStricklerTableLabel, 1, 0 );
+  aGridLayout->addWidget( myStricklerTableName, 1, 1 );
+
+  QVBoxLayout* aTopLayout = new QVBoxLayout;
+  aTopLayout->setMargin( 5 );
+  aTopLayout->setSpacing( 5 );
+  aTopLayout->addLayout( aGridLayout );
+  aTopLayout->addStretch( 1 );
+  
+  aTopContainer->setLayout( aTopLayout );
+
+  aPage->setLayout( aTopLayout );
+
+  return aPage;
+}
+
 QWizardPage* HYDROGUI_CalculationDlg::createZonesPage() {
   QWizardPage* aPage = new QWizardPage( mainFrame() );
   QFrame* aFrame = new QFrame( aPage );
 
   QGridLayout* aLayout = new QGridLayout( aPage );
+
+  QLabel* aResultsOnGeomObjectsLabel = new QLabel( tr( "RESULTS_ON_GEOMETRY_OBJECTS" ), aFrame );
   
   myBrowser = new HYDROGUI_DataBrowser( module(), NULL, aPage );
   myBrowser->setAutoOpenLevel( 3 );
   aLayout->setMargin( 5 );
   aLayout->setSpacing( 5 );
 
-  aLayout->addWidget( myBrowser, 0, 0, 1, 2 );
+  aLayout->addWidget( aResultsOnGeomObjectsLabel, 0, 0 );
+  aLayout->addWidget( myBrowser, 1, 0, 1, 2 );
 
-  myBatimetryLabel = new QLabel( tr( "BATHYMETRY" ), aFrame );
+  myBathymetryLabel = new QLabel( tr( "BATHYMETRY" ), aFrame );
   myBathymetryChoice = new QComboBox( aFrame );
 
   myBathymetryChoice->setVisible( false );
-  myBatimetryLabel->setVisible( false );
+  myBathymetryLabel->setVisible( false );
 
-  aLayout->addWidget( myBatimetryLabel, 1, 0 );
-  aLayout->addWidget( myBathymetryChoice, 1, 1 );
+  aLayout->addWidget( myBathymetryLabel, 2, 0 );
+  aLayout->addWidget( myBathymetryChoice, 2, 1 );
+
+  QPushButton* aRegenerateBtn = new QPushButton( tr( "REGENERATE_COLORS" ), this );
+  aLayout->addWidget( aRegenerateBtn, 3, 0 );
 
   aPage->setLayout( aLayout );
 
@@ -257,16 +389,16 @@ QWizardPage* HYDROGUI_CalculationDlg::createZonesPage() {
   connect( myBrowser, 
       SIGNAL( dropped( const QList<SUIT_DataObject*>&, SUIT_DataObject*, int, Qt::DropAction ) ),
       SLOT( onZonesDropped( const QList<SUIT_DataObject*>&, SUIT_DataObject*, int, Qt::DropAction ) ) );
-
+  connect( myBrowser, SIGNAL( newRegion() ), this, SLOT( OnNewRegion() ) );
+  connect( aRegenerateBtn, SIGNAL( clicked() ), this, SIGNAL( regenerateColors() ) );
   return aPage;
 }
 
-
 bool HYDROGUI_CalculationDlg::acceptCurrent() const
 {
   QString anErrorMsg;
 
-  if ( myGeomObjects->count() == 0 )
+  if ( false /*myGeomObjects->count() == 0*/ )
   {
     anErrorMsg = tr( "EMPTY_GEOMETRY_OBJECTS" );
   }
@@ -331,6 +463,11 @@ void HYDROGUI_CalculationDlg::onZonesDropped( const QList<SUIT_DataObject*>& the
   }
 }
 
+void HYDROGUI_CalculationDlg::OnNewRegion()
+{
+  emit createRegion( myBrowser->getSelected() );
+}
+
 void HYDROGUI_CalculationDlg::onMergeTypeSelected( int theIndex )
 {
   int aType = myBathymetryChoice->itemData( theIndex ).toInt();
@@ -356,7 +493,7 @@ void HYDROGUI_CalculationDlg::onSelected( SUIT_DataObject* theObject )
     myBathymetryChoice->addItem( tr("MERGE_UNKNOWN"), HYDROData_Zone::Merge_UNKNOWN );
     myBathymetryChoice->addItem( tr("MERGE_ZMIN"), HYDROData_Zone::Merge_ZMIN );
     myBathymetryChoice->addItem( tr("MERGE_ZMAX"), HYDROData_Zone::Merge_ZMAX );
-    QStringList aList = aZone->getBathymetries();
+    QStringList aList = aZone->getObjects();
     for ( int i = 0; i < aList.length(); i++ )
     {
       myBathymetryChoice->addItem( aList.at( i ), HYDROData_Zone::Merge_Object );
@@ -372,7 +509,7 @@ void HYDROGUI_CalculationDlg::onSelected( SUIT_DataObject* theObject )
         aCurIndex = 2;
         break;
       case HYDROData_Zone::Merge_Object:
-        aCurIndex = 3 + aList.indexOf( aZone->text( HYDROGUI_DataObject::BathymetryId ) );
+        aCurIndex = 3 + aList.indexOf( aZone->text( HYDROGUI_DataObject::AltitudeObjId ) );
         break;
       default:
         aCurIndex = 0; // Select unknown by default
@@ -382,7 +519,8 @@ void HYDROGUI_CalculationDlg::onSelected( SUIT_DataObject* theObject )
   }
 
   myBathymetryChoice->setVisible( doShow );
-  myBatimetryLabel->setVisible( doShow );
+  myBathymetryChoice->setEnabled( getMode() == HYDROData_CalculationCase::MANUAL );
+  myBathymetryLabel->setVisible( doShow );
 }
 
 void HYDROGUI_CalculationDlg::setObjectName( const QString& theName )
@@ -395,7 +533,7 @@ QString HYDROGUI_CalculationDlg::getObjectName() const
   return myObjectName->text();
 }
 
-void HYDROGUI_CalculationDlg::includeGeomObjects( const QStringList& theObjects )
+void moveItems( QListWidget* theSource, QListWidget* theDest, const QStringList& theObjects )
 {
   QList<QListWidgetItem*> aFoundItems;
   int anIdx;
@@ -404,37 +542,52 @@ void HYDROGUI_CalculationDlg::includeGeomObjects( const QStringList& theObjects
   for ( int i = 0, n = theObjects.length(); i < n; ++i )
   {
     QString anObjName = theObjects.at( i );
-    aFoundItems = myAvailableGeomObjects->findItems( anObjName, Qt::MatchExactly );
+    aFoundItems = theSource->findItems( anObjName, Qt::MatchExactly );
     for ( anIdx = 0; anIdx < aFoundItems.length(); anIdx++ )
     {
       anItem = aFoundItems.at( anIdx );
       // Remove this object from available objects list
-      anItem = myAvailableGeomObjects->takeItem( myAvailableGeomObjects->row( anItem ) );
+      anItem = theSource->takeItem( theSource->row( anItem ) );
       // Add the item to the included objects list
-      myGeomObjects->addItem( anItem );
+      theDest->addItem( anItem );
     }
   }
 }
 
-void HYDROGUI_CalculationDlg::excludeGeomObjects( const QStringList& theObjects )
+void HYDROGUI_CalculationDlg::includeGeomObjects( const QStringList& theObjects )
 {
   QList<QListWidgetItem*> aFoundItems;
-  int anIdx;
-  QListWidgetItem* anItem;
+  foreach ( const QString& anObjName, theObjects ) {
+    // Hide the object in the available objects list
+    aFoundItems = myAvailableGeomObjects->findItems( anObjName, Qt::MatchExactly );
+    foreach ( QListWidgetItem* anItem, aFoundItems ) {
+      anItem->setHidden( true );
+    }
 
-  for ( int i = 0, n = theObjects.length(); i < n; ++i )
-  {
-    QString anObjName = theObjects.at( i );
-    aFoundItems = myGeomObjects->findItems( anObjName, Qt::MatchExactly );
-    for ( anIdx = 0; anIdx < aFoundItems.length(); anIdx++ )
-    {
-      anItem = aFoundItems.at( anIdx );
-      // Remove this object from included objects list
-      anItem = myGeomObjects->takeItem( myGeomObjects->row( anItem ) );
-      // Add the item to the excluded objects list
-      myAvailableGeomObjects->addItem( anItem );
+    // Add the object to the list of included objects
+    Handle(HYDROData_Entity) anObject = 
+      HYDROGUI_Tool::FindObjectByName( module(), anObjName );
+    myGeomObjects->addObject( HYDROGUI_ListModel::Object2Visible( anObject, true ) );
+  }
+
+  myPriorityWidget->setObjects( getGeometryObjects() );
+}
+
+void HYDROGUI_CalculationDlg::excludeGeomObjects( const QStringList& theObjects )
+{
+  QList<QListWidgetItem*> aFoundItems;
+  foreach ( const QString& anObjName, theObjects ) {
+    // Set visible the object in the available objects list
+    aFoundItems = myAvailableGeomObjects->findItems( anObjName, Qt::MatchExactly );
+    foreach ( QListWidgetItem* anItem, aFoundItems ) {
+      anItem->setHidden( false );
     }
+
+    // Remove the object from the list of included objects
+    myGeomObjects->removeObjectByName( anObjName );
   }
+
+  myPriorityWidget->setObjects( getGeometryObjects() );
 }
 
 void HYDROGUI_CalculationDlg::setBoundary( const QString& theObjName )
@@ -454,6 +607,27 @@ void HYDROGUI_CalculationDlg::setPolylineNames( const QStringList& theObjects, c
     myPolylineName->addItem( theObjects.at( i ), theObjectsEntries.at( i ) );
   }
 }
+
+void HYDROGUI_CalculationDlg::setLandCoverMapsNames( const QStringList& theObjects, const QStringList& theObjectsEntries )
+{
+  myLandCoverMapName->clear();
+
+  for ( int i = 0, n = theObjects.length(); i < n; ++i )
+  {
+    myLandCoverMapName->addItem( theObjects.at( i ), theObjectsEntries.at( i ) );
+  }
+}
+
+void HYDROGUI_CalculationDlg::setStricklerTableNames( const QStringList& theObjects, const QStringList& theObjectsEntries )
+{
+  myStricklerTableName->clear();
+
+  for ( int i = 0, n = theObjects.length(); i < n; ++i )
+  {
+    myStricklerTableName->addItem( theObjects.at( i ), theObjectsEntries.at( i ) );
+  }
+}
+
 void HYDROGUI_CalculationDlg::setAllGeomObjects( const QStringList& theObjects, const QStringList& theObjectsEntries )
 {
   myAvailableGeomObjects->clear();
@@ -468,10 +642,10 @@ void HYDROGUI_CalculationDlg::setAllGeomObjects( const QStringList& theObjects,
   }
 }
 
-QStringList HYDROGUI_CalculationDlg::getSelectedGeomObjects() const
+QStringList getSelected( QListWidget* theWidget )
 {
   QStringList aResList;
-  QList<QListWidgetItem*> aList = myGeomObjects->selectedItems();
+  QList<QListWidgetItem*> aList = theWidget->selectedItems();
   for ( int i = 0, n = aList.length(); i < n; ++i )
   {
     aResList.append( aList.at( i )->text() );
@@ -479,15 +653,19 @@ QStringList HYDROGUI_CalculationDlg::getSelectedGeomObjects() const
   return aResList;
 }
 
+QStringList HYDROGUI_CalculationDlg::getSelectedGeomObjects() const
+{
+  return myGeomObjects->getSelectedNames();
+}
+
+QStringList HYDROGUI_CalculationDlg::getAllGeomObjects() const
+{
+  return myGeomObjects->getAllNames();
+}
+
 QStringList HYDROGUI_CalculationDlg::getSelectedAvailableGeomObjects() const
 {
-  QStringList aResList;
-  QList<QListWidgetItem*> aList = myAvailableGeomObjects->selectedItems();
-  for ( int i = 0, n = aList.length(); i < n; ++i )
-  {
-    aResList.append( aList.at( i )->text() );
-  }
-  return aResList;
+  return getSelected( myAvailableGeomObjects );
 }
 
 void HYDROGUI_CalculationDlg::setEditedObject( const Handle(HYDROData_CalculationCase) theCase )
@@ -496,14 +674,13 @@ void HYDROGUI_CalculationDlg::setEditedObject( const Handle(HYDROData_Calculatio
   myValidator->setEditedObject( theCase );
 
   // Build the calculation case subtree
-  module()->getDataModel()->buildCaseTree( myBrowser->root(), myEditedObject );
+  module()->getDataModel()->buildCaseTree( myBrowser->root(), myEditedObject);
 
   myBrowser->updateTree();
   myBrowser->openLevels();
   myBrowser->adjustColumnsWidth();
   myBrowser->setAutoUpdate( true );
   myBrowser->setUpdateModified( true );
-
 }
 
 HYDROGUI_Zone* HYDROGUI_CalculationDlg::getCurrentZone() const
@@ -524,4 +701,160 @@ void HYDROGUI_CalculationDlg::onDataChanged()
   SUIT_DataObject* aRoot = myBrowser->root();
   module()->getDataModel()->buildCaseTree( aRoot, myEditedObject );
   myBrowser->updateTree( aRoot );
-}
\ No newline at end of file
+}
+
+void HYDROGUI_CalculationDlg::setAvailableGroups( const QStringList& theGroups )
+{
+  myAvailableGroups->clear();
+  myGroups->clear();
+  foreach( QString aGroup, theGroups )
+    myAvailableGroups->addItem( aGroup );
+}
+
+QStringList HYDROGUI_CalculationDlg::getSelectedGroups() const
+{
+  return getSelected( myGroups );
+}
+
+QStringList HYDROGUI_CalculationDlg::getSelectedAvailableGroups() const
+{
+  return getSelected( myAvailableGroups );
+}
+
+void HYDROGUI_CalculationDlg::includeGroups( const QStringList& theObjects )
+{
+  moveItems( myAvailableGroups, myGroups, theObjects );
+}
+
+void HYDROGUI_CalculationDlg::excludeGroups( const QStringList& theObjects )
+{
+  moveItems( myGroups, myAvailableGroups, theObjects );
+}
+
+/**
+  Get creation mode.
+  @param theMode the mode
+*/
+int HYDROGUI_CalculationDlg::getMode() const
+{
+  return myModeButtons->checkedId();
+}
+
+/**
+  Set creation mode.
+  @param theMode the mode
+*/
+void HYDROGUI_CalculationDlg::setMode( int theMode )
+{
+  bool isBlocked = myModeButtons->blockSignals( true );
+  myModeButtons->button( theMode )->setChecked( true );
+  myModeButtons->blockSignals( isBlocked );
+
+  bool isAuto = ( theMode == HYDROData_CalculationCase::AUTOMATIC );
+
+  myGeomObjects->setOrderingEnabled( isAuto );
+  QWidget* aWidget = mySplitter->widget( 1 );
+  if ( aWidget ) {
+    aWidget->setVisible( isAuto );
+  }
+}
+
+/**
+  Enable/disable zones drag'n'drop and renaming.
+  @param theIsEnabled if true - zones drag'n'drop and renaming will be enabled
+*/
+void HYDROGUI_CalculationDlg::setEditZonesEnabled( const bool theIsEnabled )
+{
+  myBrowser->setReadOnly( !theIsEnabled );
+}
+
+/**
+  Get included geometry objects.
+  @return the list of geometry objects
+ */
+QList<Handle(HYDROData_Entity)> HYDROGUI_CalculationDlg::getGeometryObjects()
+{
+  QList<Handle(HYDROData_Entity)> anEntities = myGeomObjects->getObjects();
+  QList<Handle(HYDROData_Entity)> anObjects;
+
+  foreach ( Handle(HYDROData_Entity) anEntity, anEntities ) {
+    Handle(HYDROData_Object) anObj = Handle(HYDROData_Object)::DownCast( anEntity );
+    if ( anObj.IsNull() ) {
+      continue;
+    }
+
+    anObjects << anObj;
+  }
+
+  return anObjects;
+}
+
+/**
+  Get rules.
+  @return the list of rules
+ */
+HYDROData_ListOfRules HYDROGUI_CalculationDlg::getRules() const
+{
+  return myPriorityWidget->getRules();
+}
+
+/**
+  Set rules.
+  @param theRules the list of rules
+ */
+void  HYDROGUI_CalculationDlg::setRules( const HYDROData_ListOfRules& theRules ) const
+{
+  myPriorityWidget->setRules( theRules );
+}
+
+/**
+  Slot called when objects order is changed.
+ */
+void HYDROGUI_CalculationDlg::onOrderChanged()
+{
+  bool isConfirmed = true;
+  emit orderChanged( isConfirmed );
+  if( isConfirmed )
+    myPriorityWidget->setObjects( getGeometryObjects() );
+  else
+    myGeomObjects->undoLastMove();
+}
+
+/**
+  Slot called when priority rule for geometry objects is changed.
+ */
+void HYDROGUI_CalculationDlg::onRuleChanged()
+{
+  bool isConfirmed = true;
+  emit ruleChanged( isConfirmed );
+  if( !isConfirmed )
+    myPriorityWidget->undoLastChange();
+}
+
+void HYDROGUI_CalculationDlg::setStricklerTable( const QString& theStricklerTableName, bool theBlockSignals )
+{
+  bool isBlocked;
+  if ( theBlockSignals )
+    isBlocked = myStricklerTableName->blockSignals( true );
+  
+  myStricklerTableName->setCurrentIndex( myStricklerTableName->findText( theStricklerTableName ) );
+
+  if ( theBlockSignals )
+    myStricklerTableName->blockSignals( isBlocked );
+  else
+    emit StricklerTableSelected( theStricklerTableName );
+}
+
+void HYDROGUI_CalculationDlg::setLandCoverMap( const QString& theLandCoverMapName, bool theBlockSignals )
+{
+  bool isBlocked;
+  if ( theBlockSignals )
+    isBlocked = myLandCoverMapName->blockSignals( true );
+  
+  myLandCoverMapName->setCurrentIndex( myLandCoverMapName->findText( theLandCoverMapName ) );
+
+  if ( theBlockSignals )
+    myLandCoverMapName->blockSignals( isBlocked );
+  else
+    emit landCoverMapSelected( theLandCoverMapName );
+}