src/StdMeshers/StdMeshers_SegmentLengthAroundVertex.cxx

   1 //  Copyright (C) 2007-2008  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 //  SMESH SMESH : implementaion of SMESH idl descriptions
  23 //  File   : StdMeshers_SegmentLengthAroundVertex.cxx
  24 //  Module : SMESH
  25 //  $Header$
  26 //
  27 #include "StdMeshers_SegmentLengthAroundVertex.hxx"
  28
  29 #include "SMESH_Mesh.hxx"
  30 #include "SMESH_Algo.hxx"
  31 #include "SMDS_MeshNode.hxx"
  32 #include "SMESHDS_Mesh.hxx"
  33 #include "SMESHDS_SubMesh.hxx"
  34 #include "SMESH_MeshEditor.hxx"
  35 #include "SMESH_MesherHelper.hxx"
  36
  37 #include "utilities.h"
  38
  39 #include <BRepAdaptor_Curve.hxx>
  40 #include <GCPnts_AbscissaPoint.hxx>
  41 #include <TopTools_IndexedMapOfShape.hxx>
  42 #include <TopoDS.hxx>
  43 #include <TopoDS_Edge.hxx>
  44
  45 using namespace std;
  46
  47 //=============================================================================
  48 /*!
  49  *
  50  */
  51 //=============================================================================
  52
  53 StdMeshers_SegmentLengthAroundVertex::StdMeshers_SegmentLengthAroundVertex
  54                                        (int hypId, int studyId, SMESH_Gen * gen)
  55   :SMESH_Hypothesis(hypId, studyId, gen)
  56 {
  57   _length = 1.;
  58   _name = "SegmentLengthAroundVertex";
  59   _param_algo_dim = 0; // is used by StdMeshers_SegmentAroundVertex_0D
  60 }
  61
  62 //=============================================================================
  63 /*!
  64  *
  65  */
  66 //=============================================================================
  67
  68 StdMeshers_SegmentLengthAroundVertex::~StdMeshers_SegmentLengthAroundVertex()
  69 {
  70 }
  71
  72 //=============================================================================
  73 /*!
  74  *
  75  */
  76 //=============================================================================
  77
  78 void StdMeshers_SegmentLengthAroundVertex::SetLength(double length) throw(SALOME_Exception)
  79 {
  80   if (length <= 0)
  81     throw SALOME_Exception(LOCALIZED("length must be positive"));
  82   if (_length != length) {
  83     _length = length;
  84     NotifySubMeshesHypothesisModification();
  85   }
  86 }
  87
  88 //=============================================================================
  89 /*!
  90  *
  91  */
  92 //=============================================================================
  93
  94 double StdMeshers_SegmentLengthAroundVertex::GetLength() const
  95 {
  96   return _length;
  97 }
  98
  99 //=============================================================================
 100 /*!
 101  *
 102  */
 103 //=============================================================================
 104
 105 ostream & StdMeshers_SegmentLengthAroundVertex::SaveTo(ostream & save)
 106 {
 107   save << this->_length;
 108   return save;
 109 }
 110
 111 //=============================================================================
 112 /*!
 113  *
 114  */
 115 //=============================================================================
 116
 117 istream & StdMeshers_SegmentLengthAroundVertex::LoadFrom(istream & load)
 118 {
 119   bool isOK = true;
 120   double a;
 121   isOK = (load >> a);
 122   if (isOK)
 123     this->_length = a;
 124   else
 125     load.clear(ios::badbit | load.rdstate());
 126   return load;
 127 }
 128
 129 //=============================================================================
 130 /*!
 131  *
 132  */
 133 //=============================================================================
 134
 135 ostream & operator <<(ostream & save, StdMeshers_SegmentLengthAroundVertex & hyp)
 136 {
 137   return hyp.SaveTo( save );
 138 }
 139
 140 //=============================================================================
 141 /*!
 142  *
 143  */
 144 //=============================================================================
 145
 146 istream & operator >>(istream & load, StdMeshers_SegmentLengthAroundVertex & hyp)
 147 {
 148   return hyp.LoadFrom( load );
 149 }
 150
 151 //================================================================================
 152 /*!
 153  * \brief Initialize segment length by the mesh built on the geometry
 154  * \param theMesh - the built mesh
 155  * \param theShape - the geometry of interest
 156  * \retval bool - true if parameter values have been successfully defined
 157  */
 158 //================================================================================
 159
 160 bool StdMeshers_SegmentLengthAroundVertex::SetParametersByMesh(const SMESH_Mesh*   theMesh,
 161                                                                const TopoDS_Shape& theShape)
 162 {
 163   if ( !theMesh || theShape.IsNull() || theShape.ShapeType() != TopAbs_VERTEX )
 164     return false;
 165
 166   SMESH_MeshEditor editor( const_cast<SMESH_Mesh*>( theMesh ) );
 167   SMESH_MesherHelper helper( *editor.GetMesh() );
 168
 169   // get node built on theShape vertex
 170   SMESHDS_Mesh* meshDS = editor.GetMeshDS();
 171   SMESHDS_SubMesh* smV = meshDS->MeshElements( theShape );
 172   if ( !smV || smV->NbNodes() == 0 )
 173     return false;
 174   const SMDS_MeshNode* vNode = smV->GetNodes()->next();
 175
 176   // calculate average length of segments sharing vNode
 177
 178   _length = 0.;
 179   int nbSegs = 0;
 180
 181   SMDS_ElemIteratorPtr segIt = vNode->GetInverseElementIterator(SMDSAbs_Edge);
 182   while ( segIt->more() ) {
 183     const SMDS_MeshElement* seg = segIt->next();
 184     // get geom edge
 185     int shapeID = editor.FindShape( seg );
 186     if (!shapeID) continue;
 187     const TopoDS_Shape& s = meshDS->IndexToShape( shapeID );
 188     if ( s.IsNull() || s.ShapeType() != TopAbs_EDGE ) continue;
 189     const TopoDS_Edge& edge = TopoDS::Edge( s );
 190     // params of edge ends
 191     double u0 = helper.GetNodeU( edge, seg->GetNode(0) );
 192     double u1 = helper.GetNodeU( edge, seg->GetNode(1) );
 193     // length
 194     BRepAdaptor_Curve AdaptCurve( edge );
 195     _length += GCPnts_AbscissaPoint::Length( AdaptCurve, u0, u1);
 196     nbSegs++;
 197   }
 198
 199   if ( nbSegs > 1 )
 200     _length /= nbSegs;
 201
 202   return nbSegs;
 203 }