1 #include "SMDS_VtkEdge.hxx"
2 #include "SMDS_MeshNode.hxx"
3 #include "SMDS_Mesh.hxx"
4 #include "SMDS_VtkCellIterator.hxx"
13 SMDS_VtkEdge::SMDS_VtkEdge()
17 SMDS_VtkEdge::SMDS_VtkEdge(std::vector<vtkIdType> nodeIds, SMDS_Mesh* mesh)
22 SMDS_VtkEdge::~SMDS_VtkEdge()
26 void SMDS_VtkEdge::init(std::vector<vtkIdType> nodeIds, SMDS_Mesh* mesh)
28 SMDS_MeshEdge::init();
29 vtkUnstructuredGrid* grid = mesh->getGrid();
30 myMeshId = mesh->getMeshId();
31 vtkIdType aType = VTK_LINE;
32 if (nodeIds.size() == 3)
33 aType = VTK_QUADRATIC_EDGE;
34 myVtkID = grid->InsertNextLinkedCell(aType, nodeIds.size(), &nodeIds[0]);
35 mesh->setMyModified();
36 //MESSAGE("SMDS_VtkEdge::init myVtkID " << myVtkID);
39 bool SMDS_VtkEdge::ChangeNodes(const SMDS_MeshNode * node1, const SMDS_MeshNode * node2)
41 const SMDS_MeshNode* nodes[] = { node1, node2 };
42 SMDS_Mesh::_meshList[myMeshId]->setMyModified();
43 return ChangeNodes(nodes, 2);
46 bool SMDS_VtkEdge::ChangeNodes(const SMDS_MeshNode* nodes[], const int nbNodes)
48 vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
51 grid->GetCellPoints(myVtkID, npts, pts);
54 MESSAGE("ChangeNodes problem: not the same number of nodes " << npts << " -> " << nbNodes);
57 for (int i = 0; i < nbNodes; i++)
59 pts[i] = nodes[i]->getVtkId();
61 SMDS_Mesh::_meshList[myMeshId]->setMyModified();
65 bool SMDS_VtkEdge::IsMediumNode(const SMDS_MeshNode* node) const
67 vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
70 grid->GetCellPoints(myVtkID, npts, pts);
71 //MESSAGE("IsMediumNode " << npts << " " << (node->getVtkId() == pts[npts-1]));
72 return ((npts == 3) && (node->getVtkId() == pts[2]));
75 void SMDS_VtkEdge::Print(std::ostream & OS) const
77 OS << "edge <" << GetID() << "> : ";
80 int SMDS_VtkEdge::NbNodes() const
82 vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
83 int nbPoints = grid->GetCell(myVtkID)->GetNumberOfPoints();
84 assert(nbPoints >= 2);
88 int SMDS_VtkEdge::NbEdges() const
93 SMDSAbs_EntityType SMDS_VtkEdge::GetEntityType() const
96 return SMDSEntity_Edge;
98 return SMDSEntity_Quad_Edge;
101 vtkIdType SMDS_VtkEdge::GetVtkType() const
106 return VTK_QUADRATIC_EDGE;
111 * \brief Return node by its index
112 * \param ind - node index
113 * \retval const SMDS_MeshNode* - the node
116 SMDS_VtkEdge::GetNode(const int ind) const
119 return SMDS_MeshElement::GetNode(ind);
122 bool SMDS_VtkEdge::IsQuadratic() const
124 if (this->NbNodes() > 2)
130 SMDS_ElemIteratorPtr SMDS_VtkEdge::elementsIterator(SMDSAbs_ElementType type) const
135 return SMDS_ElemIteratorPtr(new SMDS_VtkCellIterator(SMDS_Mesh::_meshList[myMeshId], myVtkID, GetEntityType()));
137 MESSAGE("ERROR : Iterator not implemented")
139 return SMDS_ElemIteratorPtr((SMDS_ElemIterator*) NULL);
143 SMDS_ElemIteratorPtr SMDS_VtkEdge::nodesIteratorToUNV() const
145 return SMDS_ElemIteratorPtr(new SMDS_VtkCellIteratorToUNV(SMDS_Mesh::_meshList[myMeshId], myVtkID, GetEntityType()));
148 SMDS_ElemIteratorPtr SMDS_VtkEdge::interlacedNodesElemIterator() const
150 return SMDS_ElemIteratorPtr(new SMDS_VtkCellIteratorToUNV(SMDS_Mesh::_meshList[myMeshId], myVtkID, GetEntityType()));