GModelIO_SU2.cpp

Go to the documentation of this file.

// Gmsh - Copyright (C) 1997-2022 C. Geuzaine, J.-F. Remacle
//
// See the LICENSE.txt file in the Gmsh root directory for license information.
// Please report all issues on https://gitlab.onelab.info/gmsh/gmsh/issues.
 
#include <stdlib.h>
#include <string.h>
#include "GModel.h"
#include "OS.h"
#include "MElement.h"
 
static std::string physicalName(GModel *m, int dim, int num)
{
  std::string name = m->getPhysicalName(dim, num);
  if(name.empty()) {
    char tmp[256];
    sprintf(tmp, "%s%d",
            (dim == 3) ? "PhysicalVolume" :
            (dim == 2) ? "PhysicalSurface" :
                         "PhysicalLine",
            num);
    name = tmp;
  }
  for(std::size_t i = 0; i < name.size(); i++)
    if(name[i] == ' ') name[i] = '_';
  return name;
}
 
int GModel::writeSU2(const std::string &name, bool saveAll,
                     double scalingFactor)
{
  FILE *fp = Fopen(name.c_str(), "w");
  if(!fp) {
    Msg::Error("Unable to open file '%s'", name.c_str());
    return 0;
  }
 
  int ndime = getDim();
  if(ndime != 2 && ndime != 3) {
    Msg::Error("SU2 mesh output valid only for 2D or 3D models (not %dD)",
               ndime);
    fclose(fp);
    return 0;
  }
 
  if(noPhysicalGroups()) saveAll = true;
 
  fprintf(fp, "NDIME= %d\n", ndime);
 
  // all interior elements are printed in a single section; indices start at 0;
  // node ordering is the same as VTK
  int nelem = 0;
  if(ndime == 2) {
    for(auto it = firstFace(); it != lastFace(); it++)
      if(saveAll || (*it)->physicals.size())
        nelem += (*it)->getNumMeshElements();
  }
  else {
    for(auto it = firstRegion(); it != lastRegion(); it++)
      if(saveAll || (*it)->physicals.size())
        nelem += (*it)->getNumMeshElements();
  }
  int npoin = indexMeshVertices(saveAll);
 
  Msg::Info("Writing %d elements and %d nodes", nelem, npoin);
 
  // elements
  fprintf(fp, "NELEM= %d\n", nelem);
  int num = 0;
  if(ndime == 2) {
    for(auto it = firstFace(); it != lastFace(); it++)
      if(saveAll || (*it)->physicals.size())
        for(std::size_t i = 0; i < (*it)->getNumMeshElements(); i++)
          (*it)->getMeshElement(i)->writeSU2(fp, num++);
  }
  else {
    for(auto it = firstRegion(); it != lastRegion(); it++)
      if(saveAll || (*it)->physicals.size())
        for(std::size_t i = 0; i < (*it)->getNumMeshElements(); i++)
          (*it)->getMeshElement(i)->writeSU2(fp, num++);
  }
 
  // vertices
  fprintf(fp, "NPOIN= %d\n", npoin);
  std::vector<GEntity *> entities;
  getEntities(entities);
  for(std::size_t i = 0; i < entities.size(); i++)
    for(std::size_t j = 0; j < entities[i]->mesh_vertices.size(); j++)
      entities[i]->mesh_vertices[j]->writeSU2(fp, ndime, scalingFactor);
 
  // markers for physical groups of dimensions (ndime - 1) and ndime
  std::map<int, std::vector<GEntity *> > groups[4];
  getPhysicalGroups(groups);
  int nmark = groups[ndime - 1].size() + groups[ndime].size();
  if(nmark) {
    fprintf(fp, "NMARK= %d\n", nmark);
    for(int dim = ndime-1; dim <= ndime; dim++) {
      if(groups[dim].size()) {
        for(auto it = groups[dim].begin(); it != groups[dim].end(); it++) {
          std::vector<GEntity *> &entities = it->second;
          int n = 0;
          for(std::size_t i = 0; i < entities.size(); i++)
            n += entities[i]->getNumMeshElements();
          if(n) {
            fprintf(fp, "MARKER_TAG= %s\n", physicalName(this, dim, it->first).c_str());
            fprintf(fp, "MARKER_ELEMS= %d\n", n);
            for(std::size_t i = 0; i < entities.size(); i++)
              for(std::size_t j = 0; j < entities[i]->getNumMeshElements(); j++)
                entities[i]->getMeshElement(j)->writeSU2(fp, -1);
          }
        }
      }
    }
  }
 
  fclose(fp);
  return 1;
}