Salome HOME
- Geometry objects selected in Calculation case wizard are highlighted now in the...
[modules/hydro.git] / 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   connect( aWizard->button( QWizard::BackButton ), SIGNAL( clicked() ), SIGNAL( hideZones() ) );
70   
71   aWizard->show();
72 }
73
74 HYDROGUI_CalculationDlg::~HYDROGUI_CalculationDlg()
75 {
76 }
77
78 void HYDROGUI_CalculationDlg::reset()
79 {
80   myObjectName->clear();
81   myGeomObjects->clear();
82 }
83
84 QWizardPage* HYDROGUI_CalculationDlg::createObjectsPage() {
85   QWizardPage* aPage = new QWizardPage( wizard() );
86   QFrame* aFrame = new QFrame( aPage );
87
88   // Calculation name
89   myObjectName = new QLineEdit( aPage );
90   myValidator = new HYDROGUI_NameValidator(module(), myObjectName);
91   myObjectName->setValidator( myValidator );
92
93   connect( myValidator, SIGNAL( emptyName() ), SLOT( onEmptyName() ) );
94   connect( myValidator, SIGNAL( alreadyExists( QString ) ), SLOT( onAlreadyExists( QString ) ) );
95
96   myGeomObjects = new QListWidget( aPage );
97   myGeomObjects->setSelectionMode( QListWidget::SingleSelection );
98   myGeomObjects->setEditTriggers( QListWidget::NoEditTriggers );
99   myGeomObjects->setViewMode( QListWidget::ListMode );
100   myGeomObjects->setSortingEnabled( true );
101
102   connect( myGeomObjects, SIGNAL( currentTextChanged( const QString & ) ), 
103     SIGNAL( objectSelected( const QString & ) ) );
104
105   QFrame* aBtnsFrame = new QFrame( aPage );
106   QVBoxLayout* aBtnsLayout = new QVBoxLayout( aBtnsFrame );
107   aBtnsLayout->setMargin( 5 );
108   aBtnsLayout->setSpacing( 5 );
109   aBtnsFrame->setLayout( aBtnsLayout );
110   QPushButton* anAddBtn = new QPushButton( tr("ADD"), aBtnsFrame );
111   QPushButton* aRemoveBtn = new QPushButton( tr("REMOVE"), aBtnsFrame );
112   aBtnsLayout->addWidget( anAddBtn );
113   aBtnsLayout->addWidget( aRemoveBtn );
114   aBtnsLayout->addStretch( 1 );
115
116   QLabel* aNameLabel = new QLabel( tr( "NAME" ), aPage );
117   QLabel* anObjectsLabel = new QLabel( tr( "CALCULATION_REFERENCE_OBJECTS" ), aPage );
118
119   QGridLayout* aZonesLayout = new QGridLayout( aPage );
120   aZonesLayout->setMargin( 5 );
121   aZonesLayout->setSpacing( 5 );
122   aZonesLayout->setVerticalSpacing( 10 );
123   aZonesLayout->addWidget( aNameLabel,     0, 0, Qt::AlignHCenter );
124   aZonesLayout->addWidget( myObjectName,   0, 1 );
125   aZonesLayout->addWidget( anObjectsLabel, 1, 0, Qt::AlignHCenter );
126   aZonesLayout->addWidget( myGeomObjects,  1, 1, 2, 1 );
127   aZonesLayout->addWidget( aBtnsFrame,     2, 0, Qt::AlignHCenter );
128
129   aPage->setLayout( aZonesLayout );
130
131   connect( anAddBtn, SIGNAL( clicked() ), this, SIGNAL( addObjects() ) );
132   connect( aRemoveBtn, SIGNAL( clicked() ), this, SIGNAL( removeObjects() ) );
133
134   return aPage;
135 }
136
137 void HYDROGUI_CalculationDlg::onEmptyName()
138 {
139   QString aTitle = QObject::tr( "INSUFFICIENT_INPUT_DATA" );
140   QString aMessage = QObject::tr( "INCORRECT_OBJECT_NAME" );
141   SUIT_MessageBox::critical( module()->getApp()->desktop(), aTitle, aMessage );
142 }
143
144 void HYDROGUI_CalculationDlg::onAlreadyExists( QString theName )
145 {
146   QString aTitle = QObject::tr( "INSUFFICIENT_INPUT_DATA" );
147   QString aMessage = QObject::tr( "OBJECT_EXISTS_IN_DOCUMENT" ).arg( theName );
148   SUIT_MessageBox::critical( module()->getApp()->desktop(), aTitle, aMessage );
149 }
150
151 QWizardPage* HYDROGUI_CalculationDlg::createZonesPage() {
152   QWizardPage* aPage = new QWizardPage( wizard() );
153   QFrame* aFrame = new QFrame( aPage );
154
155   QGridLayout* aLayout = new QGridLayout( aPage );
156   
157   myBrowser = new HYDROGUI_DataBrowser( module(), 0, aPage );
158   myBrowser->setAutoOpenLevel( 3 );
159   aLayout->setMargin( 5 );
160   aLayout->setSpacing( 5 );
161
162   aLayout->addWidget( myBrowser, 0, 0, 1, 2 );
163
164   myBatimetryLabel = new QLabel( tr( "BATHYMETRY" ), aFrame );
165   myBathymetryChoice = new QComboBox( aFrame );
166
167   myBathymetryChoice->setVisible( false );
168   myBatimetryLabel->setVisible( false );
169
170   aLayout->addWidget( myBatimetryLabel, 1, 0 );
171   aLayout->addWidget( myBathymetryChoice, 1, 1 );
172
173   aPage->setLayout( aLayout );
174
175   connect( myBrowser, SIGNAL( clicked( SUIT_DataObject* ) ), SIGNAL( clickedInZonesBrowser( SUIT_DataObject* ) ) );
176   connect( myBrowser, SIGNAL( clicked( SUIT_DataObject* ) ), SLOT( onSelected( SUIT_DataObject* ) ) );
177   connect( myBathymetryChoice, SIGNAL( activated( int ) ), SLOT( onMergeTypeSelected( int ) ) );
178   connect( myBrowser, 
179       SIGNAL( dropped( const QList<SUIT_DataObject*>&, SUIT_DataObject*, int, Qt::DropAction ) ),
180       SLOT( onZonesDropped( const QList<SUIT_DataObject*>&, SUIT_DataObject*, int, Qt::DropAction ) ) );
181
182   return aPage;
183 }
184
185 void HYDROGUI_CalculationDlg::onZonesDropped( const QList<SUIT_DataObject*>& theList, 
186     SUIT_DataObject* theTargetParent, int theTargetRow, Qt::DropAction theDropAction )
187 {
188   QList<SUIT_DataObject*> aZonesList;
189   HYDROGUI_Zone* aZone;
190   // Get a list of dropped zones
191   for ( int i = 0; i < theList.length(); i++ )
192   {
193     aZone = dynamic_cast<HYDROGUI_Zone*>( theList.at( i ) );
194     if ( aZone )
195     {
196       aZonesList.append( aZone );
197     }
198   }
199   if ( aZonesList.length() > 0 )
200   {
201     // Get the target region
202     HYDROGUI_NamedObject* aRegionsRoot = dynamic_cast<HYDROGUI_NamedObject*>(theTargetParent);
203     if ( aRegionsRoot )
204     {
205       // Create a new region
206       emit createRegion( aZonesList );
207     }
208     else
209     {
210       HYDROGUI_Region* aRegion = dynamic_cast<HYDROGUI_Region*>(theTargetParent);
211       if ( aRegion )
212       {
213         emit moveZones( theTargetParent, aZonesList );
214       }
215     }
216   }
217 }
218
219 void HYDROGUI_CalculationDlg::onMergeTypeSelected( int theIndex )
220 {
221   int aType = myBathymetryChoice->itemData( theIndex ).toInt();
222   QString aText = myBathymetryChoice->itemText( theIndex );
223   emit setMergeType( aType, aText );
224 }
225
226 void HYDROGUI_CalculationDlg::onSelected( SUIT_DataObject* theObject )
227 {
228   bool doShow = false;
229   HYDROGUI_Zone* aZone = dynamic_cast<HYDROGUI_Zone*>( theObject );
230   if ( aZone )
231   {
232     doShow = aZone->isMergingNeed();
233   }
234
235   if ( doShow )
236   {
237     // Fill the merge type combo box
238     bool prevBlock = myBathymetryChoice->blockSignals( true );
239     myCurrentZone = aZone;
240     myBathymetryChoice->clear();
241     myBathymetryChoice->addItem( tr("MERGE_UNKNOWN"), HYDROData_Zone::Merge_UNKNOWN );
242     myBathymetryChoice->addItem( tr("MERGE_ZMIN"), HYDROData_Zone::Merge_ZMIN );
243     myBathymetryChoice->addItem( tr("MERGE_ZMAX"), HYDROData_Zone::Merge_ZMAX );
244     QStringList aList = aZone->getBathymetries();
245     for ( int i = 0; i < aList.length(); i++ )
246     {
247       myBathymetryChoice->addItem( aList.at( i ), HYDROData_Zone::Merge_Object );
248     }
249     // Select the current choice if any
250     int aCurIndex = 0;
251     switch ( aZone->getMergeType() )
252     {
253       case HYDROData_Zone::Merge_ZMIN:
254         aCurIndex = 1;
255         break;
256       case HYDROData_Zone::Merge_ZMAX:
257         aCurIndex = 2;
258         break;
259       case HYDROData_Zone::Merge_Object:
260         aCurIndex = 3 + aList.indexOf( aZone->text( HYDROGUI_DataObject::BathymetryId ) );
261         break;
262       default:
263         aCurIndex = 0; // Select unknown by default
264     }
265     myBathymetryChoice->setCurrentIndex( aCurIndex );
266     myBathymetryChoice->blockSignals( prevBlock );
267   }
268
269   myBathymetryChoice->setVisible( doShow );
270   myBatimetryLabel->setVisible( doShow );
271 }
272
273 void HYDROGUI_CalculationDlg::setObjectName( const QString& theName )
274 {
275   myObjectName->setText( theName );
276 }
277
278 QString HYDROGUI_CalculationDlg::getObjectName() const
279 {
280   return myObjectName->text();
281 }
282
283 void HYDROGUI_CalculationDlg::setGeomObjects( const QStringList& theObjects )
284 {
285   myGeomObjects->clear();
286
287   for ( int i = 0, n = theObjects.length(); i < n; ++i )
288   {
289     QString anObjName = theObjects.at( i );
290
291     QListWidgetItem* aListItem = new QListWidgetItem( anObjName, myGeomObjects );
292     aListItem->setFlags( Qt::ItemIsEnabled | Qt::ItemIsSelectable );
293   }
294 }
295
296 QStringList HYDROGUI_CalculationDlg::getSelectedGeomObjects() const
297 {
298   QStringList aResList;
299   QList<QListWidgetItem*> aList = myGeomObjects->selectedItems();
300   for ( int i = 0, n = aList.length(); i < n; ++i )
301   {
302     aResList.append( aList.at( i )->text() );
303   }
304   return aResList;
305 }
306
307 void HYDROGUI_CalculationDlg::setEditedObject( const Handle(HYDROData_CalculationCase) theCase )
308 {
309   myEditedObject = theCase;
310   myValidator->setEditedObject( theCase );
311
312   HYDROGUI_DataObject* anobj = new HYDROGUI_DataObject( 0, NULL, "" );
313   myBrowser->setRoot(anobj);
314
315   module()->getDataModel()->buildCaseTree( anobj, myEditedObject );
316   myBrowser->updateTree();
317   myBrowser->openLevels();
318   myBrowser->adjustColumnsWidth();
319 }
320
321 HYDROGUI_Zone* HYDROGUI_CalculationDlg::getCurrentZone() const
322 {
323   return myCurrentZone;
324 }
325
326 void HYDROGUI_CalculationDlg::refreshZonesBrowser()
327 {
328   myBrowser->updateTree();
329 }