import { Game } from "planetwars";
import { memory } from "planetwars/planetwars_bg";
import { Resizer, resizeCanvasToDisplaySize, FPSCounter, url_to_mesh, Mesh } from "./webgl/util";
import { Shader, Uniform4f, Uniform2fv, Uniform3fv, Uniform1i, Uniform1f, Uniform2f, ShaderFactory, Uniform3f, UniformMatrix3fv, UniformBool } from './webgl/shader';
import { Renderer } from "./webgl/renderer";
import { VertexBuffer, IndexBuffer } from "./webgl/buffer";
import { VertexBufferLayout, VertexArray } from "./webgl/vertexBufferLayout";
import { callbackify } from "util";

function f32v(ptr: number, size: number): Float32Array {
    return new Float32Array(memory.buffer, ptr, size);
}

function i32v(ptr: number, size: number): Int32Array {
    return new Int32Array(memory.buffer, ptr, size);
}

export function set_game_name(name: string) {
    GAMENAME.innerHTML = name;
}

const GAMENAME = document.getElementById("name");

const TURNCOUNTER = document.getElementById("turnCounter");

const COUNTER = new FPSCounter();
const LOADER = document.getElementById("main");

const SLIDER = <HTMLInputElement>document.getElementById("turnSlider");
const FILESELECTOR = <HTMLInputElement> document.getElementById("fileselect");
const SPEED = <HTMLInputElement> document.getElementById("speed");

export function set_loading(loading: boolean) {
    if (loading) {
        if (!LOADER.classList.contains("loading")) {
            LOADER.classList.add("loading");
        }
    } else {
        LOADER.classList.remove("loading");
    }
}

const URL = window.location.origin+window.location.pathname;
export const LOCATION = URL.substring(0, URL.lastIndexOf("/") + 1);
const CANVAS = <HTMLCanvasElement>document.getElementById("c");
const RESOLUTION = [CANVAS.width, CANVAS.height];

const GL = CANVAS.getContext("webgl");

var ms_per_frame = parseInt(SPEED.value);

resizeCanvasToDisplaySize(CANVAS);

GL.clearColor(0, 0, 0, 0);
GL.clear(GL.COLOR_BUFFER_BIT);

GL.enable(GL.BLEND);
GL.blendFunc(GL.SRC_ALPHA, GL.ONE_MINUS_SRC_ALPHA);

var SHADERFACOTRY: ShaderFactory;
ShaderFactory.create_factory(
    LOCATION + "static/shaders/frag/simple.glsl", LOCATION + "static/shaders/vert/simple.glsl"
).then((e) => SHADERFACOTRY = e);

var VOR_SHADER_FACTORY: ShaderFactory;
ShaderFactory.create_factory(
    LOCATION + "static/shaders/frag/vor.glsl", LOCATION + "static/shaders/vert/vor.glsl"
).then((e) => VOR_SHADER_FACTORY = e);

class GameInstance {
    resizer: Resizer;
    game: Game;
    shader: Shader;
    vor_shader: Shader;
    renderer: Renderer;
    planet_count: number;

    vor_counter = 3;
    use_vor = true;
    playing = true;    // 0 is paused, 1 is playing but not rerendered, 2 is playing and rerendered
    time_stopped_delta = 0;
    last_time = 0;
    frame = -1;

    ship_indices: number[];

    turn_count = 0;

