index_assign_operation_node.hpp

#ifndef CPPAD_CG_INDEX_ASSIGN_OPERATION_NODE_INCLUDED
#define CPPAD_CG_INDEX_ASSIGN_OPERATION_NODE_INCLUDED
/* --------------------------------------------------------------------------
 *  CppADCodeGen: C++ Algorithmic Differentiation with Source Code Generation:
 *    Copyright (C) 2013 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 Base>
class IndexAssignOperationNode : public OperationNode<Base> {
    friend class CodeHandler<Base>;
private:
    IndexPattern& indexPattern_;
public:

    inline OperationNode<Base>& getIndex() const {
        const std::vector<Argument<Base> >& args = this->getArguments();
        CPPADCG_ASSERT_KNOWN(!args.empty(), "Invalid number of arguments");

        OperationNode<Base>* aNode = args[0].getOperation();
        CPPADCG_ASSERT_KNOWN(aNode != nullptr && aNode->getOperationType() == CGOpCode::IndexDeclaration, "Invalid argument operation type");

        return static_cast<OperationNode<Base>&> (*aNode);
    }

    inline const IndexPattern& getIndexPattern() const {
        return indexPattern_;
    }

    inline IndexPattern& getIndexPattern() {
        return indexPattern_;
    }

    inline std::vector<const OperationNode<Base>*> getIndexPatternIndexes() const {
        std::vector<const OperationNode<Base>*> iargs;

        const std::vector<Argument<Base> >& args = this->getArguments();

        CPPADCG_ASSERT_KNOWN(args[1].getOperation() != nullptr &&
                             args[1].getOperation()->getOperationType() == CGOpCode::Index, "Invalid argument operation type");
        iargs.push_back(&static_cast<IndexOperationNode<Base>*> (args[1].getOperation())->getIndex());

        if (args.size() > 2) {
            CPPADCG_ASSERT_KNOWN(args[2].getOperation() != nullptr &&
                                 args[2].getOperation()->getOperationType() == CGOpCode::Index, "Invalid argument operation type");
            iargs.push_back(&static_cast<IndexOperationNode<Base>*> (args[2].getOperation())->getIndex());
        }

        return iargs;
    }

    inline virtual ~IndexAssignOperationNode() {
    }

protected:

    inline IndexAssignOperationNode(CodeHandler<Base>* handler,
                                    OperationNode<Base>& index,
                                    IndexPattern& indexPattern,
                                    IndexOperationNode<Base>& index1) :
        OperationNode<Base>(handler, CGOpCode::IndexAssign,{index, index1}),
        indexPattern_(indexPattern) {
    }

    inline IndexAssignOperationNode(CodeHandler<Base>* handler,
                                    OperationNode<Base>& index,
                                    IndexPattern& indexPattern,
                                    IndexOperationNode<Base>* index1,
                                    IndexOperationNode<Base>* index2) :
        OperationNode<Base>(handler, CGOpCode::IndexAssign, std::vector<size_t> (0), createArguments(index, index1, index2)),
        indexPattern_(indexPattern) {
    }

    inline static std::vector<Argument<Base> > createArguments(OperationNode<Base>& index,
                                                               IndexOperationNode<Base>* index1,
                                                               IndexOperationNode<Base>* index2) {
        std::vector<Argument<Base> > args(1 + (index1 != nullptr)+ (index2 != nullptr));

        args[0] = index;
        if (index1 != nullptr)
            args[1] = *index1;
        if (index2 != nullptr) {
            args.back() = *index2;
        }

        return args;
    }

};

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

#endif