edu.cmu.tetradapp.editor.BayesImEditorWizard Maven / Gradle / Ivy
The newest version!
///////////////////////////////////////////////////////////////////////////////
// For information as to what this class does, see the Javadoc, below. //
// Copyright (C) 1998, 1999, 2000, 2001, 2002, 2003, 2004, 2005, 2006, //
// 2007, 2008, 2009, 2010, 2014, 2015, 2022 by Peter Spirtes, Richard //
// Scheines, Joseph Ramsey, and Clark Glymour. //
// //
// 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA //
///////////////////////////////////////////////////////////////////////////////
package edu.cmu.tetradapp.editor;
import edu.cmu.tetrad.bayes.BayesIm;
import edu.cmu.tetrad.graph.Graph;
import edu.cmu.tetrad.graph.Node;
import edu.cmu.tetradapp.workbench.GraphWorkbench;
import javax.swing.*;
import javax.swing.border.MatteBorder;
import javax.swing.table.TableCellEditor;
import java.awt.*;
import java.io.Serial;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
/**
* Allows the user to choose a variable in a Bayes net and edit the parameters associated with that variable. Parameters
* are of the form P(Node=value1|Parent1=value2, Parent2=value2,...); values for these parameters are probabilities
* ranging from 0.0 to 1.0. For a given combination of parent values for node N, the probabilities for the values of N
* conditional on that combination of parent values must sum to 1.0
*
* @author josephramsey
* @version $Id: $Id
*/
public final class BayesImEditorWizard extends JPanel {
@Serial
private static final long serialVersionUID = -588986830104732678L;
/**
* The BayesIm being edited.
*/
private final BayesIm bayesIm;
/**
* The combo box that allows the user to choose a variable to edit.
*/
private final JComboBox varNamesComboBox;
/**
* The workbench that contains the Bayes net.
*/
private final GraphWorkbench workbench;
/**
* The panel that contains the table.
*/
private final JPanel tablePanel;
/**
* The table that allows the user to edit the parameters for a given variable.
*/
private BayesImNodeEditingTable editingTable;
/**
* True iff the user is allowed to edit the table.
*/
private boolean enableEditing = true;
/**
* Constructor for BayesImEditorWizard.
*
* @param bayesIm a {@link edu.cmu.tetrad.bayes.BayesIm} object
* @param workbench a {@link edu.cmu.tetradapp.workbench.GraphWorkbench} object
*/
public BayesImEditorWizard(BayesIm bayesIm, GraphWorkbench workbench) {
if (bayesIm == null) {
throw new NullPointerException();
}
if (workbench == null) {
throw new NullPointerException();
}
workbench.setAllowDoubleClickActions(false);
setBorder(new MatteBorder(10, 10, 10, 10, getBackground()));
setLayout(new BoxLayout(this, BoxLayout.Y_AXIS));
setFont(new Font("SanSerif", Font.BOLD, 12));
// Set up components.
this.varNamesComboBox = createVarNamesComboBox(bayesIm);
workbench.scrollWorkbenchToNode(
(Node) (this.varNamesComboBox.getSelectedItem()));
JButton nextButton = new JButton("Next");
nextButton.setMnemonic('N');
Node node = (Node) (this.varNamesComboBox.getSelectedItem());
this.editingTable = new BayesImNodeEditingTable(node, bayesIm);
this.editingTable.addPropertyChangeListener((evt) -> {
if ("modelChanged".equals(evt.getPropertyName())) {
firePropertyChange("modelChanged", null, null);
}
});
JScrollPane scroll = new JScrollPane(this.editingTable);
scroll.setPreferredSize(new Dimension(0, 150));
this.tablePanel = new JPanel();
this.tablePanel.setLayout(new BorderLayout());
this.tablePanel.add(scroll, BorderLayout.CENTER);
this.editingTable.grabFocus();
// Do Layout.
Box b1 = Box.createHorizontalBox();
b1.add(new JLabel("1. Choose the next variable to edit: "));
b1.add(this.varNamesComboBox);
b1.add(nextButton);
b1.add(Box.createHorizontalGlue());
Box b2 = Box.createHorizontalBox();
b2.add(new JLabel("2. Scroll to a row (that is, combination of "
+ "parent values) in the table below."));
b2.add(Box.createHorizontalGlue());
Box b3 = Box.createHorizontalBox();
b3.add(new JLabel("3. Click in the appropriate box and assign a probability"
+ " to each value of the chosen"));
b3.add(Box.createHorizontalGlue());
Box b3a = Box.createHorizontalBox();
b3a.add(new JLabel(" variable in that row."));
b3a.add(Box.createHorizontalGlue());
Box b4 = Box.createHorizontalBox();
b4.add(this.tablePanel, BorderLayout.CENTER);
Box b5 = Box.createHorizontalBox();
b5.add(new JLabel("Right click in table to randomize."));
b5.add(Box.createHorizontalGlue());
add(b1);
add(Box.createVerticalStrut(1));
add(b2);
add(Box.createVerticalStrut(5));
add(b3);
add(b3a);
add(b4);
add(b5);
// Add listeners.
this.varNamesComboBox.addActionListener((e) -> {
Node n = (Node) (this.varNamesComboBox.getSelectedItem());
getWorkbench().scrollWorkbenchToNode(n);
setCurrentNode(n);
});
nextButton.addActionListener((e) -> {
int current = this.varNamesComboBox.getSelectedIndex();
int max = this.varNamesComboBox.getItemCount();
++current;
if (current == max) {
JOptionPane.showMessageDialog(this,
"There are no more variables.");
}
int set = (current < max) ? current : 0;
this.varNamesComboBox.setSelectedIndex(set);
});
workbench.addPropertyChangeListener((evt) -> {
if (evt.getPropertyName().equals("selectedNodes")) {
List selection = (List) (evt.getNewValue());
if (selection.size() == 1) {
this.varNamesComboBox.setSelectedItem(selection.get(0));
}
}
});
this.bayesIm = bayesIm;
this.workbench = workbench;
}
private JComboBox createVarNamesComboBox(BayesIm bayesIm) {
JComboBox varNameComboBox = new JComboBox<>();
varNameComboBox.setBackground(Color.white);
Graph graph = bayesIm.getBayesPm().getDag();
List nodes = new ArrayList<>(graph.getNodes());
Collections.sort(nodes);
nodes.forEach(varNameComboBox::addItem);
if (varNameComboBox.getItemCount() > 0) {
varNameComboBox.setSelectedIndex(0);
}
return varNameComboBox;
}
/**
* Sets the getModel display to reflect the stored values of the getModel node.
*/
private void setCurrentNode(Node node) {
TableCellEditor cellEditor = this.editingTable.getCellEditor();
if (cellEditor != null) {
cellEditor.cancelCellEditing();
}
this.editingTable = new BayesImNodeEditingTable(node, getBayesIm());
this.editingTable.addPropertyChangeListener((evt) -> {
if ("modelChanged".equals(evt.getPropertyName())) {
firePropertyChange("modelChanged", null, null);
}
});
JScrollPane scroll = new JScrollPane(this.editingTable);
scroll.setPreferredSize(new Dimension(0, 150));
this.tablePanel.removeAll();
this.tablePanel.add(scroll, BorderLayout.CENTER);
this.tablePanel.revalidate();
this.tablePanel.repaint();
this.editingTable.grabFocus();
}
/**
* Getter for the field bayesIm.
*
* @return a {@link edu.cmu.tetrad.bayes.BayesIm} object
*/
public BayesIm getBayesIm() {
return this.bayesIm;
}
private GraphWorkbench getWorkbench() {
return this.workbench;
}
/**
* isEnableEditing.
*
* @return a boolean
*/
public boolean isEnableEditing() {
return this.enableEditing;
}
/**
* enableEditing.
*
* @param enableEditing a boolean
*/
public void enableEditing(boolean enableEditing) {
this.enableEditing = enableEditing;
if (this.workbench != null) {
this.workbench.setEnableEditing(enableEditing);
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy