/*
 *  demo          demo for libcaca
 *  Copyright (c) 2003 Sam Hocevar <sam@zoy.org>
 *                All Rights Reserved
 *
 *  $Id$
 *
 *  This program is free software; you can redistribute it and/or
 *  modify it under the terms of the Do What The Fuck You Want To
 *  Public License, Version 2, as published by Sam Hocevar. See
 *  http://sam.zoy.org/wtfpl/COPYING for more details.
 */

#include "config.h"

#include <math.h>
#include <string.h>
#include <stdio.h>

#include "caca.h"

static void display_menu(void);

static void demo_all(void);

static void demo_color(void);
static void demo_dots(void);
static void demo_lines(void);
static void demo_boxes(void);
static void demo_ellipses(void);
static void demo_triangles(void);
static void demo_sprites(void);
static void demo_render(void);

int bounds = 0;
int outline = 0;
int dithering = 0;
struct cucul_sprite *sprite = NULL;

cucul_t *qq;
caca_t *kk;

int main(int argc, char **argv)
{
    void (*demo)(void) = NULL;
    int quit = 0;

    qq = cucul_create(0, 0);
    if(!qq)
        return 1;
    kk = caca_attach(qq);
    if(!kk)
        return 1;

    caca_set_delay(kk, 40000);

    /* Initialize data */
    sprite = cucul_load_sprite(qq, DATADIR "/caca.txt");
    if(!sprite)
        sprite = cucul_load_sprite(qq, "caca.txt");
    if(!sprite)
        sprite = cucul_load_sprite(qq, "examples/caca.txt");

    /* Main menu */
    display_menu();
    caca_display(kk);

    /* Go ! */
    while(!quit)
    {
        int menu = 0, mouse = 0, xmouse = 0, ymouse = 0;
        int event;

        while((event = caca_get_event(kk, CACA_EVENT_ANY)))
        {
            if(demo && (event & CACA_EVENT_KEY_PRESS))
            {
                menu = 1;
                demo = NULL;
            }
            else if(event & CACA_EVENT_KEY_PRESS)
            {
                switch(event & 0xffff)
                {
                case 'q':
                case 'Q':
                    demo = NULL;
                    quit = 1;
                    break;
                case 'o':
                case 'O':
                    outline = (outline + 1) % 3;
                    display_menu();
                    break;
                case 'b':
                case 'B':
                    bounds = (bounds + 1) % 2;
                    display_menu();
                    break;
                case 'd':
                case 'D':
                    dithering = (dithering + 1) % 5;
                    cucul_set_feature(qq, dithering);
                    display_menu();
                    break;
                case 'c':
                    demo = demo_color;
                    break;
                case 'f':
                case 'F':
                    demo = demo_all;
                    break;
                case '1':
                    demo = demo_dots;
                    break;
                case '2':
                    demo = demo_lines;
                    break;
                case '3':
                    demo = demo_boxes;
                    break;
                case '4':
                    demo = demo_triangles;
                    break;
                case '5':
                    demo = demo_ellipses;
                    break;
                case 's':
                case 'S':
                    if(sprite)
                        demo = demo_sprites;
                    break;
                case 'r':
                case 'R':
                    demo = demo_render;
                    break;
                }

                if(demo)
                    cucul_clear(qq);
            }
            else if(event & CACA_EVENT_MOUSE_MOTION)
            {
                mouse = 1;
                xmouse = (event & 0xfff000) >> 12;
                ymouse = event & 0xfff;
            }
        }

        if(menu || (mouse && !demo))
        {
            display_menu();
            if(mouse && !demo)
            {
                cucul_set_color(qq, CUCUL_COLOR_RED, CUCUL_COLOR_BLACK);
                cucul_putstr(qq, xmouse, ymouse, "|\\");
            }
            caca_display(kk);
            mouse = menu = 0;
        }

        if(demo)
        {
            demo();

            cucul_set_color(qq, CUCUL_COLOR_LIGHTGRAY, CUCUL_COLOR_BLACK);
            cucul_draw_thin_box(qq, 1, 1, cucul_get_width(qq) - 2, cucul_get_height(qq) - 2);
            cucul_printf(qq, 4, 1, "[%i.%i fps]----",
                         1000000 / caca_get_rendertime(kk),
                         (10000000 / caca_get_rendertime(kk)) % 10);
            caca_display(kk);
        }
    }

    /* Clean up */
    cucul_free_sprite(sprite);
    caca_detach(kk);
    cucul_free(qq);

    return 0;
}

