Salome HOME
Fix for bug 10498. Support old storage format of NumberOfSegments hypothesis
[modules/smesh.git] / src / StdMeshers / StdMeshers_NumberOfSegments.cxx
index c16c1373ba4bca2d1bb8caa65d758a1a68c7be50..a768b6c1670b3365ba661ba31bb6acde6572fdda 100644 (file)
 
 using namespace std;
 #include "StdMeshers_NumberOfSegments.hxx"
+#include <Standard_ErrorHandler.hxx>
+#include <TCollection_AsciiString.hxx>
+#include <ExprIntrp_GenExp.hxx>
+#include <Expr_NamedUnknown.hxx>
+#include <CASCatch_CatchSignals.hxx>
+#include <CASCatch_Failure.hxx> 
+#include <CASCatch_ErrorHandler.hxx>
+#include <OSD.hxx>
+#include <Expr_Array1OfNamedUnknown.hxx>
+#include <TColStd_Array1OfReal.hxx>
+
+
+const double PRECISION = 1e-7;
 
 //=============================================================================
 /*!
@@ -37,12 +50,15 @@ using namespace std;
 //=============================================================================
 
 StdMeshers_NumberOfSegments::StdMeshers_NumberOfSegments(int hypId, int studyId,
-       SMESH_Gen * gen):SMESH_Hypothesis(hypId, studyId, gen)
+       SMESH_Gen * gen)
+  : SMESH_Hypothesis(hypId, studyId, gen),
+    _numberOfSegments(1),
+    _distrType(DT_Regular),
+    _scaleFactor(1.),
+    _expMode(false)
 {
-       _numberOfSegments = 1;
-       _scaleFactor = 1.0;
-       _name = "NumberOfSegments";
-        _param_algo_dim = 1; 
+  _name = "NumberOfSegments";
+  _param_algo_dim = 1;
 }
 
 //=============================================================================
@@ -85,31 +101,325 @@ int StdMeshers_NumberOfSegments::GetNumberOfSegments() const
        return _numberOfSegments;
 }
 
-//=============================================================================
+//================================================================================
 /*!
- *  
+ * 
  */
-//=============================================================================
+//================================================================================
+
+void StdMeshers_NumberOfSegments::SetDistrType(DistrType typ)
+  throw(SALOME_Exception)
+{
+  if (typ < DT_Regular || typ > DT_ExprFunc)
+    throw SALOME_Exception(LOCALIZED("distribution type is out of range"));
+
+  if (typ != _distrType)
+  {
+    _distrType = typ;
+    NotifySubMeshesHypothesisModification();
+  }
+}
+
+//================================================================================
+/*!
+ * 
+ */
+//================================================================================
+
+StdMeshers_NumberOfSegments::DistrType StdMeshers_NumberOfSegments::GetDistrType() const
+{
+  return _distrType;
+}
+
+//================================================================================
+/*!
+ * 
+ */
+//================================================================================
 
 void StdMeshers_NumberOfSegments::SetScaleFactor(double scaleFactor)
-throw(SALOME_Exception)
+  throw(SALOME_Exception)
 {
-       if (scaleFactor < 0)
-               throw SALOME_Exception(LOCALIZED("scale factor must be positive"));
-       _scaleFactor = scaleFactor;
+  if (_distrType != DT_Scale)
+    throw SALOME_Exception(LOCALIZED("not a scale distribution"));
+  if (scaleFactor < PRECISION)
+    throw SALOME_Exception(LOCALIZED("scale factor must be positive"));
+  if (fabs(scaleFactor - 1.0) < PRECISION)
+    throw SALOME_Exception(LOCALIZED("scale factor must not be equal to 1"));
 
-       NotifySubMeshesHypothesisModification();
+  if (fabs(_scaleFactor - scaleFactor) > PRECISION)
+  {
+    _scaleFactor = scaleFactor;
+    NotifySubMeshesHypothesisModification();
+  }
 }
 
-//=============================================================================
+//================================================================================
 /*!
- *  
+ * 
  */
-//=============================================================================
+//================================================================================
 
 double StdMeshers_NumberOfSegments::GetScaleFactor() const
