Sync to latest upstream
[pcsx_rearmed.git] / deps / lightrec / memmanager.c
1 /*
2  * Copyright (C) 2019-2020 Paul Cercueil <paul@crapouillou.net>
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * This library is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  */
14
15 #include "config.h"
16 #include "lightrec-private.h"
17 #include "memmanager.h"
18
19 #include <stdlib.h>
20 #if ENABLE_TINYMM
21 #include <tinymm.h>
22 #endif
23
24 #ifdef ENABLE_THREADED_COMPILER
25 #include <stdatomic.h>
26
27 static atomic_uint lightrec_bytes[MEM_TYPE_END];
28
29 void lightrec_register(enum mem_type type, unsigned int len)
30 {
31         atomic_fetch_add(&lightrec_bytes[type], len);
32 }
33
34 void lightrec_unregister(enum mem_type type, unsigned int len)
35 {
36         atomic_fetch_sub(&lightrec_bytes[type], len);
37 }
38
39 unsigned int lightrec_get_mem_usage(enum mem_type type)
40 {
41         return atomic_load(&lightrec_bytes[type]);
42 }
43
44 #else /* ENABLE_THREADED_COMPILER */
45
46 static unsigned int lightrec_bytes[MEM_TYPE_END];
47
48 void lightrec_register(enum mem_type type, unsigned int len)
49 {
50         lightrec_bytes[type] += len;
51 }
52
53 void lightrec_unregister(enum mem_type type, unsigned int len)
54 {
55         lightrec_bytes[type] -= len;
56 }
57
58 unsigned int lightrec_get_mem_usage(enum mem_type type)
59 {
60         return lightrec_bytes[type];
61 }
62 #endif /* ENABLE_THREADED_COMPILER */
63
64 unsigned int lightrec_get_total_mem_usage(void)
65 {
66         unsigned int i, count;
67
68         for (i = 0, count = 0; i < MEM_TYPE_END; i++)
69                 count += lightrec_get_mem_usage((enum mem_type)i);
70
71         return count;
72 }
73
74 void * lightrec_malloc(struct lightrec_state *state,
75                        enum mem_type type, unsigned int len)
76 {
77         void *ptr;
78
79 #if ENABLE_TINYMM
80         if (type == MEM_FOR_IR)
81                 ptr = tinymm_malloc(state->tinymm, len);
82         else
83 #endif
84                 ptr = malloc(len);
85         if (!ptr)
86                 return NULL;
87
88         lightrec_register(type, len);
89
90         return ptr;
91 }
92
93 void * lightrec_calloc(struct lightrec_state *state,
94                        enum mem_type type, unsigned int len)
95 {
96         void *ptr;
97
98 #if ENABLE_TINYMM
99         if (type == MEM_FOR_IR)
100                 ptr = tinymm_zalloc(state->tinymm, len);
101         else
102 #endif
103                 ptr = calloc(1, len);
104         if (!ptr)
105                 return NULL;
106
107         lightrec_register(type, len);
108
109         return ptr;
110 }
111
112 void lightrec_free(struct lightrec_state *state,
113                    enum mem_type type, unsigned int len, void *ptr)
114 {
115         lightrec_unregister(type, len);
116 #if ENABLE_TINYMM
117         if (type == MEM_FOR_IR)
118                 tinymm_free(state->tinymm, ptr);
119         else
120 #endif
121                 free(ptr);
122 }
123
124 float lightrec_get_average_ipi(void)
125 {
126         unsigned int code_mem = lightrec_get_mem_usage(MEM_FOR_CODE);
127         unsigned int native_mem = lightrec_get_mem_usage(MEM_FOR_MIPS_CODE);
128
129         return native_mem ? (float)code_mem / (float)native_mem : 0.0f;
130 }