aboutsummaryrefslogtreecommitdiff
path: root/src/ARMeilleure/Allocators.cs
blob: fba302657136d7baa4c65f9715d611432ad2d98f (plain) (blame)
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
using ARMeilleure.Common;
using System;
using System.Runtime.CompilerServices;

namespace ARMeilleure
{
    static class Allocators
    {
        [ThreadStatic] private static ArenaAllocator _default;
        [ThreadStatic] private static ArenaAllocator _operands;
        [ThreadStatic] private static ArenaAllocator _operations;
        [ThreadStatic] private static ArenaAllocator _references;
        [ThreadStatic] private static ArenaAllocator _liveRanges;
        [ThreadStatic] private static ArenaAllocator _liveIntervals;

        public static ArenaAllocator Default => GetAllocator(ref _default, 256 * 1024, 4);
        public static ArenaAllocator Operands => GetAllocator(ref _operands, 64 * 1024, 8);
        public static ArenaAllocator Operations => GetAllocator(ref _operations, 64 * 1024, 8);
        public static ArenaAllocator References => GetAllocator(ref _references, 64 * 1024, 8);
        public static ArenaAllocator LiveRanges => GetAllocator(ref _liveRanges, 64 * 1024, 8);
        public static ArenaAllocator LiveIntervals => GetAllocator(ref _liveIntervals, 64 * 1024, 8);

        [MethodImpl(MethodImplOptions.AggressiveInlining)]
        private static ArenaAllocator GetAllocator(ref ArenaAllocator alloc, uint pageSize, uint pageCount)
        {
            alloc ??= new ArenaAllocator(pageSize, pageCount);

            return alloc;
        }

        public static void ResetAll()
        {
            Default.Reset();
            Operands.Reset();
            Operations.Reset();
            References.Reset();
        }
    }
}