mangle.c

Go to the documentation of this file.

// Copyright (C) 2023 Ethan Uppal. All rights reserved.
 
#include <string.h>     // strlen
#include "mangle.h"
#include "util/itoa.h"  // ps_ztoa
#include "enum/make.h"
 
static STR ps_types_mangle[] = {
#define ENUM TO_STRING2
#include "frontend/analyzer/type_type.h"
#undef ENUM
};
 
void ps_mangle_id(struct ps_dynstr** dstr, STR str) {
    ps_dynstr_append(dstr, ps_ztoa(strlen(str)));
    ps_dynstr_append(dstr, str);
}
 
void ps_mangle_type(struct ps_dynstr** dstr, struct ps_type* type) {
    if (ps_type_is_prim(type->type)) {
        ps_dynstr_append(dstr, ps_types_mangle[type->type]);
    } else {
        switch (type->type) {
            case PS_TYPE_TUPLE: {
                auto_t tuple = type->value.tuple;
                ps_dynstr_append(dstr, ps_types_mangle[PS_TYPE_TUPLE]);
                ps_dynstr_append(dstr, ps_ztoa(tuple->length));
                ps_dynstr_append(dstr, "_");
                for (usize i = 0; i < tuple->length; i++) {
                    ps_mangle_type(dstr, $arr(tuple)[i]);
                }
                break;
            }
            case PS_TYPE_STRUCT: {
                auto_t struct_ = &type->value.struct_;
                ps_dynstr_append(dstr, ps_ztoa(struct_->name->length));
                ps_dynstr_append(dstr, struct_->name->start);
                break;
            }
            case PS_TYPE_ENUM: {
                auto_t enum_ = &type->value.enum_;
                ps_dynstr_append(dstr, ps_ztoa(enum_->name->length));
                ps_dynstr_append(dstr, enum_->name->start);
                break;
            }
            case PS_TYPE_TRAIT: {
                PS_NO_IMPL();
            }
            case PS_TYPE_UNRESOLVED: {
                auto_t unresolved = type->value.unresolved;
                ps_dynstr_append(dstr, ps_ztoa(unresolved->length));
                ps_dynstr_append(dstr, unresolved->start);
                break;
            }
            default: {
                PS_NO_IMPL();
            }
        }
    }
}