verbose.c

#include "mzcc.h"

char *ctype_to_string(Ctype *ctype)
{
    if (!ctype)
        return "(nil)";
    switch (ctype->type) {
    case CTYPE_VOID:
        return "void";
    case CTYPE_INT:
        return "int";
    case CTYPE_LONG:
        return "long";
    case CTYPE_CHAR:
        return "char";
    case CTYPE_FLOAT:
        return "float";
    case CTYPE_DOUBLE:
        return "double";
    case CTYPE_PTR: {
        String s = make_string();
        string_appendf(&s, "*%s", ctype_to_string(ctype->ptr));
        return get_cstring(s);
    }
    case CTYPE_ARRAY: {
        String s = make_string();
        string_appendf(&s, "[%d]%s", ctype->len, ctype_to_string(ctype->ptr));
        return get_cstring(s);
    }
    case CTYPE_STRUCT: {
        String s = make_string();
        string_appendf(&s, "(struct");
        for (Iter i = list_iter(dict_values(ctype->fields)); !iter_end(i);)
            string_appendf(&s, " (%s)", ctype_to_string(iter_next(&i)));
        string_appendf(&s, ")");
        return get_cstring(s);
    }
    default:
        error("Unknown ctype: %d", ctype);
        return NULL; /* non-reachable */
    }
}

static void uop_to_string(String *buf, char *op, Ast *ast)
{
    string_appendf(buf, "(%s %s)", op, ast_to_string(ast->operand));
}

static void binop_to_string(String *buf, char *op, Ast *ast)
{
    string_appendf(buf, "(%s %s %s)", op, ast_to_string(ast->left),
                   ast_to_string(ast->right));
}

