MyACEnsemblePathwiseEvolutionRecursiveBisection.hpp

#include<cmath>
using namespace std;

#include "ACEnsemblePathwise.hpp"
#include "ACEvolutionRecursiveBisection.hpp"
using namespace ACDesignPatterns;

template<typename T>
class MyACEvolutionStepUserData {
    T p;
  public:
    T& p_ref() { return p; }
};

template<typename T>
class MyACEvolutionStep : public ACModule<T> {
    stack<T> required_T;
    using ACModule<T>::x;
    using ACModule<T>::ax;
    using ACModule<T>::y;
    using ACModule<T>::ay;
  public:
    MyACEvolutionStepUserData<T> &d;
    MyACEvolutionStep(MyACEvolutionStepUserData<T> &d) : d(d) { } 
    void evaluate_primal() { 
      y[0]=sin(x[0]*d.p_ref()); 
    }
    void evaluate_split_primal() { 
      if (&x[0]==&y[0]) required_T.push(x[0]);  
      y[0]=sin(x[0]*d.p_ref()); 
    }
    void evaluate_split_adjoint() { 
      if (&x[0]==&y[0]) {
        x[0]=required_T.top(); required_T.pop(); 
        ax[0]*=d.p_ref()*cos(x[0]*d.p_ref()); 
      } else 
        ax[0]+=d.p_ref()*cos(x[0]*d.p_ref())*ay[0]; 
    }
};

template<typename T>
class MyACEvolutionRecomputeAll : public ACEvolutionRecomputeAll<T> {
    using ACEvolutionRecomputeAll<T>::step;
  public:
    MyACEvolutionRecomputeAll(size_t nsteps, ACModule<T>* s) 
        : ACEvolutionRecomputeAll<T>(nsteps) {
      step=s;
    }
};

template<typename T>
class MyACEvolutionRecursiveBisection : public ACEvolutionRecursiveBisection<T> {
  public:
    using ACEvolution<T>::step;
    using ACEvolutionRecursiveBisection<T>::m;
    MyACEvolutionRecursiveBisection(size_t nsteps, size_t ncheckpoints, MyACEvolutionStepUserData<T>& d) : ACEvolutionRecursiveBisection<T>(nsteps,ncheckpoints) {
      step=new MyACEvolutionStep<T>(d);
      m=new MyACEvolutionRecomputeAll<T>(nsteps,step);
    }
    ~MyACEvolutionRecursiveBisection() { delete step; delete m; }
};

template<typename T>
class MyACEnsemblePathwiseEvolutionRecursiveBisection : public ACEnsemblePathwise<T> {
  public:
    using ACEnsemblePathwise<T>::path;
    using ACModule<T>::x;
    using ACModule<T>::ax;
    using ACModule<T>::y;
    using ACModule<T>::ay;
    using ACModule<T>::push_arguments;
    using ACModule<T>::read_arguments;
    using ACModule<T>::pop_arguments;
    MyACEnsemblePathwiseEvolutionRecursiveBisection(size_t npaths, size_t nsteps, size_t ncheckpoints, MyACEvolutionStepUserData<T>& d) : ACEnsemblePathwise<T>(npaths) { 
      path=new MyACEvolutionRecursiveBisection<T>(nsteps,ncheckpoints,d);
    }
    ~MyACEnsemblePathwiseEvolutionRecursiveBisection() { delete path; }
};