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
|
#pragma once
// ----------------------------------------------------------------------------
// basic memory operations
/* mem functions */
int mem·move(void *dst, uintptr size, void *src);
int mem·copy(void *dst, uintptr size, void *src);
int mem·compare(void *, uintptr size, void *);
void *mem·findc(void *dst, uintptr len, int c);
int mem·set(void *dst, uintptr size, int val);
int mem·set64(void *dst, uintptr size, uint64 val);
// ----------------------------------------------------------------------------
// dynamic array
typedef struct mem·BufHead
{
vlong len;
vlong cap;
byte buf[];
} mem·BufHead;
#define mem·bufhdr(b) ((mem·BufHead*)((uint8*)(b)-offsetof(mem·BufHead, buf)))
#define mem·buflen(b) ((b) ? (mem·bufhdr(b)->len) : 0)
#define mem·bufcap(b) ((b) ? (mem·bufhdr(b)->cap) : 0)
#define mem·bufend(b) ((b) + mem·buflen(b))
#define mem·bufsize(b) ((b) ? (mem·buflen(b) * sizeof((b)[0])) : 0)
#define mem·buffree(b) ((b) ? (free(mem·bufhdr(b)), (b) = nil) : 0)
#define mem·buffit(b, n) ((n) <= mem·bufcap(b) ? 0 : ((b) = mem·bufgrow((b), (n), sizeof(*(b)))))
#define mem·bufpush(b, ...) (mem·buffit((b), 1 + mem·buflen(b)), (b)[mem·bufhdr(b)->len++] = (__VA_ARGS__))
#define mem·bufaddn(b, n) (mem·buffit(b, mem·buflen(b)+n), mem·bufhdr(b)->len += n, b+mem·bufhdr(b)->len-n)
#define mem·bufpop(b) ((b)[--mem·bufhdr(b)->len])
#define mem·bufdel(b, i) mem·bufdeln((b), (i), 1)
#define mem·bufdeln(b, i, n) (mem·move((b)+(i), (b)+(i)+(n), sizeof(*(b))*(mem·bufhdr(b)->len-(n)-(i)), mem·bufhdr(b)->len -= (n))
#define mem·bufdelswap(b, i) ((b)[i] = mem·bufend(b)[-1], mem·bufhdr(b)->len-=1)
void* mem·bufgrow(void*, vlong, vlong);
// -----------------------------------------------------------------------------
// memory allocation
/* interfaces */
typedef struct mem·Allocator {
void *(*alloc)(void *heap, uint n, ulong size);
void (*free)(void *heap, void *ptr);
} mem·Allocator;
extern mem·Allocator sys·Memory;
typedef struct mem·Reallocator {
void *(*alloc)(void *iface, uint n, ulong size);
void *(*realloc)(void *iface, void *ptr, uint n, ulong size);
void (*free)(void *iface, void *ptr);
} mem·Reallocator;
extern mem·Reallocator sys·FullMemory;
/* simple memory arena */
typedef struct mem·Arena mem·Arena;
mem·Arena *mem·makearena(mem·Allocator from, void*);
void *mem·arenaalloc(mem·Arena *A, uint n, ulong size);
void mem·freearena(mem·Arena *A);
extern mem·Allocator mem·ArenaAllocator;
|