src/SMDS/SMDS_FaceOfNodes.cxx

   1 // Copyright (C) 2007-2011  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   //MESSAGE("******************************************************* SMDS_FaceOfNodes");
 137         myNbNodes = 3;
 138         myNodes[0]=node1;
 139         myNodes[1]=node2;
 140         myNodes[2]=node3;
 141         myNodes[3]=0;
 142 }
 143
 144 SMDS_FaceOfNodes::SMDS_FaceOfNodes(const SMDS_MeshNode* node1,
 145                                    const SMDS_MeshNode* node2,
 146                                    const SMDS_MeshNode* node3,
 147                                    const SMDS_MeshNode* node4)
 148 {
 149   //MESSAGE("******************************************************* SMDS_FaceOfNodes");
 150         myNbNodes = 4;
 151         myNodes[0]=node1;
 152         myNodes[1]=node2;
 153         myNodes[2]=node3;
 154         myNodes[3]=node4;
 155 }
 156 bool SMDS_FaceOfNodes::ChangeNodes(const SMDS_MeshNode* nodes[],
 157                                    const int            nbNodes)
 158 {
 159   myNbNodes = nbNodes;
 160   myNodes[0]=nodes[0];
 161   myNodes[1]=nodes[1];
 162   myNodes[2]=nodes[2];
 163   if (nbNodes == 4)
 164     myNodes[3]=nodes[3];
 165   else if (nbNodes != 3)
 166     return false;
 167
 168   return true;
 169 }
 170
 171 /*!
 172  * \brief Return node by its index
 173  * \param ind - node index
 174  * \retval const SMDS_MeshNode* - the node
 175  */
 176 const SMDS_MeshNode* SMDS_FaceOfNodes::GetNode(const int ind) const
 177 {
 178   return myNodes[ ind ];
 179 }
 180
 181 SMDSAbs_EntityType SMDS_FaceOfNodes::GetEntityType() const
 182 {
 183   return myNbNodes == 3 ? SMDSEntity_Triangle : SMDSEntity_Quadrangle;
 184 }