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
  62 HYDROGUI_CalculationDlg::HYDROGUI_CalculationDlg( HYDROGUI_Module* theModule, const QString& theTitle )
  63 : HYDROGUI_Wizard( theModule, theTitle )
  64 {
  65   QWizard* aWizard = wizard();
  66   aWizard->addPage( createObjectsPage() );
  67   aWizard->addPage( createZonesPage() );
  68   connect( aWizard->button( QWizard::NextButton ), SIGNAL( clicked() ), SIGNAL( splitZones() ) );
  69   aWizard->show();
  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( wizard() );
  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() ), this, SLOT( onEmptyName() ) );
  92   connect( myValidator, SIGNAL( alreadyExists( QString ) ), this, 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   QFrame* aBtnsFrame = new QFrame( aPage );
 101   QVBoxLayout* aBtnsLayout = new QVBoxLayout( aBtnsFrame );
 102   aBtnsLayout->setMargin( 5 );
 103   aBtnsLayout->setSpacing( 5 );
 104   aBtnsFrame->setLayout( aBtnsLayout );
 105   QPushButton* anAddBtn = new QPushButton( tr("ADD"), aBtnsFrame );
 106   QPushButton* aRemoveBtn = new QPushButton( tr("REMOVE"), aBtnsFrame );
 107   aBtnsLayout->addWidget( anAddBtn );
 108   aBtnsLayout->addWidget( aRemoveBtn );
 109   aBtnsLayout->addStretch( 1 );
 110
 111   QLabel* aNameLabel = new QLabel( tr( "NAME" ), aPage );
 112   QLabel* anObjectsLabel = new QLabel( tr( "CALCULATION_REFERENCE_OBJECTS" ), aPage );
 113
 114   QGridLayout* aZonesLayout = new QGridLayout( aPage );
 115   aZonesLayout->setMargin( 5 );
 116   aZonesLayout->setSpacing( 5 );
 117   aZonesLayout->setVerticalSpacing( 10 );
 118   aZonesLayout->addWidget( aNameLabel,     0, 0, Qt::AlignHCenter );
 119   aZonesLayout->addWidget( myObjectName,   0, 1 );
 120   aZonesLayout->addWidget( anObjectsLabel, 1, 0, Qt::AlignHCenter );
 121   aZonesLayout->addWidget( myGeomObjects,  1, 1, 2, 1 );
 122   aZonesLayout->addWidget( aBtnsFrame,     2, 0, Qt::AlignHCenter );
 123
 124   aPage->setLayout( aZonesLayout );
 125
 126   connect( anAddBtn, SIGNAL( clicked() ), this, SIGNAL( addObjects() ) );
 127   connect( aRemoveBtn, SIGNAL( clicked() ), this, SIGNAL( removeObjects() ) );
 128
 129   return aPage;
 130 }
 131
 132 void HYDROGUI_CalculationDlg::onEmptyName()
 133 {
 134   QString aTitle = QObject::tr( "INSUFFICIENT_INPUT_DATA" );
 135   QString aMessage = QObject::tr( "INCORRECT_OBJECT_NAME" );
 136   SUIT_MessageBox::critical( module()->getApp()->desktop(), aTitle, aMessage );
 137 }
 138
 139 void HYDROGUI_CalculationDlg::onAlreadyExists( QString theName )
 140 {
 141   QString aTitle = QObject::tr( "INSUFFICIENT_INPUT_DATA" );
 142   QString aMessage = QObject::tr( "OBJECT_EXISTS_IN_DOCUMENT" ).arg( theName );
 143   SUIT_MessageBox::critical( module()->getApp()->desktop(), aTitle, aMessage );
 144 }
 145
 146 QWizardPage* HYDROGUI_CalculationDlg::createZonesPage() {
 147   QWizardPage* aPage = new QWizardPage( wizard() );
 148   QFrame* aFrame = new QFrame( aPage );
 149
 150   QGridLayout* aLayout = new QGridLayout( aPage );
 151
 152   myBrowser = new HYDROGUI_DataBrowser( module(), 0, aPage );
 153   myBrowser->setAutoOpenLevel( 3 );
 154   aLayout->setMargin( 5 );
 155   aLayout->setSpacing( 5 );
 156
 157   aLayout->addWidget( myBrowser, 0, 0, 1, 2 );
 158
 159   myBatimetryLabel = new QLabel( tr( "BATHYMETRY" ), aFrame );
 160   myBathymetryChoice = new QComboBox( aFrame );
 161
 162   myBathymetryChoice->setVisible( false );
 163   myBatimetryLabel->setVisible( false );
 164
 165   aLayout->addWidget( myBatimetryLabel, 1, 0 );
 166   aLayout->addWidget( myBathymetryChoice, 1, 1 );
 167
 168   aPage->setLayout( aLayout );
 169
 170   connect( myBrowser, SIGNAL( clicked( SUIT_DataObject* ) ), SLOT( onSelected( SUIT_DataObject* ) ) );
 171   connect( myBathymetryChoice, SIGNAL( activated( int ) ), SLOT( onMergeTypeSelected( int ) ) );
 172   connect( myBrowser,
 173       SIGNAL( dropped( const QList<SUIT_DataObject*>&, SUIT_DataObject*, int, Qt::DropAction ) ),
 174       SLOT( onZonesDropped( const QList<SUIT_DataObject*>&, SUIT_DataObject*, int, Qt::DropAction ) ) );
 175
 176   return aPage;
 177 }
 178
 179 void HYDROGUI_CalculationDlg::onZonesDropped( const QList<SUIT_DataObject*>& theList,
 180     SUIT_DataObject* theTargetParent, int theTargetRow, Qt::DropAction theDropAction )
 181 {
 182   QList<SUIT_DataObject*> aZonesList;
 183   HYDROGUI_Zone* aZone;
 184   // Get a list of dropped zones
 185   for ( int i = 0; i < theList.length(); i++ )
 186   {
 187     aZone = dynamic_cast<HYDROGUI_Zone*>( theList.at( i ) );
 188     if ( aZone )
 189     {
 190       aZonesList.append( aZone );
 191     }
 192   }
 193   if ( aZonesList.length() > 0 )
 194   {
 195     // Get the target region
 196     HYDROGUI_NamedObject* aRegionsRoot = dynamic_cast<HYDROGUI_NamedObject*>(theTargetParent);
 197     if ( aRegionsRoot )
 198     {
 199       // Create a new region
 200     }
 201     else
 202     {
 203       HYDROGUI_Region* aRegion = dynamic_cast<HYDROGUI_Region*>(theTargetParent);
 204       if ( aRegion )
 205       {
 206         aRegion->addZones( );
 207       }
 208     }
 209   }
 210 }
 211
 212 void HYDROGUI_CalculationDlg::onMergeTypeSelected( int theIndex )
 213 {
 214   int aType = myBathymetryChoice->itemData( theIndex ).toInt();
 215   QString aText = myBathymetryChoice->itemText( theIndex );
 216   emit setMergeType( aType, aText );
 217 }
 218
 219 void HYDROGUI_CalculationDlg::onSelected( SUIT_DataObject* theObject )
 220 {
 221   bool doShow = false;
 222   HYDROGUI_Zone* aZone = dynamic_cast<HYDROGUI_Zone*>( theObject );
 223   if ( aZone )
 224   {
 225     doShow = aZone->isMergingNeed();
 226   }
 227
 228   if ( doShow )
 229   {
 230     // Fill the merge type combo box
 231     bool prevBlock = myBathymetryChoice->blockSignals( true );
 232     myCurrentZone = aZone;
 233     myBathymetryChoice->clear();
 234     myBathymetryChoice->addItem( tr("MERGE_UNKNOWN"), HYDROData_Zone::Merge_UNKNOWN );
 235     myBathymetryChoice->addItem( tr("MERGE_ZMIN"), HYDROData_Zone::Merge_ZMIN );
 236     myBathymetryChoice->addItem( tr("MERGE_ZMAX"), HYDROData_Zone::Merge_ZMAX );
 237     QStringList aList = aZone->getBathymetries();
 238     for ( int i = 0; i < aList.length(); i++ )
 239     {
 240       myBathymetryChoice->addItem( aList.at( i ), HYDROData_Zone::Merge_Object );
 241     }
 242     // Select the current choice if any
 243     int aCurIndex = 0;
 244     switch ( aZone->getMergeType() )
 245     {
 246       case HYDROData_Zone::Merge_ZMIN:
 247         aCurIndex = 1;
 248         break;
 249       case HYDROData_Zone::Merge_ZMAX:
 250         aCurIndex = 2;
 251         break;
 252       case HYDROData_Zone::Merge_Object:
 253         aCurIndex = 3 + aList.indexOf( aZone->text( HYDROGUI_DataObject::BathymetryId ) );
 254         break;
 255       default:
 256         aCurIndex = 0; // Select unknown by default
 257     }
 258     myBathymetryChoice->setCurrentIndex( aCurIndex );
 259     myBathymetryChoice->blockSignals( prevBlock );
 260   }
 261
 262   myBathymetryChoice->setVisible( doShow );
 263   myBatimetryLabel->setVisible( doShow );
 264 }
 265
 266 void HYDROGUI_CalculationDlg::setObjectName( const QString& theName )
 267 {
 268   myObjectName->setText( theName );
 269 }
 270
 271 QString HYDROGUI_CalculationDlg::getObjectName() const
 272 {
 273   return myObjectName->text();
 274 }
 275
 276 void HYDROGUI_CalculationDlg::setGeomObjects( const QStringList& theObjects )
 277 {
 278 }
 279
 280 void HYDROGUI_CalculationDlg::setSelectedGeomObjects( const QStringList& theObjects )
 281 {
 282   myGeomObjects->clear();
 283
 284   for ( int i = 0, n = theObjects.length(); i < n; ++i )
 285   {
 286     QString anObjName = theObjects.at( i );
 287
 288     QListWidgetItem* aListItem = new QListWidgetItem( anObjName, myGeomObjects );
 289     aListItem->setFlags( Qt::ItemIsEnabled | Qt::ItemIsSelectable );
 290   }
 291 }
 292
 293 QStringList HYDROGUI_CalculationDlg::getSelectedGeomObjects() const
 294 {
 295   QStringList aResList;
 296   QList<QListWidgetItem*> aList = myGeomObjects->selectedItems();
 297   for ( int i = 0, n = aList.length(); i < n; ++i )
 298   {
 299     aResList.append( aList.at( i )->text() );
 300   }
 301   return aResList;
 302 }
 303
 304 void HYDROGUI_CalculationDlg::setEditedObject( const Handle(HYDROData_CalculationCase) theCase )
 305 {
 306   myEditedObject = theCase;
 307   myValidator->setEditedObject( theCase );
 308
 309   HYDROGUI_DataObject* anobj = new HYDROGUI_DataObject( 0, NULL, "" );
 310   myBrowser->setRoot(anobj);
 311
 312   LightApp_DataObject* aCaseItem = module()->getDataModel()->createObject( anobj,
 313                                      myEditedObject, "", true );
 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 }