amroc/amr/F77Interfaces/F77FileExactSolution.h

Go to the documentation of this file.

// -*- C++ -*-

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

#ifndef AMROC_F77_FILE_EXACTSOLUTION_H
#define AMROC_F77_FILE_EXACTSOLUTION_H

#include "F77Interfaces/F77ExactSolution.h"
#include "F77Interfaces/F77FileInput.h"
#include "F77Interfaces/F77UpdateLevelTransfer.h"

template <class VectorType, int dim>
class F77FileExactSolution : public F77ExactSolution<VectorType,dim> {
  typedef F77ExactSolution<VectorType,dim> base;
  typedef typename VectorType::InternalDataType DataType;
public:    
  typedef F77FileInput<VectorType,dim> file_input_type;
  typedef F77UpdateLevelTransfer<Vector<DataType,1>, dim> level_transfer_type;
  typedef typename base::vec_grid_fct_type vec_grid_fct_type;  
  typedef typename base::grid_fct_type grid_fct_type;
  typedef typename base::generic_func_type generic_func_type; 
  typedef typename base::vec_grid_data_type vec_grid_data_type;
  typedef typename base::grid_data_type grid_data_type;

  F77FileExactSolution(generic_func_type initial) : 
    base(initial), _ReadInFirst(0), _PhysicalTime(0.0), _TimeDelta(0.0) {
    _FileInput = (file_input_type*) 0;
    _LevelTransfer = (level_transfer_type*) 0;
  }
  
  F77FileExactSolution(generic_func_type exact, generic_func_type in,
                       generic_func_type prolong) : 
    base(exact), _ReadInFirst(0), _PhysicalTime(0.0), _TimeDelta(0.0) {
    _FileInput = (file_input_type*) 0;
    _LevelTransfer = (level_transfer_type*) 0;
    if (!in) return; 
    _FileInput = new file_input_type(in);
    if (prolong) 
      _FileInput->SetLevelTransfer((typename file_input_type::leveltransfer_type*)
                                   (new level_transfer_type(prolong, (generic_func_type) 0, 
                                                            (generic_func_type) 0)));
  }
  
  F77FileExactSolution(generic_func_type exact, generic_func_type out, 
                       generic_func_type in, generic_func_type prolong) : 
    base(exact), _ReadInFirst(0), _PhysicalTime(0.0), _TimeDelta(0.0) {
    _FileInput = (file_input_type*) 0;
    _LevelTransfer = (level_transfer_type*) 0;
    if (!out || !in) return; 
    _FileInput = new file_input_type(out,in);
    if (prolong) 
      _FileInput->SetLevelTransfer((typename file_input_type::leveltransfer_type*)
                                   (new level_transfer_type(prolong, (generic_func_type) 0,
                                                            (generic_func_type) 0)));
  }
  
  F77FileExactSolution(generic_func_type exact, generic_func_type out, 
                       generic_func_type in, generic_func_type prolong, 
                       generic_func_type restrct) : 
    base(exact), _ReadInFirst(0), _PhysicalTime(0.0), _TimeDelta(0.0) {
    _FileInput = (file_input_type*) 0;
    _LevelTransfer = (level_transfer_type*) 0;
    if (!out || !in) return; 
    _FileInput = new file_input_type(out,in);
    if (prolong) {
      if (!restrct)
        _FileInput->SetLevelTransfer((typename file_input_type::leveltransfer_type*)
                                     (new level_transfer_type(prolong, (generic_func_type) 0,
                                                              (generic_func_type) 0)));
      else
        _FileInput->SetLevelTransfer((typename file_input_type::leveltransfer_type*)
                                     (new level_transfer_type(prolong, restrct, (generic_func_type) 0)));
    }   
  }
  
