#include "r/r.h"
#include "u/button.h"
#include "u/pose.h"
#include "modal.h"
#include "animation.h"
#include "canvas.h"
#include "io.h"
#include "dialog.h"


static const uColor_s BG_A_COLOR = {{102, 136, 102, 255}};
static const uColor_s BG_B_COLOR = {{102, 143, 102, 255}};
static const vec4 TITLE_COLOR = {{0.6, 0.6, 0.2, 1}};
//
// private
//

typedef struct {
    RoText show_txt;
    RoSingle show;

    RoText size_txt;
    RoText size_num;
    mat4 size_hitbox;
    
    RoText repeat_txt;
    RoSingle repeat_h;
    RoSingle repeat_v;

    float time;    
    RoText time_txt;
    RoText time_num;
    mat4 time_hitbox;
    RoSingle time_sec_btn;
    RoSingle time_fps_btn;
    
    int textinput_reason;
    TextInput *textinput;
} Impl;

static void kill_fn() {
    Impl *impl = dialog.impl;
    
    ro_text_kill(&impl->show_txt);
    ro_single_kill(&impl->show);

    ro_text_kill(&impl->size_txt);
    ro_text_kill(&impl->size_num);
    
    ro_text_kill(&impl->repeat_txt);
    ro_single_kill(&impl->repeat_h);
    ro_single_kill(&impl->repeat_v);
    
    ro_text_kill(&impl->time_txt);
    ro_text_kill(&impl->time_num);
    ro_single_kill(&impl->time_sec_btn);
    ro_single_kill(&impl->time_fps_btn);
    
    textinput_kill(&impl->textinput);
    
    s_free(impl);
}

static void update(float dtime) {
    Impl *impl = dialog.impl;
    if(impl->textinput) {
        char *end;
        float value = (float) strtod(impl->textinput->text, &end);
        bool ok = end && end != impl->textinput->text
                && value>ANIMATION_SIZE_MIN && value<=ANIMATION_SIZE_MAX;
        
        impl->textinput->ok_active = ok;
        
        if (impl->textinput->state == TEXTINPUT_DONE) {
            if(impl->textinput_reason == 0) {
                s_log("set animation size: %f", value);
                animation.size = value;
                animation_save_config();
            } else if(impl->textinput_reason == 1) {
                s_log("set time field: %f", value);
                impl->time = value;
            }
        }
        if (impl->textinput->state != TEXTINPUT_IN_PROGRESS) {
            textinput_kill(&impl->textinput);
            modal_set_textinput(NULL);
        }
    }
    
    char buf[16];
    snprintf(buf, sizeof buf, "%.1f", animation.size);
    ro_text_set_text(&impl->size_num, buf);
    
    snprintf(buf, sizeof buf, "%.2f", impl->time);
    ro_text_set_text(&impl->time_num, buf);

    u_button_set_pressed(&impl->show.rect, animation.auto_show);

    u_button_set_pressed(&impl->repeat_h.rect, (animation.mode == ANIMATION_MODE_REPEAT_H
            || animation.mode == ANIMATION_MODE_REPEAT_HV) ? 1: 0);

    u_button_set_pressed(&impl->repeat_v.rect, (animation.mode == ANIMATION_MODE_REPEAT_V
            || animation.mode == ANIMATION_MODE_REPEAT_HV) ? 1: 0);

    animation.show = animation.auto_show;

}

static void render(const mat4 *cam_mat) {
    Impl *impl = dialog.impl;
    ro_text_render(&impl->show_txt, cam_mat);
    ro_single_render(&impl->show, cam_mat);
    ro_text_render(&impl->size_txt, cam_mat);
    ro_text_render(&impl->size_num, cam_mat);
    ro_text_render(&impl->repeat_txt, cam_mat);
    ro_single_render(&impl->repeat_h, cam_mat);
    ro_single_render(&impl->repeat_v, cam_mat);
    ro_text_render(&impl->time_txt, cam_mat);
    ro_text_render(&impl->time_num, cam_mat);
    ro_single_render(&impl->time_sec_btn, cam_mat);
    ro_single_render(&impl->time_fps_btn, cam_mat);
}

