Method getDatasets

src/org/opensourcephysics/tools/DataTool.java:1684–1726

Gets a list of Datasets from a self-contained source Data object. @param source the self-contained Data @return a list of Datasets

(Data source)

	 * @return a list of Datasets
	 */
	public static ArrayList<Dataset> getDatasets(Data source) {
		// if the source supplies datasets, return them
		ArrayList<Dataset> datasets = (source instanceof DatasetManager ? 
				((DatasetManager)source).getDatasetsRaw() : source.getDatasets());
		if (datasets != null) {
			return datasets;
		}
		// else create an empty dataset list and populate it from data2D
		datasets = new ArrayList<Dataset>();
		double[][] data2D = source.getData2D();
		if ((data2D == null) || (data2D.length == 0) || (data2D[0] == null)) {
			return datasets; // return empty list
		}
		// get column names--minimum 2
		// colNames[0] will be x-column name
		// colNames[1] and up will be y-column names
		String[] colNames = source.getColumnNames();
		// if colNames is null, create colNames
		if (colNames == null) {
			colNames = new String[2];
			// if only one point set in data2D, then first column is named "n"
			if (data2D.length == 1) {
				colNames[0] = "n"; //$NON-NLS-1$
			}
		}
		int n = Math.max(2, data2D.length); // number of columns
		if (colNames.length > n) {
			n++;
		}
		colNames = getColumnNames(colNames, n);
		// create datasets for double[] columns
		// if more names than double[] columns, then first double[] is row numbers
		boolean xPointsAreRowNumbers = colNames.length > data2D.length;
		double[] xPoints = xPointsAreRowNumbers ? getRowArray(data2D[0].length) : data2D[0];
		for (int i = 1; i < colNames.length; i++) {
			double[] yPoints = xPointsAreRowNumbers ? data2D[i - 1] : data2D[i];
			Dataset dataset = createDataset(xPoints, yPoints, colNames[0], colNames[i], i, source);
			if (dataset != null) {
				datasets.add(dataset);
			}
		}
		return datasets;
	}

	/**
	 * Gets a list of all Datasets from any Data object.