package org.rosuda.ibase.toolkit;

import java.util.Vector;
import java.awt.Frame;
import java.awt.Graphics;
import java.awt.*;
import java.awt.image.*;
import java.awt.print.*;
import java.awt.event.ActionListener;
import java.awt.event.ActionEvent;
import java.io.PrintStream;
import java.io.File;
import javax.imageio.*;

import net.java.games.jogl.*;

import org.rosuda.ibase.*;
import org.rosuda.pograss.*;
import org.rosuda.util.*;

/** PGScanvas - extends {@link LayerCanvas} by adding generic functionality for
exporting the content to PGS metafile or PostScript format. Any implementing
class must use PoGraSS methods instead of Graphics.
@version $Id$
*/
public class PGSJoglCanvas extends GLCanvas implements Commander, Dependent, Printable, GLEventListener {
    /** frame that owns this canvas. can be null if none does. it is mainly used
	to identify current frame in calls to dialogs */
	protected Frame myFrame=null;
    /** description of this canvas. */
    protected String desc="untitled PGS canvas";

	public int layers=0;
	
    static Notifier globalNotifier=null;
    
    /** plot manager for any additional objects */
    protected PlotManager pm;
	
    /** inProgress flag to avoid recursions in paint methods */
    protected boolean inProgress=false;
	
    /** X-axis of the plot coordinates. beware that it may be <code>null</code> */
    protected Axis ax;
    /** Y-axis of the plot coordinates. beware that it may be <code>null</code> */
    protected Axis ay;
	
    protected boolean cancel;
    protected Dialog intDlg;
    
    public PageFormat pageFormat;
    
    public PGSJoglCanvas(int layers, Axis x, Axis y) {
		super(net.java.games.jogl.impl.GLContextFactory.getFactory().chooseGraphicsConfiguration(new GLCapabilities(), new DefaultGLCapabilitiesChooser(), GraphicsEnvironment.getLocalGraphicsEnvironment().getDefaultScreenDevice()),
			  new GLCapabilities(),
			  new DefaultGLCapabilitiesChooser(),
			  null);
		ax=x; ay=y;
		this.layers=layers;
		//pm=new PlotManager(this);
		pm=null;
        if (globalNotifier==null) globalNotifier=new Notifier();
        globalNotifier.addDepend(this);
		//setSize(640,480);
        addGLEventListener(this);
    }
	
    public PGSJoglCanvas(int layers) {
		this(layers, null, null);
    }
	
    public PGSJoglCanvas() { // if no layer # specified, use 1 resulting in old behavior when DBCanvas was used
        this(1);
    }
	
    protected void finalize() {
        globalNotifier.delDepend(this);
    }
    
    public class IDlgCL implements ActionListener {
        PGSJoglCanvas c;
        public IDlgCL(PGSJoglCanvas cc) { c=cc; };
		
        /** activated if a button was pressed. It determines whether "cancel" was pressed or OK" */
        public void actionPerformed(ActionEvent e) {
            c.cancel=!e.getActionCommand().equals("OK");
            c.intDlg.setVisible(false);
        }
    }
	
    /** returns the global notifier common to all PGScanvas descendants. It is mainly used by routines which change some user settings and need to notify all plots regardless of content. */
    public static Notifier getGlobalNotifier() {
        if (globalNotifier==null) globalNotifier=new Notifier();
        return globalNotifier;
    }
    
    /** paintBuffer simply calls {@link #paintPoGraSS} on the supplied {@link Graphics}.
        Any further classes should override {@link #paintPoGraSS} instead of
	{@link #paintLayer}  */
    public void paintLayer(Graphics g, int layer) {
        if (inProgress) return; /* avoid recursions */
        inProgress=true;
        PoGraSSgraphics p=new PoGraSSgraphics(g,layer);
        beginPaint(p);
		paintPoGraSS(p);
		endPaint(p);
        inProgress=false;
    }

	int viewportWidth;
    int viewportHeight;
	
	public void init (GLDrawable drawable) {
        System.out.println ("init()");
        GL gl = drawable.getGL(); 
		
		gl.glEnable (GL.GL_LINE_SMOOTH);
		gl.glHint (GL.GL_LINE_SMOOTH_HINT, GL.GL_NICEST);
		
		gl.glEnable ( GL.GL_BLEND );
		gl.glBlendFunc ( GL.GL_SRC_ALPHA, GL.GL_ONE_MINUS_SRC_ALPHA );		
		
		gl.glClearColor( 1.0f, 1.0f, 1.0f, 1.0f ); 
        gl.glColor3f( 0.0f, 0.0f, 0.0f ); 
    }
    
