edu.cmu.tetradapp.editor.ScoredGraphsDisplay 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.graph.Graph;
import edu.cmu.tetrad.graph.GraphNode;
import edu.cmu.tetrad.graph.GraphTransforms;
import edu.cmu.tetrad.graph.Node;
import edu.cmu.tetrad.search.utils.DagScorer;
import edu.cmu.tetrad.util.NumberFormatUtil;
import edu.cmu.tetrad.util.TetradSerializable;
import edu.cmu.tetradapp.model.ScoredGraphsWrapper;
import edu.cmu.tetradapp.workbench.DisplayEdge;
import edu.cmu.tetradapp.workbench.DisplayNode;
import edu.cmu.tetradapp.workbench.GraphWorkbench;
import org.apache.commons.math3.util.FastMath;
import javax.swing.*;
import javax.swing.border.TitledBorder;
import javax.swing.event.ChangeEvent;
import javax.swing.event.ChangeListener;
import java.awt.*;
import java.awt.event.InputEvent;
import java.awt.event.KeyEvent;
import java.text.NumberFormat;
import java.util.List;
import java.util.*;
/**
* Assumes that the search method of the CPDAG search has been run and shows the various options for DAG's consistent
* with correlation information over the variables.
*
* @author josephramsey
* @version $Id: $Id
*/
public class ScoredGraphsDisplay extends JPanel implements GraphEditable {
/**
* The number format for displaying scores.
*/
private final NumberFormat nf = NumberFormatUtil.getInstance().getNumberFormat();
/**
* The workbench for displaying the graphs.
*/
private GraphWorkbench workbench;
/**
* Whether to show only the highest scoring graph.
*/
private boolean showHighestScoreOnly;
/**
* The DAGs to scores.
*/
private Map dagsToScores;
/**
* The DAGs.
*/
private List dags;
/**
* The label for the score.
*/
private JLabel scoreLabel;
/**
* The scored graphs wrapper.
*/
private ScoredGraphsWrapper scoredGraphsWrapper;
/**
* Constructor for ScoredGraphsDisplay.
*
* @param scoredGraphsWrapper a {@link edu.cmu.tetradapp.model.ScoredGraphsWrapper} object
*/
public ScoredGraphsDisplay(ScoredGraphsWrapper scoredGraphsWrapper) {
if (scoredGraphsWrapper == null) {
throw new NullPointerException();
}
this.scoredGraphsWrapper = scoredGraphsWrapper;
this.dagsToScores = scoredGraphsWrapper.getGraphsToScores();
setup();
}
/**
* Constructor for ScoredGraphsDisplay.
*
* @param graph a {@link edu.cmu.tetrad.graph.Graph} object
* @param scorer a {@link edu.cmu.tetrad.search.utils.DagScorer} object
*/
public ScoredGraphsDisplay(Graph graph, DagScorer scorer) {
List _dags = GraphTransforms.generateCpdagDags(graph, true);
for (Graph _graph : _dags) {
double score = Double.NaN;
if (scorer != null) {
score = scorer.scoreDag(_graph);
}
this.dagsToScores.put(_graph, score);
}
setup();
}
private void setup() {
if (this.dagsToScores.isEmpty()) {
throw new IllegalArgumentException("Empty map.");
}
double max = Double.NEGATIVE_INFINITY;
for (Graph dag : this.dagsToScores.keySet()) {
if (this.dagsToScores.get(dag) > max) max = this.dagsToScores.get(dag);
}
List dags = new ArrayList<>();
if (max != Double.NEGATIVE_INFINITY && this.showHighestScoreOnly) {
for (Graph dag : this.dagsToScores.keySet()) {
if (this.dagsToScores.get(dag) == max) {
dags.add(dag);
}
}
} else {
dags.addAll(this.dagsToScores.keySet());
}
if (max != Double.NEGATIVE_INFINITY) {
Collections.sort(dags, new Comparator() {
public int compare(Graph graph, Graph graph1) {
return (int) FastMath.signum(ScoredGraphsDisplay.this.dagsToScores.get(graph) - ScoredGraphsDisplay.this.dagsToScores.get(graph1));
}
});
}
this.dags = dags;
if (dags.size() == 0) {
throw new IllegalArgumentException("No graphs to display.");
}
int index = -1;
Graph dag = this.scoredGraphsWrapper.getGraph();
if (dag == null) {
this.scoredGraphsWrapper.setSelectedGraph(dags.get(0));
dag = this.scoredGraphsWrapper.getGraph();
}
index = dags.indexOf(dag);
if (index == -1) {
dag = dags.get(0);
index = 0;
}
this.workbench = new GraphWorkbench(dag);
SpinnerNumberModel model =
new SpinnerNumberModel(index + 1, 1, dags.size(), 1);
model.addChangeListener(new ChangeListener() {
public void stateChanged(ChangeEvent e) {
int index = model.getNumber().intValue() - 1;
ScoredGraphsDisplay.this.workbench.setGraph(dags.get(index));
if (ScoredGraphsDisplay.this.scoredGraphsWrapper != null) {
ScoredGraphsDisplay.this.scoredGraphsWrapper.setSelectedGraph(dags.get(index));
}
setScore(index - 1);
firePropertyChange("modelChanged", null, null);
}
});
JSpinner spinner = new JSpinner();
spinner.setModel(model);
JLabel totalLabel = new JLabel(" of " + dags.size() + " ");
this.scoreLabel = new JLabel();
setScore(dags.size() - 1);
spinner.setPreferredSize(new Dimension(50, 20));
spinner.setMaximumSize(spinner.getPreferredSize());
Box b = Box.createVerticalBox();
Box b1 = Box.createHorizontalBox();
b1.add(Box.createHorizontalGlue());
b1.add(new JLabel("DAG "));
b1.add(spinner);
b1.add(totalLabel);
b1.add(Box.createHorizontalGlue());
b1.add(new JLabel("Score = "));
b1.add(this.scoreLabel);
b1.add(Box.createHorizontalStrut(10));
b1.add(Box.createHorizontalGlue());
b1.add(new JButton(new CopySubgraphAction(this)));
b.add(b1);
Box b2 = Box.createHorizontalBox();
JPanel graphPanel = new JPanel();
graphPanel.setLayout(new BorderLayout());
JScrollPane jScrollPane = new JScrollPane(this.workbench);
jScrollPane.setPreferredSize(new Dimension(400, 400));
graphPanel.add(jScrollPane);
graphPanel.setBorder(new TitledBorder("Maximum Scoring DAGs in forbid_latent_common_causes"));
b2.add(graphPanel);
b.add(b2);
setLayout(new BorderLayout());
add(menuBar(), BorderLayout.NORTH);
add(b, BorderLayout.CENTER);
}
private void setScore(int i) {
Double score = this.dagsToScores.get(this.dags.get(i));
String text;
if (Double.isNaN(score)) {
text = "Not provided";
} else {
text = this.nf.format(score);
}
this.scoreLabel.setText(text);
}
/**
* getSelectedModelComponents.
*
* @return a {@link java.util.List} object
*/
public List getSelectedModelComponents() {
Component[] components = getWorkbench().getComponents();
List selectedModelComponents =
new ArrayList<>();
for (Component comp : components) {
if (comp instanceof DisplayNode) {
selectedModelComponents.add(
((DisplayNode) comp).getModelNode());
} else if (comp instanceof DisplayEdge) {
selectedModelComponents.add(
((DisplayEdge) comp).getModelEdge());
}
}
return selectedModelComponents;
}
/**
* {@inheritDoc}
*/
public void pasteSubsession(List © 2015 - 2025 Weber Informatics LLC | Privacy Policy