src/SMDS/SMDS_VtkEdge.cxx

   1 #include "SMDS_VtkEdge.hxx"
   2 #include "SMDS_MeshNode.hxx"
   3 #include "SMDS_Mesh.hxx"
   4 #include "SMDS_VtkCellIterator.hxx"
   5
   6 #include "utilities.h"
   7
   8 #include <vector>
   9
  10 using namespace std;
  11
  12 SMDS_VtkEdge::SMDS_VtkEdge()
  13 {
  14 }
  15
  16 SMDS_VtkEdge::SMDS_VtkEdge(std::vector<vtkIdType> nodeIds, SMDS_Mesh* mesh)
  17 {
  18         init(nodeIds, mesh);
  19 }
  20
  21 SMDS_VtkEdge::~SMDS_VtkEdge()
  22 {
  23 }
  24
  25 void SMDS_VtkEdge::init(std::vector<vtkIdType> nodeIds, SMDS_Mesh* mesh)
  26 {
  27         vtkUnstructuredGrid* grid = mesh->getGrid();
  28         myIdInShape = -1;
  29         myMeshId = mesh->getMeshId();
  30         myVtkID = grid->InsertNextLinkedCell(VTK_LINE, nodeIds.size(), &nodeIds[0]);
  31 }
  32
  33 bool SMDS_VtkEdge::ChangeNodes(const SMDS_MeshNode * node1,
  34                 const SMDS_MeshNode * node2)
  35 {
  36         return true;
  37 }
  38
  39 void SMDS_VtkEdge::Print(std::ostream & OS) const
  40 {
  41         OS << "edge <" << GetID() << "> : ";
  42 }
  43
  44 int SMDS_VtkEdge::NbNodes() const
  45 {
  46         return 2;
  47 }
  48
  49 int SMDS_VtkEdge::NbEdges() const
  50 {
  51         return 1;
  52 }
  53
  54 SMDSAbs_EntityType SMDS_VtkEdge::GetEntityType() const
  55 {
  56         return SMDSEntity_Edge;
  57 }
  58
  59 vtkIdType SMDS_VtkEdge::GetVtkType() const
  60 {
  61         return VTK_LINE;
  62 }
  63
  64 /*!
  65  * \brief Return node by its index
  66  * \param ind - node index
  67  * \retval const SMDS_MeshNode* - the node
  68  */
  69 const SMDS_MeshNode*
  70 SMDS_VtkEdge::GetNode(const int ind) const
  71 {
  72         return 0;
  73 }
  74
  75 SMDS_ElemIteratorPtr SMDS_VtkEdge::elementsIterator(SMDSAbs_ElementType type) const
  76 {
  77         switch (type)
  78         {
  79         case SMDSAbs_Node:
  80                 return SMDS_ElemIteratorPtr(new SMDS_VtkCellIterator(
  81                                 SMDS_Mesh::_meshList[myMeshId], myVtkID, GetEntityType()));
  82         default:
  83                 MESSAGE("ERROR : Iterator not implemented");
  84                 return SMDS_ElemIteratorPtr((SMDS_ElemIterator*) NULL);
  85         }
  86 }