/**
 * Copyright 2013 JogAmp Community. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without modification, are
 * permitted provided that the following conditions are met:
 *
 *    1. Redistributions of source code must retain the above copyright notice, this list of
 *       conditions and the following disclaimer.
 *
 *    2. Redistributions in binary form must reproduce the above copyright notice, this list
 *       of conditions and the following disclaimer in the documentation and/or other materials
 *       provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY JogAmp Community ``AS IS'' AND ANY EXPRESS OR IMPLIED
 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
 * FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL JogAmp Community OR
 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
 * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 * The views and conclusions contained in the software and documentation are those of the
 * authors and should not be interpreted as representing official policies, either expressed
 * or implied, of JogAmp Community.
 */
package com.jogamp.opengl.test.junit.jogl.tile;

import com.jogamp.newt.opengl.GLWindow;
import com.jogamp.opengl.test.junit.jogl.demos.es2.GearsES2;
import com.jogamp.opengl.test.junit.jogl.demos.gl2.Gears;
import com.jogamp.opengl.test.junit.util.UITestCase;
import com.jogamp.opengl.util.GLPixelBuffer;
import com.jogamp.opengl.util.TileRenderer;
import com.jogamp.opengl.util.TileRendererBase;
import com.jogamp.opengl.util.GLPixelBuffer.GLPixelAttributes;
import com.jogamp.opengl.util.texture.TextureData;
import com.jogamp.opengl.util.texture.TextureIO;

import java.io.File;
import java.io.IOException;

import com.jogamp.nativewindow.util.PixelFormat;
import com.jogamp.opengl.GL;
import com.jogamp.opengl.GLAutoDrawable;
import com.jogamp.opengl.GLCapabilities;
import com.jogamp.opengl.GLDrawableFactory;
import com.jogamp.opengl.GLEventListener;
import com.jogamp.opengl.GLProfile;
import com.jogamp.opengl.GLRunnable;

import org.junit.FixMethodOrder;
import org.junit.Test;
import org.junit.runners.MethodSorters;

/**
 * Demos offscreen {@link GLAutoDrawable} being used for
 * {@link TileRenderer} rendering to produce a PNG file.
 * <p>
 * {@link TileRenderer} is being kicked off from the main thread.
 * </p>
 * <p>
 * {@link TileRenderer} buffer allocation is performed
 * within the pre {@link GLEventListener}
 * set via {@link TileRendererBase#setGLEventListener(GLEventListener, GLEventListener)}
 * on the main thread.
 * </p>
  * <p>
 * At tile rendering finish, the viewport and
 * and the original {@link GLEventListener}'s PMV matrix as well.
 * The latter is done by calling it's {@link GLEventListener#reshape(GLAutoDrawable, int, int, int, int) reshape} method.
 * </p>
*/
@FixMethodOrder(MethodSorters.NAME_ASCENDING)
public class TestTiledRendering2NEWT extends UITestCase {
    static long duration = 500; // ms

    static GLProfile getGLProfile(final String profile) {
        if( !GLProfile.isAvailable(profile) )  {
            System.err.println("Profile "+profile+" n/a");
            return null;
        }
        return GLProfile.get(profile);
    }
    static GLProfile getGL2ES3() {
        final GLProfile glp = GLProfile.getMaxProgrammableCore(true);
        if( null == glp || !glp.isGL2ES3() ) {
            System.err.println("GL2ES3 n/a, has max-core "+glp);
            return null;
        }
        return glp;
    }

    @Test
    public void test001_off_gl2___aa0() throws IOException {
        final GLProfile glp = getGLProfile(GLProfile.GL2);
        if( null == glp ) {
            return;
        }
        doTest(false, new Gears(), glp, 0);
    }
    @Test
    public void test002_off_gl2___aa8() throws IOException {
        final GLProfile glp = getGLProfile(GLProfile.GL2);
        if( null == glp ) {
            return;
        }
        doTest(false, new Gears(), glp, 8);
    }
    @Test
    public void test011_off_gl2es3_aa0() throws IOException {
        final GLProfile glp = getGL2ES3();
        if( null == glp ) {
            return;
        }
        doTest(false, new GearsES2(), glp, 0);
    }
    @Test
    public void test012_off_gl2es3_aa8() throws IOException {
        final GLProfile glp = getGL2ES3();
        if( null == glp ) {
            return;
        }
        doTest(false, new GearsES2(), glp, 8);
    }
    @Test
    public void test101_on__gl2___aa0() throws IOException {
        final GLProfile glp = getGLProfile(GLProfile.GL2);
        if( null == glp ) {
            return;
        }
        doTest(true, new Gears(), glp, 0);
    }
    @Test
    public void test102_on__gl2___aa8() throws IOException {
        final GLProfile glp = getGLProfile(GLProfile.GL2);
        if( null == glp ) {
            return;
        }
        doTest(true, new Gears(), glp, 8);
    }
    @Test
    public void test111_on__gl2es3_aa0() throws IOException {
        final GLProfile glp = getGL2ES3();
        if( null == glp ) {
            return;
        }
        doTest(true, new GearsES2(), glp, 0);
    }
    @Test
    public void test112_on__gl2es3_aa8() throws IOException {
        final GLProfile glp = getGL2ES3();
        if( null == glp ) {
            return;
        }
        doTest(true, new GearsES2(), glp, 8);
    }