static void display_menu(void)
{
    int xo = cucul_get_width(qq) - 2;
    int yo = cucul_get_height(qq) - 2;

    cucul_clear(qq);
    cucul_set_color(qq, CUCUL_COLOR_LIGHTGRAY, CUCUL_COLOR_BLACK);
    cucul_draw_thin_box(qq, 1, 1, xo, yo);

    cucul_putstr(qq, (xo - strlen("libcaca demo")) / 2, 3, "libcaca demo");
    cucul_putstr(qq, (xo - strlen("==============")) / 2, 4, "==============");

    cucul_putstr(qq, 4, 6, "demos:");
    cucul_putstr(qq, 4, 7, "'f': full");
    cucul_putstr(qq, 4, 8, "'1': dots");
    cucul_putstr(qq, 4, 9, "'2': lines");
    cucul_putstr(qq, 4, 10, "'3': boxes");
    cucul_putstr(qq, 4, 11, "'4': triangles");
    cucul_putstr(qq, 4, 12, "'5': ellipses");
    cucul_putstr(qq, 4, 13, "'c': colour");
    cucul_putstr(qq, 4, 14, "'r': render");
    if(sprite)
        cucul_putstr(qq, 4, 15, "'s': sprites");

    cucul_putstr(qq, 4, 16, "settings:");
    cucul_printf(qq, 4, 17, "'o': outline: %s",
                 outline == 0 ? "none" : outline == 1 ? "solid" : "thin");
    cucul_printf(qq, 4, 18, "'b': drawing boundaries: %s",
                 bounds == 0 ? "screen" : "infinite");
    cucul_printf(qq, 4, 19, "'d': dithering (%s)",
                 cucul_get_feature_name(dithering));

    cucul_putstr(qq, 4, yo - 2, "'q': quit");

    caca_display(kk);
}

