package hep.io.root.util;
import hep.io.root.*;
import hep.io.root.daemon.RootAuthenticator;
import hep.io.root.daemon.RootURLStreamFactory;
import hep.io.root.interfaces.*;
import hep.io.root.util.RootDirectoryTreeCellRenderer;
import javax.swing.*;
import javax.swing.event.*;
import javax.swing.tree.*;
import java.io.*;
import java.awt.*;
import java.awt.event.*;
import jas.hist.*;
import java.net.Authenticator;
import java.net.URL;
import java.net.URLConnection;
/**
* A simple application for browsing histograms in Root Files
* @author: Tony Johnson (tonyj@slac.stanford.edu)
* @version: $Id: RootHistogramBrowser.java,v 1.2 2004/06/28 15:21:26 tonyj Exp $
*/
public class RootHistogramBrowser extends JPanel implements TreeSelectionListener
{
private final static String aboutMessage = "RootHistogramBrowser $Id: RootHistogramBrowser.java,v 1.2 2004/06/28 15:21:26 tonyj Exp $
Author: Tony Johnson (tonyj@slac.stanford.edu)";
private final static TreeModel emptyTree = null;
private JASHist plot;
private JTree tree;
RootHistogramBrowser() throws IOException
{
super(new BorderLayout());
tree = new JTree(emptyTree);
tree.setCellRenderer(new RootDirectoryTreeCellRenderer());
tree.addTreeSelectionListener(this);
tree.setRootVisible(false);
plot = new JASHist();
plot.setBackground(Color.white);
plot.setShowStatistics(true);
JSplitPane split = new JSplitPane(JSplitPane.HORIZONTAL_SPLIT,new JScrollPane(tree),plot);
add(split,BorderLayout.CENTER);
add(new RootMenuBar(),BorderLayout.NORTH);
setPreferredSize(new java.awt.Dimension(500,300));
split.setDividerLocation(245);
}
public void setRootFile(File file) throws IOException
{
Cursor old = getCursor();
setCursor(Cursor.getPredefinedCursor(Cursor.WAIT_CURSOR));
try
{
RootFileReader reader = new RootFileReader(file);
TreeModel model = new RootHistogramTreeModel(reader)
{
public void handleException(IOException x)
{
error(x);
x.printStackTrace();
super.handleException(x);
}
};
tree.setModel(model);
tree.setRowHeight(20);
tree.setLargeModel(true);
}
finally
{
setCursor(old);
}
}
public void setRootFile(URL url) throws IOException
{
Cursor old = getCursor();
setCursor(Cursor.getPredefinedCursor(Cursor.WAIT_CURSOR));
try
{
RootFileReader reader = new RootFileReader(url);
TreeModel model = new RootHistogramTreeModel(reader)
{
public void handleException(IOException x)
{
error(x);
x.printStackTrace();
super.handleException(x);
}
};
tree.setModel(model);
tree.setRowHeight(20);
tree.setLargeModel(true);
}
finally
{
setCursor(old);
}
}
public static void main(String[] argv) throws IOException
{
if (argv.length > 1) usage();
if (argv.length == 1 && argv[0].startsWith("-")) usage();
URL.setURLStreamHandlerFactory(new RootURLStreamFactory());
JFrame frame = new JFrame("Root Histogram Browser");
RootHistogramBrowser browser = new RootHistogramBrowser();
Authenticator.setDefault(new RootAuthenticator(browser));
URLConnection.setDefaultAllowUserInteraction(true);
if (argv.length == 1)
{
if (argv[0].startsWith("root:")) browser.setRootFile(new URL(argv[0]));
else browser.setRootFile(new File(argv[0]));
}
frame.setContentPane(browser);
// Make this exit when the close button is clicked.
frame.addWindowListener(new WindowAdapter()
{
public void windowClosing(WindowEvent e)
{
System.exit(0);
}
});
frame.pack();
frame.setVisible(true);
}
private static void usage()
{
System.out.println("java RootHistogramBrowser []");
System.exit(0);
}
public void valueChanged(TreeSelectionEvent event)
{
Cursor old = getCursor();
setCursor(Cursor.getPredefinedCursor(Cursor.WAIT_CURSOR));
try
{
plot.removeAllData();
TreePath path = tree.getSelectionPath();
if (path != null)
{
TKey key = (TKey) path.getLastPathComponent();
Object node = key.getObject();
if (node instanceof TH1)
{
DataSource ds = RootHistogramAdapter.create((TH1) node);
if (ds != null)
{
plot.setTitle(ds.getTitle());
plot.addData(ds).show(true);
}
}
}
}
catch (IOException x)
{
x.printStackTrace();
error(x);
}
catch (RootClassNotFound x)
{
x.printStackTrace();
error(x);
}
finally
{
setCursor(old);
}
}
private void error(Exception x)
{
JOptionPane.showMessageDialog(this,x.getMessage(),"Error",JOptionPane.ERROR_MESSAGE);
}
private class RootMenuBar extends JMenuBar
{
RootMenuBar()
{
add(new RootFileMenu());
add(new RootHelpMenu());
}
}
private class RootFileMenu extends JMenu
{
RootFileMenu()
{
super("File");
add(new JMenuItem("Open File...")
{
public void fireActionPerformed(ActionEvent e)
{
JFileChooser dlg = null;
// HACK to work around JavaWebStart bug.
try
{
if (System.getSecurityManager() != null &&
System.getProperty("os.name").indexOf("indows")>0 &&
System.getProperty("java.version").startsWith("1.3"))
{
dlg = new JFileChooser("Open Root File...");
}
}
catch (SecurityException x) {}
if (dlg == null) dlg = new JFileChooser("Open Root File...");
dlg.setFileFilter(new RootFileFilter());
int rc = dlg.showOpenDialog(RootHistogramBrowser.this);
if (rc == JFileChooser.APPROVE_OPTION)
{
try
{
setRootFile(dlg.getSelectedFile());
}
catch (IOException x)
{
error(x);
}
}
}
});
add(new JMenuItem("Open URL...")
{
public void fireActionPerformed(ActionEvent e)
{
String url = JOptionPane.showInputDialog(RootHistogramBrowser.this,"URL","root://");
if (url != null)
{
try
{
setRootFile(new URL(url));
}
catch (IOException x)
{
error(x);
}
}
}
});
add(new JMenuItem("Exit")
{
public void fireActionPerformed(ActionEvent e)
{
System.exit(0);
}
});
}
}
private class RootHelpMenu extends JMenu
{
RootHelpMenu()
{
super("Help");
add(new JMenuItem("About...")
{
public void fireActionPerformed(ActionEvent e)
{
JOptionPane.showMessageDialog(RootHistogramBrowser.this,aboutMessage);
}
});
}
}
private static class RootFileFilter extends javax.swing.filechooser.FileFilter
{
public boolean accept(File file)
{
return file.isDirectory() || file.getName().endsWith(".root");
}
public String getDescription()
{
return "Root Files (*.root)";
}
}
}