    /** Called to indicate the drawing surface has been moved and/or resized
		*/
    public void reshape (GLDrawable drawable,
                         int x,
                         int y,
                         int width,
                         int height) {
        GL gl = drawable.getGL(); 
        GLU glu = drawable.getGLU(); 
		//System.out.println("reshape (gl="+gl+", w="+width+", h="+height+")");
		
        // save size for viewport reset
        viewportWidth = width;
        viewportHeight = height;
		
		float[] rgba=getBackground().getRGBComponents(null);
		gl.glClearColor( rgba[0], rgba[1], rgba[2], rgba[3] ); 

        gl.glMatrixMode( GL.GL_PROJECTION );  
        gl.glLoadIdentity(); 
        glu.gluOrtho2D (0, width, 0, height);
        gl.glViewport( 0, 0, viewportWidth, viewportHeight ); 
		float mmx[] = { 1f, 0f, 0f, 0f, 0f, -1f, 0f, 0f, 0f, 0f, 1f, 0f, 0f, 0f, 0f, 1f };
		gl.glTranslatef(0f, (float)(height), 0f);
		gl.glMultMatrixf(mmx);
    }
    
    /** Called by drawable to initiate drawing 
		*/
    public void display (GLDrawable drawable) {
        GL gl = drawable.getGL();
        GLU glu = drawable.getGLU();
		//System.out.println("display (gl="+gl+")");
		
        // is there a pending world window change?
        //if (worldWindowChanged)
        //    resetWorldWindow(gl, glu);
		
        gl.glMatrixMode (GL.GL_MODELVIEW);
        gl.glLoadIdentity(); 
        gl.glClear (GL.GL_COLOR_BUFFER_BIT);
		
        gl.glColor3f( 0.0f, 0.0f, 0.0f );
		
		PoGraSSjogl p=new PoGraSSjogl(gl, glu, this);
		//PoGraSSgraphics p=new PoGraSSgraphics(new render.jogl.JoglGraphics(gl, glu));
		p.setTitle(desc);
		beginPaint(p);
		paintPoGraSS(p);
		endPaint(p);
		p=null;
    }
	
    /** Called by drawable to indicate mode or device has changed
		*/
    public void displayChanged (GLDrawable drawable,
                                boolean modeChanged,
                                boolean deviceChanged) {
        System.out.println ("displayChanged()");
    }
	
	
    public int print(Graphics g, PageFormat pf, int pi) {
        if (pi >= 1) {
            return Printable.NO_SUCH_PAGE;
        }
        Graphics2D g2=(Graphics2D) g;
        // move the origin such that it's inside the printable area
        g2.translate(pf.getImageableX(), pf.getImageableY());
        // TODO: support for something like "fit to page" ... (just use scale(...))
        // we must use paintLayer, becasue paint does all the buffering and rasterizing
        paintLayer(g, -1);
        return Printable.PAGE_EXISTS;
    }
    
    /** set the corresponding frame that contains this canvas. It is used mainly
		for dialog boxes to raise the correct frame before entering modal state.
		If no frame is set, default common frame is used by the dialogs. */
    public void setFrame(Frame owner) { myFrame=owner; };
    /** returns corresponding frame containing this canvas as set by {@link #setFrame} 
	@return associated frame */
    public Frame getFrame() { return myFrame; };
    /** set canvas title */
    public void setTitle(String t) { desc=t; };
    /** return canvas title
		@return canvas title */
    public String getTitle() { return desc; };
	
    /** abstract paint class to be implemented by any descendants. */
    public void paintPoGraSS(PoGraSS g) {};
	
    /** get the PlotManager associated with this plot */
    public PlotManager getPlotManager() { return pm; }
	
    public Axis getXAxis() { return ax; }
    public Axis getYAxis() { return ay; }
    
    /** this method provides an API to fetch data contents of the plot. The id is implementation-dependent, but first two variables x and y should be mapped to 0 and 1 correspondingly. Therefore every plot containting data must support getData(0). For invalid ids <code>null</code> is returned. */
    public SVar getData(int id) { return null; }
	
    protected int paintLayerCounter;
	
    protected void nextLayer(PoGraSS p) {
		if (pm!=null) pm.drawLayer(p,paintLayerCounter,layers);
		paintLayerCounter++;
		p.nextLayer();
    }
	
