ule/print.cpp

#include <stdarg.h> // va_list, va_start, va_end
#include <stdio.h> // FILE, stderr, stdout | vfprintf
#include <stdlib.h> // exit

#include "alloc.h"
#include "string.h"
#include "print.h"
#include "types.h"


void vprint(const char* format, va_list args) {
    TYPES_H_FTAG;
    vfprintf(stdout, format, args);
}

void vprintln(const char* format, va_list args) {
    TYPES_H_FTAG;
    vprint(format, args);
    print("\n");
}

/**
 * The entire purpose of this is so we don't have to #import <stdio.h> everywhere
 * +we intend to replace printf at some point with this
 */
void print(const char* format, ...) {
    TYPES_H_FTAG;
    if (format == null) { print("null"); return; }

    va_list args;
    va_start(args, format);

    vprint(format, args);

    va_end(args);
}

void println(const char* format, ...) {
    TYPES_H_FTAG;
    if (format == null) { print("null\n"); return; }

    va_list args;
    va_start(args, format);

    vprintln(format, args);

    va_end(args);
}

/**
 * Prints a stack trace.
 * Implementation varies for Win32 vs. *nix
 */
#define BACKTRACE_MAX_FRAMES 63
#ifdef _WIN32
#include <windows.h>
#include <dbghelp.h>
// if |string| is non-null, then the stack trace will be concatenated to it instead of being printed to stdout.
void trace(String* string) {
    TYPES_H_FTAG;

    #define BACKTRACE_MAX_FUNCTION_NAME_LENGTH 1024
    HANDLE processHandle = GetCurrentProcess();
    SymInitialize(processHandle, null, true);

    void* stack[BACKTRACE_MAX_FRAMES];
    unsigned short numFrames = CaptureStackBackTrace(0, BACKTRACE_MAX_FRAMES, stack, null);

    char buffer[sizeof(SYMBOL_INFO) + (BACKTRACE_MAX_FUNCTION_NAME_LENGTH - 1) * sizeof(TCHAR)];
    SYMBOL_INFO* symbol = (SYMBOL_INFO*) buffer;
    symbol->MaxNameLen = BACKTRACE_MAX_FUNCTION_NAME_LENGTH;
    symbol->SizeOfStruct = sizeof(SYMBOL_INFO);

    DWORD displacement;
    IMAGEHLP_LINE64 line;
    line.SizeOfStruct = sizeof(IMAGEHLP_LINE64);
    for (u32 i = 0; i < numFrames; i++) {
        DWORD64 address = (DWORD64) stack[i];
        SymFromAddr(processHandle, address, null, symbol);

        if (SymGetLineFromAddr64(processHandle, address, &displacement, &line)) {
            if (string == null) {
                print("\tat %s in %s: line: %lu: address %0x%0X\n", symbol->Name, line.FileName, line.LineNumber, symbol->Address);

            } else {
                string->appendf("\tat %s in %s: line: %lu: address %0x%0X\n", symbol->Name, line.FileName, line.LineNumber, symbol->Address);
            }
        } else {
            if (string == null) {
                print("\tSymGetLineFromAddr64 returned error code %lu.\n", GetLastError());
                print("\tat %s, address 0x%0X.\n", symbol->Name, symbol->Address);

            } else {
                string->appendf("\tSymGetLineFromAddr64 returned error code %lu.\n", GetLastError());
                string->appendf("\tat %s, address 0x%0X.\n", symbol->Name, symbol->Address);
            }
        }
    }
    #undef BACKTRACE_MAX_FUNCTION_NAME_LENGTH
}
#else
// OSX and Linux stacktrace stuff.
#include <execinfo.h> // backtrace, backtrace_symbols
#include <cxxabi.h> // abi::__cxa_demangle
// if |string| is non-null, then the stack trace will be concatenated to it instead of being printed to stdout.
void trace(String* string) {
    TYPES_H_FTAG;

    void* stack[BACKTRACE_MAX_FRAMES];
    u32 stackSize = backtrace(stack, BACKTRACE_MAX_FRAMES);

    // resolve addresses into strings containing "filename(function+address)"
    // this array must be free()-ed
    char** traces = backtrace_symbols(stack, stackSize);

    // iterate over the returned symbol lines. skip the first, it is the address of this function
    for (u32 i = 1; i < stackSize; i++) {

        // the names as provided by 'backtrace_symbols' are mangled for some reason.
        // we have to demangle them, using this weird api
        char buffer[1024];
        const char* mangledNameBegin = String::firstCharOccurence(traces[i], '_');
        const char* mangledNameEnd = String::lastCharOccurence(traces[i], '+');

        if (mangledNameBegin == null || mangledNameEnd == null) {
            // we can't demangle this name for some reason, just copy the mangled name to the buffer
            size_t length = String::len(traces[i]);
            String::memcpy(buffer, (void*)traces[i], length);
            buffer[length] = '\0';

        } else {
            size_t length = mangledNameEnd - 1 - mangledNameBegin;
            String::memcpy(buffer, (void*)mangledNameBegin, length);
            buffer[length] = '\0';
        }

        s32 status = -1;
        char* trace = abi::__cxa_demangle(buffer, null, null, &status);
        if (trace == null) {
            println("warning: failed to demangle name: %s", traces[i]);
            continue;
        }

        if (string == null) {
            print("%s\n", trace);

        } else {
            string->appendf("%s\n", trace);
        }
    }

    pFree(traces);
}
#undef BACKTRACE_MAX_FRAMES
#endif

