field_to_vertices.cpp

Go to the documentation of this file.

/** \file field_to_vertices.cpp
  \brief Field to vertices
  \example mofem/tools/field_to_vertices.cpp
 
*/
 
#include <MoFEM.hpp>
#include <SaveVertexDofOnTag.hpp>
 
using namespace MoFEM;
 
static char help[] = "...\n\n";
 
int main(int argc, char *argv[]) {
  MoFEM::Core::Initialize(&argc, &argv, (char *)0, help);
 
  try {
 
    // global variables
    char mesh_file_name[255];
    PetscBool flg_file = PETSC_FALSE;
    char field_name_param[255] = "RHO";
    PetscOptionsBegin(PETSC_COMM_WORLD, "", "Field to vertices options",
                      "none");
    CHKERR PetscOptionsString("-my_file", "mesh file name", "", "mesh.h5m",
                              mesh_file_name, 255, &flg_file);
    CHKERR PetscOptionsString("-my_field", "field name", "", "FIELD",
                              field_name_param, 255, PETSC_NULLPTR);
    PetscOptionsEnd();
 
    std::string field_name(field_name_param);
 
    moab::Core mb_instance;
    moab::Interface &moab = mb_instance;
    ParallelComm *pcomm = ParallelComm::get_pcomm(&moab, MYPCOMM_INDEX);
    if (pcomm == NULL)
      pcomm = new ParallelComm(&moab, PETSC_COMM_WORLD);
    const char *option;
    option = "";
    CHKERR moab.load_file(mesh_file_name, 0, option);
 
    // Create MoFEM  database
    MoFEM::Core core(moab);
    MoFEM::Interface &m_field = core;
 
    if (flg_file != PETSC_TRUE) {
      SETERRQ(PETSC_COMM_SELF, MOFEM_INVALID_DATA,
              "*** ERROR -my_file (MESH FILE NEEDED)");
    }
    BitRefLevel bit_level0;
    bit_level0.set(0);
    CHKERR m_field.getInterface<BitRefManager>()->setBitRefLevelByDim(
        0, 3, bit_level0);
    CHKERR m_field.build_fields();
 
    bool field_flg = false;
    auto fields_ptr = m_field.get_fields();
    for (auto field : (*fields_ptr)) {
      bool check_space = field->getSpace() == H1;
      if (field->getName() == field_name && check_space)
        field_flg = true;
    }
    if (!field_flg) {
      SETERRQ(PETSC_COMM_SELF, MOFEM_INVALID_DATA,
              "*** ERROR -my_field (FIELD (in H1 space) is NOT FOUND)");
    }
 
    BasicFiniteElements::SaveVertexDofOnTag ent_method(m_field,
                                                       field_name.c_str());
 
    CHKERR m_field.loop_dofs(field_name.c_str(), ent_method);
    PetscPrintf(PETSC_COMM_WORLD, "\nDone. Saving files... \n");
 
    //TODO: Higher order field mapping
    CHKERR m_field.getInterface<BitRefManager>()->writeBitLevelByType(
        bit_level0, BitRefLevel().set(), MBTET, "out_mesh.vtk", "VTK", "");
    CHKERR moab.write_file("out.h5m");
  }
  CATCH_ERRORS;
 
  CHKERR MoFEM::Core::Finalize();
 
  return 0;
}