static void demo_all(void)
{
    static int i = 0;

    int j, xo, yo, xa, ya, xb, yb, xc, yc;

    i++;

    cucul_clear(qq);

    /* Draw the sun */
    cucul_set_color(qq, CUCUL_COLOR_YELLOW, CUCUL_COLOR_BLACK);
    xo = cucul_get_width(qq) / 4;
    yo = cucul_get_height(qq) / 4 + 5 * sin(0.03*i);

    for(j = 0; j < 16; j++)
    {
        xa = xo - (30 + sin(0.03*i) * 8) * sin(0.03*i + M_PI*j/8);
        ya = yo + (15 + sin(0.03*i) * 4) * cos(0.03*i + M_PI*j/8);
        cucul_draw_thin_line(qq, xo, yo, xa, ya);
    }

    j = 15 + sin(0.03*i) * 8;
    cucul_set_color(qq, CUCUL_COLOR_WHITE, CUCUL_COLOR_BLACK);
    cucul_fill_ellipse(qq, xo, yo, j, j / 2, "#");
    cucul_set_color(qq, CUCUL_COLOR_YELLOW, CUCUL_COLOR_BLACK);
    cucul_draw_ellipse(qq, xo, yo, j, j / 2, "#");

    /* Draw the pyramid */
    xo = cucul_get_width(qq) * 5 / 8;
    yo = 2;

    xa = cucul_get_width(qq) / 8 + sin(0.03*i) * 5;
    ya = cucul_get_height(qq) / 2 + cos(0.03*i) * 5;

    xb = cucul_get_width(qq) - 10 - cos(0.02*i) * 10;
    yb = cucul_get_height(qq) * 3 / 4 - 5 + sin(0.02*i) * 5;

    xc = cucul_get_width(qq) / 4 - sin(0.02*i) * 5;
    yc = cucul_get_height(qq) * 3 / 4 + cos(0.02*i) * 5;

    cucul_set_color(qq, CUCUL_COLOR_GREEN, CUCUL_COLOR_BLACK);
    cucul_fill_triangle(qq, xo, yo, xb, yb, xa, ya, "%");
    cucul_set_color(qq, CUCUL_COLOR_YELLOW, CUCUL_COLOR_BLACK);
    cucul_draw_thin_triangle(qq, xo, yo, xb, yb, xa, ya);

    cucul_set_color(qq, CUCUL_COLOR_RED, CUCUL_COLOR_BLACK);
    cucul_fill_triangle(qq, xa, ya, xb, yb, xc, yc, "#");
    cucul_set_color(qq, CUCUL_COLOR_YELLOW, CUCUL_COLOR_BLACK);
    cucul_draw_thin_triangle(qq, xa, ya, xb, yb, xc, yc);

    cucul_set_color(qq, CUCUL_COLOR_BLUE, CUCUL_COLOR_BLACK);
    cucul_fill_triangle(qq, xo, yo, xb, yb, xc, yc, "%");
    cucul_set_color(qq, CUCUL_COLOR_YELLOW, CUCUL_COLOR_BLACK);
    cucul_draw_thin_triangle(qq, xo, yo, xb, yb, xc, yc);

    /* Draw a background triangle */
    xa = 2;
    ya = 2;

    xb = cucul_get_width(qq) - 3;
    yb = cucul_get_height(qq) / 2;

    xc = cucul_get_width(qq) / 3;
    yc = cucul_get_height(qq) - 3;

    cucul_set_color(qq, CUCUL_COLOR_CYAN, CUCUL_COLOR_BLACK);
    cucul_draw_thin_triangle(qq, xa, ya, xb, yb, xc, yc);

    xo = cucul_get_width(qq) / 2 + cos(0.027*i) * cucul_get_width(qq) / 3;
    yo = cucul_get_height(qq) / 2 - sin(0.027*i) * cucul_get_height(qq) / 2;

    cucul_draw_thin_line(qq, xa, ya, xo, yo);
    cucul_draw_thin_line(qq, xb, yb, xo, yo);
    cucul_draw_thin_line(qq, xc, yc, xo, yo);

    /* Draw a sprite on the pyramid */
    cucul_draw_sprite(qq, xo, yo, sprite, 0);

    /* Draw a trail behind the foreground sprite */
    for(j = i - 60; j < i; j++)
    {
        int delta = cucul_rand(-5, 5);
        cucul_set_color(qq, cucul_rand(0, 15), cucul_rand(0, 15));
        cucul_putchar(qq, cucul_get_width(qq) / 2
                    + cos(0.02*j) * (delta + cucul_get_width(qq) / 4),
                   cucul_get_height(qq) / 2
                    + sin(0.02*j) * (delta + cucul_get_height(qq) / 3),
                   '#');
    }

    /* Draw foreground sprite */
    cucul_draw_sprite(qq, cucul_get_width(qq) / 2 + cos(0.02*i) * cucul_get_width(qq) / 4,
                   cucul_get_height(qq) / 2 + sin(0.02*i) * cucul_get_height(qq) / 3,
                   sprite, 0);
}

static void demo_dots(void)
{
    int xmax = cucul_get_width(qq) - 1;
    int ymax = cucul_get_height(qq) - 1;
    int i;
    static char chars[10] =
    {
        '+', '-', '*', '#', 'X', '@', '%', '$', 'M', 'W'
    };

    for(i = 1000; i--;)
    {
        /* Putpixel */
        cucul_set_color(qq, cucul_rand(0, 15), cucul_rand(0, 15));
        cucul_putchar(qq, cucul_rand(0, xmax), cucul_rand(0, ymax),
                      chars[cucul_rand(0, 9)]);
    }
}

static void demo_color(void)
{
    int i, j;
    char buf[BUFSIZ];

    cucul_clear(qq);
    for(i = 0; i < 16; i++)
    {
        sprintf(buf, "'%c': %i (%s)", 'a' + i, i, cucul_get_color_name(i));
        cucul_set_color(qq, CUCUL_COLOR_LIGHTGRAY, CUCUL_COLOR_BLACK);
        cucul_putstr(qq, 4, i + (i >= 8 ? 4 : 3), buf);
        for(j = 0; j < 16; j++)
        {
            cucul_set_color(qq, i, j);
            cucul_putstr(qq, (j >= 8 ? 41 : 40) + j * 2, i + (i >= 8 ? 4 : 3), "# ");
        }
    }
}

