Salome HOME
Update to match the new implementation of SMDS
[modules/smesh.git] / src / SMESH_I / SMESH_MeshEditor_i.cxx
index f95fc67..5c79094 100644 (file)
@@ -29,9 +29,6 @@
 using namespace std;
 #include "SMESH_MeshEditor_i.hxx"
 
-#include "SMDS_MeshEdgesIterator.hxx"
-#include "SMDS_MeshFacesIterator.hxx"
-#include "SMDS_MeshVolumesIterator.hxx"
 #include "SMDS_MeshEdge.hxx"
 #include "SMDS_MeshFace.hxx"
 #include "SMDS_MeshVolume.hxx"
@@ -41,16 +38,15 @@ using namespace std;
 #include <TColStd_MapOfInteger.hxx>
 #include <TColStd_MapIteratorOfMapOfInteger.hxx>
 
-
-
 //=============================================================================
 /*!
  *  
  */
 //=============================================================================
 
-SMESH_MeshEditor_i::SMESH_MeshEditor_i(const Handle(SMESHDS_Mesh)& theMesh) {
-  _myMeshDS = theMesh;
+SMESH_MeshEditor_i::SMESH_MeshEditor_i(SMESHDS_Mesh* theMesh)
+{
+       _myMeshDS = theMesh;
 };
 
 //=============================================================================
@@ -59,13 +55,16 @@ SMESH_MeshEditor_i::SMESH_MeshEditor_i(const Handle(SMESHDS_Mesh)& theMesh) {
  */
 //=============================================================================
 
-CORBA::Boolean SMESH_MeshEditor_i::RemoveElements(const SMESH::long_array& IDsOfElements) {
-  for (int i = 0 ; i< IDsOfElements.length(); i++) {
-    CORBA::Long index = IDsOfElements[i];
-    _myMeshDS->RemoveElement(index);
-    MESSAGE ( "Element "<< index  << " was removed" )
-  }
- return true;
+CORBA::Boolean SMESH_MeshEditor_i::RemoveElements(const SMESH::
+       long_array & IDsOfElements)
+{
+       for (int i = 0; i < IDsOfElements.length(); i++)
+       {
+               CORBA::Long index = IDsOfElements[i];
+               _myMeshDS->RemoveElement(index);
+               MESSAGE("Element " << index << " was removed")
+       }
+       return true;
 };
 
 //=============================================================================
@@ -74,66 +73,23 @@ CORBA::Boolean SMESH_MeshEditor_i::RemoveElements(const SMESH::long_array& IDsOf
  */
 //=============================================================================
 
-CORBA::Boolean SMESH_MeshEditor_i::RemoveNodes(const SMESH::long_array& IDsOfNodes) {
-
-  // Here we try to collect all 1D, 2D and 3D elements which contain at least one 
-  // of <IDsOfNodes> in order to remove such elements. 
-  // This seems correct since e.g a triangle without 1 vertex looks nonsense.
-  TColStd_MapOfInteger elemsToRemove;
-
-  for (int i = 0 ; i< IDsOfNodes.length(); i++) {
-
-    CORBA::Long ID = IDsOfNodes[i];
-
-    SMDS_MeshEdgesIterator edgeIt(_myMeshDS);
-    for (; edgeIt.More(); edgeIt.Next()) {
-      Handle(SMDS_MeshEdge) anEdge = Handle(SMDS_MeshEdge)::DownCast(edgeIt.Value());
-      for (Standard_Integer i = 0; i < anEdge->NbNodes(); i++) {
-       if (anEdge->GetConnection(i) == ID) {
-         Standard_Integer elemID = anEdge->GetID();
-         if (!elemsToRemove.Contains(elemID)) elemsToRemove.Add(elemID);
-       }
-      }
-    }
-
-    SMDS_MeshFacesIterator faceIt(_myMeshDS);
-    for (; faceIt.More(); faceIt.Next()) {
-      Handle(SMDS_MeshFace) aFace = Handle(SMDS_MeshFace)::DownCast(faceIt.Value());
-      for (Standard_Integer i = 0; i < aFace->NbNodes(); i++) {
-       if (aFace->GetConnection(i) == ID) {
-         Standard_Integer elemID = aFace->GetID();
-         if (!elemsToRemove.Contains(elemID)) elemsToRemove.Add(elemID);
+CORBA::Boolean SMESH_MeshEditor_i::RemoveNodes(const SMESH::
+       long_array & IDsOfNodes)
+{
+       // It's nodes' turn to die
+       for (int i = 0; i < IDsOfNodes.length(); i++)
+       {
+               const SMDS_MeshNode * node=_myMeshDS->FindNode(IDsOfNodes[i]);
+               if(node==NULL)
+               {
+                       MESSAGE("SMESH_MeshEditor_i::RemoveNodes: Node "<<IDsOfNodes[i]
+                               <<" not found");
+                       continue;
+               }
+               _myMeshDS->RemoveNode(IDsOfNodes[i]);
+               MESSAGE("Node " << index << " was removed")
        }
-      }
-    }
-  
-    SMDS_MeshVolumesIterator volIt(_myMeshDS);
-    for (; volIt.More(); volIt.Next()) {
-      Handle(SMDS_MeshVolume) aVol = Handle(SMDS_MeshVolume)::DownCast(volIt.Value());
-      for (Standard_Integer i = 0; i < aVol->NbNodes(); i++) {
-       if (aVol->GetConnection(i) == ID) {
-         Standard_Integer elemID = aVol->GetID();
-         if (!elemsToRemove.Contains(elemID)) elemsToRemove.Add(elemID);
-       }
-      }
-    }
-  }
-
-  // Now remove them!
-  TColStd_MapIteratorOfMapOfInteger it(elemsToRemove);
-  for (; it.More(); it.Next()) {
-    Standard_Integer elemID = it.Key();
-    _myMeshDS->RemoveElement(elemID);
-    MESSAGE("RemoveNodes(): element removed: " << elemID)
-  }
-  
-  // It's nodes' turn to die
-  for (int i = 0 ; i< IDsOfNodes.length(); i++) {
-    CORBA::Long index = IDsOfNodes[i];
-    _myMeshDS->RemoveNode(index);
-    MESSAGE ( "Node "<< index  << " was removed" )
-  }
- return true;
+       return true;
 };
 
 //=============================================================================
@@ -142,14 +98,16 @@ CORBA::Boolean SMESH_MeshEditor_i::RemoveNodes(const SMESH::long_array& IDsOfNod
  */
 //=============================================================================
 
-CORBA::Boolean SMESH_MeshEditor_i::AddEdge(const SMESH::long_array& IDsOfNodes) {
-  int NbNodes = IDsOfNodes.length();
-  if ( NbNodes == 2 ) {
-    CORBA::Long index1 = IDsOfNodes[0];
-    CORBA::Long index2 = IDsOfNodes[1];
-    int idTri = _myMeshDS->AddEdge(index1,index2);
-  }
-  return true;
+CORBA::Boolean SMESH_MeshEditor_i::AddEdge(const SMESH::long_array & IDsOfNodes)
+{
+       int NbNodes = IDsOfNodes.length();
+       if (NbNodes == 2)
+       {
+               CORBA::Long index1 = IDsOfNodes[0];
+               CORBA::Long index2 = IDsOfNodes[1];
+               _myMeshDS->AddEdge(index1, index2);
+       }
+       return true;
 }
 
 //=============================================================================
@@ -159,12 +117,11 @@ CORBA::Boolean SMESH_MeshEditor_i::AddEdge(const SMESH::long_array& IDsOfNodes)
 //=============================================================================
 
 CORBA::Boolean SMESH_MeshEditor_i::AddNode(CORBA::Double x,
-                                          CORBA::Double y,
-                                          CORBA::Double z) {
-  MESSAGE( " AddNode " << x << " , " << y << " , " << z )
-  int idNode = _myMeshDS->AddNode(x,y,z);
-  MESSAGE( " idNode " << idNode )
-  return true;
+       CORBA::Double y, CORBA::Double z)
+{
+       MESSAGE(" AddNode " << x << " , " << y << " , " << z)
+               int idNode = _myMeshDS->AddNode(x, y, z)->GetID();
+       MESSAGE(" idNode " << idNode) return true;
 }
 
 //=============================================================================
@@ -173,21 +130,25 @@ CORBA::Boolean SMESH_MeshEditor_i::AddNode(CORBA::Double x,
  */
 //=============================================================================
 
-CORBA::Boolean SMESH_MeshEditor_i::AddFace(const SMESH::long_array& IDsOfNodes) {
-  int NbNodes = IDsOfNodes.length();
-  if ( NbNodes == 3 ) {
-    CORBA::Long index1 = IDsOfNodes[0];
-    CORBA::Long index2 = IDsOfNodes[1];
-    CORBA::Long index3 = IDsOfNodes[2];
-    int idTri = _myMeshDS->AddFace(index1,index2,index3);
-  } else  if ( NbNodes == 4 ) {
-    CORBA::Long index1 = IDsOfNodes[0];
-    CORBA::Long index2 = IDsOfNodes[1];
-    CORBA::Long index3 = IDsOfNodes[2];
-    CORBA::Long index4 = IDsOfNodes[3];
-    int idTri = _myMeshDS->AddFace(index1,index2,index3,index4);
-  }
- return true;
+CORBA::Boolean SMESH_MeshEditor_i::AddFace(const SMESH::long_array & IDsOfNodes)
+{
+       int NbNodes = IDsOfNodes.length();
+       if (NbNodes == 3)
+       {
+               CORBA::Long index1 = IDsOfNodes[0];
+               CORBA::Long index2 = IDsOfNodes[1];
+               CORBA::Long index3 = IDsOfNodes[2];
+               _myMeshDS->AddFace(index1, index2, index3);
+       }
+       else if (NbNodes == 4)
+       {
+               CORBA::Long index1 = IDsOfNodes[0];
+               CORBA::Long index2 = IDsOfNodes[1];
+               CORBA::Long index3 = IDsOfNodes[2];
+               CORBA::Long index4 = IDsOfNodes[3];
+               _myMeshDS->AddFace(index1, index2, index3, index4);
+       }
+       return true;
 };
 
 //=============================================================================
@@ -196,39 +157,49 @@ CORBA::Boolean SMESH_MeshEditor_i::AddFace(const SMESH::long_array& IDsOfNodes)
  */
 //=============================================================================
 
-CORBA::Boolean SMESH_MeshEditor_i::AddVolume(const SMESH::long_array& IDsOfNodes) {
-  int NbNodes = IDsOfNodes.length();
-  if ( NbNodes == 4 ) {
-    CORBA::Long index1 = IDsOfNodes[0];
-    CORBA::Long index2 = IDsOfNodes[1];
-    CORBA::Long index3 = IDsOfNodes[2];
-    CORBA::Long index4 = IDsOfNodes[3];
-    int idTetra = _myMeshDS->AddVolume(index1,index2,index3,index4);
-  } else if ( NbNodes == 5 ) {
-    CORBA::Long index1 = IDsOfNodes[0];
-    CORBA::Long index2 = IDsOfNodes[1];
-    CORBA::Long index3 = IDsOfNodes[2];
-    CORBA::Long index4 = IDsOfNodes[3];
-    CORBA::Long index5 = IDsOfNodes[4];
-    int idPyramid = _myMeshDS->AddVolume(index1,index2,index3,index4,index5);
-  } else if ( NbNodes == 6 ) {
-    CORBA::Long index1 = IDsOfNodes[0];
-    CORBA::Long index2 = IDsOfNodes[1];
-    CORBA::Long index3 = IDsOfNodes[2];
-    CORBA::Long index4 = IDsOfNodes[3];
-    CORBA::Long index5 = IDsOfNodes[4];
-    CORBA::Long index6 = IDsOfNodes[5];
-    int idPrism = _myMeshDS->AddVolume(index1,index2,index3,index4,index5,index6);
-  } else if ( NbNodes == 8 ) {
-    CORBA::Long index1 = IDsOfNodes[0];
-    CORBA::Long index2 = IDsOfNodes[1];
-    CORBA::Long index3 = IDsOfNodes[2];
-    CORBA::Long index4 = IDsOfNodes[3];
-    CORBA::Long index5 = IDsOfNodes[4];
-    CORBA::Long index6 = IDsOfNodes[5];
-    CORBA::Long index7 = IDsOfNodes[6];
-    CORBA::Long index8 = IDsOfNodes[7];
-    int idHexa = _myMeshDS->AddVolume(index1,index2,index3,index4,index5,index6,index7,index8);
-  } 
-  return true;
+CORBA::Boolean SMESH_MeshEditor_i::AddVolume(const SMESH::
+       long_array & IDsOfNodes)
+{
+       int NbNodes = IDsOfNodes.length();
+       if (NbNodes == 4)
+       {
+               CORBA::Long index1 = IDsOfNodes[0];
+               CORBA::Long index2 = IDsOfNodes[1];
+               CORBA::Long index3 = IDsOfNodes[2];
+               CORBA::Long index4 = IDsOfNodes[3];
+               _myMeshDS->AddVolume(index1, index2, index3, index4);
+       }
+       else if (NbNodes == 5)
+       {
+               CORBA::Long index1 = IDsOfNodes[0];
+               CORBA::Long index2 = IDsOfNodes[1];
+               CORBA::Long index3 = IDsOfNodes[2];
+               CORBA::Long index4 = IDsOfNodes[3];
+               CORBA::Long index5 = IDsOfNodes[4];
+               _myMeshDS->AddVolume(index1, index2, index3, index4, index5);
+       }
+       else if (NbNodes == 6)
+       {
+               CORBA::Long index1 = IDsOfNodes[0];
+               CORBA::Long index2 = IDsOfNodes[1];
+               CORBA::Long index3 = IDsOfNodes[2];
+               CORBA::Long index4 = IDsOfNodes[3];
+               CORBA::Long index5 = IDsOfNodes[4];
+               CORBA::Long index6 = IDsOfNodes[5];
+               _myMeshDS->AddVolume(index1, index2, index3, index4, index5, index6);
+       }
+       else if (NbNodes == 8)
+       {
+               CORBA::Long index1 = IDsOfNodes[0];
+               CORBA::Long index2 = IDsOfNodes[1];
+               CORBA::Long index3 = IDsOfNodes[2];
+               CORBA::Long index4 = IDsOfNodes[3];
+               CORBA::Long index5 = IDsOfNodes[4];
+               CORBA::Long index6 = IDsOfNodes[5];
+               CORBA::Long index7 = IDsOfNodes[6];
+               CORBA::Long index8 = IDsOfNodes[7];
+               _myMeshDS->AddVolume(index1, index2, index3, index4, index5, index6,
+                       index7, index8);
+       }
+       return true;
 };