aboutsummaryrefslogtreecommitdiff
path: root/include/libn.h
blob: 64e2e709d9ea7a5d0eda52aecac6e12dcaa31ec2 (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
#pragma once

// ------------------------------------------------------------------------
// Standard library

#include <assert.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <ctype.h>
#include <stdio.h>

#include <unistd.h>
#include <fcntl.h>
#include <sys/stat.h>
#include <sys/types.h>

// ----------------------------------------------------------------------------
// Dynamic array. 

typedef struct bufHdr
{
    vlong len;
    vlong cap;
    byte  buf[];
} bufHdr;

#define _bufHdr(s) ((bufHdr*)((uint8*)(s)-offsetof(bufHdr, buf)))
#define buflen(s) ((s) ? (_bufHdr(s)->len) : 0)
#define bufcap(s) ((s) ? (_bufHdr(s)->cap) : 0)
#define bufend(s) ((s) + buflen(s))
#define bufsize(s) ((s) ? (buflen(s) * sizeof((s)[0])) : 0)

#define buffree(s) ((s) ? (free(_bufHdr(s)), (s) = nil) : 0)
#define buffit(s, n) ((n) <= bufcap(s) ? 0 : ((s) = bufgrow((s), (n), sizeof(*(s)))))

#define bufresize(s, n)                                                        \
    do {                                                                       \
        (buffit(s, n));                                                        \
        ((_bufHdr(s)->len) = (n));                                             \
    } while (0)

#define bufpush(s, ...) (buffit((s), 1 + buflen(s)), (s)[_bufHdr(s)->len++] = (__VA_ARGS__))

#define bufpop(s, i) (_bufpop((s), (i), sizeof(*(s))), (s)[_bufHdr(s)->len])

void* bufgrow(void*, vlong, vlong);
void _bufpop(void*, int, vlong);

// -----------------------------------------------------------------------------
// Co-routines

typedef struct Coro Coro;

Coro*   coro·new(uintptr stk, uintptr (*func)(Coro*, uintptr));
uintptr coro·yield(Coro *c, uintptr arg);
error   coro·free(Coro *c);

// -----------------------------------------------------------------------------
// Strings

// TODO(nnoll): Move here?
#include ".include/str.h"

// -----------------------------------------------------------------------------
// Maps or dictionaries

#include ".include/map.h"

// -----------------------------------------------------------------------------
// I/O

typedef FILE Stream;
enum SeekPos
{
    seek·CUR = SEEK_CUR,
    seek·SET = SEEK_SET,
    seek·END = SEEK_END
};

Stream *io·open(byte *name, byte *mode);
error   io·close(Stream *s);
byte    io·getbyte(Stream *s);
vlong   io·read(Stream *s, int sz, int n, void *buf);
int     io·readln(Stream *s, int n, byte* buf);
error   io·putbyte(Stream *s, byte c);
int     io·putstring(Stream *s, string str);
vlong   io·write(Stream *s, int sz, int n, void *buf);
int     io·seek(Stream *s, long off, enum SeekPos origin);

// -----------------------------------------------------------------------------
// Buffered I/O

// -----------------------------------------------------------------------------
// Error handling functions.

void errorf(const byte* fmt, ...);

#define panicf(...)  (errorf(__VA_ARGS__), assert(0))