NEvaluator.hh

Go to the documentation of this file.

 
#pragma once
 
#include "Storage.hh"
#include "NTensor.hh"
 
namespace cadabra {
 
 
    class NEvaluator {
        public:
            NEvaluator(const Ex&);
 
            std::map<Ex::iterator, NTensor, Ex::iterator_base_less> subtree_values;
 
            std::map<Ex::iterator, Ex::iterator, Ex::iterator_base_less> subtree_equalities;
 
            class VariableValues {
                public:
                    Ex                        variable;
                    NTensor                   values;
                    std::vector<Ex::iterator> locations;
            };
            std::vector<VariableValues> variable_values;
 
            void    set_variable(const Ex&, const NTensor& val);
 
            NTensor evaluate();
 
 
            void find_variable_locations();
 
        private:
            const Ex& ex;
    };
 
};