amroc/amr/AMRCoupledGFMSolver.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_COUPLED_GFMSOLVER_H
#define AMROC_COUPLED_GFMSOLVER_H

#include "AMRGFMSolver.h"

template <class VectorType, class FixupType, class FlagType, int dim>
class AMRCoupledGFMSolver : public AMRGFMSolver<VectorType,FixupType,FlagType,dim> {
  typedef typename VectorType::InternalDataType DataType;
  typedef AMRGFMSolver<VectorType,FixupType,FlagType,dim> base;

public:
  typedef typename base::integrator_type integrator_type;
  typedef typename base::initial_condition_type initial_condition_type;
  typedef typename base::boundary_conditions_type boundary_conditions_type;
  typedef typename base::vec_grid_fct_type vec_grid_fct_type;  

  AMRCoupledGFMSolver(integrator_type& integ, 
                      initial_condition_type& init,
                      boundary_conditions_type& bc) : base(integ, init, bc), 
                                                      CouplingLevel(0), CouplingStrategy(0), 
                                                      _SubSteps(1) {}   
  virtual ~AMRCoupledGFMSolver() {}

  //******************************************************************************
  // Abstract class interface
  //******************************************************************************
  virtual void SendBoundaryData() = 0;  
  virtual void PostReceiveBoundaryData(bool FullDomain=false) = 0;  
  virtual void WaitReceiveBoundaryData() = 0;  
  //******************************************************************************

  virtual void register_at(ControlDevice& Ctrl, const std::string& prefix) {
    base::register_at(Ctrl,prefix);
    RegisterAt(base::LocCtrl,"CouplingLevel",CouplingLevel);
    RegisterAt(base::LocCtrl,"CouplingStrategy",CouplingStrategy);
  } 
  virtual void register_at(ControlDevice& Ctrl) {
    base::register_at(Ctrl);
  }

  virtual bool setup() {
    bool ret = base::setup();
    if (CouplingLevel>base::MaxLev-1) 
      CouplingLevel=base::MaxLev-1;    
    _SubSteps = 1;
    for (register int n=0; n<CouplingLevel; n++)
      _SubSteps *= base::_RefineFactor[n];
    return ret;
  }

  virtual void SetupData() {
    base::SetupData();
#ifdef DEBUG_PRINT_AMRSOLVER
    ( comm_service::log() <<  "*** base::SetupData() finished.\n" ).flush();
#endif 
  }

  virtual void Initialize(double& t, double &dt) {
#ifdef DEBUG_PRINT_AMRSOLVER
    ( comm_service::log() <<  "*** AMRCoupledGFMSolver::Initialize().\n" ).flush();
#endif 
    if (CouplingLevel>=0) {
      PostReceiveBoundaryData(true);
#ifdef DEBUG_PRINT_AMRSOLVER
      ( comm_service::log() <<  "*** PostReceiveBoundaryData() done.\n" ).flush();
#endif 
      WaitReceiveBoundaryData();
#ifdef DEBUG_PRINT_AMRSOLVER
      ( comm_service::log() <<  "*** WaitReceiveBoundaryData() done.\n" ).flush();
#endif 
    }
    base::Initialize(t,dt);
#ifdef DEBUG_PRINT_AMRSOLVER
    ( comm_service::log() <<  "*** base::Initialize() finished.\n" ).flush();
#endif 
    if (CouplingLevel>=0) 
      SendBoundaryData();
  }

  virtual void Restart(double& t, double& dt) {
#ifdef DEBUG_PRINT_AMRSOLVER
    ( comm_service::log() <<  "*** AMRCoupledGFMSolver::Restart().\n" ).flush();
#endif 
    if (CouplingLevel>=0) {
      PostReceiveBoundaryData(true);
#ifdef DEBUG_PRINT_AMRSOLVER
      ( comm_service::log() <<  "*** PostReceiveBoundaryData() done.\n" ).flush();
#endif 
      WaitReceiveBoundaryData();
#ifdef DEBUG_PRINT_AMRSOLVER
      ( comm_service::log() <<  "*** WaitReceiveBoundaryData() done.\n" ).flush();
#endif 
    }
    base::Restart(t,dt);                
#ifdef DEBUG_PRINT_AMRSOLVER
    ( comm_service::log() <<  "*** base::Restart() finished.\n" ).flush();
#endif 
    if (CouplingLevel>=0) 
      SendBoundaryData();
  }
  
  virtual void Advance(double& t, double &dt) {
#ifdef DEBUG_PRINT_AMRSOLVER
    ( comm_service::log() <<  "*** AMRCoupledGFMSolver::Advance().\n" ).flush();
#endif 
    if (CouplingLevel>=0) {
      PostReceiveBoundaryData(true);
#ifdef DEBUG_PRINT_AMRSOLVER
      ( comm_service::log() <<  "*** PostReceiveBoundaryData() done.\n" ).flush();
#endif 
      WaitReceiveBoundaryData();
#ifdef DEBUG_PRINT_AMRSOLVER
      ( comm_service::log() <<  "*** WaitReceiveBoundaryData() done.\n" ).flush();
#endif 
    }
    base::Advance(t,dt);
#ifdef DEBUG_PRINT_AMRSOLVER
    ( comm_service::log() <<  "*** base::Advance() finished.\n" ).flush();
#endif 
  }

  virtual void BeforeLevelStep(const int Level) {
    if (Level!=CouplingLevel) return;
    if (CouplingStrategy>0) {
      SendBoundaryData();
      if (CouplingLevel>0 && CurrentTime(base::GH(),Level)+TimeStep(base::U(),Level)<
          CurrentTime(base::GH(),0)+TimeStep(base::U(),0)) {
        if (base::RedistributeEvery>0)
          PostReceiveBoundaryData(false);
        else
          PostReceiveBoundaryData(true);
#ifdef DEBUG_PRINT_AMRSOLVER
        ( comm_service::log() <<  "*** PostReceiveBoundaryData() on level " 
          << Level << " done.\n" ).flush();
#endif 
      }
    }
  }  

  virtual void AfterLevelStep(const int Level) {
    if (Level!=CouplingLevel) return;
    if (CouplingStrategy<=0) SendBoundaryData();
    if (CouplingLevel>0 && CurrentTime(base::GH(),Level)<
        CurrentTime(base::GH(),0)+TimeStep(base::U(),0)) {
      if (CouplingStrategy<=0) {
        if (base::RedistributeEvery>0)
          PostReceiveBoundaryData(false);
        else
          PostReceiveBoundaryData(true);
#ifdef DEBUG_PRINT_AMRSOLVER
        ( comm_service::log() <<  "*** PostReceiveBoundaryData() on level " 
          << Level << " done.\n" ).flush();
#endif 
      }
      WaitReceiveBoundaryData();
#ifdef DEBUG_PRINT_AMRSOLVER
      ( comm_service::log() <<  "*** WaitReceiveBoundaryData() on level " 
        << Level << " done.\n" ).flush();
#endif 
    }
  }  

  const int& SubSteps() const { return _SubSteps; }

protected:
  int CouplingLevel, CouplingStrategy, _SubSteps;
};


#endif

---