Salome HOME
Update mail address
[modules/smesh.git] / src / DriverUNV / DriverUNV_W_SMDS_Mesh.cxx
index b0a573a96cd765310028e079ad440f9e2ccfe774..eeb2ff4b8285f45e61723de3f63d2787d612ce2f 100644 (file)
@@ -1,5 +1,3 @@
-//  SMESH DriverUNV : driver to read and write 'unv' files
-//
 //  Copyright (C) 2003  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
 //  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS 
 // 
 //  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.opencascade.org/SALOME/ or email : webmaster.salome@opencascade.org 
-//
-//
-//
-//  File   : DriverUNV_W_SMDS_Mesh.cxx
-//  Module : SMESH
+// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
 
-using namespace std;
-#include "DriverUNV_W_SMDS_Mesh.h"
+#include <algorithm>
 
-#include "SMDS_MeshElement.hxx"
-#include "SMDS_MeshNode.hxx"
-#include "SMDS_MeshEdgesIterator.hxx"
-#include "SMDS_MeshFacesIterator.hxx"
-#include "SMDS_MeshNodesIterator.hxx"
-#include "SMDS_MeshVolumesIterator.hxx"
-
-#include <utilities.h>
-
-#define sNODE_UNV_ID "  2411"
-#define sELT_UNV_ID  "  2412"
-#define sUNV_SEPARATOR "    -1"
-#define sNODE_UNV_DESCR "%10d         1         1        11\n"
-#define sELT_SURF_DESC  "%10d        %2d         1         1        11         %1d\n"
-#define sELT_VOLU_DESC  "%10d        %2d         1         1         9         %1d\n"
-#define sELT_BEAM_DESC1 "%10d        %2d         1         1         7         %1d\n"
-#define sELT_BEAM_DESC2 "         0         1         1\n"
-
-DriverUNV_W_SMDS_Mesh::DriverUNV_W_SMDS_Mesh() {
-;
-}
+#include "DriverUNV_W_SMDS_Mesh.h"
 
-DriverUNV_W_SMDS_Mesh::~DriverUNV_W_SMDS_Mesh() {
-;
-}
+#include "SMDS_Mesh.hxx"
+#include "SMESHDS_GroupBase.hxx"
+//#include "SMESH_Group.hxx"
+#include "SMDS_QuadraticEdge.hxx"
+#include "SMDS_QuadraticFaceOfNodes.hxx"
 
-void DriverUNV_W_SMDS_Mesh::SetMesh(Handle(SMDS_Mesh)& aMesh) {
-  myMesh = aMesh;
-}
+#include "utilities.h"
 
-void DriverUNV_W_SMDS_Mesh::SetFile(string aFile) {
-  myFile = aFile;
-}
+#include "UNV2411_Structure.hxx"
+#include "UNV2412_Structure.hxx"
+#include "UNV2417_Structure.hxx"
+#include "UNV_Utilities.hxx"
 
-void DriverUNV_W_SMDS_Mesh::SetFileId(FILE* aFileId) {
-  myFileId = aFileId;
-}
+using namespace std;
 
