/**
* Copyright (C) 2010-14 diirt developers. See COPYRIGHT.TXT
* All rights reserved. Use is subject to license terms. See LICENSE.TXT
*/
package org.diirt.datasource.graphene;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import org.diirt.graphene.Cell1DDataset;
import org.diirt.graphene.Cell1DDatasets;
import org.diirt.graphene.Cell2DDataset;
import org.diirt.graphene.Point2DDataset;
import org.diirt.graphene.Point2DDatasets;
import org.diirt.graphene.Point3DWithLabelDataset;
import org.diirt.graphene.Point3DWithLabelDatasets;
import org.diirt.util.array.ListDouble;
import org.diirt.util.array.ListNumber;
import org.diirt.util.array.ListNumbers;
import org.diirt.util.stats.Range;
import org.diirt.util.stats.Ranges;
import org.diirt.vtype.Display;
import org.diirt.vtype.VNumberArray;
import org.diirt.vtype.VTable;
import org.diirt.vtype.ValueUtil;
/**
* Utility class to convert VTypes into graphene datasets.
*
* @author carcassi
*/
public class DatasetConversions {
/**
* Converts a VTable into a Point2DDataset.
*
* @param vTable the table containing the data
* @param xColumn the column for the x values
* @param yColumn the column for the y values
* @return the dataset
*/
public static Point2DDataset point2DDatasetFromVTable(VTable vTable,
String xColumn, String yColumn) {
// Extract x and y using column names
ListNumber xValues = ValueUtil.numericColumnOf(vTable, xColumn);
ListNumber yValues = ValueUtil.numericColumnOf(vTable, yColumn);
// Fill the missing columns with the first available columns
for (int i = 0; i < vTable.getColumnCount(); i++) {
if (vTable.getColumnType(i).isPrimitive()) {
// Don't reuse the same column
if (!vTable.getColumnName(i).equals(xColumn) &&
!vTable.getColumnName(i).equals(yColumn)) {
if (xValues == null) {
xValues = (ListNumber) vTable.getColumnData(i);
} else if (yValues == null) {
yValues = (ListNumber) vTable.getColumnData(i);
}
}
}
}
if (xValues == null || yValues == null) {
throw new IllegalArgumentException("Couldn't find two numeric columns");
}
return Point2DDatasets.lineData(xValues, yValues);
}
/**
* Converts a VTable into a Point2DDataset.
*
* @param vTable the table containing the data
* @param xColumn the column name for the x values
* @param yColumn the column name for the y values
* @param sizeColumn the column name for the size values
* @param colorColumn the column name for the size values
* @return the dataset
*/
public static Point3DWithLabelDataset point3DDatasetFromVTable(VTable vTable,
String xColumn, String yColumn, String sizeColumn, String colorColumn) {
// Extract x and y using column names
ListNumber xValues = ValueUtil.numericColumnOf(vTable, xColumn);
ListNumber yValues = ValueUtil.numericColumnOf(vTable, yColumn);
ListNumber sizeValues = ValueUtil.numericColumnOf(vTable, sizeColumn);
List<String> colorValues = ValueUtil.stringColumnOf(vTable, colorColumn);
// If none of the columns where specified, find the first column that fits
if (xValues == null && yValues == null && sizeValues == null && colorValues == null) {
// Fill the missing columns with the first available columns
for (int i = 0; i < vTable.getColumnCount(); i++) {
if (vTable.getColumnType(i).isPrimitive()) {
if (xValues == null) {
xValues = (ListNumber) vTable.getColumnData(i);
} else if (yValues == null) {
yValues = (ListNumber) vTable.getColumnData(i);
} else if (sizeValues == null) {
sizeValues = (ListNumber) vTable.getColumnData(i);
}
} else if (vTable.getColumnType(i).equals(String.class)) {
if (colorValues == null) {
@SuppressWarnings("unchecked")
List<String> list = (List<String>) vTable.getColumnData(i);
colorValues = list;
}
}
}
if (xValues == null || yValues == null) {
throw new IllegalArgumentException("Couldn't find two numeric columns for X and Y");
}
}
if (xValues == null || yValues == null) {
throw new IllegalArgumentException("X and Y must both be specified");
}
// If sizes is missing, generate a 0 columns
final int nValues = xValues.size();
if (sizeValues == null) {
sizeValues = new ListDouble() {
@Override
public double getDouble(int index) {
return 0;
}
@Override
public int size() {
return nValues;
}
};
}
// If color is missing, generate a "None" column
if (colorValues == null) {
colorValues = Collections.nCopies(xValues.size(), "None");
}
return Point3DWithLabelDatasets.build(xValues, yValues, sizeValues, colorValues);
}
public static List<Point2DDataset> point2DDatasetsFromVTable(VTable vTable, List<String> xColumns, List<String> yColumns) {
List<ListNumber> xValues = new ArrayList<>();
List<ListNumber> yValues = new ArrayList<>();
if (xColumns != null && yColumns != null) {
for (String column : xColumns) {
xValues.add(ValueUtil.numericColumnOf(vTable, column));
}
for (String column : yColumns) {
yValues.add(ValueUtil.numericColumnOf(vTable, column));
}
} else if (xColumns == null && yColumns == null) {
for (int i = 0; i < vTable.getColumnCount(); i++) {
if (vTable.getColumnType(i).isPrimitive()) {
yValues.add((ListNumber) vTable.getColumnData(i));
}
}
} else {
throw new UnsupportedOperationException("Not supported yet.");
}
if (xValues.isEmpty()) {
List<Point2DDataset> datasets = new ArrayList<>();
for (int i = 0; i < yValues.size(); i++) {
datasets.add(Point2DDatasets.lineData(ListNumbers.linearList(0, 1, yValues.get(i).size()), yValues.get(i)));
}
return datasets;
}
if (xValues.size() == yValues.size()) {
List<Point2DDataset> datasets = new ArrayList<>();
for (int i = 0; i < xValues.size(); i++) {
datasets.add(Point2DDatasets.lineData(xValues.get(i), yValues.get(i)));
}
return datasets;
}
if (xValues.size() == 1) {
List<Point2DDataset> datasets = new ArrayList<>();
for (int i = 0; i < yValues.size(); i++) {
datasets.add(Point2DDatasets.lineData(xValues.get(0), yValues.get(i)));
}
return datasets;
}
throw new UnsupportedOperationException("Not supported yet.");
}
public static Cell2DDataset cell2DDatasetsFromVNumberArray(VNumberArray data) {
return new Cell2DDatasetFromVNumberArray(data);
}
public static Range toRange(Display display) {
if (display == null) {
return null;
}
Double min = display.getLowerDisplayLimit();
Double max = display.getUpperDisplayLimit();
if (min != null && max != null) {
return Ranges.range(min, max);
} else {
return null;
}
}
public static Cell1DDataset cell1DDatasetsFromVNumberArray(VNumberArray data) {
return Cell1DDatasets.datasetFrom(data.getData(), data.getDimensionDisplay().get(0).getCellBoundaries(), toRange(data));
}
}