edu.cmu.tetradapp.editor.SaveDataAction Maven / Gradle / Ivy
///////////////////////////////////////////////////////////////////////////////
// For information as to what this class does, see the Javadoc, below. //
// Copyright (C) 1998, 1999, 2000, 2001, 2002, 2003, 2004, 2005, 2006, //
// 2007, 2008, 2009, 2010, 2014, 2015, 2022 by Peter Spirtes, Richard //
// Scheines, Joseph Ramsey, and Clark Glymour. //
// //
// This program is free software; you can redistribute it and/or modify //
// it under the terms of the GNU General Public License as published by //
// the Free Software Foundation; either version 2 of the License, or //
// (at your option) any later version. //
// //
// This program is distributed in the hope that it will be useful, //
// but WITHOUT ANY WARRANTY; without even the implied warranty of //
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the //
// GNU General Public License for more details. //
// //
// You should have received a copy of the GNU General Public License //
// along with this program; if not, write to the Free Software //
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA //
///////////////////////////////////////////////////////////////////////////////
package edu.cmu.tetradapp.editor;
import edu.cmu.tetrad.data.*;
import edu.cmu.tetrad.util.JOptionUtils;
import edu.cmu.tetrad.util.NumberFormatUtil;
import edu.cmu.tetradapp.model.EditorUtils;
import javax.swing.*;
import java.awt.event.ActionEvent;
import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.nio.file.Files;
import java.text.NumberFormat;
/**
* Saves data to disk.
*
* @author josephramsey
*/
final class SaveDataAction extends AbstractAction {
/**
* Formats all numbers.
*/
private final NumberFormat nf = NumberFormatUtil.getInstance().getNumberFormat();
/**
* The dataEditor. -
*/
private DataEditor dataEditor;
/***
* Search editor to get data from.
*/
private MarkovBlanketSearchEditor searchEditor;
/**
* Creates a new action to save data.
*
* @param editor a {@link edu.cmu.tetradapp.editor.DataEditor} object
*/
public SaveDataAction(DataEditor editor) {
super("Save Data...");
if (editor == null) {
throw new NullPointerException("Data Editor must not be null.");
}
this.setDataEditor(editor);
}
/**
* Constructor for SaveDataAction.
*
* @param editor a {@link edu.cmu.tetradapp.editor.MarkovBlanketSearchEditor} object
*/
public SaveDataAction(MarkovBlanketSearchEditor editor) {
super("Save Data...");
if (editor == null) {
throw new NullPointerException("Editor must not be null");
}
this.searchEditor = editor;
}
/**
* {@inheritDoc}
*
* Performs the action of loading a session from a file.
*/
public void actionPerformed(ActionEvent e) {
try {
saveData();
} catch (IOException e1) {
e1.printStackTrace();
JOptionPane.showMessageDialog(JOptionUtils.centeringComp(),
"Error in saving: " + e1.getMessage());
}
}
/**
* Saves data in the selected data set to a file.
*/
private void saveData() throws IOException {
DataModel dataModel;
if (this.dataEditor != null) {
dataModel = getDataEditor().getDataModelList();
} else {
dataModel = this.searchEditor.getDataModel();
}
if (dataModel == null) {
return;
}
if (dataModel instanceof DataModelList) {
if (((DataModelList) dataModel).size() == 1) {
dataModel = ((DataModelList) dataModel).get(0);
}
}
String name = dataModel.getName();
if (name == null) name = "data";
if (dataModel instanceof DataSet dataSet) {
File file = EditorUtils.getSaveFile(name.replace(" ", "_"), "txt", getDataEditor(), false, "Save Data...");
if (file == null) {
return;
}
char delimiter = '\t';
if (file.getName().endsWith(".csv")) {
delimiter = ',';
}
PrintWriter out;
try {
out = new PrintWriter(Files.newOutputStream(file.toPath()));
} catch (IOException e) {
throw new IllegalArgumentException(
"Output file could not be opened: " + file);
}
if (dataSet.isContinuous()) {
DataWriter.writeRectangularData(dataSet, out, delimiter);
} else if (dataSet.isDiscrete()) {
DataWriter.writeRectangularData(dataSet, out, delimiter);
} else {
DataWriter.writeRectangularData(dataSet, out, delimiter);
}
out.close();
} else if (dataModel instanceof ICovarianceMatrix) {
File file = EditorUtils.getSaveFile(name.replace(" ", "_"), "txt", getDataEditor(), false, "Save Data...");
if (file == null) {
return;
}
PrintWriter out;
try {
out = new PrintWriter(Files.newOutputStream(file.toPath()));
} catch (IOException e) {
throw new IllegalArgumentException(
"Output file could not be opened: " + file);
}
DataWriter.writeCovMatrix((ICovarianceMatrix) dataModel, out, this.nf);
out.close();
} else if (dataModel instanceof DataModelList list) {
File file = EditorUtils.getSaveFile(name.replace(" ", "_"), "txt", getDataEditor(), false, "Save Data...");
final char delimiter = '\t';
for (int i = 0; i < list.size(); i++) {
DataModel _dataModel = list.get(i);
if (_dataModel == null) throw new NullPointerException("Null data model.");
if (_dataModel instanceof DataSet) {
PrintWriter out;
try {
File file1 = new File(file.getParent(), file.getName() + "_" + (i + 1) + ".txt");
System.out.println(file1);
out = new PrintWriter(Files.newOutputStream(file1.toPath()));
} catch (IOException e) {
throw new IllegalArgumentException(
"Output file could not be opened: " + name.replace(" ", "_") + ".txt");
}
DataSet dataSet = (DataSet) ((DataModelList) dataModel).get(i);
if (dataSet.isContinuous()) {
DataWriter.writeRectangularData(dataSet, out, delimiter);
} else if (dataSet.isDiscrete()) {
DataWriter.writeRectangularData(dataSet, out, delimiter);
} else {
DataWriter.writeRectangularData(dataSet, out, delimiter);
}
out.close();
} else if (_dataModel instanceof ICovarianceMatrix) {
PrintWriter out;
try {
File file1 = new File(file.getParent(), file.getName() + "_" + (i + 1) + ".txt");
System.out.println(file1);
out = new PrintWriter(Files.newOutputStream(file1.toPath()));
} catch (IOException e) {
throw new IllegalArgumentException(
"Output file could not be opened: " + name.replace(" ", "_") + ".txt");
}
DataWriter.writeCovMatrix((ICovarianceMatrix) _dataModel, out, this.nf);
}
}
} else {
JOptionPane.showMessageDialog(JOptionUtils.centeringComp(),
"Sorry, don't know how to save that.");
}
}
private DataEditor getDataEditor() {
return this.dataEditor;
}
private void setDataEditor(DataEditor dataEditor) {
this.dataEditor = dataEditor;
}
}