    constructor(game: Game, meshes: Mesh[], ship_mesh: Mesh)  {
        this.game = game;
        this.planet_count = this.game.get_planet_count();
        this.shader = SHADERFACOTRY.create_shader(GL, {"MAX_CIRCLES": ''+this.planet_count});
        this.vor_shader = VOR_SHADER_FACTORY.create_shader(GL, {"PLANETS": ''+this.planet_count});
        this.resizer = new Resizer(CANVAS, [...f32v(game.get_viewbox(), 4)], true);
        this.renderer = new Renderer();
        this.game.update_turn(0);

        const indexBuffer = new IndexBuffer(GL, [
            0, 1, 2,
            1, 2, 3,
        ]);

        const positionBuffer = new VertexBuffer(GL, [
            -1, -1,
            -1, 1,
            1, -1,
            1, 1,
        ]);

        const layout = new VertexBufferLayout();
        layout.push(GL.FLOAT, 2, 4, "a_pos");

        const vao = new VertexArray();
        vao.addBuffer(positionBuffer, layout);

        this.renderer.addToDraw(indexBuffer, vao, this.vor_shader);

        // Setup key handling
        document.addEventListener('keydown', this.handleKey.bind(this));

        const planets = f32v(game.get_planets(), this.planet_count * 3);

        for(let i=0; i < this.planet_count; i++){

            const transform = new UniformMatrix3fv([
                1, 0, 0,
                0, 1, 0,
                -planets[i*3], -planets[i*3+1], 1,
            ]);

            const indexBuffer = new IndexBuffer(GL, meshes[i % meshes.length].cells);
            const positionBuffer = new VertexBuffer(GL, meshes[i % meshes.length].positions);

            const layout = new VertexBufferLayout();
            layout.push(GL.FLOAT, 3, 4, "a_position");
            const vao = new VertexArray();
            vao.addBuffer(positionBuffer, layout);

            this.renderer.addToDraw(
                indexBuffer,
                vao,
                this.shader,
                {
                    "u_trans": transform,
                    "u_trans_next": transform,
                }
            );
        }

        this.turn_count = game.turn_count();

        this.ship_indices = [];
        const ship_ibo = new IndexBuffer(GL, ship_mesh.cells);
        const ship_positions = new VertexBuffer(GL, ship_mesh.positions);
        const ship_layout = new VertexBufferLayout();
        ship_layout.push(GL.FLOAT, 3, 4, "a_position");
        const ship_vao = new VertexArray();
        ship_vao.addBuffer(ship_positions, ship_layout);

        for (let i = 0; i < this.game.get_max_ships(); i++) {
            this.ship_indices.push(
                this.renderer.addToDraw(
                    ship_ibo,
                    ship_vao,
                    this.shader,
                    {}
                )
            );
        }

        this.vor_shader.uniform(GL, "u_planets", new Uniform3fv(planets));

        // Set slider correctly
        SLIDER.max = this.turn_count - 1 + '';
    }

    on_resize() {
        this.resizer = new Resizer(CANVAS, [...f32v(this.game.get_viewbox(), 4)], true);
    }

    _update_state() {
        const colours = f32v(this.game.get_planet_colors(), this.planet_count * 6);

        this.vor_shader.uniform(GL, "u_planet_colours", new Uniform3fv(colours));

        for(let i=0; i < this.planet_count; i++){
            const u = new Uniform3f(colours[i*6], colours[i*6 + 1], colours[i*6 + 2]);
            this.renderer.updateUniform(i+1, (us) => us["u_color"] = u);
            const u2 = new Uniform3f(colours[i*6 + 3], colours[i*6 + 4], colours[i*6 + 5]);
            this.renderer.updateUniform(i+1, (us) => us["u_color_next"] = u2);
        }

        const ships = f32v(this.game.get_ship_locations(), this.game.get_ship_count() * 9 * 2);
        const ship_colours = f32v(this.game.get_ship_colours(), this.game.get_ship_count() * 3);

        for (let i=0; i < this.game.get_max_ships(); i++) {
            const index = this.ship_indices[i];
            if (i < this.game.get_ship_count()) {

                this.renderer.enableRendershit(index);

                const u = new Uniform3f(ship_colours[i*3], ship_colours[i*3 + 1], ship_colours[i*3 + 2]);
                // const t1 = new UniformMatrix3fv(new Float32Array(ships, i * 18, 9));
                // const t2 = new UniformMatrix3fv(new Float32Array(ships, i * 18 + 9, 9));

                const t1 = new UniformMatrix3fv(ships.slice(i * 18, i * 18 + 9));
                const t2 = new UniformMatrix3fv(ships.slice(i * 18 + 9, i * 18 + 18));

                this.renderer.updateUniform(index, (us) => {
                    us["u_color"] = u;
                    us["u_color_next"] = u;
                    us["u_trans"] = t1;
                    us["u_trans_next"] = t2;
                });
            } else {
                this.renderer.disableRenderShift(index);
            }
        }
    }

