#include "tim.h"

TimCell tim_cell(cstr s, u8 fg, u8 bg) {
    TimCell c = {.fg = fg, .bg  = bg, .n = 1, .buf = {s[0]}};
    while ((s[c.n] & 192) == 128 && c.n < sizeof(c.buf)) {
        c.buf[c.n] = s[c.n];
        c.n += 1;
    }
    return c;
}

void tim_clear_cells(void) {
    size_t size = sizeof(tim->cells[0]) * tim->w * tim->h;
    memset(tim->cells, 0, size);
}

void tim_draw_chr(TimCell cell, i32 x, i32 y) {
    if (x >= 0 && x < tim->w && y >= 0 && y < tim->h) {
        tim->cells[x + y * tim->w] = cell;
    }
}

void tim_draw_row(TimCell cell, i32 x, i32 y, i32 w) {
    if (y >= 0 && y < tim->h && w > 0) {
        for (i32 i = MAX(x, 0); i < MIN(x + w, tim->w); i++) {
            tim->cells[i + y * tim->w] = cell;
        }
    }
}

void tim_draw_col(TimCell cell, i32 x, i32 y, i32 h) {
    if (x >= 0 && x < tim->w && h > 0) {
        for (i32 i = MAX(y, 0); i < MIN(y + h, tim->h); i++) {
            tim->cells[x + i * tim->w] = cell;
        }
    }
}

void tim_fill(TimCell cell, i32 x, i32 y, i32 w, i32 h) {
    if (w > 0 && h > 0) {
        for (i32 iy = MAX(y, 0); iy < MIN(y + h, tim->h); iy++) {
            for (i32 ix = MAX(x, 0); ix < MIN(x + w, tim->w); ix++) {
                tim->cells[ix + iy * tim->w] = cell;
            }
        }
    }
}

void tim_draw_str(cstr s, i32 x, i32 y, i32 w, u8 fg, u8 bg) {
    if (s && y >= 0 && x < tim->w && y < tim->h ) {
        i32  end  = MIN(x + w, tim->w);
        bool wide = false;
        for (i32 i = 0; s[i] && x < end; x++) {
            TimCell c = tim_cell(&s[i], fg, bg);
            wide = wide || tim_utf8_is_wide_perhaps(c.buf, c.n);
            if (x >= 0) {
                c.wide = wide;
                tim->cells[x + y * tim->w] = c;
            }
            i += c.n;
        }
    }
}

void tim_draw_box(i32 x, i32 y, i32 w, i32 h, u8 fg, u8 bg) {
    tim_draw_chr(tim_cell("┌", fg, bg), x        , y);
    tim_draw_chr(tim_cell("┐", fg, bg), x + w - 1, y);
    tim_draw_chr(tim_cell("└", fg, bg), x        , y + h - 1);
    tim_draw_chr(tim_cell("┘", fg, bg), x + w - 1, y + h - 1);
    tim_draw_row(tim_cell("─", fg, bg), x + 1    , y        , w - 2);
    tim_draw_row(tim_cell("─", fg, bg), x + 1    , y + h - 1, w - 2);
    tim_draw_col(tim_cell("│", fg, bg), x        , y + 1    , h - 2);
    tim_draw_col(tim_cell("│", fg, bg), x + w - 1, y + 1    , h - 2);
    tim_fill(tim_cell(" ", fg, bg), x + 1    , y + 1    , w - 2, h - 2);
}

void tim_draw_invert(i32 x, i32 y, i32 w) {
    if (y >= 0 && y < tim->h && w > 0) {
        for (i32 i = MAX(x, 0); i < MIN(x + w, tim->w); i++) {
            TimCell c = tim->cells[i + y * tim->w];
            tim->cells[i + y * tim->w].fg = c.bg;
            tim->cells[i + y * tim->w].bg = c.fg;
        }
    }
}