rich_v2_offset_supernova

#include <iostream>
#include <cmath>
#include <limits>
#include "tessellation/VoronoiMesh.hpp"
#include "newtonian/two_dimensional/hdsim2d.hpp"
#include "newtonian/two_dimensional/interpolations/pcm2d.hpp"
#include "newtonian/two_dimensional/spatial_distributions/uniform2d.hpp"
#include "newtonian/common/ideal_gas.hpp"
#include "newtonian/two_dimensional/geometric_outer_boundaries/SquareBox.hpp"
#include "newtonian/two_dimensional/hydro_boundary_conditions/RigidWallHydro.hpp"
#include "newtonian/common/hllc.hpp"
#include "newtonian/two_dimensional/source_terms/zero_force.hpp"
#include "newtonian/two_dimensional/point_motions/eulerian.hpp"
#include "newtonian/two_dimensional/point_motions/lagrangian.hpp"
#include "newtonian/two_dimensional/point_motions/eulerian.hpp"
#include "newtonian/two_dimensional/point_motions/round_cells.hpp"
#include "newtonian/two_dimensional/diagnostics.hpp"
#include "newtonian/two_dimensional/source_terms/cylindrical_complementary.hpp"
#include "newtonian/two_dimensional/source_terms/CenterGravity.hpp"
#include "newtonian/two_dimensional/source_terms/SeveralSources.hpp"
#include "misc/simple_io.hpp"
#include "newtonian/test_2d/main_loop_2d.hpp"
#include "newtonian/test_2d/kill_switch.hpp"
#include "newtonian/two_dimensional/hdf5_diagnostics.hpp"
#include "misc/mesh_generator.hpp"
#include "tessellation/RoundGrid.hpp"
#include "misc/int2str.hpp"
#include "newtonian/two_dimensional/interpolations/linear_gauss_consistent.hpp"
#include "newtonian/test_2d/consecutive_snapshots.hpp"
#include <boost/foreach.hpp>
#include "newtonian/two_dimensional/hydro_boundary_conditions/FreeFlow.hpp"
#include "misc/utils.hpp"
#include "tessellation/shape_2d.hpp"
#include "newtonian/test_2d/piecewise.hpp"
#include "mpi/MeshPointsMPI.hpp"
#include "mpi/mpi_macro.hpp"
#include "mpi/ConstNumberPerProc.hpp"
#include "misc/hdf5_utils.hpp"
#include "newtonian/test_2d/contour.hpp"
#include "newtonian/two_dimensional/custom_evolutions/Ratchet.cpp"
#include "newtonian/two_dimensional/diagnostics.hpp"
#include "newtonian/two_dimensional/simple_flux_calculator.hpp"
#include "newtonian/two_dimensional/simple_cell_updater.hpp"
#include <fenv.h>

using namespace std;
using namespace simulation2d;

namespace {

  class CellEdgesGetter: public Index2Member<Edge>
  {
  public:
    CellEdgesGetter(const Tessellation& tess,
            const int n):
      tess_(tess), edge_indices_(tess_.GetCellEdges(n)) {}

    size_t getLength(void) const
    {
      return edge_indices_.size();
    }

    Edge operator()(size_t i) const
    {
      return tess_.GetEdge(edge_indices_[i]);
    }

  private:
    const Tessellation& tess_;
    const vector<int> edge_indices_;
  };

  class WriteConserved: public DiagnosticFunction
  {
  public:

    WriteConserved(const string& fname):
      cons_(), fname_(fname) {}

    void operator()(const hdsim& sim)
    {
      cons_.push_back(total_conserved(sim));
      time_.push_back(sim.getTime());
    }

    ~WriteConserved(void)
    {
      ofstream f(fname_.c_str());
      for(size_t i=0;i<cons_.size();++i)
    f << time_[i] << " "
      << cons_[i].mass << " "
      << cons_[i].momentum.x << " "
      << cons_[i].momentum.y << " "
      << cons_[i].energy << " "
      << cons_[i].tracers["ejecta"] << "\n";
      f.close();
    }

  private:
    mutable vector<Extensive> cons_;
    mutable vector<double> time_;
    const string fname_;
  };

  class Constants
  {
  public:

    // Metric prefix
    const double kilo;
    const double centi;

    // Length / distance
    const double parsec;
    const double meter;

