#pragma once

#include <set>
#include <SDL2/SDL.h>
#include "widgets.h"

namespace Gui {

enum class Mode;
class Manager;

class Anim_Camera {
public:
    Anim_Camera(Vec2 dim) : dim(dim) {}

    Camera at(float t) const;
    void set(float t, const Camera& cam);

    Splines<Vec3, Quat, float, float> splines;

private:
    Vec2 dim;
};

class Animate {
public:
    Animate(Vec2 screen_dim) : 
        ui_camera(screen_dim), anim_camera(screen_dim), ui_render(screen_dim) {}

    void update_dim(Vec2 dim);
    bool keydown(Widgets& widgets, Undo& undo, Scene_ID sel, SDL_Keysym key);

    Vec3 selected_pos(Scene_Item& item);
    void end_transform(Undo& undo, Scene_Item& obj);
    void apply_transform(Widgets& widgets, Scene_Item& obj);
    bool select(Scene& scene, Widgets& widgets, Scene_ID selected, Scene_ID id, Vec3 cam, Vec2 spos, Vec3 dir);

    void render(Scene& scene, Scene_Maybe obj_opt, Widgets& widgets, Camera& cam);
    void timeline(Manager& manager, Undo& undo, Scene& scene, Scene_Maybe obj, Camera& user_cam);
    void UIsidebar(Manager& manager, Undo& undo, Scene_Maybe obj_opt, Camera& user_cam);
    
    void clear();
    void update(Scene& scene);
    void refresh(Scene& scene);

    std::string pump_output(Scene& scene);
    Camera set_time(Scene& scene, float time);
    float fps() const;
    int n_frames() const;
    const Anim_Camera& camera() const;
    Anim_Camera& camera();
    void set(int n_frames, int fps);

private:
    Uint64 last_frame = 0;
    bool playing = false;
    int frame_rate = 30;
    int max_frame = 90;
    int current_frame = 0;
    int displayed_frame = 0;

    Widget_Camera ui_camera;
    Widget_Render ui_render;
    Anim_Camera anim_camera;

    Joint* joint_select = nullptr;
    unsigned int joint_id_offset = 0;
    
    Pose old_pose;
    Mat4 old_p_to_j;
    Vec3 old_euler;

    bool visualize_splines = false;
    bool camera_selected = false;
    Scene_ID prev_selected = 0;
    std::unordered_map<Scene_ID, GL::Lines> spline_cache;
    void make_spline(Scene_ID id, const Anim_Pose& pose);
    void camera_spline();
};
}