src/SMDS/SMDS_FaceOfNodes.cxx

   1 // Copyright (C) 2007-2016  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, or (at your option) any later version.
  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 : implementation 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   size_t                            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( 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   //MESSAGE("******************************************************* SMDS_FaceOfNodes");
 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   //MESSAGE("******************************************************* SMDS_FaceOfNodes");
 149         myNbNodes = 4;
 150         myNodes[0]=node1;
 151         myNodes[1]=node2;
 152         myNodes[2]=node3;
 153         myNodes[3]=node4;
 154 }
 155 bool SMDS_FaceOfNodes::ChangeNodes(const SMDS_MeshNode* nodes[],
 156                                    const int            nbNodes)
 157 {
 158   myNbNodes = nbNodes;
 159   myNodes[0]=nodes[0];
 160   myNodes[1]=nodes[1];
 161   myNodes[2]=nodes[2];
 162   if (nbNodes == 4)
 163     myNodes[3]=nodes[3];
 164   else if (nbNodes != 3)
 165     return false;
 166
 167   return true;
 168 }
 169
 170 /*!
 171  * \brief Return node by its index
 172  * \param ind - node index
 173  * \retval const SMDS_MeshNode* - the node
 174  */
 175 const SMDS_MeshNode* SMDS_FaceOfNodes::GetNode(const int ind) const
 176 {
 177   return myNodes[ ind ];
 178 }
 179
 180 SMDSAbs_EntityType SMDS_FaceOfNodes::GetEntityType() const
 181 {
 182   return NbNodes() == 3 ? SMDSEntity_Triangle : SMDSEntity_Quadrangle;
 183 }
 184 SMDSAbs_GeometryType SMDS_FaceOfNodes::GetGeomType() const
 185 {
 186   return NbNodes() == 3 ? SMDSGeom_TRIANGLE : SMDSGeom_QUADRANGLE;
 187 }