    // Time
    const double year;
    const double second;

    // Mass
    const double solar_mass;
    const double gram;

    // Energy
    const double erg;

    // Force
    const double newton;

    // Parameters
    const double black_hole_mass;
    const double zeta;
    const double adiabatic_index;
    const double gravitation_constant;
    const double rho_0;
    const double R_0;
    const double omega_in;
    const double inner_density_prefactor;
    const double R_b;
    const double omega_out;
    const double outer_density_prefactor;
    const double offset;
    const double supernova_energy;
    const double supernova_radius;
    const double supernova_volume;
    const double supernova_mass;
    const double supernova_density;
    const double supernova_pressure;
    const Vector2D lower_left;
    const Vector2D upper_right;

    Constants(void):
      kilo(1e3),
      centi(1e-2),
      parsec(1),
      meter(3.24e-17*parsec),
      year(1),
      second(year/3.16e7),
      solar_mass(1),
      gram(5.03e-34*solar_mass),
      erg(gram*pow(centi*meter/second,2)),
      newton(kilo*gram*meter/pow(second,2)),
      black_hole_mass(1e6*solar_mass),
      zeta(pow(black_hole_mass/(4.3e6*solar_mass),7./15.)),
      adiabatic_index(5./3.),
      gravitation_constant(6.673e-11*newton*pow(meter/(kilo*gram),2)),
      rho_0(2.2e-22*gram/pow(centi*meter,3)),
      R_0(0.04*parsec*zeta),
      omega_in(1),
      inner_density_prefactor(rho_0*pow(R_0,omega_in)),
      R_b(0.4*parsec*zeta),
      omega_out(3),
      outer_density_prefactor
      (inner_density_prefactor*pow(R_b,omega_out-omega_in)),
      offset(0.24*R_b),
      supernova_energy(1e51*erg),
      supernova_radius(0.2*offset),
      supernova_volume((4.*M_PI/3)*pow(supernova_radius,3)),
      supernova_mass(5*solar_mass),
      supernova_density(supernova_mass/supernova_volume),
      supernova_pressure((adiabatic_index-1)*supernova_energy/supernova_volume),
      lower_left(parsec*Vector2D(0,-5*offset)),
      upper_right(parsec*Vector2D(5*offset,5*offset)) {}
  };

  class Bremsstrahlung: public DiagnosticFunction
  {
  public:

    Bremsstrahlung(const string& fname, const Constants& c):
      fname_(fname),
      // Coefficient is taken from Rybicki and Lightman 1979, page 162 equation 5.15b
      boltzmann_constant_(1.38e-16*c.erg),
      coefficient_(1.4e-27*c.erg/pow(c.centi*c.meter,3)/c.second),
      particle_mass_(1.67e-24*c.gram),
      cc_(pow(c.centi*c.meter,3)) {}

    void operator()(const hdsim& sim)
    {
      double total_luminosity = 0;
      const PhysicalGeometry& pg = sim.getPhysicalGeometry();
      const Tessellation& tess = sim.getTessellation();
      for(int i=0;i<static_cast<int>(sim.getAllCells().size());++i)
    total_luminosity += pg.calcVolume(serial_generate(CellEdgesGetter(tess,i)))*
      calcEmissivity(sim.getAllCells()[static_cast<size_t>(i)].density,
             sim.getAllCells()[static_cast<size_t>(i)].pressure);
      time_luminosity_.push_back(std::pair<double,double>(sim.getTime(),total_luminosity));
    }

    ~Bremsstrahlung(void)
    {
      ofstream f(fname_.c_str());
      for(size_t i=0;i<time_luminosity_.size();++i)
    f << time_luminosity_[i].first << " "
      << time_luminosity_[i].second << endl;
      f.close();
    }

  private:

    double calcTemperature(double density, double pressure)
    {
      return particle_mass_*(pressure/density)/boltzmann_constant_;
    }

    double calcEmissivity(double density, double pressure)
    {
      const double T = calcTemperature(density, pressure);
      const double n = (density/particle_mass_)*cc_;
      return coefficient_*sqrt(T)*pow(n,2);
    }

    const string fname_;
    const double boltzmann_constant_;
    const double coefficient_;
    const double particle_mass_;
    const double cc_;
    mutable vector<std::pair<double,double> > time_luminosity_;
  };

