org.jfree.data.xy.DefaultXYZDataset Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of jfreechart Show documentation
Show all versions of jfreechart Show documentation
JFreeChart is a class library, written in Java, for generating charts.
Utilising the Java2D API, it supports a wide range of chart types including
bar charts, pie charts, line charts, XY-plots, time series plots, Sankey charts
and more.
/* ===========================================================
* JFreeChart : a free chart library for the Java(tm) platform
* ===========================================================
*
* (C) Copyright 2000-present, by David Gilbert and Contributors.
*
* Project Info: http://www.jfree.org/jfreechart/index.html
*
* This library is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation; either version 2.1 of the License, or
* (at your option) any later version.
*
* This library 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 Lesser General Public
* License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
* USA.
*
* [Oracle and Java are registered trademarks of Oracle and/or its affiliates.
* Other names may be trademarks of their respective owners.]
*
* ----------------------
* DefaultXYZDataset.java
* ----------------------
* (C) Copyright 2006-present, by David Gilbert.
*
* Original Author: David Gilbert;
* Contributor(s): -;
*
*/
package org.jfree.data.xy;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import org.jfree.chart.util.PublicCloneable;
import org.jfree.data.DomainOrder;
import org.jfree.data.general.DatasetChangeEvent;
/**
* A default implementation of the {@link XYZDataset} interface that stores
* data values in arrays of double primitives.
*/
public class DefaultXYZDataset extends AbstractXYZDataset
implements XYZDataset, PublicCloneable {
/**
* Storage for the series keys. This list must be kept in sync with the
* seriesList.
*/
private List seriesKeys;
/**
* Storage for the series in the dataset. We use a list because the
* order of the series is significant. This list must be kept in sync
* with the seriesKeys list.
*/
private List seriesList;
/**
* Creates a new {@code DefaultXYZDataset} instance, initially
* containing no data.
*/
public DefaultXYZDataset() {
this.seriesKeys = new java.util.ArrayList();
this.seriesList = new java.util.ArrayList();
}
/**
* Returns the number of series in the dataset.
*
* @return The series count.
*/
@Override
public int getSeriesCount() {
return this.seriesList.size();
}
/**
* Returns the key for a series.
*
* @param series the series index (in the range {@code 0} to
* {@code getSeriesCount() - 1}).
*
* @return The key for the series.
*
* @throws IllegalArgumentException if {@code series} is not in the
* specified range.
*/
@Override
public Comparable getSeriesKey(int series) {
if ((series < 0) || (series >= getSeriesCount())) {
throw new IllegalArgumentException("Series index out of bounds");
}
return (Comparable) this.seriesKeys.get(series);
}
/**
* Returns the index of the series with the specified key, or -1 if there
* is no such series in the dataset.
*
* @param seriesKey the series key ({@code null} permitted).
*
* @return The index, or -1.
*/
@Override
public int indexOf(Comparable seriesKey) {
return this.seriesKeys.indexOf(seriesKey);
}
/**
* Returns the order of the domain (x-) values in the dataset. In this
* implementation, we cannot guarantee that the x-values are ordered, so
* this method returns {@code DomainOrder.NONE}.
*
* @return {@code DomainOrder.NONE}.
*/
@Override
public DomainOrder getDomainOrder() {
return DomainOrder.NONE;
}
/**
* Returns the number of items in the specified series.
*
* @param series the series index (in the range {@code 0} to
* {@code getSeriesCount() - 1}).
*
* @return The item count.
*
* @throws IllegalArgumentException if {@code series} is not in the
* specified range.
*/
@Override
public int getItemCount(int series) {
if ((series < 0) || (series >= getSeriesCount())) {
throw new IllegalArgumentException("Series index out of bounds");
}
double[][] seriesArray = (double[][]) this.seriesList.get(series);
return seriesArray[0].length;
}
/**
* Returns the x-value for an item within a series.
*
* @param series the series index (in the range {@code 0} to
* {@code getSeriesCount() - 1}).
* @param item the item index (in the range {@code 0} to
* {@code getItemCount(series)}).
*
* @return The x-value.
*
* @throws ArrayIndexOutOfBoundsException if {@code series} is not
* within the specified range.
* @throws ArrayIndexOutOfBoundsException if {@code item} is not
* within the specified range.
*
* @see #getX(int, int)
*/
@Override
public double getXValue(int series, int item) {
double[][] seriesData = (double[][]) this.seriesList.get(series);
return seriesData[0][item];
}
/**
* Returns the x-value for an item within a series.
*
* @param series the series index (in the range {@code 0} to
* {@code getSeriesCount() - 1}).
* @param item the item index (in the range {@code 0} to
* {@code getItemCount(series)}).
*
* @return The x-value.
*
* @throws ArrayIndexOutOfBoundsException if {@code series} is not
* within the specified range.
* @throws ArrayIndexOutOfBoundsException if {@code item} is not
* within the specified range.
*
* @see #getXValue(int, int)
*/
@Override
public Number getX(int series, int item) {
return getXValue(series, item);
}
/**
* Returns the y-value for an item within a series.
*
* @param series the series index (in the range {@code 0} to
* {@code getSeriesCount() - 1}).
* @param item the item index (in the range {@code 0} to
* {@code getItemCount(series)}).
*
* @return The y-value.
*
* @throws ArrayIndexOutOfBoundsException if {@code series} is not
* within the specified range.
* @throws ArrayIndexOutOfBoundsException if {@code item} is not
* within the specified range.
*
* @see #getY(int, int)
*/
@Override
public double getYValue(int series, int item) {
double[][] seriesData = (double[][]) this.seriesList.get(series);
return seriesData[1][item];
}
/**
* Returns the y-value for an item within a series.
*
* @param series the series index (in the range {@code 0} to
* {@code getSeriesCount() - 1}).
* @param item the item index (in the range {@code 0} to
* {@code getItemCount(series)}).
*
* @return The y-value.
*
* @throws ArrayIndexOutOfBoundsException if {@code series} is not
* within the specified range.
* @throws ArrayIndexOutOfBoundsException if {@code item} is not
* within the specified range.
*
* @see #getX(int, int)
*/
@Override
public Number getY(int series, int item) {
return getYValue(series, item);
}
/**
* Returns the z-value for an item within a series.
*
* @param series the series index (in the range {@code 0} to
* {@code getSeriesCount() - 1}).
* @param item the item index (in the range {@code 0} to
* {@code getItemCount(series)}).
*
* @return The z-value.
*
* @throws ArrayIndexOutOfBoundsException if {@code series} is not
* within the specified range.
* @throws ArrayIndexOutOfBoundsException if {@code item} is not
* within the specified range.
*
* @see #getZ(int, int)
*/
@Override
public double getZValue(int series, int item) {
double[][] seriesData = (double[][]) this.seriesList.get(series);
return seriesData[2][item];
}
/**
* Returns the z-value for an item within a series.
*
* @param series the series index (in the range {@code 0} to
* {@code getSeriesCount() - 1}).
* @param item the item index (in the range {@code 0} to
* {@code getItemCount(series)}).
*
* @return The z-value.
*
* @throws ArrayIndexOutOfBoundsException if {@code series} is not
* within the specified range.
* @throws ArrayIndexOutOfBoundsException if {@code item} is not
* within the specified range.
*
* @see #getZ(int, int)
*/
@Override
public Number getZ(int series, int item) {
return getZValue(series, item);
}
/**
* Adds a series or if a series with the same key already exists replaces
* the data for that series, then sends a {@link DatasetChangeEvent} to
* all registered listeners.
*
* @param seriesKey the series key ({@code null} not permitted).
* @param data the data (must be an array with length 3, containing three
* arrays of equal length, the first containing the x-values, the
* second containing the y-values and the third containing the
* z-values).
*/
public void addSeries(Comparable seriesKey, double[][] data) {
if (seriesKey == null) {
throw new IllegalArgumentException(
"The 'seriesKey' cannot be null.");
}
if (data == null) {
throw new IllegalArgumentException("The 'data' is null.");
}
if (data.length != 3) {
throw new IllegalArgumentException(
"The 'data' array must have length == 3.");
}
if (data[0].length != data[1].length
|| data[0].length != data[2].length) {
throw new IllegalArgumentException("The 'data' array must contain "
+ "three arrays all having the same length.");
}
int seriesIndex = indexOf(seriesKey);
if (seriesIndex == -1) { // add a new series
this.seriesKeys.add(seriesKey);
this.seriesList.add(data);
}
else { // replace an existing series
this.seriesList.remove(seriesIndex);
this.seriesList.add(seriesIndex, data);
}
notifyListeners(new DatasetChangeEvent(this, this));
}
/**
* Removes a series from the dataset, then sends a
* {@link DatasetChangeEvent} to all registered listeners.
*
* @param seriesKey the series key ({@code null} not permitted).
*
*/
public void removeSeries(Comparable seriesKey) {
int seriesIndex = indexOf(seriesKey);
if (seriesIndex >= 0) {
this.seriesKeys.remove(seriesIndex);
this.seriesList.remove(seriesIndex);
notifyListeners(new DatasetChangeEvent(this, this));
}
}
/**
* Tests this {@code DefaultXYZDataset} instance for equality with an
* arbitrary object. This method returns {@code true} if and only if:
*
* - {@code obj} is not {@code null};
* - {@code obj} is an instance of {@code DefaultXYDataset};
* - both datasets have the same number of series, each containing
* exactly the same values.
*
*
* @param obj the object ({@code null} permitted).
*
* @return A boolean.
*/
@Override
public boolean equals(Object obj) {
if (obj == this) {
return true;
}
if (!(obj instanceof DefaultXYZDataset)) {
return false;
}
DefaultXYZDataset that = (DefaultXYZDataset) obj;
if (!this.seriesKeys.equals(that.seriesKeys)) {
return false;
}
for (int i = 0; i < this.seriesList.size(); i++) {
double[][] d1 = (double[][]) this.seriesList.get(i);
double[][] d2 = (double[][]) that.seriesList.get(i);
double[] d1x = d1[0];
double[] d2x = d2[0];
if (!Arrays.equals(d1x, d2x)) {
return false;
}
double[] d1y = d1[1];
double[] d2y = d2[1];
if (!Arrays.equals(d1y, d2y)) {
return false;
}
double[] d1z = d1[2];
double[] d2z = d2[2];
if (!Arrays.equals(d1z, d2z)) {
return false;
}
}
return true;
}
/**
* Returns a hash code for this instance.
*
* @return A hash code.
*/
@Override
public int hashCode() {
int result;
result = this.seriesKeys.hashCode();
result = 29 * result + this.seriesList.hashCode();
return result;
}
/**
* Creates an independent copy of this dataset.
*
* @return The cloned dataset.
*
* @throws CloneNotSupportedException if there is a problem cloning the
* dataset (for instance, if a non-cloneable object is used for a
* series key).
*/
@Override
public Object clone() throws CloneNotSupportedException {
DefaultXYZDataset clone = (DefaultXYZDataset) super.clone();
clone.seriesKeys = new java.util.ArrayList(this.seriesKeys);
clone.seriesList = new ArrayList(this.seriesList.size());
for (int i = 0; i < this.seriesList.size(); i++) {
double[][] data = (double[][]) this.seriesList.get(i);
double[] x = data[0];
double[] y = data[1];
double[] z = data[2];
double[] xx = new double[x.length];
double[] yy = new double[y.length];
double[] zz = new double[z.length];
System.arraycopy(x, 0, xx, 0, x.length);
System.arraycopy(y, 0, yy, 0, y.length);
System.arraycopy(z, 0, zz, 0, z.length);
clone.seriesList.add(i, new double[][] {xx, yy, zz});
}
return clone;
}
}