-void DriverUNV_W_SMDS_Mesh::SetMeshId(int aMeshId) {
-  myMeshId = aMeshId;
-}
+namespace{
+  typedef std::vector<size_t> TConnect;
 
-void DriverUNV_W_SMDS_Mesh::Add() {
-  ;
+  int GetConnect(const SMDS_ElemIteratorPtr& theNodesIter, 
+                TConnect& theConnect)
+  {
+    theConnect.clear();
+    for(; theNodesIter->more();){
+      const SMDS_MeshElement* anElem = theNodesIter->next();
+      theConnect.push_back(anElem->GetID());
+    }
+    return theConnect.size();
+  }
+  
 }
 
-void DriverUNV_W_SMDS_Mesh::Write() {
+Driver_Mesh::Status DriverUNV_W_SMDS_Mesh::Perform()
+{
+  Status aResult = DRS_OK;
+  std::ofstream out_stream(myFile.c_str());
+  try{
+    {
+      using namespace UNV2411;
+      TDataSet aDataSet2411;
+      // Storing SMDS nodes to the UNV file
+      //-----------------------------------
+      MESSAGE("Perform - myMesh->NbNodes() = "<<myMesh->NbNodes());
+      SMDS_NodeIteratorPtr aNodesIter = myMesh->nodesIterator();
+      for(; aNodesIter->more();){
+       const SMDS_MeshNode* aNode = aNodesIter->next();
+       TRecord aRec;
+       aRec.coord[0] = aNode->X();
+       aRec.coord[1] = aNode->Y();
+       aRec.coord[2] = aNode->Z();
+       const TNodeLab& aLabel = aNode->GetID();
+       aDataSet2411.insert(TDataSet::value_type(aLabel,aRec));
+      }
+      MESSAGE("Perform - aDataSet2411.size() = "<<aDataSet2411.size());
+      UNV2411::Write(out_stream,aDataSet2411);
+    }
+    {
+      using namespace UNV2412;
+      TDataSet aDataSet2412;
+      TConnect aConnect;
+
+      // Storing SMDS Edges
+      MESSAGE("Perform - myMesh->NbEdges() = "<<myMesh->NbEdges());
+      if(myMesh->NbEdges()){
+       SMDS_EdgeIteratorPtr anIter = myMesh->edgesIterator();
+       for(; anIter->more();){
+         const SMDS_MeshEdge* anElem = anIter->next();
+         TElementLab aLabel = anElem->GetID();
+         int aNbNodes = anElem->NbNodes();
+         TRecord aRec;
+         aRec.node_labels.reserve(aNbNodes);
+         SMDS_ElemIteratorPtr aNodesIter;
+          if( anElem->IsQuadratic() ) {
+            aNodesIter = static_cast<const SMDS_QuadraticEdge* >
+              ( anElem )->interlacedNodesElemIterator();
+            aRec.fe_descriptor_id = 22;
+          } else {
+            aNodesIter = anElem->nodesIterator();
+            aRec.fe_descriptor_id = 11;
+          }
+         for(; aNodesIter->more();){
+           const SMDS_MeshElement* aNode = aNodesIter->next();
+           aRec.node_labels.push_back(aNode->GetID());
+         }
+         aDataSet2412.insert(TDataSet::value_type(aLabel,aRec));
+       }
+       MESSAGE("Perform - aDataSet2412.size() = "<<aDataSet2412.size());
+      }
 
-  int nbNodes,nbCells;
-  int i;
+      MESSAGE("Perform - myMesh->NbFaces() = "<<myMesh->NbFaces());
+      if(myMesh->NbFaces()){
+       SMDS_FaceIteratorPtr anIter = myMesh->facesIterator();
+       for(; anIter->more();){
+         const SMDS_MeshFace* anElem = anIter->next();
+         TElementLab aLabel = anElem->GetID();
+         int aNbNodes = anElem->NbNodes();
+         TRecord aRec;
+         aRec.node_labels.reserve(aNbNodes);
+         SMDS_ElemIteratorPtr aNodesIter;
+          if( anElem->IsQuadratic() )
+            aNodesIter = static_cast<const SMDS_QuadraticFaceOfNodes* >
+              ( anElem )->interlacedNodesElemIterator();
+          else
+            aNodesIter = anElem->nodesIterator();
+         for(; aNodesIter->more();){
+           const SMDS_MeshElement* aNode = aNodesIter->next();
+           aRec.node_labels.push_back(aNode->GetID());
+         }
+         switch(aNbNodes){
+         case 3:
+           aRec.fe_descriptor_id = 41;
+           break;
+         case 4:
+           aRec.fe_descriptor_id = 44;
+           break;
+         case 6:
+           aRec.fe_descriptor_id = 42;
+           break;
+         case 8:
+           aRec.fe_descriptor_id = 45;
+           break;
+         default:
+           continue;
+         }
+         aDataSet2412.insert(TDataSet::value_type(aLabel,aRec));
+       }
+       MESSAGE("Perform - aDataSet2412.size() = "<<aDataSet2412.size());
+      }
 
-  char* file2Read = (char*)myFile.c_str();
-  myFileId = fopen(file2Read,"w+");
-  if (myFileId < 0)
-    {
-      fprintf(stderr,">> ERREUR : ouverture du fichier %s \n",file2Read);
-      exit(EXIT_FAILURE);
+      MESSAGE("Perform - myMesh->NbVolumes() = "<<myMesh->NbVolumes());
+      if(myMesh->NbVolumes()){
+       SMDS_VolumeIteratorPtr anIter = myMesh->volumesIterator();
+       for(; anIter->more();){
+         const SMDS_MeshVolume* anElem = anIter->next();
+         TElementLab aLabel = anElem->GetID();
+
+         int aNbNodes = anElem->NbNodes();
+         aConnect.resize(aNbNodes);
+
+         SMDS_ElemIteratorPtr aNodesIter = anElem->nodesIterator();
+         GetConnect(aNodesIter,aConnect);
+
+         int anId = -1;
+         int* aConn = NULL;
+         switch(aNbNodes){
+         case 4: {
+           static int anIds[] = {0,2,1,3};
+           aConn = anIds;
+           anId = 111;
+           break;
+         }
+         case 6: {
+           static int anIds[] = {0,2,1,3,5,4};
+           aConn = anIds;
+           anId = 112;
+           break;
+         }
+         case 8: {
+           static int anIds[] = {0,3,2,1,4,7,6,5};
+           aConn = anIds;
+           anId = 115;
+           break;
+         }
+         case 10: {
+           static int anIds[] = {0,4,2,9,5,3, 1,6,8, 7};
+           aConn = anIds;
+           anId = 118;
+           break;
+         }
+         case 13: {
+           static int anIds[] = {0,6,4,2,7,5,3,1,8,11,10,9,12};
+           aConn = anIds;
+           anId = 114;
+           break;
+         }
+         case 15: {
+           static int anIds[] = {0,4,2,9,13,11,5,3,1,14,12,10,6,8,7};
+           aConn = anIds;
+           anId = 113;
+           break;
+         }
+         case 20: {
+           static int anIds[] = {0,6, 4,2, 12,18,16,14,7, 5, 3, 1, 19,17,15,13,8, 11,10,9};
+           aConn = anIds;
+           anId = 116;
+           break;
+         }
+         default:
+           continue;
+         }
+         if(aConn){
+           TRecord aRec;
+           aRec.fe_descriptor_id = anId;
+           aRec.node_labels.resize(aNbNodes);
+           for(int aNodeId = 0; aNodeId < aNbNodes; aNodeId++){
+             aRec.node_labels[aConn[aNodeId]] = aConnect[aNodeId];
+           }
+           aDataSet2412.insert(TDataSet::value_type(aLabel,aRec));
+         }
+       }
+       MESSAGE("Perform - aDataSet2412.size() = "<<aDataSet2412.size());
+      }
+      UNV2412::Write(out_stream,aDataSet2412);
     }
-  SCRUTE(myMesh);
-  /****************************************************************************
-  *                       NOMBRES D'OBJETS                                    *
-  ****************************************************************************/
-  fprintf(stdout,"\n(****************************)\n");
-  fprintf(stdout,"(* INFORMATIONS GENERALES : *)\n");
-  fprintf(stdout,"(****************************)\n");
-
-  /* Combien de noeuds ? */
-  nbNodes = myMesh->NbNodes();
-
-  /* Combien de mailles, faces ou aretes ? */
-  Standard_Integer nb_of_nodes, nb_of_edges,nb_of_faces, nb_of_volumes;
-  nb_of_edges = myMesh->NbEdges();
-  nb_of_faces = myMesh->NbFaces();
-  nb_of_volumes = myMesh->NbVolumes();
-  nbCells = nb_of_edges + nb_of_faces + nb_of_volumes;
-  SCRUTE(nb_of_edges);
-  SCRUTE(nb_of_faces);
-  SCRUTE(nb_of_volumes);
-
-  fprintf(stdout,"%d %d\n",nbNodes,nbCells);
-  fprintf(myFileId,"%d %d\n",nbNodes,nbCells);
-
-  /****************************************************************************
-  *                       ECRITURE DES NOEUDS                                 *
-  ****************************************************************************/
-  fprintf(stdout,"\n(************************)\n");
-  fprintf(stdout,"(* NOEUDS DU MAILLAGE : *)\n");
-  fprintf(stdout,"(************************)\n");
-
-  SMDS_MeshNodesIterator itNodes(myMesh);
-
-  fprintf(myFileId,"%s\n", sUNV_SEPARATOR);
-  fprintf(myFileId,"%s\n", sNODE_UNV_ID  );
-
-  for (;itNodes.More();itNodes.Next()) {
-    const Handle(SMDS_MeshElement)& elem = itNodes.Value();
-    const Handle(SMDS_MeshNode   )& node = myMesh->GetNode(1, elem);
-
-    fprintf(myFileId, sNODE_UNV_DESCR, node->GetID());
-    fprintf(myFileId, "%25.16E%25.16E%25.16E\n", node->X(), node->Y(), node->Z());
-  }
-  fprintf(myFileId,"%s\n", sUNV_SEPARATOR);
-
-  /****************************************************************************
-  *                       ECRITURE DES ELEMENTS                                *
-  ****************************************************************************/
-  fprintf(stdout,"\n(**************************)\n");
-  fprintf(stdout,"(* ELEMENTS DU MAILLAGE : *)\n");
-  fprintf(stdout,"(**************************)");
-  /* Ecriture des connectivites, noms, numeros des mailles */
-
-  fprintf(myFileId,"%s\n", sUNV_SEPARATOR);
-  fprintf(myFileId,"%s\n", sELT_UNV_ID   );
-
-  SMDS_MeshEdgesIterator itEdges(myMesh);
-  for (;itEdges.More();itEdges.Next()) {
-    const Handle(SMDS_MeshElement)& elem = itEdges.Value();
-
-    switch (elem->NbNodes()) {
-      case 2 : {
-        fprintf(myFileId, sELT_BEAM_DESC1, elem->GetID(), 21, elem->NbNodes());
-        fprintf(myFileId, sELT_BEAM_DESC2);
-        fprintf(myFileId, "%10d%10d\n", elem->GetConnection(1), elem->GetConnection(2));
-        break;
-      } 
-      case 3 : {
-        fprintf(myFileId, sELT_BEAM_DESC1, elem->GetID(), 24, elem->NbNodes());
-        fprintf(myFileId, sELT_BEAM_DESC2);
-        fprintf(myFileId, "%10d%10d%10d\n",elem->GetConnection(1), elem->GetConnection(2), elem->GetConnection(3));
-        break;
+    {
+      using namespace UNV2417;
+      if (myGroups.size() > 0) {
+       TDataSet aDataSet2417;
+       TGroupList::const_iterator aIter = myGroups.begin();
+       for (; aIter != myGroups.end(); aIter++) {
+         SMESHDS_GroupBase* aGroupDS = *aIter;
+         TRecord aRec;
+         aRec.GroupName = aGroupDS->GetStoreName();
+
+         int i;
+         SMDS_ElemIteratorPtr aIter = aGroupDS->GetElements();
+         if (aGroupDS->GetType() == SMDSAbs_Node) {
+           aRec.NodeList.resize(aGroupDS->Extent());
+           i = 0;
+           while (aIter->more()) {
+             const SMDS_MeshElement* aElem = aIter->next();
+             aRec.NodeList[i] = aElem->GetID(); 
+             i++;
+           }
+         } else {
+           aRec.ElementList.resize(aGroupDS->Extent());
+           i = 0;
+           while (aIter->more()) {
+             const SMDS_MeshElement* aElem = aIter->next();
+             aRec.ElementList[i] = aElem->GetID(); 
+             i++;
+           }
+         }
+         aDataSet2417.insert(TDataSet::value_type(aGroupDS->GetID(), aRec));
+       }
+       UNV2417::Write(out_stream,aDataSet2417);
+       myGroups.clear();
       }
     }
+    /*    {
+      using namespace UNV2417;
+      TDataSet aDataSet2417;
+      for ( TGroupsMap::iterator it = myGroupsMap.begin(); it != myGroupsMap.end(); it++ ) {
+       SMESH_Group*       aGroup   = it->second;
+       SMESHDS_GroupBase* aGroupDS = aGroup->GetGroupDS();
+       if ( aGroupDS ) {
+         TRecord aRec;
+         aRec.GroupName = aGroup->GetName();
+         int i;
+         SMDS_ElemIteratorPtr aIter = aGroupDS->GetElements();
+         if (aGroupDS->GetType() == SMDSAbs_Node) {
+           aRec.NodeList.resize(aGroupDS->Extent());
+           i = 0;
+           while (aIter->more()) {
+             const SMDS_MeshElement* aElem = aIter->next();
+             aRec.NodeList[i] = aElem->GetID(); 
+             i++;
+           }
+         } else {
+           aRec.ElementList.resize(aGroupDS->Extent());
+           i = 0;
+           while (aIter->more()) {
+             const SMDS_MeshElement* aElem = aIter->next();
+             aRec.ElementList[i] = aElem->GetID(); 
+             i++;
+           }
+         }
+         aDataSet2417.insert(TDataSet::value_type(aGroupDS->GetID(), aRec));
+       }
+      }
+      UNV2417::Write(out_stream,aDataSet2417);
+      }*/
   }
-
-  SMDS_MeshFacesIterator itFaces(myMesh);
-  for (;itFaces.More();itFaces.Next()) {
-    const Handle(SMDS_MeshElement)& elem = itFaces.Value();
-
-    switch (elem->NbNodes()) {
-      case 3 :
-        // linear triangle
-        fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 74, elem->NbNodes());
-        break;
-      case 4 :
-        // linear quadrilateral
-        fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 71, elem->NbNodes());
-        break;
-      case 6 :
-        // parabolic triangle
-        fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 72, elem->NbNodes());
-        break;
-      case 8 :
-        // parabolic quadrilateral
-        fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 75, elem->NbNodes());
-        break;
-      default:
-        fprintf(myFileId, "element not registered\n");
-    }
-
-    for (i=0;i<elem->NbNodes();i++)
-      fprintf(myFileId,"%10d",elem->GetConnection(i+1));
-    
-    fprintf(myFileId,"\n");
+  catch(const std::exception& exc){
+    INFOS("Follow exception was cought:\n\t"<<exc.what());
   }
-
-  SMDS_MeshVolumesIterator itVolumes(myMesh);
-  for (;itVolumes.More();itVolumes.Next()) {
-    const Handle(SMDS_MeshElement)& elem = itVolumes.Value();
-
-    switch (elem->NbNodes()) {
-      case 4 : 
-        // linear tetrahedron
-        fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 111, elem->NbNodes());
-        break;
-      case 6 : 
-        // linear tetrahedron
-        fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 112, elem->NbNodes());
-        break;
-      case 8 : 
-        // linear brick
-        fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 115, elem->NbNodes());
-        break;
-    }
-
-    for (i=0;i<elem->NbNodes();i++)
-      fprintf(myFileId,"%10d",elem->GetConnection(i+1));
-    
-    fprintf(myFileId,"\n");
+  catch(...){
+    INFOS("Unknown exception was cought !!!");
   }
-  fprintf(myFileId,"%s\n", sUNV_SEPARATOR);
-
-  fclose (myFileId);
+  return aResult;
 }