aboutsummaryrefslogtreecommitdiff
path: root/src/stdlib/optionals.c
blob: b2cd584dbc6e7ffb1c0e6b5421428314953b194f (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
// Optional types

#include <math.h>

#include "datatypes.h"
#include "floats.h"
#include "metamethods.h"
#include "optionals.h"
#include "text.h"
#include "util.h"

public
PUREFUNC bool is_none(const void *obj, const TypeInfo_t *non_optional_type) {
    if (non_optional_type->metamethods.is_none) return non_optional_type->metamethods.is_none(obj, non_optional_type);

    const bool *has_value = (const bool *)(obj + non_optional_type->size);
    return !(*has_value);
}

PUREFUNC public uint64_t Optional$hash(const void *obj, const TypeInfo_t *type) {
    return is_none(obj, type->OptionalInfo.type) ? 0 : generic_hash(obj, type->OptionalInfo.type);
}

PUREFUNC public int32_t Optional$compare(const void *x, const void *y, const TypeInfo_t *type) {
    if (x == y) return 0;
    bool x_is_null = is_none(x, type->OptionalInfo.type);
    bool y_is_null = is_none(y, type->OptionalInfo.type);
    if (x_is_null && y_is_null) return 0;
    else if (x_is_null != y_is_null) return (int32_t)y_is_null - (int32_t)x_is_null;
    else return generic_compare(x, y, type->OptionalInfo.type);
}

PUREFUNC public bool Optional$equal(const void *x, const void *y, const TypeInfo_t *type) {
    if (x == y) return true;

    bool x_is_null = is_none(x, type->OptionalInfo.type);
    bool y_is_null = is_none(y, type->OptionalInfo.type);
    if (x_is_null && y_is_null) return true;
    else if (x_is_null != y_is_null) return false;
    else return generic_equal(x, y, type->OptionalInfo.type);
}

public
Text_t Optional$as_text(const void *obj, bool colorize, const TypeInfo_t *type) {
    if (!obj) return Text$concat(generic_as_text(obj, colorize, type->OptionalInfo.type), Text("?"));

    if (is_none(obj, type->OptionalInfo.type)) return colorize ? Text("\x1b[31mnone\x1b[m") : Text("none");
    return generic_as_text(obj, colorize, type->OptionalInfo.type);
}

public
void Optional$serialize(const void *obj, FILE *out, Table_t *pointers, const TypeInfo_t *type) {
    bool has_value = !is_none(obj, type->OptionalInfo.type);
    assert(fputc((int)has_value, out) != EOF);
    if (has_value) _serialize(obj, out, pointers, type->OptionalInfo.type);
}

public
void Optional$deserialize(FILE *in, void *outval, List_t *pointers, const TypeInfo_t *type) {
    bool has_value = (bool)fgetc(in);
    const TypeInfo_t *nonnull = type->OptionalInfo.type;
    if (has_value) {
        memset(outval, 0, (size_t)type->size);
        _deserialize(in, outval, pointers, nonnull);
    } else {
        if (nonnull->tag == TextInfo) *(Text_t *)outval = NONE_TEXT;
        else if (nonnull->tag == ListInfo) *(List_t *)outval = NONE_LIST;
        else if (nonnull->tag == TableInfo) *(Table_t *)outval = NONE_TABLE;
        else if (nonnull == &Float64$info) *(double *)outval = (double)NAN;
        else if (nonnull == &Float32$info) *(float *)outval = (float)NAN;
        else if (nonnull->tag == StructInfo || (nonnull->tag == OpaqueInfo && type->size > nonnull->size))
            memset(outval + type->size, -1, (size_t)(type->size - nonnull->size));
        else memset(outval, 0, (size_t)type->size);
    }
}