static void demo_lines(void)
{
    int w = cucul_get_width(qq);
    int h = cucul_get_height(qq);
    int xa, ya, xb, yb;

    if(bounds)
    {
        xa = cucul_rand(- w, 2 * w); ya = cucul_rand(- h, 2 * h);
        xb = cucul_rand(- w, 2 * w); yb = cucul_rand(- h, 2 * h);
    }
    else
    {
        xa = cucul_rand(0, w - 1); ya = cucul_rand(0, h - 1);
        xb = cucul_rand(0, w - 1); yb = cucul_rand(0, h - 1);
    }

    cucul_set_color(qq, cucul_rand(0, 15), CUCUL_COLOR_BLACK);
    if(outline > 1)
        cucul_draw_thin_line(qq, xa, ya, xb, yb);
    else
        cucul_draw_line(qq, xa, ya, xb, yb, "#");
}

static void demo_boxes(void)
{
    int w = cucul_get_width(qq);
    int h = cucul_get_height(qq);
    int xa, ya, xb, yb;

    if(bounds)
    {
        xa = cucul_rand(- w, 2 * w); ya = cucul_rand(- h, 2 * h);
        xb = cucul_rand(- w, 2 * w); yb = cucul_rand(- h, 2 * h);
    }
    else
    {
        xa = cucul_rand(0, w - 1); ya = cucul_rand(0, h - 1);
        xb = cucul_rand(0, w - 1); yb = cucul_rand(0, h - 1);
    }

    cucul_set_color(qq, cucul_rand(0, 15), cucul_rand(0, 15));
    cucul_fill_box(qq, xa, ya, xb, yb, "#");

    cucul_set_color(qq, cucul_rand(0, 15), CUCUL_COLOR_BLACK);
    if(outline == 2)
        cucul_draw_thin_box(qq, xa, ya, xb, yb);
    else if(outline == 1)
        cucul_draw_box(qq, xa, ya, xb, yb, "#");
}

static void demo_ellipses(void)
{
    int w = cucul_get_width(qq);
    int h = cucul_get_height(qq);
    int x, y, a, b;

    if(bounds)
    {
        x = cucul_rand(- w, 2 * w); y = cucul_rand(- h, 2 * h);
        a = cucul_rand(0, w); b = cucul_rand(0, h);
    }
    else
    {
        do
        {
            x = cucul_rand(0, w); y = cucul_rand(0, h);
            a = cucul_rand(0, w); b = cucul_rand(0, h);

        } while(x - a < 0 || x + a >= w || y - b < 0 || y + b >= h);
    }

    cucul_set_color(qq, cucul_rand(0, 15), cucul_rand(0, 15));
    cucul_fill_ellipse(qq, x, y, a, b, "#");

    cucul_set_color(qq, cucul_rand(0, 15), CUCUL_COLOR_BLACK);
    if(outline == 2)
        cucul_draw_thin_ellipse(qq, x, y, a, b);
    else if(outline == 1)
        cucul_draw_ellipse(qq, x, y, a, b, "#");
}

static void demo_triangles(void)
{
    int w = cucul_get_width(qq);
    int h = cucul_get_height(qq);
    int xa, ya, xb, yb, xc, yc;

    if(bounds)
    {
        xa = cucul_rand(- w, 2 * w); ya = cucul_rand(- h, 2 * h);
        xb = cucul_rand(- w, 2 * w); yb = cucul_rand(- h, 2 * h);
        xc = cucul_rand(- w, 2 * w); yc = cucul_rand(- h, 2 * h);
    }
    else
    {

        xa = cucul_rand(0, w - 1); ya = cucul_rand(0, h - 1);
        xb = cucul_rand(0, w - 1); yb = cucul_rand(0, h - 1);
        xc = cucul_rand(0, w - 1); yc = cucul_rand(0, h - 1);
    }

    cucul_set_color(qq, cucul_rand(0, 15), cucul_rand(0, 15));
    cucul_fill_triangle(qq, xa, ya, xb, yb, xc, yc, "#");

    cucul_set_color(qq, cucul_rand(0, 15), CUCUL_COLOR_BLACK);
    if(outline == 2)
        cucul_draw_thin_triangle(qq, xa, ya, xb, yb, xc, yc);
    else if(outline == 1)
        cucul_draw_triangle(qq, xa, ya, xb, yb, xc, yc, "#");
}