  class HydrostaticPressure: public SpatialDistribution
  {
  public:

    HydrostaticPressure(double gm,
            double r0,
            double rho_0,
            double omega_1,
            double R_b,
            double omega_2):
      gm_(gm), r0_(r0), rho_0_(rho_0),
      omega_1_(omega_1), R_b_(R_b), omega_2_(omega_2) {}

    double operator()(const Vector2D& r) const
    {
      if(abs(r)<R_b_)
    return (gm_/r0_)*(rho_0_/(omega_1_+1))*pow(abs(r)/r0_,-omega_1_-1)-
      (gm_/r0_)*(rho_0_/(omega_1_+1))*pow(R_b_/r0_,-omega_1_-1)+
      (gm_/r0_)*(rho_0_/(omega_2_+1))*pow(R_b_/r0_,-omega_1_-1);
      else
    return (gm_/r0_)*(rho_0_/(omega_2_+1))*
      pow(R_b_/r0_,-omega_1_-1)*
      pow(abs(r)/R_b_,-omega_2_-1);
    }

  private:
    const double gm_;
    const double r0_;
    const double rho_0_;
    const double omega_1_;
    const double R_b_;
    const double omega_2_;
  };

  class BoundedRadialPowerLaw: public SpatialDistribution
  {
  public:

    BoundedRadialPowerLaw(const Vector2D& center,
              double index,
              double prefactor,
              double lower_bound,
              double upper_bound):
      center_(center),
      index_(index),
      prefactor_(prefactor),
      lower_bound_(lower_bound),
      upper_bound_(upper_bound) {}

    double operator()(const Vector2D& r) const
    {
      const double radius = max(min(abs(r-center_),
                    upper_bound_),
                lower_bound_);
      return prefactor_*pow(radius,index_);
    }

  private:
    const Vector2D center_;
    const double index_;
    const double prefactor_;
    const double lower_bound_;
    const double upper_bound_;
  };

  bool is_inside_rectangle(const Vector2D& point,
               const Vector2D& lower_left,
               const Vector2D& upper_right)
  {
    return ((point.x > lower_left.x) &&
        (point.x < upper_right.x) &&
        (point.y > lower_left.y) &&
        (point.y < upper_right.y));
  }

  vector<Vector2D> rectangle_clip(const vector<Vector2D>& grid,
                  const Vector2D& lower_left,
                  const Vector2D& upper_right)
  {
    vector<Vector2D> res;
    for(size_t i=0, endp=grid.size();i<endp;++i){
      const Vector2D& point = grid[i];
      if(is_inside_rectangle(point,lower_left,upper_right))
    res.push_back(point);
    }
    return res;
  }

  /*
    vector<double> calc_radius_list(void)
    {
    const double rmin = 1e-4;
    const double q = 1.01;
    const size_t n = 200;
    vector<double> res(n);
    for(size_t i=0;i<n;++i)
    res[i] = rmin*pow(q,i);
    write_number(0,"finished_calc_radius_list.txt");
    return res;
    }
  */

  /*
    vector<Vector2D> create_grid(Vector2D const& lower_left,
    Vector2D const& upper_right,
    double dx2x)
    {
    vector<Vector2D> res;
    for(double x = lower_left.x*(1+0.5*dx2x);
    x<upper_right.x; x*=(1+dx2x)){
    const double dx = x*dx2x;
    for(double y=lower_left.y+dx/2;
    y<upper_right.y; y += dx)
    res.push_back(Vector2D(x,y));
    }
    return res;
    }
  */

  /*
    vector<Vector2D> centered_hexagonal_grid(double r_min,
    double r_max)
    {
    const vector<double> r_list = arange(0,r_max,r_min);
    vector<Vector2D> res;
    for(size_t i=0;i<r_list.size();++i){
    const size_t angle_num = max<size_t>(6*i,1);
    vector<double> angle_list(angle_num,0);
    for(size_t j=0;j<angle_num;++j)
    angle_list.at(j) = 2*M_PI*(double)j/(double)(angle_num);
    for(size_t j=0;j<angle_num;++j)
    res.push_back(r_list.at(i)*Vector2D(cos(angle_list.at(j)),
    sin(angle_list.at(j))));
    }
    return res;
    }
  */

