1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
/*
* (C) Copyright 2024- ECMWF.
*
* This software is licensed under the terms of the Apache Licence Version 2.0
* which can be obtained at http://www.apache.org/licenses/LICENSE-2.0.
* In applying this licence, ECMWF does not waive the privileges and immunities
* granted to it by virtue of its status as an intergovernmental organisation
* nor does it submit to any jurisdiction.
*/
#include <cstddef>
#include <iostream>
#include <sstream>
#include "pluto/pluto.h"
[[maybe_unused]] constexpr std::size_t kb = 1024;
[[maybe_unused]] constexpr std::size_t mb = 1024 * kb;
[[maybe_unused]] constexpr std::size_t gb = 1024 * mb;
void use_allocator() {
// The allocator will use the default memory resource
std::size_t size = 1 * mb;
pluto::allocator<std::byte> allocator;
std::byte* array = allocator.allocate(size);
allocator.deallocate(array, size);
}
void recursive_scope_push_pop(int recursion) {
// Manually push new scope
pluto::scope::push();
pluto::TraceMemoryResource mr{"scope_" + std::to_string(recursion), pluto::new_delete_resource()};
pluto::set_default_resource(&mr);
use_allocator();
if (recursion > 1) {
recursive_scope_push_pop(recursion - 1);
}
use_allocator();
// Manually pop end scope
pluto::scope::pop();
}
void recursive_scope_object(int recursion) {
// Use scope object lifetime to push/pop scope
pluto::scope scope;
pluto::TraceMemoryResource mr{"scope_" + std::to_string(recursion), pluto::new_delete_resource()};
pluto::set_default_resource(&mr);
use_allocator();
if (recursion > 1) {
recursive_scope_object(recursion - 1);
}
use_allocator();
}
int main([[maybe_unused]] int argc, [[maybe_unused]] char* argv[]) {
std::cout << "BEGIN" << std::endl;
pluto::trace::enable(true);
// Just wrap default resource for tracing allocations
pluto::TraceMemoryResource default_memory_resource{"default", pluto::get_default_resource()};
pluto::set_default_resource(&default_memory_resource);
use_allocator(); // should print default allocations
recursive_scope_push_pop(3);
use_allocator(); // should print default allocations
recursive_scope_object(3);
use_allocator(); // should print default allocations
// Illustrate that we can enable or disable tracing within a pluto::scope
// and within the scope change the output stream
{
auto print = [](const std::string& msg) {
if (pluto::trace::enabled()) {
pluto::trace::out << "PLUTO_TRACE " << msg << std::endl;
}
};
std::stringstream trace_out;
{
pluto::scope scope_1;
pluto::trace::set(trace_out); // --> start printing to the stringstream
pluto::trace::enable(true);
print("scope 1 begin");
{
pluto::scope scope_2;
pluto::trace::enable(false);
print("scope 2 begin");
{
pluto::scope scope_3;
pluto::trace::enable(true);
print("scope 3 begin");
print("scope 3 end");
}
print("scope 2 end");
}
print("scope 1 end");
}
print("back in main scope"); // should print back to std::cout
std::cout << "pluto::trace:\n" << trace_out.str() << std::endl;
}
std::cout << "END" << std::endl;
}
|
