use of edu.cmu.tetradapp.workbench.GraphWorkbench in project tetrad by cmu-phil.
the class MimSearchEditor method workbenchScroll.
private JScrollPane workbenchScroll(String resultLabel) {
Graph resultGraph = resultGraph();
Graph sourceGraph = getMimRunner().getSourceGraph();
Graph latestWorkbenchGraph = (Graph) getMimRunner().getParams().get("sourceGraph", null);
boolean arrangedAll = GraphUtils.arrangeBySourceGraph(resultGraph, latestWorkbenchGraph);
if (!arrangedAll) {
GraphUtils.arrangeBySourceGraph(resultGraph, sourceGraph);
}
this.workbench = new GraphWorkbench(resultGraph);
this.workbench.setAllowDoubleClickActions(false);
this.workbenchScroll = new JScrollPane(getWorkbench());
getWorkbenchScroll().setPreferredSize(new Dimension(450, 450));
getWorkbenchScroll().setBorder(new TitledBorder(resultLabel));
this.workbench.addMouseListener(new MouseAdapter() {
public void mouseExited(MouseEvent e) {
storeLatestWorkbenchGraph();
}
});
return getWorkbenchScroll();
}
use of edu.cmu.tetradapp.workbench.GraphWorkbench in project tetrad by cmu-phil.
the class MimSearchEditor method updateDisplayPanel.
private void updateDisplayPanel() {
displayPanel.removeAll();
JTabbedPane tabbedPane = new JTabbedPane();
if (getMimRunner().getStructureGraph() != null) {
if (getMimRunner().getStructureGraph() != null) {
// DataGraphUtils.circleLayout(structureGraph, 200, 200, 150);
Graph structureGraph = getMimRunner().getStructureGraph();
doDefaultArrangement(structureGraph);
GraphWorkbench structureWorkbench = new GraphWorkbench(structureGraph);
structureWorkbench.setAllowDoubleClickActions(false);
tabbedPane.add("Structure Model", new JScrollPane(structureWorkbench));
}
}
if (getMimRunner().getClusters() != null) {
ClusterEditor editor = new ClusterEditor(getMimRunner().getClusters(), getMimRunner().getData().getVariableNames());
ClusterEditor clusterEditor = editor;
tabbedPane.add("Measurement Model", editor);
}
if (getMimRunner().getFullGraph() != null) {
Graph fullGraph = getMimRunner().getFullGraph();
doDefaultArrangement(fullGraph);
GraphUtils.fruchtermanReingoldLayout(fullGraph);
GraphWorkbench fullGraphBench = new GraphWorkbench(fullGraph);
tabbedPane.add("Full Graph", new JScrollPane(fullGraphBench));
}
displayPanel.add(tabbedPane, BorderLayout.CENTER);
displayPanel.revalidate();
displayPanel.repaint();
}
use of edu.cmu.tetradapp.workbench.GraphWorkbench in project tetrad by cmu-phil.
the class PcGesSearchEditor method layoutByKnowledge.
public void layoutByKnowledge() {
GraphWorkbench resultWorkbench = getWorkbench();
Graph graph = resultWorkbench.getGraph();
IKnowledge knowledge = (IKnowledge) getAlgorithmRunner().getParams().get("knowledge", new Knowledge2());
SearchGraphUtils.arrangeByKnowledgeTiers(graph, knowledge);
}
use of edu.cmu.tetradapp.workbench.GraphWorkbench in project tetrad by cmu-phil.
the class PcGesSearchEditor method addSpecialMenus.
protected void addSpecialMenus(JMenuBar menuBar) {
if (!(getAlgorithmRunner() instanceof IGesRunner)) {
JMenu test = new JMenu("Independence");
menuBar.add(test);
IndTestMenuItems.addIndependenceTestChoices(test, this);
// test.addSeparator();
//
// AlgorithmRunner algorithmRunner = getAlgorithmRunner();
// if (algorithmRunner instanceof IndTestProducer) {
// IndTestProducer p = (IndTestProducer) algorithmRunner;
// IndependenceFactsAction action =
// new IndependenceFactsAction(this, p, "Independence Facts...");
// test.add(action);
// }
}
JMenu graph = new JMenu("Graph");
JMenuItem showDags = new JMenuItem("Show DAGs in forbid_latent_common_causes");
// JMenuItem meekOrient = new JMenuItem("Meek Orientation");
JMenuItem dagInPattern = new JMenuItem("Choose DAG in forbid_latent_common_causes");
JMenuItem gesOrient = new JMenuItem("Global Score-based Reorientation");
JMenuItem nextGraph = new JMenuItem("Next Graph");
JMenuItem previousGraph = new JMenuItem("Previous Graph");
// graph.add(new LayoutMenu(this));
graph.add(new GraphPropertiesAction(getWorkbench()));
graph.add(new PathsAction(getWorkbench()));
// graph.add(new DirectedPathsAction(getWorkbench()));
// graph.add(new TreksAction(getWorkbench()));
// graph.add(new AllPathsAction(getWorkbench()));
// graph.add(new NeighborhoodsAction(getWorkbench()));
graph.add(new TriplesAction(getWorkbench().getGraph(), getAlgorithmRunner()));
graph.addSeparator();
// graph.add(meekOrient);
graph.add(dagInPattern);
graph.add(gesOrient);
graph.addSeparator();
graph.add(previousGraph);
graph.add(nextGraph);
graph.addSeparator();
graph.add(showDags);
graph.addSeparator();
graph.add(new JMenuItem(new SelectBidirectedAction(getWorkbench())));
graph.add(new JMenuItem(new SelectUndirectedAction(getWorkbench())));
menuBar.add(graph);
showDags.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent e) {
Window owner = (Window) getTopLevelAncestor();
new WatchedProcess(owner) {
public void watch() {
// Needs to be a pattern search; this isn't checked
// before running the algorithm because of allowable
// "slop"--e.g. bidirected edges.
AlgorithmRunner runner = getAlgorithmRunner();
Graph graph = runner.getGraph();
if (graph == null) {
JOptionPane.showMessageDialog(JOptionUtils.centeringComp(), "No result gaph.");
return;
}
// if (runner instanceof ImagesRunner) {
// GraphScorer scorer = ((ImagesRunner) runner).getGraphScorer();
// Graph _graph = ((ImagesRunner) runner).getTopGraphs().get(getIndex()).getGraph();
//
// ScoredGraphsDisplay display = new ScoredGraphsDisplay(_graph, scorer);
// GraphWorkbench workbench = getWorkbench();
//
// EditorWindow editorWindow =
// new EditorWindow(display, "Independence Facts",
// "Close", false, workbench);
// DesktopController.getInstance().addEditorWindow(editorWindow, JLayeredPane.PALETTE_LAYER);
// editorWindow.setVisible(true);
// }
// else {
PatternDisplay display = new PatternDisplay(graph);
GraphWorkbench workbench = getWorkbench();
EditorWindow editorWindow = new EditorWindow(display, "Independence Facts", "Close", false, workbench);
DesktopController.getInstance().addEditorWindow(editorWindow, JLayeredPane.PALETTE_LAYER);
editorWindow.setVisible(true);
// }
}
};
}
});
// meekOrient.addActionListener(new ActionListener() {
// public void actionPerformed(ActionEvent e) {
// ImpliedOrientation rules = getAlgorithmRunner().getMeekRules();
// rules.setKnowledge((IKnowledge) getAlgorithmRunner().getParams().get("knowledge", new Knowledge2()));
// rules.orientImplied(getGraph());
// getGraphHistory().add(getGraph());
// getWorkbench().setGraph(getGraph());
// firePropertyChange("modelChanged", null, null);
// }
// });
dagInPattern.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent e) {
Graph graph = new EdgeListGraph(getGraph());
// // Removing bidirected edges from the pattern before selecting a DAG. 4
// for (Edge edge : graph.getEdges()) {
// if (Edges.isBidirectedEdge(edge)) {
// graph.removeEdge(edge);
// }
// }
Graph dag = SearchGraphUtils.dagFromPattern(graph);
// PatternToDag search = new PatternToDag(new forbid_latent_common_causes(graph));
// Graph dag = search.patternToDagMeek();
getGraphHistory().add(dag);
getWorkbench().setGraph(dag);
((AbstractAlgorithmRunner) getAlgorithmRunner()).setResultGraph(dag);
firePropertyChange("modelChanged", null, null);
}
});
gesOrient.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent e) {
DataModel dataModel = getAlgorithmRunner().getDataModel();
final Graph graph = SearchGraphUtils.reorient(getGraph(), dataModel, getKnowledge());
getGraphHistory().add(graph);
getWorkbench().setGraph(graph);
firePropertyChange("modelChanged", null, null);
}
});
nextGraph.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent e) {
Graph next = getGraphHistory().next();
getWorkbench().setGraph(next);
((AbstractAlgorithmRunner) getAlgorithmRunner()).setResultGraph(next);
firePropertyChange("modelChanged", null, null);
}
});
previousGraph.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent e) {
Graph previous = getGraphHistory().previous();
getWorkbench().setGraph(previous);
((AbstractAlgorithmRunner) getAlgorithmRunner()).setResultGraph(previous);
firePropertyChange("modelChanged", null, null);
}
});
// if (getAlgorithmRunner().supportsKnowledge()) {
// menuBar.add(new Knowledge2Menu(this));
// }
menuBar.add(new LayoutMenu(this));
}
use of edu.cmu.tetradapp.workbench.GraphWorkbench in project tetrad by cmu-phil.
the class DagEditor method editGraph.
private void editGraph(Graph graph) {
this.workbench = new GraphWorkbench(graph);
DagGraphToolbar toolbar = new DagGraphToolbar(getWorkbench());
JMenuBar menuBar = createGraphMenuBar();
JScrollPane scroll = new JScrollPane();
scroll.setPreferredSize(new Dimension(450, 450));
scroll.setViewportView(getWorkbench());
add(scroll, BorderLayout.CENTER);
add(toolbar, BorderLayout.WEST);
add(menuBar, BorderLayout.NORTH);
JLabel label = new JLabel("Double click variable to change name. More information on graph edge types");
label.setFont(new Font("SansSerif", Font.PLAIN, 12));
// Info button added by Zhou to show edge types
JButton infoBtn = new JButton(new ImageIcon(ImageUtils.getImage(this, "info.png")));
infoBtn.setBorder(new EmptyBorder(0, 0, 0, 0));
// Clock info button to show edge types instructions - Zhou
infoBtn.addActionListener(new ActionListener() {
@Override
public void actionPerformed(ActionEvent e) {
helpSet.setHomeID("graph_edge_types");
HelpBroker broker = helpSet.createHelpBroker();
ActionListener listener = new CSH.DisplayHelpFromSource(broker);
listener.actionPerformed(e);
}
});
Box b = Box.createHorizontalBox();
b.add(Box.createHorizontalStrut(2));
b.add(label);
b.add(infoBtn);
b.add(Box.createHorizontalGlue());
b.setBorder(new MatteBorder(0, 0, 1, 0, Color.GRAY));
add(b, BorderLayout.SOUTH);
validate();
}
Aggregations