src/HYDROGUI/HYDROGUI_CalculationDlg.cxx

   1 // Copyright (C) 2007-2013  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   5 //
   6 // This library is free software; you can redistribute it and/or
   7 // modify it under the terms of the GNU Lesser General Public
   8 // License as published by the Free Software Foundation; either
   9 // version 2.1 of the License.
  10 //
  11 // This library is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 // Lesser General Public License for more details.
  15 //
  16 // You should have received a copy of the GNU Lesser General Public
  17 // License along with this library; if not, write to the Free Software
  18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  19 //
  20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21 //
  22
  23 #include "HYDROGUI_CalculationDlg.h"
  24
  25 #include "HYDROGUI_ObjSelector.h"
  26 #include "HYDROGUI_Tool.h"
  27 #include "HYDROGUI_DataBrowser.h"
  28 #include "HYDROGUI_DataModel.h"
  29 #include "HYDROGUI_Module.h"
  30 #include "HYDROGUI_DataObject.h"
  31 #include "HYDROGUI_NameValidator.h"
  32 #include "HYDROGUI_Region.h"
  33 #include "HYDROGUI_Zone.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 <SUIT_Desktop.h>
  49 #include <SUIT_MessageBox.h>
  50
  51 #include <QGroupBox>
  52 #include <QLabel>
  53 #include <QLayout>
  54 #include <QLineEdit>
  55 #include <QListWidget>
  56 #include <QPicture>
  57 #include <QPushButton>
  58 #include <QToolButton>
  59 #include <QWizardPage>
  60 #include <QComboBox>
  61 #include <QStackedWidget>
  62
  63 HYDROGUI_CalculationDlg::HYDROGUI_CalculationDlg( HYDROGUI_Module* theModule, const QString& theTitle )
  64 : HYDROGUI_Wizard( theModule, theTitle )
  65 {
  66   addPage( createObjectsPage() );
  67   addPage( createZonesPage() );
  68   connect( button( QWizard::NextButton ), SIGNAL( clicked() ), SIGNAL( splitZones() ) );
  69   connect( button( QWizard::BackButton ), SIGNAL( clicked() ), SIGNAL( hideZones() ) );
  70 }
  71
  72 HYDROGUI_CalculationDlg::~HYDROGUI_CalculationDlg()
  73 {
  74 }
  75
  76 void HYDROGUI_CalculationDlg::reset()
  77 {
  78   myObjectName->clear();
  79   myGeomObjects->clear();
  80 }
  81
  82 QWizardPage* HYDROGUI_CalculationDlg::createObjectsPage() {
  83   QWizardPage* aPage = new QWizardPage( mainFrame() );
  84   QFrame* aFrame = new QFrame( aPage );
  85
  86   // Calculation name
  87   myObjectName = new QLineEdit( aPage );
  88   myValidator = new HYDROGUI_NameValidator(module(), myObjectName);
  89   myObjectName->setValidator( myValidator );
  90
  91   connect( myValidator, SIGNAL( emptyName() ), SLOT( onEmptyName() ) );
  92   connect( myValidator, SIGNAL( alreadyExists( QString ) ), SLOT( onAlreadyExists( QString ) ) );
  93
  94   myGeomObjects = new QListWidget( aPage );
  95   myGeomObjects->setSelectionMode( QListWidget::SingleSelection );
  96   myGeomObjects->setEditTriggers( QListWidget::NoEditTriggers );
  97   myGeomObjects->setViewMode( QListWidget::ListMode );
  98   myGeomObjects->setSortingEnabled( true );
  99
 100   connect( myGeomObjects, SIGNAL( currentTextChanged( const QString & ) ),
 101     SIGNAL( objectSelected( const QString & ) ) );
 102
 103   QFrame* aBtnsFrame = new QFrame( aPage );
 104   QVBoxLayout* aBtnsLayout = new QVBoxLayout( aBtnsFrame );
 105   aBtnsLayout->setMargin( 5 );
 106   aBtnsLayout->setSpacing( 5 );
 107   aBtnsFrame->setLayout( aBtnsLayout );
 108   QPushButton* anAddBtn = new QPushButton( tr("ADD"), aBtnsFrame );
 109   QPushButton* aRemoveBtn = new QPushButton( tr("REMOVE"), aBtnsFrame );
 110   aBtnsLayout->addWidget( anAddBtn );
 111   aBtnsLayout->addWidget( aRemoveBtn );
 112   aBtnsLayout->addStretch( 1 );
 113
 114   QLabel* aNameLabel = new QLabel( tr( "NAME" ), aPage );
 115   QLabel* anObjectsLabel = new QLabel( tr( "CALCULATION_REFERENCE_OBJECTS" ), aPage );
 116
 117   QGridLayout* aZonesLayout = new QGridLayout( aPage );
 118   aZonesLayout->setMargin( 5 );
 119   aZonesLayout->setSpacing( 5 );
 120   aZonesLayout->setVerticalSpacing( 10 );
 121   aZonesLayout->addWidget( aNameLabel,     0, 0, Qt::AlignHCenter );
 122   aZonesLayout->addWidget( myObjectName,   0, 1 );
 123   aZonesLayout->addWidget( anObjectsLabel, 1, 0, Qt::AlignHCenter );
 124   aZonesLayout->addWidget( myGeomObjects,  1, 1, 2, 1 );
 125   aZonesLayout->addWidget( aBtnsFrame,     2, 0, Qt::AlignHCenter );
 126
 127   aPage->setLayout( aZonesLayout );
 128
 129   connect( anAddBtn, SIGNAL( clicked() ), this, SIGNAL( addObjects() ) );
 130   connect( aRemoveBtn, SIGNAL( clicked() ), this, SIGNAL( removeObjects() ) );
 131
 132   return aPage;
 133 }
 134
 135 void HYDROGUI_CalculationDlg::onEmptyName()
 136 {
 137   QString aTitle = QObject::tr( "INSUFFICIENT_INPUT_DATA" );
 138   QString aMessage = QObject::tr( "INCORRECT_OBJECT_NAME" );
 139   SUIT_MessageBox::critical( module()->getApp()->desktop(), aTitle, aMessage );
 140 }
 141
 142 void HYDROGUI_CalculationDlg::onAlreadyExists( QString theName )
 143 {
 144   QString aTitle = QObject::tr( "INSUFFICIENT_INPUT_DATA" );
 145   QString aMessage = QObject::tr( "OBJECT_EXISTS_IN_DOCUMENT" ).arg( theName );
 146   SUIT_MessageBox::critical( module()->getApp()->desktop(), aTitle, aMessage );
 147 }
 148
 149 QWizardPage* HYDROGUI_CalculationDlg::createZonesPage() {
 150   QWizardPage* aPage = new QWizardPage( mainFrame() );
 151   QFrame* aFrame = new QFrame( aPage );
 152
 153   QGridLayout* aLayout = new QGridLayout( aPage );
 154
 155   myBrowser = new HYDROGUI_DataBrowser( module(), 0, aPage );
 156   myBrowser->setAutoOpenLevel( 3 );
 157   aLayout->setMargin( 5 );
 158   aLayout->setSpacing( 5 );
 159
 160   aLayout->addWidget( myBrowser, 0, 0, 1, 2 );
 161
 162   myBatimetryLabel = new QLabel( tr( "BATHYMETRY" ), aFrame );
 163   myBathymetryChoice = new QComboBox( aFrame );
 164
 165   myBathymetryChoice->setVisible( false );
 166   myBatimetryLabel->setVisible( false );
 167
 168   aLayout->addWidget( myBatimetryLabel, 1, 0 );
 169   aLayout->addWidget( myBathymetryChoice, 1, 1 );
 170
 171   aPage->setLayout( aLayout );
 172
 173   connect( myBrowser, SIGNAL( clicked( SUIT_DataObject* ) ), SIGNAL( clickedInZonesBrowser( SUIT_DataObject* ) ) );
 174   connect( myBrowser, SIGNAL( clicked( SUIT_DataObject* ) ), SLOT( onSelected( SUIT_DataObject* ) ) );
 175   connect( myBathymetryChoice, SIGNAL( activated( int ) ), SLOT( onMergeTypeSelected( int ) ) );
 176   connect( myBrowser,
 177       SIGNAL( dropped( const QList<SUIT_DataObject*>&, SUIT_DataObject*, int, Qt::DropAction ) ),
 178       SLOT( onZonesDropped( const QList<SUIT_DataObject*>&, SUIT_DataObject*, int, Qt::DropAction ) ) );
 179
 180   return aPage;
 181 }
 182
 183 void HYDROGUI_CalculationDlg::onZonesDropped( const QList<SUIT_DataObject*>& theList,
 184     SUIT_DataObject* theTargetParent, int theTargetRow, Qt::DropAction theDropAction )
 185 {
 186   QList<SUIT_DataObject*> aZonesList;
 187   HYDROGUI_Zone* aZone;
 188   // Get a list of dropped zones
 189   for ( int i = 0; i < theList.length(); i++ )
 190   {
 191     aZone = dynamic_cast<HYDROGUI_Zone*>( theList.at( i ) );
 192     if ( aZone )
 193     {
 194       aZonesList.append( aZone );
 195     }
 196   }
 197   if ( aZonesList.length() > 0 )
 198   {
 199     // Get the target region
 200     HYDROGUI_NamedObject* aRegionsRoot = dynamic_cast<HYDROGUI_NamedObject*>(theTargetParent);
 201     if ( aRegionsRoot )
 202     {
 203       // Create a new region
 204       emit createRegion( aZonesList );
 205     }
 206     else
 207     {
 208       HYDROGUI_Region* aRegion = dynamic_cast<HYDROGUI_Region*>(theTargetParent);
 209       if ( aRegion )
 210       {
 211         emit moveZones( theTargetParent, aZonesList );
 212       }
 213     }
 214   }
 215 }
 216
 217 void HYDROGUI_CalculationDlg::onMergeTypeSelected( int theIndex )
 218 {
 219   int aType = myBathymetryChoice->itemData( theIndex ).toInt();
 220   QString aText = myBathymetryChoice->itemText( theIndex );
 221   emit setMergeType( aType, aText );
 222 }
 223
 224 void HYDROGUI_CalculationDlg::onSelected( SUIT_DataObject* theObject )
 225 {
 226   bool doShow = false;
 227   HYDROGUI_Zone* aZone = dynamic_cast<HYDROGUI_Zone*>( theObject );
 228   if ( aZone )
 229   {
 230     doShow = aZone->isMergingNeed();
 231   }
 232
 233   if ( doShow )
 234   {
 235     // Fill the merge type combo box
 236     bool prevBlock = myBathymetryChoice->blockSignals( true );
 237     myCurrentZone = aZone;
 238     myBathymetryChoice->clear();
 239     myBathymetryChoice->addItem( tr("MERGE_UNKNOWN"), HYDROData_Zone::Merge_UNKNOWN );
 240     myBathymetryChoice->addItem( tr("MERGE_ZMIN"), HYDROData_Zone::Merge_ZMIN );
 241     myBathymetryChoice->addItem( tr("MERGE_ZMAX"), HYDROData_Zone::Merge_ZMAX );
 242     QStringList aList = aZone->getBathymetries();
 243     for ( int i = 0; i < aList.length(); i++ )
 244     {
 245       myBathymetryChoice->addItem( aList.at( i ), HYDROData_Zone::Merge_Object );
 246     }
 247     // Select the current choice if any
 248     int aCurIndex = 0;
 249     switch ( aZone->getMergeType() )
 250     {
 251       case HYDROData_Zone::Merge_ZMIN:
 252         aCurIndex = 1;
 253         break;
 254       case HYDROData_Zone::Merge_ZMAX:
 255         aCurIndex = 2;
 256         break;
 257       case HYDROData_Zone::Merge_Object:
 258         aCurIndex = 3 + aList.indexOf( aZone->text( HYDROGUI_DataObject::BathymetryId ) );
 259         break;
 260       default:
 261         aCurIndex = 0; // Select unknown by default
 262     }
 263     myBathymetryChoice->setCurrentIndex( aCurIndex );
 264     myBathymetryChoice->blockSignals( prevBlock );
 265   }
 266
 267   myBathymetryChoice->setVisible( doShow );
 268   myBatimetryLabel->setVisible( doShow );
 269 }
 270
 271 void HYDROGUI_CalculationDlg::setObjectName( const QString& theName )
 272 {
 273   myObjectName->setText( theName );
 274 }
 275
 276 QString HYDROGUI_CalculationDlg::getObjectName() const
 277 {
 278   return myObjectName->text();
 279 }
 280
 281 void HYDROGUI_CalculationDlg::setGeomObjects( const QStringList& theObjects )
 282 {
 283   myGeomObjects->clear();
 284
 285   for ( int i = 0, n = theObjects.length(); i < n; ++i )
 286   {
 287     QString anObjName = theObjects.at( i );
 288
 289     QListWidgetItem* aListItem = new QListWidgetItem( anObjName, myGeomObjects );
 290     aListItem->setFlags( Qt::ItemIsEnabled | Qt::ItemIsSelectable );
 291   }
 292 }
 293
 294 QStringList HYDROGUI_CalculationDlg::getSelectedGeomObjects() const
 295 {
 296   QStringList aResList;
 297   QList<QListWidgetItem*> aList = myGeomObjects->selectedItems();
 298   for ( int i = 0, n = aList.length(); i < n; ++i )
 299   {
 300     aResList.append( aList.at( i )->text() );
 301   }
 302   return aResList;
 303 }
 304
 305 void HYDROGUI_CalculationDlg::setEditedObject( const Handle(HYDROData_CalculationCase) theCase )
 306 {
 307   myEditedObject = theCase;
 308   myValidator->setEditedObject( theCase );
 309
 310   HYDROGUI_DataObject* anobj = new HYDROGUI_DataObject( 0, NULL, "" );
 311   myBrowser->setRoot(anobj);
 312
 313   module()->getDataModel()->buildCaseTree( anobj, myEditedObject );
 314   myBrowser->updateTree();
 315   myBrowser->openLevels();
 316   myBrowser->adjustColumnsWidth();
 317 }
 318
 319 HYDROGUI_Zone* HYDROGUI_CalculationDlg::getCurrentZone() const
 320 {
 321   return myCurrentZone;
 322 }
 323
 324 void HYDROGUI_CalculationDlg::refreshZonesBrowser()
 325 {
 326   myBrowser->updateTree();
 327 }