    render(time: number) {
        COUNTER.frame(time);

        if (COUNTER.delta(time) < 30) {
            this.vor_counter = Math.min(3, this.vor_counter + 1);
        } else {
            this.vor_counter = Math.max(-3, this.vor_counter - 1);
        }

        if (this.vor_counter < -2) {
            this.use_vor = false;
        }

        if (!this.playing) {
            this.last_time = time;

            this.shader.uniform(GL, "u_viewbox", new Uniform4f(this.resizer.get_viewbox()));
            this.vor_shader.uniform(GL, "u_viewbox", new Uniform4f(this.resizer.get_viewbox()));

            this.renderer.render(GL);
            return;
        }
        if (time > this.last_time + ms_per_frame) {

            this.last_time = time;
            this.updateTurn(this.frame + 1);
        }

        GL.bindFramebuffer(GL.FRAMEBUFFER, null);
        GL.viewport(0, 0, GL.canvas.width, GL.canvas.height);
        GL.clear(GL.COLOR_BUFFER_BIT | GL.DEPTH_BUFFER_BIT);

        this.vor_shader.uniform(GL, "u_viewbox", new Uniform4f(this.resizer.get_viewbox()));
        this.vor_shader.uniform(GL, "u_resolution", new Uniform2f(RESOLUTION));
        this.vor_shader.uniform(GL, "u_vor", new UniformBool(this.use_vor));

        this.shader.uniform(GL, "u_time", new Uniform1f((time - this.last_time) / ms_per_frame));
        this.shader.uniform(GL, "u_mouse", new Uniform2f(this.resizer.get_mouse_pos()));
        this.shader.uniform(GL, "u_viewbox", new Uniform4f(this.resizer.get_viewbox()));
        this.shader.uniform(GL, "u_resolution", new Uniform2f(RESOLUTION));
        this.shader.uniform(GL, "u_vor", new UniformBool(true));

        this.renderer.render(GL);
    }

    updateTurn(turn: number) {
        this.frame = Math.max(0, turn);
        const new_frame = this.game.update_turn(this.frame);
        if (new_frame < this.frame) {
            this.frame = new_frame;
            this.playing = false;
        } else {
            this._update_state();
            this.playing = true;
        }

        TURNCOUNTER.innerHTML = this.frame + " / " + this.turn_count;
        SLIDER.value = this.frame + '';
    }

    handleKey(event: KeyboardEvent) {
        // Space
        if (event.keyCode == 32) {
            if (this.playing) {
                this.playing = false;
            } else {
                this.playing = true;
            }
        }

        // Arrow left
        if (event.keyCode == 37) {
            // This feels more natural than -1 what it should be, I think
            this.updateTurn(this.frame - 2);
        }

        // Arrow right
        if (event.keyCode == 39) {
            this.updateTurn(this.frame + 1);
        }

        // d key
        if (event.keyCode == 68) {
            SPEED.value = ms_per_frame + 10 + '';
            SPEED.onchange(undefined);
        }

        // a key
        if (event.keyCode == 65) {
            SPEED.value = Math.max(ms_per_frame - 10, 0) + '';
            SPEED.onchange(undefined);
        }
    }
}

var game_instance: GameInstance;
var meshes;

export async function set_instance(source: string) {
    if (!meshes) {
        meshes = await Promise.all(
            ["ship.svg", "earth.svg", "mars.svg", "venus.svg"].map(
                (name) => "static/res/assets/" + name
            ).map(url_to_mesh)
        );
    }

    resizeCanvasToDisplaySize(CANVAS);

    game_instance = new GameInstance(Game.new(source), meshes.slice(1), meshes[0]);

    set_loading(false);
}

window.addEventListener('resize', function() {
    resizeCanvasToDisplaySize(CANVAS);

    if (game_instance) {
        game_instance.on_resize();
    }
}, { capture: false, passive: true})

SLIDER.oninput = function() {
    if (game_instance) {
        game_instance.updateTurn(parseInt(SLIDER.value));
    }
}

FILESELECTOR.onchange = function(){
    const file = FILESELECTOR.files[0];
    if(!file) { return; }
    var reader = new FileReader();

    reader.onload = function() {
        set_instance(<string> reader.result);
    }

    reader.readAsText(file);
}

SPEED.onchange = function() {
    ms_per_frame = parseInt(SPEED.value);
}

function step(time: number) {
    if (game_instance) {
        game_instance.render(time);
    }

    requestAnimationFrame(step);
}
// set_loading(false);

requestAnimationFrame(step);