src/SMDS/SMDS_FaceOfNodes.cxx

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