SparseGridsDefine.cpp

Go to the documentation of this file.

#include <sparse_grids_simulation.h>

int SparseGridsSimulation::define(  int a_rank, 
                                    int a_nproc 
                                 )
{
  m_rank = a_rank;
  m_nproc = a_nproc;

  if (m_sim_fg != NULL) {
    fprintf(stderr, "Error in SparseGridsSimulation::define() -\n");
    fprintf(stderr, "  m_sim_fg not NULL!\n");
    exit(1);
  }

  /* default values */
  m_imin = 2;
  m_write_sg_solutions = 0;
  m_print_sg_errors = 0;
  m_interp_order = 6;

  if (!m_rank) {

    FILE *in;
    in = fopen(_SPARSEGRIDS_SIM_INP_FNAME_,"r");

    if (!in) {

      fprintf(stderr, "Error in SparseGridsSimulation::Define() -\n");
      fprintf(stderr, "  %s file not found.\n", _SPARSEGRIDS_SIM_INP_FNAME_);
      exit(1);

    } else {

      int ferr;
      char word[_MAX_STRING_SIZE_];
      ferr = fscanf(in,"%s", word); if (ferr != 1) return(1);

      if (std::string(word) == "begin") {

        while (std::string(word) != "end") {

          ferr = fscanf(in,"%s",word); if (ferr != 1) return(1);

          if (std::string(word) == "log2_imin") {

            ferr = fscanf(in,"%d",&m_imin); if (ferr != 1) return(1);

          } else if (std::string(word) == "interp_order") {

            ferr = fscanf(in,"%d",&m_interp_order); if (ferr != 1) return(1);

          } else if (std::string(word) == "write_sg_solutions") {

            char answer[_MAX_STRING_SIZE_];
            ferr = fscanf(in,"%s",answer); if (ferr != 1) return(1);
            m_write_sg_solutions = (strcmp(answer,"yes") ? 0 : 1);

          } else if (std::string(word) == "write_sg_errors") {

            char answer[_MAX_STRING_SIZE_];
            ferr = fscanf(in,"%s",answer); if (ferr != 1) return(1);
            m_print_sg_errors = (strcmp(answer,"yes") ? 0 : 1);

          } else if (std::string(word) != "end") {

            char useless[_MAX_STRING_SIZE_];
            ferr = fscanf(in,"%s",useless);
            printf("Warning: keyword %s in file \"%s\" with value %s not recognized or extraneous. Ignoring.\n",
                    _SPARSEGRIDS_SIM_INP_FNAME_, word, useless );

          }

          if (ferr != 1) return(1);
        }

      } else {
        fprintf( stderr, "Error: Illegal format in file \"%s\". Word read is: %s\n",
                 _SPARSEGRIDS_SIM_INP_FNAME_, word);
        return 1;
      }

      fclose(in);

    }

    /* print useful stuff to screen */
    printf("Sparse grids inputs:\n");
    printf("  log2 of minimum grid size:  %d\n", m_imin);
    printf("  interpolation order:  %d\n", m_interp_order);
    printf( "  write sparse grids solutions?  %s\n",
            ( m_write_sg_solutions == 1 ? "yes" : "no" ) );
  }

#ifndef serial
  MPI_Bcast(&m_imin,1,MPI_INT,0,MPI_COMM_WORLD);
  MPI_Bcast(&m_interp_order,1,MPI_INT,0,MPI_COMM_WORLD);
  MPI_Bcast(&m_write_sg_solutions,1,MPI_INT,0,MPI_COMM_WORLD);
  MPI_Bcast(&m_print_sg_errors,1,MPI_INT,0,MPI_COMM_WORLD);
#endif

  m_sim_fg = new SimulationObject;
  /* the following are deliberately set to junk values
   * to ensure they are never used */
  m_sim_fg->solver.my_idx = -1;
  m_sim_fg->solver.nsims = -1;
  /* these are okay */
  m_sim_fg->mpi.rank = m_rank;
  m_sim_fg->mpi.nproc = m_nproc;

  if (!m_rank) {
    printf("Allocated full grid simulation object(s).\n");
  }

  m_is_defined = true;
  return 0;
}