  vector<Vector2D> centered_logarithmic_spiral(double r_min,
                           double r_max,
                           double alpha,
                           const Vector2D& center)
  {
    const double theta_max = log(r_max/r_min)/alpha;
    const vector<double> theta_list =
      arange(0,theta_max,2*M_PI*alpha/(1-0.5*alpha));

    vector<double> r_list(theta_list.size(),0);
    for(size_t i=0;i<r_list.size();++i)
      r_list.at(i) = r_min*exp(alpha*theta_list.at(i));

    vector<Vector2D> res(r_list.size());
    for(size_t i=0;i<res.size();++i)
      res[i] = center+r_list[i]*Vector2D(cos(theta_list.at(i)),
                     sin(theta_list.at(i)));
    return res;
  }

  /*
    vector<Vector2D> complete_grid(double r_inner,
    double r_outer,
    double alpha)
    {
    const vector<Vector2D> inner =
    centered_hexagonal_grid(r_inner*alpha*2*M_PI,
    r_inner);
    const vector<Vector2D> outer =
    centered_logarithmic_spiral(r_inner,
    r_outer,
    alpha);
    return join(inner, outer);
    }
  */

  template<class T> class VectorInitializer
  {
  public:

    VectorInitializer(const T& t):
      buf_(1,t) {}

    VectorInitializer& operator()(const T& t)
    {
      buf_.push_back(t);
      return *this;
    }

    vector<T> operator()(void)
    {
      return buf_;
    }

  private:
    vector<T> buf_;
  };

  vector<ComputationalCell> calc_init_cond(const Tessellation& tess,
                       const Constants& c)
  {
    vector<ComputationalCell> res(static_cast<size_t>(tess.GetPointNo()));
    for(size_t i=0;i<res.size();++i){
      const Vector2D r = tess.GetMeshPoint(static_cast<int>(i));
      res[i].density = Piecewise(Circle(Vector2D(0,-c.offset),c.supernova_radius),
                 Uniform2D(c.supernova_radius),
                 Piecewise(Circle(Vector2D(0,0), c.R_b),
                       BoundedRadialPowerLaw(Vector2D(0,0),
                                 -c.omega_in,
                                 c.inner_density_prefactor,
                                 1e-6, 10),
                       BoundedRadialPowerLaw(Vector2D(0,0),
                                 -c.omega_out,
                                 c.outer_density_prefactor,
                                 1e-6,10)))(r);
      res[i].pressure = Piecewise(Circle(Vector2D(0,-c.offset),c.supernova_radius),
                  Uniform2D(c.supernova_pressure),
                  HydrostaticPressure(c.gravitation_constant*c.black_hole_mass,
                              c.R_0,
                              c.rho_0,
                              c.omega_in,
                              c.R_b,
                              c.omega_out))(r);
      res[i].velocity = Vector2D(0,0);
      res[i].tracers["ejecta"] = Piecewise
    (Circle(Vector2D(0,-c.offset),c.supernova_radius),
     Uniform2D(1), Uniform2D(0))(r);
      res[i].stickers["dummy"] = Circle(Vector2D(0,0),c.supernova_radius)(r);
    }
    return res;
  }

  class SinkFlux: public FluxCalculator
  {
  public:

    SinkFlux(const RiemannSolver& rs):
      rs_(rs) {}

    vector<Extensive> operator()
    (const Tessellation& tess,
     const vector<Vector2D>& point_velocities,
     const vector<ComputationalCell>& cells,
     const EquationOfState& eos,
     const double /*time*/,
     const double /*dt*/) const
    {
      vector<Extensive> res(tess.getAllEdges().size());
      for(size_t i=0;i<tess.getAllEdges().size();++i){
    const Conserved hydro_flux =
      calcHydroFlux(tess, point_velocities,
            cells, eos, i);
    res[i].mass = hydro_flux.Mass;
    res[i].momentum = hydro_flux.Momentum;
    res[i].energy = hydro_flux.Energy;
    for(std::map<std::string,double>::const_iterator it =
          cells.front().tracers.begin();
        it!=cells.front().tracers.end();++it)
      res[i].tracers[it->first] = (it->second)*hydro_flux.Mass;
      }
      return res;
    }

  private:
    const RiemannSolver& rs_;

