src/SMDS/SMDS_Mesh0DElement.cxx

   1 // Copyright (C) 2007-2011  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.
   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 //  File   : SMDS_Mesh0DElement.cxx
  22 //  Author : Jean-Michel BOULCOURT
  23 //  Module : SMESH
  24 //
  25 #ifdef _MSC_VER
  26 #pragma warning(disable:4786)
  27 #endif
  28
  29 #include "SMDS_Mesh0DElement.hxx"
  30 #include "SMDS_IteratorOfElements.hxx"
  31 #include "SMDS_MeshNode.hxx"
  32 #include "utilities.h"
  33
  34 using namespace std;
  35
  36 //=======================================================================
  37 //function : SMDS_Mesh0DElement
  38 //purpose  :
  39 //=======================================================================
  40 SMDS_Mesh0DElement::SMDS_Mesh0DElement (const SMDS_MeshNode * node)
  41 {
  42   MESSAGE("SMDS_Mesh0DElement " << GetID());
  43   myNode = node;
  44 }
  45
  46 //=======================================================================
  47 //function : Print
  48 //purpose  :
  49 //=======================================================================
  50 void SMDS_Mesh0DElement::Print (ostream & OS) const
  51 {
  52   OS << "0D Element <" << GetID() << "> : (" << myNode << ") " << endl;
  53 }
  54
  55 //=======================================================================
  56 //function : NbNodes
  57 //purpose  :
  58 //=======================================================================
  59 int SMDS_Mesh0DElement::NbNodes() const
  60 {
  61   return 1;
  62 }
  63
  64 //=======================================================================
  65 //function : NbEdges
  66 //purpose  :
  67 //=======================================================================
  68 int SMDS_Mesh0DElement::NbEdges() const
  69 {
  70   return 0;
  71 }
  72
  73 //=======================================================================
  74 //function : GetType
  75 //purpose  :
  76 //=======================================================================
  77 SMDSAbs_ElementType SMDS_Mesh0DElement::GetType() const
  78 {
  79   return SMDSAbs_0DElement;
  80 }
  81
  82 vtkIdType SMDS_Mesh0DElement::GetVtkType() const
  83 {
  84   return VTK_VERTEX;
  85 }
  86
  87 //=======================================================================
  88 //function : elementsIterator
  89 //purpose  :
  90 //=======================================================================
  91 class SMDS_Mesh0DElement_MyNodeIterator: public SMDS_ElemIterator
  92 {
  93   const SMDS_MeshNode * myNode;
  94   int myIndex;
  95  public:
  96   SMDS_Mesh0DElement_MyNodeIterator(const SMDS_MeshNode * node):
  97     myNode(node),myIndex(0) {}
  98
  99   bool more()
 100   {
 101     return myIndex < 1;
 102   }
 103
 104   const SMDS_MeshElement* next()
 105   {
 106     myIndex++;
 107     if (myIndex == 1)
 108       return myNode;
 109     return NULL;
 110   }
 111 };
 112
 113 SMDS_ElemIteratorPtr SMDS_Mesh0DElement::elementsIterator (SMDSAbs_ElementType type) const
 114 {
 115   switch(type)
 116   {
 117   case SMDSAbs_0DElement:
 118     return SMDS_MeshElement::elementsIterator(SMDSAbs_0DElement);
 119   case SMDSAbs_Node:
 120     return SMDS_ElemIteratorPtr(new SMDS_Mesh0DElement_MyNodeIterator(myNode));
 121   default:
 122     return SMDS_ElemIteratorPtr
 123       (new SMDS_IteratorOfElements
 124        (this,type, SMDS_ElemIteratorPtr(new SMDS_Mesh0DElement_MyNodeIterator(myNode))));
 125   }
 126 }
 127
 128 //=======================================================================
 129 //function : operator<
 130 //purpose  :
 131 //=======================================================================
 132 bool operator< (const SMDS_Mesh0DElement & e1, const SMDS_Mesh0DElement & e2)
 133 {
 134   int id1 = e1.myNode->getVtkId();
 135   int id2 = e2.myNode->getVtkId();
 136
 137   return (id1 < id2);
 138 }
 139
 140 /*!
 141  * \brief Return node by its index
 142  * \param ind - node index
 143  * \retval const SMDS_MeshNode* - the node
 144  */
 145 const SMDS_MeshNode* SMDS_Mesh0DElement::GetNode(const int ind) const
 146 {
 147   if (ind == 0)
 148     return myNode;
 149   return NULL;
 150 }
 151
 152 //=======================================================================
 153 //function : ChangeNode
 154 //purpose  :
 155 //=======================================================================
 156 bool SMDS_Mesh0DElement::ChangeNode (const SMDS_MeshNode * node)
 157 {
 158   myNode = node;
 159   return true;
 160 }