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
23 // SMESH OBJECT : interactive object for SMESH visualization
24 // File : SMESH_Grid.cxx
25 // Author : Nicolas REJNERI
28 #include "SMESH_ObjectDef.h"
29 #include "SMESH_ActorUtils.h"
31 #include "SMDS_BallElement.hxx"
32 #include "SMDS_Mesh.hxx"
33 #include "SMDS_MeshCell.hxx"
34 #include "SMDS_PolyhedralVolumeOfNodes.hxx"
35 #include "SMESHDS_Mesh.hxx"
36 #include "SMESHDS_Script.hxx"
37 #include "SMESH_Actor.h"
38 #include "SMESH_ControlsDef.hxx"
40 #include <SalomeApp_Application.h>
41 #include <VTKViewer_ExtractUnstructuredGrid.h>
42 #include <VTKViewer_CellLocationsArray.h>
44 #include CORBA_SERVER_HEADER(SMESH_Gen)
45 #include CORBA_SERVER_HEADER(SALOME_Exception)
48 #include <vtkIdList.h>
49 #include <vtkCellArray.h>
50 #include <vtkUnsignedCharArray.h>
51 #include <vtkCellData.h>
52 #include <vtkUnstructuredGrid.h>
59 #include <utilities.h>
64 #define EXCEPTION(TYPE, MSG) {\
65 std::ostringstream aStream;\
66 aStream<<__FILE__<<"["<<__LINE__<<"]::"<<MSG;\
67 throw TYPE(aStream.str());\
72 static int MYDEBUG = 0;
73 static int MYDEBUGWITHFILES = 0;//1;
75 static int MYDEBUG = 0;
76 static int MYDEBUGWITHFILES = 0;
81 Class : SMESH_VisualObjDef
82 Description : Base class for all mesh objects to be visuilised
85 //=================================================================================
86 // function : getCellType
87 // purpose : Get type of VTK cell
88 //=================================================================================
89 // static inline vtkIdType getCellType( const SMDSAbs_ElementType theType,
90 // const bool thePoly,
91 // const int theNbNodes )
95 // case SMDSAbs_0DElement: return VTK_VERTEX;
97 // case SMDSAbs_Ball: return VTK_POLY_VERTEX;
100 // if( theNbNodes == 2 ) return VTK_LINE;
101 // else if ( theNbNodes == 3 ) return VTK_QUADRATIC_EDGE;
102 // else return VTK_EMPTY_CELL;
104 // case SMDSAbs_Face :
105 // if (thePoly && theNbNodes>2 ) return VTK_POLYGON;
106 // else if ( theNbNodes == 3 ) return VTK_TRIANGLE;
107 // else if ( theNbNodes == 4 ) return VTK_QUAD;
108 // else if ( theNbNodes == 6 ) return VTK_QUADRATIC_TRIANGLE;
109 // else if ( theNbNodes == 8 ) return VTK_QUADRATIC_QUAD;
110 // else if ( theNbNodes == 9 ) return VTK_BIQUADRATIC_QUAD;
111 // else if ( theNbNodes == 7 ) return VTK_BIQUADRATIC_TRIANGLE;
112 // else return VTK_EMPTY_CELL;
114 // case SMDSAbs_Volume:
115 // if (thePoly && theNbNodes>3 ) return VTK_POLYHEDRON; //VTK_CONVEX_POINT_SET;
116 // else if ( theNbNodes == 4 ) return VTK_TETRA;
117 // else if ( theNbNodes == 5 ) return VTK_PYRAMID;
118 // else if ( theNbNodes == 6 ) return VTK_WEDGE;
119 // else if ( theNbNodes == 8 ) return VTK_HEXAHEDRON;
120 // else if ( theNbNodes == 12 ) return VTK_HEXAGONAL_PRISM;
121 // else if ( theNbNodes == 10 ) return VTK_QUADRATIC_TETRA;
122 // else if ( theNbNodes == 20 ) return VTK_QUADRATIC_HEXAHEDRON;
123 // else if ( theNbNodes == 27 ) return VTK_TRIQUADRATIC_HEXAHEDRON;
124 // else if ( theNbNodes == 15 ) return VTK_QUADRATIC_WEDGE;
125 // else if ( theNbNodes == 13 ) return VTK_QUADRATIC_PYRAMID; //VTK_CONVEX_POINT_SET;
126 // else return VTK_EMPTY_CELL;
128 // default: return VTK_EMPTY_CELL;
132 //=================================================================================
133 // functions : SMESH_VisualObjDef
134 // purpose : Constructor
135 //=================================================================================
136 SMESH_VisualObjDef::SMESH_VisualObjDef()
138 if ( MYDEBUG ) MESSAGE("-------------------------------SMESH_VisualObjDef::SMESH_VisualObjDef");
139 myGrid = vtkUnstructuredGrid::New();
141 ClearEntitiesFlags();
142 SMESH::GetEntitiesFromObject(NULL);
144 SMESH_VisualObjDef::~SMESH_VisualObjDef()
146 if ( MYDEBUG ) MESSAGE("--------------------------------SMESH_VisualObjDef::~SMESH_VisualObjDef");
147 if ( MYDEBUG ) MESSAGE( "myGrid->GetReferenceCount() = " << myGrid->GetReferenceCount() );
151 //=================================================================================
152 // functions : GetNodeObjId, GetNodeVTKId, GetElemObjId, GetElemVTKId
153 // purpose : Methods for retrieving VTK IDs by SMDS IDs and vice versa
154 //=================================================================================
155 vtkIdType SMESH_VisualObjDef::GetNodeObjId( int theVTKID )
159 TMapOfIds::const_iterator i = myVTK2SMDSNodes.find(theVTKID);
160 return i == myVTK2SMDSNodes.end() ? -1 : i->second;
162 const SMDS_MeshNode* aNode = 0;
163 if( this->GetMesh() )
164 aNode = this->GetMesh()->FindNodeVtk( theVTKID );
166 return aNode ? aNode->GetID() : -1;
169 vtkIdType SMESH_VisualObjDef::GetNodeVTKId( int theObjID )
173 TMapOfIds::const_iterator i = mySMDS2VTKNodes.find(theObjID);
174 return i == mySMDS2VTKNodes.end() ? -1 : i->second;
177 const SMDS_MeshNode* aNode = 0;
178 if( this->GetMesh() ) {
179 aNode = this->GetMesh()->FindNode(theObjID);
181 return aNode ? aNode->getVtkId() : -1;
184 vtkIdType SMESH_VisualObjDef::GetElemObjId( int theVTKID )
188 TMapOfIds::const_iterator i = myVTK2SMDSElems.find(theVTKID);
189 return i == myVTK2SMDSElems.end() ? -1 : i->second;
191 return this->GetMesh()->fromVtkToSmds(theVTKID);
194 vtkIdType SMESH_VisualObjDef::GetElemVTKId( int theObjID )
198 TMapOfIds::const_iterator i = mySMDS2VTKElems.find(theObjID);
199 return i == mySMDS2VTKElems.end() ? -1 : i->second;
202 const SMDS_MeshElement* e = 0;
203 if ( this->GetMesh() )
204 e = this->GetMesh()->FindElement(theObjID);
206 return e ? e->getVtkId() : -1;
209 //=================================================================================
210 // function : SMESH_VisualObjDef::createPoints
211 // purpose : Create points from nodes
212 //=================================================================================
213 /*! fills a vtkPoints structure for a submesh.
214 * fills a std::list of SMDS_MeshElements*, then extract the points.
215 * fills also conversion id maps between SMDS and VTK.
217 void SMESH_VisualObjDef::createPoints( vtkPoints* thePoints )
219 if ( thePoints == 0 )
223 vtkIdType nbNodes = GetEntities( SMDSAbs_Node, aNodes );
224 thePoints->SetNumberOfPoints( nbNodes );
228 TEntityList::const_iterator anIter;
229 for ( anIter = aNodes.begin(); anIter != aNodes.end(); ++anIter )
231 const SMDS_MeshNode* aNode = ( const SMDS_MeshNode* )(*anIter);
234 thePoints->SetPoint( nbPoints, aNode->X(), aNode->Y(), aNode->Z() );
235 int anId = aNode->GetID();
236 mySMDS2VTKNodes.insert( mySMDS2VTKNodes.end(), std::make_pair( anId, nbPoints ));
237 myVTK2SMDSNodes.insert( myVTK2SMDSNodes.end(), std::make_pair( nbPoints, anId ));
242 if ( nbPoints != nbNodes )
243 thePoints->SetNumberOfPoints( nbPoints );
246 //=================================================================================
247 // function : buildPrs
248 // purpose : create VTK cells( fill unstructured grid )
249 //=================================================================================
250 void SMESH_VisualObjDef::buildPrs(bool buildGrid)
252 if ( MYDEBUG ) MESSAGE("---------------------------SMESH_VisualObjDef::buildPrs " << buildGrid);
258 mySMDS2VTKNodes.clear();
259 myVTK2SMDSNodes.clear();
260 mySMDS2VTKElems.clear();
261 myVTK2SMDSElems.clear();
270 mySMDS2VTKNodes.clear();
271 myVTK2SMDSNodes.clear();
272 mySMDS2VTKElems.clear();
273 myVTK2SMDSElems.clear();
275 myGrid->SetPoints( 0 );
276 myGrid->SetCells( 0, 0, 0, 0, 0 );
283 if (!GetMesh()->isCompacted())
285 NulData(); // detach from the SMDS grid to allow immediate memory de-allocation in compactMesh()
286 if ( MYDEBUG ) MESSAGE("*** buildPrs ==> compactMesh!");
287 GetMesh()->compactMesh();
288 if ( SMESHDS_Mesh* m = dynamic_cast<SMESHDS_Mesh*>( GetMesh() )) // IPAL53915
289 m->GetScript()->SetModified(false); // drop IsModified set in compactMesh()
291 vtkUnstructuredGrid *theGrid = GetMesh()->getGrid();
292 updateEntitiesFlags();
293 myGrid->ShallowCopy(theGrid);
294 //MESSAGE(myGrid->GetReferenceCount());
295 //MESSAGE( "Update - myGrid->GetNumberOfCells() = "<<myGrid->GetNumberOfCells() );
296 //MESSAGE( "Update - myGrid->GetNumberOfPoints() = "<<myGrid->GetNumberOfPoints() );
297 if( MYDEBUGWITHFILES ) {
298 SMESH::WriteUnstructuredGrid( myGrid,"myPrs.vtu" );
303 //=================================================================================
304 // function : buildNodePrs
305 // purpose : create VTK cells for nodes
306 //=================================================================================
308 void SMESH_VisualObjDef::buildNodePrs()
310 // PAL16631: without swap, bad_alloc is not thrown but hung up and crash instead,
311 // so check remaining memory size for safety
312 SMDS_Mesh::CheckMemory(); // PAL16631
313 vtkPoints* aPoints = vtkPoints::New();
314 createPoints( aPoints );
315 SMDS_Mesh::CheckMemory();
316 myGrid->SetPoints( aPoints );
319 myGrid->SetCells( 0, 0, 0, 0, 0 );
322 //=================================================================================
323 // function : buildElemPrs
324 // purpose : Create VTK cells for elements
325 //=================================================================================
328 typedef std::vector<const SMDS_MeshElement*> TConnect;
330 int GetConnect(const SMDS_ElemIteratorPtr& theNodesIter,
331 TConnect& theConnect)
334 for(; theNodesIter->more();)
335 theConnect.push_back(theNodesIter->next());
336 return theConnect.size();
340 void SetId(vtkIdList *theIdList,
341 const SMESH_VisualObjDef::TMapOfIds& theSMDS2VTKNodes,
342 const TConnect& theConnect,
346 theIdList->SetId(thePosition,theSMDS2VTKNodes.find(theConnect[theId]->GetID())->second);
352 void SMESH_VisualObjDef::buildElemPrs()
356 vtkPoints* aPoints = vtkPoints::New();
357 createPoints( aPoints );
358 myGrid->SetPoints( aPoints );
362 MESSAGE("Update - myGrid->GetNumberOfPoints() = "<<myGrid->GetNumberOfPoints());
364 // Calculate cells size
366 const int nbTypes = 5;
367 static SMDSAbs_ElementType aTypes[ nbTypes ] =
368 { SMDSAbs_Edge, SMDSAbs_Face, SMDSAbs_Volume, SMDSAbs_Ball, SMDSAbs_0DElement };
371 map<SMDSAbs_ElementType,int> nbEnts;
372 map<SMDSAbs_ElementType,TEntityList> anEnts;
374 vtkIdType aNbCells = 0;
376 for ( int i = 0; i < nbTypes; i++ )
378 nbEnts[ aTypes[ i ] ] = GetEntities( aTypes[ i ], anEnts[ aTypes[ i ] ] );
379 aNbCells += nbEnts[ aTypes [ i ]];
381 // PAL16631: without swap, bad_alloc is not thrown but hung up and crash instead,
382 // so check remaining memory size for safety
383 SMDS_Mesh::CheckMemory(); // PAL16631
385 vtkIdType aCellsSize = 2 * nbEnts[ SMDSAbs_0DElement ] + 3 * nbEnts[ SMDSAbs_Edge ];
386 aCellsSize += 2 * nbEnts[ SMDSAbs_Ball ];
388 for ( int i = 1; i <= 2; i++ ) // iterate through faces and volumes
390 if ( nbEnts[ aTypes[ i ] ] )
392 const TEntityList& aList = anEnts[ aTypes[ i ] ];
393 TEntityList::const_iterator anIter;
394 for ( anIter = aList.begin(); anIter != aList.end(); ++anIter ) {
395 if((*anIter)->GetEntityType() != SMDSEntity_Polyhedra &&
396 (*anIter)->GetEntityType() != SMDSEntity_Quad_Polyhedra) {
397 aCellsSize += (*anIter)->NbNodes() + 1;
399 // Special case for the VTK_POLYHEDRON:
400 // itsinput cellArray is of special format.
401 // [nCellFaces, nFace0Pts, i, j, k, nFace1Pts, i, j, k, ...]
403 if( const SMDS_VtkVolume* ph = dynamic_cast<const SMDS_VtkVolume*>(*anIter) ) {
404 int nbFaces = ph->NbFaces();
405 aCellsSize += (1 + ph->NbFaces());
406 for( int i = 1; i <= nbFaces; i++ ) {
407 aCellsSize += ph->NbFaceNodes(i);
415 MESSAGE( "Update - aNbCells = "<<aNbCells<<"; aCellsSize = "<<aCellsSize );
419 vtkCellArray* aConnectivity = vtkCellArray::New();
420 aConnectivity->Allocate( aCellsSize, 0 );
422 SMDS_Mesh::CheckMemory(); // PAL16631
424 vtkUnsignedCharArray* aCellTypesArray = vtkUnsignedCharArray::New();
425 aCellTypesArray->SetNumberOfComponents( 1 );
426 aCellTypesArray->Allocate( aNbCells * aCellTypesArray->GetNumberOfComponents() );
428 SMDS_Mesh::CheckMemory(); // PAL16631
430 vtkIdList *anIdList = vtkIdList::New();
434 aConnect.reserve(VTK_CELL_SIZE);
436 SMDS_Mesh::CheckMemory(); // PAL16631
437 bool hasBalls = nbEnts[ SMDSAbs_Ball ] > 0;
438 vtkDataArray* aScalars = 0;
440 aScalars = vtkDataArray::CreateDataArray(VTK_DOUBLE);
441 aScalars->SetNumberOfComponents(1);
442 aScalars->SetNumberOfTuples(aNbCells);
444 for ( int i = 0; i < nbTypes; i++ ) // iterate through all types of elements
446 if ( nbEnts[ aTypes[ i ] ] > 0 ) {
448 const SMDSAbs_ElementType& aType = aTypes[ i ];
449 const TEntityList& aList = anEnts[ aType ];
450 TEntityList::const_iterator anIter;
451 for ( anIter = aList.begin(); anIter != aList.end(); ++anIter )
453 const SMDS_MeshElement* anElem = *anIter;
455 vtkIdType aNbNodes = anElem->NbNodes();
456 anIdList->SetNumberOfIds( aNbNodes );
457 const vtkIdType vtkElemType = SMDS_MeshCell::toVtkType( anElem->GetEntityType() );
459 int anId = anElem->GetID();
461 mySMDS2VTKElems.insert( mySMDS2VTKElems.end(), std::make_pair( anId, iElem ));
462 myVTK2SMDSElems.insert( myVTK2SMDSElems.end(), std::make_pair( iElem, anId ));
464 SMDS_ElemIteratorPtr aNodesIter = anElem->nodesIterator();
466 // Convert connectivities from SMDS to VTK
468 if (aType == SMDSAbs_Volume && anElem->IsPoly() && aNbNodes > 3) { // POLYEDRE
470 if ( const SMDS_VtkVolume* ph = dynamic_cast<const SMDS_VtkVolume*>(anElem) ) {
471 int nbFaces = ph->NbFaces();
472 anIdList->InsertNextId(nbFaces);
473 for( int i = 1; i <= nbFaces; i++ ) {
474 anIdList->InsertNextId(ph->NbFaceNodes(i));
475 for(int j = 1; j <= ph->NbFaceNodes(i); j++) {
476 const SMDS_MeshNode* n = ph->GetFaceNode(i,j);
478 anIdList->InsertNextId(mySMDS2VTKNodes[n->GetID()]);
485 const std::vector<int>& aConnectivities =
486 SMDS_MeshCell::toVtkOrder( VTKCellType( vtkElemType ));
487 if (aConnectivities.size() > 0) {
489 GetConnect(aNodesIter,aConnect);
490 for (vtkIdType aNodeId = 0; aNodeId < aNbNodes; aNodeId++)
491 SetId(anIdList,mySMDS2VTKNodes,aConnect,aNodeId,aConnectivities[aNodeId]);
494 for( vtkIdType aNodeId = 0; aNodesIter->more(); aNodeId++ ){
495 const SMDS_MeshElement* aNode = aNodesIter->next();
496 anIdList->SetId( aNodeId, mySMDS2VTKNodes[aNode->GetID()] );
501 vtkIdType aCurId = aConnectivity->InsertNextCell( anIdList );
502 aCellTypesArray->InsertNextValue( vtkElemType );
504 //Store diameters of the balls
507 if(aType == SMDSAbs_Ball) {
508 if (const SMDS_BallElement* ball = dynamic_cast<const SMDS_BallElement*>(anElem) ) {
509 aDiam = ball->GetDiameter();
512 aScalars->SetTuple(aCurId,&aDiam);
518 SMDS_Mesh::CheckMemory(); // PAL16631
521 // Insert cells in grid
523 VTKViewer_CellLocationsArray* aCellLocationsArray = VTKViewer_CellLocationsArray::New();
524 aCellLocationsArray->SetNumberOfComponents( 1 );
525 aCellLocationsArray->SetNumberOfTuples( aNbCells );
527 SMDS_Mesh::CheckMemory(); // PAL16631
529 aConnectivity->InitTraversal();
530 for( vtkIdType idType = 0, *pts, npts; aConnectivity->GetNextCell( npts, pts ); idType++ )
531 aCellLocationsArray->SetValue( idType, aConnectivity->GetTraversalLocation( npts ) );
533 myGrid->SetCells( aCellTypesArray, aCellLocationsArray,aConnectivity );
534 myGrid->GetCellData()->SetScalars(aScalars);
536 aCellLocationsArray->Delete();
537 aCellTypesArray->Delete();
538 aConnectivity->Delete();
541 SMDS_Mesh::CheckMemory(); // PAL16631
544 //=================================================================================
545 // function : GetEdgeNodes
546 // purpose : Retrieve ids of nodes from edge of elements ( edge is numbered from 0 )
547 //=================================================================================
548 bool SMESH_VisualObjDef::GetEdgeNodes( const int theElemId,
549 const int theEdgeNum,
551 int& theNodeId2 ) const
553 const SMDS_Mesh* aMesh = GetMesh();
557 const SMDS_MeshElement* anElem = aMesh->FindElement( theElemId );
561 int nbNodes = anElem->NbCornerNodes();
563 if (( theEdgeNum < 0 || theEdgeNum > 3 ) ||
564 ( nbNodes != 3 && nbNodes != 4 ) ||
565 ( theEdgeNum >= nbNodes ))
568 theNodeId1 = anElem->GetNode( theEdgeNum )->GetID();
569 theNodeId2 = anElem->GetNode(( theEdgeNum + 1 ) % nbNodes )->GetID();
574 vtkUnstructuredGrid* SMESH_VisualObjDef::GetUnstructuredGrid()
576 if ( !myLocalGrid && !GetMesh()->isCompacted() )
578 NulData(); // detach from the SMDS grid to allow immediate memory de-allocation in compactMesh()
579 GetMesh()->compactMesh();
580 if ( SMESHDS_Mesh* m = dynamic_cast<SMESHDS_Mesh*>( GetMesh() )) // IPAL53915
581 m->GetScript()->SetModified(false); // drop IsModified set in compactMesh()
582 updateEntitiesFlags();
583 vtkUnstructuredGrid *theGrid = GetMesh()->getGrid();
584 myGrid->ShallowCopy(theGrid);
590 //=================================================================================
591 // function : IsValid
592 // purpose : Return true if there are some entities
593 //=================================================================================
594 bool SMESH_VisualObjDef::IsValid() const
596 return ( GetNbEntities(SMDSAbs_0DElement) > 0 ||
597 GetNbEntities(SMDSAbs_Ball ) > 0 ||
598 GetNbEntities(SMDSAbs_Edge ) > 0 ||
599 GetNbEntities(SMDSAbs_Face ) > 0 ||
600 GetNbEntities(SMDSAbs_Volume ) > 0 ||
601 GetNbEntities(SMDSAbs_Node ) > 0 );
604 //=================================================================================
605 // function : updateEntitiesFlags
606 // purpose : Update entities flags
607 //=================================================================================
608 void SMESH_VisualObjDef::updateEntitiesFlags()
610 unsigned int tmp = myEntitiesState;
611 ClearEntitiesFlags();
613 map<SMDSAbs_ElementType,int> entities = SMESH::GetEntitiesFromObject(this);
616 if( myEntitiesCache[SMDSAbs_0DElement] != 0 ||
617 myEntitiesCache[SMDSAbs_0DElement] >= entities[SMDSAbs_0DElement] )
618 myEntitiesState &= ~SMESH_Actor::e0DElements;
620 if( myEntitiesCache[SMDSAbs_Ball] != 0 ||
621 myEntitiesCache[SMDSAbs_Ball] >= entities[SMDSAbs_Ball] )
622 myEntitiesState &= ~SMESH_Actor::eBallElem;
624 if( myEntitiesCache[SMDSAbs_Edge] != 0 ||
625 myEntitiesCache[SMDSAbs_Edge] >= entities[SMDSAbs_Edge] )
626 myEntitiesState &= ~SMESH_Actor::eEdges;
628 if( myEntitiesCache[SMDSAbs_Face] != 0 ||
629 myEntitiesCache[SMDSAbs_Face] >= entities[SMDSAbs_Face] )
630 myEntitiesState &= ~SMESH_Actor::eFaces;
632 if( myEntitiesCache[SMDSAbs_Volume] != 0 ||
633 myEntitiesCache[SMDSAbs_Volume] >= entities[SMDSAbs_Volume] )
634 myEntitiesState &= ~SMESH_Actor::eVolumes;
636 if( tmp != myEntitiesState ) {
637 myEntitiesFlag = true;
640 myEntitiesCache = entities;
643 //=================================================================================
644 // function : ClearEntitiesFlags
645 // purpose : Clear the entities flags
646 //=================================================================================
647 void SMESH_VisualObjDef::ClearEntitiesFlags()
649 myEntitiesState = SMESH_Actor::eAllEntity;
650 myEntitiesFlag = false;
653 //=================================================================================
654 // function : GetEntitiesFlag
655 // purpose : Return the entities flag
656 //=================================================================================
657 bool SMESH_VisualObjDef::GetEntitiesFlag()
659 return myEntitiesFlag;
662 //=================================================================================
663 // function : GetEntitiesState
664 // purpose : Return the entities state
665 //=================================================================================
666 unsigned int SMESH_VisualObjDef::GetEntitiesState()
668 return myEntitiesState;
672 Class : SMESH_MeshObj
673 Description : Class for visualisation of mesh
676 //=================================================================================
677 // function : SMESH_MeshObj
678 // purpose : Constructor
679 //=================================================================================
680 SMESH_MeshObj::SMESH_MeshObj(SMESH::SMESH_Mesh_ptr theMesh):
681 myClient(SalomeApp_Application::orb(),theMesh)
685 MESSAGE("SMESH_MeshObj - this = "<<this<<"; theMesh->_is_nil() = "<<theMesh->_is_nil());
688 //=================================================================================
689 // function : ~SMESH_MeshObj
690 // purpose : Destructor
691 //=================================================================================
692 SMESH_MeshObj::~SMESH_MeshObj()
695 MESSAGE("SMESH_MeshObj - this = "<<this<<"\n");
697 myEmptyGrid->Delete();
700 //=================================================================================
702 // purpose : Update mesh and fill grid with new values if necessary
703 //=================================================================================
704 bool SMESH_MeshObj::Update( int theIsClear )
706 // Update SMDS_Mesh on client part
707 if ( MYDEBUG ) MESSAGE("SMESH_MeshObj::Update " << this);
708 if ( myClient.Update(theIsClear) || GetUnstructuredGrid()->GetNumberOfPoints()==0) {
709 if ( MYDEBUG ) MESSAGE("buildPrs");
710 buildPrs(); // Fill unstructured grid
716 bool SMESH_MeshObj::NulData()
718 if ( MYDEBUG ) MESSAGE ("SMESH_MeshObj::NulData() =============================================");
721 myEmptyGrid = SMDS_UnstructuredGrid::New();
722 myEmptyGrid->Initialize();
723 myEmptyGrid->Allocate();
724 vtkPoints* points = vtkPoints::New();
725 points->SetNumberOfPoints(0);
726 myEmptyGrid->SetPoints( points );
728 //myEmptyGrid->BuildLinks();
730 myGrid->ShallowCopy(myEmptyGrid);
733 //=================================================================================
734 // function : GetElemDimension
735 // purpose : Get dimension of element
736 //=================================================================================
737 int SMESH_MeshObj::GetElemDimension( const int theObjId )
739 const SMDS_MeshElement* anElem = myClient->FindElement( theObjId );
743 int aType = anElem->GetType();
746 case SMDSAbs_0DElement : return 0;
747 case SMDSAbs_Ball : return 0;
748 case SMDSAbs_Edge : return 1;
749 case SMDSAbs_Face : return 2;
750 case SMDSAbs_Volume: return 3;
755 //=================================================================================
756 // function : GetEntities
757 // purpose : Get entities of specified type. Return number of entities
758 //=================================================================================
759 int SMESH_MeshObj::GetNbEntities( const SMDSAbs_ElementType theType) const
765 return myClient->NbNodes();
768 case SMDSAbs_0DElement:
770 return myClient->Nb0DElements();
774 return myClient->NbBalls();
779 return myClient->NbEdges();
784 return myClient->NbFaces();
789 return myClient->NbVolumes();
798 int SMESH_MeshObj::GetEntities( const SMDSAbs_ElementType theType, TEntityList& theObjs ) const
806 SMDS_NodeIteratorPtr anIter = myClient->nodesIterator();
807 while ( anIter->more() ) theObjs.push_back( anIter->next() );
810 case SMDSAbs_0DElement:
812 SMDS_ElemIteratorPtr anIter = myClient->elementsIterator(SMDSAbs_0DElement);
813 while ( anIter->more() ) theObjs.push_back( anIter->next() );
818 SMDS_ElemIteratorPtr anIter = myClient->elementGeomIterator(SMDSGeom_BALL);
819 while ( anIter->more() ) theObjs.push_back( anIter->next() );
824 SMDS_EdgeIteratorPtr anIter = myClient->edgesIterator();
825 while ( anIter->more() ) theObjs.push_back( anIter->next() );
830 SMDS_FaceIteratorPtr anIter = myClient->facesIterator();
831 while ( anIter->more() ) theObjs.push_back( anIter->next() );
836 SMDS_VolumeIteratorPtr anIter = myClient->volumesIterator();
837 while ( anIter->more() ) theObjs.push_back( anIter->next() );
844 return theObjs.size();
847 //=================================================================================
848 // function : UpdateFunctor
849 // purpose : Update functor in accordance with current mesh
850 //=================================================================================
851 void SMESH_MeshObj::UpdateFunctor( const SMESH::Controls::FunctorPtr& theFunctor )
853 theFunctor->SetMesh( GetMesh() );
856 //=================================================================================
857 // function : IsNodePrs
858 // purpose : Return true if node presentation is used
859 //=================================================================================
860 bool SMESH_MeshObj::IsNodePrs() const
862 return myClient->Nb0DElements() + myClient->NbEdges() + myClient->NbFaces() + myClient->NbVolumes() + myClient->NbBalls() == 0 ;
867 Class : SMESH_SubMeshObj
868 Description : Base class for visualisation of submeshes and groups
871 //=================================================================================
872 // function : SMESH_SubMeshObj
873 // purpose : Constructor
874 //=================================================================================
875 SMESH_SubMeshObj::SMESH_SubMeshObj( SMESH_MeshObj* theMeshObj )
877 if ( MYDEBUG ) MESSAGE( "SMESH_SubMeshObj - theMeshObj = " << theMeshObj );
879 myMeshObj = theMeshObj;
882 SMESH_SubMeshObj::~SMESH_SubMeshObj()
886 //=================================================================================
887 // function : GetElemDimension
888 // purpose : Get dimension of element
889 //=================================================================================
890 int SMESH_SubMeshObj::GetElemDimension( const int theObjId )
892 return myMeshObj == 0 ? 0 : myMeshObj->GetElemDimension( theObjId );
895 //=================================================================================
896 // function : UpdateFunctor
897 // purpose : Update functor in accordance with current mesh
898 //=================================================================================
900 void SMESH_SubMeshObj::UpdateFunctor( const SMESH::Controls::FunctorPtr& theFunctor )
902 theFunctor->SetMesh( myMeshObj->GetMesh() );
905 //=================================================================================
907 // purpose : Update mesh object and fill grid with new values
908 //=================================================================================
909 bool SMESH_SubMeshObj::Update( int theIsClear )
911 if ( MYDEBUG ) MESSAGE("SMESH_SubMeshObj::Update " << this)
912 bool changed = myMeshObj->Update( theIsClear );
919 Class : SMESH_GroupObj
920 Description : Class for visualisation of groups
923 //=================================================================================
924 // function : SMESH_GroupObj
925 // purpose : Constructor
926 //=================================================================================
927 SMESH_GroupObj::SMESH_GroupObj( SMESH::SMESH_GroupBase_ptr theGroup,
928 SMESH_MeshObj* theMeshObj )
929 : SMESH_SubMeshObj( theMeshObj ),
930 myGroupServer( SMESH::SMESH_GroupBase::_duplicate(theGroup) )
932 if ( MYDEBUG ) MESSAGE("SMESH_GroupObj - theGroup->_is_nil() = "<<theGroup->_is_nil());
933 myGroupServer->Register();
936 SMESH_GroupObj::~SMESH_GroupObj()
938 if ( MYDEBUG ) MESSAGE("~SMESH_GroupObj");
939 myGroupServer->UnRegister();
942 //=================================================================================
943 // function : IsNodePrs
944 // purpose : Return true if node presentation is used
945 //=================================================================================
946 bool SMESH_GroupObj::IsNodePrs() const
948 return myGroupServer->GetType() == SMESH::NODE;
951 //=================================================================================
952 // function : GetElementType
953 // purpose : Return type of elements of group
954 //=================================================================================
955 SMDSAbs_ElementType SMESH_GroupObj::GetElementType() const
957 return SMDSAbs_ElementType(myGroupServer->GetType());
960 //=================================================================================
961 // function : getNodesFromElems
962 // purpose : Retrieve nodes from elements
963 //=================================================================================
964 static int getNodesFromElems( SMESH::long_array_var& theElemIds,
965 const SMDS_Mesh* theMesh,
966 std::list<const SMDS_MeshElement*>& theResList )
968 set<const SMDS_MeshElement*> aNodeSet;
970 for ( CORBA::Long i = 0, n = theElemIds->length(); i < n; i++ )
972 const SMDS_MeshElement* anElem = theMesh->FindElement( theElemIds[ i ] );
975 SMDS_ElemIteratorPtr anIter = anElem->nodesIterator();
976 while ( anIter->more() )
978 const SMDS_MeshElement* aNode = anIter->next();
980 aNodeSet.insert( aNode );
985 set<const SMDS_MeshElement*>::const_iterator anIter;
986 for ( anIter = aNodeSet.begin(); anIter != aNodeSet.end(); ++anIter )
987 theResList.push_back( *anIter );
989 return theResList.size();
992 //=================================================================================
993 // function : getPointers
994 // purpose : Get std::list<const SMDS_MeshElement*> from list of IDs
995 //=================================================================================
996 static int getPointers( const SMDSAbs_ElementType theRequestType,
997 SMESH::long_array_var& theElemIds,
998 const SMDS_Mesh* theMesh,
999 std::list<const SMDS_MeshElement*>& theResList )
1001 for ( CORBA::Long i = 0, n = theElemIds->length(); i < n; i++ )
1003 const SMDS_MeshElement* anElem = theRequestType == SMDSAbs_Node
1004 ? theMesh->FindNode( theElemIds[ i ] ) : theMesh->FindElement( theElemIds[ i ] );
1007 theResList.push_back( anElem );
1010 return theResList.size();
1014 //=================================================================================
1015 // function : GetEntities
1016 // purpose : Get entities of specified type. Return number of entities
1017 //=================================================================================
1018 int SMESH_GroupObj::GetNbEntities( const SMDSAbs_ElementType theType) const
1020 if(SMDSAbs_ElementType(myGroupServer->GetType()) == theType) {
1021 return myGroupServer->Size();
1023 if ( theType == SMDSAbs_Node ) {
1024 return myGroupServer->GetNumberOfNodes();
1029 int SMESH_GroupObj::GetEntities( const SMDSAbs_ElementType theType, TEntityList& theResList ) const
1032 SMDS_Mesh* aMesh = myMeshObj->GetMesh();
1037 SMDSAbs_ElementType aGrpType = SMDSAbs_ElementType(myGroupServer->GetType());
1038 if ( aGrpType != theType && theType != SMDSAbs_Node )
1041 SMESH::long_array_var anIds = myGroupServer->GetListOfID();
1042 if ( anIds->length() == 0 )
1045 if ( aGrpType == theType )
1046 return getPointers( theType, anIds, aMesh, theResList );
1047 else if ( theType == SMDSAbs_Node )
1048 return getNodesFromElems( anIds, aMesh, theResList );
1056 Class : SMESH_subMeshObj
1057 Description : Class for visualisation of submeshes
1060 //=================================================================================
1061 // function : SMESH_subMeshObj
1062 // purpose : Constructor
1063 //=================================================================================
1064 SMESH_subMeshObj::SMESH_subMeshObj( SMESH::SMESH_subMesh_ptr theSubMesh,
1065 SMESH_MeshObj* theMeshObj )
1066 : SMESH_SubMeshObj( theMeshObj ),
1067 mySubMeshServer( SMESH::SMESH_subMesh::_duplicate( theSubMesh ) )
1069 if ( MYDEBUG ) MESSAGE( "SMESH_subMeshObj - theSubMesh->_is_nil() = " << theSubMesh->_is_nil() );
1071 mySubMeshServer->Register();
1074 SMESH_subMeshObj::~SMESH_subMeshObj()
1076 if ( MYDEBUG ) MESSAGE( "~SMESH_subMeshObj" );
1077 mySubMeshServer->UnRegister();
1080 //=================================================================================
1081 // function : GetEntities
1082 // purpose : Get entities of specified type. Return number of entities
1083 //=================================================================================
1084 int SMESH_subMeshObj::GetNbEntities( const SMDSAbs_ElementType theType) const
1090 return mySubMeshServer->GetNumberOfNodes( /*all=*/true );
1094 case SMDSAbs_0DElement:
1097 case SMDSAbs_Volume:
1099 SMESH::long_array_var anIds =
1100 mySubMeshServer->GetElementsByType( SMESH::ElementType(theType) );
1101 return anIds->length();
1109 int SMESH_subMeshObj::GetEntities( const SMDSAbs_ElementType theType, TEntityList& theResList ) const
1113 SMDS_Mesh* aMesh = myMeshObj->GetMesh();
1117 bool isNodal = IsNodePrs();
1121 if ( theType == SMDSAbs_Node )
1123 SMESH::long_array_var anIds = mySubMeshServer->GetNodesId();
1124 return getPointers( SMDSAbs_Node, anIds, aMesh, theResList );
1129 if ( theType == SMDSAbs_Node )
1131 SMESH::long_array_var anIds = mySubMeshServer->GetElementsId();
1132 return getNodesFromElems( anIds, aMesh, theResList );
1136 SMESH::long_array_var anIds =
1137 mySubMeshServer->GetElementsByType( SMESH::ElementType(theType) );
1138 return getPointers( theType, anIds, aMesh, theResList );
1145 //=================================================================================
1146 // function : IsNodePrs
1147 // purpose : Return true if node presentation is used
1148 //=================================================================================
1149 bool SMESH_subMeshObj::IsNodePrs() const
1151 return mySubMeshServer->GetNumberOfElements() == 0;