Rythmos_ThetaStepper_decl.hpp

//@HEADER
// ***********************************************************************
//
//                           Rythmos Package
//                 Copyright (2006) Sandia Corporation
//
// Under terms of Contract DE-AC04-94AL85000, there is a non-exclusive
// license for use of this work by or on behalf of the U.S. Government.
//
// This library is free software; you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as
// published by the Free Software Foundation; either version 2.1 of the
// License, or (at your option) any later version.
//
// This library is distributed in the hope that it will be useful, but
// WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301
// USA
// Questions? Contact Todd S. Coffey (tscoffe@sandia.gov)
//
// ***********************************************************************
//@HEADER

#ifndef Rythmos_THETA_STEPPER_DECL_H
#define Rythmos_THETA_STEPPER_DECL_H

#include "Rythmos_ConfigDefs.h"
#ifdef HAVE_RYTHMOS_EXPERIMENTAL

#include "Rythmos_StepperBase.hpp"
#include "Rythmos_DataStore.hpp"
#include "Rythmos_LinearInterpolator.hpp"
#include "Rythmos_InterpolatorAcceptingObjectBase.hpp"
#include "Rythmos_InterpolatorBaseHelpers.hpp"
#include "Rythmos_SingleResidualModelEvaluator.hpp"
#include "Rythmos_SolverAcceptingStepperBase.hpp"
#include "Rythmos_StepperHelpers.hpp"

#include "Thyra_VectorBase.hpp"
#include "Thyra_ModelEvaluator.hpp"
#include "Thyra_ModelEvaluatorHelpers.hpp"
#include "Thyra_AssertOp.hpp"
#include "Thyra_NonlinearSolverBase.hpp"
#include "Thyra_TestingTools.hpp"

#include "Teuchos_VerboseObjectParameterListHelpers.hpp"
#include "Teuchos_as.hpp"


namespace {
  const std::string ThetaStepperType_name = "Theta Stepper Type";
  const std::string ThetaStepperType_default = "Implicit Euler";

  const std::string PredictorOrder_name = "Predictor Order";
  const int PredictorOrder_default = 2;
}

namespace Rythmos {
  
enum ThetaStepperType 
{
  ImplicitEuler = 0,
  Trapezoid,
  INVALID_THETA_STEPPER_TYPE
};
  

template<class Scalar>
class ThetaStepper : 
  virtual public SolverAcceptingStepperBase<Scalar>,
  virtual public InterpolatorAcceptingObjectBase<Scalar>
{
public:
  
  typedef typename Teuchos::ScalarTraits<Scalar>::magnitudeType ScalarMag;
  

  ThetaStepper();

  bool isImplicit() const;
  
  
  void setInterpolator(const RCP<InterpolatorBase<Scalar> >& interpolator);

  RCP<InterpolatorBase<Scalar> >
    getNonconstInterpolator();

  RCP<const InterpolatorBase<Scalar> >
    getInterpolator() const;
  
  RCP<InterpolatorBase<Scalar> > unSetInterpolator();



  void setSolver(
    const RCP<Thyra::NonlinearSolverBase<Scalar> > &solver
    );

  RCP<Thyra::NonlinearSolverBase<Scalar> >
  getNonconstSolver();

  RCP<const Thyra::NonlinearSolverBase<Scalar> >
  getSolver() const;


 
  bool supportsCloning() const;

  RCP<StepperBase<Scalar> > cloneStepperAlgorithm() const;

  void setModel(const RCP<const Thyra::ModelEvaluator<Scalar> >& model);

  void setNonconstModel(const RCP<Thyra::ModelEvaluator<Scalar> >& model);
  
  RCP<const Thyra::ModelEvaluator<Scalar> > getModel() const;

  RCP<Thyra::ModelEvaluator<Scalar> > getNonconstModel();

  void setInitialCondition(
    const Thyra::ModelEvaluatorBase::InArgs<Scalar> &initialCondition
    );

  Thyra::ModelEvaluatorBase::InArgs<Scalar> getInitialCondition() const;

  Scalar takeStep(Scalar dt, StepSizeType flag);
  
  const StepStatus<Scalar> getStepStatus() const;
  


  RCP<const Thyra::VectorSpaceBase<Scalar> >
  get_x_space() const;

  void addPoints(
    const Array<Scalar>& time_vec,
    const Array<RCP<const Thyra::VectorBase<Scalar> > >& x_vec,
    const Array<RCP<const Thyra::VectorBase<Scalar> > >& xdot_vec
    );
  
  TimeRange<Scalar> getTimeRange() const;
  
  void getPoints(
    const Array<Scalar>& time_vec,
    Array<RCP<const Thyra::VectorBase<Scalar> > >* x_vec,
    Array<RCP<const Thyra::VectorBase<Scalar> > >* xdot_vec,
    Array<ScalarMag>* accuracy_vec
    ) const;
  
  void getNodes(Array<Scalar>* time_vec) const;
  
  void removeNodes(Array<Scalar>& time_vec);

  int getOrder() const;

  

  void setParameterList(RCP<Teuchos::ParameterList> const& paramList);
  
  RCP<Teuchos::ParameterList> getNonconstParameterList();
  
  RCP<Teuchos::ParameterList> unsetParameterList();
  
  RCP<const Teuchos::ParameterList> getValidParameters() const;
 

  
  void describe(
    Teuchos::FancyOStream  &out,
    const Teuchos::EVerbosityLevel verbLevel
    ) const;


private:

  // ///////////////////////
  // Private date members

  bool isInitialized_;
  bool haveInitialCondition_;
  RCP<const Thyra::ModelEvaluator<Scalar> > model_;
  RCP<Thyra::NonlinearSolverBase<Scalar> > solver_;

  Thyra::ModelEvaluatorBase::InArgs<Scalar> basePoint_;

  RCP<Thyra::VectorBase<Scalar> > x_;
  RCP<Thyra::VectorBase<Scalar> > x_old_;
  RCP<Thyra::VectorBase<Scalar> > x_pre_;

  RCP<Thyra::VectorBase<Scalar> > x_dot_;
  RCP<Thyra::VectorBase<Scalar> > x_dot_old_;
  RCP<Thyra::VectorBase<Scalar> > x_dot_really_old_;
  RCP<Thyra::VectorBase<Scalar> > x_dot_base_;

  Scalar t_;
  Scalar t_old_;

  Scalar dt_;
  Scalar dt_old_;
  int numSteps_;

  ThetaStepperType thetaStepperType_;
  Scalar theta_;
  int predictor_corrector_begin_after_step_;
  int default_predictor_order_;

  RCP<Rythmos::SingleResidualModelEvaluator<Scalar> >  neModel_;

  RCP<Teuchos::ParameterList> parameterList_;

  RCP<InterpolatorBase<Scalar> > interpolator_;


  // //////////////////////////
  // Private member functions

  void defaultInitializeAll_();
  void initialize_();
  void obtainPredictor_();
};


template<class Scalar>
RCP<ThetaStepper<Scalar> >
thetaStepper(
  const RCP<Thyra::ModelEvaluator<Scalar> >& model,
  const RCP<Thyra::NonlinearSolverBase<Scalar> >& solver,
  RCP<Teuchos::ParameterList>& parameterList
  );

} // namespace Rythmos

#endif // HAVE_RYTHMOS_EXPERIMENTAL

#endif //Rythmos_THETA_STEPPER_DECL_H