type.h

Go to the documentation of this file.

// Copyright (C) 2023 Ethan Uppal. All rights reserved.
 
#pragma once
 
#include "def.h"
#include "enum/make.h"             // TO_ENUM
#include "frontend/lexer/token.h"  // struct ps_token
#include "util/print.h"            // PS_PRINT_DECL, PS_PRINT_IMPL
#include "util/dynarr.h"
 
struct ps_type;
struct ps_type_tuple;
 
struct ps_type_field {
    STR name;
    struct ps_type* type;
};
 
struct ps_type_variant {
    STR name;
    struct ps_type_tuple* args;
};
 
struct ps_type_field_arr ps_dynarr(struct ps_type_field);
#define ps_type_field_arr_new() ps_dynarr_new(struct ps_type_field_arr)
#define ps_type_field_arr_free(arr) ps_dynarr_free(arr)
 
struct ps_type_variant_arr ps_dynarr(struct ps_type_variant);
#define ps_type_variant_arr_new() ps_dynarr_new(struct ps_type_variant_arr)
#define ps_type_variant_arr_free(arr) ps_dynarr_free(arr)
 
struct ps_type_tuple ps_dynarr(struct ps_type*);
#define ps_type_tuple_new() ps_dynarr_new(struct ps_type_tuple)
#define ps_type_tuple_free(arr) ps_dynarr_free(arr)
 
struct ps_type_struct {
    struct ps_token* name;
    struct ps_type_field_arr* fields;
};
 
struct ps_type_enum {
    struct ps_token* name;
    usize length;
    struct ps_type_variant_arr* variants;
};
 
enum ps_type_type {
#define ENUM TO_ENUM
#include "type_type.h"
#undef ENUM
};
 
struct ps_type {
    enum ps_type_type type;
    union {
        struct ps_type_tuple* tuple;
        struct ps_type_struct struct_;
        struct ps_type_enum enum_;
        // TODO: trait type
        struct ps_token* unresolved;
    } value;
};
 
bool ps_type_is_prim(enum ps_type_type type);
 
struct ps_type* ps_type_prim_new(enum ps_type_type prim_type);
 
struct ps_type* ps_type_struct_new(struct ps_token* name);
 
void ps_type_struct_add(struct ps_type* type, struct ps_type_field field);
 
struct ps_type* ps_type_enum_new(struct ps_token* name);
 
void ps_type_enum_add(struct ps_type* type, struct ps_type_variant variant);
 
struct ps_type* ps_type_unresolved_new(struct ps_token* name);
 
// factory function for identifier-based types
struct ps_type* ps_type_from_name(struct ps_token* name);
 
// factory function for tuples
struct ps_type* ps_type_from_tuple(struct ps_type_tuple* tuple);
 
PS_PRINT_DECL(ps_type);
PS_PRINT_DECL(ps_type_tuple);
PS_PRINT_DECL(ps_type_variant);
PS_IPRINT_DECL(ps_type_struct);
PS_IPRINT_DECL(ps_type_enum);
PS_IPRINT_DECL(ps_type_field);
PS_PRINT_DECL(ps_type_field_arr);