    const Conserved calcHydroFlux
    (const Tessellation& tess,
     const vector<Vector2D>& point_velocities,
     const vector<ComputationalCell>& cells,
     const EquationOfState& eos,
     const size_t i) const
    {
      const Edge& edge = tess.GetEdge(static_cast<int>(i));
      const std::pair<bool,bool> flags
    (edge.neighbors.first>=0 && edge.neighbors.first<tess.GetPointNo(),
     edge.neighbors.second>=0 && edge.neighbors.second<tess.GetPointNo());
      assert(flags.first || flags.second);
      if(!flags.first){
    const size_t right_index =
      static_cast<size_t>(edge.neighbors.second);
    const ComputationalCell& right_cell = cells[right_index];
    if(right_cell.stickers.find("dummy")->second)
      return Conserved();
    const Vector2D p = Parallel(edge);
    const Primitive right = convert_to_primitive(right_cell,eos);
    const Primitive left = reflect(right,p);
    const Vector2D n = remove_parallel_component
      (tess.GetMeshPoint(edge.neighbors.second) -
       edge.vertices.first, p);
    return rotate_solve_rotate_back
      (rs_, left, right, 0, n, p);
      }
      if(!flags.second){
    const size_t left_index =
      static_cast<size_t>(edge.neighbors.first);
    const ComputationalCell& left_cell = cells[left_index];
    if(left_cell.stickers.find("dummy")->second)
      return Conserved();
    const Primitive left = convert_to_primitive(left_cell, eos);
    const Vector2D p = Parallel(edge);
    const Primitive right = reflect(left,p);
    const Vector2D n = remove_parallel_component
      (edge.vertices.second -
       tess.GetMeshPoint(edge.neighbors.first), p);
    return rotate_solve_rotate_back
      (rs_, left, right, 0, n, p);
      }
      const size_t left_index =
    static_cast<size_t>(edge.neighbors.first);
      const size_t right_index =
    static_cast<size_t>(edge.neighbors.second);
      const ComputationalCell& left_cell = cells[left_index];
      const ComputationalCell& right_cell = cells[right_index];
      if(left_cell.stickers.find("dummy")->second &&
     right_cell.stickers.find("dummy")->second)
    return Conserved();
      const Vector2D p = Parallel(edge);
      const Vector2D n =
    tess.GetMeshPoint(edge.neighbors.second) -
    tess.GetMeshPoint(edge.neighbors.first);
      const double velocity = Projection
    (tess.CalcFaceVelocity
     (point_velocities[left_index],
      point_velocities[right_index],
      tess.GetCellCM(edge.neighbors.first),
      tess.GetCellCM(edge.neighbors.second),
      calc_centroid(edge)),n);
      if(left_cell.stickers.find("dummy")->second){
    const Primitive right =
      convert_to_primitive(right_cell, eos);
    const Primitive left =
      ScalarProd(n,right.Velocity) < 0 ? right :
      reflect(right,p);
    return rotate_solve_rotate_back
                     (rs_,left,right,velocity,n,p);
      }
      if(right_cell.stickers.find("dummy")->second){
    const Primitive left =
      convert_to_primitive(left_cell, eos);
    const Primitive right =
      ScalarProd(n,left.Velocity)>0 ?
      left : reflect(left,p);
    return rotate_solve_rotate_back
      (rs_,left,right,velocity,n,p);
      }
      const Primitive left =
    convert_to_primitive(left_cell, eos);
      const Primitive right =
    convert_to_primitive(right_cell, eos);
      return rotate_solve_rotate_back
    (rs_,left,right,velocity,n,p);
    }
  };

  class TimeStepCalculator: public Index2Member<double>
  {
  public:

    TimeStepCalculator(const Tessellation& tess,
               const vector<ComputationalCell>& cells,
               const EquationOfState& eos,
               const vector<Vector2D>& point_velocities):
      tess_(tess), cells_(cells),
      point_velocities_(point_velocities), eos_(eos) {}

    size_t getLength(void) const
    {
      return cells_.size();
    }

    double operator()(size_t i) const
    {
      return tess_.GetWidth(static_cast<int>(i))/
    (eos_.dp2c(cells_[i].density, cells_[i].pressure)+
     abs(cells_[i].velocity)+
     abs(point_velocities_[i]));
    }

