alloc_F.pass.cpp revision 1b92188a82b01e76ac6e8ad5f997293c2a078adc 
1//===----------------------------------------------------------------------===//
2//
3//                     The LLVM Compiler Infrastructure
4//
5// This file is dual licensed under the MIT and the University of Illinois Open
6// Source Licenses. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9
10// <functional>
11
12// class function<R(ArgTypes...)>
13
14// template<class F, class A> function(allocator_arg_t, const A&, F);
15
16#include <functional>
17#include <cassert>
18
19#include "test_allocator.h"
20
21class A
22{
23    int data_[10];
24public:
25    static int count;
26
27    A()
28    {
29        ++count;
30        for (int i = 0; i < 10; ++i)
31            data_[i] = i;
32    }
33
34    A(const A&) {++count;}
35
36    ~A() {--count;}
37
38    int operator()(int i) const
39    {
40        for (int j = 0; j < 10; ++j)
41            i += data_[j];
42        return i;
43    }
44
45    int foo(int) const {return 1;}
46};
47
48int A::count = 0;
49
50int g(int) {return 0;}
51
52class Foo {
53public:
54  void bar(int k) { }
55};
56
57int main()
58{
59    {
60    std::function<int(int)> f(std::allocator_arg, test_allocator<A>(), A());
61    assert(A::count == 1);
62    assert(f.target<A>());
63    assert(f.target<int(*)(int)>() == 0);
64    }
65    assert(A::count == 0);
66    {
67    std::function<int(int)> f(std::allocator_arg, test_allocator<int(*)(int)>(), g);
68    assert(f.target<int(*)(int)>());
69    assert(f.target<A>() == 0);
70    }
71    {
72    std::function<int(int)> f(std::allocator_arg, test_allocator<int(*)(int)>(),
73                              (int (*)(int))0);
74    assert(!f);
75    assert(f.target<int(*)(int)>() == 0);
76    assert(f.target<A>() == 0);
77    }
78    {
79    std::function<int(const A*, int)> f(std::allocator_arg,
80                                        test_allocator<int(A::*)(int)const>(),
81                                        &A::foo);
82    assert(f);
83    assert(f.target<int (A::*)(int) const>() != 0);
84    }
85    {
86    Foo f;
87    std::function<void(int)> fun = std::bind(&Foo::bar, &f, std::placeholders::_1);
88    fun(10);
89    }
90}
91