ACEvolutionRecomputeAll.hpp

#ifndef ACEVOLUTION_RECOMPUTE_ALL_INCLUDED
#define ACEVOLUTION_RECOMPUTE_ALL_INCLUDED


#include "ACEvolution.hpp"

namespace ACDesignPatterns {

template<typename T>
class ACEvolutionRecomputeAll : public ACEvolution<T> {
  protected:
    using ACEvolution<T>::step;
  public:
    using ACEvolution<T>::nsteps;
    using ACEvolution<T>::evaluate_primal;
    using ACModule<T>::push_arguments;
    using ACModule<T>::read_arguments;
    using ACModule<T>::pop_arguments;
    using ACModule<T>::x;
    using ACModule<T>::ax;
    using ACModule<T>::y;
    using ACModule<T>::ay;
    ACEvolutionRecomputeAll(size_t nsteps) : ACEvolution<T>(nsteps) {};
    void evaluate_primal() {
      step->register_inoutput(x,ax,y,ay);
      ACEvolution<T>::evaluate_primal();
    }
    void evaluate_split_primal() {
      step->register_inoutput(x,ax,y,ay);
      push_arguments();
      ACEvolution<T>::evaluate_primal();
    }
    void evaluate_split_adjoint() {
      step->register_inoutput(x,ax,y,ay);
      for (size_t t=nsteps;t>0;t--) {
        read_arguments();
        for (size_t i=0;i<t-1;i++)
          step->evaluate_primal();
        step->evaluate_split_primal();
        step->evaluate_split_adjoint();
      }
      pop_arguments();
    }
};

}

#endif