package com.badlogic.gdx.tests.extensions;

import com.badlogic.gdx.ApplicationListener;
import com.badlogic.gdx.Gdx;
import com.badlogic.gdx.graphics.Color;
import com.badlogic.gdx.graphics.GL10;
import com.badlogic.gdx.graphics.OrthographicCamera;
import com.badlogic.gdx.graphics.Pixmap.Format;
import com.badlogic.gdx.graphics.Texture;
import com.badlogic.gdx.graphics.g2d.SpriteBatch;
import com.badlogic.gdx.graphics.g2d.TextureRegion;
import com.badlogic.gdx.graphics.glutils.FrameBuffer;
import com.badlogic.gdx.graphics.glutils.ShaderProgram;
 
public class LightingGame implements ApplicationListener {
        private OrthographicCamera camera;
        private SpriteBatch batch;
       
        private Texture light;
        private Texture background;
       
        private FrameBuffer lightMap;
        private TextureRegion fboRegion;
 
        private ShaderProgram shadowShader;
        private ShaderProgram defaultShader;
       
        @Override
        public void create() {
               
                camera = new OrthographicCamera(1024/2, 300);
                camera.position.set(1024/4, 300/2, 0);
                camera.update();
               
                batch = new SpriteBatch();
               
                background = new Texture(Gdx.files.internal("data/tiles.png"));
                light = new Texture(Gdx.files.internal("data/light.png"));
               
                lightMap = new FrameBuffer(Format.RGBA8888, Gdx.graphics.getWidth(), Gdx.graphics.getHeight(), false);
               
                fboRegion = new TextureRegion(lightMap.getColorBufferTexture(), 0, 0, lightMap.getWidth(), lightMap.getHeight());
                fboRegion.flip(false, true);
               
                setupShaders();
        }
       
        public void setupShaders(){
                defaultShader = SpriteBatch.createDefaultShader();
               
                String vertexShader = "attribute vec4 " + ShaderProgram.POSITION_ATTRIBUTE + ";\n" //
                                + "attribute vec4 " + ShaderProgram.COLOR_ATTRIBUTE + ";\n" //
                                + "attribute vec2 " + ShaderProgram.TEXCOORD_ATTRIBUTE + "0;\n" //
                                + "uniform mat4 u_projTrans;\n" //
                                + "varying vec4 v_color;\n" //
                                + "varying vec2 v_texCoords;\n" //
                                + "\n" //
                                + "void main()\n" //
                                + "{\n" //
                                + "   v_color = " + ShaderProgram.COLOR_ATTRIBUTE + ";\n" //
                                + "   v_texCoords = " + ShaderProgram.TEXCOORD_ATTRIBUTE + "0;\n" //
                                + "   gl_Position =  u_projTrans * " + ShaderProgram.POSITION_ATTRIBUTE + ";\n"
                                + "}\n";
                String fragmentShader = "#ifdef GL_ES\n" //
                                + "#define LOWP lowp\n" //
                                + "precision mediump float;\n" //
                                + "#else\n" //
                                + "#define LOWP \n" //
                                + "#endif\n" //
                                + "varying LOWP vec4 v_color;\n" //
                                + "varying vec2 v_texCoords;\n" //
                                + "uniform sampler2D u_texture;\n"
                                + "uniform sampler2D u_texture1;\n"
                                + "uniform vec2 lightMapResolution;\n"
                                + "void main()\n"//
                                + "{\n" //
                                + "  vec4 texColor0 = texture2D(u_texture, v_texCoords); \n"
                                + "  vec4 texColor1 = texture2D(u_texture1, gl_FragCoord.xy / lightMapResolution); \n"
                                + "  vec4 shadow = texColor0 * vec4(0.5, 0.5, 0.5, 1.0);\n"
                                + "  vec4 light = texColor0;\n"
                                + "  gl_FragColor = mix(shadow, light, texColor1.a);"
                                + "}";
 
                shadowShader = new ShaderProgram(vertexShader, fragmentShader);
                if (!shadowShader.isCompiled()) {
                        System.err.println(shadowShader.getLog());
                        System.exit(0);
                }
                if (shadowShader.getLog().length()!=0)
                        System.out.println(shadowShader.getLog());
        }
 
        @Override
        public void dispose() {
                batch.dispose();
        }
 
        @Override
        public void render() {         
                // Draw lightMap
               
               
                lightMap.begin();
               
                Gdx.gl.glClearColor(0, 0, 0, 0.0f);
                Gdx.gl.glClear(GL10.GL_COLOR_BUFFER_BIT);
 
                batch.setShader(defaultShader);
                batch.begin();
                batch.draw(light, Gdx.input.getX()-light.getWidth()/2f, Gdx.graphics.getHeight()-Gdx.input.getY()-light.getHeight()/2f);
                batch.end();
               
                batch.flush();
               
                lightMap.end();
               
                // Draw normal stuff
                Gdx.gl.glClearColor(0, 0, 0, 0);
                Gdx.gl.glClear(GL10.GL_COLOR_BUFFER_BIT);
               
                batch.setProjectionMatrix(camera.combined);
               
                batch.setShader(shadowShader);
                batch.begin();
                shadowShader.setUniformf("lightMapResolution", lightMap.getWidth(), lightMap.getHeight());
                shadowShader.setUniformi("u_texture1", 1);
               
                lightMap.getColorBufferTexture().bind(1);
                background.bind(0);
                
                batch.draw(background, 0, 0);
       
                batch.end();
        }
 
        @Override
        public void resize(int width, int height) {
                camera.setToOrtho(false, width, height);
               
        }
 
        @Override
        public void pause() {
        }
 
        @Override
        public void resume() {
        }
}