Functional.hh

Go to the documentation of this file.

 
#pragma once
 
#include <functional>
#include "Storage.hh"
 
namespace cadabra {
 
 
    void do_list(const Ex& tr, Ex::iterator it, std::function<bool(Ex::iterator)> f);
 
 
    int list_size(const Ex& tr, Ex::iterator it);
 
 
    Ex::iterator find_in_list(const Ex& tr, Ex::iterator it, std::function<Ex::iterator(Ex::iterator)> f);
 
 
    Ex::iterator find_in_subtree(const Ex& tr, Ex::iterator it, std::function<bool(Ex::iterator)> f, bool including_head=true);
 
 
    Ex make_list(Ex el);
 
 
 
 
    template<typename T>
    typename T::iterator do_subtree(const T& tr, typename T::iterator it, std::function<typename T::iterator(typename T::iterator)> f)
        {
        if(it==tr.end()) return it;
 
        class T::post_order_iterator walk=it, last=it;
        ++last;
        walk.descend_all();
 
        do {
            auto nxt=walk;
            ++nxt;
 
            bool cpy=false;
            if(walk==it) cpy=true;
            walk = f(walk);
            if(cpy) it=walk;
 
            walk=nxt;
            }
        while(walk!=last);
 
        return it;
        }
 
    };