In libstdc++'s std::function
we use a union type that is suitably sized and aligned to store pointers, function pointers or pointers to member functions. We avoid a heap allocation for any function object that can be stored in that size and alignment, but only if it is "location invariant"
/**
* Trait identifying "location-invariant" types, meaning that the
* address of the object (or any of its members) will not escape.
* Also implies a trivial copy constructor and assignment operator.
*/
The code is based on the std::tr1::function
implementation and that part hasn't changed significantly. I think that could be simplified using std::aligned_storage
and could be improved by specializing the trait so that more types are identified as location invariant.
Invoking the target object is done without any virtual function calls, the type erasure is done by storing a single function pointer in the std::function
which is the address of a function template specialization. All operations are done by calling that function template through the stored pointer and passing in an enum identifying what operation it is being asked to perform. This means no vtable and only a single function pointer needs to be stored in the object.
This design was contributed by the original boost::function
author and I believe it is close to the boost implementation. See the Performance docs for Boost.Function for some rationale. That means it's pretty unlikely that GCC's std::function
is any faster than boost::function
, because it's a similar design by the same person.
N.B. our std::function
doesn't support construction with an allocator yet, any allocations it needs to do will be done using new
.
In response to Emile's comment expressing a desire to avoid a heap allocation for a std::function
which holds a pointer to member function and an object, here's a little hack to do it (but you didn't hear it from me ;-)
struct A {
int i = 0;
int foo() const { return 0; }
};
struct InvokeA
{
int operator()() const { return a->foo(); }
A* a;
};
namespace std
{
template<> struct __is_location_invariant<InvokeA>
{ static const bool value = true; };
}
int main()
{
A a;
InvokeA inv{ &a };
std::function<int()> f2(inv);
return f2();
}
The trick is that InvokeA
is small enough to fit in the function
's small object buffer, and the trait specialization says it's safe to store in there, so the function
holds a copy of that object directly, not on the heap. This requires a
to persist as long as the pointer to it persists, but that would be the case anyway if the function
's target was bind(&A::foo, &a)
.