+  throw(SALOME_Exception)
 {
-       return _scaleFactor;
+  if (_distrType != DT_Scale)
+    throw SALOME_Exception(LOCALIZED("not a scale distribution"));
+  return _scaleFactor;
+}
+
+//================================================================================
+/*!
+ * 
+ */
+//================================================================================
+
+void StdMeshers_NumberOfSegments::SetTableFunction(const std::vector<double>& table)
+  throw(SALOME_Exception)
+{
+  if (_distrType != DT_TabFunc)
+    throw SALOME_Exception(LOCALIZED("not a table function distribution"));
+  if ( (table.size() % 2) != 0 )
+    throw SALOME_Exception(LOCALIZED("odd size of vector of table function"));
+
+  int i;
+  double prev = -PRECISION;
+  bool isSame = table.size() == _table.size();
+
+  bool pos = false;
+  for (i=0; i < table.size()/2; i++) {
+    double par = table[i*2];
+    double val = table[i*2+1];
+    if ( par<0 || par > 1)
+      throw SALOME_Exception(LOCALIZED("parameter of table function is out of range [0,1]"));
+    if ( fabs(par-prev)<PRECISION )
+      throw SALOME_Exception(LOCALIZED("two parameters are the same"));
+    if ( val < 0 )
+      throw SALOME_Exception(LOCALIZED("value of table function is not positive"));
+    if( val>PRECISION )
+      pos = true;
+    if (isSame)
+    {
+      double oldpar = _table[i*2];
+      double oldval = _table[i*2+1];
+      if (fabs(par - oldpar) > PRECISION || fabs(val - oldval) > PRECISION)
+        isSame = false;
+    }
+    prev = par;
+  }
+
+  if( !pos )
+    throw SALOME_Exception(LOCALIZED("value of table function is not positive"));
+
+  if( pos && !isSame )
+  {
+    _table = table;
+    NotifySubMeshesHypothesisModification();
+  }
+}
+
+//================================================================================
+/*!
+ * 
+ */
+//================================================================================
+
+const std::vector<double>& StdMeshers_NumberOfSegments::GetTableFunction() const
+  throw(SALOME_Exception)
+{
+  if (_distrType != DT_TabFunc)
+    throw SALOME_Exception(LOCALIZED("not a table function distribution"));
+  return _table;
+}
+
+//================================================================================
+/*! check if only 't' is unknown variable in expression
+ */
+//================================================================================
+bool isCorrectArg( const Handle( Expr_GeneralExpression )& expr )
+{
+  Handle( Expr_NamedUnknown ) sub = Handle( Expr_NamedUnknown )::DownCast( expr );
+  if( !sub.IsNull() )
+    return sub->GetName()=="t";
+
+  bool res = true;
+  for( int i=1, n=expr->NbSubExpressions(); i<=n && res; i++ )
+  {
+    Handle( Expr_GeneralExpression ) sub = expr->SubExpression( i );
+    Handle( Expr_NamedUnknown ) name = Handle( Expr_NamedUnknown )::DownCast( sub );
+    if( !name.IsNull() )
+    {
+      if( name->GetName()!="t" )
+       res = false;
+    }
+    else
+      res = isCorrectArg( sub );
+  }
+  return res;
+}
+
+//================================================================================
+/*! this function parses the expression 'str' in order to check if syntax is correct
+ *  ( result in 'syntax' ) and if only 't' is unknown variable in expression ( result in 'args' )
+ */
+//================================================================================
+bool process( const TCollection_AsciiString& str,
+             bool& syntax, bool& args,
+             bool& non_neg, bool& non_zero,
+             bool& singulars, double& sing_point )
+{
+  Handle( ExprIntrp_GenExp ) myExpr = ExprIntrp_GenExp::Create();
+  myExpr->Process( str.ToCString() );
+
+  if( myExpr->IsDone() )
+  {
+    syntax = true;
+    args = isCorrectArg( myExpr->Expression() );
+  }
+
+  bool res = syntax && args;
+  if( !res )
+    myExpr.Nullify();
+
+  non_neg = true;
+  singulars = false;
+  non_zero = false;
+
+  if( res )
+  {
+    OSD::SetSignal( true );
+    CASCatch_CatchSignals aCatchSignals;
+    aCatchSignals.Activate();
+    double res;
+    Expr_Array1OfNamedUnknown myVars( 1, 1 );
+    TColStd_Array1OfReal  myValues( 1, 1 );
+    myVars.ChangeValue( 1 ) = new Expr_NamedUnknown( "t" );
+
+    const int max = 500;
+    for( int i=0; i<=max; i++ )
+    {
+      double t = double(i)/double(max);
+      myValues.ChangeValue( 1 ) = t;
+      CASCatch_TRY
+      {   
+       res = myExpr->Expression()->Evaluate( myVars, myValues );
+      }
+      CASCatch_CATCH(CASCatch_Failure)
+      {
+       aCatchSignals.Deactivate();
+       Handle(CASCatch_Failure) aFail = CASCatch_Failure::Caught();
+       sing_point = t;
+       singulars = true;
+       break;
+      }
+      if( res<0 )
+      {
+       non_neg = false;
+       break;
+      }
+      if( res>PRECISION )
+       non_zero = true;
+    }
+    aCatchSignals.Deactivate();
+  }
+  return res && non_neg && ( !singulars );
+}
+
+//================================================================================
+/*!
+ * 
+ */
+//================================================================================
+
+void StdMeshers_NumberOfSegments::SetExpressionFunction(const char* expr)
+  throw(SALOME_Exception)
+{
+  if (_distrType != DT_ExprFunc)
+    throw SALOME_Exception(LOCALIZED("not an expression function distribution"));
+
+  // remove white spaces
+  TCollection_AsciiString str((Standard_CString)expr);
+  str.RemoveAll(' ');
+  str.RemoveAll('\t');
+  str.RemoveAll('\r');
+  str.RemoveAll('\n');
+
+  bool syntax, args, non_neg, singulars, non_zero;
+  double sing_point;
+  bool res = true;//process( str, syntax, args, non_neg, non_zero, singulars, sing_point );
+  if( !res )
+  {
+    if( !syntax )
+      throw SALOME_Exception(LOCALIZED("invalid expression syntax"));
+    if( !args )
+      throw SALOME_Exception(LOCALIZED("only 't' may be used as function argument"));
+    if( !non_neg )
+      throw SALOME_Exception(LOCALIZED("only non-negative function can be used as density"));
+    if( singulars )
+    {
+      char buf[1024];
+      sprintf( buf, "Function has singular point in %.3f", sing_point );
+      throw SALOME_Exception( buf );
+    }
+    if( !non_zero )
+      throw SALOME_Exception(LOCALIZED("f(t)=0 cannot be used as density"));
+
+    return;
+  }
+  
+  std::string func = expr;
+  if( _func != func )
+  {
+    _func = func;
+    NotifySubMeshesHypothesisModification();
+  }
+}
+
+//================================================================================
+/*!
+ * 
+ */
+//================================================================================
+
+const char* StdMeshers_NumberOfSegments::GetExpressionFunction() const
+  throw(SALOME_Exception)
+{
+  if (_distrType != DT_ExprFunc)
+    throw SALOME_Exception(LOCALIZED("not an expression function distribution"));
+  return _func.c_str();
+}
+
+//================================================================================
+/*!
+ * 
+ */
+//================================================================================
+
+void StdMeshers_NumberOfSegments::SetExponentMode(bool isExp)
+  throw(SALOME_Exception)
+{
+  if (_distrType != DT_TabFunc && _distrType != DT_ExprFunc)
+    throw SALOME_Exception(LOCALIZED("not a functional distribution"));
+
+  if (isExp != _expMode)
+  {
+    _expMode = isExp;
+    NotifySubMeshesHypothesisModification();
+  }
+}
+
+//================================================================================
+/*!
+ * 
+ */
+//================================================================================
+
+bool StdMeshers_NumberOfSegments::IsExponentMode() const
+  throw(SALOME_Exception)
+{
+  if (_distrType != DT_TabFunc && _distrType != DT_ExprFunc)
+    throw SALOME_Exception(LOCALIZED("not a functional distribution"));
+  return _expMode;
 }
 
 //=============================================================================