  F77FileExactSolution(generic_func_type exact, generic_func_type out, 
                       generic_func_type in, generic_func_type prolong, 
                       generic_func_type restrct, generic_func_type update) : 
  base(exact), _ReadInFirst(0), _PhysicalTime(0.0, _TimeDelta(0.0)) {
    _FileInput = (file_input_type*) 0;
    _LevelTransfer = (level_transfer_type*) 0;
    if (!out || !in) return; 
    _FileInput = new file_input_type(out,in);
    if (prolong) {
      if (!restrct) {
        if (!update)
          _FileInput->SetLevelTransfer((typename file_input_type::leveltransfer_type*)
                                       (new level_transfer_type(prolong, (generic_func_type) 0,
                                                                (generic_func_type) 0)));
        else
          _FileInput->SetLevelTransfer((typename file_input_type::leveltransfer_type*)
                                       (new level_transfer_type(prolong, (generic_func_type) 0, update)));
      }
      else {
        if (!update)
          _FileInput->SetLevelTransfer((typename file_input_type::leveltransfer_type*)
                                       (new level_transfer_type(prolong, restrct, (generic_func_type) 0)));
        else
          _FileInput->SetLevelTransfer((typename file_input_type::leveltransfer_type*)
                                       (new level_transfer_type(prolong, restrct, update)));
      }
    }   
  }

  virtual ~F77FileExactSolution() {
    if (_FileInput) delete _FileInput;
    if (_LevelTransfer) delete _LevelTransfer;
  }

  virtual void register_at(ControlDevice& Ctrl) { base::register_at(Ctrl); }
  virtual void register_at(ControlDevice& Ctrl, const std::string& prefix) {
    base::register_at(Ctrl,prefix);
    if (_FileInput) _FileInput->register_at(base::LocCtrl, "");
    RegisterAt(base::LocCtrl,"ReadInFirst",_ReadInFirst);
    RegisterAt(base::LocCtrl,"Time",_PhysicalTime);
    RegisterAt(base::LocCtrl,"TimeDelta",_TimeDelta);
  }

  virtual void init() { if (_FileInput) _FileInput->init(); }
  virtual void update() { if (_FileInput) _FileInput->update(); }
  virtual void finish() { if (_FileInput) _FileInput->finish(); } 

  virtual void SetupData(GridHierarchy* gh, const int& ghosts) {
    base::SetupData(gh,ghosts);
    if (_FileInput) _FileInput->SetupData(gh,ghosts);
    if (_ReadInFirst!=0 && _FileInput) 
      _FileInput->SetOutFunc((generic_func_type) 0);
  }

  virtual bool Set(vec_grid_fct_type& u, grid_fct_type& work, const double& t) {
    if (_PhysicalTime<=0.)
      return base::Set(u,work,t);
    else if (t<_PhysicalTime-_TimeDelta || t>_PhysicalTime+_TimeDelta) 
      return false;

    for (int lev=0; lev<=FineLevel(base::GH()); lev++) {
      int Time = CurrentTime(base::GH(),lev); 
      int TStep = TimeStep(u,lev);
      SwapTimeLevels(u,lev,Time,Time+TStep);
      SwapTimeLevels(work,lev,Time,Time+TStep);
    }

    if (_FileInput && _ReadInFirst>0 && !_FileInput->GetOutFunc()) 
      _FileInput->ReadIn(u,work);

    for (int lev=0; lev<=FineLevel(base::GH()); lev++) {
      int Time = CurrentTime(base::GH(),lev); 

      forall (u,Time,lev,c) 
        base::SetGrid(u(Time,lev,c),work(Time,lev,c),lev,t);
      end_forall      

      BoundaryUpdate(u,Time,lev);
    }

    if (_FileInput && _ReadInFirst==0) {
      _FileInput->ReadIn(u,work);
      for (int lev=0; lev<=FineLevel(base::GH()); lev++) {
        int Time = CurrentTime(base::GH(),lev); 
        BoundaryUpdate(u,Time,lev);
      }
    }     

    for (int lev=0; lev<=FineLevel(base::GH()); lev++) {
      int Time = CurrentTime(base::GH(),lev); 
      int TStep = TimeStep(u,lev);
      SwapTimeLevels(u,lev,Time,Time+TStep);
      SwapTimeLevels(work,lev,Time,Time+TStep);
    }
    return true;
  }

protected:
  int _ReadInFirst;
  double _PhysicalTime, _TimeDelta;
  file_input_type* _FileInput;
  level_transfer_type* _LevelTransfer;
};
   
#endif

---