aboutsummaryrefslogtreecommitdiff
path: root/core/log/log_allocator.odin
blob: 934f0d6437ed825bee06566dcc071b516e151eec (plain)
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
package log

import "core:runtime"

Log_Allocator :: struct {
	allocator: runtime.Allocator,
	level:     Level,
	prefix:    string,
	locked:    bool,
}

log_allocator_init :: proc(la: ^Log_Allocator, level: Level, allocator := context.allocator, prefix := "") {
	la.allocator = allocator
	la.level = level
	la.prefix = prefix
	la.locked = false
}


log_allocator :: proc(la: ^Log_Allocator) -> runtime.Allocator {
	return runtime.Allocator{
		procedure = log_allocator_proc,
		data = la,
	}
}

log_allocator_proc :: proc(allocator_data: rawptr, mode: runtime.Allocator_Mode,
                           size, alignment: int,
                           old_memory: rawptr, old_size: int, location := #caller_location) -> ([]byte, runtime.Allocator_Error)  {
	la := (^Log_Allocator)(allocator_data)

	padding := " " if la.prefix != "" else ""

	if !la.locked {
		la.locked = true
		defer la.locked = false

		switch mode {
		case .Alloc:
			logf(
				la.level,
				"%s%s>>> ALLOCATOR(mode=.Alloc, size=%d, alignment=%d)",
				la.prefix, padding, size, alignment,
				location = location,
			)
		case .Alloc_Non_Zeroed:
			logf(
				la.level,
				"%s%s>>> ALLOCATOR(mode=.Alloc_Non_Zeroed, size=%d, alignment=%d)",
				la.prefix, padding, size, alignment,
				location = location,
			)
		case .Free:
			if old_size != 0 {
				logf(
					la.level,
					"%s%s<<< ALLOCATOR(mode=.Free, ptr=%p, size=%d)",
					la.prefix, padding, old_memory, old_size,
					location = location,
				)
			} else {
				logf(
					la.level,
					"%s%s<<< ALLOCATOR(mode=.Free, ptr=%p)",
					la.prefix, padding, old_memory,
					location = location,
				)
			}
		case .Free_All:
			logf(
				la.level,
				"%s%s<<< ALLOCATOR(mode=.Free_All)",
				la.prefix, padding,
				location = location,
			)
		case .Resize:
			logf(
				la.level,
				"%s%s>>> ALLOCATOR(mode=.Resize, ptr=%p, old_size=%d, size=%d, alignment=%d)",
				la.prefix, padding, old_memory, old_size, size, alignment,
				location = location,
			)
		case .Query_Features:
			logf(
				la.level,
				"%s%ALLOCATOR(mode=.Query_Features)",
				la.prefix, padding,
				location = location,
			)
		case .Query_Info:
			logf(
				la.level,
				"%s%ALLOCATOR(mode=.Query_Info)",
				la.prefix, padding,
				location = location,
			)
		}
	}

	data, err := la.allocator.procedure(la.allocator.data, mode, size, alignment, old_memory, old_size, location)
	if !la.locked {
		la.locked = true
		defer la.locked = false
		if err != nil {
			logf(
				la.level,
				"%s%ALLOCATOR ERROR=%v",
				la.prefix, padding, error,
				location = location,
			)
		}
	}
	return data, err
}