Salome HOME
Merge from BR_V5_DEV 16Feb09
[modules/smesh.git] / src / SMDS / SMDS_MeshEdge.cxx
index cbb448286ed7224851c619f6807fdd06a0e0ff78..9d17cd439bd043c60d9dd89a88fdfcc21f419d72 100644 (file)
-using namespace std;
-// File:       SMDS_MeshEdge.cxx
-// Created:    Wed Jan 23 17:02:20 2002
-// Author:     Jean-Michel BOULCOURT
-//             <jmb@coulox.paris1.matra-dtv.fr>
+//  Copyright (C) 2007-2008  CEA/DEN, EDF R&D, OPEN CASCADE
+//
+//  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
+//  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
+//
+//  This library is free software; you can redistribute it and/or
+//  modify it under the terms of the GNU Lesser General Public
+//  License as published by the Free Software Foundation; either
+//  version 2.1 of the License.
+//
+//  This library is distributed in the hope that it will be useful,
+//  but WITHOUT ANY WARRANTY; without even the implied warranty of
+//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+//  Lesser General Public License for more details.
+//
+//  You should have received a copy of the GNU Lesser General Public
+//  License along with this library; if not, write to the Free Software
+//  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
+//
+//  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
+//
+//  SMESH SMDS : implementaion of Salome mesh data structure
+//  File   : SMDS_MeshEdge.cxx
+//  Author : Jean-Michel BOULCOURT
+//  Module : SMESH
+//
+#ifdef _MSC_VER
+#pragma warning(disable:4786)
+#endif
 
+#include "SMDS_MeshEdge.hxx"
+#include "SMDS_IteratorOfElements.hxx"
+#include "SMDS_MeshNode.hxx"
 
-#include "SMDS_MeshEdge.ixx"
+using namespace std;
 
 //=======================================================================
 //function : SMDS_MeshEdge
 //purpose  : 
 //=======================================================================
 
-SMDS_MeshEdge::SMDS_MeshEdge(const Standard_Integer ID,
-                            const Standard_Integer idnode1, 
-                            const Standard_Integer idnode2) :
-SMDS_MeshElement(ID,2,SMDSAbs_Edge)
-{
-  SetConnections(idnode1,idnode2);
-  ComputeKey();
+SMDS_MeshEdge::SMDS_MeshEdge(const SMDS_MeshNode * node1,
+                             const SMDS_MeshNode * node2)
+{      
+       myNodes[0]=node1;
+       myNodes[1]=node2;
 }
 
 //=======================================================================
-//function : SetConnections
+//function : Print
 //purpose  : 
 //=======================================================================
-void SMDS_MeshEdge::SetConnections(const Standard_Integer idnode1, 
-                                  const Standard_Integer idnode2)
+
+void SMDS_MeshEdge::Print(ostream & OS) const
+{
+       OS << "edge <" << GetID() << "> : (" << myNodes[0] << " , " << myNodes[1] <<
+               ") " << endl;
+}
+
+int SMDS_MeshEdge::NbNodes() const
+{
+       return 2;
+}
+
+int SMDS_MeshEdge::NbEdges() const
+{
+       return 1;
+}
+
+SMDSAbs_ElementType SMDS_MeshEdge::GetType() const
+{
+       return SMDSAbs_Edge;
+}
+
+class SMDS_MeshEdge_MyNodeIterator:public SMDS_ElemIterator
 {
-  Standard_Integer idmin = (idnode1 < idnode2? idnode1 : idnode2);
+  const SMDS_MeshNode *const* myNodes;
+  int myIndex;
+ public:
+  SMDS_MeshEdge_MyNodeIterator(const SMDS_MeshNode * const* nodes):
+    myNodes(nodes),myIndex(0) {}
 
-  myNodes[0] = idmin;
-  if (idmin == idnode1) {
-    myNodes[1] = idnode2;
-  } else {
-    myNodes[1] = idnode1;
+  bool more()
+  {
+    return myIndex<2;
+  }
+
+  const SMDS_MeshElement* next()
+  {
+    myIndex++;
+    return myNodes[myIndex-1];
+  }
+};
+
+SMDS_ElemIteratorPtr SMDS_MeshEdge::
+       elementsIterator(SMDSAbs_ElementType type) const
+{
+  switch(type)
+  {
+  case SMDSAbs_Edge:
+    return SMDS_MeshElement::elementsIterator(SMDSAbs_Edge); 
+  case SMDSAbs_Node:
+    return SMDS_ElemIteratorPtr(new SMDS_MeshEdge_MyNodeIterator(myNodes));
+  default:
+    return SMDS_ElemIteratorPtr
+      (new SMDS_IteratorOfElements
+       (this,type, SMDS_ElemIteratorPtr(new SMDS_MeshEdge_MyNodeIterator(myNodes))));
   }
 }
 
+bool operator<(const SMDS_MeshEdge & e1, const SMDS_MeshEdge & e2)
+{
+       int id11=e1.myNodes[0]->GetID();
+       int id21=e2.myNodes[0]->GetID();
+       int id12=e1.myNodes[1]->GetID();
+       int id22=e2.myNodes[1]->GetID();
+       int tmp;
+
+       if(id11>=id12) 
+       {
+               tmp=id11;
+               id11=id12;
+               id12=tmp;       
+       }
+       if(id21>=id22) 
+       {
+               tmp=id21;
+               id21=id22;
+               id22=tmp;       
+       }
+
+       if(id11<id21) return true;
+       else if(id11==id21) return (id21<id22);
+       else return false;
+}
+
+/*!
+ * \brief Return node by its index
+ * \param ind - node index
+ * \retval const SMDS_MeshNode* - the node
+ * 
+ * Index is wrapped if it is out of a valid range
+ */
+const SMDS_MeshNode* SMDS_MeshEdge::GetNode(const int ind) const
+{
+  return myNodes[ WrappedIndex( ind )];
+}
+
 //=======================================================================
-//function : Print
+//function : ChangeNodes
 //purpose  : 
 //=======================================================================
 
-void SMDS_MeshEdge::Print(Standard_OStream& OS) const
+bool SMDS_MeshEdge::ChangeNodes(const SMDS_MeshNode * node1,
+                                const SMDS_MeshNode * node2)
 {
-  OS << "edge <" << myID <<"> : (" << myNodes[0] << " , " << myNodes[1] << ") " << endl;  
+  myNodes[0]=node1;
+  myNodes[1]=node2;
+  return true;
 }
 
-
-
-