package Plot;

import Proc.*;
import Util.*;

import javax.swing.*;
import java.awt.*;
import java.awt.event.*;
import com.sun.jimi.core.*;
import com.sun.jimi.core.raster.*;
import java.awt.image.BufferedImage;
import java.io.*;
import jas.hist.*;

public class RecmbPlotManager {
    
    private JFrame target;
    private int dataset;
    private AppProc proc;
    
    Plot  stg;
    JASHist sth;  
    private ActionListener listener = new ActionListener(){
	    public void actionPerformed(ActionEvent e){
		try{
		    saveCurveImage();
		}catch( Exception exception){
		}
	    }
	};
    
    private ActionListener hlistener = new ActionListener(){
	    public void actionPerformed(ActionEvent e){
		try{
		    saveHistImage();
		}catch( Exception exception){
		}
	    }
	};



    public RecmbPlotManager(){
    }

    public void plot(double[][] plotdata, String title){
	 
	
	stg = new Plot(plotdata[0], "Rec", "Recombination Rate Vs. Iterations");
	
	
	JPanel p = new JPanel();
	p.setLayout(new GridLayout(0,1));
	
	JPanel gPane = new JPanel();
	gPane.setLayout(new BorderLayout()); 
	gPane.add(stg,BorderLayout.CENTER); 
	JButton saveButton = new JButton("Save image");
	saveButton.addActionListener(listener);
	JPanel bPane = new JPanel();
	bPane.add(saveButton);
	gPane.add(bPane,BorderLayout.SOUTH);
	p.add(gPane);
      	
	JScrollPane sp = new JScrollPane(p);
	sp.setPreferredSize(new Dimension(500,350));
	if(target!= null)
	    target.dispose();
	target = new JFrame("Structure Plotting: Recombination Rate - " + title);
	target.addWindowListener(new WindowAdapter() {
		public void windowClosing(WindowEvent e) {
		    target.setVisible(false);
		}
		 
	    });
	
	 
	target.setLocation(300,200);
	target.getContentPane().add(sp);
	target.pack();
	target.setVisible(true);
    }
    

    
    public void plot(AppProc proc){
	
	this.proc = proc;
	
	final Plot g =  new Plot(proc,"Rec", 0, "Recombination Rate Vs. Iterations");
	g.start();
	JPanel p = new JPanel();
	p.setLayout(new GridLayout(0,1));	
	p.add(g);
	
	JScrollPane sp = new JScrollPane(p);
	sp.setPreferredSize(new Dimension(500,350));
	if(target!= null)
	    target.dispose();
	target = new JFrame("Structure Plotting: Recombination Rate");
	target.addWindowListener(new WindowAdapter() {
                public void windowClosing(WindowEvent e) {
		    if(g != null)
			g.stop();
		    target.setVisible(false);
		}
	    });
	
	target.setLocation(300,200);
	target.getContentPane().add(sp);
	target.pack();
	target.setVisible(true);
    }
    
