/*
 * Console replacement routines and miscellaneous graphics routines.
 *
 * Copyright (c) 2019 Dr. D'nar
 * SPDX-License-Identifier: BSD-2-clause
 */

#include <graphx.h>
#include <compression.h>
#include <fontlibc.h>

#include "dungeon.h"
#include "calc.h"
#include "style.h"
#include "splash.h"

unsigned char foreground_color = gfx_white;
unsigned char background_color = gfx_black;


/**
 * Switching to the homescreen erases the cached copy of the decompressed splash
 * graphic; call this routine to switch back to GraphX mode and decompress the
 * splash graphic again.
 */
void gfx_resume_render_splash(void)
{
    gfx_sprite_t* splash;
    gfx_Begin();
    gfx_SetPalette(splash_pal, sizeof_splash_pal, 0);
    gfx_SetDraw(gfx_buffer);
    splash = gfx_AllocSprite(splashl_width, splashl_height, &malloc_safe);
    zx7_Decompress(splash, splashl_data);
    gfx_Sprite_NoClip(splash, 0, 0);
    free(splash);
    splash = gfx_AllocSprite(splashr_width, splashr_height, &malloc_safe);
    zx7_Decompress(splash, splashr_data);
    gfx_Sprite_NoClip(splash, splashl_width, 0);
    free(splash);
    asm("; Quick and dirty vertical mirror routine");
    asm("	ld	hl, 0D52C00h + (320 * 129)");
    asm("	ld	de, 0D52C00h + (320 * 130)");
    asm("	ld	a, 110");
    asm("_main_mirror_loop:");
    asm("	ld	bc, 320");
    asm("	ldir");
    asm("	ld	bc, -(320 * 2)");
    asm("	add	hl, bc");
    asm("	dec	a");
    asm("	jr	nz, _main_mirror_loop");
    gfx_SetDraw(gfx_screen);
}


/*******************************************************************************
 * Font Stuff
 ******************************************************************************/

char* times_pack_name = "Times";
char* drsans_pack_name = "DrSans";

/**
 * Display an error about a missing font file and terminate.
 */
void font_missing(char* name)
{
    gfx_SetTextFGColor(gfx_red);
    gfx_SetTextBGColor(gfx_black);
    gfx_PrintStringXY(get_arbitrary_message(BAD_ERROR), 1, 1);
    gfx_PrintStringXY(name, 1, 10);
    gfx_PrintString(get_arbitrary_message(MISSING_FONT));
        while (!os_GetCSC());
    exit_clean(1);
}

/**
 * Select a font with given properties.
 * Display an error and terminate if the font cannot be located.
 * @param weight If 0, choose any weight; otherwise, accept ONLY the given weight
 */
fontlib_font_t* set_font(char* name, uint8_t size, uint8_t weight, uint8_t style_set, uint8_t style_reset, fontlib_load_options_t options)
{
    fontlib_font_t* font;
    font = fontlib_GetFontByStyle(name, size, size, weight, weight > 0 ? weight : 0xFF, style_set, style_reset);
    if (font && fontlib_SetFont(font, options))
        return font;
    font_missing(name);
    /* Make ZDS shutup about no return value. */
    return NULL;
}

/**
 * Select the Times font.
 */
fontlib_font_t* set_times(uint8_t size, fontlib_load_options_t options)
{
    return set_font(times_pack_name, size, 0, 0, 0, options);
}

/**
 * Select the Dr. Sans font.
 */
fontlib_font_t* set_drsans(uint8_t size, uint8_t weight, fontlib_load_options_t options)
{
    return set_font(drsans_pack_name, size, weight, 0, 0, options);
}


/*******************************************************************************
 * Some Text Drawing Routines
 ******************************************************************************/

/**
 * Does fontlib_DrawString, but using the given arbitrary message reference.
 * Not part of the console replacement routines.
 */
void draw_compressed(int message)
{
    char* string = get_arbitrary_message(message);
    fontlib_DrawString(string);
}

/**
 * Prints a string, centering it in the current text window.
 * Not part of the console replacement routines.
 */
