package vtk.sample.rendering;

import com.jogamp.opengl.DefaultGLCapabilitiesChooser;
import com.jogamp.opengl.GLAutoDrawable;
import com.jogamp.opengl.GLCapabilities;
import com.jogamp.opengl.GLContext;
import com.jogamp.opengl.GLEventListener;
import com.jogamp.opengl.GLProfile;
import com.jogamp.opengl.awt.GLCanvas;
import java.awt.BorderLayout;
import java.awt.GraphicsConfiguration;
import java.awt.GraphicsDevice;
import java.awt.GraphicsEnvironment;

import javax.swing.JFrame;
import javax.swing.SwingUtilities;

public class JoglContextTestCase
{
  private static JFrame mainFrame = null;
  private static int screenNum = 0;

  private static void buildSceneAndCheckConfigurations()
  {
    GraphicsEnvironment graphEnv = GraphicsEnvironment.getLocalGraphicsEnvironment();
    GraphicsDevice[] graphDevsOrig = graphEnv.getScreenDevices();

    GraphicsConfiguration gcToUse = graphDevsOrig[screenNum].getDefaultConfiguration();

    // *************************************************************************
    // NOTE: if I comment the following loop, which apparently does NOTHING
    // (except calling .getGraphicsConfigurations()), this test case
    // returns "GL version: 4.6" on my PC, while with the following loop
    // uncommented, it returns "GL version: 1.1" --> this would cause VTK
    // to CRASH!
    // The single call responsible for the "GL 1.1" issue, for some reason,
    // seems to be 'graphDevsOrig[i].getConfigurations();'
    // *************************************************************************    
    for (int i = 0; i < graphDevsOrig.length; i++)
    {
      System.out.println("Screen device # " + i + ": " + graphDevsOrig[i].getIDstring());
      GraphicsConfiguration[] graphConfs = graphDevsOrig[i].getConfigurations();

      for (int j = 0; j < graphConfs.length; j++)
      {
        System.out.println("Screen device # " + i + ", configuration # " + j + ", " + 
                           "bounds: " + graphConfs[j].getBounds());
      }
    }

    buildScene(gcToUse);
  }

  private static void buildScene(GraphicsConfiguration graphConf)
  {
    GLCanvas glCanvas = new GLCanvas(new GLCapabilities(GLProfile.getMaximum(true)), new DefaultGLCapabilitiesChooser(), graphConf.getDevice());
    glCanvas.addGLEventListener(new GLEventListener()
    {
      @Override
      public void init(final GLAutoDrawable drawable)
      {
        // Make sure the JOGL Context is current
        GLContext ctx = drawable.getContext();
        if (! ctx.isCurrent())
        {
          ctx.makeCurrent();
        }
//        final GL gl = drawable.getGL();
//        System.out.println(JoglVersion.getGLInfo(gl, null));
//        System.out.println("* Context = " + drawable.getContext().toString());
        System.out.println("* Context GL version: " + drawable.getContext().getGLVersion());
      }

      @Override
      public void reshape(final GLAutoDrawable drawable, final int x, final int y, final int width, final int height)
      {
      }

      @Override
      public void display(final GLAutoDrawable drawable)
      {
      }

      @Override
      public void dispose(final GLAutoDrawable drawable)
      {
      }
    });

    // UI part
    mainFrame = new JFrame("SimpleVTK", graphConf);
    mainFrame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
    mainFrame.getContentPane().setLayout(new BorderLayout());

    mainFrame.setSize(1000, 600);
//    mainFrame.setLocationRelativeTo(null);
    mainFrame.setVisible(true);
    mainFrame.setExtendedState(JFrame.MAXIMIZED_BOTH);
    mainFrame.getContentPane().add(glCanvas, BorderLayout.CENTER);
    glCanvas.requestFocus();

    System.out.println("NewFrame created in thread [" + Thread.currentThread().getId() + "], isEDT: " + SwingUtilities.isEventDispatchThread());
  }

  public static void main(String[] args)
  {
    if (args.length > 0)
    {
      try
      {
        String screenNumStr = args[0];
        screenNum = Integer.parseInt(screenNumStr);
      }
      catch (Throwable ex)
      {
        screenNum = 0;
        ex.printStackTrace();
      }
    }
    
    SwingUtilities.invokeLater(new Runnable()
    {
      public void run()
      {
        buildSceneAndCheckConfigurations();
      }
    });
  }
}