Rythmos_CompositeIntegratorObserver.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., 59 Temple Place, Suite 330, Boston, MA 02111-1307
// USA
// Questions? Contact Todd S. Coffey (tscoffe@sandia.gov)
//
// ***********************************************************************
//@HEADER

#ifndef RYTHMOS_COMPOSITE_INTEGRATOR_OBSERVER_HPP
#define RYTHMOS_COMPOSITE_INTEGRATOR_OBSERVER_HPP


#include "Rythmos_IntegrationObserverBase.hpp"
#include "Teuchos_as.hpp"


namespace Rythmos {


template<class Scalar>
class CompositeIntegratorObserver
  : public IntegrationObserverBase<Scalar>
{
public:


  CompositeIntegratorObserver();

  void addObserver(
    const RCP<IntegrationObserverBase<Scalar> > &observer
    );

  // ToDo: Add functions to add observers



  virtual RCP<IntegrationObserverBase<Scalar> >
  cloneIntegrationObserver() const;

  virtual void resetIntegrationObserver(
    const TimeRange<Scalar> &integrationTimeDomain
    );

  virtual void observeCompletedTimeStep(
    const StepperBase<Scalar> &stepper,
    const StepControlInfo<Scalar> &stepCtrlInfo,
    const int timeStepIter
    );


private:

  Array<RCP<IntegrationObserverBase<Scalar> > > observers_;

};


template<class Scalar>
RCP<CompositeIntegratorObserver<Scalar> > compositeIntegratorObserver()
{
  RCP<CompositeIntegratorObserver<Scalar> >
    frsco(new CompositeIntegratorObserver<Scalar>());
  return frsco;
}


//
// Implementations
//


// Constructors/Initializers/Accessors


template<class Scalar>
CompositeIntegratorObserver<Scalar>::CompositeIntegratorObserver()
{}


template<class Scalar>
void CompositeIntegratorObserver<Scalar>::addObserver(
  const RCP<IntegrationObserverBase<Scalar> > &observer
  )
{
#ifdef RYTHMOS_DEBUG
  TEST_FOR_EXCEPT(is_null(observer));
#endif
  observers_.push_back(observer);
}


// Overridden from IntegrationObserverBase


template<class Scalar>
RCP<IntegrationObserverBase<Scalar> >
CompositeIntegratorObserver<Scalar>::cloneIntegrationObserver() const
{
  using Teuchos::as;
  RCP<CompositeIntegratorObserver<Scalar> >
    compositeObserver = compositeIntegratorObserver<Scalar>();
  for (int i = 0; i < as<int>(observers_.size()); ++i ) {
    compositeObserver->addObserver(observers_[i]->cloneIntegrationObserver());
  }
  TEST_FOR_EXCEPTION( true, std::logic_error,
    "Error, I have not tested this function yet so look over this very carefully before you"
    " remove this this macro call!" );
  return compositeObserver;
}


template<class Scalar>
void CompositeIntegratorObserver<Scalar>::resetIntegrationObserver(
  const TimeRange<Scalar> &integrationTimeDomain
  )
{
  using Teuchos::as;
  for (int i = 0; i < as<int>(observers_.size()); ++i ) {
    observers_[i]->resetIntegrationObserver(integrationTimeDomain);
  }
}


template<class Scalar>
void CompositeIntegratorObserver<Scalar>::observeCompletedTimeStep(
  const StepperBase<Scalar> &stepper,
  const StepControlInfo<Scalar> &stepCtrlInfo,
  const int timeStepIter
  )
{
  using Teuchos::as;

  const RCP<FancyOStream> out = this->getOStream();
  const Teuchos::EVerbosityLevel verbLevel = this->getVerbLevel();

  for (int i = 0; i < as<int>(observers_.size()); ++i ) {
    RCP<IntegrationObserverBase<Scalar> > observer = observers_[i];
    observer->setOStream(out);
    observer->setVerbLevel(verbLevel);
    observer->observeCompletedTimeStep(stepper,stepCtrlInfo,timeStepIter);
  }
}


} // namespace Rythmos


#endif //RYTHMOS_COMPOSITE_INTEGRATOR_OBSERVER_HPP

---