    /** before using {@link #paintPoGraSS} this method should be called to ensure that a consistent state while painting. Currently the main goal of this function is to reset the paintLayerCounter. */
    protected void beginPaint(PoGraSS p) {
        paintLayerCounter=0;
    }
	
    /** this methods finalizes painting tasks. If {@link #paintPoGraSS} is called directly this method should not be ommitted. */
    protected void endPaint(PoGraSS p) {
		while (paintLayerCounter<layers)
			nextLayer(p);
    }

	// for compatibility with LayerCanvas
	public void setUpdateRoot(int ur) {
	}
	
    public void forcedFlush() {
        Rectangle r=getBounds();
        setUpdateRoot(0);
        //setSize(r.width-1,r.height-1);
        setSize(r.width,r.height);
    }
    
    /** default handing of commands "exportPGS" and "exportPS". Any descendant should
		call <code>super.run(o,cmd)</code> to retain this functionality */
    public Object run(Object o, String cmd) {
        if (cmd=="BREAK" && Common.breakDispatcher!=null) {
            Common.breakDispatcher.NotifyAll(new NotifyMsg(this,Common.NM_BREAK));
        }
        if (cmd=="prefs") {
            Platform.getPlatform().handlePrefs();
        }
        if (cmd=="exportPGS") {
            PoGraSSmeta p=new PoGraSSmeta();
            paintPoGraSS(p);
			endPaint(p);
			PrintStream outs=Tools.getNewOutputStreamDlg(myFrame,"Export as PoGraSS to ...","output.pgs");
			if (outs!=null) {
				outs.print(p.getMeta());
				outs.close(); outs=null;
                //(new MsgDialog(null,"PGS Export","Current plot has been exported to output.pgs.")).show();
            };
        };
		if (cmd=="exportPS") {
			PrintStream outs=Tools.getNewOutputStreamDlg(myFrame,"Export as PostScript to ...","output.ps");
			if (outs!=null) {
				PoGraSSPS p=new PoGraSSPS(outs);
				p.setTitle(desc);
                beginPaint(p);
                paintPoGraSS(p);
				endPaint(p);
				p=null;
				outs.close();
				outs=null;
			};
		};
        if (cmd=="exportPDF") {
            new MsgDialog(myFrame,"Information","Please note that the PDF export is very experimental. If the result is unsatisfactory, please use the PostScript export instead.");
            PrintStream outs=Tools.getNewOutputStreamDlg(myFrame,"Export as PDF to ...","output.pdf");
            if (outs!=null) {
                PoGraSSPDF p=new PoGraSSPDF(outs);
                p.setTitle(desc);
                beginPaint(p);
                paintPoGraSS(p);
				endPaint(p);
                p=null;
                outs.close();
                outs=null;
            };
        };
        if (cmd=="sizeDlg") {
            Dialog d=intDlg=new Dialog(myFrame,"Set plot size",true);
            IDlgCL ic=new IDlgCL(this);
            d.setBackground(Color.white);
            d.setLayout(new BorderLayout());
            d.add(new SpacingPanel(),BorderLayout.WEST);
            d.add(new SpacingPanel(),BorderLayout.EAST);
            Panel bp=new Panel(); bp.setLayout(new FlowLayout());
            Button b,b2;
            bp.add(b=new Button("OK"));bp.add(b2=new Button("Cancel"));
            d.add(bp,BorderLayout.SOUTH);
            d.add(new Label(" "),BorderLayout.NORTH);
            Panel cp=new Panel(); cp.setLayout(new FlowLayout());
            d.add(cp);
            cp.add(new Label("width: "));
            TextField tw=new TextField(""+getSize().width,6);
            TextField th=new TextField(""+getSize().height,6);
            cp.add(tw);
            cp.add(new Label(", height: "));
            cp.add(th);
            d.pack();
            b.addActionListener(ic);b2.addActionListener(ic);
            d.setVisible(true);
            if (!cancel) {
                int w=Tools.parseInt(tw.getText());
                int h=Tools.parseInt(th.getText());
                if(w<10) w=getSize().width;
                if(h<10) w=getSize().height;
                setSize(w,h);
                if (myFrame!=null) myFrame.pack();
            };
            d.dispose();
        };
        if (cmd=="javaPrint") {
            PrinterJob printJob = PrinterJob.getPrinterJob();
            if (pageFormat==null) pageFormat=printJob.defaultPage();
            printJob.setPrintable(this, pageFormat);
            if (printJob.printDialog()) {
                try {
                    printJob.print();
                } catch (Exception ex) {
                    ex.printStackTrace();
                }
            }       
        }
        if (cmd=="pageSetup") {
            PrinterJob printJob = PrinterJob.getPrinterJob();
            if (pageFormat==null) pageFormat=printJob.defaultPage();
            pageFormat=printJob.pageDialog(pageFormat);
        }
        if (cmd=="exportBitmapDlg") {
            Dialog d=intDlg=new Dialog(myFrame,"Export as bitmap with size",true);
            IDlgCL ic=new IDlgCL(this);
            d.setBackground(Color.white);
            d.setLayout(new BorderLayout());
            d.add(new SpacingPanel(),BorderLayout.WEST);
            d.add(new SpacingPanel(),BorderLayout.EAST);
            Panel bp=new Panel(); bp.setLayout(new FlowLayout());
            Button b,b2;
            bp.add(b=new Button("OK"));bp.add(b2=new Button("Cancel"));
            d.add(bp,BorderLayout.SOUTH);
            d.add(new Label(" "),BorderLayout.NORTH);
            Panel cp=new Panel(); cp.setLayout(new FlowLayout());
            d.add(cp);
            cp.add(new Label("width: "));
            int ow=getSize().width;
            int oh=getSize().height;
            TextField tw=new TextField(""+ow,6);
            TextField th=new TextField(""+oh,6);
            TextField tfs=new TextField("10",6);
            cp.add(tw);
            cp.add(new Label(", height: "));
            cp.add(th);
            cp.add(new Label(", font size: "));
            cp.add(tfs);
            d.pack();
            b.addActionListener(ic);b2.addActionListener(ic);
            d.setVisible(true);
            if (!cancel) {
                int w=Tools.parseInt(tw.getText());
                int h=Tools.parseInt(th.getText());
                double fs=Tools.parseDouble(tfs.getText());
                if(w<10) w=getSize().width;
                if(h<10) w=getSize().height;
                
                setSize(w,h);
                BufferedImage img=new BufferedImage(w, h, BufferedImage.TYPE_INT_ARGB);
                Graphics2D g=img.createGraphics();
                PoGraSSgraphics p=new PoGraSSgraphics(g);
				p.setTitle(desc);
                beginPaint(p);
                p.setFontSize(fs);
                paintPoGraSS(p);
				endPaint(p);
                setSize(ow,oh);
				
                FileDialog fd=new FileDialog(myFrame,desc,FileDialog.SAVE);
                fd.setModal(true);
                fd.show();
                String fnam="";
                
                if (fd.getDirectory()!=null) fnam+=fd.getDirectory();
                if (fd.getFile()!=null) fnam+=fd.getFile();
                
                if (!PngEncoder.savePNG(new File(fnam),new PngEncoder(img, true, PngEncoder.FILTER_NONE,7)))
                    System.err.println("PGSCanvas.run.exportBitmapDlg: Unable to write PNG file \""+fnam+"\"");
            }
            d.dispose();
        }
        if (cmd=="exportSVG") {
            boolean svgExtensionPresent=false;
            PoGraSS p=null;
            try {
                Class c=Class.forName("PoGraSSSVG");
                p=(PoGraSS)c.newInstance();
                svgExtensionPresent=true;
            } catch (Throwable ee) {
                System.out.println("Cannot load SVG classes: "+ee.getMessage());
                ee.printStackTrace();
            };
            if (svgExtensionPresent) {
                PrintStream outs=Tools.getNewOutputStreamDlg(myFrame,"Export as SVG to ...","output.svg");
                if (outs!=null) {
                    p.setOutPrintStream(outs);
                    p.setTitle(desc);
                    beginPaint(p);
                    paintPoGraSS(p);
					endPaint(p);
                    p=null;
                    outs.close();
                    outs=null;
                };
            } else {
                (new MsgDialog(myFrame,"PGS Export","Cannot find SVG-extensions. Please make sure that PoGraSSSVG is properly installed.")).show();
            };
        };
        return null;
    };
	
    public void Notifying(NotifyMsg msg, Object o, Vector path) {
        if (myFrame!=null)
            myFrame.setBackground(Common.backgroundColor);
        setBackground(Common.backgroundColor);
        setUpdateRoot(0);
        repaint();
    }    
}