amroc/amr/F77Interfaces/F77FileOutput.h

Go to the documentation of this file.

// -*- C++ -*-

// Copyright (C) 2002 Ralf Deiterding
// Brandenburgische Universitaet Cottbus
//
// Copyright (C) 2003-2007 California Institute of Technology
// Ralf Deiterding, ralf@cacr.caltech.edu

#ifndef AMROC_F77_FILEOUTPUT_H
#define AMROC_F77_FILEOUTPUT_H

#include "FileOutput.h"

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

public:
  typedef generic_fortran_func generic_func_type;

  typedef void (*out_1_func_type) ( FI(1,VectorType), FI(1,DataType), BI, const INTEGER& meqn, 
                                    const INTEGER& cnt, const DOUBLE& t );
  typedef void (*out_2_func_type) ( FI(2,VectorType), FI(2,DataType), BI, const INTEGER& meqn, 
                                    const INTEGER& cnt, const DOUBLE& t );
  typedef void (*out_3_func_type) ( FI(3,VectorType), FI(3,DataType), BI, const INTEGER& meqn, 
                                    const INTEGER& cnt, const DOUBLE& t );

  F77OutBase(generic_func_type out) : f_out(out), _Equations(VectorType::Length()) {}

  virtual ~F77OutBase() {}

  virtual void Transform(vec_grid_fct_type& u, grid_fct_type& work, const int Time, const int& Level, 
                         const int cnt, const double& t) {
    if (!f_out) return;    
    forall(u,Time,Level,c)
      BBox bb = u(Time,Level,c).bbox();
      if (dim == 1) 
        ((out_1_func_type) f_out)(FA(1,u(Time,Level,c)),FA(1,work(Time,Level,c)),
                                  BOUNDING_BOX(bb),_Equations,cnt,t);
      else if (dim == 2) 
        ((out_2_func_type) f_out)(FA(2,u(Time,Level,c)),FA(2,work(Time,Level,c)),
                                  BOUNDING_BOX(bb),_Equations,cnt,t);
      else if (dim == 3) 
        ((out_3_func_type) f_out)(FA(3,u(Time,Level,c)),FA(3,work(Time,Level,c)),
                                  BOUNDING_BOX(bb),_Equations,cnt,t);
    end_forall 
  }

  inline void SetOutFunc(generic_func_type out) { f_out = out; }
  generic_func_type GetOutFunc() const { return f_out; }

protected:
  generic_func_type f_out;
  int _Equations;
};


template <class VectorType, int dim>
class F77FileOutput : public F77OutBase<VectorType,dim>,
  public FileOutput<VectorType,dim> {
  typedef typename VectorType::InternalDataType DataType;
  typedef FileOutput<VectorType,dim> base;
  typedef F77OutBase<VectorType,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;
  typedef typename out_base::generic_func_type generic_func_type;

  F77FileOutput(generic_func_type out) : out_base(out), base() {}

  virtual ~F77FileOutput() {}
  
  virtual void WriteOut(vec_grid_fct_type& u, grid_fct_type& IOfunc) {
    if (!out_base::f_out) return;    
    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      

---