src/DriverCGNS/DriverCGNS_Write.cxx

   1 // Copyright (C) 2007-2020  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 // File      : DriverCGNS_Write.cxx
  23 // Created   : Fri Aug  5 17:43:54 2011
  24 // Author    : Edward AGAPOV (eap)
  25
  26 #include "DriverCGNS_Write.hxx"
  27
  28 #include "SMDS_IteratorOnIterators.hxx"
  29 #include "SMDS_MeshNode.hxx"
  30 #include "SMDS_VolumeTool.hxx"
  31 #include "SMESHDS_GroupBase.hxx"
  32 #include "SMESHDS_Mesh.hxx"
  33 #include "SMESH_Comment.hxx"
  34
  35 #include <limits>
  36 #include <cgnslib.h>
  37
  38 #if CGNS_VERSION < 3100
  39 # define cgsize_t int
  40 #endif
  41
  42 using namespace std;
  43
  44 namespace
  45 {
  46   //================================================================================
  47   /*!
  48    * \brief Return interlace and type of CGNS element for the given SMDSAbs_EntityType
  49    */
  50   //================================================================================
  51
  52   const int* getInterlaceAndType( const SMDSAbs_EntityType      smType,
  53                                   CGNS_ENUMT( ElementType_t ) & cgType )
  54   {
  55     static vector< const int* >                 interlaces;
  56     static vector< CGNS_ENUMT( ElementType_t )> cgTypes;
  57     if ( interlaces.empty() )
  58     {
  59       interlaces.resize( SMDSEntity_Last, 0 );
  60       cgTypes.resize( SMDSEntity_Last, CGNS_ENUMV( ElementTypeNull ));
  61       {
  62         static int ids[] = {0};
  63         interlaces[SMDSEntity_0D] = ids;
  64         cgTypes   [SMDSEntity_0D] = CGNS_ENUMV( NODE );
  65       }
  66       {
  67         static int ids[] = { 0, 1 };
  68         interlaces[SMDSEntity_Edge] = ids;
  69         cgTypes   [SMDSEntity_Edge] = CGNS_ENUMV( BAR_2 );
  70       }
  71       {
  72         static int ids[] = { 0, 1, 2 };
  73         interlaces[SMDSEntity_Quad_Edge] = ids;
  74         cgTypes   [SMDSEntity_Quad_Edge] = CGNS_ENUMV( BAR_3 );
  75       }
  76       {
  77         static int ids[] = { 0, 2, 1 };
  78         interlaces[SMDSEntity_Triangle] = ids;
  79         cgTypes   [SMDSEntity_Triangle] = CGNS_ENUMV( TRI_3 );
  80       }
  81       {
  82         static int ids[] = { 0, 2, 1, 5, 4, 3 };
  83         interlaces[SMDSEntity_Quad_Triangle] = ids;
  84         cgTypes   [SMDSEntity_Quad_Triangle] = CGNS_ENUMV( TRI_6 );
  85         interlaces[SMDSEntity_BiQuad_Triangle] = ids;
  86         cgTypes   [SMDSEntity_BiQuad_Triangle] = CGNS_ENUMV( TRI_6 );
  87       }
  88       {
  89         static int ids[] = { 0, 3, 2, 1 };
  90         interlaces[SMDSEntity_Quadrangle] = ids;
  91         cgTypes   [SMDSEntity_Quadrangle] = CGNS_ENUMV( QUAD_4 );
  92       }
  93       {
  94         static int ids[] = { 0,3,2,1,7,6,5,4 };
  95         interlaces[SMDSEntity_Quad_Quadrangle] = ids;
  96         cgTypes   [SMDSEntity_Quad_Quadrangle] = CGNS_ENUMV( QUAD_8 );
  97       }
  98       {
  99         static int ids[] = { 0,3,2,1,7,6,5,4,8 };
 100         interlaces[SMDSEntity_BiQuad_Quadrangle] = ids;
 101         cgTypes   [SMDSEntity_BiQuad_Quadrangle] = CGNS_ENUMV( QUAD_9 );
 102       }
 103       {
 104         static int ids[] = { 0, 2, 1, 3 };
 105         interlaces[SMDSEntity_Tetra] = ids;
 106         cgTypes   [SMDSEntity_Tetra] = CGNS_ENUMV( TETRA_4 );
 107       }
 108       {
 109         static int ids[] = { 0,2,1,3,6,5,4,7,9,8 };
 110         interlaces[SMDSEntity_Quad_Tetra] = ids;
 111         cgTypes   [SMDSEntity_Quad_Tetra] = CGNS_ENUMV( TETRA_10 );
 112       }
 113       {
 114         static int ids[] = { 0,3,2,1,4 };
 115         interlaces[SMDSEntity_Pyramid] = ids;
 116         cgTypes   [SMDSEntity_Pyramid] = CGNS_ENUMV( PYRA_5 );
 117       }
 118       {
 119         static int ids[] = { 0,3,2,1,4,8,7,6,5,9,12,11,10 };
 120         interlaces[SMDSEntity_Quad_Pyramid] = ids;
 121         cgTypes   [SMDSEntity_Quad_Pyramid] = CGNS_ENUMV( PYRA_13 );
 122       }
 123       {
 124         static int ids[] = { 0,2,1,3,5,4 };
 125         interlaces[SMDSEntity_Penta] = ids;
 126         cgTypes   [SMDSEntity_Penta] = CGNS_ENUMV( PENTA_6 );
 127       }
 128       {
 129         static int ids[] = { 0,2,1,3,5,4,8,7,6,9,11,10,14,13,12 };
 130         interlaces[SMDSEntity_Quad_Penta] = ids;
 131         cgTypes   [SMDSEntity_Quad_Penta] = CGNS_ENUMV( PENTA_15 );
 132       }
 133       {
 134         static int ids[] = { 0,2,1,3,5,4,8,7,6,9,11,10,14,13,12,15,16,17 }; // TODO: check CGNS ORDER
 135         interlaces[SMDSEntity_BiQuad_Penta] = ids;
 136         cgTypes   [SMDSEntity_BiQuad_Penta] = CGNS_ENUMV( PENTA_18 );
 137       }
 138       {
 139         static int ids[] = { 0,3,2,1,4,7,6,5 };
 140         interlaces[SMDSEntity_Hexa] = ids;
 141         cgTypes   [SMDSEntity_Hexa] = CGNS_ENUMV( HEXA_8 );
 142       }
 143       {
 144         static int ids[] = { 0,3,2,1,4,7,6,5,11,10,9,8,12,15,14,13,19,18,17,16 };
 145         interlaces[SMDSEntity_Quad_Hexa] = ids;
 146         cgTypes   [SMDSEntity_Quad_Hexa] = CGNS_ENUMV( HEXA_20 );
 147       }
 148       {
 149         static int ids[] = { 0,3,2,1,4,7,6,5,11,10,9,8,12,15,14,13,19,18,17,16,
 150                              20, 24,23,22,21, 25, 26};
 151         interlaces[SMDSEntity_TriQuad_Hexa] = ids;
 152         cgTypes   [SMDSEntity_TriQuad_Hexa] = CGNS_ENUMV( HEXA_27 );
 153       }
 154       {
 155         cgTypes[SMDSEntity_Polygon]         = CGNS_ENUMV( NGON_n );
 156         cgTypes[SMDSEntity_Quad_Polygon]    = CGNS_ENUMV( NGON_n );
 157         cgTypes[SMDSEntity_Polyhedra]       = CGNS_ENUMV( NFACE_n );
 158         cgTypes[SMDSEntity_Hexagonal_Prism] = CGNS_ENUMV( NFACE_n );
 159       }
 160     }
 161     cgType  = cgTypes[ smType ];
 162     return interlaces[ smType ];
 163   }
 164
 165   //================================================================================
 166   /*!
 167    * \brief Cut off type of boundary condition from the group name
 168    */
 169   //================================================================================
 170
 171   CGNS_ENUMT( BCType_t ) getBCType( string& groupName )
 172   {
 173     CGNS_ENUMT( BCType_t ) bcType = CGNS_ENUMV( BCGeneral ); // default type
 174
 175     // boundary condition type starts from "BC"
 176     size_t bcBeg = groupName.find("BC");
 177     if ( bcBeg != string::npos )
 178     {
 179       for ( int t = 0; t < NofValidBCTypes; ++t )
 180       {
 181         CGNS_ENUMT( BCType_t ) type = CGNS_ENUMT( BCType_t )( t );
 182         string typeName = cg_BCTypeName( type );
 183         if ( typeName == &groupName[0] + bcBeg )
 184         {
 185           bcType = type;
 186           while ( bcBeg > 0 && isspace( bcBeg-1 ))
 187             --bcBeg;
 188           if ( bcBeg == 0 )
 189             groupName = "Group";
 190           else
 191             groupName = groupName.substr( 0, bcBeg-1 );
 192         }
 193       }
 194     }
 195     return bcType;
 196   }
 197
 198   //================================================================================
 199   /*!
 200    * \brief Sortable face of a polyhedron
 201    */
 202   struct TPolyhedFace
 203   {
 204     smIdType _id; // id of NGON_n
 205     vector< smIdType > _nodes; // lowest node IDs used for sorting
 206
 207     TPolyhedFace( const SMDS_MeshNode** nodes, const smIdType nbNodes, smIdType ID):_id(ID)
 208     {
 209       set< smIdType > ids;
 210       for ( smIdType i = 0; i < nbNodes; ++i )
 211         ids.insert( nodes[i]->GetID() );
 212
 213       _nodes.resize( 3 ); // std::min( nbNodes, 4 )); hope 3 nodes is enough
 214       set< smIdType >::iterator idIt = ids.begin();
 215       for ( size_t j = 0; j < _nodes.size(); ++j, ++idIt )
 216         _nodes[j] = *idIt;
 217     }
 218     bool operator< (const TPolyhedFace& o ) const
 219     {
 220       return _nodes < o._nodes;
 221     }
 222   };
 223   //================================================================================
 224   /*!
 225    * \brief Return CGNS id of an element
 226    */
 227   //================================================================================
 228
 229   cgsize_t cgnsID( const SMDS_MeshElement*                         elem,
 230                    const map< const SMDS_MeshElement*, cgsize_t >& elem2cgID )
 231   {
 232     map< const SMDS_MeshElement*, cgsize_t >::const_iterator e2id = elem2cgID.find( elem );
 233     return ( e2id == elem2cgID.end() ? elem->GetID() : e2id->second );
 234   }
 235
 236 } // namespace
 237
 238 //================================================================================
 239 /*!
 240  * \brief Write the mesh into the CGNS file
 241  */
 242 //================================================================================
 243
 244 Driver_Mesh::Status DriverCGNS_Write::Perform()
 245 {
 246   myErrorMessages.clear();
 247
 248   if ( !myMesh || myMesh->GetMeshInfo().NbElements() < 1 )
 249     return addMessage( !myMesh ? "NULL mesh" : "Empty mesh (no elements)", /*fatal = */true );
 250
 251   if ( Driver_Mesh::IsMeshTooLarge< cgsize_t >( myMesh, /*checkIDs =*/ false))
 252     return DRS_TOO_LARGE_MESH;
 253
 254   // open the file
 255   if ( cg_open(myFile.c_str(), CG_MODE_MODIFY, &_fn) != CG_OK &&
 256        cg_open(myFile.c_str(), CG_MODE_WRITE,  &_fn) != CG_OK )
 257     return addMessage( cg_get_error(), /*fatal = */true );
 258
 259   // create a Base
 260   // --------------
 261
 262   const int spaceDim = 3;
 263   int        meshDim = 1;
 264   if ( myMesh->NbFaces()   > 0 ) meshDim = 2;
 265   if ( myMesh->NbVolumes() > 0 ) meshDim = 3;
 266
 267   if ( myMeshName.empty() )
 268   {
 269     int nbases = 0;
 270     if ( cg_nbases( _fn, &nbases) == CG_OK )
 271       myMeshName = ( SMESH_Comment("Base_") << nbases+1 );
 272     else
 273       myMeshName = "Base_0";
 274   }
 275   int iBase;
 276   if ( cg_base_write( _fn, myMeshName.c_str(), meshDim, spaceDim, &iBase ))
 277     return addMessage( cg_get_error(), /*fatal = */true );
 278
 279   // create a Zone
 280   // --------------
 281
 282   smIdType nbCells = myMesh->NbEdges();
 283   if ( meshDim == 3 )
 284     nbCells = myMesh->NbVolumes();
 285   else if ( meshDim == 2 )
 286     nbCells = myMesh->NbFaces();
 287
 288   cgsize_t size[9] = { FromIdType<cgsize_t>( myMesh->NbNodes() ),
 289                        FromIdType<cgsize_t>( nbCells ),
 290                        /*NBoundVertex=*/0, 0,0,0,0,0,0 };
 291   int iZone;
 292   if ( cg_zone_write( _fn, iBase, "SMESH_Mesh", size,
 293                       CGNS_ENUMV( Unstructured ), &iZone) != CG_OK )
 294     return addMessage( cg_get_error(), /*fatal = */true );
 295
 296   // Map to store only elements whose an SMDS ID differs from a CGNS one
 297   typedef map< const SMDS_MeshElement*, cgsize_t > TElem2cgIDMap;
 298   vector< TElem2cgIDMap > elem2cgIDByEntity( SMDSEntity_Last );
 299   TElem2cgIDMap::iterator elem2cgIDIter;
 300
 301   TElem2cgIDMap & n2cgID = elem2cgIDByEntity[ SMDSEntity_Node ];
 302
 303   // Write nodes
 304   // ------------
 305   {
 306     vector< double > coords( myMesh->NbNodes() );
 307     int iC;
 308     // X
 309     SMDS_NodeIteratorPtr nIt = myMesh->nodesIterator();
 310     for ( int i = 0; nIt->more(); ++i ) coords[i] = nIt->next()->X();
 311     if ( cg_coord_write( _fn, iBase, iZone, CGNS_ENUMV(RealDouble),
 312                           "CoordinateX", &coords[0], &iC) != CG_OK )
 313       return addMessage( cg_get_error(), /*fatal = */true );
 314     // Y
 315     nIt = myMesh->nodesIterator();
 316     for ( int i = 0; nIt->more(); ++i ) coords[i] = nIt->next()->Y();
 317     if ( cg_coord_write( _fn, iBase, iZone, CGNS_ENUMV(RealDouble),
 318                           "CoordinateY", &coords[0], &iC) != CG_OK )
 319       return addMessage( cg_get_error(), /*fatal = */true );
 320     // Z
 321     nIt = myMesh->nodesIterator();
 322     for ( int i = 0; nIt->more(); ++i ) coords[i] = nIt->next()->Z();
 323     if ( cg_coord_write( _fn, iBase, iZone, CGNS_ENUMV(RealDouble),
 324                           "CoordinateZ", &coords[0], &iC) != CG_OK )
 325       return addMessage( cg_get_error(), /*fatal = */true );
 326
 327     // store CGNS ids of nodes
 328     nIt = myMesh->nodesIterator();
 329     for ( int i = 0; nIt->more(); ++i )
 330     {
 331       const SMDS_MeshElement* n = nIt->next();
 332       if ( n->GetID() != i+1 )
 333         n2cgID.insert( n2cgID.end(), make_pair( n, i+1 ));
 334     }
 335   }
 336   // Write elements
 337   // ---------------
 338
 339   cgsize_t cgID = 1, startID;
 340
 341   // write into a section all successive elements of one geom type
 342   int iSec;
 343   vector< cgsize_t > elemData;
 344   SMDS_ElemIteratorPtr elemIt = myMesh->elementsIterator();
 345   vector< SMDS_ElemIteratorPtr > elemItVec;
 346   if ( _elementsByType )
 347   {
 348     // create an iterator returning all elements by type
 349     for ( int type = SMDSEntity_Node + 1; type < SMDSEntity_Last; ++type )
 350     {
 351       if ( type == SMDSEntity_Ball )
 352         continue; // not supported
 353       elemIt = myMesh->elementEntityIterator( SMDSAbs_EntityType( type ));
 354       if ( elemIt->more() )
 355         elemItVec.push_back( elemIt );
 356     }
 357     typedef SMDS_IteratorOnIterators< const SMDS_MeshElement*,
 358                                       vector< SMDS_ElemIteratorPtr > > TVecIterator;
 359     elemIt.reset( new TVecIterator( elemItVec ));
 360   }
 361
 362   const SMDS_MeshElement* elem = elemIt->next();
 363   while ( elem )
 364   {
 365     const SMDSAbs_EntityType elemType = elem->GetEntityType();
 366     CGNS_ENUMT( ElementType_t ) cgType;
 367     const int* interlace = getInterlaceAndType( elemType, cgType );
 368
 369     TElem2cgIDMap & elem2cgID = elem2cgIDByEntity[ elemType ];
 370
 371     elemData.clear();
 372     startID = cgID;
 373
 374     if ( interlace ) // STANDARD elements
 375     {
 376       int cgnsNbNodes; // get nb nodes by element type, that can be less that elem->NbNodes()
 377       cg_npe( cgType, &cgnsNbNodes );
 378       do
 379       {
 380         for ( int i = 0; i < cgnsNbNodes; ++i )
 381           elemData.push_back( cgnsID( elem->GetNode( interlace[i] ), n2cgID ));
 382         if ( elem->GetID() != cgID )
 383           elem2cgID.insert( elem2cgID.end(), make_pair( elem, cgID ));
 384         ++cgID;
 385         elem = elemIt->more() ? elemIt->next() : 0;
 386       }
 387       while ( elem && elem->GetEntityType() == elemType );
 388     }
 389     else if ( elemType == SMDSEntity_Polygon ) // POLYGONS
 390       do
 391       {
 392         elemData.push_back( elem->NbNodes() );
 393         for ( int i = 0, nb = elem->NbNodes(); i < nb; ++i )
 394           elemData.push_back( cgnsID( elem->GetNode(i), n2cgID ));
 395         if ( elem->GetID() != cgID )
 396           elem2cgID.insert( elem2cgID.end(), make_pair( elem, cgID ));
 397         ++cgID;
 398         elem = elemIt->more() ? elemIt->next() : 0;
 399       }
 400       while ( elem && elem->GetEntityType() == elemType );
 401
 402     else if ( elemType == SMDSEntity_Quad_Polygon ) // QUADRATIC POLYGONS
 403       do // write as linear NGON_n
 404       {
 405         elemData.push_back( elem->NbNodes() );
 406         interlace = & SMDS_MeshCell::interlacedSmdsOrder( SMDSEntity_Quad_Polygon,
 407                                                           elem->NbNodes() )[0];
 408         for ( int i = 0, nb = elem->NbNodes(); i < nb; ++i )
 409           elemData.push_back( cgnsID( elem->GetNode( interlace[i] ), n2cgID ));
 410         if ( elem->GetID() != cgID )
 411           elem2cgID.insert( elem2cgID.end(), make_pair( elem, cgID ));
 412         ++cgID;
 413         elem = elemIt->more() ? elemIt->next() : 0;
 414       }
 415       while ( elem && elem->GetEntityType() == elemType );
 416
 417     else if ( elemType == SMDSEntity_Polyhedra ||
 418               elemType == SMDSEntity_Hexagonal_Prism) // POLYHEDRA
 419     {
 420       // to save polyhedrons after all
 421       const SMDS_MeshInfo& meshInfo = myMesh->GetMeshInfo();
 422       if ( meshInfo.NbPolyhedrons() == meshInfo.NbElements() - cgID + 1 )
 423         break; // only polyhedrons remain
 424       while ( elem && elem->GetEntityType() == elemType )
 425         elem = elemIt->more() ? elemIt->next() : 0;
 426       continue;
 427     }
 428     else // skip NOT SUPPORTED elements
 429     {
 430       while ( elemIt->more() )
 431       {
 432         elem = elemIt->next();
 433         if ( elem->GetEntityType() != elemType )
 434           break;
 435       }
 436     }
 437
 438     SMESH_Comment sectionName( cg_ElementTypeName( cgType ));
 439     sectionName << " " << startID << " - " << cgID-1;
 440
 441     if ( cg_section_write(_fn, iBase, iZone, sectionName.c_str(), cgType, startID,
 442                           cgID-1, /*nbndry=*/0, &elemData[0], &iSec) != CG_OK )
 443       return addMessage( cg_get_error(), /*fatal = */true );
 444   }
 445
 446   // Write polyhedral volumes
 447   // -------------------------
 448
 449   if ( myMesh->GetMeshInfo().NbElements() > cgID-1 ) // polyhedra or hexagonal prisms remain
 450   {
 451     // the polyhedron (NFACE_n) is described as a set of signed face IDs,
 452     // so first we are to write all polygones (NGON_n) bounding polyhedrons
 453
 454     vector< cgsize_t > faceData;
 455     set< TPolyhedFace > faces;
 456     set< TPolyhedFace >::iterator faceInSet;
 457     vector<const SMDS_MeshNode *> faceNodesVec;
 458     int nbPolygones = 0, faceID;
 459
 460     SMDS_VolumeTool vol;
 461
 462     elemData.clear();
 463
 464     int nbPolyhTreated = 0;
 465
 466     TElem2cgIDMap * elem2cgID = 0;
 467     TElem2cgIDMap & n2cgID    = elem2cgIDByEntity[ SMDSEntity_Node ];
 468
 469     SMDS_ElemIteratorPtr elemIt = myMesh->elementsIterator();
 470     while ( elemIt->more() )
 471     {
 472       elem = elemIt->next();
 473       SMDSAbs_EntityType type = elem->GetEntityType();
 474       if ( type == SMDSEntity_Polyhedra ||
 475            type == SMDSEntity_Hexagonal_Prism )
 476       {
 477         ++nbPolyhTreated;
 478         vol.Set( elem );
 479         vol.SetExternalNormal();
 480         const int nbFaces = vol.NbFaces();
 481         elemData.push_back( nbFaces );
 482         for ( int iF = 0; iF < nbFaces; ++iF )
 483         {
 484           const int nbNodes = vol.NbFaceNodes( iF );
 485           const SMDS_MeshNode** faceNodes = vol.GetFaceNodes( iF );
 486           faceNodesVec.assign( faceNodes, faceNodes + nbNodes );
 487           if (( elem = myMesh->FindElement( faceNodesVec, SMDSAbs_Face, /*noMedium=*/false)))
 488           {
 489             // a face of the polyhedron is present in the mesh
 490             faceID = cgnsID( elem, elem2cgIDByEntity[ elem->GetEntityType() ]);
 491           }
 492           else if ( vol.IsFreeFace( iF ))
 493           {
 494             // the face is not shared by volumes
 495             faceID = cgID++;
 496             ++nbPolygones;
 497             faceData.push_back( nbNodes );
 498             for ( int i = 0; i < nbNodes; ++i )
 499               faceData.push_back( cgnsID( faceNodes[i], n2cgID ));
 500           }
 501           else
 502           {
 503             TPolyhedFace face( faceNodes, nbNodes, cgID );
 504             faceInSet = faces.insert( faces.end(), face );
 505             if ( faceInSet->_id == cgID ) // the face encounters for the 1st time
 506             {
 507               faceID = cgID++;
 508               ++nbPolygones;
 509               faceData.push_back( nbNodes );
 510               for ( int i = 0; i < nbNodes; ++i )
 511                 faceData.push_back( cgnsID( faceNodes[i], n2cgID ));
 512             }
 513             else
 514             {
 515               // the face encounters for the 2nd time; we hope it won't encounter once more,
 516               // for that we can erase it from the set of faces
 517               faceID = -faceInSet->_id;
 518               faces.erase( faceInSet );
 519             }
 520           }
 521           elemData.push_back( faceID );
 522         }
 523       }
 524     }
 525
 526     if ( nbPolygones > 0 )
 527     {
 528       if ( cg_section_write(_fn, iBase, iZone, "Faces of Polyhedrons",
 529                              CGNS_ENUMV( NGON_n ), cgID - nbPolygones, cgID-1,
 530                              /*nbndry=*/0, &faceData[0], &iSec) != CG_OK )
 531         return addMessage( cg_get_error(), /*fatal = */true );
 532     }
 533
 534     if ( cg_section_write(_fn, iBase, iZone, "Polyhedrons",
 535                              CGNS_ENUMV( NFACE_n ), cgID, cgID+nbPolyhTreated-1,
 536                            /*nbndry=*/0, &elemData[0], &iSec) != CG_OK )
 537       return addMessage( cg_get_error(), /*fatal = */true );
 538
 539     if ( !myMesh->GetGroups().empty() )
 540     {
 541       // store CGNS ids of polyhedrons
 542       elem2cgID = &elem2cgIDByEntity[ SMDSEntity_Polyhedra ];
 543       elemIt = myMesh->elementsIterator();
 544       while ( elemIt->more() )
 545       {
 546         elem = elemIt->next();
 547         if ( elem->GetEntityType() == SMDSEntity_Polyhedra )
 548         {
 549           if ( elem->GetID() != cgID )
 550             elem2cgID->insert( elem2cgID->end(), make_pair( elem, cgID ));
 551           ++cgID;
 552         }
 553       }
 554     }
 555   } // write polyhedral volumes
 556
 557
 558   // Write groups as boundary conditions
 559   // ------------------------------------
 560
 561   const set<SMESHDS_GroupBase*>& groups = myMesh->GetGroups();
 562   set<SMESHDS_GroupBase*>::const_iterator grpIt = groups.begin();
 563   set< string > groupNames; groupNames.insert(""); // to avoid duplicated and empty names
 564   for ( ; grpIt != groups.end(); ++grpIt )
 565   {
 566     const SMESHDS_GroupBase* group = *grpIt;
 567
 568     // write BC location (default is Vertex)
 569     CGNS_ENUMT( GridLocation_t ) location = CGNS_ENUMV( Vertex );
 570     if ( group->GetType() != SMDSAbs_Node )
 571     {
 572       switch ( meshDim ) {
 573       case 3:
 574         switch ( group->GetType() ) {
 575 #if CGNS_VERSION > 3130
 576         case SMDSAbs_Volume: location = CGNS_ENUMV( CellCenter ); break;
 577 #else
 578         case SMDSAbs_Volume: location = CGNS_ENUMV( FaceCenter ); break;
 579 #endif
 580         case SMDSAbs_Face:   location = CGNS_ENUMV( FaceCenter ); break;
 581         case SMDSAbs_Edge:   location = CGNS_ENUMV( EdgeCenter ); break;
 582         default:;
 583         }
 584         break;
 585       case 2:
 586         switch ( group->GetType() ) {
 587 #if CGNS_VERSION > 3130
 588         case SMDSAbs_Face: location = CGNS_ENUMV( CellCenter ); break;
 589 #else
 590         case SMDSAbs_Face: location = CGNS_ENUMV( FaceCenter ); break;
 591 #endif
 592         case SMDSAbs_Edge: location = CGNS_ENUMV( EdgeCenter ); break;
 593         default:;
 594         }
 595         break;
 596       case 1:
 597 #if CGNS_VERSION > 3130
 598         location = CGNS_ENUMV( CellCenter ); break;
 599 #else
 600         location = CGNS_ENUMV( EdgeCenter ); break;
 601 #endif
 602         break;
 603       }
 604     }
 605
 606     // try to extract type of boundary condition from the group name
 607     string name = group->GetStoreName();
 608     CGNS_ENUMT( BCType_t ) bcType = getBCType( name );
 609     if ( !groupNames.insert( name ).second ) // assure name uniqueness
 610     {
 611       int index = 1;
 612       string newName;
 613       do {
 614         newName = SMESH_Comment( name ) << "_" << index++;
 615       }
 616       while ( !groupNames.insert( newName ).second );
 617       name = newName;
 618     }
 619     // write IDs of elements
 620     vector< cgsize_t > pnts;
 621     pnts.reserve( group->Extent() );
 622     SMDS_ElemIteratorPtr elemIt = group->GetElements();
 623     while ( elemIt->more() )
 624     {
 625       const SMDS_MeshElement* elem = elemIt->next();
 626       pnts.push_back( cgnsID( elem, elem2cgIDByEntity[ elem->GetEntityType() ]));
 627     }
 628     if ( pnts.size() == 0 )
 629       continue; // can't store empty group
 630     int iBC;
 631     if ( cg_boco_write( _fn, iBase, iZone, name.c_str(), bcType,
 632                         CGNS_ENUMV( PointList ), pnts.size(), &pnts[0], &iBC) != CG_OK )
 633       return addMessage( cg_get_error(), /*fatal = */true);
 634
 635     // write BC location
 636     if ( location != CGNS_ENUMV( Vertex ) || meshDim == 1 )
 637     {
 638       if ( cg_boco_gridlocation_write( _fn, iBase, iZone, iBC, location) != CG_OK )
 639         return addMessage( cg_get_error(), /*fatal = */false);
 640     }
 641   }
 642   return DRS_OK;
 643 }
 644
 645 //================================================================================
 646 /*!
 647  * \brief Constructor
 648  */
 649 //================================================================================
 650
 651 DriverCGNS_Write::DriverCGNS_Write(): _fn(0), _elementsByType( false )
 652 {
 653 }
 654
 655 //================================================================================
 656 /*!
 657  * \brief Close the cgns file at destruction
 658  */
 659 //================================================================================
 660
 661 DriverCGNS_Write::~DriverCGNS_Write()
 662 {
 663   if ( _fn > 0 )
 664     cg_close( _fn );
 665 }