amroc/amr/Interfaces/SchemeFileOutput.h

Go to the documentation of this file.

// -*- C++ -*-

// Copyright (C) 2013 Oak Ridge National Laboratory
// Ralf Deiterding, deiterdingr@ornl.gov

#ifndef SCHEME_FILEOUTPUT_H
#define SCHEME_FILEOUTPUT_H

#include "FileOutput.h"

template <class SchemeType, int dim>
class SchemeOutBase {
  typedef typename SchemeType::VectorType VectorType;
  typedef typename VectorType::InternalDataType DataType;
  typedef GridFunction<VectorType,dim> vec_grid_fct_type;  
  typedef GridFunction<DataType,dim> grid_fct_type;  

public:
  SchemeOutBase(SchemeType &scheme) : _scheme(scheme) {}

  virtual ~SchemeOutBase() {}

  virtual void Transform(vec_grid_fct_type& u, grid_fct_type& work, const int Time, const int& Level, 
                         const int cnt, const double& t) {
    forall(u,Time,Level,c)
      Scheme().Output(u(Time,Level,c),work(Time,Level,c),cnt,0);
    end_forall 
  }

  inline SchemeType& Scheme() { return _scheme; }
  inline const SchemeType& Scheme() const { return _scheme; }

protected:
  SchemeType &_scheme;
};


template <class SchemeType, int dim>
class SchemeFileOutput : public SchemeOutBase<SchemeType,dim>,
  public FileOutput<typename SchemeType::VectorType,dim> {
  typedef typename SchemeType::VectorType VectorType;
  typedef typename VectorType::InternalDataType DataType;
  typedef FileOutput<VectorType,dim> base;
  typedef SchemeOutBase<SchemeType,dim> out_base;
public:
  typedef typename base::vec_grid_fct_type vec_grid_fct_type;  
  typedef typename base::grid_fct_type grid_fct_type;
  typedef typename base::grid_data_type grid_data_type;

  SchemeFileOutput(SchemeType &scheme) : out_base(scheme), base() {}

  virtual ~SchemeFileOutput() {}
  
  virtual void WriteOut(vec_grid_fct_type& u, grid_fct_type& IOfunc) {
    for (int cnt=1; cnt<=base::Ncnt(); cnt++) { 
      if (base::CompName[cnt-1].c_str()[0] == '-') 
        continue;
      for (int lev=0; lev<=FineLevel(base::GH()); lev++) {
        int Time = CurrentTime(base::GH(),lev); 
        double t = GetPhysicalTime(u,Time,lev);
        this->Transform(u,IOfunc,Time,lev,cnt,t);
        base::WriteOut(IOfunc,base::CompName[cnt-1].c_str(),Time,lev,t);
      }
#ifndef DAGH_NO_MPI
      MPI_Barrier(comm_service::comm()); 
#endif
    }
  }

  virtual void WriteOut(grid_fct_type& IOfunc, const char* name) 
  { base::WriteOut(IOfunc,name); }  

  virtual void WriteOut(grid_fct_type& IOfunc, const char* name, 
                        const int& Time, const int& Level, const double& t) 
  { base::WriteOut(IOfunc,name,Time,Level,t); }

  virtual void WriteOut(grid_data_type& IOdata, const char* name, 
                        const int& Time, const int& Level, const double& t) {
    base::WriteOut(IOdata,name,Time,Level,t);
  }
};

#endif      

---