NEvaluator.hh

Go to the documentation of this file.

 
#pragma once
 
#include "Storage.hh"
#include "NTensor.hh"
#include "Compare.hh"
#include <functional>
 
namespace cadabra {
 
 
    class NEvaluator {
        public:
            NEvaluator();
            NEvaluator(Ex::iterator);
 
            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;
            std::map<Ex, size_t, tree_exact_less_no_wildcards_obj> variable_values_locs;
 
            void    set_function(Ex::iterator);
            
            void    set_variable(const Ex&, const NTensor& val);
 
            typedef std::function<std::complex<double>(const Ex&)> lookup_function_t;
            void    set_lookup_function(lookup_function_t);
            
            NTensor evaluate();
 
 
            void find_variable_locations();
 
        private:
            Ex                ex;
            lookup_function_t lookup_function;
 
            // The shape of the tensor that we will produce.
            std::vector<size_t> fullshape;
    };
 
};