src/SMDS/SMDS_FaceOfNodes.cxx

   1 //  Copyright (C) 2007-2010  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 //  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   4 //  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   5 //
   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.
  10 //
  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.
  15 //
  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
  19 //
  20 //  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21 //
  22
  23 //  SMESH SMDS : implementaion of Salome mesh data structure
  24 //
  25 #ifdef _MSC_VER
  26 #pragma warning(disable:4786)
  27 #endif
  28
  29 #include "SMDS_SetIterator.hxx"
  30 #include "SMDS_FaceOfNodes.hxx"
  31 #include "SMDS_IteratorOfElements.hxx"
  32 #include "SMDS_MeshNode.hxx"
  33 #include "SMDS_Mesh.hxx"
  34
  35 #include "utilities.h"
  36
  37 using namespace std;
  38
  39 //=======================================================================
  40 //function : NbEdges
  41 //purpose  :
  42 //=======================================================================
  43
  44 int SMDS_FaceOfNodes::NbEdges() const
  45 {
  46         return NbNodes();
  47 }
  48
  49 int SMDS_FaceOfNodes::NbFaces() const
  50 {
  51         return 1;
  52 }
  53
  54 int SMDS_FaceOfNodes::NbNodes() const
  55 {
  56         return myNbNodes;
  57 }
  58
  59 //=======================================================================
  60 //function : Print
  61 //purpose  :
  62 //=======================================================================
  63
  64 void SMDS_FaceOfNodes::Print(ostream & OS) const
  65 {
  66         OS << "face <" << GetID() << " > : ";
  67         int i;
  68         for (i = 0; i < NbNodes() - 1; i++) OS << myNodes[i] << ",";
  69         OS << myNodes[i] << ") " << endl;
  70 }
  71
  72 //=======================================================================
  73 //function : elementsIterator
  74 //purpose  :
  75 //=======================================================================
  76
  77 class SMDS_FaceOfNodes_MyIterator:public SMDS_NodeArrayElemIterator
  78 {
  79  public:
  80   SMDS_FaceOfNodes_MyIterator(const SMDS_MeshNode* const *s, int l):
  81     SMDS_NodeArrayElemIterator( s, & s[ l ] ) {}
  82 };
  83
  84 /// ===================================================================
  85 /*!
  86  * \brief Iterator on edges of face
  87  */
  88 /// ===================================================================
  89
  90 class _MyEdgeIterator : public SMDS_ElemIterator
  91 {
  92   vector< const SMDS_MeshElement* > myElems;
  93   int myIndex;
  94 public:
  95   _MyEdgeIterator(const SMDS_FaceOfNodes* face):myIndex(0) {
  96     myElems.reserve( face->NbNodes() );
  97     for ( int i = 0; i < face->NbNodes(); ++i ) {
  98       const SMDS_MeshElement* edge =
  99         SMDS_Mesh::FindEdge( face->GetNode( i ), face->GetNodeWrap( i + 1 ));
 100       if ( edge )
 101         myElems.push_back( edge );
 102     }
 103   }
 104   /// Return true if and only if there are other object in this iterator
 105   virtual bool more() { return myIndex < myElems.size(); }
 106
 107   /// Return the current object and step to the next one
 108   virtual const SMDS_MeshElement* next() { return myElems[ myIndex++ ]; }
 109 };
 110
 111 SMDS_ElemIteratorPtr SMDS_FaceOfNodes::elementsIterator
 112                          (SMDSAbs_ElementType type) const
 113 {
 114   switch(type)
 115   {
 116   case SMDSAbs_Face:
 117     return SMDS_MeshElement::elementsIterator(SMDSAbs_Face);
 118   case SMDSAbs_Node:
 119     return SMDS_ElemIteratorPtr(new SMDS_FaceOfNodes_MyIterator(myNodes,myNbNodes));
 120   case SMDSAbs_Edge:
 121     return SMDS_ElemIteratorPtr(new _MyEdgeIterator( this ));
 122     break;
 123   default:
 124     return SMDS_ElemIteratorPtr
 125       (new SMDS_IteratorOfElements
 126        (this,type,SMDS_ElemIteratorPtr
 127         (new SMDS_FaceOfNodes_MyIterator(myNodes,myNbNodes))));
 128   }
 129   return SMDS_ElemIteratorPtr();
 130 }
 131
 132 SMDS_FaceOfNodes::SMDS_FaceOfNodes(const SMDS_MeshNode* node1,
 133                                    const SMDS_MeshNode* node2,
 134                                    const SMDS_MeshNode* node3)
 135 {
 136         myNbNodes = 3;
 137         myNodes[0]=node1;
 138         myNodes[1]=node2;
 139         myNodes[2]=node3;
 140         myNodes[3]=0;
 141 }
 142
 143 SMDS_FaceOfNodes::SMDS_FaceOfNodes(const SMDS_MeshNode* node1,
 144                                    const SMDS_MeshNode* node2,
 145                                    const SMDS_MeshNode* node3,
 146                                    const SMDS_MeshNode* node4)
 147 {
 148         myNbNodes = 4;
 149         myNodes[0]=node1;
 150         myNodes[1]=node2;
 151         myNodes[2]=node3;
 152         myNodes[3]=node4;
 153 }
 154 bool SMDS_FaceOfNodes::ChangeNodes(const SMDS_MeshNode* nodes[],
 155                                    const int            nbNodes)
 156 {
 157   myNbNodes = nbNodes;
 158   myNodes[0]=nodes[0];
 159   myNodes[1]=nodes[1];
 160   myNodes[2]=nodes[2];
 161   if (nbNodes == 4)
 162     myNodes[3]=nodes[3];
 163   else if (nbNodes != 3)
 164     return false;
 165
 166   return true;
 167 }
 168
 169 /*!
 170  * \brief Return node by its index
 171  * \param ind - node index
 172  * \retval const SMDS_MeshNode* - the node
 173  */
 174 const SMDS_MeshNode* SMDS_FaceOfNodes::GetNode(const int ind) const
 175 {
 176   return myNodes[ ind ];
 177 }
 178
 179 SMDSAbs_EntityType SMDS_FaceOfNodes::GetEntityType() const
 180 {
 181   return myNbNodes == 3 ? SMDSEntity_Triangle : SMDSEntity_Quadrangle;
 182 }