evaluator_adcg.hpp

#ifndef CPPAD_CG_EVALUATOR_ADCG_INCLUDED
#define CPPAD_CG_EVALUATOR_ADCG_INCLUDED
/* --------------------------------------------------------------------------
 *  CppADCodeGen: C++ Algorithmic Differentiation with Source Code Generation:
 *    Copyright (C) 2016 Ciengis
 *
 *  CppADCodeGen is distributed under multiple licenses:
 *
 *   - Eclipse Public License Version 1.0 (EPL1), and
 *   - GNU General Public License Version 3 (GPL3).
 *
 *  EPL1 terms and conditions can be found in the file "epl-v10.txt", while
 *  terms and conditions for the GPL3 can be found in the file "gpl3.txt".
 * ----------------------------------------------------------------------------
 * Author: Joao Leal
 */

namespace CppAD {
namespace cg {

template<class ScalarIn, class BaseOut>
class Evaluator<ScalarIn, CG<BaseOut>, CppAD::AD<CG<BaseOut> > > : public EvaluatorAD<ScalarIn, CG<BaseOut>, Evaluator<ScalarIn, CG<BaseOut>, CppAD::AD<CG<BaseOut> > > > {
    friend EvaluatorBase<ScalarIn, CG<BaseOut>, CppAD::AD<CG<BaseOut> >, Evaluator<ScalarIn, CG<BaseOut>, CppAD::AD<CG<BaseOut> > > >;
public:
    using ScalarOut = CG<BaseOut>;
    using ActiveOut = CppAD::AD<ScalarOut>;
    using Super = EvaluatorAD<ScalarIn, ScalarOut, Evaluator<ScalarIn, CG<BaseOut>, CppAD::AD<CG<BaseOut> > > >;
protected:
    using Super::evalsAtomic_;
    using Super::atomicFunctions_;
    using Super::handler_;
    using Super::evalArrayCreationOperation;
protected:
    bool printFor_;
    ActiveOut printForPos_;
    bool adcgName_;
public:

    inline Evaluator(CodeHandler<ScalarIn>& handler) :
        Super(handler),
        printFor_(false),
        printForPos_(0),
        adcgName_(true) {
    }

    inline virtual ~Evaluator() {
    }

    inline void setPrintFor(bool printFor) {
        printFor_ = printFor;
    }

    inline bool isPrintFor() const {
        return printFor_;
    }

    inline void setPrintForPos(const ActiveOut& pos) {
        printForPos_ = pos;
    }

    inline const ActiveOut& getPrintForPos() const {
        return printForPos_;
    }

    inline void setCopyAdCgName(bool adcgName) {
        adcgName_ = adcgName;
    }

    inline bool isCopyAdCgName() const {
        return adcgName_;
    }

protected:

    void processActiveOut(const OperationNode<ScalarIn>& node,
                          ActiveOut& a) {
        if (node.getName() != nullptr) {
            if(adcgName_ && CppAD::Variable(a)) {
                ScalarOut a2(CppAD::Value(CppAD::Var2Par(a)));
                if (a2.getOperationNode() != nullptr) {
                    a2.getOperationNode()->setName(*node.getName());
                }
            }

            if (printFor_) {
                CppAD::PrintFor(printForPos_, "", a, node.getName()->c_str());
            }
        }
    }

};

} // END cg namespace
} // END CppAD namespace

#endif