     public void plotHist(double[][] data, String title){
     
	 HistPlot hist = new HistPlot(data[0],"Recombination rate", "Histogram");
 	
	 JPanel p = new JPanel();
	 p.setLayout(new GridLayout(0,1));
	 p.setBackground(Color.white);
	 

	 JPanel ph = new JPanel();
	 ph.setBackground(Color.white);
	 sth=hist.draw();
	 ph.setLayout(new BorderLayout()); 
	 ph.add(sth,BorderLayout.CENTER); 
	 JButton saveButton = new JButton("Save image");
	 saveButton.addActionListener(hlistener);
	 JPanel bPane = new JPanel();
	 bPane.add(saveButton);
	 ph.add(bPane,BorderLayout.SOUTH);   	 
	 ph.setPreferredSize(new Dimension(550,350));
	 p.add(ph);
	 
	 
	JScrollPane sp = new JScrollPane(p);
	sp.setPreferredSize(new Dimension(550,360));
	if(target!= null)
	    target.dispose();
	target = new JFrame("Histogram of distribution of recombination rate - " + title);
	target.addWindowListener(new WindowAdapter() {
                public void windowClosing(WindowEvent e) {
		    target.setVisible(false);
		}});
	
	target.setLocation(300,200);
	target.getContentPane().add(sp);
	target.pack();
	target.setVisible(true);
	
     }
    
      
    protected void saveCurveImage(){
	String filename="";
	// ask for file name/dir to save image
	JFileChooser fc = new JFileChooser();
	GeneralFileFilter jpgFilter = new GeneralFileFilter("jpg", "JPEG image files");
	fc.addChoosableFileFilter(jpgFilter);
	int returnVal = fc.showDialog(target,"Save");
	
	if (returnVal == JFileChooser.APPROVE_OPTION){ 
	    
	    File file = fc.getSelectedFile();
	    String fname = file.getAbsolutePath();
	    if(!fname.endsWith(".jpg")&&!fname.endsWith(".jpeg"))
		file = new File(new String(fname+".jpg"));
	    //check if file exists
	    if(file.exists()){
		Object[] options = {"Yes",
				    "No "};
		
		int n=JOptionPane.showOptionDialog(target,
						   "Target file "+file.getName()+" already exists, Overwrite it?",
						   "Save image",
						   JOptionPane.YES_NO_OPTION,
						   JOptionPane.QUESTION_MESSAGE,
						   null,
						   options,
						   options[1]);
		if(n==JOptionPane.NO_OPTION)
		    return;
	    }
	    File fc_dir = file.getParentFile();
	    if(!fc_dir.canWrite()){
		JOptionPane.showMessageDialog(null,
					      "Can not write image file: permission denied",
					      "Error",
					      JOptionPane.ERROR_MESSAGE);
		return;
	    }
	    filename = file.getAbsolutePath();
	    
	}
	
	Rectangle rect = stg.getBounds();
	BufferedImage image = new BufferedImage(rect.width,rect.height,BufferedImage.TYPE_INT_RGB);	
	Graphics2D g = (Graphics2D)image.getGraphics();
	g.setBackground(Color.white);	
        g.setPaint(Color.white);
	g.fill(rect);
	(stg.getComponent(0)).paint(g);
	
	try {
	    File f = new File(filename);
	    JimiRasterImage jrf = Jimi.createRasterImage(image.getSource());
	    FileOutputStream fos=null;
	    try { 
		fos =new FileOutputStream(f);
	    }catch (Exception ee) { };
	    
	    if(fos != null)
		Jimi.putImage("image/jpeg",jrf,fos);
	}catch (JimiException je) {je.printStackTrace();}
	
    }
    
    
     protected void saveHistImage(){
	String filename="";
	// ask for file name/dir to save image
	JFileChooser fc = new JFileChooser();
	int returnVal = fc.showDialog(target,"Save");
	
	if (returnVal == JFileChooser.APPROVE_OPTION){ 
	    
	    File file = fc.getSelectedFile();
	    String fname = file.getAbsolutePath();
	    if(!fname.endsWith(".jpg")&&!fname.endsWith(".jpeg"))
		file = new File(new String(fname+".jpg"));
	    //check if file exists
	    if(file.exists()){
		Object[] options = {"Yes",
				    "No "};
		
		int n=JOptionPane.showOptionDialog(target,
						   "Target file "+file.getName()+" already exists, Overwrite it?",
						   "Save image",
						   JOptionPane.YES_NO_OPTION,
						   JOptionPane.QUESTION_MESSAGE,
						   null,
						   options,
						   options[1]);
		if(n==JOptionPane.NO_OPTION)
		    return;
	    }
	    File fc_dir = file.getParentFile();
	    if(!fc_dir.canWrite()){
		JOptionPane.showMessageDialog(null,
					      "Can not write image file: permission denied",
					      "Error",
					      JOptionPane.ERROR_MESSAGE);
		return;
	    }
	    filename = file.getAbsolutePath();
	    
	}
	
	Rectangle rect = sth.getBounds();
	BufferedImage image = new BufferedImage(rect.width,rect.height,BufferedImage.TYPE_INT_RGB);	
	Graphics2D g = (Graphics2D)image.getGraphics();
	g.setBackground(Color.white);	
	sth.paint(g);
	
	try {
	    File f = new File(filename);
	    JimiRasterImage jrf = Jimi.createRasterImage(image.getSource());
	    FileOutputStream fos=null;
	    try { 
		fos =new FileOutputStream(f);
	    }catch (Exception ee) { };
	    
	    if(fos != null)
		Jimi.putImage("image/jpeg",jrf,fos);
	}catch (JimiException je) {je.printStackTrace();}
	
    }

 

}