static void demo_sprites(void)
{
    cucul_draw_sprite(qq, cucul_rand(0, cucul_get_width(qq) - 1),
                   cucul_rand(0, cucul_get_height(qq) - 1), sprite, 0);
}

#if 0
static void demo_render(void)
{
    struct cucul_bitmap *bitmap;
    //short buffer[256*256];
    //short *dest = buffer;
    int buffer[256*256];
    int *dest = buffer;
    int x, y, z;
    static int i = 0;

    i = (i + 1) % 512;
    z = i < 256 ? i : 511 - i;

    for(x = 0; x < 256; x++)
        for(y = 0; y < 256; y++)
    {
        //*dest++ = ((x >> 3) << 11) | ((y >> 2) << 5) | ((z >> 3));
        *dest++ = (x << 16) | (y << 8) | (z);
    }

    //bitmap = cucul_create_bitmap(16, 256, 256, 2 * 256, 0xf800, 0x07e0, 0x001f, 0x0000);
    bitmap = cucul_create_bitmap(32, 256, 256, 4 * 256, 0x00ff0000, 0x0000ff00, 0x000000ff, 0xff000000);
    cucul_draw_bitmap(qq, 0, 0, cucul_get_width(qq) - 1, cucul_get_height(qq) - 1,
                     bitmap, buffer);
    cucul_free_bitmap(bitmap);
}
#endif

static void draw_circle(int *buffer, int xo, int yo, int r, int mask, int val);

static void demo_render(void)
{
    struct cucul_bitmap *bitmap;
    int buffer[256*256];
    int *dest;
    int x, y, z, xo, yo;
    static int i = 0;

    i++;

    dest = buffer;
    for(x = 0; x < 256; x++)
        for(y = 0; y < 256; y++)
    {
        *dest++ = 0xff000000;
    }

    /* red */
    xo = 128 + 48 * sin(0.02 * i);
    yo = 128 + 48 * cos(0.03 * i);
    for(z = 0; z < 240; z++)
        draw_circle(buffer, xo, yo, z, 0x00ff0000, 200 << 16);

    /* green */
    xo = 128 + 48 * sin(2 + 0.06 * i);
    yo = 128 + 48 * cos(2 + 0.05 * i);
    for(z = 0; z < 240; z++)
        draw_circle(buffer, xo, yo, z, 0x0000ff00, 200 << 8);

    /* blue */
    xo = 128 + 48 * sin(1 + 0.04 * i);
    yo = 128 + 48 * cos(1 + 0.03 * i);
    for(z = 0; z < 240; z++)
        draw_circle(buffer, xo, yo, z, 0x000000ff, 200);

    bitmap = cucul_create_bitmap(32, 256, 256, 4 * 256, 0x00ff0000, 0x0000ff00, 0x000000ff, 0xff000000);
    cucul_draw_bitmap(qq, 0, 0, cucul_get_width(qq) - 1, cucul_get_height(qq) - 1, bitmap, (char *)buffer);
    cucul_free_bitmap(bitmap);
}

static void draw_circle(int *buffer, int x, int y, int r, int mask, int val)
{
    int t, dx, dy;

#define POINT(X,Y) \
    buffer[(X) + 256 * (Y)] = 0xff000000 | (buffer[(X) + 256 * (Y)] & ~mask) | val;

    for(t = 0, dx = 0, dy = r; dx <= dy; dx++)
    {
        POINT(x - dx / 3, y - dy / 3);
        POINT(x + dx / 3, y - dy / 3);
        POINT(x - dx / 3, y + dy / 3);
        POINT(x + dx / 3, y + dy / 3);

        POINT(x - dy / 3, y - dx / 3);
        POINT(x + dy / 3, y - dx / 3);
        POINT(x - dy / 3, y + dx / 3);
        POINT(x + dy / 3, y + dx / 3);

        t += t > 0 ? dx - dy-- : dx;
    }
}