edu.cmu.tetradapp.editor.BayesImEditorObs Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of tetrad-gui Show documentation
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.bayes.BayesPm;
import edu.cmu.tetrad.graph.Graph;
import edu.cmu.tetradapp.model.BayesImWrapperObs;
import edu.cmu.tetradapp.workbench.GraphWorkbench;

import javax.swing.*;
import java.awt.*;
import java.io.Serial;

/**
 * An editor for Bayes net instantiated models. Assumes that the workbench and parameterized model have been established
 * (that is, that the nodes have been identified and named and that the number and names of the values for the nodes
 * have been specified).
 *
 * @author Aaron Powers
 * @author josephramsey
 * @version $Id: $Id
 */
/////////////////////////////////////////////////////////////////
// allow the user to set the probabilities of given combinations
// of node values
//
public class BayesImEditorObs extends JPanel {

    @Serial
    private static final long serialVersionUID = 3600349007912545105L;

    /**
     * The wrapper for the Bayes IM.
     */
    private final BayesImWrapperObs wrapper;
    /**
     * The wizard that allows the user to modify parameter values for this IM.
     */
    private BayesImEditorWizardObs wizard;

    /**
     * Constructs a new instanted model editor from a Bayes IM.
     *
     * @param wrapper a {@link edu.cmu.tetradapp.model.BayesImWrapperObs} object
     * @param bayesIm a {@link edu.cmu.tetrad.bayes.BayesIm} object
     */
    public BayesImEditorObs(BayesImWrapperObs wrapper, BayesIm bayesIm) {
        if (wrapper == null) {
            throw new NullPointerException();
        }

        this.wrapper = wrapper;
        init(bayesIm);
    }

    /**
     * Constructs a new instanted model editor from a Bayes IM wrapper.
     *
     * @param bayesImWrapperObs a {@link edu.cmu.tetradapp.model.BayesImWrapperObs} object
     */
    public BayesImEditorObs(BayesImWrapperObs bayesImWrapperObs) {
        this(bayesImWrapperObs, bayesImWrapperObs.getBayesIm());
    }

    private void init(BayesIm bayesIm) {
        if (bayesIm == null) {
            throw new NullPointerException("Bayes IM must not be null.");
        }

        BayesPm bayesPm = bayesIm.getBayesPm();
        Graph graph = bayesPm.getDag();
        GraphWorkbench workbench = new GraphWorkbench(graph);

        JMenuBar menuBar = new JMenuBar();
        JMenu file = new JMenu("File");
        menuBar.add(file);
        file.add(new SaveScreenshot(this, true, "Save Screenshot..."));
        file.add(new SaveComponentImage(workbench, "Save Graph Image..."));
        setLayout(new BorderLayout());
        add(menuBar, BorderLayout.NORTH);

        this.wizard = new BayesImEditorWizardObs(bayesIm, workbench);
        this.wizard.enableEditing(false);

        this.wizard.addPropertyChangeListener(evt -> {
            if ("editorValueChanged".equals(evt.getPropertyName())) {
                firePropertyChange("modelChanged", null, null);
            }
        });

        JScrollPane workbenchScroll = new JScrollPane(workbench);
        JScrollPane wizardScroll = new JScrollPane(getWizard());

        workbenchScroll.setPreferredSize(new Dimension(450, 450));

        JSplitPane splitPane = new JSplitPane(JSplitPane.HORIZONTAL_SPLIT,
                workbenchScroll, wizardScroll);
        splitPane.setOneTouchExpandable(true);
        splitPane.setDividerLocation(workbenchScroll.getPreferredSize().width);
        add(splitPane, BorderLayout.CENTER);

        setName("Bayes IM Obs Editor");
        getWizard().addPropertyChangeListener(evt -> {
            if ("editorClosing".equals(evt.getPropertyName())) {
                firePropertyChange("editorClosing", null, getName());
            }

            if ("closeFrame".equals(evt.getPropertyName())) {
                firePropertyChange("closeFrame", null, null);
                firePropertyChange("editorClosing", true, true);
            }

            if ("modelChanged".equals(evt.getPropertyName())) {
                firePropertyChange("modelChanged", evt.getOldValue(),
                        evt.getNewValue());
            }
        });
    }

    /**
     * {@inheritDoc}
     * 
     * Sets the name of this editor.
     */
    public void setName(String name) {
        String oldName = getName();
        super.setName(name);
        firePropertyChange("name", oldName, getName());
    }

    /**
     * @return a reference to this editor.
     */
    private BayesImEditorWizardObs getWizard() {
        return this.wizard;
    }

    /**
     * 
getBayesIm.
     *
     * @param bayesIm a {@link edu.cmu.tetrad.bayes.BayesIm} object
     */
    public void getBayesIm(BayesIm bayesIm) {
        this.wrapper.setBayesIm(bayesIm);
        removeAll();
        init(this.wrapper.getBayesIm());
        revalidate();
        repaint();
        firePropertyChange("modelChanged", null, null);
    }
}