Feature #86: The hierarchy in the Object Browser (T 19).

[modules/hydro.git] / src / HYDROGUI / HYDROGUI_CalculationDlg.cxx

diff --git a/src/HYDROGUI/HYDROGUI_CalculationDlg.cxx b/src/HYDROGUI/HYDROGUI_CalculationDlg.cxx
index 9b53a2499e4713e20e8ee912a51c05189650f139..42ba2cbf9f7c76f1ecea73970530269d3e36d58d 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_CalculationDlg.cxx
+++ b/src/HYDROGUI/HYDROGUI_CalculationDlg.cxx
@@ -77,6 +77,8 @@ void HYDROGUI_CalculationDlg::reset()
 {
   myObjectName->clear();
   myGeomObjects->clear();
+  myPolylineName->clear();
+  myAvailableGeomObjects->clear();
 }
 
 QWizardPage* HYDROGUI_CalculationDlg::createObjectsPage() {
@@ -91,43 +93,73 @@ QWizardPage* HYDROGUI_CalculationDlg::createObjectsPage() {
   connect( myValidator, SIGNAL( emptyName() ), SLOT( onEmptyName() ) );
   connect( myValidator, SIGNAL( alreadyExists( QString ) ), SLOT( onAlreadyExists( QString ) ) );
 
+  myPolylineName = new QComboBox( aPage );
+  connect( myPolylineName, SIGNAL( activated( const QString & ) ), 
+    SIGNAL( boundarySelected( const QString & ) ) );
+
   myGeomObjects = new QListWidget( aPage );
   myGeomObjects->setSelectionMode( QListWidget::SingleSelection );
   myGeomObjects->setEditTriggers( QListWidget::NoEditTriggers );
   myGeomObjects->setViewMode( QListWidget::ListMode );
   myGeomObjects->setSortingEnabled( true );
 
+  myAvailableGeomObjects = new QListWidget( aPage );
+  myAvailableGeomObjects->setSelectionMode( QListWidget::SingleSelection );
+  myAvailableGeomObjects->setEditTriggers( QListWidget::NoEditTriggers );
+  myAvailableGeomObjects->setViewMode( QListWidget::ListMode );
+  myAvailableGeomObjects->setSortingEnabled( true );
+
   connect( myGeomObjects, SIGNAL( currentTextChanged( const QString & ) ), 
     SIGNAL( objectSelected( const QString & ) ) );
 
-  QFrame* aBtnsFrame = new QFrame( aPage );
+  QFrame* anObjectsFrame = new QFrame( aPage );
+  QGridLayout* anObjsLayout = new QGridLayout( anObjectsFrame );
+  anObjsLayout->setMargin( 5 );
+  anObjsLayout->setSpacing( 5 );
+  anObjectsFrame->setLayout( anObjsLayout );
+
+  QFrame* aBtnsFrame = new QFrame( anObjectsFrame );
   QVBoxLayout* aBtnsLayout = new QVBoxLayout( aBtnsFrame );
   aBtnsLayout->setMargin( 5 );
   aBtnsLayout->setSpacing( 5 );
   aBtnsFrame->setLayout( aBtnsLayout );
-  QPushButton* anAddBtn = new QPushButton( tr("ADD"), aBtnsFrame );
-  QPushButton* aRemoveBtn = new QPushButton( tr("REMOVE"), aBtnsFrame );
+  QPushButton* anAddBtn = new QPushButton( tr("INCLUDE"), aBtnsFrame );
+  QPushButton* aRemoveBtn = new QPushButton( tr("EXCLUDE"), aBtnsFrame );
+
+  // 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
+  anObjsLayout->addWidget( anObjectsLabel, 0, 0, Qt::AlignHCenter );
+  anObjsLayout->addWidget( anIncludedLabel, 0, 2, Qt::AlignHCenter );
+  anObjsLayout->addWidget( myAvailableGeomObjects, 1, 0, Qt::AlignHCenter );
+  anObjsLayout->addWidget( aBtnsFrame, 1, 1, Qt::AlignHCenter );
+  anObjsLayout->addWidget( myGeomObjects, 1, 2, Qt::AlignHCenter );
+
+
   QLabel* aNameLabel = new QLabel( tr( "NAME" ), aPage );
-  QLabel* anObjectsLabel = new QLabel( tr( "CALCULATION_REFERENCE_OBJECTS" ), aPage );
+  QLabel* aLimitsLabel = new QLabel( tr( "LIMITS" ), aPage );
 
-  QGridLayout* aZonesLayout = new QGridLayout( aPage );
-  aZonesLayout->setMargin( 5 );
-  aZonesLayout->setSpacing( 5 );
-  aZonesLayout->setVerticalSpacing( 10 );
-  aZonesLayout->addWidget( aNameLabel,     0, 0, Qt::AlignHCenter );
-  aZonesLayout->addWidget( myObjectName,   0, 1 );
-  aZonesLayout->addWidget( anObjectsLabel, 1, 0, Qt::AlignHCenter );
-  aZonesLayout->addWidget( myGeomObjects,  1, 1, 2, 1 );
-  aZonesLayout->addWidget( aBtnsFrame,     2, 0, Qt::AlignHCenter );
+  // Fill the page
+  QGridLayout* aPageLayout = new QGridLayout( aPage );
+  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 );
 
-  aPage->setLayout( aZonesLayout );
+  aPage->setLayout( aPageLayout );
 
-  connect( anAddBtn, SIGNAL( clicked() ), this, SIGNAL( addObjects() ) );
-  connect( aRemoveBtn, SIGNAL( clicked() ), this, SIGNAL( removeObjects() ) );
+  connect( anAddBtn, SIGNAL( clicked() ), SIGNAL( addObjects() ) );
+  connect( aRemoveBtn, SIGNAL( clicked() ), SIGNAL( removeObjects() ) );
 
   return aPage;
 }