static bool pointer_event(ePointer_s pointer) {
    Impl *impl = dialog.impl;

    if(u_button_toggled(&impl->show.rect, pointer)) {
        s_log("animation auto_show");
        animation.auto_show = u_button_is_pressed(&impl->show.rect);
        animation_save_config();
    }
    
    if(pointer.id == 0 
            && pointer.action == E_POINTER_DOWN 
            && u_pose_aa_contains(impl->size_hitbox, pointer.pos.xy)) {
        s_log("edit animation size");
        impl->textinput = textinput_new("Size", 0);
        modal_set_textinput(impl->textinput);
        snprintf(impl->textinput->text, TEXTINPUT_MAX_CHARS, "%.1f", animation.size);
        impl->textinput->shiftstate = TEXTINPUT_SHIFT_ALT;
        impl->textinput_reason = 0;
    }
    
    if(pointer.id == 0 
            && pointer.action == E_POINTER_DOWN 
            && u_pose_aa_contains(impl->time_hitbox, pointer.pos.xy)) {
        s_log("edit time");
        impl->textinput = textinput_new("Time", 0);
        modal_set_textinput(impl->textinput);
        snprintf(impl->textinput->text, TEXTINPUT_MAX_CHARS, "%.2f", impl->time);
        impl->textinput->shiftstate = TEXTINPUT_SHIFT_ALT;
        impl->textinput_reason = 1;
    }
    
    if(u_button_toggled(&impl->repeat_h.rect, pointer)) {
        if(u_button_is_pressed(&impl->repeat_h.rect)) {
            if(animation.mode == ANIMATION_MODE_SINGLE) 
                animation.mode = ANIMATION_MODE_REPEAT_H;
            else
                animation.mode = ANIMATION_MODE_REPEAT_HV;
        } else {
            if(animation.mode == ANIMATION_MODE_REPEAT_H) 
                animation.mode = ANIMATION_MODE_SINGLE;
            else
                animation.mode = ANIMATION_MODE_REPEAT_V;
        }

        animation_save_config();
    }
    
    if(u_button_toggled(&impl->repeat_v.rect, pointer)) {
        if(u_button_is_pressed(&impl->repeat_v.rect)) {
            if(animation.mode == ANIMATION_MODE_SINGLE) 
                animation.mode = ANIMATION_MODE_REPEAT_V;
            else
                animation.mode = ANIMATION_MODE_REPEAT_HV;
        } else {
            if(animation.mode == ANIMATION_MODE_REPEAT_V) 
                animation.mode = ANIMATION_MODE_SINGLE;
            else
                animation.mode = ANIMATION_MODE_REPEAT_H;
        }

        animation_save_config();
    }
    
    if(u_button_clicked(&impl->time_sec_btn.rect, pointer)) {
        s_log("update all frame times sec: %f", impl->time);
        for(int i=0; i<CANVAS_MAX_FRAMES; i++)
            canvas.frame_times[i] = impl->time;
        canvas_save_config();
    }
    
    if(u_button_clicked(&impl->time_fps_btn.rect, pointer)) {
        s_log("update all frame times fps: %f", impl->time);
        float sec = 1.0/impl->time;
        for(int i=0; i<CANVAS_MAX_FRAMES; i++)
            canvas.frame_times[i] = sec;
        canvas_save_config();
    }
    
    return true;
}

static void on_action(bool ok) {
    dialog_hide();
}

//
// public
//

