amroc/amr/F77Interfaces/F77FileInitialCondition.h

Go to the documentation of this file.

// -*- C++ -*-

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

#ifndef AMROC_F77_FILE_INITIALCONDITION_H
#define AMROC_F77_FILE_INITIALCONDITION_H

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

template <class VectorType, int dim>
class F77FileInitialCondition : public F77InitialCondition<VectorType,dim> {
  typedef F77InitialCondition<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; 

  F77FileInitialCondition(generic_func_type initial) : base(initial), _ReadInFirst(0) {
    _FileInput = (file_input_type*) 0;
    _LevelTransfer = (level_transfer_type*) 0;
  }
  
  F77FileInitialCondition(generic_func_type initial, generic_func_type in,
                          generic_func_type prolong) : base(initial), _ReadInFirst(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)));
  }
  
  F77FileInitialCondition(generic_func_type initial, generic_func_type out, 
                          generic_func_type in, generic_func_type prolong) : 
  base(initial), _ReadInFirst(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)));
  }
  
  F77FileInitialCondition(generic_func_type initial, generic_func_type out, 
                          generic_func_type in, generic_func_type prolong, 
                          generic_func_type restrct) : 
  base(initial), _ReadInFirst(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)));
    }   
  }
  
  F77FileInitialCondition(generic_func_type initial, generic_func_type out, 
                          generic_func_type in, generic_func_type prolong, 
                          generic_func_type restrct, generic_func_type update) : 
  base(initial), _ReadInFirst(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 ~F77FileInitialCondition() {
    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);
  }

  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 void Set(vec_grid_fct_type& u, grid_fct_type& work, const int Level) { 
    SetPhysicalTime(u,0,Level,0.0);
   
    if (_FileInput && _ReadInFirst>0 && !_FileInput->GetOutFunc()) 
      _FileInput->ReadIn(u,work);

    forall (u,0,Level,c)
      base::SetGrid(u(0,Level,c),work(0,Level,c),Level);    
    end_forall

    if (_FileInput && _ReadInFirst==0) {
      BoundaryUpdate(u,0,Level);
      _FileInput->ReadIn(u,work);
    }

    BoundaryUpdate(u,0,Level);
  }

protected:
  int _ReadInFirst;
  file_input_type* _FileInput;
  level_transfer_type* _LevelTransfer;
};
   
#endif

---