@@ -278,16 +310,76 @@ QString HYDROGUI_CalculationDlg::getObjectName() const
   return myObjectName->text();
 }
 
-void HYDROGUI_CalculationDlg::setGeomObjects( const QStringList& theObjects )
+void HYDROGUI_CalculationDlg::includeGeomObjects( const QStringList& theObjects )
 {
-  myGeomObjects->clear();
+  QList<QListWidgetItem*> aFoundItems;
+  int anIdx;
+  QListWidgetItem* anItem;
+
+  for ( int i = 0, n = theObjects.length(); i < n; ++i )
+  {
+    QString anObjName = theObjects.at( i );
+    aFoundItems = myAvailableGeomObjects->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 ) );
+      // Add the item to the included objects list
+      myGeomObjects->addItem( anItem );
+    }
+  }
+}
+
+void HYDROGUI_CalculationDlg::excludeGeomObjects( const QStringList& theObjects )
+{
+  QList<QListWidgetItem*> aFoundItems;
+  int anIdx;
+  QListWidgetItem* anItem;
+
+  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 );
+    }
+  }
+}
+
+void HYDROGUI_CalculationDlg::setBoundary( const QString& theObjName )
+{
+  bool isBlocked = myPolylineName->blockSignals( true );
+  myPolylineName->setCurrentIndex( myPolylineName->findText( theObjName ) );
+  myPolylineName->blockSignals( isBlocked );
+}
+
+void HYDROGUI_CalculationDlg::setPolylineNames( const QStringList& theObjects, const QStringList& theObjectsEntries )
+{
+  myPolylineName->clear();
+  myPolylineName->addItem( "", "" ); // No boundary item
+
+  for ( int i = 0, n = theObjects.length(); i < n; ++i )
+  {
+    myPolylineName->addItem( theObjects.at( i ), theObjectsEntries.at( i ) );
+  }
+}
+void HYDROGUI_CalculationDlg::setAllGeomObjects( const QStringList& theObjects, const QStringList& theObjectsEntries )
+{
+  myAvailableGeomObjects->clear();
 
   for ( int i = 0, n = theObjects.length(); i < n; ++i )
   {
     QString anObjName = theObjects.at( i );
 
-    QListWidgetItem* aListItem = new QListWidgetItem( anObjName, myGeomObjects );
+    QListWidgetItem* aListItem = new QListWidgetItem( anObjName, myAvailableGeomObjects );
     aListItem->setFlags( Qt::ItemIsEnabled | Qt::ItemIsSelectable );
+    aListItem->setData( Qt::UserRole, theObjectsEntries.at( i ) );
   }
 }
 
@@ -302,6 +394,17 @@ QStringList HYDROGUI_CalculationDlg::getSelectedGeomObjects() const
   return aResList;
 }
 
+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;
+}
+
 void HYDROGUI_CalculationDlg::setEditedObject( const Handle(HYDROData_CalculationCase) theCase )
 {
   myEditedObject = theCase;