    void doTest(final boolean onscreen, final GLEventListener demo, final GLProfile glp, final int msaaCount) throws IOException {
        final GLCapabilities caps = new GLCapabilities(glp);
        caps.setDoubleBuffered(onscreen);
        if( msaaCount > 0 ) {
            caps.setSampleBuffers(true);
            caps.setNumSamples(msaaCount);
        }

        final int maxTileSize = 256;
        final GLAutoDrawable glad;
        if( onscreen ) {
            final GLWindow glWin = GLWindow.create(caps);
            glWin.setSize(maxTileSize, maxTileSize);
            glWin.setVisible(true);
            glad = glWin;
        } else {
            final GLDrawableFactory factory = GLDrawableFactory.getFactory(glp);
            glad = factory.createOffscreenAutoDrawable(null, caps, null, maxTileSize, maxTileSize);
        }

        glad.addGLEventListener( demo );

        // Fix the image size for now
        final int imageWidth = glad.getSurfaceWidth() * 6;
        final int imageHeight = glad.getSurfaceHeight() * 4;

        final String filename = this.getSnapshotFilename(0, "-tile", glad.getChosenGLCapabilities(), imageWidth, imageHeight, false, TextureIO.PNG, null);
        final File file = new File(filename);

        // Initialize the tile rendering library
        final TileRenderer renderer = new TileRenderer();
        renderer.setImageSize(imageWidth, imageHeight);
        renderer.setTileSize(glad.getSurfaceWidth(), glad.getSurfaceHeight(), 0);
        renderer.attachAutoDrawable(glad);

        final GLPixelBuffer.GLPixelBufferProvider pixelBufferProvider = GLPixelBuffer.defaultProviderWithRowStride;
        final boolean[] flipVertically = { false };

        final GLEventListener preTileGLEL = new GLEventListener() {
            @Override
            public void init(final GLAutoDrawable drawable) {
                final GL gl = drawable.getGL();
                final PixelFormat.Composition hostPixelComp = pixelBufferProvider.getHostPixelComp(gl.getGLProfile(), 3);
                final GLPixelAttributes pixelAttribs = pixelBufferProvider.getAttributes(gl, 3, true);
                final GLPixelBuffer pixelBuffer = pixelBufferProvider.allocate(gl, hostPixelComp, pixelAttribs, true, imageWidth, imageHeight, 1, 0);
                renderer.setImageBuffer(pixelBuffer);
                if( drawable.isGLOriented() ) {
                    flipVertically[0] = false;
                } else {
                    flipVertically[0] = true;
                }
            }
            @Override
            public void dispose(final GLAutoDrawable drawable) {}
            @Override
            public void display(final GLAutoDrawable drawable) {}
            @Override
            public void reshape(final GLAutoDrawable drawable, final int x, final int y, final int width, final int height) {}
        };
        renderer.setGLEventListener(preTileGLEL, null);

        while ( !renderer.eot() ) {
            renderer.display();
        }

        renderer.detachAutoDrawable();

        // Restore viewport and Gear's PMV matrix
        // .. even though we close the demo, this is for documentation!
        glad.invoke(true, new GLRunnable() {
            @Override
            public boolean run(final GLAutoDrawable drawable) {
                drawable.getGL().glViewport(0, 0, drawable.getSurfaceWidth(), drawable.getSurfaceHeight());
                demo.reshape(drawable, 0, 0, drawable.getSurfaceWidth(), drawable.getSurfaceHeight());
                return false;
            }
        });

        final GLPixelBuffer imageBuffer = renderer.getImageBuffer();
        final TextureData textureData = new TextureData(
                caps.getGLProfile(),
                0 /* internalFormat */,
                imageWidth, imageHeight,
                0,
                imageBuffer.pixelAttributes,
                false, false,
                flipVertically[0],
                imageBuffer.buffer,
                null /* Flusher */);

        TextureIO.write(textureData, file);

        glad.destroy();
    }

    public static void main(final String args[]) {
        for(int i=0; i<args.length; i++) {
            if(args[i].equals("-time")) {
                i++;
                try {
                    duration = Integer.parseInt(args[i]);
                } catch (final Exception ex) { ex.printStackTrace(); }
            }
        }
        org.junit.runner.JUnitCore.main(TestTiledRendering2NEWT.class.getName());
    }
}