void _debug(const char* format, ...) {
    TYPES_H_FTAG;
    if (format == null) {
        print("%sdebug:%s null\n", ANSI_BLUE, ANSI_RESET);
        return;
    }

    va_list args;
    va_start(args, format);

    print("%sdebug:%s ", ANSI_BLUE, ANSI_RESET);
    vprintln(format, args);

    va_end(args);
}

void _warn(const char* format, ...) {
    TYPES_H_FTAG;
    if (format == null) {
        print("%swarning:%s null\n", ANSI_YELLOW, ANSI_RESET);
        return;
    }

    va_list args;
    va_start(args, format);

    print("%swarning:%s ", ANSI_YELLOW, ANSI_RESET);
    vprintln(format, args);

    va_end(args);
}

static void (*customDie)(const char* string) = null;
// if you want to override what happens by default when your program calls 'die', you can do so here.
// just keep in mind the intention is for 'die' to be for when your program has encountered a fatal, unrecoverable error.
void setCustomDieBehavior(void (*dieBehavior)(const char* string)) {
    customDie = dieBehavior;
}

// for fatal errors which may occur at runtime, even on a release binary.
// if a fatal error should not occur at runtime on a release binary, consider preferring 'massert'
// it's unclear when you should use asserts vs. die actually. idk man
void die(const char* format, ...) {
    TYPES_H_FTAG;
    if (format == null) {
        if (customDie == null) {
            print("%serror:%s (unspecified error)\n", ANSI_RED, ANSI_RESET);
            trace();
            exit(1);
            return;

        } else {
            String string = String128f("error: (unspecified error)\n");
            trace(&string);
            customDie(string.c_str());
            return;
        }
    }

    va_list args;
    va_start(args, format);

    if (customDie == null) {
        println("%serror:%s", ANSI_RED, ANSI_RESET);
        vprintln(format, args);
        println();

        va_end(args);

        trace();
        exit(1);

    } else {
        String string = String128f("");
        string.setfv(format, args);
        string.append("\n");
        trace(&string);

        va_end(args);

        customDie(string.c_str());
    }
}

void print(bool b)         { TYPES_H_FTAG; print("%s", b ? "true" : "false"); }
void print(char c)         { TYPES_H_FTAG; print("%c", c); }
void print(signed int i)   { TYPES_H_FTAG; print("%d", i); }
void print(unsigned int i) { TYPES_H_FTAG; print("%u", i); }
void print(float f)        { TYPES_H_FTAG; print("%.14g", f); }
void print(double d)       { TYPES_H_FTAG; print("%.14g", d); }
void print(void* p)        { TYPES_H_FTAG; print("%p", p); }
void print(char* s)        { TYPES_H_FTAG; print("%s", s); }

#ifndef _WIN32
void print(size_t i) { TYPES_H_FTAG; print("%u", i); }
void println(size_t i) { TYPES_H_FTAG; print(i); print("\n"); }
#endif

void println(bool b)         { TYPES_H_FTAG; print(b); print("\n"); }
void println(char c)         { TYPES_H_FTAG; print(c); print("\n"); }
void println(signed int i)   { TYPES_H_FTAG; print(i); print("\n"); }
void println(unsigned int i) { TYPES_H_FTAG; print(i); print("\n"); }
void println(float f)        { TYPES_H_FTAG; print(f); print("\n"); }
void println(double d)       { TYPES_H_FTAG; print(d); print("\n"); }
void println(void* p)        { TYPES_H_FTAG; print(p); print("\n"); }
void println(char* s)        { TYPES_H_FTAG; print(s); print("\n"); }
void println()               { TYPES_H_FTAG;           print("\n"); }

#ifdef _USING_GLM_TYPES__
void print(glm::vec<2, float, (glm::qualifier) 3> v)      { TYPES_H_FTAG; print("vec2: %.14g,%.14g", v.x, v.y); }
void print(glm::vec<3, float, (glm::qualifier) 3> v)      { TYPES_H_FTAG; print("vec3: %.14g,%.14g,%.14g", v.x, v.y, v.z); }
void print(glm::vec<4, float, (glm::qualifier) 3> v)      { TYPES_H_FTAG; print("vec4: %.14g,%.14g,%.14g,%.14g", v.x, v.y, v.z, v.w); }
void print(glm::mat<2, 2, float, (glm::qualifier) 3> m)   { TYPES_H_FTAG; print("mat2: "); print(m[0]); print(m[1]); }
void print(glm::mat<3, 3, float, (glm::qualifier) 3> m)   { TYPES_H_FTAG; print("mat3: "); print(m[0]); print(m[1]); print(m[2]); }
void print(glm::mat<4, 4, float, (glm::qualifier) 3> m)   { TYPES_H_FTAG; print("mat4: "); print(m[0]); print(m[1]); print(m[2]); print(m[3]); }

void println(glm::vec<2, float, (glm::qualifier) 3> v)    { TYPES_H_FTAG; print(v); print("\n"); }
void println(glm::vec<3, float, (glm::qualifier) 3> v)    { TYPES_H_FTAG; print(v); print("\n"); }
void println(glm::vec<4, float, (glm::qualifier) 3> v)    { TYPES_H_FTAG; print(v); print("\n"); }
void println(glm::mat<2, 2, float, (glm::qualifier) 3> m) { TYPES_H_FTAG; print(m); print("\n"); }
void println(glm::mat<3, 3, float, (glm::qualifier) 3> m) { TYPES_H_FTAG; print(m); print("\n"); }
void println(glm::mat<4, 4, float, (glm::qualifier) 3> m) { TYPES_H_FTAG; print(m); print("\n"); }
#endif