visualization.cpp

#include "include/problemGeneration.hpp"

using namespace std;
using namespace mfem;

void Visualize(HYPRE_ParVector X, HYPRE_Int dump_mesh, ProblemOptionsList &options, ProblemInfo &probInfo, string custom_sol_name)
{
   int num_procs, myid;
   MPI_Comm_size(MPI_COMM_WORLD, &num_procs);
   MPI_Comm_rank(MPI_COMM_WORLD, &myid);

   if (options.problem >= 0)
   {
      // Recover the parallel grid function corresponding to X. This is the
      // local finite element solution on each processor.

      // Save the refined mesh and the solution in parallel. This output can
      // be viewed later using GLVis: "glvis -np <np> -m mesh -g sol".
      ostringstream mesh_name, sol_name, nodes_name;
      if (custom_sol_name == "*") 
      {
         mesh_name << "outputs/mesh" << "." << setfill('0') << setw(6) << myid;
         sol_name << "outputs/sol" << "." << setfill('0') << setw(6) << myid;
         nodes_name << "outputs/nodes" << "." << setfill('0') << setw(6) << myid; 
      }
      else 
      {
         sol_name << "outputs/" << custom_sol_name << "." << setfill('0') << setw(6) << myid;
         mesh_name << "outputs/mesh_" << custom_sol_name << "." << setfill('0') << setw(6) << myid;
         nodes_name << "outputs/nodes_" << custom_sol_name << "." << setfill('0') << setw(6) << myid; 
      }

      if (dump_mesh)
      {
         ofstream mesh_ofs(mesh_name.str().c_str());
         mesh_ofs.precision(8);
         probInfo.pmesh->Print(mesh_ofs);
      }
      

      HypreParVector X_mfem(X);
      probInfo.a->RecoverFEMSolution(X_mfem, *(probInfo.b), *(probInfo.x));
      ofstream sol_ofs(sol_name.str().c_str());
      sol_ofs.precision(8);
      probInfo.x->Save(sol_ofs);

      // !!! Try
      ParFiniteElementSpace nodes_fespace(probInfo.pmesh, probInfo.fec, probInfo.pmesh->Dimension());
      ParGridFunction nodes(&nodes_fespace);
      probInfo.pmesh->GetNodes(nodes);
      HypreParVector *nodes_true = nodes.GetTrueDofs();

      char test_name[256];
      sprintf(test_name, "outputs/nodes");
      /* nodes_true->Print(nodes_name.str().c_str()); */
      nodes_true->Print(test_name);

   }
   else
   {
      ostringstream sol_name, suffix;
      suffix << "_problem" << options.problem << "P" << num_procs << "n" << options.n;
      if (custom_sol_name == "*") sol_name << "outputs/sol" << suffix.str();
      else sol_name << "outputs/" << custom_sol_name;

      HYPRE_ParVectorPrint(X, sol_name.str().c_str());
      if (probInfo.x_coords.size() && dump_mesh)
      {
         ostringstream coord_name;
         coord_name << "outputs/x_coords" << suffix.str() << "." << myid; 
         ofstream coords_out(coord_name.str().c_str());
         for (auto coord = probInfo.x_coords.begin(); coord != probInfo.x_coords.end(); ++coord)
            coords_out << *coord << endl;
      }
      if (probInfo.y_coords.size() && dump_mesh)
      {
         ostringstream coord_name;
         coord_name << "outputs/y_coords" << suffix.str() << "." << myid; 
         ofstream coords_out(coord_name.str().c_str());
         for (auto coord = probInfo.y_coords.begin(); coord != probInfo.y_coords.end(); ++coord)
            coords_out << *coord << endl;
      }
      if (probInfo.z_coords.size() && dump_mesh)
      {
         ostringstream coord_name;
         coord_name << "outputs/z_coords" << suffix.str() << "." << myid; 
         ofstream coords_out(coord_name.str().c_str());
         for (auto coord = probInfo.z_coords.begin(); coord != probInfo.z_coords.end(); ++coord)
            coords_out << *coord << endl;
      }
   }
}