  private:
    const Tessellation& tess_;
    const vector<ComputationalCell>& cells_;
    const vector<Vector2D>& point_velocities_;
    const EquationOfState& eos_;
  };

  class LogCFL: public TimeStepFunction
  {
  public:

    LogCFL(double cfl, const string& fname):
      cfl_(cfl), fname_(fname) {}

    double operator()
    (const Tessellation& tess,
     const vector<ComputationalCell>& cells,
     const EquationOfState& eos,
     const vector<Vector2D>& point_velocities,
     const double /*time*/) const
    {
      const TimeStepCalculator tsc(tess,cells,eos,point_velocities);
      double res = tsc(0);
      size_t argmin = 0;
      for(size_t i=1;i<tsc.getLength();++i){
    const double candidate = tsc(i);
    if(candidate<res && !cells[i].stickers.find("dummy")->second){
      res = candidate;
      argmin = i;
    }
      }
      ofstream f(fname_.c_str());
      f << argmin << endl;
      f << tess.GetMeshPoint(static_cast<int>(argmin)).x << ", ";
      f << tess.GetMeshPoint(static_cast<int>(argmin)).y << endl;
      f << tess.GetWidth(static_cast<int>(argmin)) << endl;
      f << eos.dp2c(cells[argmin].density, cells[argmin].pressure) << endl;
      f << cells[argmin].velocity.x << ", ";
      f << cells[argmin].velocity.y << endl;
      f << point_velocities[argmin].x << ", ";
      f << point_velocities[argmin].y << endl;
      f.close();
      return cfl_*res;
    }

  private:
    const double cfl_;
    const string fname_;
  };
}

class SimData
{

public:

  SimData(const Constants& c):
    pg_(Vector2D(0,0), Vector2D(0,1)),
    outer_(c.lower_left, c.upper_right),
    init_points_(rectangle_clip
         (centered_logarithmic_spiral(0.001,
                          abs(c.upper_right-c.lower_left),
                          0.005,
                          Vector2D(-0.1*c.offset,0)),
          c.lower_left, c.upper_right)),
    tess_(init_points_, outer_),
    eos_(c.adiabatic_index),
    rs_(),
    hbc_(rs_),
    interpm_(),
    raw_point_motion_(),
    point_motion_(raw_point_motion_,eos_),
    alt_point_motion_(),
    gravity_acc_(c.gravitation_constant*c.black_hole_mass,
         0.01*c.parsec,
         c.parsec*Vector2D(-0.01*c.offset,0)),
    gravity_force_(gravity_acc_),
    geom_force_(pg_.getAxis()),
    force_(VectorInitializer<SourceTerm*>(&gravity_force_)
       (&geom_force_)()),
    tsf_(0.3, "dt_log.txt"),
    fc_(rs_),
    cu_(),
    sim_(tess_,
     outer_,
     pg_,
     calc_init_cond(tess_,c),
     eos_,
     alt_point_motion_,
     force_,
     tsf_,
     fc_,
     cu_) {}

  hdsim& getSim(void)
  {
    return sim_;
  }

private:
  const CylindricalSymmetry pg_;
  const SquareBox outer_;
  const vector<Vector2D> init_points_;
  VoronoiMesh tess_;
  VoronoiMesh proc_tess_;
  const IdealGas eos_;
  const Hllc rs_;
  const RigidWallHydro hbc_;
  PCM2D interpm_;
  Lagrangian raw_point_motion_;
  RoundCells point_motion_;
  Eulerian alt_point_motion_;
  CenterGravity gravity_acc_;
  ConservativeForce gravity_force_;
  CylindricalComplementary geom_force_;
  SeveralSources force_;
  //  const SimpleCFL tsf_;
  const LogCFL tsf_;
  const SinkFlux fc_;
  const SimpleCellUpdater cu_;
  hdsim sim_;
};

class CustomDiagnostic
{
public:

  CustomDiagnostic(double dt,
           double t_start):
    dt_(dt),
    t_next_(t_start),
    counter_(0) {}

  void operator()(const hdsim& sim)
  {
    if(sim.getTime()>t_next_){
      write_snapshot_to_hdf5(sim, "snapshot_"+int2str(counter_)+".h5");

      t_next_ += dt_;
      ++counter_;
    }
  }

private:
  const double dt_;
  double t_next_;
  int counter_;
};