@@ -120,7 +430,29 @@ double StdMeshers_NumberOfSegments::GetScaleFactor() const
 
 ostream & StdMeshers_NumberOfSegments::SaveTo(ostream & save)
 {
-  save << this->_numberOfSegments << " " << this->_scaleFactor;
+  save << _numberOfSegments << " " << (int)_distrType;
+  switch (_distrType)
+  {
+  case DT_Scale:
+    save << " " << _scaleFactor;
+    break;
+  case DT_TabFunc:
+    int i;
+    save << " " << _table.size();
+    for (i=0; i < _table.size(); i++)
+      save << " " << _table[i];
+    break;
+  case DT_ExprFunc:
+    save << " " << _func;
+    break;
+  case DT_Regular:
+  default:
+    break;
+  }
+
+  if (_distrType == DT_TabFunc || _distrType == DT_ExprFunc)
+    save << " " << (int)_expMode;
+  
   return save;
 }
 
@@ -134,17 +466,108 @@ istream & StdMeshers_NumberOfSegments::LoadFrom(istream & load)
 {
   bool isOK = true;
   int a;
+
+  // read number of segments
   isOK = (load >> a);
   if (isOK)
-    this->_numberOfSegments = a;
+    _numberOfSegments = a;
   else
     load.clear(ios::badbit | load.rdstate());
-  double b;
-  isOK = (load >> b);
+
+  // read second stored value. It can be two variants here:
+  // 1. If the hypothesis is stored in old format (nb.segments and scale factor),
+  //    we wait here the scale factor, which is double.
+  // 2. If the hypothesis is stored in new format
+  //    (nb.segments, distr.type, some other params.),
+  //    we wait here the ditribution type, which is integer
+  double scale_factor;
+  isOK = (load >> scale_factor);
+  a = (int)scale_factor;
+
+  // try to interprete ditribution type,
+  // supposing that this hypothesis was written in the new format
   if (isOK)
-    this->_scaleFactor = b;
+  {
+    if (a < DT_Regular || a > DT_ExprFunc)
+      _distrType = DT_Regular;
+    else
+      _distrType = (DistrType) a;
+  }
   else
     load.clear(ios::badbit | load.rdstate());
+
+  // parameters of distribution
+  double b;
+  switch (_distrType)
+  {
+  case DT_Scale:
+    {
+      isOK = (load >> b);
+      if (isOK)
+        _scaleFactor = b;
+      else
+      {
+        load.clear(ios::badbit | load.rdstate());
+        // this can mean, that the hypothesis is stored in old format
+        _distrType = DT_Regular;
+        _scaleFactor = scale_factor;
+      }
+    }
+    break;
+  case DT_TabFunc:
+    {
+      isOK = (load >> a);
+      if (isOK)
+      {
+        _table.resize(a, 0.);
+        int i;
+        for (i=0; i < _table.size(); i++)
+        {
+          isOK = (load >> b);
+          if (isOK)
+            _table[i] = b;
+          else
+            load.clear(ios::badbit | load.rdstate());
+        }
+      }
+      else
+      {
+        load.clear(ios::badbit | load.rdstate());
+        // this can mean, that the hypothesis is stored in old format
+        _distrType = DT_Regular;
+        _scaleFactor = scale_factor;
+      }
+    }
+    break;
+  case DT_ExprFunc:
+    {
+      string str;
+      isOK = (load >> str);
+      if (isOK)
+        _func = str;
+      else
+      {
+        load.clear(ios::badbit | load.rdstate());
+        // this can mean, that the hypothesis is stored in old format
+        _distrType = DT_Regular;
+        _scaleFactor = scale_factor;
+      }
+    }
+    break;
+  case DT_Regular:
+  default:
+    break;
+  }
+
+  if (_distrType == DT_TabFunc || _distrType == DT_ExprFunc)
+  {
+    isOK = (load >> a);
+    if (isOK)
+      _expMode = (bool) a;
+    else
+      load.clear(ios::badbit | load.rdstate());
+  }
+
   return load;
 }