ACModule.hpp

#ifndef ACMODULE_INCLUDED
#define ACMODULE_INCLUDED

#include<stack>
#include<vector>
using namespace std;

namespace ACDesignPatterns {

template<typename T>
class VarRefs {
    vector<T*> v;
  public:
    size_t register_var(T& z) {
      v.push_back(&z); return v.size()-1;
    }
    void clear() {
      v.clear();
    }
    T& operator[](size_t i) {
      return *v[i];
    }
    size_t size() { return v.size(); }
};

template<typename T>
class ACModule {
  protected:
    stack<vector<T> > argument_checkpoint;
  public:
    virtual ~ACModule() {};
    VarRefs<T> x,ax,y,ay;
    void register_input(T& v, T& a) {
      x.register_var(v); ax.register_var(a); 
    } 
    void register_input(VarRefs<T>& v, VarRefs<T>& a) {
      x=v; ax=a; 
    } 
    void register_output(T& v, T& a) {
      y.register_var(v); ay.register_var(a); 
    } 
    void register_output(VarRefs<T>& v, VarRefs<T>& a) {
      y=v; ay=a; 
    } 
    void register_inoutput(T& xv, T& xa, T& yv, T& ya) {
      x.register_var(xv); ax.register_var(xa); 
      y.register_var(yv); ay.register_var(ya); 
    }
    void register_inoutput(VarRefs<T>& xv, VarRefs<T>& xa, VarRefs<T>& yv, VarRefs<T>& ya) {
      x=xv; ax=xa; y=yv; ay=ya; 
    } 
    void reset_inoutput() { x.clear(); ax.clear(); y.clear(); ay.clear(); }
    virtual void push_arguments() { 
      vector<T> a(x.size());
      for (size_t i=0;i<x.size();i++) a[i]=x[i];
      argument_checkpoint.push(a); 
    }
    virtual void read_arguments() { 
      vector<T> a(x.size()); a=argument_checkpoint.top(); 
      for (size_t i=0;i<x.size();i++) x[i]=a[i];
    }
    virtual void pop_arguments() { 
      argument_checkpoint.pop(); 
    }
    virtual void evaluate_primal()=0;
    virtual void evaluate_joint_primal() {
      push_arguments();
      evaluate_primal();
    }
    virtual void evaluate_split_primal()=0;
    virtual void evaluate_split_adjoint()=0;
    virtual void evaluate_joint_adjoint() {
      read_arguments(); pop_arguments();
      evaluate_split_primal();
      evaluate_split_adjoint();
    }
};

}

#endif