lang_latex_default_var_name_gen.hpp

#ifndef CPPAD_CG_LANG_LATEX_DEFAULT_VAR_NAME_GEN_INCLUDED
#define CPPAD_CG_LANG_LATEX_DEFAULT_VAR_NAME_GEN_INCLUDED
/* --------------------------------------------------------------------------
 *  CppADCodeGen: C++ Algorithmic Differentiation with Source Code Generation:
 *    Copyright (C) 2014 Ciengis
 *    Copyright (C) 2018 Joao Leal
 *
 *  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 Base>
class LangLatexDefaultVariableNameGenerator : public VariableNameGenerator<Base> {
protected:
    // auxiliary string stream
    std::stringstream _ss;
    // array name of the dependent variables
    std::string _depName;
    // array name of the independent variables
    std::string _indepName;
    // array name of the temporary variables
    std::string _tmpName;
    // array name of the temporary array variables
    std::string _tmpArrayName;
    // sparse array name of the temporary array variables
    std::string _tmpSparseArrayName;
    // the lowest variable ID used for the temporary variables
    size_t _minTemporaryID;
    // the highest variable ID used for the temporary variables
    size_t _maxTemporaryID;
    // the highest ID used for the temporary array variables
    size_t _maxTemporaryArrayID;
    // the highest ID used for the temporary sparse array variables
    size_t _maxTemporarySparseArrayID;
public:

    inline LangLatexDefaultVariableNameGenerator(std::string depName = "y",
                                                 std::string indepName = "x",
                                                 std::string tmpName = "v",
                                                 std::string tmpArrayName = "\\mathbf{a}",
                                                 std::string tmpSparseArrayName = "\\mathbf{s}") :
        _depName(std::move(depName)),
        _indepName(std::move(indepName)),
        _tmpName(std::move(tmpName)),
        _tmpArrayName(std::move(tmpArrayName)),
        _tmpSparseArrayName(std::move(tmpSparseArrayName)),
        _minTemporaryID(0), // not really required (but it avoids warnings)
        _maxTemporaryID(0), // not really required (but it avoids warnings)
        _maxTemporaryArrayID(0), // not really required (but it avoids warnings)
        _maxTemporarySparseArrayID(0) { // not really required (but it avoids warnings)

        this->_independent.push_back(FuncArgument(_indepName));
        this->_dependent.push_back(FuncArgument(_depName));
        this->_temporary.push_back(FuncArgument(_tmpName));
        this->_temporary.push_back(FuncArgument(_tmpArrayName));
        this->_temporary.push_back(FuncArgument(_tmpSparseArrayName));
    }

    inline virtual ~LangLatexDefaultVariableNameGenerator() = default;

    inline size_t getMinTemporaryVariableID() const override {
        return _minTemporaryID;
    }

    inline size_t getMaxTemporaryVariableID() const override {
        return _maxTemporaryID;
    }

    inline size_t getMaxTemporaryArrayVariableID() const override {
        return _maxTemporaryArrayID;
    }

    size_t getMaxTemporarySparseArrayVariableID() const override {
        return _maxTemporarySparseArrayID;
    }

    inline std::string generateDependent(size_t index) override {
        _ss.clear();
        _ss.str("");

        _ss << _depName;
        latexIndex(_ss, index);

        return _ss.str();
    }

    inline std::string generateIndependent(const OperationNode<Base>& independent,
                                           size_t id) override {
        _ss.clear();
        _ss.str("");

        _ss << _indepName;
        latexIndex(_ss, id - 1);

        return _ss.str();
    }

    inline std::string generateTemporary(const OperationNode<Base>& variable,
                                         size_t id) override {
        _ss.clear();
        _ss.str("");

        if (this->_temporary[0].array) {
            _ss << _tmpName;
            latexIndex(_ss, id - this->_minTemporaryID);
        } else {
            _ss << _tmpName << id;
            latexIndex(_ss, id);
        }

        return _ss.str();
    }

    std::string generateTemporaryArray(const OperationNode<Base>& variable,
                                       size_t id) override {
        _ss.clear();
        _ss.str("");

        CPPADCG_ASSERT_UNKNOWN(variable.getOperationType() == CGOpCode::ArrayCreation)

        _ss << "\\&" << _tmpArrayName; 
        latexIndex(_ss, id - 1);

        return _ss.str();
    }

    std::string generateTemporarySparseArray(const OperationNode<Base>& variable,
                                             size_t id) override {
        _ss.clear();
        _ss.str("");

        CPPADCG_ASSERT_UNKNOWN(variable.getOperationType() == CGOpCode::SparseArrayCreation)

        _ss << "\\&" << _tmpSparseArrayName; 
        latexIndex(_ss, id - 1);

        return _ss.str();
    }

    std::string generateIndexedDependent(const OperationNode<Base>& var,
                                         size_t id,
                                         const IndexPattern& ip) override {
        CPPADCG_ASSERT_KNOWN(var.getOperationType() == CGOpCode::LoopIndexedDep, "Invalid node type")
        CPPADCG_ASSERT_KNOWN(!var.getArguments().empty(), "Invalid number of arguments")

        _ss.clear();
        _ss.str("");

        std::string index = LanguageLatex<Base>::indexPattern2String(ip, getIndexes(var, 1));
        _ss << _depName << "_";
        if (index.size() > 1)
            _ss << "{" << index << "}";
        else
            _ss << index;

        return _ss.str();
    }

    std::string generateIndexedIndependent(const OperationNode<Base>& independent,
                                           size_t id,
                                           const IndexPattern& ip) override {
        CPPADCG_ASSERT_KNOWN(independent.getOperationType() == CGOpCode::LoopIndexedIndep, "Invalid node type")
        CPPADCG_ASSERT_KNOWN(independent.getArguments().size() > 0, "Invalid number of arguments")

        _ss.clear();
        _ss.str("");

        std::string index = LanguageLatex<Base>::indexPattern2String(ip, getIndexes(independent));
        _ss << _indepName << "_";
        if (index.size() > 1)
            _ss << "{" << index << "}";
        else
            _ss << index;

        return _ss.str();
    }

    inline void setTemporaryVariableID(size_t minTempID,
                                       size_t maxTempID,
                                       size_t maxTempArrayID,
                                       size_t maxTempSparseArrayID) override {
        _minTemporaryID = minTempID;
        _maxTemporaryID = maxTempID;
        _maxTemporaryArrayID = maxTempArrayID;
        _maxTemporarySparseArrayID = maxTempSparseArrayID;

        // if
        //  _minTemporaryID == _maxTemporaryID + 1
        // then no temporary variables are being used
        CPPADCG_ASSERT_UNKNOWN(_minTemporaryID <= _maxTemporaryID + 1)
    }

    const std::string& getIndependentArrayName(const OperationNode<Base>& indep,
                                               size_t id) override {
        return _indepName;
    }

    size_t getIndependentArrayIndex(const OperationNode<Base>& indep,
                                    size_t id) override {
        return id - 1;
    }

    bool isConsecutiveInIndepArray(const OperationNode<Base>& indepFirst,
                                   size_t idFirst,
                                   const OperationNode<Base>& indepSecond,
                                   size_t idSecond) override {
        return idFirst + 1 == idSecond;
    }

    bool isInSameIndependentArray(const OperationNode<Base>& indep1,
                                  size_t id1,
                                  const OperationNode<Base>& indep2,
                                  size_t id2) override {
        return true;
    }

    const std::string& getTemporaryVarArrayName(const OperationNode<Base>& var,
                                                size_t id) override {
        return _tmpName;
    }

    size_t getTemporaryVarArrayIndex(const OperationNode<Base>& var,
                                     size_t id) override {
        return id - this->_minTemporaryID;
    }

    bool isConsecutiveInTemporaryVarArray(const OperationNode<Base>& varFirst,
                                          size_t idFirst,
                                          const OperationNode<Base>& varSecond,
                                          size_t idSecond) override {
        return idFirst + 1 == idSecond;
    }

    bool isInSameTemporaryVarArray(const OperationNode<Base>& var1,
                                   size_t id1,
                                   const OperationNode<Base>& var2,
                                   size_t id2) override {
        return true;
    }

protected:

    static inline std::stringstream& latexIndex(std::stringstream& ss,
                                                size_t index) {
        ss << "_";
        if (index < 10)
            ss << index;
        else
            ss << "{" << index << "}";

        return ss;
    }

    static inline std::vector<const OperationNode<Base>*> getIndexes(const OperationNode<Base>& var,
                                                                     size_t offset = 0) {
        const std::vector<Argument<Base> >& args = var.getArguments();
        std::vector<const OperationNode<Base>*> indexes(args.size() - offset);

        for (size_t a = offset; a < args.size(); a++) {
            CPPADCG_ASSERT_KNOWN(args[a].getOperation() != nullptr, "Invalid argument")
            CPPADCG_ASSERT_KNOWN(args[a].getOperation()->getOperationType() == CGOpCode::Index, "Invalid argument")

            indexes[a - offset] = &static_cast<const IndexOperationNode<Base>*> (args[a].getOperation())->getIndex();
        }

        return indexes;
    }
};

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

#endif