CODEWITHSUNDEEP
c++templatestemplate-meta-programming
zounds
zounds

Reputation: 783

Passing a template member function to a template

I have a set of types templated on an integer N, and a class to contain all the types where N <= K and N >= 0, something like

template <int N> struct T { int i; }

template <int N> struct TContainer{
    std::vector<T<N>> vec;
    TContainer<N - 1> prev;

    template <int target_n> TContainer<target_n> & get() { return prev.get(); }
    template <> TContainer<N> & get<N>() { return *this; }
};

template <> struct TContainer<0>{
    std::vector<T<N>> vec;

    template <int target_n> TContainer<target_n> & get() {  }
    template <> TContainer<0> & get<0>() { return *this; }
};

class ContainsAll{
    TContainer<K> my_data;
    int my_instance_data;
};

Is it possible for ContainsAll to have some kind of foreach function, which would be able to take a template member function to act on each element for all N? Something like

template<int N>
void for_each(TContainer<N> & container, template_function){
    for(auto it = container.vec.begin(); it != container.vec.end(); ++it){
        template_function(*it);
    }
    for_each<K - 1>(container.prev, template_function);
}

template<> void for_each<0>(TContainer<0> & container, template_function){
    for(auto it = container.vec.begin(); it != container.vec.end(); ++it){
        template_function(*it);
    }
}

So I can do

template <int N> add_to_T(T<N> & in){
    in.i += my_instance_data;
}

for_each(my_container, add_to_T);

Upvotes: 3

Views: 181

Answers (1)

kennytm
kennytm

Reputation: 523214

You can't pass a template function. But you can pass a instance of a structure (a function object), which contains a template function, e.g.

template <typename F>
void for_each(TContainer<0>& container, F f)
{
    for (auto& t : container.vec)
        f(t);
}

template <int N, typename F>
void for_each(TContainer<N>& container, F f)
{
    for (auto& t : container.vec)
        f(t);
    for_each(container.prev, std::move(f));
}


struct TAdder {
    template <int N>
    void operator()(T<N>& t) const
    {
        t.i += N;
    }
};

for_each(c, TAdder{});

Upvotes: 2

Related Questions