package eu.tankernn.gameEngine;

import java.io.IOException;
import java.util.Collection;
import java.util.stream.Collectors;

import org.lwjgl.opengl.Display;
import org.lwjgl.opengl.GL30;
import org.lwjgl.util.vector.Vector4f;

import eu.tankernn.gameEngine.entities.Camera;
import eu.tankernn.gameEngine.entities.Entity3D;
import eu.tankernn.gameEngine.entities.GameContext;
import eu.tankernn.gameEngine.environmentMap.EnvironmentMapRenderer;
import eu.tankernn.gameEngine.loader.Loader;
import eu.tankernn.gameEngine.loader.models.AABB;
import eu.tankernn.gameEngine.loader.textures.Texture;
import eu.tankernn.gameEngine.particles.ParticleMaster;
import eu.tankernn.gameEngine.postProcessing.PostProcessor;
import eu.tankernn.gameEngine.renderEngine.DisplayManager;
import eu.tankernn.gameEngine.renderEngine.Fbo;
import eu.tankernn.gameEngine.renderEngine.MasterRenderer;
import eu.tankernn.gameEngine.renderEngine.MultisampleMultitargetFbo;
import eu.tankernn.gameEngine.renderEngine.Scene;
import eu.tankernn.gameEngine.renderEngine.gui.floating.FloatingTextureRenderer;
import eu.tankernn.gameEngine.renderEngine.skybox.Skybox;
import eu.tankernn.gameEngine.renderEngine.water.WaterMaster;
import eu.tankernn.gameEngine.terrains.TerrainPack;
import eu.tankernn.gameEngine.util.DistanceSorter;
import eu.tankernn.gameEngine.util.InternalFile;
import eu.tankernn.gameEngine.util.MousePicker;

public class TankernnGame3D extends TankernnGame {
	protected MasterRenderer renderer;
	protected WaterMaster waterMaster;
	protected ParticleMaster particleMaster;
	protected FloatingTextureRenderer floatingRenderer;
	protected PostProcessor postProcessor;
	protected Camera camera;
	protected Skybox sky;
	protected MousePicker picker;

	protected World world;
	protected Entity3D player;

	private MultisampleMultitargetFbo multisampleFbo = new MultisampleMultitargetFbo(Display.getWidth(),
			Display.getHeight());
	private Fbo outputFbo = new Fbo(Display.getWidth(), Display.getHeight(), Fbo.DEPTH_TEXTURE),
			outputFbo2 = new Fbo(Display.getWidth(), Display.getHeight(), Fbo.DEPTH_TEXTURE);

	public TankernnGame3D(String name, String[] dayTextures, String[] nightTextures) {
		super(name);
		try {
			loader.readModelSpecification(new InternalFile("models.json"));
		} catch (IOException e) {
			e.printStackTrace();
		}
		this.sky = new Skybox(loader, Texture.newCubeMap(InternalFile.fromFilenames("skybox", dayTextures, "png"), 400),
				Texture.newCubeMap(InternalFile.fromFilenames("skybox", nightTextures, "png"), 400), 400);
	}

	public void update() {
		super.update();

		GameContext ctx = new GameContext(true, DisplayManager.getFrameTimeSeconds(),
				world.getEntities().values().stream().map(Entity3D::getState).collect(Collectors.toSet())) {

			@Override
			public float getTerrainHeight(float x, float z) {
				return getTerrain().getTerrainHeightByWorldPos(x, z);
			}

			@Override
			public float getHeight(int entityId) {
				return world.getEntities().get(entityId).getHeight();
			}

			@Override
			public AABB getBoundingBox(int entityId) {
				return world.getEntities().get(entityId).getBoundingBox();
			}
		};

		world.update(ctx);

		picker.update(world.getTerrainPack(), world.getEntities().values(), guiMaster.getGuis());
		camera.update();
		world.getTerrainPack().update(player);
		particleMaster.update(camera);
		DistanceSorter.sort(world.getLights(), camera);

		audioMaster.setListenerPosition(player.getPosition());
	}

	protected void preRender() {

	}

	protected void render() {
		Scene scene = new Scene(world.getEntities().values(), world.getTerrainPack(), world.getLights(), camera, sky);

		EnvironmentMapRenderer.renderEnvironmentMap(scene.getEnvironmentMap(), scene, player.getPosition(), renderer);

		waterMaster.renderBuffers(renderer, scene);

		multisampleFbo.bindFrameBuffer();

		renderer.renderScene(scene, new Vector4f(0, 1, 0, Float.MAX_VALUE));
		waterMaster.renderWater(camera, world.getLights());
		particleMaster.renderParticles(camera);
		floatingRenderer.render(world.getFloatTextures(), camera);
	}

	protected void postRender() {
		multisampleFbo.unbindFrameBuffer();

		multisampleFbo.resolveToFbo(GL30.GL_COLOR_ATTACHMENT0, outputFbo);
		multisampleFbo.resolveToFbo(GL30.GL_COLOR_ATTACHMENT1, outputFbo2);

		postProcessor.doPostProcessing(outputFbo.getColourTexture(), outputFbo2.getColourTexture());
		super.render();
	}

	public void fullRender() {
		this.preRender();
		this.render();
		this.postRender();
	}

	public void cleanUp() {
		super.cleanUp();
		world.finalize();
		particleMaster.finalize();
		postProcessor.finalize();
		waterMaster.finalize();
		multisampleFbo.finalize();
		outputFbo.finalize();
		outputFbo2.finalize();
		renderer.finalize();
	}

	public Collection<Entity3D> getEntities() {
		return world.getEntities().values();
	}

	public Loader getLoader() {
		return loader;
	}

	public TerrainPack getTerrain() {
		return world.getTerrainPack();
	}

	public World getWorld() {
		return world;
	}

	public Entity3D getPlayer() {
		return player;
	}
}