namespace {
  /*
    vector<double> calc_decade_vector(double val, size_t decades)
    {
    vector<double> res;
    for(size_t i=0;i<decades;++i)
    res.push_back(val/pow(2.0,static_cast<double>(i)+1));
    return res;
    }
  */
}

namespace {

  /*
    double highest_shocked_point(const hdsim& sim)
    {
    const double mn_thres = 0.01;
    double ys = -10;
    for(int i=0;i<sim.GetCellNo();++i){
    if(sim.GetCell(i).Velocity.y/sim.GetCell(i).SoundSpeed>mn_thres){
    ys = max(ys,sim.GetMeshPoint(i).y);
    }
    }
    return ys;
    }
  */
}

namespace {
  /*
    class MyDiagnostic: public DiagnosticFunction
    {
    public:

    MyDiagnostic(double y_start,
    size_t decades):
    snapshot_times_(calc_decade_vector(y_start, decades)),
    y_next_(y_start), counter_(0), active_(true) {}

    void operator()(const hdsim& sim)
    {
    if(!active_)
    return;

    if(highest_shocked_point(sim)>y_next_){
    write_snapshot_to_hdf5(sim,"snapshot_"+int2str(counter_)+".h5");
    if(static_cast<size_t>(counter_)>snapshot_times_.size()-1)
    active_ = false;
    else{
    y_next_ = snapshot_times_.at(static_cast<size_t>(counter_));
    ++counter_;
    }
    }
    }

    private:
    const vector<double> snapshot_times_;
    mutable double y_next_;
    mutable int counter_;
    mutable bool active_;
    };
  */
}

namespace {
  class MultipleDiagnostics: public DiagnosticFunction
  {
  public:

    MultipleDiagnostics(const vector<DiagnosticFunction*>& diag_list):
      diag_list_(diag_list) {}

    void operator()(const hdsim& sim)
    {
      BOOST_FOREACH(DiagnosticFunction* df, diag_list_)
    {
      (*df)(sim);
    }
    }

  private:
    const vector<DiagnosticFunction*> diag_list_;
  };
}

namespace {

  class WriteCycle: public DiagnosticFunction
  {
  public:

    WriteCycle(const string& fname):
      fname_(fname) {}

    void operator()(const hdsim& sim)
    {
      write_number(sim.getCycle(),fname_);
    }

  private:
    const string fname_;
  };

  void my_main_loop(hdsim& sim, const Constants& c)
  {
    //  sim.SetCfl(0.1);
    const double tf = 400*c.year;
    SafeTimeTermination term_cond_raw(tf,1e6);
    ConsecutiveSnapshots diag1(new ConstantTimeInterval(1*c.year),
                   new Rubric("snapshot_",".h5"));
    WriteTime diag2("time.txt");
    WriteConserved diag4("conserved.txt");
    Bremsstrahlung diag5("luminosity_history.txt",c);
    WriteCycle diag6("cycle.txt");
    vector<DiagnosticFunction*> diag_list;
    diag_list.push_back(&diag1);
    diag_list.push_back(&diag2);
    //    diag_list.push_back(&diag3);
    diag_list.push_back(&diag4);
    diag_list.push_back(&diag5);
    diag_list.push_back(&diag6);
    MultipleDiagnostics diag(diag_list);

    main_loop(sim,
          term_cond_raw,
          &hdsim::TimeAdvance,
          &diag);
  }
}

namespace {
  void report_error(UniversalError const& eo)
  {
    cout << eo.GetErrorMessage() << endl;
    cout.precision(14);
    for(size_t i=0;i<eo.GetFields().size();++i)
      cout << eo.GetFields()[i] << " = "
       << eo.GetValues()[i] << endl;
  }
}

int main(void)
{
  feenableexcept(FE_DIVBYZERO | FE_INVALID | FE_OVERFLOW);
  //  MPI_Init(NULL, NULL);

  try{

    Constants c;

    SimData sim_data(c);
    hdsim& sim = sim_data.getSim();

    write_snapshot_to_hdf5(sim,
               "initial.h5");

    my_main_loop(sim,c);

    write_snapshot_to_hdf5(sim,
               "final.h5");

  }
  catch(const UniversalError& eo){
    report_error(eo);
    throw;
  }

  return 0;
}