#include <grf_pde_1d.h>
#include <pde_1d.h>
#include <cstring>
#include <fstream>
#include <iostream>

using namespace std;

/******************* GRF_PDE_1D ******************************************/
/************************************************************************/
/* Creator for pde_1d graph.                                            */
/* pde : pointer to the m_pde_1d to display (do not free!)              */
/* file_name : filename prefix                                          */
/************************************************************************/
grf_pde_1d::grf_pde_1d(pde_1d *pde,string file_name) : file_name_(file_name)
{ 
  nx_    = pde->nx();
  xmin_  = pde->xmin();
  xmax_  = pde->xmax();
  dx_    = pde->dx();
};

/**************************************************************/
/* Ouput a single txt 1D graphic file                         */
/* name : file descriptor (Energy of fct name)                */
/* findex : file index                                        */
/* p : pointer to the data array                              */
/* vsize : size of data vector (No of fcts)                   */
/* t : value of t for the snapshot                            */
/* Output format:                                             */
/* The file should be a .m file. It can be execute by matlab  */
/*    # t = t                                                 */ 
/*    ti Fi                                                   */
/* Throw a string on error                                    */
/**************************************************************/
void grf_pde_1d::write_1_grf(const char *name,int findex,double *p, int vsize,
                                double t)
{ int ix,index;
  double x;
  string fname = make_file_name(file_name_+"_"+name,findex,"txt");

  ofstream ofs(fname.c_str());

  if(!ofs)
  { throw((string)" Error: Can't open file : "+fname);
  }

  ofs <<"# t = "<<t<<";\n";

  index = 0 ;
  x = xmin_;
  for(ix = 0; ix < nx_; ix += 1)
  { ofs << x << " " << p[index] << "\n";
    index += vsize;
    x += dx_;
  }
}

// Create a file name of the form:
//    NAME_DDD.XXX
// where NAME is the string name
//       DDD is the index
//       XXX is the suffix
// Return the nameas a string   
//*******************************
string grf_pde_1d::make_file_name(string name,int index,string suffix)
{ char Buff[40];

  sprintf(Buff,"%d",index);

  return(name+"_"+Buff+"."+suffix);
}

/*********************** PERIODIC GRF ***************************/
/**************************************************************/
/* Ouput a single txt 1D graphic file for a periodic PDE      */
/* The only difference with grf_pde_1d is that the value of   */
/* the point, p{x_min}=p{xmax}, is added at the end           */
/* name : file descriptor (Energy of fct name)                */
/* findex : file index                                        */
/* p : pointer to the data array                              */
/* vsize : size of data vector (No of fcts)                   */
/* t : value of t for the snapshot                            */
/* Output format:                                             */
/* The file should be a .m file. It can be execute by matlab  */
/*    # t = t                                                 */ 
/*    ti Fi                                                   */
/* Throw a string on error                                    */
/**************************************************************/
void grf_pde_1d_periodic::write_1_grf(const char *name,int findex,double *p, 
          int vsize,double t)
{ int ix,index;
  double x;
  string fname = make_file_name(file_name_+"_"+name,findex,"txt");

  ofstream ofs(fname.c_str());

  if(!ofs)
  { throw((string)" Error: Can't open file : "+fname);
  }

  ofs <<"# t = "<<t<<";\n";

  index = 0 ;
  x = xmin_;
  for(ix = 0; ix < nx_; ix += 1)
  { ofs << x << " " << p[index] << "\n";
    index += vsize;
    x += dx_;
  }
  ofs << x << " " << p[0] << "\n";
}