void dialog_create_animation() {
    dialog_hide();
    canvas_reload();
    s_log("create");
    Impl *impl = s_malloc0(sizeof *impl);
    dialog.impl = impl;
    
    int pos = 24;

    impl->show_txt = ro_text_new_font55(16);
    ro_text_set_text(&impl->show_txt, "show animation:");
    ro_text_set_color(&impl->show_txt, DIALOG_TEXT_COLOR);
    impl->show_txt.pose = u_pose_new(DIALOG_LEFT + 6, DIALOG_TOP - pos - 2, 1, 2);

    impl->show = ro_single_new(r_texture_new_file(2, 1, "res/button_ok.png"));
    impl->show.rect.pose = u_pose_new_aa(DIALOG_LEFT + DIALOG_WIDTH - 20, DIALOG_TOP - pos, 16, 16);

    pos += 24;

    impl->size_txt = ro_text_new_font55(5);
    ro_text_set_text(&impl->size_txt, "size:");
    ro_text_set_color(&impl->size_txt, DIALOG_TEXT_COLOR);
    impl->size_num = ro_text_new_font55(8);
    ro_text_set_color(&impl->size_num, DIALOG_TEXT_EDIT_COLOR);
    impl->size_txt.pose = u_pose_new(DIALOG_LEFT + 6, DIALOG_TOP - pos, 1, 2);
    impl->size_num.pose = u_pose_new(DIALOG_LEFT + 40, DIALOG_TOP - pos, 1, 2);
    impl->size_hitbox = u_pose_new_aa(DIALOG_LEFT, DIALOG_TOP - pos + 4, DIALOG_WIDTH, 10 + 8);
    
    pos += 16;
    
    impl->repeat_txt = ro_text_new_font55(8);
    ro_text_set_text(&impl->repeat_txt, "repeat:");
    
    ro_text_set_color(&impl->repeat_txt, DIALOG_TEXT_COLOR);
    impl->repeat_txt.pose = u_pose_new(DIALOG_LEFT + 6, DIALOG_TOP - pos - 2, 1, 2);

    impl->repeat_h = ro_single_new(r_texture_new_file(2, 2, "res/button_repeat.png"));
    impl->repeat_h.rect.pose = u_pose_new_aa(DIALOG_LEFT + DIALOG_WIDTH - 40, DIALOG_TOP - pos, 16, 16);
    
    impl->repeat_v = ro_single_new(impl->repeat_h.tex);
    impl->repeat_v.owns_tex = false;
    impl->repeat_v.rect.sprite.y = 1;
    impl->repeat_v.rect.pose = u_pose_new_aa(DIALOG_LEFT + DIALOG_WIDTH - 20, DIALOG_TOP - pos, 16, 16);
    
    pos += 24;
    
    impl->time = 0.25;
    impl->time_txt = ro_text_new_font55(32);
    ro_text_set_text(&impl->time_txt, "reset all \n"
            "frame times:");
    ro_text_set_color(&impl->time_txt, DIALOG_TEXT_COLOR);
    impl->time_num = ro_text_new_font55(8);
    ro_text_set_color(&impl->time_num, DIALOG_TEXT_EDIT_COLOR);
    impl->time_txt.pose = u_pose_new(DIALOG_LEFT + 6, DIALOG_TOP - pos - 2, 1, 2);
    impl->time_num.pose = u_pose_new(DIALOG_LEFT + 84, DIALOG_TOP - pos, 1, 2);
    impl->time_hitbox = u_pose_new_aa(DIALOG_LEFT, DIALOG_TOP - pos + 4, DIALOG_WIDTH, 10 + 8);
    
    impl->time_sec_btn = ro_single_new(r_texture_new_file(2, 2, "res/button_sec_fps.png"));
    impl->time_sec_btn.rect.pose = u_pose_new_aa(DIALOG_LEFT + 82, DIALOG_TOP - pos - 12 , 16, 16);
    impl->time_fps_btn = ro_single_new(impl->time_sec_btn.tex);
    impl->time_fps_btn.owns_tex = false;
    impl->time_fps_btn.rect.sprite.y = 1;
    impl->time_fps_btn.rect.pose = u_pose_new_aa(DIALOG_LEFT + 100, DIALOG_TOP - pos - 12, 16, 16);
    
    pos += 16;
    
    dialog.impl_height = pos;
    
    dialog_set_title("animation", TITLE_COLOR);
    dialog_set_bg_color(BG_A_COLOR, BG_B_COLOR);
    dialog.kill = kill_fn;
    dialog.update = update;
    dialog.render = render;
    dialog.pointer_event = pointer_event;
    dialog.opt_on_ok_cb = on_action;
}