/* * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. */ /* * GUI.java * Copyright (C) 2007 University of Waikato, Hamilton, New Zealand * */ package weka.classifiers.bayes.net; import weka.classifiers.bayes.net.MarginCalculator.JunctionTreeNode; import weka.core.FastVector; import weka.core.Instances; import weka.core.OptionHandler; import weka.core.SerializedObject; import weka.core.Utils; import weka.core.converters.AbstractFileLoader; import weka.core.converters.AbstractFileSaver; import weka.core.converters.ArffSaver; import weka.core.converters.ConverterUtils; import weka.gui.ConverterFileChooser; import weka.gui.ExtensionFileFilter; import weka.gui.GenericObjectEditor; import weka.gui.LookAndFeel; import weka.gui.PropertyDialog; import weka.gui.graphvisualizer.BIFFormatException; import weka.gui.graphvisualizer.BIFParser; import weka.gui.graphvisualizer.GraphNode; import weka.gui.graphvisualizer.HierarchicalBCEngine; import weka.gui.graphvisualizer.LayoutCompleteEvent; import weka.gui.graphvisualizer.LayoutCompleteEventListener; import weka.gui.graphvisualizer.LayoutEngine; import weka.gui.visualize.PrintablePanel; import java.awt.BorderLayout; import java.awt.Color; import java.awt.Container; import java.awt.Dimension; import java.awt.Font; import java.awt.FontMetrics; import java.awt.Frame; import java.awt.Graphics; import java.awt.Graphics2D; import java.awt.GridBagConstraints; import java.awt.GridBagLayout; import java.awt.GridLayout; import java.awt.Insets; import java.awt.Rectangle; import java.awt.RenderingHints; import java.awt.event.ActionEvent; import java.awt.event.ActionListener; import java.awt.event.MouseAdapter; import java.awt.event.MouseEvent; import java.awt.event.MouseMotionAdapter; import java.awt.event.WindowAdapter; import java.awt.event.WindowEvent; import java.awt.image.BufferedImage; import java.awt.print.PageFormat; import java.awt.print.Printable; import java.awt.print.PrinterException; import java.awt.print.PrinterJob; import java.beans.PropertyEditor; import java.io.File; import java.io.FileReader; import java.io.FileWriter; import java.io.IOException; import java.util.Random; import javax.swing.AbstractAction; import javax.swing.Action; import javax.swing.BorderFactory; import javax.swing.ImageIcon; import javax.swing.JButton; import javax.swing.JCheckBox; import javax.swing.JCheckBoxMenuItem; import javax.swing.JDialog; import javax.swing.JFileChooser; import javax.swing.JFrame; import javax.swing.JLabel; import javax.swing.JMenu; import javax.swing.JMenuBar; import javax.swing.JMenuItem; import javax.swing.JOptionPane; import javax.swing.JPanel; import javax.swing.JPopupMenu; import javax.swing.JScrollPane; import javax.swing.JTable; import javax.swing.JTextField; import javax.swing.JToolBar; import javax.swing.KeyStroke; import javax.swing.table.AbstractTableModel; /** * GUI interface to Bayesian Networks. Allows editing Bayesian networks * on screen and provides GUI interface to various Bayesian network facilities * in Weka, including random network generation, data set generation and * Bayesion network inference. * * @author Remco Bouckaert (remco@cs.waikato.ac.nz) * @version $Revision: 4902 $ */ public class GUI extends JPanel implements LayoutCompleteEventListener { /** for serialization */ private static final long serialVersionUID = -2038911085935515624L; /** The current LayoutEngine */ protected LayoutEngine m_layoutEngine; /** Panel actually displaying the graph */ protected GraphPanel m_GraphPanel; /** Container of Bayesian network */ EditableBayesNet m_BayesNet = new EditableBayesNet(true); /** String containing file name storing current network */ protected String m_sFileName = ""; /** used for calculating marginals in Bayesian netwowrks */ MarginCalculator m_marginCalculator = null; /** * used for calculating marginals in Bayesian netwowrks when evidence is * present */ MarginCalculator m_marginCalculatorWithEvidence = null; /** flag indicating whether marginal distributions of each of the nodes * should be shown in display. */ boolean m_bViewMargins = false; boolean m_bViewCliques = false; /** The menu bar */ private JMenuBar m_menuBar; /** data selected from file. Used to train a Bayesian network on */ Instances m_Instances = null; /** Text field for specifying zoom */ final JTextField m_jTfZoom; /** toolbar containing buttons at top of window */ final JToolBar m_jTbTools; /** status bar at bottom of window */ final JLabel m_jStatusBar; /** TextField for node's width */ private final JTextField m_jTfNodeWidth = new JTextField(3); /** TextField for nodes height */ private final JTextField m_jTfNodeHeight = new JTextField(3); /** this contains the m_GraphPanel GraphPanel */ JScrollPane m_jScrollPane; /** path for icons */ private final String ICONPATH = "weka/classifiers/bayes/net/icons/"; /** current zoom value */ private double m_fScale = 1; /** standard width of node */ private int m_nNodeHeight = 2 * getFontMetrics(getFont()).getHeight(); /** standard height of node */ final static int DEFAULT_NODE_WIDTH = 50; private int m_nNodeWidth = DEFAULT_NODE_WIDTH; /** width of node, allowing for some padding */ final static int PADDING = 10; private int m_nPaddedNodeWidth = DEFAULT_NODE_WIDTH + PADDING; /** used when using zoomIn and zoomOut buttons */ private int [] m_nZoomPercents = { 10, 25, 50, 75, 100, 125, 150, 175, 200, 225, 250, 275, 300, 350, 400, 450, 500, 550, 600, 650, 700, 800, 900, 999 }; /** actions triggered by GUI events */ Action a_new = new ActionNew(); Action a_quit = new ActionQuit(); Action a_save = new ActionSave(); ActionExport a_export = new ActionExport(); ActionPrint a_print = new ActionPrint(); Action a_load = new ActionLoad(); Action a_zoomin = new ActionZoomIn(); Action a_zoomout = new ActionZoomOut(); Action a_layout = new ActionLayout(); Action a_saveas = new ActionSaveAs(); Action a_viewtoolbar = new ActionViewToolbar(); Action a_viewstatusbar = new ActionViewStatusbar(); Action a_networkgenerator = new ActionGenerateNetwork(); Action a_datagenerator = new ActionGenerateData(); Action a_datasetter = new ActionSetData(); Action a_learn = new ActionLearn(); Action a_learnCPT = new ActionLearnCPT(); Action a_help = new ActionHelp(); Action a_about = new ActionAbout(); ActionAddNode a_addnode = new ActionAddNode(); Action a_delnode = new ActionDeleteNode(); Action a_cutnode = new ActionCutNode(); Action a_copynode = new ActionCopyNode(); Action a_pastenode = new ActionPasteNode(); Action a_selectall = new ActionSelectAll(); Action a_addarc = new ActionAddArc(); Action a_delarc = new ActionDeleteArc(); Action a_undo = new ActionUndo(); Action a_redo= new ActionRedo(); Action a_alignleft = new ActionAlignLeft(); Action a_alignright = new ActionAlignRight(); Action a_aligntop = new ActionAlignTop(); Action a_alignbottom = new ActionAlignBottom(); Action a_centerhorizontal = new ActionCenterHorizontal(); Action a_centervertical = new ActionCenterVertical(); Action a_spacehorizontal = new ActionSpaceHorizontal(); Action a_spacevertical = new ActionSpaceVertical(); /** node currently selected through right clicking */ int m_nCurrentNode = -1; /** selection of nodes */ Selection m_Selection = new Selection(); /** selection rectangle drawn through dragging with left mouse button */ Rectangle m_nSelectedRect = null; class Selection { FastVector m_selected; public Selection() { m_selected = new FastVector(); } // c'tor public FastVector getSelected() {return m_selected;} void updateGUI() { if (m_selected.size() > 0) { a_cutnode.setEnabled(true); a_copynode.setEnabled(true); } else { a_cutnode.setEnabled(false); a_copynode.setEnabled(false); } if (m_selected.size() > 1) { a_alignleft.setEnabled(true); a_alignright.setEnabled(true); a_aligntop.setEnabled(true); a_alignbottom.setEnabled(true); a_centerhorizontal.setEnabled(true); a_centervertical.setEnabled(true); a_spacehorizontal.setEnabled(true); a_spacevertical.setEnabled(true); } else { a_alignleft.setEnabled(false); a_alignright.setEnabled(false); a_aligntop.setEnabled(false); a_alignbottom.setEnabled(false); a_centerhorizontal.setEnabled(false); a_centervertical.setEnabled(false); a_spacehorizontal.setEnabled(false); a_spacevertical.setEnabled(false); } } // updateGUI public void addToSelection(int nNode) { for (int iNode = 0; iNode < m_selected.size(); iNode++) { if (nNode == (Integer) m_selected.elementAt(iNode)) { return; } } m_selected.addElement(nNode); updateGUI(); } // addToSelection public void addToSelection(int [] iNodes) { for (int iNode = 0; iNode < iNodes.length; iNode++) { addToSelection(iNodes[iNode]); } updateGUI(); } // addToSelection public void addToSelection(Rectangle selectedRect) { for (int iNode = 0; iNode < m_BayesNet.getNrOfNodes(); iNode++) { if (contains(selectedRect, iNode)) { addToSelection(iNode); } } } // addToSelection public void selectAll() { m_selected.removeAllElements(); for (int iNode = 0; iNode < m_BayesNet.getNrOfNodes(); iNode++) { m_selected.addElement(iNode); } updateGUI(); } // selectAll boolean contains(Rectangle rect, int iNode) { return rect.intersects((m_BayesNet.getPositionX(iNode)) * m_fScale, (m_BayesNet.getPositionY(iNode)) * m_fScale, m_nPaddedNodeWidth * m_fScale, m_nNodeHeight * m_fScale); } // contains public void removeFromSelection(int nNode) { for (int iNode = 0; iNode < m_selected.size(); iNode++) { if (nNode == (Integer) m_selected.elementAt(iNode)) { m_selected.removeElementAt(iNode); } } updateGUI(); } // removeFromSelection public void toggleSelection(int nNode) { for (int iNode = 0; iNode < m_selected.size(); iNode++) { if (nNode == (Integer) m_selected.elementAt(iNode)) { m_selected.removeElementAt(iNode); updateGUI(); return; } } addToSelection(nNode); } // toggleSelection public void toggleSelection(Rectangle selectedRect) { for (int iNode = 0; iNode < m_BayesNet.getNrOfNodes(); iNode++) { if (contains(selectedRect, iNode)) { toggleSelection(iNode); } } } // toggleSelection public void clear() { m_selected.removeAllElements(); updateGUI(); } public void draw(Graphics g) { if (m_selected.size() == 0) { return; } for (int iNode = 0; iNode < m_selected.size(); iNode++) { int nNode = (Integer) m_selected.elementAt(iNode); int nPosX = m_BayesNet.getPositionX(nNode); int nPosY = m_BayesNet.getPositionY(nNode); g.setColor(Color.BLACK); int nXRC = nPosX + m_nPaddedNodeWidth - m_nNodeWidth - (m_nPaddedNodeWidth - m_nNodeWidth) / 2; int nYRC = nPosY; int d = 5; g.fillRect(nXRC, nYRC, d, d); g.fillRect(nXRC, nYRC + m_nNodeHeight, d, d); g.fillRect(nXRC + m_nNodeWidth, nYRC, d, d); g.fillRect(nXRC + m_nNodeWidth, nYRC + m_nNodeHeight, d, d); } } // draw } // Selection ClipBoard m_clipboard = new ClipBoard(); class ClipBoard { String m_sText = null; public ClipBoard() { if (a_pastenode != null) { a_pastenode.setEnabled(false); } } public boolean hasText() {return m_sText != null;} public String getText() { return m_sText; } public void setText(String sText) { m_sText = sText; a_pastenode.setEnabled(true); } } // class ClipBoard /** Base class used for definining actions * with a name, tool tip text, possibly an icon and accelerator key. * */ class MyAction extends AbstractAction { /** for serialization */ private static final long serialVersionUID = -2038911111935517L; public MyAction(String sName, String sToolTipText, String sIcon, String sAcceleratorKey) { super(sName); //setToolTipText(sToolTipText); putValue(Action.SHORT_DESCRIPTION, sToolTipText); putValue(Action.LONG_DESCRIPTION, sToolTipText); if (sAcceleratorKey.length() > 0) { KeyStroke keyStroke = KeyStroke.getKeyStroke(sAcceleratorKey); putValue(Action.ACCELERATOR_KEY, keyStroke); } putValue(Action.MNEMONIC_KEY, (int) sName.charAt(0)); java.net.URL tempURL = ClassLoader.getSystemResource(ICONPATH + sIcon + ".png"); if (tempURL != null) { putValue(Action.SMALL_ICON, new ImageIcon(tempURL)); } else { putValue(Action.SMALL_ICON, new ImageIcon(new BufferedImage(20,20, BufferedImage.TYPE_4BYTE_ABGR))); //System.err.println(ICONPATH + sIcon + ".png not found for weka.gui.graphvisualizer.Graph"); } } // c'tor /* Place holder. Should be implemented by derived classes. * (non-Javadoc) * @see java.awt.event.ActionListener#actionPerformed(java.awt.event.ActionEvent) */ public void actionPerformed(ActionEvent ae) {} } // class MyAction class ActionGenerateNetwork extends MyAction { /** for serialization */ private static final long serialVersionUID = -2038911085935517L; public ActionGenerateNetwork() { super("Generate Network", "Generate Random Bayesian Network", "generate.network", "ctrl N"); } // c'tor int m_nNrOfNodes = 10; int m_nNrOfArcs = 15; int m_nCardinality = 2; int m_nSeed = 123; JDialog dlg = null; public void actionPerformed(ActionEvent ae) { if (dlg == null) { dlg = new JDialog(); dlg.setTitle("Generate Random Bayesian Network Options"); final JLabel jLbNrOfNodes = new JLabel("Nr of nodes"); final JTextField jTfNrOfNodes = new JTextField(3); jTfNrOfNodes.setHorizontalAlignment(JTextField.CENTER); jTfNrOfNodes.setText("" + m_nNrOfNodes); final JLabel jLbNrOfArcs = new JLabel("Nr of arcs"); final JTextField jTfNrOfArcs = new JTextField(3); jTfNrOfArcs.setHorizontalAlignment(JTextField.CENTER); jTfNrOfArcs.setText("" + m_nNrOfArcs); final JLabel jLbCardinality = new JLabel("Cardinality"); final JTextField jTfCardinality = new JTextField(3); jTfCardinality.setHorizontalAlignment(JTextField.CENTER); jTfCardinality.setText("" + m_nCardinality); final JLabel jLbSeed = new JLabel("Random seed"); final JTextField jTfSeed = new JTextField(3); jTfSeed.setHorizontalAlignment(JTextField.CENTER); jTfSeed.setText("" + m_nSeed); JButton jBtGo; jBtGo = new JButton("Generate Network"); jBtGo.addActionListener(new ActionListener() { public void actionPerformed(ActionEvent ae) { try { BayesNetGenerator generator = new BayesNetGenerator(); m_BayesNet = generator; m_BayesNet.clearUndoStack(); String[] options = new String[8]; options[0] = "-N"; options[1] = "" + jTfNrOfNodes.getText(); options[2] = "-A"; options[3] = "" + jTfNrOfArcs.getText(); options[4] = "-C"; options[5] = "" + jTfCardinality.getText(); options[6] = "-S"; options[7] = "" + jTfSeed.getText(); generator.setOptions(options); generator.generateRandomNetwork(); // Convert to EditableBayesNet // This ensures the getOptions() called by GenericObjectEditor to get the correct result. BIFReader bifReader = new BIFReader(); bifReader.processString(m_BayesNet.toXMLBIF03()); m_BayesNet = new EditableBayesNet(bifReader); updateStatus(); layoutGraph(); a_datagenerator.setEnabled(true); m_Instances = null;; a_learn.setEnabled(false); a_learnCPT.setEnabled(false); dlg.setVisible(false); } catch (Exception e) { e.printStackTrace(); } } }); JButton jBtCancel; jBtCancel = new JButton("Cancel"); jBtCancel.setMnemonic('C'); jBtCancel.addActionListener(new ActionListener() { public void actionPerformed(ActionEvent ae) { dlg.setVisible(false); } }); GridBagConstraints gbc = new GridBagConstraints(); dlg.setLayout(new GridBagLayout()); Container c = new Container(); c.setLayout(new GridBagLayout()); gbc.gridwidth = 2; gbc.insets = new Insets(8, 0, 0, 0); gbc.anchor = GridBagConstraints.NORTHWEST; gbc.gridwidth = GridBagConstraints.RELATIVE; gbc.fill = GridBagConstraints.HORIZONTAL; c.add(jLbNrOfNodes, gbc); gbc.gridwidth = GridBagConstraints.REMAINDER; c.add(jTfNrOfNodes, gbc); gbc.gridwidth = GridBagConstraints.RELATIVE; c.add(jLbNrOfArcs, gbc); gbc.gridwidth = GridBagConstraints.REMAINDER; c.add(jTfNrOfArcs, gbc); gbc.gridwidth = GridBagConstraints.RELATIVE; c.add(jLbCardinality, gbc); gbc.gridwidth = GridBagConstraints.REMAINDER; c.add(jTfCardinality, gbc); gbc.gridwidth = GridBagConstraints.RELATIVE; c.add(jLbSeed, gbc); gbc.gridwidth = GridBagConstraints.REMAINDER; c.add(jTfSeed, gbc); gbc.fill = GridBagConstraints.HORIZONTAL; dlg.add(c, gbc); dlg.add(jBtGo); gbc.gridwidth = GridBagConstraints.REMAINDER; dlg.add(jBtCancel); } dlg.pack(); dlg.setLocation(100, 100); dlg.setVisible(true); dlg.setSize(dlg.getPreferredSize()); dlg.setVisible(false); dlg.setVisible(true); dlg.repaint(); } // actionPerformed } // class ActionGenerate class ActionGenerateData extends MyAction { /** for serialization */ private static final long serialVersionUID = -2038911085935516L; public ActionGenerateData() { super("Generate Data", "Generate Random Instances from Network", "generate.data", "ctrl D"); } // c'tor int m_nNrOfInstances = 100; int m_nSeed = 1234; String m_sFile = ""; JDialog dlg = null; public void actionPerformed(ActionEvent ae) { if (dlg == null) { dlg = new JDialog(); dlg.setTitle("Generate Random Data Options"); final JLabel jLbNrOfInstances = new JLabel("Nr of instances"); final JTextField jTfNrOfInstances = new JTextField(3); jTfNrOfInstances.setHorizontalAlignment(JTextField.CENTER); jTfNrOfInstances.setText("" + m_nNrOfInstances); final JLabel jLbSeed = new JLabel("Random seed"); final JTextField jTfSeed = new JTextField(3); jTfSeed.setHorizontalAlignment(JTextField.CENTER); jTfSeed.setText("" + m_nSeed); final JLabel jLbFile = new JLabel("Output file (optional)"); final JTextField jTfFile = new JTextField(12); jTfFile.setHorizontalAlignment(JTextField.CENTER); jTfFile.setText(m_sFile); JButton jBtGo; jBtGo = new JButton("Generate Data"); jBtGo.addActionListener(new ActionListener() { public void actionPerformed(ActionEvent ae) { try { String tmpfilename = "tmp.bif.file.xml"; BayesNetGenerator generator = new BayesNetGenerator(); String[] options = new String[4]; options[0] = "-M"; options[1] = "" + jTfNrOfInstances.getText(); options[2] = "-F"; options[3] = tmpfilename; FileWriter outfile = new FileWriter(tmpfilename); StringBuffer text = new StringBuffer(); if (m_marginCalculator == null) { m_marginCalculator = new MarginCalculator(); m_marginCalculator.calcMargins(m_BayesNet); } text.append(m_marginCalculator.toXMLBIF03()); outfile.write(text.toString()); outfile.close(); generator.setOptions(options); generator.generateRandomNetwork(); generator.generateInstances(); m_Instances = generator.m_Instances; a_learn.setEnabled(true); a_learnCPT.setEnabled(true); m_sFile = jTfFile.getText(); if (m_sFile != null && !m_sFile.equals("")) { AbstractFileSaver saver = ConverterUtils.getSaverForFile(m_sFile); // no idea what the format is, so let's save it as ARFF file if (saver == null) saver = new ArffSaver(); saver.setFile(new File(m_sFile)); saver.setInstances(m_Instances); saver.writeBatch(); } } catch (Exception e) { e.printStackTrace(); } dlg.setVisible(false); } }); JButton jBtFile = new JButton("Browse"); jBtFile.addActionListener(new ActionListener() { public void actionPerformed(ActionEvent ae) { ConverterFileChooser fc = new ConverterFileChooser(System.getProperty("user.dir")); fc.setDialogTitle("Save Instances As"); int rval = fc.showSaveDialog(GUI.this); if (rval == JFileChooser.APPROVE_OPTION) { String filename = fc.getSelectedFile().toString(); jTfFile.setText(filename); } dlg.setVisible(true); } }); JButton jBtCancel; jBtCancel = new JButton("Cancel"); jBtCancel.setMnemonic('C'); jBtCancel.addActionListener(new ActionListener() { public void actionPerformed(ActionEvent ae) { dlg.setVisible(false); } }); GridBagConstraints gbc = new GridBagConstraints(); dlg.setLayout(new GridBagLayout()); Container c = new Container(); c.setLayout(new GridBagLayout()); gbc.gridwidth = 2; gbc.insets = new Insets(8, 0, 0, 0); gbc.anchor = GridBagConstraints.NORTHWEST; gbc.gridwidth = GridBagConstraints.RELATIVE; gbc.fill = GridBagConstraints.HORIZONTAL; c.add(jLbNrOfInstances, gbc); gbc.gridwidth = GridBagConstraints.REMAINDER; c.add(jTfNrOfInstances, gbc); gbc.gridwidth = GridBagConstraints.RELATIVE; c.add(jLbSeed, gbc); gbc.gridwidth = GridBagConstraints.REMAINDER; c.add(jTfSeed, gbc); gbc.gridwidth = GridBagConstraints.RELATIVE; c.add(jLbFile, gbc); gbc.gridwidth = GridBagConstraints.REMAINDER; c.add(jTfFile, gbc); gbc.gridwidth = GridBagConstraints.REMAINDER; c.add(jBtFile, gbc); gbc.fill = GridBagConstraints.HORIZONTAL; dlg.add(c, gbc); dlg.add(jBtGo); gbc.gridwidth = GridBagConstraints.REMAINDER; dlg.add(jBtCancel); } dlg.setLocation(100, 100); dlg.setVisible(true); dlg.setSize(dlg.getPreferredSize()); dlg.setVisible(false); dlg.setVisible(true); dlg.repaint(); } // actionPerformed } // class ActionGenerateData class ActionLearn extends MyAction { /** for serialization */ private static final long serialVersionUID = -2038911085935516L; public ActionLearn() { super("Learn Network", "Learn Bayesian Network", "learn", "ctrl L"); setEnabled(false); } // c'tor JDialog dlg = null; public void actionPerformed(ActionEvent ae) { if (dlg == null) { dlg = new JDialog(); dlg.setTitle("Learn Bayesian Network"); final JButton jBtOptions = new JButton("Options"); jBtOptions.addActionListener(new ActionListener() { public void actionPerformed(ActionEvent ae) { //m_BayesNet = new EditableBayesNet(); try { GenericObjectEditor.registerEditors(); GenericObjectEditor ce = new GenericObjectEditor(true); ce.setClassType(weka.classifiers.Classifier.class); ce.setValue(m_BayesNet); PropertyDialog pd; if (PropertyDialog.getParentDialog(GUI.this) != null) pd = new PropertyDialog(PropertyDialog.getParentDialog(GUI.this), ce, 100, 100); else pd = new PropertyDialog(PropertyDialog.getParentFrame(GUI.this), ce, 100, 100); pd.addWindowListener(new WindowAdapter() { public void windowClosing(WindowEvent e) { PropertyEditor pe = ((PropertyDialog) e.getSource()).getEditor(); Object c = (Object) pe.getValue(); String options = ""; if (c instanceof OptionHandler) { options = Utils.joinOptions(((OptionHandler) c).getOptions()); try { m_BayesNet.setOptions(((OptionHandler) c).getOptions()); } catch (Exception e2) { e2.printStackTrace(); } } System.out.println(c.getClass().getName() + " " + options); System.exit(0); } }); pd.setVisible(true); } catch (Exception ex) { ex.printStackTrace(); System.err.println(ex.getMessage()); } m_BayesNet.clearUndoStack(); a_undo.setEnabled(false); a_redo.setEnabled(false); } }); final JTextField jTfOptions = new JTextField(40); jTfOptions.setHorizontalAlignment(JTextField.CENTER); jTfOptions.setText("" + Utils.joinOptions(m_BayesNet.getOptions())); JButton jBtGo; jBtGo = new JButton("Learn"); jBtGo.addActionListener(new ActionListener() { public void actionPerformed(ActionEvent ae) { try { m_BayesNet.buildClassifier(m_Instances); layoutGraph(); updateStatus(); m_BayesNet.clearUndoStack(); dlg.setVisible(false); } catch (Exception e) { e.printStackTrace(); } dlg.setVisible(false); } }); JButton jBtCancel; jBtCancel = new JButton("Cancel"); jBtCancel.setMnemonic('C'); jBtCancel.addActionListener(new ActionListener() { public void actionPerformed(ActionEvent ae) { dlg.setVisible(false); } }); GridBagConstraints gbc = new GridBagConstraints(); dlg.setLayout(new GridBagLayout()); Container c = new Container(); c.setLayout(new GridBagLayout()); gbc.gridwidth = 2; gbc.insets = new Insets(8, 0, 0, 0); gbc.anchor = GridBagConstraints.NORTHWEST; gbc.gridwidth = GridBagConstraints.RELATIVE; gbc.fill = GridBagConstraints.HORIZONTAL; c.add(jBtOptions, gbc); gbc.gridwidth = GridBagConstraints.REMAINDER; c.add(jTfOptions, gbc); gbc.fill = GridBagConstraints.HORIZONTAL; dlg.add(c, gbc); dlg.add(jBtGo); gbc.gridwidth = GridBagConstraints.REMAINDER; dlg.add(jBtCancel); } dlg.setLocation(100, 100); dlg.setVisible(true); dlg.setSize(dlg.getPreferredSize()); dlg.setVisible(false); dlg.setVisible(true); dlg.repaint(); } // actionPerformed } // class ActionLearn class ActionLearnCPT extends MyAction { /** for serialization */ private static final long serialVersionUID = -2022211085935516L; public ActionLearnCPT() { super("Learn CPT", "Learn conditional probability tables", "learncpt", ""); setEnabled(false); } // c'tor public void actionPerformed(ActionEvent ae) { if (m_Instances == null) { JOptionPane.showMessageDialog(null, "Select instances to learn from first (menu Tools/Set Data)"); return; } try { m_BayesNet.setData(m_Instances); } catch (Exception e) { JOptionPane.showMessageDialog(null, "Data set is not compatible with network.\n"+e.getMessage() + "\nChoose other instances (menu Tools/Set Data)"); return; } try { m_BayesNet.estimateCPTs(); m_BayesNet.clearUndoStack(); } catch (Exception e) { e.printStackTrace(); } updateStatus(); } // actionPerformed } // class ActionLearnCPT class ActionSetData extends MyAction { /** for serialization */ private static final long serialVersionUID = -2038911085935519L; public ActionSetData() { super("Set Data", "Set Data File", "setdata", "ctrl A"); } // c'tor public void actionPerformed(ActionEvent ae) { ConverterFileChooser fc = new ConverterFileChooser(System.getProperty("user.dir")); fc.setDialogTitle("Set Data File"); int rval = fc.showOpenDialog(GUI.this); if (rval == JFileChooser.APPROVE_OPTION) { AbstractFileLoader loader = fc.getLoader(); try { if (loader != null) m_Instances = loader.getDataSet(); if (m_Instances.classIndex() == -1) m_Instances.setClassIndex(m_Instances.numAttributes() - 1); a_learn.setEnabled(true); a_learnCPT.setEnabled(true); repaint(); } catch (Exception e) { e.printStackTrace(); } } } } // class ActionSetData class ActionUndo extends MyAction { /** for serialization */ private static final long serialVersionUID = -3038910085935519L; public ActionUndo() { super("Undo", "Undo", "undo", "ctrl Z"); setEnabled(false); } // c'tor public boolean isEnabled() { return m_BayesNet.canUndo(); } public void actionPerformed(ActionEvent ae) { String sMsg = m_BayesNet.undo(); m_jStatusBar.setText("Undo action performed: " + sMsg); //if (!sMsg.equals("")) { // JOptionPane.showMessageDialog(null, sMsg, "Undo action successful", JOptionPane.INFORMATION_MESSAGE); //} a_redo.setEnabled(m_BayesNet.canRedo()); a_undo.setEnabled(m_BayesNet.canUndo()); m_Selection.clear(); updateStatus(); repaint(); } } // ActionUndo class ActionRedo extends MyAction { /** for serialization */ private static final long serialVersionUID = -4038910085935519L; public ActionRedo() { super("Redo", "Redo", "redo", "ctrl Y"); setEnabled(false); } // c'tor public boolean isEnabled() { return m_BayesNet.canRedo(); } public void actionPerformed(ActionEvent ae) { String sMsg = m_BayesNet.redo(); m_jStatusBar.setText("Redo action performed: " + sMsg); //if (!sMsg.equals("")) { // JOptionPane.showMessageDialog(null, sMsg, "Redo action successful", JOptionPane.INFORMATION_MESSAGE); //} m_Selection.clear(); updateStatus(); repaint(); } } // ActionRedo class ActionAddNode extends MyAction { /** for serialization */ private static final long serialVersionUID = -2038910085935519L; public ActionAddNode() { super("Add Node", "Add Node", "addnode", ""); } // c'tor JDialog dlg = null; JTextField jTfName = new JTextField(20); JTextField jTfCard = new JTextField(3); int m_X = Integer.MAX_VALUE; int m_Y; public void addNode(int nX, int nY) { m_X = nX; m_Y = nY; addNode(); } // addNode void addNode() { if (dlg == null) { dlg = new JDialog(); dlg.setTitle("Add node"); JLabel jLbName = new JLabel("Name"); jTfName.setHorizontalAlignment(JTextField.CENTER); JLabel jLbCard = new JLabel("Cardinality"); jTfCard.setHorizontalAlignment(JTextField.CENTER); jTfCard.setText("2"); JButton jBtCancel; jBtCancel = new JButton("Cancel"); jBtCancel.setMnemonic('C'); jBtCancel.addActionListener(new ActionListener() { public void actionPerformed(ActionEvent ae) { dlg.setVisible(false); } }); JButton jBtOk = new JButton("Ok"); jBtOk.setMnemonic('O'); jBtOk.addActionListener(new ActionListener() { public void actionPerformed(ActionEvent ae) { String sName = jTfName.getText(); if (sName.length() <= 0) { JOptionPane.showMessageDialog(null, "Name should have at least one character"); return; } int nCard = new Integer(jTfCard.getText()).intValue(); if (nCard <= 1) { JOptionPane.showMessageDialog(null, "Cardinality should be larger than 1"); return; } try { if (m_X < Integer.MAX_VALUE) { m_BayesNet.addNode(sName, nCard, m_X, m_Y); } else { m_BayesNet.addNode(sName, nCard); } m_jStatusBar.setText(m_BayesNet.lastActionMsg()); a_undo.setEnabled(true); a_redo.setEnabled(false); //GraphNode n = new GraphNode("id" + m_nodes.size(), sName); //n.probs = m_BayesNet.getDistribution(sName); //n.outcomes = m_BayesNet.getValues(sName); //n.x = 100 + m_nodes.size() * 10; //n.y = 100 + m_nodes.size() * 10; //m_nodes.addElement(n); } catch (Exception e) { e.printStackTrace(); } repaint(); dlg.setVisible(false); } }); dlg.setLayout(new GridLayout(3, 2, 10, 10)); dlg.add(jLbName); dlg.add(jTfName); dlg.add(jLbCard); dlg.add(jTfCard); dlg.add(jBtOk); dlg.add(jBtCancel); dlg.setSize(dlg.getPreferredSize()); } jTfName.setText("Node" + (m_BayesNet.getNrOfNodes() + 1)); dlg.setVisible(true); } // addNode public void actionPerformed(ActionEvent ae) { m_X = Integer.MAX_VALUE; addNode(); } } // class ActionAddNode class ActionDeleteNode extends MyAction { /** for serialization */ private static final long serialVersionUID = -2038912085935519L; public ActionDeleteNode() { super("Delete Node", "Delete Node", "delnode", "DELETE"); } // c'tor public void actionPerformed(ActionEvent ae) { if (m_Selection.getSelected().size() > 0) { m_BayesNet.deleteSelection(m_Selection.getSelected()); m_jStatusBar.setText(m_BayesNet.lastActionMsg()); m_Selection.clear(); updateStatus(); repaint(); } else { String[] options = new String[m_BayesNet.getNrOfNodes()]; for (int i = 0; i < options.length; i++) { options[i] = m_BayesNet.getNodeName(i); } String sResult = (String) JOptionPane.showInputDialog(null, "Select node to delete", "Nodes", 0, null, options, options[0]); if (sResult != null && !sResult.equals("")) { int iNode = m_BayesNet.getNode2(sResult); deleteNode(iNode); } } } } // class ActionDeleteNode class ActionCopyNode extends MyAction { /** for serialization */ private static final long serialVersionUID = -2038732085935519L; public ActionCopyNode() { super("Copy", "Copy Nodes", "copy", "ctrl C"); } // c'tor public ActionCopyNode(String sName, String sToolTipText, String sIcon, String sAcceleratorKey) { super(sName, sToolTipText, sIcon, sAcceleratorKey); } // c'rot public void actionPerformed(ActionEvent ae) { copy(); } public void copy() { String sXML = m_BayesNet.toXMLBIF03(m_Selection.getSelected()); m_clipboard.setText(sXML); } // copy } // class ActionCopyNode class ActionCutNode extends ActionCopyNode { /** for serialization */ private static final long serialVersionUID = -2038822085935519L; public ActionCutNode() { super("Cut", "Cut Nodes", "cut", "ctrl X"); } // c'tor public void actionPerformed(ActionEvent ae) { copy(); m_BayesNet.deleteSelection(m_Selection.getSelected()); m_jStatusBar.setText(m_BayesNet.lastActionMsg()); m_Selection.clear(); a_undo.setEnabled(true); a_redo.setEnabled(false); repaint(); } } // class ActionCutNode class ActionPasteNode extends MyAction { /** for serialization */ private static final long serialVersionUID = -2038732085935519L; public ActionPasteNode() { super("Paste", "Paste Nodes", "paste", "ctrl V"); } // c'tor public void actionPerformed(ActionEvent ae) { try { m_BayesNet.paste(m_clipboard.getText()); updateStatus(); m_jStatusBar.setText(m_BayesNet.lastActionMsg()); } catch (Exception e) { e.printStackTrace(); } } public boolean isEnabled() { return m_clipboard.hasText(); } } // class ActionPasteNode class ActionSelectAll extends MyAction { /** for serialization */ private static final long serialVersionUID = -2038642085935519L; public ActionSelectAll() { super("Select All", "Select All Nodes", "selectall", "ctrl A"); } // c'tor public void actionPerformed(ActionEvent ae) { m_Selection.selectAll(); repaint(); } } // class ActionSelectAll class ActionExport extends MyAction { boolean m_bIsExporting = false; /** for serialization */ private static final long serialVersionUID = -3027642085935519L; public ActionExport() { super("Export", "Export to graphics file", "export", ""); } // c'tor public void actionPerformed(ActionEvent ae) { m_bIsExporting = true; m_GraphPanel.saveComponent(); m_bIsExporting = false; repaint(); } public boolean isExporting() {return m_bIsExporting;} } // class ActionExport class ActionAlignLeft extends MyAction { /** for serialization */ private static final long serialVersionUID = -3138642085935519L; public ActionAlignLeft() { super("Align Left", "Align Left", "alignleft", ""); } // c'tor public void actionPerformed(ActionEvent ae) { m_BayesNet.alignLeft(m_Selection.getSelected()); m_jStatusBar.setText(m_BayesNet.lastActionMsg()); a_undo.setEnabled(true); a_redo.setEnabled(false); repaint(); } } // class ActionAlignLeft class ActionAlignRight extends MyAction { /** for serialization */ private static final long serialVersionUID = -4238642085935519L; public ActionAlignRight() { super("Align Right", "Align Right", "alignright", ""); } // c'tor public void actionPerformed(ActionEvent ae) { m_BayesNet.alignRight(m_Selection.getSelected()); m_jStatusBar.setText(m_BayesNet.lastActionMsg()); a_undo.setEnabled(true); a_redo.setEnabled(false); repaint(); } } // class ActionAlignRight class ActionAlignTop extends MyAction { /** for serialization */ private static final long serialVersionUID = -5338642085935519L; public ActionAlignTop() { super("Align Top", "Align Top", "aligntop", ""); } // c'tor public void actionPerformed(ActionEvent ae) { m_BayesNet.alignTop(m_Selection.getSelected()); m_jStatusBar.setText(m_BayesNet.lastActionMsg()); a_undo.setEnabled(true); a_redo.setEnabled(false); repaint(); } } // class ActionAlignTop class ActionAlignBottom extends MyAction { /** for serialization */ private static final long serialVersionUID = -6438642085935519L; public ActionAlignBottom() { super("Align Bottom", "Align Bottom", "alignbottom", ""); } // c'tor public void actionPerformed(ActionEvent ae) { m_BayesNet.alignBottom(m_Selection.getSelected()); m_jStatusBar.setText(m_BayesNet.lastActionMsg()); a_undo.setEnabled(true); a_redo.setEnabled(false); repaint(); } } // class ActionAlignBottom class ActionCenterHorizontal extends MyAction { /** for serialization */ private static final long serialVersionUID = -7538642085935519L; public ActionCenterHorizontal() { super("Center Horizontal", "Center Horizontal", "centerhorizontal", ""); } // c'tor public void actionPerformed(ActionEvent ae) { m_BayesNet.centerHorizontal(m_Selection.getSelected()); m_jStatusBar.setText(m_BayesNet.lastActionMsg()); a_undo.setEnabled(true); a_redo.setEnabled(false); repaint(); } } // class ActionCenterHorizontal class ActionCenterVertical extends MyAction { /** for serialization */ private static final long serialVersionUID = -8638642085935519L; public ActionCenterVertical() { super("Center Vertical", "Center Vertical", "centervertical", ""); } // c'tor public void actionPerformed(ActionEvent ae) { m_BayesNet.centerVertical(m_Selection.getSelected()); m_jStatusBar.setText(m_BayesNet.lastActionMsg()); a_undo.setEnabled(true); a_redo.setEnabled(false); repaint(); } } // class ActionCenterVertical class ActionSpaceHorizontal extends MyAction { /** for serialization */ private static final long serialVersionUID = -9738642085935519L; public ActionSpaceHorizontal() { super("Space Horizontal", "Space Horizontal", "spacehorizontal", ""); } // c'tor public void actionPerformed(ActionEvent ae) { m_BayesNet.spaceHorizontal(m_Selection.getSelected()); m_jStatusBar.setText(m_BayesNet.lastActionMsg()); a_undo.setEnabled(true); a_redo.setEnabled(false); repaint(); } } // class ActionSpaceHorizontal class ActionSpaceVertical extends MyAction { /** for serialization */ private static final long serialVersionUID = -838642085935519L; public ActionSpaceVertical() { super("Space Vertical", "Space Vertical", "spacevertical", ""); } // c'tor public void actionPerformed(ActionEvent ae) { m_BayesNet.spaceVertical(m_Selection.getSelected()); m_jStatusBar.setText(m_BayesNet.lastActionMsg()); a_undo.setEnabled(true); a_redo.setEnabled(false); repaint(); } } // class ActionSpaceVertical class ActionAddArc extends MyAction { /** for serialization */ private static final long serialVersionUID = -2038913085935519L; public ActionAddArc() { super("Add Arc", "Add Arc", "addarc", ""); } // c'tor public void actionPerformed(ActionEvent ae) { try { String[] options = new String[m_BayesNet.getNrOfNodes()]; for (int i = 0; i < options.length; i++) { options[i] = m_BayesNet.getNodeName(i); } String sChild = (String) JOptionPane.showInputDialog(null, "Select child node", "Nodes", 0, null, options, options[0]); if (sChild == null || sChild.equals("")) { return; } int iChild = m_BayesNet.getNode(sChild); addArcInto(iChild); } catch (Exception e) { e.printStackTrace(); } } } // class ActionAddArc class ActionDeleteArc extends MyAction { /** for serialization */ private static final long serialVersionUID = -2038914085935519L; public ActionDeleteArc() { super("Delete Arc", "Delete Arc", "delarc", ""); } // c'tor public void actionPerformed(ActionEvent ae) { int nEdges = 0; for (int iNode = 0; iNode < m_BayesNet.getNrOfNodes(); iNode++) { nEdges += m_BayesNet.getNrOfParents(iNode); } String[] options = new String[nEdges]; int i = 0; for (int iNode = 0; iNode < m_BayesNet.getNrOfNodes(); iNode++) { for (int iParent = 0; iParent < m_BayesNet.getNrOfParents(iNode); iParent++) { int nParent = m_BayesNet.getParent(iNode, iParent); String sEdge = m_BayesNet.getNodeName(nParent); sEdge += " -> "; sEdge += m_BayesNet.getNodeName(iNode); options[i++] = sEdge; } } deleteArc(options); } } // class ActionDeleteArc class ActionNew extends MyAction { /** for serialization */ private static final long serialVersionUID = -2038911085935515L; public ActionNew() { super("New", "New Network", "new", ""); } // c'tor public void actionPerformed(ActionEvent ae) { m_sFileName = ""; m_BayesNet = new EditableBayesNet(true); updateStatus(); layoutGraph(); a_datagenerator.setEnabled(false); m_BayesNet.clearUndoStack(); m_jStatusBar.setText("New Network"); m_Selection = new Selection(); repaint(); } } // class ActionNew class ActionLoad extends MyAction { /** for serialization */ private static final long serialVersionUID = -2038911085935515L; public ActionLoad() { super("Load", "Load Graph", "open", "ctrl O"); } // c'tor public void actionPerformed(ActionEvent ae) { JFileChooser fc = new JFileChooser(System.getProperty("user.dir")); ExtensionFileFilter ef1 = new ExtensionFileFilter(".arff", "ARFF files"); ExtensionFileFilter ef2 = new ExtensionFileFilter(".xml", "XML BIF files"); fc.addChoosableFileFilter(ef1); fc.addChoosableFileFilter(ef2); fc.setDialogTitle("Load Graph"); int rval = fc.showOpenDialog(GUI.this); if (rval == JFileChooser.APPROVE_OPTION) { String sFileName = fc.getSelectedFile().toString(); if (sFileName.endsWith(ef1.getExtensions()[0])) { initFromArffFile(sFileName); } else { try { readBIFFromFile(sFileName); } catch (Exception e) { e.printStackTrace(); } } m_jStatusBar.setText("Loaded " + sFileName); updateStatus(); } } } // class ActionLoad class ActionViewStatusbar extends MyAction { /** for serialization */ private static final long serialVersionUID = -20389330812354L; public ActionViewStatusbar() { super("View statusbar", "View statusbar", "statusbar", ""); } // c'tor public void actionPerformed(ActionEvent ae) { m_jStatusBar.setVisible(!m_jStatusBar.isVisible()); } // actionPerformed } // class ActionViewStatusbar class ActionViewToolbar extends MyAction { /** for serialization */ private static final long serialVersionUID = -20389110812354L; public ActionViewToolbar() { super("View toolbar", "View toolbar", "toolbar", ""); } // c'tor public void actionPerformed(ActionEvent ae) { m_jTbTools.setVisible(!m_jTbTools.isVisible()); } // actionPerformed } // class ActionViewToolbar class ActionSave extends MyAction { /** for serialization */ private static final long serialVersionUID = -20389110859355156L; public ActionSave() { super("Save", "Save Graph", "save", "ctrl S"); } // c'tor public ActionSave(String sName, String sToolTipText, String sIcon, String sAcceleratorKey) { super(sName, sToolTipText, sIcon, sAcceleratorKey); } // c'tor public void actionPerformed(ActionEvent ae) { if (!m_sFileName.equals("")) { saveFile(m_sFileName); m_BayesNet.isSaved(); m_jStatusBar.setText("Saved as " + m_sFileName); } else { if (saveAs()) { m_BayesNet.isSaved(); m_jStatusBar.setText("Saved as " + m_sFileName); } } } // actionPerformed ExtensionFileFilter ef1 = new ExtensionFileFilter(".xml", "XML BIF files"); boolean saveAs() { JFileChooser fc = new JFileChooser(System.getProperty("user.dir")); fc.addChoosableFileFilter(ef1); fc.setDialogTitle("Save Graph As"); if (!m_sFileName.equals("")) { // can happen on actionQuit fc.setSelectedFile(new File(m_sFileName)); } int rval = fc.showSaveDialog(GUI.this); if (rval == JFileChooser.APPROVE_OPTION) { // System.out.println("Saving to file \""+ // f.getAbsoluteFile().toString()+"\""); String sFileName = fc.getSelectedFile().toString(); if (!sFileName.endsWith(".xml")) sFileName = sFileName.concat(".xml"); saveFile(sFileName); return true; } return false; } // saveAs protected void saveFile(String sFileName) { try { FileWriter outfile = new FileWriter(sFileName); outfile.write(m_BayesNet.toXMLBIF03()); outfile.close(); m_sFileName = sFileName; m_jStatusBar.setText("Saved as " + m_sFileName); } catch(IOException e) { e.printStackTrace(); } } // saveFile } // class ActionSave class ActionSaveAs extends ActionSave { /** for serialization */ private static final long serialVersionUID = -20389110859354L; public ActionSaveAs() { super("Save As", "Save Graph As", "saveas", ""); } // c'tor public void actionPerformed(ActionEvent ae) { saveAs(); } // actionPerformed } // class ActionSaveAs class ActionPrint extends ActionSave { /** for serialization */ private static final long serialVersionUID = -20389001859354L; boolean m_bIsPrinting = false; public ActionPrint() { super("Print", "Print Graph", "print", "ctrl P"); } // c'tor public void actionPerformed(ActionEvent ae) { PrinterJob printJob = PrinterJob.getPrinterJob(); printJob.setPrintable(m_GraphPanel); if (printJob.printDialog()) try { m_bIsPrinting = true; printJob.print(); m_bIsPrinting = false; } catch(PrinterException pe) { m_jStatusBar.setText("Error printing: " + pe); m_bIsPrinting = false; } m_jStatusBar.setText("Print"); } // actionPerformed public boolean isPrinting() {return m_bIsPrinting;} } // class ActionPrint class ActionQuit extends ActionSave { /** for serialization */ private static final long serialVersionUID = -2038911085935515L; public ActionQuit() { super("Exit", "Exit Program", "exit", ""); } // c'tor public void actionPerformed(ActionEvent ae) { if (m_BayesNet.isChanged()) { int result = JOptionPane.showConfirmDialog(null, "Network changed. Do you want to save it?", "Save before closing?", JOptionPane.YES_NO_CANCEL_OPTION); if (result == JOptionPane.CANCEL_OPTION) { return; } if (result == JOptionPane.YES_OPTION) { if (!saveAs()) { return; } } } System.exit(0); } } // class ActionQuit class ActionHelp extends MyAction { /** for serialization */ private static final long serialVersionUID = -20389110859354L; public ActionHelp() { super("Help", "Bayesian Network Workbench Help", "help", ""); } // c'tor public void actionPerformed(ActionEvent ae) { JOptionPane.showMessageDialog(null, "See Weka Homepage\nhttp://www.cs.waikato.ac.nz/ml", "Help Message", JOptionPane.PLAIN_MESSAGE); } } // class ActionHelp class ActionAbout extends MyAction { /** for serialization */ private static final long serialVersionUID = -20389110859353L; public ActionAbout() { super("About", "Help about", "about", ""); } // c'tor public void actionPerformed(ActionEvent ae) { JOptionPane.showMessageDialog(null, "Bayesian Network Workbench\nPart of Weka\n2007", "About Message", JOptionPane.PLAIN_MESSAGE); } } // class ActionAbout class ActionZoomIn extends MyAction { /** for serialization */ private static final long serialVersionUID = -2038911085935515L; public ActionZoomIn() { super("Zoom in", "Zoom in", "zoomin", "+"); } // c'tor public void actionPerformed(ActionEvent ae) { int i = 0, s = (int) (m_fScale * 100); if (s < 300) i = s / 25; else if (s < 700) i = 6 + s / 50; else i = 13 + s / 100; if (s >= 999) { setEnabled(false); return; } else if (s >= 10) { if (i >= 22) { setEnabled(false); } if (s == 10 && !a_zoomout.isEnabled()) { a_zoomout.setEnabled(true); } m_jTfZoom.setText(m_nZoomPercents[i + 1] + "%"); m_fScale = m_nZoomPercents[i + 1] / 100D; } else { if (!a_zoomout.isEnabled()) a_zoomout.setEnabled(true); m_jTfZoom.setText(m_nZoomPercents[0] + "%"); m_fScale = m_nZoomPercents[0] / 100D; } setAppropriateSize(); m_GraphPanel.repaint(); m_GraphPanel.invalidate(); m_jScrollPane.revalidate(); m_jStatusBar.setText("Zooming in"); } } // class ActionZoomIn class ActionZoomOut extends MyAction { /** for serialization */ private static final long serialVersionUID = -203891108593551L; public ActionZoomOut() { super("Zoom out", "Zoom out", "zoomout", "-"); } // c'tor public void actionPerformed(ActionEvent ae) { int i = 0, s = (int) (m_fScale * 100); if (s < 300) i = (int) Math.ceil(s / 25D); else if (s < 700) i = 6 + (int) Math.ceil(s / 50D); else i = 13 + (int) Math.ceil(s / 100D); if (s <= 10) { setEnabled(false); } else if (s < 999) { if (i <= 1) { setEnabled(false); } m_jTfZoom.setText(m_nZoomPercents[i - 1] + "%"); m_fScale = m_nZoomPercents[i - 1] / 100D; } else { if (!a_zoomin.isEnabled()) a_zoomin.setEnabled(true); m_jTfZoom.setText(m_nZoomPercents[22] + "%"); m_fScale = m_nZoomPercents[22] / 100D; } setAppropriateSize(); m_GraphPanel.repaint(); m_GraphPanel.invalidate(); m_jScrollPane.revalidate(); m_jStatusBar.setText("Zooming out"); } } // class ActionZoomOut class ActionLayout extends MyAction { /** for serialization */ private static final long serialVersionUID = -203891108593551L; public ActionLayout() { super("Layout", "Layout Graph", "layout", "ctrl L"); } // c'tor JDialog dlg = null; public void actionPerformed(ActionEvent ae) { if (dlg == null) { dlg = new JDialog(); dlg.setTitle("Graph Layout Options"); final JCheckBox jCbCustomNodeSize = new JCheckBox("Custom Node Size"); final JLabel jLbNodeWidth = new JLabel("Width"); final JLabel jLbNodeHeight = new JLabel("Height"); m_jTfNodeWidth.setHorizontalAlignment(JTextField.CENTER); m_jTfNodeWidth.setText("" + m_nNodeWidth); m_jTfNodeHeight.setHorizontalAlignment(JTextField.CENTER); m_jTfNodeHeight.setText("" + m_nNodeHeight); jLbNodeWidth.setEnabled(false); m_jTfNodeWidth.setEnabled(false); jLbNodeHeight.setEnabled(false); m_jTfNodeHeight.setEnabled(false); jCbCustomNodeSize.addActionListener(new ActionListener() { public void actionPerformed(ActionEvent ae) { if (((JCheckBox) ae.getSource()).isSelected()) { jLbNodeWidth.setEnabled(true); m_jTfNodeWidth.setEnabled(true); jLbNodeHeight.setEnabled(true); m_jTfNodeHeight.setEnabled(true); } else { jLbNodeWidth.setEnabled(false); m_jTfNodeWidth.setEnabled(false); jLbNodeHeight.setEnabled(false); m_jTfNodeHeight.setEnabled(false); setAppropriateSize(); setAppropriateNodeSize(); } } }); JButton jBtLayout; jBtLayout = new JButton("Layout Graph"); jBtLayout.setMnemonic('L'); jBtLayout.addActionListener(new ActionListener() { public void actionPerformed(ActionEvent ae) { int tmpW, tmpH; if (jCbCustomNodeSize.isSelected()) { try { tmpW = Integer.parseInt(m_jTfNodeWidth.getText()); } catch (NumberFormatException ne) { JOptionPane.showMessageDialog(GUI.this.getParent(), "Invalid integer entered for node width.", "Error", JOptionPane.ERROR_MESSAGE); tmpW = m_nNodeWidth; m_jTfNodeWidth.setText("" + m_nNodeWidth); } try { tmpH = Integer.parseInt(m_jTfNodeHeight.getText()); } catch (NumberFormatException ne) { JOptionPane.showMessageDialog(GUI.this.getParent(), "Invalid integer entered for node height.", "Error", JOptionPane.ERROR_MESSAGE); tmpH = m_nNodeHeight; m_jTfNodeWidth.setText("" + m_nNodeHeight); } if (tmpW != m_nNodeWidth || tmpH != m_nNodeHeight) { m_nNodeWidth = tmpW; m_nPaddedNodeWidth = m_nNodeWidth + PADDING; m_nNodeHeight = tmpH; } } // JButton bt = (JButton) ae.getSource(); // bt.setEnabled(false); dlg.setVisible(false); updateStatus(); layoutGraph(); m_jStatusBar.setText("Laying out Bayes net"); } }); JButton jBtCancel; jBtCancel = new JButton("Cancel"); jBtCancel.setMnemonic('C'); jBtCancel.addActionListener(new ActionListener() { public void actionPerformed(ActionEvent ae) { dlg.setVisible(false); } }); GridBagConstraints gbc = new GridBagConstraints(); dlg.setLayout(new GridBagLayout()); //dlg.add(m_le.getControlPanel()); Container c = new Container(); c.setLayout(new GridBagLayout()); gbc.gridwidth = 1; gbc.insets = new Insets(8, 0, 0, 0); gbc.anchor = GridBagConstraints.NORTHWEST; gbc.gridwidth = GridBagConstraints.REMAINDER; c.add(jCbCustomNodeSize, gbc); gbc.gridwidth = GridBagConstraints.RELATIVE; c.add(jLbNodeWidth, gbc); gbc.gridwidth = GridBagConstraints.REMAINDER; c.add(m_jTfNodeWidth, gbc); gbc.gridwidth = GridBagConstraints.RELATIVE; c.add(jLbNodeHeight, gbc); gbc.gridwidth = GridBagConstraints.REMAINDER; c.add(m_jTfNodeHeight, gbc); gbc.fill = GridBagConstraints.HORIZONTAL; dlg.add(c, gbc); dlg.add(jBtLayout); gbc.gridwidth = GridBagConstraints.REMAINDER; dlg.add(jBtCancel); } dlg.setLocation(100, 100); dlg.setVisible(true); dlg.setSize(dlg.getPreferredSize()); dlg.setVisible(false); dlg.setVisible(true); dlg.repaint(); } } // class ActionLayout /** * Constructor
* Sets up the gui and initializes all the other previously uninitialized * variables. */ public GUI() { m_GraphPanel = new GraphPanel(); m_jScrollPane = new JScrollPane(m_GraphPanel); // creating a new layout engine and adding this class as its listener // to receive layoutComplete events m_jTfZoom = new JTextField("100%"); m_jTfZoom.setMinimumSize(m_jTfZoom.getPreferredSize()); m_jTfZoom.setHorizontalAlignment(JTextField.CENTER); m_jTfZoom.setToolTipText("Zoom"); m_jTfZoom.addActionListener(new ActionListener() { public void actionPerformed(ActionEvent ae) { JTextField jt = (JTextField) ae.getSource(); try { int i = -1; i = jt.getText().indexOf('%'); if (i == -1) i = Integer.parseInt(jt.getText()); else i = Integer.parseInt(jt.getText().substring(0, i)); if (i <= 999) m_fScale = i / 100D; jt.setText((int) (m_fScale * 100) + "%"); if (m_fScale > 0.1) { if (!a_zoomout.isEnabled()) a_zoomout.setEnabled(true); } else a_zoomout.setEnabled(false); if (m_fScale < 9.99) { if (!a_zoomin.isEnabled()) a_zoomin.setEnabled(true); } else a_zoomin.setEnabled(false); setAppropriateSize(); // m_GraphPanel.clearBuffer(); m_GraphPanel.repaint(); m_GraphPanel.invalidate(); m_jScrollPane.revalidate(); } catch (NumberFormatException ne) { JOptionPane.showMessageDialog(GUI.this.getParent(), "Invalid integer entered for zoom.", "Error", JOptionPane.ERROR_MESSAGE); jt.setText((m_fScale * 100) + "%"); } } }); GridBagConstraints gbc = new GridBagConstraints(); final JPanel p = new JPanel(new GridBagLayout()); p.setBorder(BorderFactory.createCompoundBorder(BorderFactory.createTitledBorder("ExtraControls"), BorderFactory .createEmptyBorder(4, 4, 4, 4))); p.setPreferredSize(new Dimension(0, 0)); m_jTbTools = new JToolBar(); m_jTbTools.setFloatable(false); m_jTbTools.setLayout(new GridBagLayout()); gbc.anchor = GridBagConstraints.NORTHWEST; gbc.gridwidth = GridBagConstraints.REMAINDER; gbc.insets = new Insets(0, 0, 0, 0); m_jTbTools.add(p, gbc); gbc.gridwidth = 1; m_jTbTools.add(a_new); m_jTbTools.add(a_save); m_jTbTools.add(a_load); m_jTbTools.addSeparator(new Dimension(2, 2)); m_jTbTools.add(a_cutnode); m_jTbTools.add(a_copynode); m_jTbTools.add(a_pastenode); m_jTbTools.addSeparator(new Dimension(2, 2)); m_jTbTools.add(a_undo); m_jTbTools.add(a_redo); m_jTbTools.addSeparator(new Dimension(2, 2)); m_jTbTools.add(a_alignleft); m_jTbTools.add(a_alignright); m_jTbTools.add(a_aligntop); m_jTbTools.add(a_alignbottom); m_jTbTools.add(a_centerhorizontal); m_jTbTools.add(a_centervertical); m_jTbTools.add(a_spacehorizontal); m_jTbTools.add(a_spacevertical); m_jTbTools.addSeparator(new Dimension(2, 2)); m_jTbTools.add(a_zoomin); gbc.fill = GridBagConstraints.VERTICAL; gbc.weighty = 1; JPanel p2 = new JPanel(new BorderLayout()); p2.setPreferredSize(m_jTfZoom.getPreferredSize()); p2.setMinimumSize(m_jTfZoom.getPreferredSize()); p2.add(m_jTfZoom, BorderLayout.CENTER); m_jTbTools.add(p2, gbc); gbc.weighty = 0; gbc.fill = GridBagConstraints.NONE; m_jTbTools.add(a_zoomout); m_jTbTools.addSeparator(new Dimension(2, 2)); // jTbTools.add(jBtExtraControls, gbc); m_jTbTools.add(a_layout); m_jTbTools.addSeparator(new Dimension(4, 2)); gbc.weightx = 1; gbc.fill = GridBagConstraints.BOTH; //jTbTools.add(m_layoutEngine.getProgressBar(), gbc); m_jStatusBar = new JLabel("Status bar"); this.setLayout(new BorderLayout()); this.add(m_jTbTools, BorderLayout.NORTH); this.add(m_jScrollPane, BorderLayout.CENTER); this.add(m_jStatusBar, BorderLayout.SOUTH); updateStatus(); a_datagenerator.setEnabled(false); makeMenuBar(); } /** * Get the menu bar for this application. * * @return the menu bar */ public JMenuBar getMenuBar() { return m_menuBar; } private void makeMenuBar() { m_menuBar = new JMenuBar(); JMenu fileMenu = new JMenu("File"); fileMenu.setMnemonic('F'); m_menuBar.add(fileMenu); fileMenu.add(a_new); fileMenu.add(a_load); fileMenu.add(a_save); fileMenu.add(a_saveas); fileMenu.addSeparator(); fileMenu.add(a_print); fileMenu.add(a_export); fileMenu.addSeparator(); fileMenu.add(a_quit); JMenu editMenu = new JMenu("Edit"); editMenu.setMnemonic('E'); m_menuBar.add(editMenu); editMenu.add(a_undo); editMenu.add(a_redo); editMenu.addSeparator(); editMenu.add(a_selectall); editMenu.add(a_delnode); editMenu.add(a_cutnode); editMenu.add(a_copynode); editMenu.add(a_pastenode); editMenu.addSeparator(); editMenu.add(a_addnode); editMenu.add(a_addarc); editMenu.add(a_delarc); editMenu.addSeparator(); editMenu.add(a_alignleft); editMenu.add(a_alignright); editMenu.add(a_aligntop); editMenu.add(a_alignbottom); editMenu.add(a_centerhorizontal); editMenu.add(a_centervertical); editMenu.add(a_spacehorizontal); editMenu.add(a_spacevertical); JMenu toolMenu = new JMenu("Tools"); toolMenu.setMnemonic('T'); toolMenu.add(a_networkgenerator); toolMenu.add(a_datagenerator); toolMenu.add(a_datasetter); toolMenu.add(a_learn); toolMenu.add(a_learnCPT); toolMenu.addSeparator(); toolMenu.add(a_layout); toolMenu.addSeparator(); final JCheckBoxMenuItem viewMargins = new JCheckBoxMenuItem("Show Margins", false); viewMargins.addActionListener(new ActionListener() { public void actionPerformed(ActionEvent ae) { boolean bPrev = m_bViewMargins; m_bViewMargins = viewMargins.getState(); if (bPrev == false && viewMargins.getState() == true) { updateStatus(); } repaint(); } }); toolMenu.add(viewMargins); final JCheckBoxMenuItem viewCliques = new JCheckBoxMenuItem("Show Cliques", false); viewCliques.addActionListener(new ActionListener() { public void actionPerformed(ActionEvent ae) { boolean bPrev = m_bViewCliques; m_bViewCliques = viewCliques.getState(); if (bPrev == false && viewCliques.getState() == true) { updateStatus(); } repaint(); } }); toolMenu.add(viewCliques); m_menuBar.add(toolMenu); JMenu viewMenu = new JMenu("View"); viewMenu.setMnemonic('V'); m_menuBar.add(viewMenu); viewMenu.add(a_zoomin); viewMenu.add(a_zoomout); viewMenu.addSeparator(); viewMenu.add(a_viewtoolbar); viewMenu.add(a_viewstatusbar); JMenu helpMenu = new JMenu("Help"); helpMenu.setMnemonic('H'); m_menuBar.add(helpMenu); helpMenu.add(a_help); helpMenu.add(a_about); } /** * This method sets the node size that is appropriate considering the * maximum label size that is present. It is used internally when custom * node size checkbox is unchecked. */ protected void setAppropriateNodeSize() { int strWidth; FontMetrics fm = this.getFontMetrics(this.getFont()); int nMaxStringWidth = DEFAULT_NODE_WIDTH; if (nMaxStringWidth == 0) for (int iNode = 0; iNode < m_BayesNet.getNrOfNodes(); iNode++) { strWidth = fm.stringWidth(m_BayesNet.getNodeName(iNode)); if (strWidth > nMaxStringWidth) nMaxStringWidth = strWidth; } m_nNodeWidth = nMaxStringWidth + 4; m_nPaddedNodeWidth = m_nNodeWidth + PADDING; m_jTfNodeWidth.setText("" + m_nNodeWidth); m_nNodeHeight = 2 * fm.getHeight(); m_jTfNodeHeight.setText("" + m_nNodeHeight); } /** * Sets the preferred size for m_GraphPanel GraphPanel to the minimum size that is * neccessary to display the graph. */ public void setAppropriateSize() { int maxX = 0, maxY = 0; m_GraphPanel.setScale(m_fScale, m_fScale); for (int iNode = 0; iNode < m_BayesNet.getNrOfNodes(); iNode++) { int nPosX = m_BayesNet.getPositionX(iNode); int nPosY = m_BayesNet.getPositionY(iNode); if (maxX < nPosX) maxX = nPosX + 100; if (maxY < nPosY) maxY = nPosY; } m_GraphPanel.setPreferredSize(new Dimension((int) ((maxX + m_nPaddedNodeWidth + 2) * m_fScale), (int) ((maxY + m_nNodeHeight + 2) * m_fScale))); m_GraphPanel.revalidate(); } // setAppropriateSize /** * This method is an implementation for LayoutCompleteEventListener class. * It sets the size appropriate for m_GraphPanel GraphPanel and and revalidates it's * container JScrollPane once a LayoutCompleteEvent is received from the * LayoutEngine. Also, it updates positions of the Bayesian network stored * in m_BayesNet. */ public void layoutCompleted(LayoutCompleteEvent le) { LayoutEngine layoutEngine = m_layoutEngine; // (LayoutEngine) le.getSource(); FastVector nPosX = new FastVector(m_BayesNet.getNrOfNodes()); FastVector nPosY = new FastVector(m_BayesNet.getNrOfNodes()); for (int iNode = 0; iNode < layoutEngine.getNodes().size(); iNode++) { GraphNode gNode = (GraphNode) layoutEngine.getNodes().elementAt(iNode); if (gNode.nodeType == GraphNode.NORMAL) { nPosX.addElement(gNode.x); nPosY.addElement(gNode.y); } } m_BayesNet.layoutGraph(nPosX, nPosY); m_jStatusBar.setText("Graph layed out"); a_undo.setEnabled(true); a_redo.setEnabled(false); setAppropriateSize(); m_GraphPanel.invalidate(); m_jScrollPane.revalidate(); m_GraphPanel.repaint(); } // layoutCompleted /** * BIF reader
* Reads a graph description in XMLBIF03 from an file * with name sFileName */ public void readBIFFromFile(String sFileName) throws BIFFormatException, IOException { m_sFileName = sFileName; try { BIFReader bayesNet = new BIFReader(); bayesNet.processFile(sFileName); m_BayesNet = new EditableBayesNet(bayesNet); updateStatus(); a_datagenerator.setEnabled(m_BayesNet.getNrOfNodes() > 0); m_BayesNet.clearUndoStack(); } catch (Exception ex) { ex.printStackTrace(); return; } setAppropriateNodeSize(); setAppropriateSize(); } // readBIFFromFile /* read arff file from file sFileName * and start new Bayesian network with nodes * representing attributes in data set. */ void initFromArffFile(String sFileName) { try { Instances instances = new Instances(new FileReader(sFileName)); m_BayesNet = new EditableBayesNet(instances); m_Instances = instances; a_learn.setEnabled(true); a_learnCPT.setEnabled(true); setAppropriateNodeSize(); setAppropriateSize(); } catch (Exception ex) { ex.printStackTrace(); return; } } // initFromArffFile /** * The panel which contains the actual Bayeian network. */ private class GraphPanel extends PrintablePanel implements Printable { /** for serialization */ private static final long serialVersionUID = -3562813603236753173L; /** node drawing modes */ final static int HIGHLIGHTED = 1; final static int NORMAL = 0; public GraphPanel() { super(); this.addMouseListener(new GraphVisualizerMouseListener()); this.addMouseMotionListener(new GraphVisualizerMouseMotionListener()); this.setToolTipText(""); } // c'tor /* For showing instructions when hovering over a node * (non-Javadoc) * @see javax.swing.JComponent#getToolTipText(java.awt.event.MouseEvent) */ public String getToolTipText(MouseEvent me) { int x, y; Rectangle r; x = y = 0; r = new Rectangle(0, 0, (int) (m_nPaddedNodeWidth * m_fScale), (int) (m_nNodeHeight * m_fScale)); x += me.getX(); y += me.getY(); for (int iNode = 0; iNode < m_BayesNet.getNrOfNodes(); iNode++) { r.x = (int) (m_BayesNet.getPositionX(iNode) * m_fScale); r.y = (int) (m_BayesNet.getPositionY(iNode) * m_fScale); if (r.contains(x, y)) { return m_BayesNet.getNodeName(iNode) + " (right click to manipulate this node)"; } } return null; } // getToolTipText /* Code for showing the graph in the panel. * (non-Javadoc) * @see javax.swing.JComponent#paintComponent(java.awt.Graphics) */ public void paintComponent(Graphics gr) { Graphics2D g = (Graphics2D) gr; RenderingHints rh = new RenderingHints(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON); rh.put(RenderingHints.KEY_RENDERING, RenderingHints.VALUE_RENDER_SPEED); g.setRenderingHints(rh); g.scale(m_fScale, m_fScale); Rectangle r = g.getClipBounds(); g.clearRect(r.x, r.y, r.width, r.height); if (m_bViewCliques) { m_nClique = 1; viewCliques(g, m_marginCalculator.m_root); } for (int iNode = 0; iNode < m_BayesNet.getNrOfNodes(); iNode++) { drawNode(g, iNode, NORMAL); } if (!a_export.isExporting() && !a_print.isPrinting()) { m_Selection.draw(g); } if (m_nSelectedRect != null) { g.drawRect((int)(m_nSelectedRect.x/ m_fScale), (int)(m_nSelectedRect.y/ m_fScale), (int)(m_nSelectedRect.width/ m_fScale), (int)(m_nSelectedRect.height/ m_fScale)); } } // paintComponent /** number of the clique being drawn. Used for selecting the color of the clique */ int m_nClique = 1; /* draws cliques in junction tree. * */ void viewCliques(Graphics g, JunctionTreeNode node) { int [] nodes = node.m_nNodes; g.setColor( new Color(m_nClique % 7 * 256 /7, (m_nClique % 2 * 256 / 2), (m_nClique % 3 * 256 / 3)) ); int dX = m_nPaddedNodeWidth / 2 + m_nClique; int dY = m_nNodeHeight / 2; int nPosX = 0; int nPosY = 0; String sStr = ""; for (int j = 0; j < nodes.length; j++) { nPosX += m_BayesNet.getPositionX(nodes[j]); nPosY += m_BayesNet.getPositionY(nodes[j]); sStr += " " + nodes[j]; for (int k = j+1; k < nodes.length; k++) { g.drawLine( m_BayesNet.getPositionX(nodes[j]) + dX, m_BayesNet.getPositionY(nodes[j]) + dY, m_BayesNet.getPositionX(nodes[k]) + dX, m_BayesNet.getPositionY(nodes[k]) + dY ); } } m_nClique++; nPosX /= nodes.length; nPosY /= nodes.length; g.drawString("Clique " + m_nClique + "("+sStr+")", nPosX, nPosY); for (int iChild = 0; iChild < node.m_children.size(); iChild++) { viewCliques(g, (JunctionTreeNode) node.m_children.elementAt(iChild)); } } // viewCliques /* Draw a node with index iNode on Graphics g at position * Drawing mode can be NORMAL or HIGHLIGHTED. */ protected void drawNode(Graphics g, int iNode, int mode) { int nPosX = m_BayesNet.getPositionX(iNode); int nPosY = m_BayesNet.getPositionY(iNode); g.setColor(this.getBackground().darker().darker()); FontMetrics fm = getFontMetrics(getFont()); if (mode == HIGHLIGHTED) { g.setXORMode(Color.green); // g.setColor(Color.green); } g.fillOval(nPosX + m_nPaddedNodeWidth - m_nNodeWidth - (m_nPaddedNodeWidth - m_nNodeWidth) / 2, nPosY, m_nNodeWidth, m_nNodeHeight); g.setColor(Color.white); if (mode == HIGHLIGHTED) { g.setXORMode(Color.red); } // Draw the node's label if it can fit inside the node's // current width otherwise just display its node nr // if it can fit in node's current width if (fm.stringWidth(m_BayesNet.getNodeName(iNode)) <= m_nNodeWidth) { g.drawString(m_BayesNet.getNodeName(iNode), nPosX + m_nPaddedNodeWidth / 2 - fm.stringWidth(m_BayesNet.getNodeName(iNode)) / 2, nPosY + m_nNodeHeight / 2 + fm.getHeight() / 2 - 2); } else if (fm.stringWidth("" + iNode) <= m_nNodeWidth) { g.drawString("" + iNode, nPosX + m_nPaddedNodeWidth / 2 - fm.stringWidth("" + iNode) / 2, nPosY + m_nNodeHeight / 2 + fm.getHeight() / 2 - 2); } if (mode == HIGHLIGHTED) { g.setXORMode(Color.green); } if (m_bViewMargins) { if (m_BayesNet.getEvidence(iNode) < 0) { g.setColor(new Color(0, 128, 0)); } else { g.setColor(new Color(128, 0, 0)); } double[] P = m_BayesNet.getMargin(iNode); for (int iValue = 0; iValue < P.length; iValue++) { String sP = P[iValue] + ""; if (sP.charAt(0) == '0') { sP = sP.substring(1); } if (sP.length() > 5) { sP = sP.substring(1, 5); } g.fillRect(nPosX + m_nPaddedNodeWidth, nPosY + iValue * 10 + 2, (int) (P[iValue] * 100), 8); g.drawString(m_BayesNet.getNodeValue(iNode, iValue) + " " + sP, nPosX + m_nPaddedNodeWidth + (int) (P[iValue] * 100), nPosY + iValue * 10 + 10); } } if (m_bViewCliques) { return; } g.setColor(Color.black); // Drawing all incoming edges into the node, for (int iParent = 0; iParent < m_BayesNet.getNrOfParents(iNode); iParent++) { int nParent = m_BayesNet.getParent(iNode, iParent); int nPosX1 = nPosX + m_nPaddedNodeWidth / 2; int nPosY1 = nPosY + m_nNodeHeight; int nPosX2 = m_BayesNet.getPositionX(nParent); int nPosY2 = m_BayesNet.getPositionY(nParent); int nPosX2b = nPosX2 + m_nPaddedNodeWidth / 2; int nPosY2b = nPosY2; double phi = Math.atan2((nPosX2b - nPosX1 + 0.0) * m_nNodeHeight, (nPosY2b - nPosY1 + 0.0) * m_nNodeWidth); nPosX1 = (int) (nPosX + m_nPaddedNodeWidth / 2 + Math.sin(phi) * m_nNodeWidth / 2); nPosY1 = (int) (nPosY + m_nNodeHeight / 2 + Math.cos(phi) * m_nNodeHeight / 2); nPosX2b = (int) (nPosX2 + m_nPaddedNodeWidth / 2 - Math.sin(phi) * m_nNodeWidth / 2); nPosY2b = (int) (nPosY2 + m_nNodeHeight / 2 - Math.cos(phi) * m_nNodeHeight / 2); drawArrow(g, nPosX2b, nPosY2b, nPosX1, nPosY1); } if (mode == HIGHLIGHTED) { FastVector children = m_BayesNet.getChildren(iNode); for (int iChild = 0; iChild < children.size(); iChild++) { int nChild = (Integer) children.elementAt(iChild); int nPosX1 = nPosX + m_nPaddedNodeWidth / 2; int nPosY1 = nPosY; int nPosX2 = m_BayesNet.getPositionX(nChild); int nPosY2 = m_BayesNet.getPositionY(nChild); int nPosX2b = nPosX2 + m_nPaddedNodeWidth / 2; int nPosY2b = nPosY2 + m_nNodeHeight; double phi = Math.atan2((nPosX2b - nPosX1 + 0.0) * m_nNodeHeight, (nPosY2b - nPosY1 + 0.0) * m_nNodeWidth); nPosX1 = (int) (nPosX + m_nPaddedNodeWidth / 2 + Math.sin(phi) * m_nNodeWidth / 2); nPosY1 = (int) (nPosY + m_nNodeHeight / 2 + Math.cos(phi) * m_nNodeHeight / 2); nPosX2b = (int) (nPosX2 + m_nPaddedNodeWidth / 2 - Math.sin(phi) * m_nNodeWidth / 2); nPosY2b = (int) (nPosY2 + m_nNodeHeight / 2 - Math.cos(phi) * m_nNodeHeight / 2); drawArrow(g, nPosX1, nPosY1, nPosX2b, nPosY2b); } } } // drawNode /** * This method draws an arrow on a line from (x1,y1) to (x2,y2). The * arrow head is seated on (x2,y2) and is in the direction of the line. * If the arrow is needed to be drawn in the opposite direction then * simply swap the order of (x1, y1) and (x2, y2) when calling this * function. */ protected void drawArrow(Graphics g, int nPosX1, int nPosY1, int nPosX2, int nPosY2) { g.drawLine(nPosX1, nPosY1, nPosX2, nPosY2); if (nPosX1 == nPosX2) { if (nPosY1 < nPosY2) { g.drawLine(nPosX2, nPosY2, nPosX2 + 4, nPosY2 - 8); g.drawLine(nPosX2, nPosY2, nPosX2 - 4, nPosY2 - 8); } else { g.drawLine(nPosX2, nPosY2, nPosX2 + 4, nPosY2 + 8); g.drawLine(nPosX2, nPosY2, nPosX2 - 4, nPosY2 + 8); } } else { // theta=line's angle from base, beta=angle of arrow's side from // line double hyp = 0, base = 0, perp = 0, theta, beta; int nPosX3 = 0, nPosY3 = 0; if (nPosX2 < nPosX1) { base = nPosX1 - nPosX2; hyp = Math.sqrt((nPosX2 - nPosX1) * (nPosX2 - nPosX1) + (nPosY2 - nPosY1) * (nPosY2 - nPosY1)); theta = Math.acos(base / hyp); } else { // x1>x2 as we already checked x1==x2 before base = nPosX1 - nPosX2; hyp = Math.sqrt((nPosX2 - nPosX1) * (nPosX2 - nPosX1) + (nPosY2 - nPosY1) * (nPosY2 - nPosY1)); theta = Math.acos(base / hyp); } beta = 30 * Math.PI / 180; hyp = 8; base = Math.cos(theta - beta) * hyp; perp = Math.sin(theta - beta) * hyp; nPosX3 = (int) (nPosX2 + base); if (nPosY1 < nPosY2) nPosY3 = (int) (nPosY2 - perp); else nPosY3 = (int) (nPosY2 + perp); g.drawLine(nPosX2, nPosY2, nPosX3, nPosY3); base = Math.cos(theta + beta) * hyp; perp = Math.sin(theta + beta) * hyp; nPosX3 = (int) (nPosX2 + base); if (nPosY1 < nPosY2) nPosY3 = (int) (nPosY2 - perp); else nPosY3 = (int) (nPosY2 + perp); g.drawLine(nPosX2, nPosY2, nPosX3, nPosY3); } } // drawArrow /** * This method highlights a given node and all its incoming and outgoing arcs */ public void highLight(int iNode) { Graphics2D g = (Graphics2D) this.getGraphics(); RenderingHints rh = new RenderingHints(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON); rh.put(RenderingHints.KEY_RENDERING, RenderingHints.VALUE_RENDER_SPEED); g.setRenderingHints(rh); g.setPaintMode(); g.scale(m_fScale, m_fScale); drawNode(g, iNode, HIGHLIGHTED); } // highlight /** implementation of Printable, used for printing * @see Printable */ public int print(Graphics g, PageFormat pageFormat, int pageIndex) { if (pageIndex > 0) { return(NO_SUCH_PAGE); } else { Graphics2D g2d = (Graphics2D)g; g2d.translate(pageFormat.getImageableX(), pageFormat.getImageableY()); double fHeight = pageFormat.getImageableHeight(); double fWidth = pageFormat.getImageableWidth(); int xMax = 1; int yMax = 1; for (int iNode = 0; iNode < m_BayesNet.getNrOfNodes(); iNode++) { if (xMax < m_BayesNet.getPositionX(iNode)) { xMax = m_BayesNet.getPositionX(iNode); } if (yMax < m_BayesNet.getPositionY(iNode)) { yMax = m_BayesNet.getPositionY(iNode); } } double fCurrentScale = m_fScale; xMax += m_nPaddedNodeWidth + 100; if (fWidth/xMax < fHeight/yMax) { m_fScale = fWidth/xMax; } else { m_fScale = fHeight/yMax; } // Turn off double buffering paint(g2d); m_fScale = fCurrentScale; // Turn double buffering back on return(PAGE_EXISTS); } } // print } // class GraphPanel /** * Table Model for the Table for editing CPTs */ private class GraphVisualizerTableModel extends AbstractTableModel { /** for serialization */ private static final long serialVersionUID = -4789813491347366596L; /** labels for the columns */ final String [] m_sColumnNames; /** probability table data **/ final double [][] m_fProbs; /** nr of node for currently editted CPT */ int m_iNode; public GraphVisualizerTableModel(int iNode) { m_iNode = iNode; double [][] probs = m_BayesNet.getDistribution(iNode); m_fProbs = new double[probs.length][probs[0].length]; for (int i = 0; i < probs.length; i++) { for (int j = 0; j < probs[0].length; j++) { m_fProbs[i][j] = probs[i][j]; } } m_sColumnNames = m_BayesNet.getValues(iNode); } // c'tor /** method that generates random CPTs */ public void randomize() { int nProbs = m_fProbs[0].length; Random random = new Random(); for (int i = 0; i < m_fProbs.length; i++) { // get random nrs for (int j = 0; j < nProbs-1; j++) { m_fProbs[i][j] = random.nextDouble(); } // sort for (int j = 0; j < nProbs-1; j++) { for (int k = j+1; k < nProbs-1; k++) { if (m_fProbs[i][j] > m_fProbs[i][k]) { double h = m_fProbs[i][j]; m_fProbs[i][j] = m_fProbs[i][k]; m_fProbs[i][k] = h; } } } double sum = m_fProbs[i][0]; for (int j = 1; j < nProbs-1; j++) { m_fProbs[i][j] = m_fProbs[i][j] - sum; sum += m_fProbs[i][j]; } m_fProbs[i][nProbs - 1] = 1.0 - sum; } } // randomize public void setData() {} /** return nr of colums */ public int getColumnCount() { return m_sColumnNames.length; } /** return nr of rows */ public int getRowCount() { return m_fProbs.length; } /** return name of specified colum * @param iCol index of the column */ public String getColumnName(int iCol) { return m_sColumnNames[iCol]; } /** return data point * @param iRow index of row in table * @param iCol index of column in table */ public Object getValueAt(int iRow, int iCol) { return new Double(m_fProbs[iRow][iCol]); } /** Set data point, assigns value to CPT entry * specified by row and column. The remainder of the * CPT is normalized so that the values add up to 1. * IF a value below zero of over 1 is given, no changes * take place. * @param oProb data point * @param iRow index of row in table * @param iCol index of column in table */ public void setValueAt(Object oProb, int iRow, int iCol) { Double fProb = (Double) oProb; if (fProb < 0 || fProb > 1) { return; } m_fProbs[iRow][iCol] = (double) fProb; double sum = 0; for (int i = 0; i < m_fProbs[iRow].length; i++) { sum += m_fProbs[iRow][i]; } if (sum > 1) { // handle overflow int i = m_fProbs[iRow].length - 1; while (sum > 1) { if (i != iCol) { if (m_fProbs[iRow][i] > sum - 1) { m_fProbs[iRow][i] -= sum - 1; sum = 1; } else { sum -= m_fProbs[iRow][i]; m_fProbs[iRow][i] = 0; } } i--; } } else { // handle underflow int i = m_fProbs[iRow].length - 1; while (sum < 1) { if (i != iCol) { m_fProbs[iRow][i] += 1 - sum; sum = 1; } i--; } } validate(); } // setData /* * JTable uses this method to determine the default renderer/ editor for * each cell. */ public Class getColumnClass(int c) { return getValueAt(0, c).getClass(); } /* * Implemented this to make sure the table is uneditable. */ public boolean isCellEditable(int row, int col) { return true; } } // class GraphVisualizerTableModel /** * Listener class for processing mouseClicked */ private class GraphVisualizerMouseListener extends MouseAdapter { /** A left mouseclick on a node adds node to selection (depending * on shift and ctrl keys). * A right mouseclick on a node pops up menu with actions to be * performed on the node. * A right mouseclick outside another node pops up menu. */ public void mouseClicked(MouseEvent me) { int x, y; Rectangle r = new Rectangle(0, 0, (int) (m_nPaddedNodeWidth * m_fScale), (int) (m_nNodeHeight * m_fScale)); x = me.getX(); y = me.getY(); for (int iNode = 0; iNode < m_BayesNet.getNrOfNodes(); iNode++) { r.x = (int) (m_BayesNet.getPositionX(iNode) * m_fScale); r.y = (int) (m_BayesNet.getPositionY(iNode) * m_fScale); if (r.contains(x, y)) { m_nCurrentNode = iNode; if (me.getButton() == MouseEvent.BUTTON3) { handleRightNodeClick(me); } if (me.getButton() == MouseEvent.BUTTON1) { if ((me.getModifiersEx() & MouseEvent.CTRL_DOWN_MASK) != 0) { m_Selection.toggleSelection(m_nCurrentNode); } else if ((me.getModifiersEx() & MouseEvent.SHIFT_DOWN_MASK) != 0) { m_Selection.addToSelection(m_nCurrentNode); } else { m_Selection.clear(); m_Selection.addToSelection(m_nCurrentNode); } repaint(); } return; } } if (me.getButton() == MouseEvent.BUTTON3) { handleRightClick(me, (int)(x/m_fScale), (int)(y/m_fScale)); } } // mouseClicked /* update selection * (non-Javadoc) * @see java.awt.event.MouseListener#mouseReleased(java.awt.event.MouseEvent) */ public void mouseReleased(MouseEvent me) { if (m_nSelectedRect != null) { if ((me.getModifiersEx() & MouseEvent.CTRL_DOWN_MASK) != 0) { m_Selection.toggleSelection(m_nSelectedRect); } else if ((me.getModifiersEx() & MouseEvent.SHIFT_DOWN_MASK) != 0) { m_Selection.addToSelection(m_nSelectedRect); } else { m_Selection.clear(); m_Selection.addToSelection(m_nSelectedRect); } m_nSelectedRect = null; repaint(); } } // mouseReleased /** position clicked on */ int m_nPosX = 0, m_nPosY = 0; /* pop up menu with actions that apply in general or to selection (if any exists) */ void handleRightClick(MouseEvent me, int nPosX, int nPosY) { ActionListener act = new ActionListener() { public void actionPerformed(ActionEvent ae) { if (ae.getActionCommand().equals("Add node")) { a_addnode.addNode(m_nPosX, m_nPosY); return; } repaint(); } }; JPopupMenu popupMenu = new JPopupMenu("Choose a value"); JMenuItem addNodeItem = new JMenuItem("Add node"); addNodeItem.addActionListener(act); popupMenu.add(addNodeItem); FastVector selected = m_Selection.getSelected(); JMenu addArcMenu = new JMenu("Add parent"); popupMenu.add(addArcMenu); if (selected.size() == 0) { addArcMenu.setEnabled(false); } else { int nNodes = m_BayesNet.getNrOfNodes(); boolean[] isNotAllowedAsParent = new boolean[nNodes]; // prevent it being a parent of itself for (int iNode = 0; iNode < selected.size(); iNode++) { isNotAllowedAsParent[(Integer) selected.elementAt(iNode)] = true; } // prevent a descendant being a parent, since it introduces cycles for (int i = 0; i < nNodes; i++) { for (int iNode = 0; iNode < nNodes; iNode++) { for (int iParent = 0; iParent < m_BayesNet.getNrOfParents(iNode); iParent++) { if (isNotAllowedAsParent[m_BayesNet.getParent(iNode, iParent)]) { isNotAllowedAsParent[iNode] = true; } } } } // prevent nodes that are already a parent for (int iNode = 0; iNode < selected.size(); iNode++) { int nNode = (Integer) selected.elementAt(iNode); for (int iParent = 0; iParent < m_BayesNet.getNrOfParents(nNode); iParent++) { isNotAllowedAsParent[m_BayesNet.getParent(nNode, iParent)] = true; } } ActionListener addParentAction = new ActionListener() { public void actionPerformed(ActionEvent ae) { try { m_BayesNet.addArc(ae.getActionCommand(), m_Selection.getSelected()); m_jStatusBar.setText(m_BayesNet.lastActionMsg()); updateStatus(); } catch (Exception e) { e.printStackTrace(); } } }; // count nr of remaining candidates int nCandidates = 0; for (int i = 0; i < nNodes; i++) { if (!isNotAllowedAsParent[i]) { JMenuItem item = new JMenuItem(m_BayesNet.getNodeName(i)); item.addActionListener(addParentAction); addArcMenu.add(item); nCandidates++; } } if (nCandidates == 0) { addArcMenu.setEnabled(false); } } m_nPosX = nPosX; m_nPosY = nPosY; popupMenu.setLocation(me.getX(), me.getY()); popupMenu.show(m_GraphPanel, me.getX(), me.getY()); } // handleRightClick /* pop up menu with actions that apply to node that was clicked on */ void handleRightNodeClick(MouseEvent me) { m_Selection.clear(); repaint(); ActionListener renameValueAction = new ActionListener() { public void actionPerformed(ActionEvent ae) { renameValue(m_nCurrentNode, ae.getActionCommand()); } }; ActionListener delValueAction = new ActionListener() { public void actionPerformed(ActionEvent ae) { delValue(m_nCurrentNode, ae.getActionCommand()); } }; ActionListener addParentAction = new ActionListener() { public void actionPerformed(ActionEvent ae) { try { m_BayesNet.addArc(ae.getActionCommand(), m_BayesNet.getNodeName(m_nCurrentNode)); m_jStatusBar.setText(m_BayesNet.lastActionMsg()); updateStatus(); } catch (Exception e) { e.printStackTrace(); } } }; ActionListener delParentAction = new ActionListener() { public void actionPerformed(ActionEvent ae) { deleteArc(m_nCurrentNode, ae.getActionCommand()); } }; ActionListener delChildAction = new ActionListener() { public void actionPerformed(ActionEvent ae) { deleteArc(ae.getActionCommand(), m_nCurrentNode); } }; ActionListener setAvidenceAction = new ActionListener() { public void actionPerformed(ActionEvent ae) { try { String [] outcomes = m_BayesNet.getValues(m_nCurrentNode); int iValue = 0; while (iValue < outcomes.length && !outcomes[iValue].equals(ae.getActionCommand())) { iValue++; } if (iValue == outcomes.length) { iValue = -1; } if (iValue < outcomes.length) { m_jStatusBar.setText("Set evidence for " + m_BayesNet.getNodeName(m_nCurrentNode)); if (m_BayesNet.getEvidence(m_nCurrentNode) < 0 && iValue >= 0) { m_BayesNet.setEvidence(m_nCurrentNode, iValue); m_marginCalculatorWithEvidence.setEvidence(m_nCurrentNode, iValue); } else { m_BayesNet.setEvidence(m_nCurrentNode, iValue); SerializedObject so = new SerializedObject(m_marginCalculator); m_marginCalculatorWithEvidence = (MarginCalculator) so.getObject(); for (int iNode = 0; iNode < m_BayesNet.getNrOfNodes(); iNode++) { if (m_BayesNet.getEvidence(iNode) >= 0) { m_marginCalculatorWithEvidence.setEvidence(iNode, m_BayesNet.getEvidence(iNode)); } } } for (int iNode = 0; iNode < m_BayesNet.getNrOfNodes(); iNode++) { m_BayesNet.setMargin(iNode, m_marginCalculatorWithEvidence.getMargin(iNode)); } } } catch (Exception e) { e.printStackTrace(); } repaint(); } }; ActionListener act = new ActionListener() { public void actionPerformed(ActionEvent ae) { if (ae.getActionCommand().equals("Rename")) { renameNode(m_nCurrentNode); return; } if (ae.getActionCommand().equals("Add parent")) { addArcInto(m_nCurrentNode); return; } if (ae.getActionCommand().equals("Add value")) { addValue(); return; } if (ae.getActionCommand().equals("Delete node")) { deleteNode(m_nCurrentNode); return; } if (ae.getActionCommand().equals("Edit CPT")) { editCPT(m_nCurrentNode); return; } repaint(); } }; try { JPopupMenu popupMenu = new JPopupMenu("Choose a value"); JMenu setEvidenceMenu = new JMenu("Set evidence"); String [] outcomes = m_BayesNet.getValues(m_nCurrentNode); for (int iValue = 0; iValue < outcomes.length; iValue++) { JMenuItem item = new JMenuItem(outcomes[iValue]); item.addActionListener(setAvidenceAction); setEvidenceMenu.add(item); } setEvidenceMenu.addSeparator(); JMenuItem item = new JMenuItem("Clear"); item.addActionListener(setAvidenceAction); setEvidenceMenu.add(item); popupMenu.add(setEvidenceMenu); setEvidenceMenu.setEnabled(m_bViewMargins); popupMenu.addSeparator(); JMenuItem renameItem = new JMenuItem("Rename"); renameItem.addActionListener(act); popupMenu.add(renameItem); JMenuItem delNodeItem = new JMenuItem("Delete node"); delNodeItem.addActionListener(act); popupMenu.add(delNodeItem); JMenuItem editCPTItem = new JMenuItem("Edit CPT"); editCPTItem.addActionListener(act); popupMenu.add(editCPTItem); popupMenu.addSeparator(); JMenu addArcMenu = new JMenu("Add parent"); popupMenu.add(addArcMenu); int nNodes = m_BayesNet.getNrOfNodes(); boolean[] isNotAllowedAsParent = new boolean[nNodes]; // prevent it being a parent of itself isNotAllowedAsParent[m_nCurrentNode] = true; // prevent a descendant being a parent, since it introduces cycles for (int i = 0; i < nNodes; i++) { for (int iNode = 0; iNode < nNodes; iNode++) { for (int iParent = 0; iParent < m_BayesNet.getNrOfParents(iNode); iParent++) { if (isNotAllowedAsParent[m_BayesNet.getParent(iNode, iParent)]) { isNotAllowedAsParent[iNode] = true; } } } } // prevent nodes that are already a parent for (int iParent = 0; iParent < m_BayesNet.getNrOfParents(m_nCurrentNode); iParent++) { isNotAllowedAsParent[m_BayesNet.getParent(m_nCurrentNode, iParent)] = true; } // count nr of remaining candidates int nCandidates = 0; for (int i = 0; i < nNodes; i++) { if (!isNotAllowedAsParent[i]) { item = new JMenuItem(m_BayesNet.getNodeName(i)); item.addActionListener(addParentAction); addArcMenu.add(item); nCandidates++; } } if (nCandidates == 0) { addArcMenu.setEnabled(false); } JMenu delArcMenu = new JMenu("Delete parent"); popupMenu.add(delArcMenu); if (m_BayesNet.getNrOfParents(m_nCurrentNode) == 0) { delArcMenu.setEnabled(false); } for (int iParent = 0; iParent < m_BayesNet.getNrOfParents(m_nCurrentNode); iParent++) { item = new JMenuItem(m_BayesNet.getNodeName(m_BayesNet.getParent(m_nCurrentNode, iParent))); item.addActionListener(delParentAction); delArcMenu.add(item); } JMenu delChildMenu = new JMenu("Delete child"); popupMenu.add(delChildMenu); FastVector nChildren = m_BayesNet.getChildren(m_nCurrentNode); if (nChildren.size() == 0) { delChildMenu.setEnabled(false); } for (int iChild = 0; iChild < nChildren.size(); iChild++) { item = new JMenuItem(m_BayesNet.getNodeName((Integer) nChildren.elementAt(iChild))); item.addActionListener(delChildAction); delChildMenu.add(item); } popupMenu.addSeparator(); JMenuItem addValueItem = new JMenuItem("Add value"); addValueItem.addActionListener(act); popupMenu.add(addValueItem); JMenu renameValue = new JMenu("Rename value"); popupMenu.add(renameValue); for (int iValue = 0; iValue < outcomes.length; iValue++) { item = new JMenuItem(outcomes[iValue]); item.addActionListener(renameValueAction); renameValue.add(item); } JMenu delValue = new JMenu("Delete value"); popupMenu.add(delValue); if (m_BayesNet.getCardinality(m_nCurrentNode) <= 2) { delValue.setEnabled(false); } for (int iValue = 0; iValue < outcomes.length; iValue++) { JMenuItem delValueItem = new JMenuItem(outcomes[iValue]); delValueItem.addActionListener(delValueAction); delValue.add(delValueItem); } popupMenu.setLocation(me.getX(), me.getY()); popupMenu.show(m_GraphPanel, me.getX(), me.getY()); } catch (Exception e) { e.printStackTrace(); } } // handleRightNodeClick } // class GraphVisualizerMouseListener /** * private class for handling mouseMoved events to highlight nodes if the * the mouse is moved on one, move it around or move selection around */ private class GraphVisualizerMouseMotionListener extends MouseMotionAdapter { /* last node moved over. Used for turning highlight on and off */ int m_nLastNode = -1; /* current mouse position clicked */ int m_nPosX, m_nPosY; /* identify the node under the mouse * @returns node index of node under mouse, or -1 if there is no such node */ int getGraphNode(MouseEvent me) { m_nPosX = m_nPosY = 0; Rectangle r = new Rectangle(0, 0, (int) (m_nPaddedNodeWidth * m_fScale), (int) (m_nNodeHeight * m_fScale)); m_nPosX += me.getX(); m_nPosY += me.getY(); for (int iNode = 0; iNode < m_BayesNet.getNrOfNodes(); iNode++) { r.x = (int) (m_BayesNet.getPositionX(iNode) * m_fScale); r.y = (int) (m_BayesNet.getPositionY(iNode) * m_fScale); if (r.contains(m_nPosX, m_nPosY)) { return iNode; } } return -1; } // getGraphNode /* handle mouse dragging event * (non-Javadoc) * @see java.awt.event.MouseMotionListener#mouseDragged(java.awt.event.MouseEvent) */ public void mouseDragged(MouseEvent me) { if (m_nSelectedRect != null) { m_nSelectedRect.width = me.getPoint().x - m_nSelectedRect.x; m_nSelectedRect.height = me.getPoint().y - m_nSelectedRect.y; repaint(); return; } int iNode = getGraphNode(me); if (iNode >= 0) { if (m_Selection.getSelected().size() > 0) { if (m_Selection.getSelected().contains(iNode)) { m_BayesNet.setPosition(iNode, (int) ((m_nPosX / m_fScale - m_nPaddedNodeWidth / 2)), (int) ((m_nPosY / m_fScale - m_nNodeHeight / 2)), m_Selection.getSelected()); } else { m_Selection.clear(); m_BayesNet.setPosition(iNode, (int) ((m_nPosX / m_fScale - m_nPaddedNodeWidth / 2)), (int) ((m_nPosY / m_fScale - m_nNodeHeight / 2))); } repaint(); } else { m_BayesNet.setPosition(iNode, (int) ((m_nPosX / m_fScale - m_nPaddedNodeWidth / 2)), (int) ((m_nPosY / m_fScale - m_nNodeHeight / 2))); } m_jStatusBar.setText(m_BayesNet.lastActionMsg()); a_undo.setEnabled(true); a_redo.setEnabled(false); m_GraphPanel.highLight(iNode); } if (iNode < 0) { if (m_nLastNode >= 0) { m_GraphPanel.repaint(); m_nLastNode = -1; } else { m_nSelectedRect = new Rectangle(me.getPoint().x, me.getPoint().y, 1, 1); m_GraphPanel.repaint(); } } } // mouseDragged /* handles mouse move event * (non-Javadoc) * @see java.awt.event.MouseMotionListener#mouseMoved(java.awt.event.MouseEvent) */ public void mouseMoved(MouseEvent me) { int iNode = getGraphNode(me); if (iNode >= 0) { if (iNode != m_nLastNode) { m_GraphPanel.highLight(iNode); if (m_nLastNode >= 0) { m_GraphPanel.highLight(m_nLastNode); } m_nLastNode = iNode; } } if (iNode < 0 && m_nLastNode >= 0) { m_GraphPanel.repaint(); m_nLastNode = -1; } } // mouseMoved } // class GraphVisualizerMouseMotionListener /* apply graph layout algorithm to Bayesian network */ void layoutGraph() { if (m_BayesNet.getNrOfNodes() == 0) { return; } try { FastVector m_nodes = new FastVector(); FastVector m_edges = new FastVector(); BIFParser bp = new BIFParser(m_BayesNet.toXMLBIF03(), m_nodes, m_edges); bp.parse(); updateStatus(); m_layoutEngine = new HierarchicalBCEngine(m_nodes, m_edges, m_nPaddedNodeWidth, m_nNodeHeight); m_layoutEngine.addLayoutCompleteEventListener(this); m_layoutEngine.layoutGraph(); } catch (Exception e) { e.printStackTrace(); } } // layoutGraph /* Update status of various items that need regular updating * such as enabled status of some menu items, marginal distributions * if shown, repainting of graph. */ void updateStatus() { a_undo.setEnabled(m_BayesNet.canUndo()); a_redo.setEnabled(m_BayesNet.canRedo()); a_datagenerator.setEnabled(m_BayesNet.getNrOfNodes() > 0); if (!m_bViewMargins && !m_bViewCliques) { repaint(); return; } try { m_marginCalculator = new MarginCalculator(); m_marginCalculator.calcMargins(m_BayesNet); SerializedObject so = new SerializedObject(m_marginCalculator); m_marginCalculatorWithEvidence = (MarginCalculator) so.getObject(); for (int iNode = 0; iNode < m_BayesNet.getNrOfNodes(); iNode++) { if (m_BayesNet.getEvidence(iNode) >= 0) { m_marginCalculatorWithEvidence.setEvidence(iNode, m_BayesNet.getEvidence(iNode)); } } for (int iNode = 0; iNode < m_BayesNet.getNrOfNodes(); iNode++) { m_BayesNet.setMargin(iNode, m_marginCalculatorWithEvidence.getMargin(iNode)); } } catch (Exception e) { e.printStackTrace(); } repaint(); } // updateStatus /* add arc with node iChild as child. * This pops up a selection list with potential parents for the child. * All decendants and current parents are excluded from the list as is * the child node itself. * @param iChild index of the node for which to add an arc */ void addArcInto(int iChild) { String sChild = m_BayesNet.getNodeName(iChild); try { int nNodes = m_BayesNet.getNrOfNodes(); boolean[] isNotAllowedAsParent = new boolean[nNodes]; // prevent it being a parent of itself isNotAllowedAsParent[iChild] = true; // prevent a descendant being a parent, since it introduces cycles for (int i = 0; i < nNodes; i++) { for (int iNode = 0; iNode < nNodes; iNode++) { for (int iParent = 0; iParent < m_BayesNet.getNrOfParents(iNode); iParent++) { if (isNotAllowedAsParent[m_BayesNet.getParent(iNode, iParent)]) { isNotAllowedAsParent[iNode] = true; } } } } // prevent nodes that are already a parent for (int iParent = 0; iParent < m_BayesNet.getNrOfParents(iChild); iParent++) { isNotAllowedAsParent[m_BayesNet.getParent(iChild, iParent)] = true; } // count nr of remaining candidates int nCandidates = 0; for (int i = 0; i < nNodes; i++) { if (!isNotAllowedAsParent[i]) { nCandidates++; } } if (nCandidates == 0) { JOptionPane.showMessageDialog(null, "No potential parents available for this node (" + sChild + "). Choose another node as child node."); return; } String[] options = new String[nCandidates]; int k = 0; for (int i = 0; i < nNodes; i++) { if (!isNotAllowedAsParent[i]) { options[k++] = m_BayesNet.getNodeName(i); } } String sParent = (String) JOptionPane.showInputDialog(null, "Select parent node for " + sChild, "Nodes", 0, null, options, options[0]); if (sParent == null || sParent.equals("")) { return; } // update all data structures m_BayesNet.addArc(sParent, sChild); m_jStatusBar.setText(m_BayesNet.lastActionMsg()); updateStatus(); } catch (Exception e) { e.printStackTrace(); } } // addArcInto /* deletes arc from node with name sParent into child with index iChild * */ void deleteArc(int iChild, String sParent) { try { m_BayesNet.deleteArc(m_BayesNet.getNode(sParent), iChild); m_jStatusBar.setText(m_BayesNet.lastActionMsg()); } catch (Exception e) { e.printStackTrace(); } updateStatus(); } // deleteArc /* deletes arc from node with index iParent into child with name sChild * */ void deleteArc(String sChild, int iParent) { try { m_BayesNet.deleteArc(iParent, m_BayesNet.getNode(sChild)); m_jStatusBar.setText(m_BayesNet.lastActionMsg()); } catch (Exception e) { e.printStackTrace(); } updateStatus(); } // deleteArc /* deletes arc. Pops up list of arcs listed in 'options' as * " -> ". */ void deleteArc(String[] options) { String sResult = (String) JOptionPane.showInputDialog(null, "Select arc to delete", "Arcs", 0, null, options, options[0]); if (sResult != null && !sResult.equals("")) { int nPos = sResult.indexOf(" -> "); String sParent = sResult.substring(0, nPos); String sChild = sResult.substring(nPos + 4); try { m_BayesNet.deleteArc(sParent, sChild); m_jStatusBar.setText(m_BayesNet.lastActionMsg()); } catch (Exception e) { e.printStackTrace(); } updateStatus(); } } // deleteArc /* Rename node with index nTargetNode. * Pops up window that allwos for entering a new name. */ void renameNode(int nTargetNode) { String sName = (String) JOptionPane.showInputDialog(null, m_BayesNet.getNodeName(nTargetNode), "New name for node", JOptionPane.OK_CANCEL_OPTION); if (sName == null || sName.equals("")) { return; } try { while (m_BayesNet.getNode2(sName) >= 0) { sName = (String) JOptionPane.showInputDialog(null, "Cannot rename to " + sName + ".\nNode with that name already exists."); if (sName == null || sName.equals("")) { return; } } m_BayesNet.setNodeName(nTargetNode, sName); m_jStatusBar.setText(m_BayesNet.lastActionMsg()); } catch (Exception e) { e.printStackTrace(); } repaint(); } // renameNode /* Rename value with name sValeu of a node with index nTargetNode. * Pops up window that allows entering a new name. */ void renameValue(int nTargetNode, String sValue) { String sNewValue = (String) JOptionPane.showInputDialog(null, "New name for value " + sValue, "Node " + m_BayesNet.getNodeName(nTargetNode), JOptionPane.OK_CANCEL_OPTION); if (sNewValue == null || sNewValue.equals("")) { return; } m_BayesNet.renameNodeValue(nTargetNode, sValue, sNewValue); m_jStatusBar.setText(m_BayesNet.lastActionMsg()); a_undo.setEnabled(true); a_redo.setEnabled(false); repaint(); } // renameValue /* delete a single node with index iNode */ void deleteNode(int iNode) { try { m_BayesNet.deleteNode(iNode); m_jStatusBar.setText(m_BayesNet.lastActionMsg()); } catch (Exception e) { e.printStackTrace(); } updateStatus(); } // deleteNode /* Add a value to currently selected node. * Shows window that allows to enter the name of the value. */ void addValue() { //GraphNode n = (GraphNode) m_nodes.elementAt(m_nCurrentNode); String sValue = "Value" + (m_BayesNet.getCardinality(m_nCurrentNode) + 1); String sNewValue = (String) JOptionPane.showInputDialog(null, "New value " + sValue, "Node " + m_BayesNet.getNodeName(m_nCurrentNode), JOptionPane.OK_CANCEL_OPTION); if (sNewValue == null || sNewValue.equals("")) { return; } try { m_BayesNet.addNodeValue(m_nCurrentNode, sNewValue); m_jStatusBar.setText(m_BayesNet.lastActionMsg()); //n.outcomes = m_BayesNet.getValues(m_nCurrentNode); //for (int iNode = 0; iNode < m_BayesNet.getNrOfNodes(); iNode++) { // n = (GraphNode) m_nodes.elementAt(iNode); // n.probs = m_BayesNet.getDistribution(iNode); //} } catch (Exception e) { e.printStackTrace(); } updateStatus(); } // addValue /* remove value with name sValue from the node with index nTargetNode */ void delValue(int nTargetNode, String sValue) { try { m_BayesNet.delNodeValue(nTargetNode, sValue); m_jStatusBar.setText(m_BayesNet.lastActionMsg()); } catch (Exception e) { e.printStackTrace(); } updateStatus(); } // delValue /* Edits CPT of node with index nTargetNode. * Pops up table with probability table that the user can change or just view. */ void editCPT(int nTargetNode) { m_nCurrentNode = nTargetNode; final GraphVisualizerTableModel tm = new GraphVisualizerTableModel(nTargetNode); JTable jTblProbs = new JTable(tm); JScrollPane js = new JScrollPane(jTblProbs); int nParents = m_BayesNet.getNrOfParents(nTargetNode); if (nParents > 0) { GridBagConstraints gbc = new GridBagConstraints(); JPanel jPlRowHeader = new JPanel(new GridBagLayout()); // indices of the parent nodes in the Vector int[] idx = new int[nParents]; // max length of values of each parent int[] lengths = new int[nParents]; // Adding labels for rows gbc.anchor = GridBagConstraints.NORTHWEST; gbc.fill = GridBagConstraints.HORIZONTAL; gbc.insets = new Insets(0, 1, 0, 0); int addNum = 0, temp = 0; boolean dark = false; while (true) { gbc.gridwidth = 1; for (int k = 0; k < nParents; k++) { int iParent2 = m_BayesNet.getParent(nTargetNode, k); JLabel lb = new JLabel(m_BayesNet.getValueName(iParent2,idx[k])); lb.setFont(new Font("Dialog", Font.PLAIN, 12)); lb.setOpaque(true); lb.setBorder(BorderFactory.createEmptyBorder(1, 2, 1, 1)); lb.setHorizontalAlignment(JLabel.CENTER); if (dark) { lb.setBackground(lb.getBackground().darker()); lb.setForeground(Color.white); } else lb.setForeground(Color.black); temp = lb.getPreferredSize().width; lb.setPreferredSize(new Dimension(temp, jTblProbs.getRowHeight())); if (lengths[k] < temp) lengths[k] = temp; temp = 0; if (k == nParents - 1) { gbc.gridwidth = GridBagConstraints.REMAINDER; dark = (dark == true) ? false : true; } jPlRowHeader.add(lb, gbc); addNum++; } for (int k = nParents - 1; k >= 0; k--) { int iParent2 = m_BayesNet.getParent(m_nCurrentNode, k); if (idx[k] == m_BayesNet.getCardinality(iParent2) - 1 && k != 0) { idx[k] = 0; continue; } else { idx[k]++; break; } } int iParent2 = m_BayesNet.getParent(m_nCurrentNode, 0); if (idx[0] == m_BayesNet.getCardinality(iParent2)) { JLabel lb = (JLabel) jPlRowHeader.getComponent(addNum - 1); jPlRowHeader.remove(addNum - 1); lb.setPreferredSize(new Dimension(lb.getPreferredSize().width, jTblProbs .getRowHeight())); gbc.gridwidth = GridBagConstraints.REMAINDER; gbc.weighty = 1; jPlRowHeader.add(lb, gbc); gbc.weighty = 0; break; } } gbc.gridwidth = 1; // The following panel contains the names of the // parents // and is displayed above the row names to identify // which value belongs to which parent JPanel jPlRowNames = new JPanel(new GridBagLayout()); for (int j = 0; j < nParents; j++) { JLabel lb2; JLabel lb1 = new JLabel(m_BayesNet.getNodeName(m_BayesNet.getParent(nTargetNode, j))); lb1.setBorder(BorderFactory.createEmptyBorder(1, 2, 1, 1)); Dimension tempd = lb1.getPreferredSize(); if (tempd.width < lengths[j]) { lb1.setPreferredSize(new Dimension(lengths[j], tempd.height)); lb1.setHorizontalAlignment(JLabel.CENTER); lb1.setMinimumSize(new Dimension(lengths[j], tempd.height)); } else if (tempd.width > lengths[j]) { lb2 = (JLabel) jPlRowHeader.getComponent(j); lb2.setPreferredSize(new Dimension(tempd.width, lb2.getPreferredSize().height)); } jPlRowNames.add(lb1, gbc); } js.setRowHeaderView(jPlRowHeader); js.setCorner(JScrollPane.UPPER_LEFT_CORNER, jPlRowNames); } final JDialog dlg = new JDialog((Frame) GUI.this.getTopLevelAncestor(), "Probability Distribution Table For " + m_BayesNet.getNodeName(nTargetNode), true); dlg.setSize(500, 400); dlg.setLocation(GUI.this.getLocation().x + GUI.this.getWidth() / 2 - 250, GUI.this.getLocation().y + GUI.this.getHeight() / 2 - 200); dlg.getContentPane().setLayout(new BorderLayout()); dlg.getContentPane().add(js, BorderLayout.CENTER); JButton jBtRandomize = new JButton("Randomize"); jBtRandomize.setMnemonic('R'); jBtRandomize.addActionListener(new ActionListener() { public void actionPerformed(ActionEvent ae) { tm.randomize(); dlg.repaint(); } }); JButton jBtOk = new JButton("Ok"); jBtOk.setMnemonic('O'); jBtOk.addActionListener(new ActionListener() { public void actionPerformed(ActionEvent ae) { tm.setData(); try { m_BayesNet.setDistribution(m_nCurrentNode, tm.m_fProbs); m_jStatusBar.setText(m_BayesNet.lastActionMsg()); updateStatus(); } catch (Exception e) { e.printStackTrace(); } dlg.setVisible(false); } }); JButton jBtCancel = new JButton("Cancel"); jBtCancel.setMnemonic('C'); jBtCancel.addActionListener(new ActionListener() { public void actionPerformed(ActionEvent ae) { dlg.setVisible(false); } }); Container c = new Container(); c.setLayout(new GridBagLayout()); c.add(jBtRandomize); c.add(jBtOk); c.add(jBtCancel); dlg.getContentPane().add(c, BorderLayout.SOUTH); dlg.setVisible(true); } // editCPT /** * Main method. Builds up menus and reads from file if one is specified. */ public static void main(String[] args) { weka.core.logging.Logger.log(weka.core.logging.Logger.Level.INFO, "Logging started"); LookAndFeel.setLookAndFeel(); JFrame jf = new JFrame("Bayes Network Editor"); final GUI g = new GUI(); JMenuBar menuBar = g.getMenuBar(); if (args.length>0) { try { g.readBIFFromFile(args[0]); } catch (IOException ex) { ex.printStackTrace(); } catch (BIFFormatException bf) { bf.printStackTrace(); System.exit(-1); } } jf.setJMenuBar(menuBar); jf.getContentPane().add(g); jf.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE); jf.setSize(800, 600); jf.setVisible(true); g.m_Selection.updateGUI(); GenericObjectEditor.registerEditors(); } // main } // end of class