static void ast_to_string_int(String *buf, Ast *ast)
{
    if (!ast) {
        string_appendf(buf, "(nil)");
        return;
    }
    switch (ast->type) {
    case AST_LITERAL:
        switch (ast->ctype->type) {
        case CTYPE_CHAR:
            if (ast->ival == '\n')
                string_appendf(buf, "'\n'");
            else if (ast->ival == '\\')
                string_appendf(buf, "'\\\\'");
            else
                string_appendf(buf, "'%c'", ast->ival);
            break;
        case CTYPE_INT:
            string_appendf(buf, "%d", ast->ival);
            break;
        case CTYPE_LONG:
            string_appendf(buf, "%ldL", ast->ival);
            break;
        case CTYPE_FLOAT:
        case CTYPE_DOUBLE:
            string_appendf(buf, "%f", ast->fval);
            break;
        default:
            error("internal error");
        }
        break;
    case AST_STRING:
        string_appendf(buf, "\"%s\"", quote_cstring(ast->sval));
        break;
    case AST_LVAR:
    case AST_GVAR:
        string_appendf(buf, "%s", ast->varname);
        break;
    case AST_FUNCALL: {
        string_appendf(buf, "(%s)%s(", ctype_to_string(ast->ctype), ast->fname);
        for (Iter i = list_iter(ast->args); !iter_end(i);) {
            string_appendf(buf, "%s", ast_to_string(iter_next(&i)));
            if (!iter_end(i))
                string_appendf(buf, ",");
        }
        string_appendf(buf, ")");
        break;
    }
    case AST_FUNC: {
        string_appendf(buf, "(%s)%s(", ctype_to_string(ast->ctype), ast->fname);
        for (Iter i = list_iter(ast->params); !iter_end(i);) {
            Ast *param = iter_next(&i);
            string_appendf(buf, "%s %s", ctype_to_string(param->ctype),
                           ast_to_string(param));
            if (!iter_end(i))
                string_appendf(buf, ",");
        }
        string_appendf(buf, ")");
        ast_to_string_int(buf, ast->body);
        break;
    }
    case AST_DECL:
        string_appendf(buf, "(decl %s %s", ctype_to_string(ast->declvar->ctype),
                       ast->declvar->varname);
        if (ast->declinit)
            string_appendf(buf, " %s)", ast_to_string(ast->declinit));
        else
            string_appendf(buf, ")");
        break;
    case AST_ARRAY_INIT:
        string_appendf(buf, "{");
        for (Iter i = list_iter(ast->arrayinit); !iter_end(i);) {
            ast_to_string_int(buf, iter_next(&i));
            if (!iter_end(i))
                string_appendf(buf, ",");
        }
        string_appendf(buf, "}");
        break;
    case AST_IF:
        string_appendf(buf, "(if %s %s", ast_to_string(ast->cond),
                       ast_to_string(ast->then));
        if (ast->els)
            string_appendf(buf, " %s", ast_to_string(ast->els));
        string_appendf(buf, ")");
        break;
    case AST_TERNARY:
        string_appendf(buf, "(? %s %s %s)", ast_to_string(ast->cond),
                       ast_to_string(ast->then), ast_to_string(ast->els));
        break;
    case AST_FOR:
        string_appendf(buf, "(for %s %s %s ", ast_to_string(ast->forinit),
                       ast_to_string(ast->forcond),
                       ast_to_string(ast->forstep));
        string_appendf(buf, "%s)", ast_to_string(ast->forbody));
        break;
    case AST_RETURN:
        string_appendf(buf, "(return %s)", ast_to_string(ast->retval));
        break;
    case AST_COMPOUND_STMT: {
        string_appendf(buf, "{");
        for (Iter i = list_iter(ast->stmts); !iter_end(i);) {
            ast_to_string_int(buf, iter_next(&i));
            string_appendf(buf, ";");
        }
        string_appendf(buf, "}");
        break;
    }
    case AST_STRUCT_REF:
        ast_to_string_int(buf, ast->struc);
        string_appendf(buf, ".");
        string_appendf(buf, ast->field);
        break;
    case AST_ADDR:
        uop_to_string(buf, "addr", ast);
        break;
    case AST_DEREF:
        uop_to_string(buf, "deref", ast);
        break;
    case PUNCT_INC:
        uop_to_string(buf, "++", ast);
        break;
    case PUNCT_DEC:
        uop_to_string(buf, "--", ast);
        break;
    case PUNCT_LOGAND:
        binop_to_string(buf, "and", ast);
        break;
    case PUNCT_LOGOR:
        binop_to_string(buf, "or", ast);
        break;
    case '!':
        uop_to_string(buf, "!", ast);
        break;
    case '&':
        binop_to_string(buf, "&", ast);
        break;
    case '|':
        binop_to_string(buf, "|", ast);
        break;
    default: {
        char *left = ast_to_string(ast->left);
        char *right = ast_to_string(ast->right);
        if (ast->type == PUNCT_EQ)
            string_appendf(buf, "(== ");
        else
            string_appendf(buf, "(%c ", ast->type);
        string_appendf(buf, "%s %s)", left, right);
    }
    }
}

char *ast_to_string(Ast *ast)
{
    String s = make_string();
    ast_to_string_int(&s, ast);
    return get_cstring(s);
}

char *token_to_string(const Token tok)
{
    enum TokenType ttype = get_ttype(tok);
    if (ttype == TTYPE_NULL)
        return "(null)";
    String s = make_string();
    switch (ttype) {
    case TTYPE_NULL:
        error("internal error: unknown token type: %d", get_ttype(tok));
    case TTYPE_IDENT:
        return get_ident(tok);
    case TTYPE_PUNCT:
        if (is_punct(tok, PUNCT_EQ))
            string_appendf(&s, "==");
        else
            string_appendf(&s, "%c", get_punct(tok));
        return get_cstring(s);
    case TTYPE_CHAR:
        string_append(&s, get_char(tok));
        return get_cstring(s);
    case TTYPE_NUMBER:
        return get_number(tok);
    case TTYPE_STRING:
        string_appendf(&s, "\"%s\"", get_strtok(tok));
        return get_cstring(s);
    }
    error("internal error: unknown token type: %d", get_ttype(tok));
    return NULL; /* non-reachable */
}