void print_centered(const char* string)
{
    fontlib_SetCursorPosition(fontlib_GetWindowWidth() / 2 + fontlib_GetWindowXMin() - (fontlib_GetStringWidth(string) / 2), fontlib_GetCursorY());
    fontlib_DrawString(string);
}

/**
 * Prints a comprssed string centered.
 * Not part of the console replacement routines.
 */
void print_centered_compressed(int message)
{
    print_centered(get_arbitrary_message(message));
}

/**
 * Prints a compressed string right-aligned.
 * Not part of the console replacement routines.
 */
void print_right(int message)
{
    char* string = get_arbitrary_message(message);
    fontlib_SetCursorPosition(fontlib_GetWindowWidth() + fontlib_GetWindowXMin() - fontlib_GetStringWidth(string), fontlib_GetCursorY());
    fontlib_DrawString(string);
}


/*******************************************************************************
 * Console Replacement
 ******************************************************************************/
unsigned char print_lines_printed;
char* print_font_pack_name;
unsigned char print_size, print_weight, print_lines;
fontlib_load_options_t print_options;
unsigned char print_window_height;
unsigned int print_cursor_x;
unsigned char print_cursor_y;

/**
 * Resets the count of lines printed for pagination purposes.
 */
void print_reset_pagination(void)
{
    print_lines_printed = 0;
}

/**
 * Prints a string, but with word wrap!
 * This prints one line at a time; call it in a loop.
 * @param string Text to print
 * @param fake_print Set to true to perform the exact same layout logic, but
 * without actually printing anything.  This allows finding word wrap points.
 * @note fake_print DOES care what the current cursor X position is---it uses
 * that to figure out how to deal with words too big to fit into the text
 * window.  Such words will get force-printed starting on their own line.
 * @return Returns a pointer to the last character processed, which will either
 * be '\0', a control code (such as newline), or the first character of the next
 * line of text.
 */
char* print_word_wrap(const char* string, bool fake_print)
{
    char old_stop = fontlib_GetAlternateStopCode();
    unsigned int left = fontlib_GetWindowXMin();
    unsigned int width = fontlib_GetWindowWidth();
    unsigned int right = left + width;
    unsigned int str_width;
    unsigned int x = fontlib_GetCursorX();
    unsigned char first_printable = (unsigned char)fontlib_GetFirstPrintableCodePoint();
    unsigned char c;
    unsigned int space_width = fontlib_GetGlyphWidth(' ');
    if (first_printable == '\0')
        first_printable = '\1';
    fontlib_SetAlternateStopCode(' ');
    do
    {
        /* Check if the next word can fit on the current line */
        str_width = fontlib_GetStringWidth(string);
        if (x + str_width < right)
            if (!fake_print)
                x = fontlib_DrawString(string);
            else
                x += str_width;
        else
        {
            /* If the word is super-long such that it won't fit in the window,
             * then forcibly print it starting on a new line. */
            if (str_width != 0)
            {
                if (str_width > width && x == left)
                    if (!fake_print)    
                        x = fontlib_DrawString(string);
                    else
                    {
                        do
                            x += (str_width = fontlib_GetGlyphWidth(*string++));
                        while (x < right);
                        string--;
                        break;
                    }
                else
                    break;
            }
            /* If the width returned was zero, that means either another space
             * is waiting to be printed, which will be handled below; or a
             * control code is next, which also will be handled below.  This can
             * occur, for example, if a control code immediately follows a
             * space. */
        }
        /* FontLibC will kindly tell us exactly where it left off. */
        string = fontlib_GetLastCharacterRead();
        /* Now we need to deal with why the last word was terminated. */
        c = (unsigned char)(*string);
        /* If it's a control code, we either process a newline or give up. */
        if (c < first_printable)
        {
            if (c == ZERO_WIDTH_SPACE)
                string++;
            else if (c == '\t')
            {
                string++;
                x += 16;
                x &= 0xFFFFF0;
                if (!fake_print)
                {
                    fontlib_ClearEOL();
                    fontlib_SetCursorPosition(x, fontlib_GetCursorY());
                }
            }
            else
                break;
        }
        /* If it's a space, we need to process that manually since DrawString
         * won't handle it because we set space as a stop code. */
        if (c == ' ')
        {
            string++;
            /* We do actually need to check if there's space to print the
             * space. */
            if (x + space_width < right)
            {
                if (!fake_print)
                    fontlib_DrawGlyph(' ');
                x += space_width;
            }
            else
            {
                /* If there isn't room, we need to eat the space; it would look
                 * weird to print a space at the start of the next line. 
                 * However, we do not eat ALL the spaces if there's more than
                 * one, just the first one or two. */
                if (*string == ' ') /* Take care of possible second space. */
                    string++;
                break;
            }
        }
    } while (true);
    if (!fake_print)
        fontlib_ClearEOL();
    fontlib_SetAlternateStopCode(old_stop);
    return string;
}

