]> SALOME platform Git repositories - modules/smesh.git/blob - src/SMDS/SMDS_BallElement.cxx
Salome HOME
Correction of mistakes: 0022537: EDF 2877 SMESH : wrong information about the numbre...
[modules/smesh.git] / src / SMDS / SMDS_BallElement.cxx
1 // Copyright (C) 2010-2014  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
3 // This library is free software; you can redistribute it and/or
4 // modify it under the terms of the GNU Lesser General Public
5 // License as published by the Free Software Foundation; either
6 // version 2.1 of the License, or (at your option) any later version.
7 //
8 // This library is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11 // Lesser General Public License for more details.
12 //
13 // You should have received a copy of the GNU Lesser General Public
14 // License along with this library; if not, write to the Free Software
15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
16 //
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
18 //
19
20 //  SMESH SMDS : implementaion of Salome mesh data structure
21 //  Module     : SMESH
22 //  File       : SMDS_BallElement.cxx
23 //  Author     : Edward AGAPOV (eap)
24
25 #include "SMDS_BallElement.hxx"
26
27 #include "SMDS_ElemIterator.hxx"
28 #include "SMDS_Mesh.hxx"
29 #include "SMDS_MeshNode.hxx"
30 #include "SMDS_VtkCellIterator.hxx"
31
32 SMDS_BallElement::SMDS_BallElement()
33 {
34   SMDS_MeshCell::init();
35 }
36
37 SMDS_BallElement::SMDS_BallElement (const SMDS_MeshNode * node, double diameter)
38 {
39   init( node->getVtkId(), diameter, SMDS_Mesh::_meshList[ node->getMeshId() ] );
40 }
41
42 SMDS_BallElement::SMDS_BallElement(vtkIdType nodeId, double diameter, SMDS_Mesh* mesh)
43 {
44   init( nodeId, diameter, mesh );
45 }
46
47 void SMDS_BallElement::init(vtkIdType nodeId, double diameter, SMDS_Mesh* mesh)
48 {
49   SMDS_MeshCell::init();
50   SMDS_UnstructuredGrid* grid = mesh->getGrid();
51   myVtkID = grid->InsertNextLinkedCell( GetVtkType(), 1, &nodeId );
52   myMeshId = mesh->getMeshId();
53   grid->SetBallDiameter( myVtkID, diameter );
54   mesh->setMyModified();
55 }
56
57 double SMDS_BallElement::GetDiameter() const
58 {
59   return SMDS_Mesh::_meshList[myMeshId]->getGrid()->GetBallDiameter( myVtkID );
60 }
61
62 void SMDS_BallElement::SetDiameter(double diameter)
63 {
64   SMDS_Mesh::_meshList[myMeshId]->getGrid()->SetBallDiameter( myVtkID, diameter );
65 }
66
67 bool SMDS_BallElement::ChangeNode (const SMDS_MeshNode * node)
68 {
69   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
70   vtkIdType npts = 0;
71   vtkIdType* pts = 0;
72   grid->GetCellPoints(myVtkID, npts, pts);
73   pts[0] = node->getVtkId();
74   SMDS_Mesh::_meshList[myMeshId]->setMyModified();
75   return true;
76 }
77
78 void SMDS_BallElement::Print (std::ostream & OS) const
79 {
80   OS << "ball<" << GetID() << "> : ";
81 }
82
83 const SMDS_MeshNode* SMDS_BallElement::GetNode (const int ind) const
84 {
85   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
86   vtkIdType npts, *pts;
87   grid->GetCellPoints( myVtkID, npts, pts );
88   return SMDS_Mesh::_meshList[myMeshId]->FindNodeVtk( pts[ 0 ]);
89 }
90
91 SMDS_ElemIteratorPtr SMDS_BallElement::elementsIterator (SMDSAbs_ElementType type) const
92 {
93   switch (type)
94   {
95     case SMDSAbs_Node:
96       return SMDS_ElemIteratorPtr(new SMDS_VtkCellIterator(SMDS_Mesh::_meshList[myMeshId], myVtkID, GetEntityType()));
97     default:
98       ;
99       return SMDS_ElemIteratorPtr((SMDS_ElemIterator*) NULL);
100   }
101 }
102