/* ReadIRQFiles.java
 */

//import io.ascii.pairs.PairData;

import jas.hep.*;
import jas.hist.*;
import jas.hist.normalization.*;

import hep.analysis.*;

import java.awt.GridLayout;
import java.io.*;
import java.util.*;
import javax.swing.*;


public class ReadIRQFiles
{

    static int nbins = 0;

    public static void main(String[] argv)
    {
	boolean started = false;
	try {
	    int nargs = 0;
	    String directoryName = argv[nargs]; nargs++;
	    int first = Integer.parseInt(argv[nargs]); nargs++;
	    String firstFileName = directoryName+"/RDat."+first;
	    int last  = Integer.parseInt(argv[nargs]); nargs++;
	    String lastFileName  = directoryName+"/RDat."+last;
	    nbins = Integer.parseInt(argv[nargs]); nargs++;
	    System.out.println("Read IRQ files "+first+"-"+last+" ("+firstFileName+" to "+lastFileName+")"+
			       " w/ "+nbins+" time bins.");
	    started = true;

	    // Create Histograms
	    String title1 = "Readout glitch amplitude";  Histogram h1 = new Histogram(title1);
	    String title2 = "Amplitude vs. Time";        Histogram2D h2 = new Histogram2D(title2);

	    for (int i=first; i<=last; i++) {

		// Set up the number
		String number = ""+i;
		//if (i<10) number = "0"+number;

		// File name
		String fileName = directoryName+"/RDat."+number;
		if (i<first+10) System.out.println(" Ready to read file "+fileName);

		// Set file
		File file = new File(fileName);

		// Read STAR IRQ data
		StarDumpData data = new StarDumpData(file,nbins);
		if (!data.isGood()) continue;

		// Dump specific channel
		for (int j=0; j<5; j++) {
		// for (int j=0; j<nbins; j++) {
		    System.out.println("  Loc 22B, ch 0, bin "+j+": data = "+data.getLoc(22,"B",0,j));
		}

		// Histogram readout glitch
		h1.fill(data.getLoc(22,"B",0,2));

		// Histogram readout
		fillAmpvsTime(data,h2);
	    }

	    DataSource ds1 = PartitionAdapter.create(h1);
	    DataSource ds2 = PartitionAdapter.create(h2);

	    JASHist plot1 = new JASHist();
	    plot1.addData(ds1).show(true);
	    plot1.setTitle(title1);
	    JASHist plot2 = new JASHist();
	    plot2.addData(ds2).show(true);
	    plot2.setTitle(title2);

	    // Make plots
	    IRQPlotter p = new IRQPlotter();
	    p.displayPlot(plot1);
	    p.displayPlot(plot2);
	}
	catch (Exception x) 
	{
	    if (!started) {
		System.err.println("Usage: jre -mx64M ReadIRQFiles <directoryName> <first File #> <last File #> <# bins read>");
	    }
	    else {
		System.err.println("Exception "+x);
		x.printStackTrace();
	    }
	}
    }

    static void fillAmpvsTime(StarDumpData data, Histogram2D h)
    {
	// Fill it
	for (int j=0; j<nbins; j++) {
	    int amp = data.getLoc(22,"B",0,j);
	    int time = j;
	    h.fill(time,amp);
	}
    }
}


/** IRQPlotter
 */
class IRQPlotter
{

    IRQPlotter()
    {
    }

    void displayPlot(JASHist plot)
    {
	//System.out.println(" Display plot "+plot);

	// Display the plot
	JPanel panel = new JPanel(new GridLayout(1,1));
	panel.add(plot);
	JFrame frame = new JFrame("STAR IRQ Data");
	frame.setContentPane(panel);
	frame.setDefaultCloseOperation(frame.EXIT_ON_CLOSE);
	frame.setSize(600,600);
	frame.show();
    }

    void displayPlots(JASHist[] plots)
    {
	int nplots = plots.length;
	//System.out.println(" Display "+nplots+" plot(s).");

	// Display the plots
	int n = (nplots+1)/2;
	JPanel panel = new JPanel(new GridLayout(n,n));
	for (int i=0; i<nplots; i++) {
	    panel.add(plots[i]);
	}
	JFrame frame = new JFrame("STAR IRQ Data");
	frame.setContentPane(panel);
	frame.setDefaultCloseOperation(frame.EXIT_ON_CLOSE);
	frame.setSize(600,600);
	frame.show();
    }
}