1 // Copyright (C) 2007-2016 CEA/DEN, EDF R&D, OPEN CASCADE
3 // Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
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, or (at your option) any later version.
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.
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
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
22 // File : SMESH_Measurements_i.cxx
23 // Author : Pavel TELKOV, Open CASCADE S.A.S. (pavel.telkov@opencascade.com)
29 #include "SMESH_Measurements_i.hxx"
31 #include "SMDS_ElemIterator.hxx"
32 #include "SMDS_Mesh.hxx"
33 #include "SMDS_MeshElement.hxx"
34 #include "SMDS_MeshNode.hxx"
35 #include "SMESHDS_Mesh.hxx"
36 #include "SMESH_Filter_i.hxx"
37 #include "SMESH_Gen_i.hxx"
38 #include "SMESH_MeshAlgos.hxx"
39 #include "SMESH_PythonDump.hxx"
42 using namespace SMESH;
45 * this local function to avoid uninitialized fields
47 static void initMeasure( SMESH::Measure& theMeasure)
50 theMeasure.minX = theMeasure.minY = theMeasure.minZ = 0.;
51 theMeasure.maxX = theMeasure.maxY = theMeasure.maxZ = 0.;
52 theMeasure.node1 = theMeasure.node2 = -1;
53 theMeasure.elem1 = theMeasure.elem2 = -1;
54 theMeasure.value = 0.;
57 //=============================================================================
59 * SMESH_Gen_i::CreateMeasurements
61 * Create measurement instance
63 //=============================================================================
65 SMESH::Measurements_ptr SMESH_Gen_i::CreateMeasurements()
67 SMESH::Measurements_i* aMeasure = new SMESH::Measurements_i();
68 SMESH::Measurements_var anObj = aMeasure->_this();
75 Description : make measure of mesh qunatities
78 //=======================================================================
79 // name : Measurements_i
80 // Purpose : Constructor
81 //=======================================================================
82 Measurements_i::Measurements_i()
83 : SALOME::GenericObj_i( SMESH_Gen_i::GetPOA() )
85 //Base class Salome_GenericObject do it inmplicitly by overriding PortableServer::POA_ptr _default_POA() method
86 //PortableServer::ObjectId_var anObjectId =
87 // SMESH_Gen_i::GetPOA()->activate_object( this );
90 //=======================================================================
91 // name : ~Measurements_i
92 // Purpose : Destructor
93 //=======================================================================
94 Measurements_i::~Measurements_i()
96 //TPythonDump()<<this<<".UnRegister()";
99 static bool getNodeNodeDistance (SMESH::Measure& theMeasure,
100 const SMDS_MeshNode* theNode1,
101 const SMDS_MeshNode* theNode2 = 0)
103 double dist = 0., dd = 0.;
108 dd = theNode1->X(); if (theNode2) dd -= theNode2->X(); theMeasure.minX = dd; dd *= dd; dist += dd;
109 dd = theNode1->Y(); if (theNode2) dd -= theNode2->Y(); theMeasure.minY = dd; dd *= dd; dist += dd;
110 dd = theNode1->Z(); if (theNode2) dd -= theNode2->Z(); theMeasure.minZ = dd; dd *= dd; dist += dd;
115 theMeasure.value = sqrt(dist);
116 theMeasure.node1 = theNode1->GetID();
117 theMeasure.node2 = theNode2 ? theNode2->GetID() : 0;
122 static SMESHDS_Mesh* getMesh(SMESH::SMESH_IDSource_ptr theSource)
124 if (!CORBA::is_nil( theSource ))
126 SMESH_Mesh_i* anImplPtr = DownCast<SMESH_Mesh_i*>(theSource->GetMesh());
128 return anImplPtr->GetImpl().GetMeshDS();
133 static bool isNodeType (SMESH::array_of_ElementType_var theTypes)
135 return theTypes->length() > 0 && theTypes[0] == SMESH::NODE;
138 static double getNumericalValue(SMESH::SMESH_IDSource_ptr theSource,
139 SMESH::Controls::NumericalFunctorPtr theFunctor)
143 if ( !CORBA::is_nil( theSource ) ) {
144 const SMESHDS_Mesh* aMesh = getMesh( theSource );
146 theFunctor->SetMesh( aMesh );
148 SMESH::long_array_var anElementsId = theSource->GetIDs();
149 for ( CORBA::ULong i = 0; i < anElementsId->length(); i++) {
150 value += theFunctor->GetValue( anElementsId[i] );
157 //=======================================================================
158 // name : MinDistance
159 // Purpose : minimal distance between two given entities
160 //=======================================================================
161 SMESH::Measure Measurements_i::MinDistance
162 (SMESH::SMESH_IDSource_ptr theSource1,
163 SMESH::SMESH_IDSource_ptr theSource2)
165 SMESH::Measure aMeasure;
166 initMeasure(aMeasure);
168 if (CORBA::is_nil( theSource1 ))
171 // if second source is null, min distance from theSource1 to the origin is calculated
172 bool isOrigin = CORBA::is_nil( theSource2 );
174 // calculate minimal distance between two mesh entities
175 SMESH::array_of_ElementType_var types1 = theSource1->GetTypes();
176 SMESH::array_of_ElementType_var types2;
177 if ( !isOrigin ) types2 = theSource2->GetTypes();
179 // here we assume that type of all IDs defined by first type in array
180 bool isNode1 = isNodeType(types1);
181 bool isNode2 = isOrigin || isNodeType(types2);
183 SMESH::long_array_var aElementsId1 = theSource1->GetIDs();
184 SMESH::long_array_var aElementsId2;
185 if ( !isOrigin ) aElementsId2 = theSource2->GetIDs();
187 // compute distance between two entities
188 /* NOTE: currently only node-to-node case is implemented
189 * all other cases will be implemented later
190 * below IF should be replaced by complete switch
191 * on mesh entities types
193 if (isNode1 && isNode2)
196 const SMESHDS_Mesh* aMesh1 = getMesh( theSource1 );
197 const SMESHDS_Mesh* aMesh2 = isOrigin ? 0 : getMesh( theSource2 );
198 const SMDS_MeshNode* theNode1 = aMesh1 ? aMesh1->FindNode( aElementsId1[0] ) : 0;
199 const SMDS_MeshNode* theNode2 = aMesh2 ? aMesh2->FindNode( aElementsId2[0] ) : 0;
200 getNodeNodeDistance( aMeasure, theNode1, theNode2 );
210 //=======================================================================
211 // name : enlargeBoundingBox
213 //=======================================================================
214 static void enlargeBoundingBox(const SMDS_MeshNode* theNode,
215 SMESH::Measure& theMeasure)
219 if ( theMeasure.node1 == -1 ) {
220 // we use this attribute as a flag that it is the first node added to the bnd box
221 theMeasure.minX = theMeasure.maxX = theNode->X();
222 theMeasure.minY = theMeasure.maxY = theNode->Y();
223 theMeasure.minZ = theMeasure.maxZ = theNode->Z();
224 theMeasure.node1 = theNode->GetID();
227 theMeasure.minX = std::min( theMeasure.minX, theNode->X() );
228 theMeasure.maxX = std::max( theMeasure.maxX, theNode->X() );
229 theMeasure.minY = std::min( theMeasure.minY, theNode->Y() );
230 theMeasure.maxY = std::max( theMeasure.maxY, theNode->Y() );
231 theMeasure.minZ = std::min( theMeasure.minZ, theNode->Z() );
232 theMeasure.maxZ = std::max( theMeasure.maxZ, theNode->Z() );
236 //=======================================================================
237 // name : enlargeBoundingBox
239 //=======================================================================
240 static void enlargeBoundingBox(const SMESH::SMESH_IDSource_ptr theObject,
241 SMESH::Measure& theMeasure)
243 if ( CORBA::is_nil( theObject ) )
245 const SMESHDS_Mesh* aMesh = getMesh( theObject );
248 SMESH::array_of_ElementType_var types = theObject->GetTypes();
249 SMESH::long_array_var aElementsId = theObject->GetIDs();
250 // here we assume that type of all IDs defined by first type in array
251 const bool isNode = isNodeType( types );
252 for(int i = 0, n = aElementsId->length(); i < n; i++)
255 enlargeBoundingBox( aMesh->FindNode( aElementsId[i] ), theMeasure);
258 if ( const SMDS_MeshElement* elem = aMesh->FindElement( aElementsId[i] ))
259 for (SMDS_NodeIteratorPtr aNodeIter = elem->nodeIterator(); aNodeIter->more(); )
260 enlargeBoundingBox( aNodeIter->next(), theMeasure);
265 //=======================================================================
266 // name : BoundingBox
267 // Purpose : compute common bounding box of entities
268 //=======================================================================
269 SMESH::Measure Measurements_i::BoundingBox (const SMESH::ListOfIDSources& theSources)
271 SMESH::Measure aMeasure;
272 initMeasure(aMeasure);
274 // calculate bounding box on sources
275 for ( int i = 0, n = theSources.length(); i < n ; ++i )
276 enlargeBoundingBox( theSources[i], aMeasure );
281 //=======================================================================
283 // Purpose : sum of length of 1D elements of the source
284 //=======================================================================
285 double Measurements_i::Length(SMESH::SMESH_IDSource_ptr theSource)
287 return getNumericalValue( theSource, SMESH::Controls::NumericalFunctorPtr(new SMESH::Controls::Length()) );
290 //=======================================================================
292 // Purpose : sum of area of 2D elements of the source
293 //=======================================================================
294 double Measurements_i::Area(SMESH::SMESH_IDSource_ptr theSource)
296 return getNumericalValue( theSource, SMESH::Controls::NumericalFunctorPtr(new SMESH::Controls::Area()) );
299 //=======================================================================
301 // Purpose : sum of volume of 3D elements of the source
302 //=======================================================================
303 double Measurements_i::Volume(SMESH::SMESH_IDSource_ptr theSource)
305 return getNumericalValue( theSource, SMESH::Controls::NumericalFunctorPtr(new SMESH::Controls::Volume()) );
308 //=======================================================================
309 //function : GravityCenter
310 //purpose : return gravity center of the source: average coordinates of all nodes
311 //=======================================================================
313 SMESH::PointStruct Measurements_i::GravityCenter(SMESH::SMESH_IDSource_ptr theSource)
315 SMESH::PointStruct grCenter = { 0.,0.,0. };
316 const SMESHDS_Mesh* mesh = getMesh( theSource );
320 // unmark all nodes; visited nodes will be marked
321 SMESH_MeshAlgos::MarkElems( mesh->nodesIterator(), /*isMarked=*/false );
323 gp_XYZ sumCoord( 0,0,0 );
326 SMDS_ElemIteratorPtr eIt = SMESH_Mesh_i::GetElements( theSource, SMESH::ALL );
327 while ( eIt->more() )
329 const SMDS_MeshElement* elem = eIt->next();
330 for ( SMDS_NodeIteratorPtr nIt = elem->nodeIterator(); nIt->more(); )
332 const SMDS_MeshNode* n = nIt->next();
333 if ( !n->isMarked() )
335 sumCoord += SMESH_NodeXYZ( n );
337 n->setIsMarked( true );
341 sumCoord /= nodeCount;
343 grCenter.x = sumCoord.X();
344 grCenter.y = sumCoord.Y();
345 grCenter.z = sumCoord.Z();