Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
edu.cmu.tetradapp.editor.GridSearchEditor Maven / Gradle / Ivy
package edu.cmu.tetradapp.editor;
import edu.cmu.tetrad.algcomparison.algorithm.Algorithm;
import edu.cmu.tetrad.algcomparison.graph.*;
import edu.cmu.tetrad.algcomparison.independence.IndependenceWrapper;
import edu.cmu.tetrad.algcomparison.score.ScoreWrapper;
import edu.cmu.tetrad.algcomparison.simulation.Simulation;
import edu.cmu.tetrad.algcomparison.simulation.Simulations;
import edu.cmu.tetrad.algcomparison.utils.TakesIndependenceWrapper;
import edu.cmu.tetrad.algcomparison.utils.UsesScoreWrapper;
import edu.cmu.tetrad.annotation.AnnotatedClass;
import edu.cmu.tetrad.annotation.Score;
import edu.cmu.tetrad.annotation.TestOfIndependence;
import edu.cmu.tetrad.data.DataSet;
import edu.cmu.tetrad.data.DataType;
import edu.cmu.tetrad.data.Knowledge;
import edu.cmu.tetrad.graph.EdgeListGraph;
import edu.cmu.tetrad.graph.Graph;
import edu.cmu.tetrad.graph.GraphSaveLoadUtils;
import edu.cmu.tetrad.graph.LayoutUtil;
import edu.cmu.tetrad.util.*;
import edu.cmu.tetradapp.editor.simulation.ParameterTab;
import edu.cmu.tetradapp.model.GridSearchModel;
import edu.cmu.tetradapp.ui.PaddingPanel;
import edu.cmu.tetradapp.ui.model.*;
import edu.cmu.tetradapp.util.*;
import edu.cmu.tetradapp.workbench.GraphWorkbench;
import org.jetbrains.annotations.NotNull;
import javax.swing.*;
import javax.swing.border.EmptyBorder;
import javax.swing.event.ChangeEvent;
import javax.swing.event.ChangeListener;
import javax.swing.event.DocumentEvent;
import javax.swing.event.DocumentListener;
import javax.swing.table.AbstractTableModel;
import javax.swing.table.DefaultTableCellRenderer;
import javax.swing.table.TableColumnModel;
import javax.swing.table.TableRowSorter;
import javax.swing.text.BadLocationException;
import java.awt.*;
import java.io.*;
import java.lang.reflect.InvocationTargetException;
import java.nio.charset.StandardCharsets;
import java.text.DecimalFormat;
import java.text.NumberFormat;
import java.util.List;
import java.util.*;
import java.util.function.Function;
import java.util.regex.PatternSyntaxException;
import java.util.stream.Collectors;
import java.util.stream.Stream;
import static edu.cmu.tetradapp.model.GridSearchModel.getAllSimulationParameters;
/**
* The AlgcomparisonEditor class represents a JPanel that contains different tabs for simulation, algorithm, table
* columns, comparison, and help. It is used for editing an GridSearchModel.
*
* The reference is here:
*
* Ramsey, J. D., Malinsky, D., & Bui, K. V. (2020). Algcomparison: Comparing the performance of graphical structure
* learning algorithms with tetrad. Journal of Machine Learning Research, 21(238), 1-6.
*
* @author josephramsey
*/
public class GridSearchEditor extends JPanel {
/**
* JLabel representing a message indicating that there are no parameters to edit.
*/
private static final JLabel NO_PARAM_LBL = new JLabel("No parameters to edit");
/**
* A JComboBox that holds instances of IndependenceTestModel.
*/
private static JComboBox indTestComboBox;
/**
* ComboBox for selecting a ScoreModel.
*/
private static JComboBox scoreModelComboBox;
/**
* The GridSearchModel class represents a model used in an algorithm comparison application. It contains methods and
* properties related to the comparison of algorithms.
*/
private final GridSearchModel model;
/**
* JTextArea used for displaying verbose output.
*/
private transient JTextArea verboseOutputTextArea;
/**
* JTextArea used for displaying simulation choice information.
*/
private transient JTextArea simulationChoiceTextArea;
/**
* The TextArea component used for displaying algorithm choices.
*/
private transient JTextArea algorithmChoiceTextArea;
/**
* JTextArea used for displaying table column choices.
*/
private transient JTextArea tableColumnsChoiceTextArea;
/**
* JTextArea used for displaying comparison results.
*/
private transient JTextArea comparisonTextArea;
/**
* JTextArea used for displaying help choice information.
*/
private transient JTextArea helpChoiceTextArea;
/**
* Button used to add a simulation.
*/
private transient JButton addSimulation;
/**
* Button used to add an algorithm.
*/
private transient JButton addAlgorithm;
/**
* Button used to add table columns.
*/
private transient JButton addTableColumns;
/**
* Represents a drop-down menu for selecting an algorithm.
*/
private transient JComboBox algorithmDropdown;
/**
* Private variable representing a JScrollPane used for comparing variables.
*/
private transient JScrollPane comparisonScroll;
/**
* The comparisonTabbedPane represents a tabbed pane component in the user interface for displaying comparison
* related data and functionality.
*
* It is a private instance variable of type JTabbedPane.
*/
private transient JTabbedPane comparisonTabbedPane;
/**
* Initializes an instance of AlgcomparisonEditor which is a JPanel containing a JTabbedPane that displays different
* tabs for simulation, algorithm, table columns, comparison and help.
*
* @param model the GridSearchModel to use for the editor
*/
public GridSearchEditor(GridSearchModel model) {
this.model = model;
JTabbedPane tabbedPane = new JTabbedPane();
tabbedPane.setTabPlacement(JTabbedPane.BOTTOM);
tabbedPane.setPreferredSize(new Dimension(800, 400));
addSimulationTab(tabbedPane);
addAlgorithmTab(tabbedPane);
addTableColumnsTab(tabbedPane);
addComparisonTab(tabbedPane);
// addXmlTab(tabbedPane); // todo work on this later.
addHelpTab(tabbedPane);
setLayout(new BorderLayout());
add(tabbedPane, BorderLayout.CENTER);
model.getParameters().set("algcomparisonSaveData", model.getParameters().getBoolean("algcomparisonSaveData", true));
model.getParameters().set("algcomparisonSaveGraphs", model.getParameters().getBoolean("algcomparisonSaveGraphs", true));
model.getParameters().set("algcomparisonSaveCPDAGs", model.getParameters().getBoolean("algcomparisonSaveCPDAGs", false));
model.getParameters().set("algcomparisonSavePAGs", model.getParameters().getBoolean("algcomparisonSavePAGs", false));
model.getParameters().set("algcomparisonSortByUtility", model.getParameters().getBoolean("algcomparisonSortByUtility", false));
model.getParameters().set("algcomparisonShowUtilities", model.getParameters().getBoolean("algcomparisonShowUtilities", false));
model.getParameters().set("algcomparisonSetAlgorithmKnowledge", model.getParameters().getBoolean("algcomparisonSetAlgorithmKnowledge", true));
model.getParameters().set("algcomparisonParallelism", model.getParameters().getInt("algcomparisonParallelism", Runtime.getRuntime().availableProcessors()));
model.getParameters().set("algcomparisonGraphType", model.getParameters().getString("algcomparisonGraphType", "DAG"));
comparisonTextArea.setText(model.getLastComparisonText());
verboseOutputTextArea.setText(model.getLastVerboseOutputText());
SwingUtilities.invokeLater(() -> {
try {
scrollToWord(comparisonTextArea, comparisonScroll, "AVERAGE VALUE");
} catch (BadLocationException ex) {
System.out.println("Scrolling operation failed.");
}
});
}
/**
* Creates a map of parameter components for the given set of parameters and a Parameters object.
*
* @param params the set of parameter names
* @param parameters the Parameters object containing the parameter values
* @return a map of parameter names to corresponding Box components
*/
public static Map createParameterComponents(Set params, Parameters parameters,
boolean listOptionAllowed, boolean bothOptionAllowed) {
ParamDescriptions paramDescriptions = ParamDescriptions.getInstance();
return params.stream()
.collect(Collectors.toMap(
Function.identity(),
e -> createParameterComponent(e, parameters, paramDescriptions.get(e), listOptionAllowed, bothOptionAllowed),
(u, v) -> {
throw new IllegalStateException(String.format("Duplicate key %s.", u));
},
TreeMap::new));
}
/**
* Converts a map of parameter components to an array of Box objects.
*
* @param parameterComponents a map of parameter names to corresponding Box components
* @return an array of Box objects containing the parameter components
*/
public static Box[] toArray(Map parameterComponents) {
ParamDescriptions paramDescriptions = ParamDescriptions.getInstance();
List boolComps = new LinkedList<>();
List otherComps = new LinkedList<>();
parameterComponents.forEach((k, v) -> {
if (paramDescriptions.get(k).getDefaultValue() instanceof Boolean) {
boolComps.add(v);
} else {
otherComps.add(v);
}
});
return Stream.concat(otherComps.stream(), boolComps.stream())
.toArray(Box[]::new);
}
/**
* Creates a component for a specific parameter based on its type and default value.
*
* @param parameter the name of the parameter
* @param parameters the Parameters object containing the parameter values
* @param paramDesc the ParamDescription object containing information about the parameter
* @return a Box component representing the parameter
*/
private static Box createParameterComponent(String parameter, Parameters parameters, ParamDescription paramDesc,
boolean listOptionAllowed, boolean bothOptionAllowed) {
JComponent component;
Object defaultValue = paramDesc.getDefaultValue();
Object[] defaultValues = parameters.getValues(parameter);
if (defaultValue instanceof Double) {
double lowerBoundDouble = paramDesc.getLowerBoundDouble();
double upperBoundDouble = paramDesc.getUpperBoundDouble();
Double[] defValues = new Double[defaultValues.length];
for (int i = 0; i < defaultValues.length; i++) {
if (defaultValues[i] instanceof Number) {
defValues[i] = ((Number) defaultValues[i]).doubleValue();
} else {
throw new IllegalArgumentException("Unexpected type: " + defaultValues[i].getClass());
}
}
if (listOptionAllowed) {
component = getListDoubleTextField(parameter, parameters, defValues, lowerBoundDouble, upperBoundDouble);
} else {
component = getDoubleTextField(parameter, parameters, (Double) defaultValue, lowerBoundDouble, upperBoundDouble);
}
} else if (defaultValue instanceof Integer) {
int lowerBoundInt = paramDesc.getLowerBoundInt();
int upperBoundInt = paramDesc.getUpperBoundInt();
Integer[] defValues = new Integer[defaultValues.length];
for (int i = 0; i < defaultValues.length; i++) {
try {
defValues[i] = (int) defaultValues[i];
} catch (Exception e) {
throw new RuntimeException("Parameter " + parameter + " has a default value that is not an integer: " + defaultValues[i]);
}
}
if (listOptionAllowed) {
component = getListIntTextField(parameter, parameters, defValues, lowerBoundInt, upperBoundInt);
} else {
component = getIntTextField(parameter, parameters, (Integer) defaultValue, lowerBoundInt, upperBoundInt);
}
} else if (defaultValue instanceof Long) {
long lowerBoundLong = paramDesc.getLowerBoundLong();
long upperBoundLong = paramDesc.getUpperBoundLong();
Long[] defValues = new Long[defaultValues.length];
for (int i = 0; i < defaultValues.length; i++) {
try {
defValues[i] = (Long) defaultValues[i];
} catch (Exception e) {
throw new RuntimeException("Parameter " + parameter + " has a default value that is not a long: " + defaultValues[i]);
}
}
if (listOptionAllowed) {
component = getListLongTextField(parameter, parameters, defValues, lowerBoundLong, upperBoundLong);
} else {
component = getLongTextField(parameter, parameters, (Long) defaultValue, lowerBoundLong, upperBoundLong);
}
} else if (defaultValue instanceof Boolean) {
component = getBooleanSelectionBox(parameter, parameters, bothOptionAllowed);
} else if (defaultValue instanceof String) {
component = getStringField(parameter, parameters, (String) defaultValue);
} else {
throw new IllegalArgumentException("Unexpected type: " + defaultValue.getClass());
}
Box paramRow = Box.createHorizontalBox();
JLabel paramLabel = new JLabel(paramDesc.getShortDescription());
String longDescription = paramDesc.getLongDescription();
if (longDescription != null) {
paramLabel.setToolTipText(longDescription);
}
paramRow.add(paramLabel);
paramRow.add(Box.createHorizontalGlue());
paramRow.add(component);
return paramRow;
}
/**
* Returns a customized DoubleTextField with specified parameters.
*
* @param parameter the name of the parameter to be set in the Parameters object
* @param parameters the Parameters object to store the parameter values
* @param defaultValue the default value to set in the DoubleTextField
* @param lowerBound the lowerbound limit for valid input values in the DoubleTextField
* @param upperBound the upperbound limit for valid input values in the DoubleTextField
* @return a DoubleTextField with the specified parameters
*/
public static DoubleTextField getDoubleTextField(String parameter, Parameters parameters,
double defaultValue, double lowerBound, double upperBound) {
DoubleTextField field = new DoubleTextField(defaultValue,
8, new DecimalFormat("0.####"), new DecimalFormat("0.0#E0"), 0.001);
field.setFilter((value, oldValues) -> {
if (Double.isNaN(value)) {
return oldValues;
}
if (value < lowerBound) {
return oldValues;
}
if (value > upperBound) {
return oldValues;
}
try {
parameters.set(parameter, value);
} catch (Exception e) {
// Ignore.
}
return value;
});
return field;
}
/**
* Creates a ListDoubleTextField component with the given parameters.
*
* @param parameter the name of the parameter
* @param parameters the Parameters object containing the parameter values
* @param defaultValues the default values for the component
* @param lowerBound the lower bound for the values
* @param upperBound the upper bound for the values
* @return a ListDoubleTextField component with the specified parameters
*/
public static ListDoubleTextField getListDoubleTextField(String parameter, Parameters parameters,
Double[] defaultValues, double lowerBound, double upperBound) {
ListDoubleTextField field = new ListDoubleTextField(defaultValues,
8, new DecimalFormat("0.####"), new DecimalFormat("0.0#E0"), 0.001);
field.setFilter((values, oldValues) -> {
if (values.length == 0) {
return oldValues;
}
List valuesList = new ArrayList<>();
for (Double value : values) {
if (Double.isNaN(value)) {
continue;
}
if (value < lowerBound) {
continue;
}
if (value > upperBound) {
continue;
}
valuesList.add(value);
}
if (valuesList.isEmpty()) {
return oldValues;
}
Double[] newValues = valuesList.toArray(new Double[0]);
try {
parameters.set(parameter, (Object[]) newValues);
} catch (Exception e) {
// Ignore.
}
return newValues;
});
return field;
}
/**
* Returns an IntTextField with the specified parameters.
*
* @param parameter the name of the parameter
* @param parameters the Parameters object to update with the new value
* @param defaultValue the default value for the IntTextField
* @param lowerBound the lower bound for valid values
* @param upperBound the upper bound for valid values
* @return an IntTextField with the specified parameters
*/
public static IntTextField getIntTextField(String parameter, Parameters parameters,
int defaultValue, double lowerBound, double upperBound) {
IntTextField field = new IntTextField(defaultValue, 8);
field.setFilter((value, oldValue) -> {
if (value < lowerBound) {
return oldValue;
}
if (value > upperBound) {
return oldValue;
}
try {
parameters.set(parameter, value);
} catch (Exception e) {
// Ignore.
}
return value;
});
return field;
}
/**
* Returns a ListIntTextField component with the specified parameters.
*
* @param parameter the name of the parameter
* @param parameters the Parameters object containing the parameter values
* @param defaultValues the default values for the component
* @param lowerBound the lower bound for the values
* @param upperBound the upper bound for the values
* @return a ListIntTextField component with the specified parameters
*/
public static ListIntTextField getListIntTextField(String parameter, Parameters parameters,
Integer[] defaultValues, double lowerBound, double upperBound) {
ListIntTextField field = new ListIntTextField(defaultValues, 8);
field.setFilter((values, oldValues) -> {
if (values.length == 0) {
return oldValues;
}
List valuesList = new ArrayList<>();
for (Integer value : values) {
if (value < lowerBound) {
continue;
}
if (value > upperBound) {
continue;
}
valuesList.add(value);
}
if (valuesList.isEmpty()) {
return oldValues;
}
Integer[] newValues = valuesList.toArray(new Integer[0]);
try {
parameters.set(parameter, (Object[]) newValues);
} catch (Exception e) {
// Ignore.
}
return newValues;
});
return field;
}
/**
* Returns a LongTextField object with the specified parameters.
*
* @param parameter The name of the parameter to set in the Parameters object.
* @param parameters The Parameters object to set the parameter in.
* @param defaultValue The default value to use for the LongTextField.
* @param lowerBound The lower bound for the LongTextField value.
* @param upperBound The upper bound for the LongTextField value.
* @return A LongTextField object with the specified parameters.
*/
public static LongTextField getLongTextField(String parameter, Parameters parameters,
long defaultValue, long lowerBound, long upperBound) {
LongTextField field = new LongTextField(defaultValue, 8);
field.setFilter((value, oldValue) -> {
if (value < lowerBound) {
return oldValue;
}
if (value > upperBound) {
return oldValue;
}
try {
parameters.set(parameter, value);
} catch (Exception e) {
// Ignore.
}
return value;
});
return field;
}
public static ListLongTextField getListLongTextField(String parameter, Parameters parameters,
Long[] defaultValues, long lowerBound, long upperBound) {
ListLongTextField field = new ListLongTextField(defaultValues, 8);
field.setFilter((values, oldValues) -> {
if (values.length == 0) {
return oldValues;
}
List valuesList = new ArrayList<>();
for (Long value : values) {
if (value < lowerBound) {
continue;
}
if (value > upperBound) {
continue;
}
valuesList.add(value);
}
if (valuesList.isEmpty()) {
return oldValues;
}
Long[] newValues = valuesList.toArray(new Long[0]);
try {
parameters.set(parameter, (Object[]) newValues);
} catch (Exception e) {
// Ignore.
}
return newValues;
});
return field;
}
/**
* Returns a Box component representing a boolean selection box.
*
* @param parameter the name of the parameter
* @param parameters the Parameters object containing the parameter values
* @param bothOptionAllowed whether the option allows one to select both true and false
* @return a Box component representing the boolean selection box
*/
public static Box getBooleanSelectionBox(String parameter, Parameters parameters, boolean bothOptionAllowed) {
Box selectionBox = Box.createHorizontalBox();
JRadioButton yesButton = new JRadioButton("Yes");
JRadioButton noButton = new JRadioButton("No");
JRadioButton bothButton = null;
if (bothOptionAllowed) {
bothButton = new JRadioButton("Both");
}
// Button group to ensure only one option can be selected
ButtonGroup selectionBtnGrp = new ButtonGroup();
selectionBtnGrp.add(yesButton);
selectionBtnGrp.add(noButton);
if (bothOptionAllowed) {
selectionBtnGrp.add(bothButton);
}
Object[] values = parameters.getValues(parameter);
Boolean[] booleans = new Boolean[values.length];
try {
for (int i = 0; i < values.length; i++) {
try {
booleans[i] = (Boolean) values[i];
} catch (Exception e) {
throw new RuntimeException("Parameter " + parameter + " has a value that is not a boolean: " + values[i]);
}
}
} catch (Exception e) {
throw new RuntimeException(e);
}
// Set default selection
if (booleans.length == 1 && booleans[0]) {
yesButton.setSelected(true);
} else if (booleans.length == 1) {
noButton.setSelected(true);
} else if (booleans.length == 2 && bothOptionAllowed) {
bothButton.setSelected(true);
}
// Add to containing box
selectionBox.add(yesButton);
selectionBox.add(noButton);
if (bothOptionAllowed) {
selectionBox.add(bothButton);
}
// Event listener
yesButton.addActionListener((e) -> {
JRadioButton button = (JRadioButton) e.getSource();
if (button.isSelected()) {
Object[] objects = new Object[1];
objects[0] = Boolean.TRUE;
parameters.set(parameter, objects);
}
});
// Event listener
noButton.addActionListener((e) -> {
JRadioButton button = (JRadioButton) e.getSource();
if (button.isSelected()) {
Object[] objects = new Object[1];
objects[0] = Boolean.FALSE;
parameters.set(parameter, objects);
}
});
if (bothOptionAllowed) {
bothButton.addActionListener((e) -> {
JRadioButton button = (JRadioButton) e.getSource();
if (button.isSelected()) {
Object[] objects = new Object[2];
objects[0] = Boolean.TRUE;
objects[1] = Boolean.FALSE;
parameters.set(parameter, objects);
}
});
}
return selectionBox;
}
// /**
// * Returns the XML text used for the XML tab in the AlgcomparisonEditor.
// *
// * @return the XML text
// */
// @NotNull
// private static String getXmlText() {
// return """
// ** This is placeholder text **
//
//
//
//
//
//
//
// src/test/resources/data/simulation
//
//
// RandomForward
// SemSimulation
//
//
//
//
// fisher-z-test
// sem-bic-score
//
//
// sem-bic-score
//
//
//
// 1
// 4,6
// 4
//
//
//
//
// adjacencyPrecision
// arrowheadRecall
// adjacencyRecall
//
//
// true
// true
// true
// true
// true
// true
//
// """;
// }
/**
* Creates a StringTextField component with the specified parameters.
*
* @param parameter the name of the parameter
* @param parameters the Parameters object containing the parameter values
* @param defaultValue the default value for the component
* @return a StringTextField component with the specified parameters
*/
public static StringTextField getStringField(String parameter, Parameters parameters, String defaultValue) {
StringTextField field = new StringTextField(parameters.getString(parameter, defaultValue), 20);
field.setFilter((value, oldValue) -> {
if (value.equals(field.getValue().trim())) {
return oldValue;
}
try {
parameters.set(parameter, value);
} catch (Exception e) {
// Ignore.
}
return value;
});
return field;
}
/**
* Retrieves the parameter text for the given set of parameter names and parameters.
*
* @param paramNamesSet the set of parameter names
* @param parameters the Parameters object containing the parameter values
* @return the parameter text
*/
@NotNull
private static String getParameterText(Set paramNamesSet, Parameters parameters) {
List paramNames = new ArrayList<>(paramNamesSet);
Collections.sort(paramNames);
StringBuilder paramText = new StringBuilder();
ParamDescriptions paramDescriptions = ParamDescriptions.getInstance();
NumberFormat nf = new DecimalFormat("0.####");
for (String name : paramNames) {
ParamDescription description = paramDescriptions.get(name);
Object[] values = parameters.getValues(name);
StringBuilder sb = new StringBuilder();
for (int i = 0; i < values.length; i++) {
if (i > 0) {
sb.append(", ");
}
if (values[i] instanceof Double)
sb.append(nf.format((double) values[i]));
else if (values[i] instanceof Integer)
sb.append((int) values[i]);
else if (values[i] instanceof Long)
sb.append((long) values[i]);
else
sb.append(values[i]);
}
paramText.append("\n\n- ").append(name).append(" = ").append(sb);
paramText.append("\n").append(description.getShortDescription());
paramText.append(". ").append(description.getLongDescription());
}
return paramText.toString();
}
public static void scrollToWord(JTextArea textArea, JScrollPane scrollPane, String word) throws BadLocationException {
String text = textArea.getText();
int pos = text.indexOf(word);
if (pos >= 0) {
Rectangle viewRect = textArea.modelToView2D(pos).getBounds();
if (viewRect != null) {
JViewport viewport = scrollPane.getViewport();
viewRect = new Rectangle(viewRect.x, viewRect.y, viewport.getWidth() - 10, viewport.getHeight() - 10);
textArea.scrollRectToVisible(viewRect);
}
}
}
/**
* Updates the indices in the graph index combo box based on the selected simulation and algorithm.
*
* @param simulationComboBox The combo box that contains the available simulation options.
* @param algorithmComboBox The combo box that contains the available algorithm options.
* @param graphIndexComboBox The combo box to update with the graph indices.
* @param resultsDir The directory where the graph results are stored.
*/
private void updateAlgorithmBoxIndices(JComboBox simulationComboBox, JComboBox algorithmComboBox,
JComboBox graphIndexComboBox, File resultsDir) {
int savedAlgorithm = model.getSelectedAlgorithm();
Object selectedSimulation = simulationComboBox.getSelectedItem();
if (selectedSimulation == null) {
algorithmComboBox.removeAllItems();
graphIndexComboBox.removeAllItems();
return;
}
List algorithmIndices = new ArrayList<>();
if (resultsDir.exists()) {
File[] dirs = resultsDir.listFiles();
if (dirs != null) {
// The dirs array should contain directories for each simulation/algorithm combination. These
// are formatted as, e.g., "5.2" for simulation5 and algorithm 2. We need to iterate through
// all of these directories and find the highest simulation number and the highest
// algorithm number. The number of graphs will be determined once we have the simulation and
// algorithm numbers. These are listed as "graph1.txt", "graph2.txt", etc., in each of these
// directories.
for (File dir : dirs) {
String name = dir.getName();
String[] parts = name.split("\\.");
try {
int simulation = Integer.parseInt(parts[0]);
int algorithm = Integer.parseInt(parts[1]);
if (simulation == (int) selectedSimulation) {
if (!algorithmIndices.contains(algorithm)) {
algorithmIndices.add(algorithm);
}
}
} catch (NumberFormatException e) {
// These aren't directories/files written out by the tool.
}
}
}
}
algorithmComboBox.removeAllItems();
Collections.sort(algorithmIndices);
for (int i : algorithmIndices) {
algorithmComboBox.addItem(i);
}
if (savedAlgorithm > 0) {
algorithmComboBox.setSelectedItem(savedAlgorithm);
}
updateGraphBoxIndices(simulationComboBox, algorithmComboBox, graphIndexComboBox, resultsDir);
}
/**
* Updates the indices in the graph index combo box based on the selected simulation and algorithm.
*
* @param simulationComboBox The combo box that contains the available simulation options.
* @param algorithmComboBox The combo box that contains the available algorithm options.
* @param graphIndexComboBox The combo box to update with the graph indices.
* @param resultsDir The directory where the graph results are stored.
*/
private void updateGraphBoxIndices(JComboBox simulationComboBox, JComboBox algorithmComboBox,
JComboBox graphIndexComboBox, File resultsDir) {
int savedGraphIndex = model.getSelectedGraphIndex();
Object selectedSimulation = simulationComboBox.getSelectedItem();
Object selectedAlgorithm = algorithmComboBox.getSelectedItem();
if (selectedSimulation == null || selectedAlgorithm == null) {
graphIndexComboBox.removeAllItems();
return;
}
int simulation = (int) selectedSimulation;
int algorithm = (int) selectedAlgorithm;
File dir = new File(resultsDir, simulation + "." + algorithm);
List indices = new ArrayList<>();
if (dir.exists()) {
File[] graphs = dir.listFiles();
if (graphs != null) {
for (File graph : graphs) {
String name = graph.getName();
if (!name.startsWith("graph.")) {
continue;
}
String[] parts = name.split("\\.");
int graphIndex;
try {
graphIndex = Integer.parseInt(parts[1]);
if (!indices.contains(graphIndex)) {
indices.add(graphIndex);
}
} catch (NumberFormatException e) {
// These aren't directories/files written out by the tool.
}
}
}
}
graphIndexComboBox.removeAllItems();
Collections.sort(indices);
for (int i : indices) {
graphIndexComboBox.addItem(i);
}
if (savedGraphIndex > 0) {
graphIndexComboBox.setSelectedItem(savedGraphIndex);
}
}
private void updateSelectedGraph(JComboBox simulationComboBox, JComboBox algorithmComboBox,
JComboBox graphIndexComboBox, File resultsDir,
GraphWorkbench workbench) {
Object selectedSimulation = simulationComboBox.getSelectedItem();
Object selectedAlgorithm = algorithmComboBox.getSelectedItem();
Object selectedGraphIndex = graphIndexComboBox.getSelectedItem();
if (selectedSimulation == null || selectedAlgorithm == null || selectedGraphIndex == null) {
return;
}
File dir = new File(resultsDir, (int) selectedSimulation + "." + (int) selectedAlgorithm);
File graphFile = new File(dir, "graph." + selectedGraphIndex + ".txt");
if (graphFile.exists()) {
Graph graph = GraphSaveLoadUtils.loadGraphTxt(graphFile);
LayoutUtil.defaultLayout(graph);
workbench.setGraph(graph);
model.setSelectedGraph(graph);
model.setSelectedSimulation((int) selectedSimulation);
model.setSelectedAlgorithm((int) selectedAlgorithm);
model.setSelectedGraphIndex((int) selectedGraphIndex);
firePropertyChange("modelChanged", null, null);
}
}
private void refreshGraphSelectionContent(JTabbedPane tabbedPane) {
Box tab = (Box) tabbedPane.getComponentAt(4);
tab.removeAll();
tab.add(getGraphSelectorBox());
tab.revalidate();
tab.repaint();
}
/**
* Retrieves a simulation object based on the provided graph and simulation classes.
*
* @param graphClazz The class of the random graph object.
* @param simulationClazz The class of the simulation object.
* @return The simulation object.
* @throws NoSuchMethodException If the constructor for the graph or simulation class cannot be found.
* @throws InvocationTargetException If an error occurs while invoking the graph or simulation constructor.
* @throws InstantiationException If the graph or simulation class cannot be instantiated.
* @throws IllegalAccessException If the graph or simulation constructor or class is inaccessible.
*/
@NotNull
private edu.cmu.tetrad.algcomparison.simulation.Simulation getSimulation(
Class graphClazz,
Class simulationClazz)
throws NoSuchMethodException, InvocationTargetException, InstantiationException, IllegalAccessException {
RandomGraph randomGraph;
if (graphClazz == SingleGraph.class) {
if (model.getSuppliedGraph() == null) {
throw new IllegalArgumentException("No graph supplied.");
}
randomGraph = new SingleGraph(model.getSuppliedGraph());
} else {
randomGraph = graphClazz.getConstructor().newInstance();
}
return simulationClazz.getConstructor(RandomGraph.class).newInstance(randomGraph);
}
@NotNull
private Class getGraphClazz(String graphString) {
List graphTypeStrings = new ArrayList<>(Arrays.asList(ParameterTab.GRAPH_TYPE_ITEMS));
if (model.getSuppliedGraph() != null) {
graphTypeStrings.add("User Supplied Graph");
}
return switch (graphTypeStrings.indexOf(graphString)) {
case 0:
yield RandomForward.class;
case 1:
yield ErdosRenyi.class;
case 2:
yield ScaleFree.class;
case 3:
yield Cyclic.class;
case 4:
yield RandomSingleFactorMim.class;
case 5:
yield RandomTwoFactorMim.class;
case 6:
yield SingleGraph.class;
default:
throw new IllegalArgumentException("Unexpected value: " + graphString);
};
}
/**
* Adds the dropdowns for selecting an independence test and a score to the given Box.
*
* @param vert1 the Box to which the dropdowns will be added
*/
private void addTestAndScoreDropdowns(Box vert1) {
IndependenceTestModels independenceTestModels = IndependenceTestModels.getInstance();
List models = independenceTestModels.getModels();
indTestComboBox = new JComboBox<>();
for (IndependenceTestModel model : models) {
indTestComboBox.addItem(model);
}
String lastIndependenceTest = model.getLastIndependenceTest();
for (int i = 0; i < indTestComboBox.getItemCount(); i++) {
IndependenceTestModel independenceTestModel = indTestComboBox.getItemAt(i);
if (independenceTestModel.getName().equals(lastIndependenceTest)) {
indTestComboBox.setSelectedIndex(i);
break;
}
}
Box horiz4 = Box.createHorizontalBox();
horiz4.add(new JLabel("Choose an independence test:"));
horiz4.add(Box.createHorizontalGlue());
horiz4.add(indTestComboBox);
vert1.add(horiz4);
ScoreModels scoreModels = ScoreModels.getInstance();
List scoreModelsList = scoreModels.getModels();
scoreModelComboBox = new JComboBox<>();
for (ScoreModel model : scoreModelsList) {
scoreModelComboBox.addItem(model);
}
String lastScore = model.getLastScore();
for (int i = 0; i < scoreModelComboBox.getItemCount(); i++) {
ScoreModel scoreModel = scoreModelComboBox.getItemAt(i);
if (scoreModel.getName().equals(lastScore)) {
scoreModelComboBox.setSelectedIndex(i);
break;
}
}
Box horiz5 = Box.createHorizontalBox();
horiz5.add(new JLabel("Choose a score:"));
horiz5.add(Box.createHorizontalGlue());
horiz5.add(scoreModelComboBox);
vert1.add(horiz5);
}
/**
* Adds a simulation tab to the provided JTabbedPane.
*
* @param tabbedPane the JTabbedPane to which the simulation tab is added
*/
private void addSimulationTab(JTabbedPane tabbedPane) {
simulationChoiceTextArea = new JTextArea();
simulationChoiceTextArea.setLineWrap(true);
simulationChoiceTextArea.setWrapStyleWord(true);
simulationChoiceTextArea.setEditable(false);
setSimulationText();
JPanel simulationChoice = new JPanel();
simulationChoice.setLayout(new BorderLayout());
simulationChoice.add(new JScrollPane(simulationChoiceTextArea, JScrollPane.VERTICAL_SCROLLBAR_AS_NEEDED, JScrollPane.HORIZONTAL_SCROLLBAR_AS_NEEDED), BorderLayout.CENTER);
Box simulationSelectionBox = Box.createHorizontalBox();
simulationSelectionBox.add(Box.createHorizontalGlue());
addSimulation = new JButton("Add Simulation");
addAddSimulationListener();
JButton removeLastSimulation = new JButton("Remove Last Simulation");
removeLastSimulation.addActionListener(e -> {
model.removeLastSimulation();
setSimulationText();
setComparisonText();
});
JButton editSimulationParameters = new JButton("Edit Parameters");
editSimulationParameters.addActionListener(e -> {
List simulations = model.getSelectedSimulations().getSimulations();
Set params = getAllSimulationParameters(simulations);
Box parameterBox = getParameterBox(params, true, false);
new PaddingPanel(parameterBox);
JDialog dialog = new JDialog(SwingUtilities.getWindowAncestor(this), "Edit Simulation Parameters", Dialog.ModalityType.APPLICATION_MODAL);
dialog.setLayout(new BorderLayout());
// Add your panel to the center of the dialog
dialog.add(parameterBox, BorderLayout.CENTER);
// Create a panel for the buttons
JPanel buttonPanel = betButtonPanel(dialog);
// Add the button panel to the bottom of the dialog
dialog.add(buttonPanel, BorderLayout.SOUTH);
dialog.pack(); // Adjust dialog size to fit its contents
dialog.setLocationRelativeTo(this); // Center dialog relative to the parent component
dialog.setVisible(true);
});
simulationSelectionBox.add(addSimulation);
simulationSelectionBox.add(removeLastSimulation);
simulationSelectionBox.add(editSimulationParameters);
simulationSelectionBox.add(Box.createHorizontalGlue());
simulationChoice.add(simulationSelectionBox, BorderLayout.SOUTH);
tabbedPane.addTab("Simulations", simulationChoice);
}
@NotNull
private JPanel betButtonPanel(JDialog dialog) {
JPanel buttonPanel = new JPanel(new FlowLayout(FlowLayout.CENTER));
JButton doneButton = new JButton("Done");
doneButton.addActionListener(e1 -> {
setSimulationText();
setComparisonText();
dialog.dispose();
SwingUtilities.invokeLater(() -> {
revalidate();
repaint();
});
});
buttonPanel.add(doneButton);
return buttonPanel;
}
// /**
// * Adds an XML tab to the provided JTabbedPane.
// *
// * @param tabbedPane the JTabbedPane to which the XML tab is added
// */
// private void addXmlTab(JTabbedPane tabbedPane) {
// JPanel xmlPanel = new JPanel();
// xmlPanel.setLayout(new BorderLayout());
// JTextArea xmlTextArea = new JTextArea();
// xmlTextArea.setLineWrap(false);
// xmlTextArea.setWrapStyleWord(false);
// xmlTextArea.setEditable(false);
// xmlTextArea.setFont(new Font("Monospaced", Font.PLAIN, 12));
// xmlTextArea.setText(getXmlText());
// xmlPanel.add(new JScrollPane(xmlTextArea, JScrollPane.VERTICAL_SCROLLBAR_AS_NEEDED, JScrollPane.HORIZONTAL_SCROLLBAR_AS_NEEDED), BorderLayout.CENTER);
//
// JButton loadXml = new JButton("Load XML");
// JButton saveXml = new JButton("Save XML");
//
// loadXml.addActionListener(e -> {
// JOptionPane.showMessageDialog(this, "This will load and XML file and parse it to set the" + " configuration of this tool.");
// setSimulationText();
// setAlgorithmText();
// setTableColumnsText();
// });
//
// saveXml.addActionListener(e -> {
// JOptionPane.showMessageDialog(this, "This will save the XML file shown in this panel.");
// setSimulationText();
// setAlgorithmText();
// setTableColumnsText();
// });
//
// Box xmlSelectionBox = Box.createHorizontalBox();
// xmlSelectionBox.add(Box.createHorizontalGlue());
// xmlSelectionBox.add(loadXml);
// xmlSelectionBox.add(saveXml);
// xmlSelectionBox.add(Box.createHorizontalGlue());
//
// xmlPanel.add(xmlSelectionBox, BorderLayout.SOUTH);
// tabbedPane.addTab("XML", xmlPanel);
// }
/**
* Adds an algorithm tab to the given JTabbedPane.
*
* @param tabbedPane the JTabbedPane to add the algorithm tab to
*/
private void addAlgorithmTab(JTabbedPane tabbedPane) {
algorithmChoiceTextArea = new JTextArea();
algorithmChoiceTextArea.setLineWrap(true);
algorithmChoiceTextArea.setWrapStyleWord(true);
algorithmChoiceTextArea.setEditable(false);
setAlgorithmText();
Box algorithSelectionBox = Box.createHorizontalBox();
algorithSelectionBox.add(Box.createHorizontalGlue());
addAlgorithm = new JButton("Add Algorithm");
addAddAlgorithmListener();
JButton removeLastAlgorithm = new JButton("Remove Last Algorithm");
removeLastAlgorithm.addActionListener(e -> {
model.removeLastAlgorithm();
setAlgorithmText();
setComparisonText();
});
JButton editAlgorithmParameters = new JButton("Edit Parameters");
editAlgorithmParameters.addActionListener(e -> {
List algorithms = model.getSelectedAlgorithms();
JTabbedPane tabbedPane1 = new JTabbedPane();
tabbedPane1.setTabPlacement(JTabbedPane.TOP);
Set allAlgorithmParameters = GridSearchModel.getAllAlgorithmParameters(algorithms);
Set allTestParameters = GridSearchModel.getAllTestParameters(algorithms);
Set allBootstrapParameters = GridSearchModel.getAllBootstrapParameters(algorithms);
Set allScoreParameters = GridSearchModel.getAllScoreParameters(algorithms);
if (allAlgorithmParameters.isEmpty() && allTestParameters.isEmpty() && allBootstrapParameters.isEmpty()
&& allScoreParameters.isEmpty()) {
JLabel noParamLbl = NO_PARAM_LBL;
noParamLbl.setBorder(new EmptyBorder(10, 10, 10, 10));
tabbedPane1.addTab("No Parameters", new PaddingPanel(noParamLbl));
}
tabbedPane1.addTab("Algorithm", new PaddingPanel(getParameterBox(allAlgorithmParameters, true, true)));
tabbedPane1.addTab("Test", new PaddingPanel(getParameterBox(allTestParameters, true, true)));
tabbedPane1.addTab("Score", new PaddingPanel(getParameterBox(allScoreParameters, true, true)));
tabbedPane1.addTab("Bootstrapping", new PaddingPanel(getParameterBox(allBootstrapParameters, false, false)));
JDialog dialog = new JDialog(SwingUtilities.getWindowAncestor(this), "Edit Algorithm Parameters", Dialog.ModalityType.APPLICATION_MODAL);
dialog.setLayout(new BorderLayout());
// Add your panel to the center of the dialog
dialog.add(tabbedPane1, BorderLayout.CENTER);
// Create a panel for the buttons
JPanel buttonPanel = new JPanel(new FlowLayout(FlowLayout.CENTER));
JButton doneButton = new JButton("Done");
doneButton.addActionListener(e1 -> {
setAlgorithmText();
setComparisonText();
dialog.dispose();
});
buttonPanel.add(doneButton);
// Add the button panel to the bottom of the dialog
dialog.add(buttonPanel, BorderLayout.SOUTH);
dialog.pack(); // Adjust dialog size to fit its contents
dialog.setLocationRelativeTo(this); // Center dialog relative to the parent component
dialog.setVisible(true);
});
algorithSelectionBox.add(addAlgorithm);
algorithSelectionBox.add(removeLastAlgorithm);
algorithSelectionBox.add(editAlgorithmParameters);
algorithSelectionBox.add(Box.createHorizontalGlue());
JPanel algorithmChoice = new JPanel();
algorithmChoice.setLayout(new BorderLayout());
algorithmChoice.add(new JScrollPane(algorithmChoiceTextArea, JScrollPane.VERTICAL_SCROLLBAR_ALWAYS, JScrollPane.HORIZONTAL_SCROLLBAR_ALWAYS), BorderLayout.CENTER);
algorithmChoice.add(algorithSelectionBox, BorderLayout.SOUTH);
tabbedPane.addTab("Algorithms", algorithmChoice);
}
@NotNull
private Box getParameterBox(Set params, boolean listOptionAllowed, boolean bothOptionAllowed) {
Box parameterBox = Box.createVerticalBox();
parameterBox.removeAll();
if (params.isEmpty()) {
JLabel noParamLbl = NO_PARAM_LBL;
noParamLbl.setBorder(new EmptyBorder(10, 10, 10, 10));
parameterBox.add(noParamLbl, BorderLayout.NORTH);
} else {
Box parameters = Box.createVerticalBox();
Box[] paramBoxes = ParameterComponents.toArray(
createParameterComponents(params, model.getParameters(), listOptionAllowed, bothOptionAllowed));
int lastIndex = paramBoxes.length - 1;
for (int i = 0; i < lastIndex; i++) {
parameters.add(paramBoxes[i]);
parameters.add(Box.createVerticalStrut(10));
}
parameters.add(paramBoxes[lastIndex]);
Box horiz = Box.createHorizontalBox();
if (listOptionAllowed) {
horiz.add(new JLabel("Please type comma-separated lists of values, thus: 10, 100, 1000"));
} else {
horiz.add(new JLabel("Please type a single value."));
}
horiz.add(Box.createHorizontalGlue());
horiz.setBorder(new EmptyBorder(0, 0, 10, 0));
parameterBox.add(horiz, BorderLayout.NORTH);
parameterBox.add(new JScrollPane(new PaddingPanel(parameters)), BorderLayout.CENTER);
parameterBox.setBorder(new EmptyBorder(10, 10, 10, 10));
parameterBox.setPreferredSize(new Dimension(800, 400));
}
return parameterBox;
}
/**
* Adds a table columns tab to the provided JTabbedPane.
*
* @param tabbedPane the JTabbedPane to add the table columns tab to
*/
private void addTableColumnsTab(JTabbedPane tabbedPane) {
tableColumnsChoiceTextArea = new JTextArea();
tableColumnsChoiceTextArea.setLineWrap(true);
tableColumnsChoiceTextArea.setWrapStyleWord(true);
tableColumnsChoiceTextArea.setEditable(false);
setTableColumnsText();
Box tableColumnsSelectionBox = Box.createHorizontalBox();
tableColumnsSelectionBox.add(Box.createHorizontalGlue());
addTableColumns = new JButton("Add Table Column(s)");
addAddTableColumnsListener(tabbedPane);
JButton removeLastTableColumn = new JButton("Remove Last Column");
removeLastTableColumn.addActionListener(e -> {
model.removeLastTableColumn();
setTableColumnsText();
setComparisonText();
});
tableColumnsSelectionBox.add(addTableColumns);
tableColumnsSelectionBox.add(removeLastTableColumn);
// tableColumnsSelectionBox.add(createEditutilitiesButton());
tableColumnsSelectionBox.add(Box.createHorizontalGlue());
JPanel tableColumnsChoice = new JPanel();
tableColumnsChoice.setLayout(new BorderLayout());
tableColumnsChoice.add(new JScrollPane(tableColumnsChoiceTextArea, JScrollPane.VERTICAL_SCROLLBAR_AS_NEEDED,
JScrollPane.HORIZONTAL_SCROLLBAR_AS_NEEDED), BorderLayout.CENTER);
tableColumnsChoice.add(tableColumnsSelectionBox, BorderLayout.SOUTH);
tabbedPane.addTab("Table Columns", tableColumnsChoice);
}
private @NotNull JButton createEditutilitiesButton() {
JButton editUtilities = new JButton("Edit Utilities");
editUtilities.addActionListener(e -> {
List columns = model.getSelectedTableColumns();
Set params = new HashSet<>();
for (GridSearchModel.MyTableColumn column : columns) {
params.add("algcomparison." + column.getColumnName());
Double weight = model.getParameters().getDouble("algcomparison." + column.getColumnName());
Parameters.serializableInstance().remove("algcomparison." + column.getColumnName());
ParamDescriptions.getInstance().put("algcomparison." + column.getColumnName(),
new ParamDescription("algcomparison." + column.getColumnName(),
"Utility for " + column.getColumnName() + " in [0, 1]",
"Utility for " + column.getColumnName(),
weight, 0.0, 1.0));
}
Box parameterBox = getParameterBox(params, false, false);
new PaddingPanel(parameterBox);
JDialog dialog = new JDialog(SwingUtilities.getWindowAncestor(GridSearchEditor.this), "Edit Utilities", Dialog.ModalityType.APPLICATION_MODAL);
dialog.setLayout(new BorderLayout());
JLabel label = new JLabel("To sort comparison tables by utility please adjust parameters in Comparison.");
label.setBorder(new EmptyBorder(10, 10, 10, 10));
dialog.add(label, BorderLayout.NORTH);
// Add your panel to the center of the dialog
dialog.add(parameterBox, BorderLayout.CENTER);
// Create a panel for the buttons
JPanel buttonPanel = betButtonPanel(dialog);
// Add the button panel to the bottom of the dialog
dialog.add(buttonPanel, BorderLayout.SOUTH);
dialog.pack(); // Adjust dialog size to fit its contents
dialog.setLocationRelativeTo(GridSearchEditor.this); // Center dialog relative to the parent component
dialog.setVisible(true);
});
return editUtilities;
}
/**
* Adds a comparison tab to the given JTabbedPane.
*
* @param tabbedPane the JTabbedPane to add the comparison tab to
*/
private void addComparisonTab(JTabbedPane tabbedPane) {
comparisonTextArea = new JTextArea();
comparisonTextArea.setLineWrap(false);
comparisonTextArea.setWrapStyleWord(false);
comparisonTextArea.setEditable(false);
comparisonTextArea.setFont(new Font("Monospaced", Font.PLAIN, 12));
verboseOutputTextArea = new JTextArea();
verboseOutputTextArea.setLineWrap(true);
verboseOutputTextArea.setWrapStyleWord(true);
verboseOutputTextArea.setEditable(false);
verboseOutputTextArea.setFont(new Font("Monospaced", Font.PLAIN, 12));
setComparisonText();
JButton runComparison = runComparisonButton();
JButton setComparisonParameters = new JButton("Edit Parameters");
setComparisonParameters.addActionListener(e -> {
Box parameterBox = Box.createVerticalBox();
Box horiz1 = Box.createHorizontalBox();
horiz1.add(new JLabel("Save Data:"));
horiz1.add(Box.createHorizontalGlue());
horiz1.add(getBooleanSelectionBox("algcomparisonSaveData", model.getParameters(), false));
Box horiz2 = Box.createHorizontalBox();
horiz2.add(new JLabel("Save Graphs:"));
horiz2.add(Box.createHorizontalGlue());
horiz2.add(getBooleanSelectionBox("algcomparisonSaveGraphs", model.getParameters(), false));
Box horiz2b = Box.createHorizontalBox();
horiz2b.add(new JLabel("Save CPDAGs:"));
horiz2b.add(Box.createHorizontalGlue());
horiz2b.add(getBooleanSelectionBox("algcomparisonSaveCPDAGs", model.getParameters(), false));
Box horiz2c = Box.createHorizontalBox();
horiz2c.add(new JLabel("Save PAGs:"));
horiz2c.add(Box.createHorizontalGlue());
horiz2c.add(getBooleanSelectionBox("algcomparisonSavePAGs", model.getParameters(), false));
Box horiz4a = Box.createHorizontalBox();
horiz4a.add(new JLabel("Sort by Utility:"));
horiz4a.add(Box.createHorizontalGlue());
horiz4a.add(getBooleanSelectionBox("algcomparisonSortByUtility", model.getParameters(), false));
Box horiz4b = Box.createHorizontalBox();
horiz4b.add(new JLabel("Show Utilities:"));
horiz4b.add(Box.createHorizontalGlue());
horiz4b.add(getBooleanSelectionBox("algcomparisonShowUtilities", model.getParameters(), false));
Box horiz4c = Box.createHorizontalBox();
horiz4c.add(new JLabel("Set Knowledge on Algorithm If Available:"));
horiz4c.add(Box.createHorizontalGlue());
horiz4c.add(getBooleanSelectionBox("algcomparisonSetAlgorithmKnowledge", model.getParameters(), false));
Box horiz5 = Box.createHorizontalBox();
horiz5.add(new JLabel("Parallelism:"));
horiz5.add(Box.createHorizontalGlue());
horiz5.add(getIntTextField("algcomparisonParallelism", model.getParameters(),
model.getParameters().getInt("algcomparisonParallelism", Runtime.getRuntime().availableProcessors()),
1, 1000));
Box horiz6 = Box.createHorizontalBox();
horiz6.add(new JLabel("Comparison Graph Type:"));
horiz6.add(Box.createHorizontalGlue());
JComboBox comparisonGraphTypeComboBox = new JComboBox<>();
for (GridSearchModel.ComparisonGraphType comparisonGraphType : GridSearchModel.ComparisonGraphType.values()) {
comparisonGraphTypeComboBox.addItem(comparisonGraphType.toString());
}
comparisonGraphTypeComboBox.setSelectedItem(model.getParameters().getString("algcomparisonGraphType"));
comparisonGraphTypeComboBox.addActionListener(e1 -> {
String selectedItem = (String) comparisonGraphTypeComboBox.getSelectedItem();
// ComparisonGraphType comparisonGraphType1 = ComparisonGraphType.valueOf(selectedItem);
model.getParameters().set("algcomparisonGraphType", selectedItem);
});
horiz6.add(comparisonGraphTypeComboBox);
parameterBox.add(horiz1);
parameterBox.add(horiz2);
parameterBox.add(horiz2b);
parameterBox.add(horiz2c);
parameterBox.add(horiz4a);
parameterBox.add(horiz4b);
parameterBox.add(horiz4c);
parameterBox.add(horiz5);
parameterBox.add(horiz6);
parameterBox.setBorder(new EmptyBorder(10, 10, 10, 10));
JDialog dialog = new JDialog(SwingUtilities.getWindowAncestor(this), "Edit Comparison Parameters", Dialog.ModalityType.APPLICATION_MODAL);
dialog.setLayout(new BorderLayout());
// Add your panel to the center of the dialog
dialog.add(parameterBox, BorderLayout.CENTER);
// Create a panel for the buttons
JPanel buttonPanel = new JPanel(new FlowLayout(FlowLayout.CENTER));
JButton doneButton = new JButton("Done");
doneButton.addActionListener(e1 -> {
SwingUtilities.invokeLater(dialog::dispose);
setComparisonText();
});
buttonPanel.add(doneButton);
// Add the button panel to the bottom of the dialog
dialog.add(buttonPanel, BorderLayout.SOUTH);
dialog.pack(); // Adjust dialog size to fit its contents
dialog.setLocationRelativeTo(this); // Center dialog relative to the parent component
dialog.setVisible(true);
});
Box comparisonSelectionBox = Box.createHorizontalBox();
comparisonSelectionBox.add(Box.createHorizontalGlue());
comparisonSelectionBox.add(runComparison);
comparisonSelectionBox.add(createEditutilitiesButton());
comparisonSelectionBox.add(setComparisonParameters);
comparisonSelectionBox.add(Box.createHorizontalGlue());
comparisonTabbedPane = new JTabbedPane();
comparisonScroll = new JScrollPane(comparisonTextArea, JScrollPane.VERTICAL_SCROLLBAR_AS_NEEDED, JScrollPane.HORIZONTAL_SCROLLBAR_AS_NEEDED);
comparisonTabbedPane.addTab("Comparison", comparisonScroll);
comparisonTabbedPane.addTab("Verbose Output", new JScrollPane(verboseOutputTextArea, JScrollPane.VERTICAL_SCROLLBAR_AS_NEEDED, JScrollPane.HORIZONTAL_SCROLLBAR_AS_NEEDED));
// comparisonTabbedPane.addTab("Graphs", getGraphSelectorBox());
//
// comparisonTabbedPane.addChangeListener(new ChangeListener() {
// public void stateChanged(ChangeEvent e) {
// JTabbedPane sourceTabbedPane = (JTabbedPane) e.getSource();
// refreshGraphSelectionContent(sourceTabbedPane);
// }
// });
JPanel comparisonPanel = new JPanel();
comparisonPanel.setLayout(new BorderLayout());
comparisonPanel.add(comparisonTabbedPane, BorderLayout.CENTER);
comparisonPanel.add(comparisonSelectionBox, BorderLayout.SOUTH);
tabbedPane.addTab("Comparison", comparisonPanel);
tabbedPane.addTab("View Graphs", getGraphSelectorBox());
tabbedPane.addChangeListener(e -> {
JTabbedPane sourceTabbedPane = (JTabbedPane) e.getSource();
refreshGraphSelectionContent(sourceTabbedPane);
});
}
/**
* Returns a Box component with selectors for simulation, algorithm, and graph index.
*
* @return a Box component with selectors for simulation, algorithm, and graph index
*/
private @NotNull Box getGraphSelectorBox() {
String resultsPath = model.getResultsPath();
File resultsDir = new File(resultsPath, "results");
List simulationIndices = new ArrayList<>();
if (resultsDir.exists()) {
File[] dirs = resultsDir.listFiles();
if (dirs != null) {
// The dirs array should contain directories for each simulation/algorithm combination. These
// are formatted as, e.g., "5.2" for simulation5 and algorithm 2. We need to iterate through
// all of these directories and find the highest simulation number and the highest
// algorithm number. The number of graphs will be determined once we have the simulation and
// algorithm numbers. These are listed as "graph1.txt", "graph2.txt", etc., in each of these
// directories.
for (File dir : dirs) {
String name = dir.getName();
String[] parts = name.split("\\.");
int simulation;
int algorithm;
try {
simulation = Integer.parseInt(parts[0]);
algorithm = Integer.parseInt(parts[1]);
if (!simulationIndices.contains(simulation)) {
simulationIndices.add(simulation);
}
} catch (NumberFormatException e) {
// These aren't directories/files written out by the tool.
}
}
}
}
Collections.sort(simulationIndices);
Box graphSelectorBox = Box.createVerticalBox();
Box instructions = Box.createHorizontalBox();
instructions.add(new JLabel("Select the simulation, algorithm, and graph index to view, from the comparison table:"));
instructions.add(Box.createHorizontalGlue());
graphSelectorBox.add(Box.createVerticalStrut(4));
graphSelectorBox.add(instructions);
graphSelectorBox.add(Box.createVerticalStrut(4));
Box selectors = Box.createHorizontalBox();
JComboBox simulationComboBox = new JComboBox<>();
JComboBox algorithmComboBox = new JComboBox<>();
JComboBox graphIndexComboBox = new JComboBox<>();
for (int i : simulationIndices) {
simulationComboBox.addItem(i);
}
if (model.getSelectedSimulation() > 0) {
simulationComboBox.setSelectedItem(model.getSelectedSimulation());
}
updateAlgorithmBoxIndices(simulationComboBox, algorithmComboBox, graphIndexComboBox, resultsDir);
updateGraphBoxIndices(simulationComboBox, algorithmComboBox, graphIndexComboBox, resultsDir);
if (model.getSelectedGraphIndex() > 0) {
graphIndexComboBox.setSelectedItem(model.getSelectedGraphIndex());
}
selectors.add(new JLabel("Simulation:"));
selectors.add(simulationComboBox);
selectors.add(new JLabel("Algorithm:"));
selectors.add(algorithmComboBox);
selectors.add(new JLabel("Graph Index:"));
selectors.add(graphIndexComboBox);
graphSelectorBox.add(selectors);
graphSelectorBox.add(Box.createVerticalStrut(4));
GraphWorkbench workbench = new GraphWorkbench();
workbench.setGraph(new EdgeListGraph());
graphSelectorBox.add(new JScrollPane(workbench));
// Add listeners to the algorithm and simulation combo boxes to update the graph index combo box
// when the algorithm or simulation is changed.
simulationComboBox.addActionListener(e -> {
updateAlgorithmBoxIndices(simulationComboBox, algorithmComboBox, graphIndexComboBox, resultsDir);
updateGraphBoxIndices(simulationComboBox, algorithmComboBox, graphIndexComboBox, resultsDir);
updateSelectedGraph(simulationComboBox, algorithmComboBox, graphIndexComboBox, resultsDir, workbench);
});
algorithmComboBox.addActionListener(e -> {
updateGraphBoxIndices(simulationComboBox, algorithmComboBox, graphIndexComboBox, resultsDir);
updateSelectedGraph(simulationComboBox, algorithmComboBox, graphIndexComboBox, resultsDir, workbench);
});
graphIndexComboBox.addActionListener(e -> {
updateSelectedGraph(simulationComboBox, algorithmComboBox, graphIndexComboBox, resultsDir, workbench);
});
updateAlgorithmBoxIndices(simulationComboBox, algorithmComboBox, graphIndexComboBox, resultsDir);
updateGraphBoxIndices(simulationComboBox, algorithmComboBox, graphIndexComboBox, resultsDir);
updateSelectedGraph(simulationComboBox, algorithmComboBox, graphIndexComboBox, resultsDir, workbench);
return graphSelectorBox;
}
@NotNull
private JButton runComparisonButton() {
JButton runComparison = new JButton("Run Comparison");
runComparison.addActionListener(e -> {
class MyWatchedProcess extends WatchedProcess {
public void watch() {
SwingUtilities.invokeLater(() -> comparisonTabbedPane.setSelectedIndex(1));
ByteArrayOutputStream baos = new BufferedListeningByteArrayOutputStream();
java.io.PrintStream ps1 = new java.io.PrintStream(baos);
verboseOutputTextArea.setText("");
TextAreaOutputStream baos2 = new TextAreaOutputStream(verboseOutputTextArea);
PrintStream ps2 = new PrintStream(baos2);
TetradLogger.getInstance().addOutputStream(baos2);
try {
model.runComparison(ps1, ps2);
String resultsPath = model.getResultsPath();
if (resultsPath != null && simulationChoiceTextArea != null) {
// Write contents of simulation text area to a file at resultsPath + "/simulation.txt"
try (PrintWriter writer = new PrintWriter(resultsPath + "/simulation.txt")) {
writer.println(simulationChoiceTextArea.getText());
} catch (FileNotFoundException ex) {
throw new RuntimeException(ex);
}
// Write contents of algorithm text area to a file at resultsPath + "/algorithm.txt"
try (PrintWriter writer = new PrintWriter(resultsPath + "/algorithm.txt")) {
writer.println(algorithmChoiceTextArea.getText());
} catch (FileNotFoundException ex) {
throw new RuntimeException(ex);
}
// Write contents of table columns text area to a file at resultsPath + "/tableColumns.txt"
try (PrintWriter writer = new PrintWriter(resultsPath + "/tableColumns.txt")) {
writer.println(tableColumnsChoiceTextArea.getText());
} catch (FileNotFoundException ex) {
throw new RuntimeException(ex);
}
// Write contents of verbose output text area to a file at resultsPath + "/verboseOutput.txt"
try (PrintWriter writer = new PrintWriter(resultsPath + "/verboseOutput.txt")) {
writer.println(verboseOutputTextArea.getText());
} catch (FileNotFoundException ex) {
throw new RuntimeException(ex);
}
}
} catch (Exception ex) {
throw new RuntimeException(ex);
}
ps1.flush();
comparisonTextArea.setText(baos.toString());
TetradLogger.getInstance().removeOutputStream(baos2);
SwingUtilities.invokeLater(() -> {
try {
scrollToWord(comparisonTextArea, comparisonScroll, "AVERAGE VALUE");
} catch (BadLocationException ex) {
System.out.println("Scrolling operation failed.");
}
});
SwingUtilities.invokeLater(() -> comparisonTabbedPane.setSelectedIndex(0));
if (comparisonTextArea != null) {
model.setLastComparisonText(comparisonTextArea.getText());
}
if (verboseOutputTextArea != null) {
model.setLastVerboseOutputText(verboseOutputTextArea.getText());
}
}
}
new MyWatchedProcess();
});
return runComparison;
}
/**
* Adds a help tab to the provided JTabbedPane.
*
* @param tabbedPane the JTabbedPane to which the help tab is added
*/
private void addHelpTab(JTabbedPane tabbedPane) {
JPanel helpChoice = new JPanel();
helpChoice.setLayout(new BorderLayout());
helpChoiceTextArea = new JTextArea();
helpChoiceTextArea.setLineWrap(true);
helpChoiceTextArea.setWrapStyleWord(true);
helpChoiceTextArea.setEditable(false);
setHelpText();
helpChoice.add(helpChoiceTextArea, BorderLayout.CENTER);
Box helpSelectionBox = Box.createHorizontalBox();
helpSelectionBox.add(Box.createHorizontalGlue());
helpSelectionBox.add(Box.createHorizontalGlue());
JPanel helpPanel = new JPanel();
helpPanel.setLayout(new BorderLayout());
helpPanel.add(new JScrollPane(helpChoiceTextArea, JScrollPane.VERTICAL_SCROLLBAR_AS_NEEDED, JScrollPane.HORIZONTAL_SCROLLBAR_AS_NEEDED), BorderLayout.CENTER);
helpPanel.add(helpSelectionBox, BorderLayout.SOUTH);
tabbedPane.addTab("Help", helpPanel);
}
/**
* Adds a listener to the "Add Simulation" button. When the button is clicked, a dialog is displayed to choose the
* graph type and simulation type. If the "Add" button in the dialog is clicked, the selected graph and simulation
* types are used to create a simulation and add it to the model.
*/
private void addAddSimulationListener() {
addSimulation.addActionListener(e -> {
JPanel panel = new JPanel();
panel.setLayout(new BorderLayout());
Box vert1 = Box.createVerticalBox();
Box horiz2 = Box.createHorizontalBox();
horiz2.add(new JLabel("Choose a graph type:"));
horiz2.add(Box.createHorizontalGlue());
JComboBox graphsDropdown = getGraphsDropdown();
Arrays.stream(ParameterTab.GRAPH_TYPE_ITEMS).forEach(graphsDropdown::addItem);
if (model.getSuppliedGraph() != null) {
graphsDropdown.addItem("User Supplied Graph");
}
graphsDropdown.setMaximumSize(graphsDropdown.getPreferredSize());
graphsDropdown.setSelectedItem(model.getLastGraphChoice());
horiz2.add(graphsDropdown);
vert1.add(horiz2);
Box horiz3 = Box.createHorizontalBox();
horiz3.add(new JLabel("Choose a simulation type:"));
horiz3.add(Box.createHorizontalGlue());
JComboBox simulationsDropdown = new JComboBox<>();
Arrays.stream(ParameterTab.MODEL_TYPE_ITEMS).forEach(simulationsDropdown::addItem);
simulationsDropdown.setMaximumSize(simulationsDropdown.getPreferredSize());
simulationsDropdown.setSelectedItem(model.getLastSimulationChoice());
simulationsDropdown.addActionListener(e1 -> {
String selectedItem = (String) simulationsDropdown.getSelectedItem();
if (selectedItem != null) {
model.setLastSimulationChoice(selectedItem);
}
});
horiz3.add(simulationsDropdown);
vert1.add(horiz3);
panel.add(vert1, BorderLayout.NORTH);
// Create the JDialog. Use the parent frame to make it modal.
JDialog dialog = new JDialog(SwingUtilities.getWindowAncestor(this), "Add Simulation", Dialog.ModalityType.APPLICATION_MODAL);
dialog.setLayout(new BorderLayout());
// Add your panel to the center of the dialog
dialog.add(panel, BorderLayout.CENTER);
// Create a panel for the buttons
JPanel buttonPanel = getButtonPanel(graphsDropdown, simulationsDropdown, dialog);
// Add the button panel to the bottom of the dialog
dialog.add(buttonPanel, BorderLayout.SOUTH);
// Set the dialog size, position, and visibility
dialog.pack(); // Adjust dialog size to fit its contents
dialog.setLocationRelativeTo(this); // Center dialog relative to the parent component
dialog.setVisible(true);
});
}
@NotNull
private JComboBox getGraphsDropdown() {
JComboBox graphsDropdown = new JComboBox<>();
String lastGraphChoice = model.getLastGraphChoice();
if (lastGraphChoice != null) {
graphsDropdown.addItem(lastGraphChoice);
}
graphsDropdown.addActionListener(e1 -> {
String selectedItem = (String) graphsDropdown.getSelectedItem();
if (selectedItem != null) {
model.setLastGraphChoice(selectedItem);
}
});
return graphsDropdown;
}
@NotNull
private JPanel getButtonPanel(JComboBox graphsDropdown, JComboBox simulationsDropdown, JDialog dialog) {
JPanel buttonPanel = new JPanel(new FlowLayout(FlowLayout.RIGHT));
JButton addButton = new JButton("Add");
JButton cancelButton = new JButton("Cancel");
// Add action listeners for the buttons
addButton.addActionListener(e1 -> {
String graphString = (String) graphsDropdown.getSelectedItem();
String simulationString = (String) simulationsDropdown.getSelectedItem();
Class graphClazz = getGraphClazz(graphString);
List simulationTypeStrings = Arrays.asList(ParameterTab.MODEL_TYPE_ITEMS);
Class simulationClass = switch (simulationTypeStrings.indexOf(simulationString)) {
case 0:
yield edu.cmu.tetrad.algcomparison.simulation.BayesNetSimulation.class;
case 1:
yield edu.cmu.tetrad.algcomparison.simulation.SemSimulation.class;
case 2:
yield edu.cmu.tetrad.algcomparison.simulation.LinearFisherModel.class;
case 3:
yield edu.cmu.tetrad.algcomparison.simulation.NLSemSimulation.class;
case 4:
yield edu.cmu.tetrad.algcomparison.simulation.LeeHastieSimulation.class;
case 5:
yield edu.cmu.tetrad.algcomparison.simulation.ConditionalGaussianSimulation.class;
case 6:
yield edu.cmu.tetrad.algcomparison.simulation.TimeSeriesSemSimulation.class;
default:
throw new IllegalArgumentException("Unexpected value: " + simulationString);
};
GridSearchModel.SimulationSpec spec = new GridSearchModel.SimulationSpec("name", graphClazz, simulationClass);
model.addSimulationSpec(spec);
setComparisonText();
setSimulationText();
dialog.dispose(); // Close the dialog
});
cancelButton.addActionListener(e12 -> {
dialog.dispose(); // Close the dialog
});
// Add the buttons to the button panel
buttonPanel.add(addButton);
buttonPanel.add(cancelButton);
return buttonPanel;
}
/**
* Adds an ActionListener to the addAlgorithm button
*/
private void addAddAlgorithmListener() {
addAlgorithm.addActionListener(e -> {
AlgorithmModels algorithmModels = AlgorithmModels.getInstance();
List algorithmModels1 = algorithmModels.getModels(DataType.Continuous, false);
JPanel panel = new JPanel();
panel.setLayout(new BorderLayout());
algorithmDropdown = new JComboBox<>();
for (AlgorithmModel model : algorithmModels1) {
algorithmDropdown.addItem(model);
}
String lastAlgorithmChoice = model.getLastAlgorithmChoice();
for (int i = 0; i < algorithmDropdown.getItemCount(); i++) {
AlgorithmModel itemAt = (AlgorithmModel) algorithmDropdown.getItemAt(i);
if (itemAt.getName().equals(lastAlgorithmChoice)) {
algorithmDropdown.setSelectedIndex(i);
break;
}
}
algorithmDropdown.addActionListener(e1 -> {
AlgorithmModel selectedItem = (AlgorithmModel) algorithmDropdown.getSelectedItem();
if (selectedItem != null) {
model.setLastAlgorithmChoice(selectedItem.getName());
}
});
Box vert1 = Box.createVerticalBox();
Box horiz2 = Box.createHorizontalBox();
horiz2.add(new JLabel("Choose an algorithm:"));
horiz2.add(Box.createHorizontalGlue());
horiz2.add(algorithmDropdown);
vert1.add(horiz2);
addTestAndScoreDropdowns(vert1);
panel.add(vert1, BorderLayout.NORTH);
// Create the JDialog. Use the parent frame to make it modal.
JDialog dialog = new JDialog(SwingUtilities.getWindowAncestor(this), "Add Simulation", Dialog.ModalityType.APPLICATION_MODAL);
dialog.setLayout(new BorderLayout());
dialog.add(panel, BorderLayout.CENTER);
// Create a panel for the buttons
JPanel buttonPanel = getAddButton(dialog);
// Add the button panel to the bottom of the dialog
dialog.add(buttonPanel, BorderLayout.SOUTH);
// Set the dialog size, position, and visibility
dialog.pack(); // Adjust dialog size to fit its contents
dialog.setLocationRelativeTo(this); // Center dialog relative to the parent component
dialog.setVisible(true);
});
}
@NotNull
private JPanel getAddButton(JDialog dialog) {
JPanel buttonPanel = new JPanel(new FlowLayout(FlowLayout.RIGHT));
JButton addButton = new JButton("Add");
JButton cancelButton = new JButton("Cancel");
// Add action listeners for the buttons
addButton.addActionListener(e1 -> {
AlgorithmModel algorithmModel = (AlgorithmModel) algorithmDropdown.getSelectedItem();
if (algorithmModel == null) {
return;
}
Class algorithm = algorithmModel.getAlgorithm().clazz();
IndependenceTestModel testModel = (IndependenceTestModel) indTestComboBox.getSelectedItem();
AnnotatedClass test = null;
if (testModel != null) {
test = testModel.getIndependenceTest();
}
ScoreModel scoreModel = (ScoreModel) scoreModelComboBox.getSelectedItem();
AnnotatedClass score = null;
if (scoreModel != null) {
score = scoreModel.getScore();
}
IndependenceWrapper independenceWrapper = null;
ScoreWrapper scoreWrapper = null;
try {
if (test != null) {
independenceWrapper = (IndependenceWrapper) test.clazz().getConstructor().newInstance();
}
if (score != null) {
scoreWrapper = (ScoreWrapper) score.clazz().getConstructor().newInstance();
}
if (testModel != null) {
model.setLastIndependenceTest(testModel.getName());
}
if (scoreModel != null) {
model.setLastScore(scoreModel.getName());
}
} catch (InstantiationException | IllegalAccessException | InvocationTargetException |
NoSuchMethodException ex) {
throw new RuntimeException(ex);
}
try {
Algorithm algorithmImpl = (Algorithm) algorithm.getConstructor().newInstance();
if (algorithmImpl instanceof TakesIndependenceWrapper && independenceWrapper != null) {
((TakesIndependenceWrapper) algorithmImpl).setIndependenceWrapper(independenceWrapper);
}
if (algorithmImpl instanceof UsesScoreWrapper && scoreWrapper != null) {
((UsesScoreWrapper) algorithmImpl).setScoreWrapper(scoreWrapper);
}
model.addAlgorithm(new GridSearchModel.AlgorithmSpec("name", algorithmModel, test, score));
} catch (InstantiationException | IllegalAccessException | InvocationTargetException |
NoSuchMethodException ex) {
throw new RuntimeException(ex);
}
setAlgorithmText();
setComparisonText();
dialog.dispose();
});
cancelButton.addActionListener(e12 -> {
dialog.dispose(); // Close the dialog
});
// Add the buttons to the button panel
buttonPanel.add(addButton);
buttonPanel.add(cancelButton);
return buttonPanel;
}
/**
* Adds an action listener to the addTableColumns button. This action listener creates a table model, adds a JTable
* to the panel, and sets a row sorter for the table based on user input. It also creates a text field with a
* document listener to filter the table based on user input.
*/
private void addAddTableColumnsListener(JTabbedPane tabbedPane) {
addTableColumns.addActionListener(e -> {
java.util.Set selectedColumns = new HashSet<>();
List allTableColumns = model.getAllTableColumns();
// Create a table idaCheckEst for the results of the IDA check
TableColumnSelectionModel columnSelectionTableModel = new TableColumnSelectionModel(allTableColumns, selectedColumns);
this.setLayout(new BorderLayout());
// Add the table to the left
JTable table = new JTable(columnSelectionTableModel);
NumberFormat numberFormat = NumberFormatUtil.getInstance().getNumberFormat();
IdaEditor.NumberFormatRenderer numberRenderer = new IdaEditor.NumberFormatRenderer(numberFormat);
table.setDefaultRenderer(Double.class, numberRenderer);
table.setAutoCreateRowSorter(true);
table.setFillsViewportHeight(true);
((TableColumnSelectionModel) table.getModel()).setTableRef(table); // Set the table reference
DefaultTableCellRenderer centerRenderer = new DefaultTableCellRenderer();
centerRenderer.setHorizontalAlignment(JLabel.CENTER);
TableColumnModel columnModel = table.getColumnModel();
columnModel.getColumn(0).setCellRenderer(centerRenderer);
this.add(new JScrollPane(table));
// Create a TableRowSorter and set it to the JTable
TableRowSorter sorter = new TableRowSorter<>(columnSelectionTableModel);
table.setRowSorter(sorter);
// sorter.addRowSorterListener(e2 -> {
//
// if (e2.getType() == RowSorterEvent.Type.SORTED) {
// List visiblePairs = new ArrayList<>();
// int rowCount = table.getRowCount();
//
// for (int i = 0; i < rowCount; i++) {
// int modelIndex = table.convertRowIndexToModel(i);
// visiblePairs.add(allTableColumns.get(modelIndex));
// }
// }
// });
// Create the text field
JLabel label = new JLabel("Regexes (semicolon separated):");
JTextField filterText = new JTextField(15);
filterText.setMaximumSize(new Dimension(500, 20));
label.setLabelFor(filterText);
// Create a listener for the text field that will update the table's row sort
filterText.getDocument().addDocumentListener(new DocumentListener() {
/**
* Filters the table based on the text in the text field.
*/
private void filter() {
String text = filterText.getText();
if (text.trim().isEmpty()) {
sorter.setRowFilter(null);
} else {
String[] textParts = text.split(";");
List> filters = new ArrayList<>(textParts.length);
for (String part : textParts) {
try {
String trim = part.trim();
filters.add(RowFilter.regexFilter(trim));
} catch (PatternSyntaxException e) {
// ignore
}
}
sorter.setRowFilter(RowFilter.orFilter(filters));
}
}
/**
* Inserts text into the text field.
*
* @param e the document event.
*/
@Override
public void insertUpdate(DocumentEvent e) {
filter();
}
/**
* Removes text from the text field.
*
* @param e the document event.
*/
@Override
public void removeUpdate(DocumentEvent e) {
filter();
}
/**
* Changes text in the text field.
*
* @param e the document event.
*/
@Override
public void changedUpdate(DocumentEvent e) {
// this method won't be called for plain text fields
}
});
JPanel panel = new JPanel();
panel.setLayout(new BorderLayout());
Box vert1 = Box.createVerticalBox();
vert1.add(new JLabel("Choose new columns for your tables:"));
Box horiz2 = Box.createHorizontalBox();
horiz2.add(label);
horiz2.add(filterText);
vert1.add(horiz2);
vert1.add(Box.createVerticalStrut(5));
vert1.add(new JScrollPane(table));
vert1.add(Box.createVerticalStrut(10));
Box horiz3 = Box.createHorizontalBox();
horiz3.add(Box.createHorizontalGlue());
JButton selectUsedParameters = new JButton("Select Parameters Used");
horiz3.add(selectUsedParameters);
JButton selectLastStatisticsUsed = new JButton("Select Last Statistics Used");
horiz3.add(selectLastStatisticsUsed);
horiz3.add(Box.createHorizontalGlue());
vert1.add(horiz3);
selectUsedParameters.addActionListener(e1 -> {
for (int i = 0; i < table.getRowCount(); i++) {
GridSearchModel.MyTableColumn myTableColumn = columnSelectionTableModel.getMyTableColumn(i);
if (myTableColumn.getType() == GridSearchModel.MyTableColumn.ColumnType.PARAMETER
&& myTableColumn.isSetByUser()) {
columnSelectionTableModel.selectRow(i);
}
}
});
selectLastStatisticsUsed.addActionListener(e1 -> {
for (int i = 0; i < table.getRowCount(); i++) {
GridSearchModel.MyTableColumn myTableColumn = columnSelectionTableModel.getMyTableColumn(i);
List lastStatisticsUsed = model.getLastStatisticsUsed();
if (myTableColumn.getType() == GridSearchModel.MyTableColumn.ColumnType.STATISTIC
&& lastStatisticsUsed.contains(myTableColumn.getColumnName())) {
columnSelectionTableModel.selectRow(i);
}
}
});
panel.add(vert1, BorderLayout.CENTER);
// Create the JDialog. Use the parent frame to make it modal.
JDialog dialog = new JDialog(SwingUtilities.getWindowAncestor(this), "Add Table Column", Dialog.ModalityType.APPLICATION_MODAL);
dialog.setLayout(new BorderLayout());
dialog.add(panel, BorderLayout.CENTER);
dialog.setPreferredSize(new Dimension(500, 500));
// Create a panel for the buttons
JPanel buttonPanel = getButtonPanel(columnSelectionTableModel, dialog, tabbedPane);
// Add the button panel to the bottom of the dialog
dialog.add(buttonPanel, BorderLayout.SOUTH);
// Set the dialog size, position, and visibility
dialog.pack(); // Adjust dialog size to fit its contents
dialog.setLocationRelativeTo(this);
dialog.setVisible(true);
});
}
@NotNull
private JPanel getButtonPanel(TableColumnSelectionModel columnSelectionTableModel, JDialog dialog, JTabbedPane tabbedPane) {
JPanel buttonPanel = new JPanel(new FlowLayout(FlowLayout.RIGHT));
JButton addButton = new JButton("Add");
JButton cancelButton = new JButton("Cancel");
// Add action listeners for the buttons
addButton.addActionListener(e1 -> {
columnSelectionTableModel.setTableRef(null);
SwingUtilities.invokeLater(dialog::dispose);
List selectedTableColumns = new ArrayList<>(
columnSelectionTableModel.getSelectedTableColumns());
for (GridSearchModel.MyTableColumn column : selectedTableColumns) {
model.addTableColumn(column);
}
setLayout(new BorderLayout());
add(tabbedPane, BorderLayout.CENTER);
setTableColumnsText();
setComparisonText();
});
cancelButton.addActionListener(e12 -> {
columnSelectionTableModel.setTableRef(null);
setLayout(new BorderLayout());
add(tabbedPane, BorderLayout.CENTER);
dialog.dispose();
});
// Add the buttons to the button panel
buttonPanel.add(addButton);
buttonPanel.add(cancelButton);
return buttonPanel;
}
/**
* Sets the text in the simulationChoiceTextArea based on the selected simulations in the model. If no simulations
* are selected, it displays a message indicating that at least one simulation needs to be selected. If only one
* simulation is selected, it displays information about the selected simulation. If multiple simulations are
* selected, it displays information about each of the selected simulations. It also appends the simulation
* parameter text obtained from getSimulationParameterText() method.
*/
private void setSimulationText() {
simulationChoiceTextArea.setText("");
Simulations selectedSimulations = model.getSelectedSimulations();
List simulations = selectedSimulations.getSimulations();
DataSet dataSet = model.getSuppliedData();
if (dataSet != null) {
simulationChoiceTextArea.append("A data set has been supplied with " + dataSet.getNumColumns() + " variables and " + dataSet.getNumRows() + " rows.");
simulationChoiceTextArea.append("\n\nThe variables for the data are as follow: " + dataSet.getVariableNames() + "\n\n");
}
Graph graph = model.getSuppliedGraph();
if (graph != null) {
simulationChoiceTextArea.append("A graph has been supplied with " + graph.getNumNodes() + " nodes and " + graph.getNumEdges() + " edges.");
simulationChoiceTextArea.append("\n\nThe nodes for the graph are as follow: " + graph.getNodeNames() + "\n\n");
}
Knowledge knowledge = model.getKnowledge();
if (knowledge != null) {
simulationChoiceTextArea.append("Knowledge has been set, as follows:");
simulationChoiceTextArea.append("\n\n" + knowledge + "\n\n");
}
if (simulations.isEmpty()) {
simulationChoiceTextArea.append("""
** No simulations have been selected. Please select at least one simulation using the Add Simulation button below. **
""");
return;
} else if (simulations.size() == 1) {
simulationChoiceTextArea.append("""
The following simulation has been selected. This simulations will be run with the selected algorithms.
""");
Simulation simulation = simulations.get(0);
Class randomGraphClass = simulation.getRandomGraphClass();
Class simulationClass = simulation.getSimulationClass();
simulationChoiceTextArea.append("Selected graph type = " + (randomGraphClass == null ? "None" : randomGraphClass.getSimpleName() + "\n"));
simulationChoiceTextArea.append("Selected simulation type = " + simulationClass.getSimpleName() + "\n");
} else {
simulationChoiceTextArea.append("""
The following simulations have been selected. These simulations will be run with the selected algorithms.
""");
for (int i = 0; i < simulations.size(); i++) {
simulationChoiceTextArea.append("\n");
simulationChoiceTextArea.append("Simulation #" + (i + 1) + ":\n");
simulationChoiceTextArea.append("Selected graph type = " + simulations.get(i).getRandomGraphClass().getSimpleName() + "\n");
simulationChoiceTextArea.append("Selected simulation type = " + simulations.get(i).getSimulationClass().getSimpleName() + "\n");
}
}
simulationChoiceTextArea.append(getSimulationParameterText());
simulationChoiceTextArea.setCaretPosition(0);
}
private void setAlgorithmText() {
algorithmChoiceTextArea.setText("");
List selectedAlgorithms = model.getSelectedAlgorithms();
if (selectedAlgorithms.isEmpty()) {
algorithmChoiceTextArea.append("""
** No algorithm have been selected. Please select at least one algorithm using the Add Algorithm button below. **
""");
return;
} else if (selectedAlgorithms.size() == 1) {
algorithmChoiceTextArea.append("""
The following algorithm has been selected. This algorithm will be run with the selected simulations.
""");
GridSearchModel.AlgorithmSpec algorithm = selectedAlgorithms.get(0);
algorithmChoiceTextArea.append("Selected algorithm: " + algorithm.getAlgorithmImpl().getDescription() + "\n");
if (algorithm instanceof TakesIndependenceWrapper) {
algorithmChoiceTextArea.append("Selected independence test = " + ((TakesIndependenceWrapper) algorithm).getIndependenceWrapper().getDescription() + "\n");
}
if (algorithm instanceof UsesScoreWrapper) {
algorithmChoiceTextArea.append("Selected score = " + ((UsesScoreWrapper) algorithm).getScoreWrapper().getDescription() + "\n");
}
} else {
algorithmChoiceTextArea.append("""
The following algorithms have been selected. These algorithms will be run with the selected simulations.
""");
for (int i = 0; i < selectedAlgorithms.size(); i++) {
GridSearchModel.AlgorithmSpec algorithm = selectedAlgorithms.get(i);
algorithmChoiceTextArea.append("\nAlgorithm #" + (i + 1) + ". " + algorithm.getAlgorithmImpl().getDescription() + "\n");
if (algorithm instanceof TakesIndependenceWrapper) {
algorithmChoiceTextArea.append("Selected independence test = " + ((TakesIndependenceWrapper) algorithm).getIndependenceWrapper().getDescription() + "\n");
}
if (algorithm instanceof UsesScoreWrapper) {
algorithmChoiceTextArea.append("Selected score = " + ((UsesScoreWrapper) algorithm).getScoreWrapper().getDescription() + "\n");
}
}
}
algorithmChoiceTextArea.append(getAlgorithmParameterText());
List selectedAlgorithmModels = model.getSelectedAlgorithms();
Set algorithmDescriptions = new HashSet<>();
if (!selectedAlgorithmModels.isEmpty()) {
algorithmChoiceTextArea.append("\n\nAlgorithm Descriptions:");
}
for (GridSearchModel.AlgorithmSpec algorithmSpec : selectedAlgorithmModels) {
if (algorithmDescriptions.contains(algorithmSpec.getName())) {
continue;
}
algorithmChoiceTextArea.append("\n\n" + algorithmSpec.getName());
algorithmChoiceTextArea.append("\n\n" + algorithmSpec.getAlgorithm().getDescription().replace("\n", "\n\n"));
algorithmDescriptions.add(algorithmSpec.getName());
}
Set independenceWrappers = new HashSet<>();
for (GridSearchModel.AlgorithmSpec algorithm : selectedAlgorithms) {
if (algorithm instanceof TakesIndependenceWrapper) {
independenceWrappers.add(((TakesIndependenceWrapper) algorithm).getIndependenceWrapper());
}
}
Set scoreWrappers = new HashSet<>();
Set scoreDescriptions = new HashSet<>();
for (GridSearchModel.AlgorithmSpec algorithm : selectedAlgorithms) {
if (algorithm instanceof UsesScoreWrapper) {
if (scoreDescriptions.contains(algorithm.getAlgorithmImpl().getDescription())) {
continue;
}
scoreWrappers.add(((UsesScoreWrapper) algorithm).getScoreWrapper());
scoreDescriptions.add(algorithm.getAlgorithmImpl().getDescription());
}
}
IndependenceTestModels independenceTestModels = IndependenceTestModels.getInstance();
Set independenceTestModels1 = new HashSet<>(independenceTestModels.getModels());
Set independenceDescriptions = new HashSet<>();
if (!independenceWrappers.isEmpty()) {
algorithmChoiceTextArea.append("\n\nIndependence Test Descriptions:");
}
for (IndependenceTestModel independenceTestModel : independenceTestModels1) {
independenceWrappers.forEach(independenceWrapper -> {
if (independenceTestModel.getIndependenceTest().clazz().equals(independenceWrapper.getClass())) {
if (independenceDescriptions.contains(independenceTestModel.getName())) {
return;
}
algorithmChoiceTextArea.append("\n\n" + independenceTestModel.getName());
algorithmChoiceTextArea.append("\n\n" + independenceTestModel.getDescription().replace("\n", "\n\n"));
independenceDescriptions.add(independenceTestModel.getName());
}
});
}
ScoreModels scoreModels = ScoreModels.getInstance();
Set scoreModels1 = new HashSet<>(scoreModels.getModels());
Set scoreDescriptions1 = new HashSet<>();
if (!scoreWrappers.isEmpty()) {
algorithmChoiceTextArea.append("\n\nScore Descriptions:");
}
for (ScoreModel scoreModel : scoreModels1) {
scoreWrappers.forEach(scoreWrapper -> {
if (scoreModel.getScore().clazz().equals(scoreWrapper.getClass())) {
if (scoreDescriptions1.contains(scoreModel.getName())) {
return;
}
algorithmChoiceTextArea.append("\n\n" + scoreModel.getName());
algorithmChoiceTextArea.append("\n\n" + scoreModel.getDescription().replace("\n", "\n\n"));
scoreDescriptions1.add(scoreModel.getName());
}
});
}
algorithmChoiceTextArea.setCaretPosition(0);
}
/**
* Sets the text in the tableColumnsChoiceTextArea based on the selected table columns.
*/
private void setTableColumnsText() {
tableColumnsChoiceTextArea.setText("");
List selectedTableColumns = model.getSelectedTableColumns();
if (selectedTableColumns.isEmpty()) {
tableColumnsChoiceTextArea.append("""
** No columns have been selected. Please select at least one column using the Add Table Column button below. **
""");
return;
} else if (selectedTableColumns.size() == 1) {
tableColumnsChoiceTextArea.setText("""
The following table column has been selected. The comparison table will include these columns in the table.
""");
} else {
tableColumnsChoiceTextArea.setText("The comparison table will include these columns in the table.");
}
for (int i = 0; i < selectedTableColumns.size(); i++) {
GridSearchModel.MyTableColumn tableColumn = selectedTableColumns.get(i);
tableColumnsChoiceTextArea.append("\n\n" + (i + 1) + ". " + tableColumn.getColumnName() + " (" + tableColumn.getDescription() + ")");
}
tableColumnsChoiceTextArea.setCaretPosition(0);
}
/**
* Sets the text of the comparisonTextArea based on the selected simulations, algorithms, and table columns. If any
* of the simulations, algorithms, or table columns are empty, it sets a message indicating that the selection is
* empty. Otherwise, it sets a message indicating that a comparison has not been run for the selection.
*/
private void setComparisonText() {
if (model.getSelectedSimulations().getSimulations().isEmpty() || model.getSelectedAlgorithms().isEmpty()
|| model.getSelectedTableColumns().isEmpty()) {
comparisonTextArea.setText(
"""
** You have made an empty selection; look back at the Simulation, Algorithm, and Table Columns tabs **
""");
} else if (comparisonTextArea.getText().isBlank()) {
comparisonTextArea.setText
("""
** Your selection is non-empty, but you have not yet run a comparison for it **
""");
}
}
/**
* Sets the help text for the tool.
*/
private void setHelpText() {
helpChoiceTextArea.setText("""
This tool may be used to do a comparison of multiple algorithms (in Tetrad for now) for a range of simulations types, algorithms, table columns, and parameter settings.
To run a Grid Search comparison, select one or more simulations, one or more algorithms, and one or more table columns (statistics or parameter columns). Then in the Comparison tab, click the "Run Comparison" button.
The comparison will be displayed in the "comparison" tab.
Some combinations you may select could take a very long time to run; you may need to experiment. One problem is that you may select too many combinations of parameters, and the tool will try every combination of these parameters that is sensible, and perhaps this may take a very long time to do. Or you may, for instance, opt for graphs that have too many variables or are too dense. Or, some of the algorithms may simply take a very long time to run, even for small graphs. We will run your request in a thread with a stop button so you can gracefully exit and try a smaller problem. In fact, it may not make sense to run larger comparisons in this interface at all; you may wish to use the command line tool or Python to do it.
If you think the problem is that you need more memory, you can increase the memory available to the JVM by starting Tetrad from the command line changing the -Xmx option in at startup. That is, you can start Tetrad with a command like this:
java -Xmx4g -jar [tetrad.jar]
Here, "[tetrad.jar]" should be replaced by the name of the Tetrad jar you have downloaded. This would set the maximum memory available to the JVM to 4 gigabytes. You can increase this number to increase the memory available to the JVM up to the limit of what you have available on your machine. The default is 1 gigabyte.
In the Simulation tab, simulations may be added by clicking the Add Simulation button. The last one in the list may be removed by clicking the Remove Last Simulation button.
A simulation selection requires one to select a graph type and a simulation type.
This selection implies a list of parameters for all of the simulations. These parameters may be edited by clicking the Edit Parameters button. Note that parameters may be given a list of comma-separated values; each combination of parameters will be explored in the comparison.
The Algorithm tab and TableColumns tab work similarly. An algorithm selection requires one to select an algorithm type and then an independence test and/or a score depending on the requirements of the algorithm.
For the Algorithm tab, once one has selected all algorithms, one may edit the parameters for these algorithms.
For the TableColumns tab, one may select the columns to be included in the comparison table. These columns may be selected by clicking the Add Table Column button. The last column in the list may be removed by clicking the Remove Last Table Column button. For parameter columns, parameters that have been set by the user may be selected by clicking the Select Parameters Used button. For statistic columns, the last statistics used may be selected by clicking the Select Last Statistics Used button. This will select all statistics that were used in the last comparison.
In the Comparison tab, there is a button to run the comparison and display the results.
Full results are saved to the user's hard drive. The location of these files is displayed in the comparison tab, at the top of the page. This includes all the output from the comparison, including the true dataset and graphs for all simulations, the estimated graph, elapsed times for all algorithm runs, and the results displayed in the Comparison tab for the comparison. These datasets and graphs may be used for analayis by other tools, such as in R or Python.
The reference is here:
Ramsey, J. D., Malinsky, D., & Bui, K. V. (2020). Algcomparison: Comparing the performance of graphical structure learning algorithms with tetrad. Journal of Machine Learning Research, 21(238), 1-6.
""");
}
/**
* Retrieves the simulation parameter text for displaying the parameter choices
*
* @return a string containing the simulation parameter text
*/
@NotNull
private String getSimulationParameterText() {
List simulations = model.getSelectedSimulations().getSimulations();
Set paramNamesSet = getAllSimulationParameters(simulations);
StringBuilder paramText;
if (simulations.size() == 1) {
paramText = new StringBuilder("\nParameter choices for this simulation:");
} else {
paramText = new StringBuilder("\nParameter choices for all simulations:");
}
paramText.append(getParameterText(paramNamesSet, model.getParameters()));
return paramText.toString();
}
/**
* Retrieves the algorithm parameter choices as text.
*
* @return The algorithm parameter choices as text.
*/
private String getAlgorithmParameterText() {
List algorithm = model.getSelectedAlgorithms();
Set allAlgorithmParameters = GridSearchModel.getAllAlgorithmParameters(algorithm);
Set allTestParameters = GridSearchModel.getAllTestParameters(algorithm);
Set allScoreParameters = GridSearchModel.getAllScoreParameters(algorithm);
Set allBootstrappingParameters = GridSearchModel.getAllBootstrapParameters(algorithm);
StringBuilder paramText = new StringBuilder();
if (algorithm.size() == 1) {
if (!allAlgorithmParameters.isEmpty()) {
paramText.append("\nParameter choices for this algorithm:");
paramText.append(getParameterText(allAlgorithmParameters, model.getParameters()));
}
if (!allTestParameters.isEmpty()) {
paramText.append("\n\nParameter choices for this test:");
paramText.append(getParameterText(allTestParameters, model.getParameters()));
}
if (!allScoreParameters.isEmpty()) {
paramText.append("\n\nParameter choices for this score:");
paramText.append(getParameterText(allScoreParameters, model.getParameters()));
}
if (!allBootstrappingParameters.isEmpty()) {
paramText.append("\n\nParameter choices for bootstrapping:");
paramText.append(getParameterText(allBootstrappingParameters, model.getParameters()));
}
} else {
if (!allAlgorithmParameters.isEmpty()) {
paramText.append("\nParameter choices for all algorithms:");
paramText.append(getParameterText(allAlgorithmParameters, model.getParameters()));
}
if (!allTestParameters.isEmpty()) {
paramText.append("\n\nParameter choices for all tests:");
paramText.append(getParameterText(allTestParameters, model.getParameters()));
}
if (!allScoreParameters.isEmpty()) {
paramText.append("\n\nParameter choices for all scores:");
paramText.append(getParameterText(allScoreParameters, model.getParameters()));
}
if (!allBootstrappingParameters.isEmpty()) {
paramText.append("\n\nParameter choices for bootstrapping:");
paramText.append(getParameterText(allBootstrappingParameters, model.getParameters()));
}
}
return paramText.toString();
}
/**
* This class extends ByteArrayOutputStream and adds buffering and listening functionality. It overrides the write
* methods to capture the data being written and process it when a newline character is encountered.
*/
public static class BufferedListeningByteArrayOutputStream extends ByteArrayOutputStream {
private final StringBuilder buffer = new StringBuilder();
@Override
public void write(int b) {
// Convert byte to string and add to buffer
String s = new String(new byte[]{(byte) b}, StandardCharsets.UTF_8);
buffer.append(s);
// Process buffer if newline character is found
if (s.contains("\n")) {
super.write(buffer.toString().getBytes(StandardCharsets.UTF_8), 0, buffer.length());
buffer.setLength(0); // Clear the buffer for next data
}
}
@Override
public void write(byte[] b, int off, int len) {
// Convert the byte array to string and add to buffer
String s = new String(b, off, len, StandardCharsets.UTF_8);
buffer.append(s);
// Process buffer if newline character is found
if (s.contains("\n")) {
super.write(buffer.toString().getBytes(StandardCharsets.UTF_8), 0, buffer.length());
buffer.setLength(0); // Clear the buffer for next data
}
}
}
/**
* A table model for the results of the IDA check. This table can be sorted by clicking on the column headers, up or
* down. The table can be copied and pasted into a text file or into Excel.
*/
private static class TableColumnSelectionModel extends AbstractTableModel {
/**
* The column names for the table. The first column is the pair of nodes, the second column is the minimum total
* effect, the third column is the maximum total effect, the fourth column is the minimum absolute total effect,
* the fifth column is the true total effect, and the sixth column is the squared distance from the true total
* effect, where if the true total effect falls between the minimum and maximum total effect zero is reported.
* If the true model is not given, the last two columns are not included.
*/
private final String[] columnNames = {"Index", "Column Name", "Description", "Selected"};
/**
* The data for the table.
*/
private final Object[][] data;
private final List allTableColumns;
private final Set selectedTableColumns;
private JTable tableRef;
/**
* Constructs a new table estModel for the results of the IDA check.
*/
public TableColumnSelectionModel(List allTableColumns, Set selectedTableColumns) {
if (allTableColumns == null) {
throw new IllegalArgumentException("allTableColumns is null");
}
if (selectedTableColumns == null) {
throw new IllegalArgumentException("selectedTableColumns is null");
}
if (!new HashSet<>(allTableColumns).containsAll(selectedTableColumns)) {
throw new IllegalArgumentException("selectedTableColumns contains elements not in allTableColumns");
}
// Create the data for the table
this.data = new Object[allTableColumns.size()][3];
this.allTableColumns = allTableColumns;
this.selectedTableColumns = new HashSet<>(selectedTableColumns);
for (int i = 0; i < allTableColumns.size(); i++) {
GridSearchModel.MyTableColumn tableColumn = allTableColumns.get(i);
this.data[i][0] = i + 1; // 1-based index (not 0-based index)
this.data[i][1] = tableColumn.getColumnName();
this.data[i][2] = tableColumn.getDescription();
}
}
/**
* Returns the number of rows in the table.
*
* @return the number of rows in the table.
*/
@Override
public int getRowCount() {
return allTableColumns.size();
}
/**
* Returns the number of columns in the table.
*
* @return the number of columns in the table.
*/
@Override
public int getColumnCount() {
return 4;
}
/**
* Returns the name of the column at the given index.
*
* @param col the index of the column.
* @return the name of the column at the given index.
*/
@Override
public String getColumnName(int col) {
return this.columnNames[col];
}
/**
* Returns the value at the given row and column.
*
* @param row the row.
* @param col the column.
* @return the value at the given row and column.
*/
@Override
public Object getValueAt(int row, int col) {
if (col == 3) {
if (tableRef == null || tableRef.getSelectionModel() == null) {
return "";
}
int index = tableRef.convertRowIndexToView(row); // Convert the row index to the model index (in case the table is sorted)
if (index < 0 || index >= allTableColumns.size()) {
return "";
}
boolean rowSelected = tableRef.getSelectionModel().isSelectedIndex(index);
if (rowSelected) {
selectedTableColumns.add(allTableColumns.get(((Integer) data[row][0]) - 1));
} else {
selectedTableColumns.remove(allTableColumns.get(((Integer) data[row][0]) - 1));
}
return rowSelected ? "Selected" : "";
} else {
return this.data[row][col];
}
}
/**
* Returns the class of the column at the given index.
*
* @param c the index of the column.
* @return the class of the column at the given index.
*/
@Override
public Class getColumnClass(int c) {
return getValueAt(0, c).getClass();
}
public Set getSelectedTableColumns() {
return selectedTableColumns;
}
public void setTableRef(JTable tableRef) {
this.tableRef = tableRef;
}
public GridSearchModel.MyTableColumn getMyTableColumn(int row) {
return allTableColumns.get(row);
}
public void selectRow(int row) {
tableRef.getSelectionModel().addSelectionInterval(row, row);
}
}
public static class TextAreaOutputStream extends OutputStream {
private final JTextArea textArea;
private final StringBuilder sb = new StringBuilder();
public TextAreaOutputStream(JTextArea textArea) {
this.textArea = textArea;
}
@Override
public void write(int b) throws IOException {
if (b == '\r') return; // Ignore carriage return on Windows
if (b == '\n') {
final String text = sb + "\n";
SwingUtilities.invokeLater(() -> textArea.append(text));
sb.setLength(0); // Reset StringBuilder
} else {
sb.append((char) b);
}
}
}
}
