Salome HOME
13aea929f96392d576c5ec4709f4c996e39ffe91
[plugins/netgenplugin.git] / src / GUI / NETGENPluginGUI_HypothesisCreator.cxx
1 //  Copyright (C) 2007-2010  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 //  NETGENPlugin GUI: GUI for plugged-in mesher NETGENPlugin
24 //  File   : NETGENPluginGUI_HypothesisCreator.cxx
25 //  Author : Michael Zorin
26 //  Module : NETGENPlugin
27 //  $Header: 
28 //
29 #include "NETGENPluginGUI_HypothesisCreator.h"
30
31 #include <SMESHGUI_Utils.h>
32 #include <SMESHGUI_HypothesesUtils.h>
33 #include <SMESHGUI.h>
34 #include <SMESHGUI_SpinBox.h>
35
36 #include CORBA_SERVER_HEADER(NETGENPlugin_Algorithm)
37
38 #include <SUIT_Session.h>
39 #include <SUIT_ResourceMgr.h>
40
41 #include <SalomeApp_Tools.h>
42
43 #include <QComboBox>
44 #include <QLabel>
45 #include <QGroupBox>
46 #include <QFrame>
47 #include <QLayout>
48 #include <QLineEdit>
49 #include <QCheckBox>
50 #include <QPixmap>
51
52  enum Fineness
53    {
54      VeryCoarse,
55      Coarse,
56      Moderate,
57      Fine,
58      VeryFine,
59      UserDefined
60    };
61
62 NETGENPluginGUI_HypothesisCreator::NETGENPluginGUI_HypothesisCreator( const QString& theHypType )
63 : SMESHGUI_GenericHypothesisCreator( theHypType ),
64   myIs2D(false)
65 {
66 }
67
68 NETGENPluginGUI_HypothesisCreator::~NETGENPluginGUI_HypothesisCreator()
69 {
70 }
71
72 bool NETGENPluginGUI_HypothesisCreator::checkParams(QString& msg) const
73 {
74   NetgenHypothesisData data_old, data_new;
75   readParamsFromHypo( data_old );
76   readParamsFromWidgets( data_new );
77   bool res = storeParamsToHypo( data_new );
78   storeParamsToHypo( data_old );
79   
80   res = myMaxSize->isValid(msg,true) && res;
81   res = myGrowthRate->isValid(msg,true) && res; ;
82   res = myNbSegPerEdge->isValid(msg,true) && res;
83   res = myNbSegPerRadius->isValid(msg,true) && res;
84   return res;
85 }
86
87 QFrame* NETGENPluginGUI_HypothesisCreator::buildFrame()
88 {
89   QFrame* fr = new QFrame( 0 );
90   fr->setObjectName( "myframe" );
91   QVBoxLayout* lay = new QVBoxLayout( fr );
92   lay->setMargin( 5 );
93   lay->setSpacing( 0 );
94
95   QGroupBox* GroupC1 = new QGroupBox( tr( "SMESH_ARGUMENTS" ), fr );
96   lay->addWidget( GroupC1 );
97   
98   QGridLayout* aGroupLayout = new QGridLayout( GroupC1 );
99   aGroupLayout->setSpacing( 6 );
100   aGroupLayout->setMargin( 11 );
101   
102   int row = 0;
103   myName = 0;
104   if( isCreation() )
105   {
106     aGroupLayout->addWidget( new QLabel( tr( "SMESH_NAME" ), GroupC1 ), row, 0 );
107     myName = new QLineEdit( GroupC1 );
108     aGroupLayout->addWidget( myName, row, 1 );
109     row++;
110   }
111
112   aGroupLayout->addWidget( new QLabel( tr( "NETGEN_MAX_SIZE" ), GroupC1 ), row, 0 );
113   myMaxSize = new SMESHGUI_SpinBox( GroupC1 );
114   myMaxSize->RangeStepAndValidator( 1e-07, 1e+06, 10., "length_precision" );
115   aGroupLayout->addWidget( myMaxSize, row, 1 );
116   row++;
117   
118   mySecondOrder = new QCheckBox( tr( "NETGEN_SECOND_ORDER" ), GroupC1 );
119   aGroupLayout->addWidget( mySecondOrder, row, 0 );
120   row++;
121   
122   aGroupLayout->addWidget( new QLabel( tr( "NETGEN_FINENESS" ), GroupC1 ), row, 0 );
123   myFineness = new QComboBox( GroupC1 );
124   QStringList types;
125   types << tr( "NETGEN_VERYCOARSE" ) << tr( "NETGEN_COARSE" )   << tr( "NETGEN_MODERATE" ) <<
126            tr( "NETGEN_FINE" )       << tr( "NETGEN_VERYFINE" ) << tr( "NETGEN_CUSTOM" );
127   myFineness->addItems( types );
128   aGroupLayout->addWidget( myFineness, row, 1 );
129   row++;
130
131   aGroupLayout->addWidget( new QLabel( tr( "NETGEN_GROWTH_RATE" ), GroupC1 ), row, 0 );
132   myGrowthRate = new SMESHGUI_SpinBox( GroupC1 );
133   myGrowthRate->RangeStepAndValidator( .1, 10., .1, "parametric_precision" );
134   aGroupLayout->addWidget( myGrowthRate, row, 1 );
135   row++;
136
137   const double VALUE_MAX = 1.0e+6;
138
139   aGroupLayout->addWidget( new QLabel( tr( "NETGEN_SEG_PER_EDGE" ), GroupC1 ), row, 0 );
140   myNbSegPerEdge = new SMESHGUI_SpinBox( GroupC1 );
141   myNbSegPerEdge->RangeStepAndValidator( .2, VALUE_MAX, .1, "parametric_precision" );
142   aGroupLayout->addWidget( myNbSegPerEdge, row, 1 );
143   row++;
144   
145   aGroupLayout->addWidget( new QLabel( tr( "NETGEN_SEG_PER_RADIUS" ), GroupC1 ), row, 0 );
146   myNbSegPerRadius = new SMESHGUI_SpinBox( GroupC1 );
147   myNbSegPerRadius->RangeStepAndValidator( .2, VALUE_MAX, .1, "parametric_precision" );
148   aGroupLayout->addWidget( myNbSegPerRadius, row, 1 );
149   row++;
150
151   if ( hypType()=="NETGEN_Parameters_2D" )
152   {
153     myAllowQuadrangles = new QCheckBox( tr( "NETGEN_ALLOW_QUADRANGLES" ), GroupC1 );
154     aGroupLayout->addWidget( myAllowQuadrangles, row, 0 );
155     myIs2D = true;
156     row++;
157   }
158
159   myOptimize = new QCheckBox( tr( "NETGEN_OPTIMIZE" ), GroupC1 );
160   aGroupLayout->addWidget( myOptimize, row, 0 );
161   row++;
162   
163   connect( myFineness, SIGNAL( activated( int ) ), this, SLOT( onFinenessChanged() ) );
164   
165   return fr;
166 }
167
168 void NETGENPluginGUI_HypothesisCreator::retrieveParams() const
169 {
170   NetgenHypothesisData data;
171   readParamsFromHypo( data );
172
173   if( myName )
174     myName->setText( data.myName );
175   if(data.myMaxSizeVar.isEmpty())
176     myMaxSize->setValue( data.myMaxSize );
177   else
178     myMaxSize->setText( data.myMaxSizeVar );
179   
180   mySecondOrder->setChecked( data.mySecondOrder );
181   myOptimize->setChecked( data.myOptimize );
182   myFineness->setCurrentIndex( data.myFineness );
183
184   if(data.myGrowthRateVar.isEmpty())
185     myGrowthRate->setValue( data.myGrowthRate );
186   else
187     myGrowthRate->setText( data.myGrowthRateVar );
188
189   if(data.myNbSegPerEdgeVar.isEmpty())
190     myNbSegPerEdge->setValue( data.myNbSegPerEdge );
191   else
192     myNbSegPerEdge->setText( data.myNbSegPerEdgeVar );
193   
194   if(data.myNbSegPerRadiusVar.isEmpty())
195     myNbSegPerRadius->setValue( data.myNbSegPerRadius );
196   else
197     myNbSegPerRadius->setText( data.myNbSegPerRadiusVar );
198   
199   if (myIs2D)
200     myAllowQuadrangles->setChecked( data.myAllowQuadrangles );
201
202   // update widgets
203   bool isCustom = (myFineness->currentIndex() == UserDefined);
204   myGrowthRate->setEnabled(isCustom);
205   myNbSegPerEdge->setEnabled(isCustom);
206   myNbSegPerRadius->setEnabled(isCustom);
207 }
208
209 QString NETGENPluginGUI_HypothesisCreator::storeParams() const
210 {
211   NetgenHypothesisData data;
212   readParamsFromWidgets( data );
213   storeParamsToHypo( data );
214   
215   QString valStr = tr("NETGEN_MAX_SIZE") + " = " + QString::number( data.myMaxSize ) + "; ";
216   if ( data.mySecondOrder )
217     valStr +=  tr("NETGEN_SECOND_ORDER") + "; ";
218   if ( data.myOptimize )
219     valStr +=  tr("NETGEN_OPTIMIZE") + "; ";
220   valStr += myFineness->currentText() + "(" +  QString::number( data.myGrowthRate )     + ", " +
221                                                QString::number( data.myNbSegPerEdge )   + ", " +
222                                                QString::number( data.myNbSegPerRadius ) + ")";
223
224   if ( myIs2D && data.myAllowQuadrangles )
225     valStr += "; " + tr("NETGEN_ALLOW_QUADRANGLES");
226   
227   return valStr;
228 }
229
230 bool NETGENPluginGUI_HypothesisCreator::readParamsFromHypo( NetgenHypothesisData& h_data ) const
231 {
232   NETGENPlugin::NETGENPlugin_Hypothesis_var h =
233     NETGENPlugin::NETGENPlugin_Hypothesis::_narrow( initParamsHypothesis() );
234
235   HypothesisData* data = SMESH::GetHypothesisData( hypType() );
236   h_data.myName = isCreation() && data ? data->Label : "";
237
238   SMESH::ListOfParameters_var aParameters = h->GetLastParameters();
239
240   h_data.myMaxSize = h->GetMaxSize();
241   h_data.myMaxSizeVar = (aParameters->length() > 0) ? QString(aParameters[0].in()) : QString("");
242   h_data.mySecondOrder = h->GetSecondOrder();
243   h_data.myOptimize = h->GetOptimize();
244
245   h_data.myFineness = (int) h->GetFineness();
246   h_data.myGrowthRate = h->GetGrowthRate();
247   h_data.myGrowthRateVar = (aParameters->length() > 1) ? QString(aParameters[1].in()) : QString("");
248   h_data.myNbSegPerEdge = h->GetNbSegPerEdge();
249   h_data.myNbSegPerEdgeVar  = (aParameters->length() > 2) ? QString(aParameters[2].in()) : QString("");
250   h_data.myNbSegPerRadius = h->GetNbSegPerRadius();
251   h_data.myNbSegPerRadiusVar = (aParameters->length() > 3) ? QString(aParameters[3].in()) : QString("");
252
253   if ( myIs2D )
254     {
255       NETGENPlugin::NETGENPlugin_Hypothesis_2D_var h_2d =
256         NETGENPlugin::NETGENPlugin_Hypothesis_2D::_narrow( initParamsHypothesis() );
257
258       if ( !h_2d->_is_nil() )
259         h_data.myAllowQuadrangles = h_2d->GetQuadAllowed();
260     }
261   
262   return true;
263 }
264
265 bool NETGENPluginGUI_HypothesisCreator::storeParamsToHypo( const NetgenHypothesisData& h_data ) const
266 {
267   NETGENPlugin::NETGENPlugin_Hypothesis_var h =
268     NETGENPlugin::NETGENPlugin_Hypothesis::_narrow( hypothesis() );
269
270   bool ok = true;
271   try
272   {
273     if( isCreation() )
274       SMESH::SetName( SMESH::FindSObject( h ), h_data.myName.toLatin1().data() );
275     QStringList aVariablesList;
276     h->SetMaxSize( h_data.myMaxSize );
277     aVariablesList.append(h_data.myMaxSizeVar);
278     h->SetSecondOrder( h_data.mySecondOrder );
279     h->SetOptimize( h_data.myOptimize );
280     int fineness = h_data.myFineness;
281     h->SetFineness( fineness );
282
283     if( fineness==UserDefined )
284       {
285         h->SetGrowthRate( h_data.myGrowthRate );
286         h->SetNbSegPerEdge( h_data.myNbSegPerEdge );
287         h->SetNbSegPerRadius( h_data.myNbSegPerRadius );
288         
289         aVariablesList.append(h_data.myGrowthRateVar);
290         aVariablesList.append(h_data.myNbSegPerEdgeVar);
291         aVariablesList.append(h_data.myNbSegPerRadiusVar);
292       }
293     
294     if ( myIs2D )
295       {
296         NETGENPlugin::NETGENPlugin_Hypothesis_2D_var h_2d =
297           NETGENPlugin::NETGENPlugin_Hypothesis_2D::_narrow( h );
298         
299         if ( !h_2d->_is_nil() )
300           h_2d->SetQuadAllowed( h_data.myAllowQuadrangles );
301       }
302
303     h->SetParameters(aVariablesList.join(":").toLatin1().constData());
304     if( fineness==UserDefined )
305       {
306         h->SetParameters(aVariablesList.join(":").toLatin1().constData());
307         h->SetParameters(aVariablesList.join(":").toLatin1().constData());
308         h->SetParameters(aVariablesList.join(":").toLatin1().constData());
309       }
310         
311   }
312   catch(const SALOME::SALOME_Exception& ex)
313   {
314     SalomeApp_Tools::QtCatchCorbaException(ex);
315     ok = false;
316   }
317   return ok;
318 }
319
320 bool NETGENPluginGUI_HypothesisCreator::readParamsFromWidgets( NetgenHypothesisData& h_data ) const
321 {
322   h_data.myName           = myName ? myName->text() : "";
323   h_data.myMaxSize        = myMaxSize->value();
324   h_data.myMaxSizeVar     = myMaxSize->text();
325   h_data.mySecondOrder    = mySecondOrder->isChecked();
326   h_data.myOptimize       = myOptimize->isChecked();
327   h_data.myFineness       = myFineness->currentIndex();
328   h_data.myGrowthRate     = myGrowthRate->value();
329   h_data.myNbSegPerEdge   = myNbSegPerEdge->value();
330   h_data.myNbSegPerRadius = myNbSegPerRadius->value();
331
332   h_data.myGrowthRateVar     = myGrowthRate->text();
333   h_data.myNbSegPerEdgeVar   = myNbSegPerEdge->text();
334   h_data.myNbSegPerRadiusVar = myNbSegPerRadius->text();
335
336   
337   if ( myIs2D )
338     h_data.myAllowQuadrangles = myAllowQuadrangles->isChecked();
339   
340   return true;
341 }
342
343 void NETGENPluginGUI_HypothesisCreator::onFinenessChanged()
344 {
345   bool isCustom = (myFineness->currentIndex() == UserDefined);
346   
347   myGrowthRate->setEnabled(isCustom);
348   myNbSegPerEdge->setEnabled(isCustom);
349   myNbSegPerRadius->setEnabled(isCustom);
350
351   if (!isCustom)
352     {
353       double aGrowthRate, aNbSegPerEdge, aNbSegPerRadius;
354       
355       switch ( myFineness->currentIndex() )
356         {
357         case VeryCoarse:
358           aGrowthRate = 0.7;
359           aNbSegPerEdge = 0.3;
360           aNbSegPerRadius = 1;
361           break;
362         case Coarse:
363           aGrowthRate = 0.5;
364           aNbSegPerEdge = 0.5;
365           aNbSegPerRadius = 1.5;
366           break;
367         case Fine:
368           aGrowthRate = 0.2;
369           aNbSegPerEdge = 2;
370           aNbSegPerRadius = 3;
371           break;
372         case VeryFine:
373           aGrowthRate = 0.1;
374           aNbSegPerEdge = 3;
375           aNbSegPerRadius = 5;
376           break;
377         case Moderate:
378         default:
379           aGrowthRate = 0.3;
380           aNbSegPerEdge = 1;
381           aNbSegPerRadius = 2;
382           break;
383         }
384       
385       myGrowthRate->setValue( aGrowthRate );
386       myNbSegPerEdge->setValue( aNbSegPerEdge );
387       myNbSegPerRadius->setValue( aNbSegPerRadius );
388     }
389 }
390
391 QString NETGENPluginGUI_HypothesisCreator::caption() const
392 {
393   return tr( QString( "NETGEN_%1_TITLE" ).arg(myIs2D?QString("2D"):QString("3D")).toLatin1().data() );
394 }
395
396 QPixmap NETGENPluginGUI_HypothesisCreator::icon() const
397 {
398   QString hypIconName = tr( QString("ICON_DLG_NETGEN_PARAMETERS%1").arg(myIs2D?QString("_2D"):QString("")).toLatin1().data() );
399   return SUIT_Session::session()->resourceMgr()->loadPixmap( "NETGENPlugin", hypIconName );
400 }
401
402 QString NETGENPluginGUI_HypothesisCreator::type() const
403 {
404   return tr( QString( "NETGEN_%1_HYPOTHESIS" ).arg(myIs2D?QString("2D"):QString("3D")).toLatin1().data() );
405 }
406
407 QString NETGENPluginGUI_HypothesisCreator::helpPage() const
408 {
409   return "netgen_2d_3d_hypo_page.html";
410 }