/**
 * Configures the console replacement.
 * This is hard-coded to accept input at the bottom of the screen.
 */
void print_configure(char* name, uint8_t size, uint8_t weight, fontlib_load_options_t options)
{
    uint8_t fheight;
    print_font_pack_name = name;
    print_size = size;
    print_weight = weight;
    print_options = options;
    set_font(name, size, weight, 0, 0, options);
    fheight = fontlib_GetCurrentFontHeight();
    /* Make the window one line smaller so there's room for the text entry box
     * at the bottom. */
    print_lines = ((LCD_HEIGHT / fheight) - 1);
    print_window_height = fheight * print_lines;
}

/**
 * Internal routine called before printing to the console replacement.
 * Configures the text window, cursor, and font.
 */
static void print_setup_internal(void)
{
    fontlib_SetWindow(0, 0, LCD_WIDTH, print_window_height);
    set_font(print_font_pack_name, print_size, print_weight, 0, 0, print_options);
    fontlib_SetCursorPosition(print_cursor_x, print_cursor_y);
    fontlib_SetNewlineOptions(FONTLIB_AUTO_CLEAR_TO_EOL | FONTLIB_AUTO_SCROLL | FONTLIB_ENABLE_AUTO_WRAP);
}

/**
 * Internal routine called after printing to the console replacement.
 * Saves cursor position.
 */
static void print_cleanup_internal(void)
{
    print_cursor_x = fontlib_GetCursorX();
    print_cursor_y = fontlib_GetCursorY();
}

/**
 * Clears the console-replacement window of all text.
 */
void print_clear(void)
{
    print_setup_internal();
    fontlib_ClearWindow();
    /*fontlib_HomeUp();*/
    print_reset_pagination();
    fontlib_SetCursorPosition(0, print_window_height - fontlib_GetCurrentFontHeight());
    print_cleanup_internal();
}

/**
 * Prints a newline in the console replacement.
 * Pauses if the console is full.
 */
void print_newline(void)
{
    if (++print_lines_printed >= print_lines)
    {
        print_reset_pagination();
        wait_any_key_msg(paginate_message);
    }
    print_setup_internal();
    fontlib_Newline();
    print_cleanup_internal();
}

/**
 * Prints an arbitrary message to the console replacement.
 */
void print_compressed(int message)
{
    char* string = get_arbitrary_message(message);
    print(string);
}

/**
 * Prints the given string to the console replacement.
 * Control codes are processed.
 */
void print(const char* string)
{
    unsigned char next;
    char* stop;
    unsigned int len, width;
    bool centering = false;
    char old_newline = fontlib_GetNewlineCode();
    fontlib_SetNewlineCode('\0');
    do
    {
        print_setup_internal();
        if (centering)
        {
            stop = print_word_wrap(string, true);
            len = stop - string;
            width = fontlib_GetStringWidthL(string, len);
            fontlib_ClearEOL();
            fontlib_ShiftCursorPosition((fontlib_GetWindowWidth() - width) / 2, 0);
        }
        string = print_word_wrap(string, false);
        print_cleanup_internal();
        next = (unsigned char)*string;
        if (next < (unsigned char)fontlib_GetFirstPrintableCodePoint())
        {
            if (next == FORMAT_CENTERING)
            {
                string++;
                centering = true;
            }
            else if (next == '\n')
            {
                string++;
                print_newline();
                centering = false;
            }
            else
                break;
        }
        else
            print_newline();
    } while (true);
    fontlib_SetNewlineCode(old_newline);
}