Salome HOME
Merge from V5_1_4_BR (5_1_4rc2) 09/06/2010
[modules/smesh.git] / src / StdMeshersGUI / StdMeshersGUI_DistrPreview.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 // File   : StdMeshersGUI_DistrPreview.cxx
24 // Author : Open CASCADE S.A.S.
25 // SMESH includes
26 //
27 #include "StdMeshersGUI_DistrPreview.h"
28
29 // Qwt includes
30 #include <qwt_plot_curve.h>
31 #include <qwt_plot_marker.h>
32 #include <qwt_plot_grid.h>
33 #include <qwt_symbol.h>
34 #include <qwt_legend.h>
35
36 // OCCT includes
37 #include <Expr_NamedUnknown.hxx>
38 #include <Expr_GeneralExpression.hxx>
39
40 #if (OCC_VERSION_MAJOR << 16 | OCC_VERSION_MINOR << 8 | OCC_VERSION_MAINTENANCE) > 0x060100
41 #define NO_CAS_CATCH
42 #endif
43
44 #include <Standard_Failure.hxx>
45
46 #ifdef NO_CAS_CATCH
47 #include <Standard_ErrorHandler.hxx>
48 #endif
49
50 #ifdef WIN32
51 # include <algorithm>
52 #endif
53 #include <math.h>
54 #include <limits>
55
56 #include <Basics_Utils.hxx>
57
58 StdMeshersGUI_DistrPreview::StdMeshersGUI_DistrPreview( QWidget* p, StdMeshers::StdMeshers_NumberOfSegments_ptr h )
59 : QwtPlot( p ),
60   myPoints( 50 ),
61   myIsTable( false ),
62   myVars( 1, 1 ),
63   myValues( 1, 1 ),
64   myConv( CUT_NEGATIVE ),
65   myIsDone( true ),
66   myNbSeg( 1 )
67 {
68   Kernel_Utils::Localizer loc;
69   myHypo = StdMeshers::StdMeshers_NumberOfSegments::_duplicate( h );
70   myVars.ChangeValue( 1 ) = new Expr_NamedUnknown( "t" );
71   myDensity = new QwtPlotCurve( QString() );
72   myDensity->attach( this );
73   myDistr = new QwtPlotCurve( QString() );
74   myDistr->attach( this );
75   myMsg = new QwtPlotMarker();
76   myMsg->attach( this );
77   myMsg->setValue( 0.5, 0.5 );
78   QwtText mt = myMsg->label();
79   mt.setBackgroundPen( QPen( Qt::red, 1 ) );
80   QFont f = mt.font();
81   f.setPointSize( 14 ); f.setBold( true );
82   mt.setFont( f );
83   myMsg->setLabel( mt );
84   myDensity->setPen( QPen( Qt::red, 1 ) );
85
86   QColor dc = Qt::blue;
87   myDistr->setPen( QPen( dc, 1 ) );
88   myDistr->setSymbol( QwtSymbol( QwtSymbol::XCross, QBrush( dc ), QPen( dc ), QSize( 5, 5 ) ) );
89
90   QwtLegend* l = legend();
91   if ( !l ) {
92     l = new QwtLegend( this );
93     l->setFrameStyle( QFrame::Box | QFrame::Sunken );
94   }
95   insertLegend( l, QwtPlot::BottomLegend );
96
97   myDensity->setTitle( tr( "SMESH_DENSITY_FUNC" ) );
98   myDistr->setTitle( tr( "SMESH_DISTR" ) );
99   
100   QwtPlotGrid* aGrid = new QwtPlotGrid();
101   QPen aMajPen = aGrid->majPen();
102   aMajPen.setStyle( Qt::DashLine );
103   aGrid->setPen( aMajPen );
104
105   aGrid->enableX( true );
106   aGrid->enableY( true );
107
108   aGrid->attach( this );
109 }
110
111 StdMeshersGUI_DistrPreview::~StdMeshersGUI_DistrPreview()
112 {
113 }
114
115 bool StdMeshersGUI_DistrPreview::isTableFunc() const
116 {
117   return myIsTable;
118 }
119
120 void StdMeshersGUI_DistrPreview::tableFunc( SMESH::double_array& f ) const
121 {
122   f = myTableFunc;
123 }
124
125 QString StdMeshersGUI_DistrPreview::function() const
126 {
127   return myFunction;
128 }
129
130 int StdMeshersGUI_DistrPreview::nbSeg() const
131 {
132   return myNbSeg;
133 }
134
135 int StdMeshersGUI_DistrPreview::pointsCount() const
136 {
137   return myPoints;
138 }
139
140 void StdMeshersGUI_DistrPreview::setConversion( Conversion conv, const bool upd )
141 {
142   myConv = conv;
143   if( upd )
144     update();
145 }
146
147 bool StdMeshersGUI_DistrPreview::setParams( const QString& func, const int nbSeg, const int points, const bool upd )
148 {
149   myIsTable = false;
150   myTableFunc = SMESH::double_array();
151   myFunction = func.isEmpty() ? "0" : func;
152   myPoints = points>0 ? points : 2;
153   myNbSeg = nbSeg>0 ? nbSeg : 1;
154   bool res = init( func );
155   if( upd )
156     update();
157   return res;
158 }
159
160 bool StdMeshersGUI_DistrPreview::setParams( const SMESH::double_array& f, const int nbSeg, const bool upd )
161 {
162   myIsTable = true;
163   myTableFunc = f;
164   if( myTableFunc.length()%2==1 )
165     myTableFunc.length( myTableFunc.length()-1 );
166
167   myFunction = "0";
168   myPoints = myTableFunc.length()/2;
169   myNbSeg = nbSeg>0 ? nbSeg : 1;
170
171   if( upd )
172     update();
173
174   return myTableFunc.length()>0;
175 }
176
177 bool StdMeshersGUI_DistrPreview::createTable( SMESH::double_array& func )
178 {
179   if( myExpr.IsNull() )
180   {
181     func.length( 0 );
182     return false;
183   }
184
185   const double xmin = 0.0, xmax = 1.0;
186
187   double d = (xmax-xmin)/double(myPoints-1);
188   func.length( 2*myPoints );
189   int err = 0;
190   for( int i=0, j=0; i<myPoints; j++ )
191   {
192     bool ok;
193     double t = xmin + d*j, f = funcValue( t, ok );
194     if( ok )
195     {
196       func[2*i] = t;
197       func[2*i+1] = f;
198       i++;
199     }
200     else
201       err++;
202   }
203   func.length( func.length()-2*err );
204   return err==0;
205 }
206
207 void StdMeshersGUI_DistrPreview::update()
208 {
209   Kernel_Utils::Localizer loc;
210   SMESH::double_array graph, distr;
211   if( isTableFunc() )
212   {
213     myIsDone = true;
214     graph = myTableFunc;
215   }
216   else
217     myIsDone = createTable( graph );
218
219   if( graph.length()>=2 )
220   {
221     StdMeshers::StdMeshers_NumberOfSegments_var h = 
222       StdMeshers::StdMeshers_NumberOfSegments::_narrow( myHypo );
223
224     if( !CORBA::is_nil( h.in() ) )
225     {
226       SMESH::double_array* arr = 0;
227       if( isTableFunc() )
228         arr = h->BuildDistributionTab( myTableFunc, myNbSeg, ( int )myConv );
229       else
230         arr = h->BuildDistributionExpr( myFunction.toLatin1().data(), myNbSeg, ( int )myConv );
231       if( arr )
232       {
233         distr = *arr;
234         delete arr;
235       }
236     }
237   }
238
239   bool correct = graph.length()>=2 && distr.length()>=2;
240   if( !correct )
241   {
242     showError();
243     return;
244   }
245   else
246   {
247     QwtText mt = myMsg->label();
248     mt.setText( QString() );
249     myMsg->setLabel( mt );
250   }
251
252   int size = graph.length()/2;
253   double* x = new double[size], *y = new double[size];
254   double min_x, max_x, min_y, max_y;
255   for( int i=0; i<size; i++ )
256   {
257     x[i] = graph[2*i];
258     y[i] = graph[2*i+1];
259     if( !convert( y[i] ) )
260     {
261       min_x = 0.0; max_x = 1.0; min_y = 0.0; max_y = 1.0;
262       delete[] x; delete[] y;
263       x = y = 0;
264       showError();
265       return;
266     }
267 #ifdef WIN32
268     if ( std::fabs(y[i]) >= HUGE_VAL)
269       y[i] = HUGE_VAL/100.;
270 #else
271     if ( isinf(y[i]))
272       y[i] = std::numeric_limits<double>::max()/100.;
273 #endif
274 //     if ( y[i] > 1e3 )
275 //       y[i] = 1e3;
276     if( i==0 || y[i]<min_y )
277       min_y = y[i];
278     if( i==0 || y[i]>max_y )
279       max_y = y[i];
280     if( i==0 || x[i]<min_x )
281       min_x = x[i];
282     if( i==0 || x[i]>max_x )
283       max_x = x[i];
284   }
285
286   setAxisScale( myDensity->xAxis(), min_x, max_x );
287   setAxisScale( myDensity->yAxis(),
288 #ifdef WIN32
289     min( 0.0, min_y ),
290     max( 0.0, max_y )
291 #else
292     std::min( 0.0, min_y ),
293     std::max( 0.0, max_y )
294 #endif
295     );
296   myDensity->setData( x, y, size );
297   if( x )
298     delete[] x;
299   if( y )
300     delete[] y;
301   x = y = 0;
302
303   size = distr.length();
304   x = new double[size];
305   y = new double[size];
306   for( int i=0; i<size; i++ )
307   {
308     x[i] = distr[i];
309     y[i] = 0;
310   }
311   myDistr->setData( x, y, size );
312   delete[] x;
313   delete[] y;
314   x = y = 0;
315
316   try {   
317 #ifdef NO_CAS_CATCH
318     OCC_CATCH_SIGNALS;
319 #endif
320     replot();
321   } catch(Standard_Failure) {
322     Handle(Standard_Failure) aFail = Standard_Failure::Caught();
323   }
324 }
325
326 void StdMeshersGUI_DistrPreview::showError()
327 {
328   setAxisScale( myDensity->xAxis(), 0.0, 1.0 );
329   setAxisScale( myDensity->yAxis(), 0.0, 1.0 );
330   myDensity->setData( 0, 0, 0 );
331   myDistr->setData( 0, 0, 0 );
332   QwtText mt = myMsg->label();
333   mt.setText( tr( "SMESH_INVALID_FUNCTION" ) );
334   myMsg->setLabel( mt );
335   replot();
336 }
337
338 bool isCorrectArg( const Handle( Expr_GeneralExpression )& expr )
339 {
340   Handle( Expr_NamedUnknown ) sub = Handle( Expr_NamedUnknown )::DownCast( expr );
341   if( !sub.IsNull() )
342     return sub->GetName()=="t";
343
344   bool res = true;
345   for( int i=1, n=expr->NbSubExpressions(); i<=n && res; i++ )
346   {
347     Handle( Expr_GeneralExpression ) sub = expr->SubExpression( i );
348     Handle( Expr_NamedUnknown ) name = Handle( Expr_NamedUnknown )::DownCast( sub );
349     if( !name.IsNull() )
350     {
351       if( name->GetName()!="t" )
352         res = false;
353     }
354     else
355       res = isCorrectArg( sub );
356   }
357   return res;
358 }
359
360 bool StdMeshersGUI_DistrPreview::init( const QString& str )
361 {
362   Kernel_Utils::Localizer loc;
363   bool parsed_ok = true;
364   try {
365 #ifdef NO_CAS_CATCH
366     OCC_CATCH_SIGNALS;
367 #endif
368     myExpr = ExprIntrp_GenExp::Create();
369     myExpr->Process( ( Standard_CString ) str.toLatin1().data() );
370   } catch(Standard_Failure) {
371     Handle(Standard_Failure) aFail = Standard_Failure::Caught();
372     parsed_ok = false;
373   }
374
375   bool syntax = false, args = false;
376   if( parsed_ok && myExpr->IsDone() )
377   {
378     syntax = true;
379     args = isCorrectArg( myExpr->Expression() );
380   }
381
382   bool res = parsed_ok && syntax && args;
383   if( !res )
384     myExpr.Nullify();
385   return res;
386 }
387
388 double StdMeshersGUI_DistrPreview::funcValue( const double t, bool& ok )
389 {
390   if( myExpr.IsNull() )
391     return 0;
392
393   myValues.ChangeValue( 1 ) = t;
394
395   ok = true;
396   double res = calc( ok );
397
398   return res;
399 }
400
401 double StdMeshersGUI_DistrPreview::calc( bool& ok )
402 {
403   double res = 0.0;
404
405   ok = true;
406   try {   
407 #ifdef NO_CAS_CATCH
408     OCC_CATCH_SIGNALS;
409 #endif
410     res = myExpr->Expression()->Evaluate( myVars, myValues );
411   } catch(Standard_Failure) {
412     Handle(Standard_Failure) aFail = Standard_Failure::Caught();
413     ok = false;
414     res = 0.0;
415   }
416
417   return res;
418 }
419
420 bool StdMeshersGUI_DistrPreview::isDone() const
421 {
422   return myIsDone;
423 }
424
425 bool StdMeshersGUI_DistrPreview::convert( double& v ) const
426 {
427   bool ok = true;
428   switch( myConv )
429   {
430   case EXPONENT:
431     {
432       try { 
433 #ifdef NO_CAS_CATCH
434         OCC_CATCH_SIGNALS;
435 #endif
436         // in StdMeshers_NumberOfSegments.cc
437         // const double PRECISION = 1e-7;
438         //
439         if(v < -7) v = -7.0;
440         v = pow( 10.0, v );
441       } catch(Standard_Failure) {
442         Handle(Standard_Failure) aFail = Standard_Failure::Caught();
443         v = 0.0;
444         ok = false;
445       }
446     }
447     break;
448
449   case CUT_NEGATIVE:
450     if( v<0 )
451       v = 0;
452     break;
453   }
454
455   return ok;
456 }