org.jfree.data.xy.DefaultXYZDataset Java Examples

/**
 * Confirm that the equals method can distinguish all the required fields.
 */
public void testEquals() {

    DefaultXYZDataset d1 = new DefaultXYZDataset();
    DefaultXYZDataset d2 = new DefaultXYZDataset();
    assertTrue(d1.equals(d2));
    assertTrue(d2.equals(d1));

    double[] x1 = new double[] {1.0, 2.0, 3.0};
    double[] y1 = new double[] {4.0, 5.0, 6.0};
    double[] z1 = new double[] {7.0, 8.0, 9.0};
    double[][] data1 = new double[][] {x1, y1, z1};
    double[] x2 = new double[] {1.0, 2.0, 3.0};
    double[] y2 = new double[] {4.0, 5.0, 6.0};
    double[] z2 = new double[] {7.0, 8.0, 9.0};
    double[][] data2 = new double[][] {x2, y2, z2};
    d1.addSeries("S1", data1);
    assertFalse(d1.equals(d2));
    d2.addSeries("S1", data2);
    assertTrue(d1.equals(d2));
}
 

/**
 * Creates a sample dataset for testing.
 * 
 * @return A sample dataset.
 */
public DefaultXYZDataset createSampleDataset1() {
    DefaultXYZDataset d = new DefaultXYZDataset();
    double[] x1 = new double[] {1.0, 2.0, 3.0};
    double[] y1 = new double[] {4.0, 5.0, 6.0};
    double[] z1 = new double[] {7.0, 8.0, 9.0};
    double[][] data1 = new double[][] {x1, y1, z1};
    d.addSeries("S1", data1);
    
    double[] x2 = new double[] {1.0, 2.0, 3.0};
    double[] y2 = new double[] {4.0, 5.0, 6.0};
    double[] z2 = new double[] {7.0, 8.0, 9.0};
    double[][] data2 = new double[][] {x2, y2, z2};
    d.addSeries("S2", data2);
    return d;
}
 

/**
 * Some tests for the addSeries() method.
 */
public void testAddSeries() {
    DefaultXYZDataset d = new DefaultXYZDataset();
    d.addSeries("S1", new double[][] {{1.0}, {2.0}, {3.0}});
    assertEquals(1, d.getSeriesCount());
    assertEquals("S1", d.getSeriesKey(0));
    
    // check that adding a series will overwrite the old series
    d.addSeries("S1", new double[][] {{11.0}, {12.0}, {13.0}});
    assertEquals(1, d.getSeriesCount());
    assertEquals(12.0, d.getYValue(0, 0), EPSILON);
    
    // check null key
    boolean pass = false;
    try
    {
      d.addSeries(null, new double[][] {{1.0}, {2.0}, {3.0}});
    }
    catch (IllegalArgumentException e)
    {
      pass = true;
    }
    assertTrue(pass);
}
 

/**
 * Confirm that the equals method can distinguish all the required fields.
 */
public void testEquals() {
 
    DefaultXYZDataset d1 = new DefaultXYZDataset();
    DefaultXYZDataset d2 = new DefaultXYZDataset();
    assertTrue(d1.equals(d2));
    assertTrue(d2.equals(d1));

    double[] x1 = new double[] {1.0, 2.0, 3.0};
    double[] y1 = new double[] {4.0, 5.0, 6.0};
    double[] z1 = new double[] {7.0, 8.0, 9.0};
    double[][] data1 = new double[][] {x1, y1, z1};
    double[] x2 = new double[] {1.0, 2.0, 3.0};
    double[] y2 = new double[] {4.0, 5.0, 6.0};
    double[] z2 = new double[] {7.0, 8.0, 9.0};
    double[][] data2 = new double[][] {x2, y2, z2};
    d1.addSeries("S1", data1);
    assertFalse(d1.equals(d2));
    d2.addSeries("S1", data2);
    assertTrue(d1.equals(d2));
}
 

/**
 * Some tests for the addSeries() method.
 */
public void testAddSeries() {
    DefaultXYZDataset d = new DefaultXYZDataset();
    d.addSeries("S1", new double[][] {{1.0}, {2.0}, {3.0}});
    assertEquals(1, d.getSeriesCount());
    assertEquals("S1", d.getSeriesKey(0));

    // check that adding a series will overwrite the old series
    d.addSeries("S1", new double[][] {{11.0}, {12.0}, {13.0}});
    assertEquals(1, d.getSeriesCount());
    assertEquals(12.0, d.getYValue(0, 0), EPSILON);

    // check null key
    boolean pass = false;
    try
    {
      d.addSeries(null, new double[][] {{1.0}, {2.0}, {3.0}});
    }
    catch (IllegalArgumentException e)
    {
      pass = true;
    }
    assertTrue(pass);
}
 

/**
 * Creates a sample dataset for testing.
 *
 * @return A sample dataset.
 */
public DefaultXYZDataset createSampleDataset1() {
    DefaultXYZDataset d = new DefaultXYZDataset();
    double[] x1 = new double[] {1.0, 2.0, 3.0};
    double[] y1 = new double[] {4.0, 5.0, 6.0};
    double[] z1 = new double[] {7.0, 8.0, 9.0};
    double[][] data1 = new double[][] {x1, y1, z1};
    d.addSeries("S1", data1);

    double[] x2 = new double[] {1.0, 2.0, 3.0};
    double[] y2 = new double[] {4.0, 5.0, 6.0};
    double[] z2 = new double[] {7.0, 8.0, 9.0};
    double[][] data2 = new double[][] {x2, y2, z2};
    d.addSeries("S2", data2);
    return d;
}
 

private XYZDataset createXyzDataset()
{
	if (sampleXyzDataset == null)
	{
		DefaultXYZDataset dataset = new DefaultXYZDataset(); 
		
		dataset.addSeries(
			"First", 
			new double[][]{ 
				{2.1, 2.3, 2.3, 2.2, 2.2, 1.8, 1.8, 1.9, 2.3, 3.8}, 
				{14.1, 11.1, 10.0, 8.8, 8.7, 8.4, 5.4, 4.1, 4.1, 25}, 
				{2.4, 2.7, 2.7, 2.2, 2.2, 2.2, 2.1, 2.2, 1.6, 4}
				}
			);
		
		sampleXyzDataset = dataset;
	}
	
	return sampleXyzDataset;
}
 

/**
 * A simple test for bug 1766646.
 */
public void testBug1766646B() {
    XYBlockRenderer r = new XYBlockRenderer();
    Range range = r.findRangeBounds(null);
    assertTrue(range == null);
    DefaultXYZDataset emptyDataset = new DefaultXYZDataset();
    range = r.findRangeBounds(emptyDataset);
    assertTrue(range == null);
}
 

/**
 * A simple test for bug 1766646.
 */
@Test
public void testBug1766646A() {
    XYBlockRenderer r = new XYBlockRenderer();
    Range range = r.findDomainBounds(null);
    assertTrue(range == null);
    DefaultXYZDataset emptyDataset = new DefaultXYZDataset();
    range = r.findDomainBounds(emptyDataset);
    assertTrue(range == null);
}
 

/**
 * A simple test for bug 1766646.
 */
public void testBug1766646A() {
    XYBlockRenderer r = new XYBlockRenderer();
    Range range = r.findDomainBounds(null);
    assertTrue(range == null);
    DefaultXYZDataset emptyDataset = new DefaultXYZDataset();
    range = r.findDomainBounds(emptyDataset);
    assertTrue(range == null);
}
 

/**
 * A simple test for bug 1766646.
 */
public void testBug1766646B() {
    XYBlockRenderer r = new XYBlockRenderer();
    Range range = r.findRangeBounds(null);
    assertTrue(range == null);
    DefaultXYZDataset emptyDataset = new DefaultXYZDataset();
    range = r.findRangeBounds(emptyDataset);
    assertTrue(range == null);
}
 

/**
 * Some checks for the indexOf(Comparable) method.
 */
public void testIndexOf() {
    DefaultXYZDataset d = createSampleDataset1();
    assertEquals(0, d.indexOf("S1"));
    assertEquals(1, d.indexOf("S2"));
    assertEquals(-1, d.indexOf("Green Eggs and Ham"));
    assertEquals(-1, d.indexOf(null));
}
 

/**
 * A simple test for bug 1766646.
 */
public void testBug1766646A() {
    XYBlockRenderer r = new XYBlockRenderer();
    Range range = r.findDomainBounds(null);
    assertTrue(range == null);
    DefaultXYZDataset emptyDataset = new DefaultXYZDataset();
    range = r.findDomainBounds(emptyDataset);
    assertTrue(range == null);
}
 

/**
 * Check if finding the bounds in z-dimension of an XYZDataset works. 
 */
public void testFindZBounds() {
    XYShapeRenderer r = new XYShapeRenderer();
    assertNull(r.findZBounds(null));

    DefaultXYZDataset dataset = new DefaultXYZDataset();
    Range range;

    double data1[][] = { {1,1,1}, {1,1,1}, {1,2,3} };
    dataset.addSeries("series1", data1);
    range = r.findZBounds(dataset);
    assertNotNull(range);
    assertEquals(1d, range.getLowerBound(), EPSILON);
    assertEquals(3d, range.getUpperBound(), EPSILON);

    double data2[][] = { {1,1,1}, {1,1,1}, {-1,-2,-3} };
    dataset.removeSeries("series1");
    dataset.addSeries("series2", data2);
    range = r.findZBounds(dataset);
    assertNotNull(range);
    assertEquals(-3d, range.getLowerBound(), EPSILON);
    assertEquals(-1d, range.getUpperBound(), EPSILON);

    double data3[][] = { {1,1,1}, {1,1,1}, {-1.2,2.9,3.8} };
    dataset.removeSeries("series2");
    dataset.addSeries("series3", data3);
    range = r.findZBounds(dataset);
    assertNotNull(range);
    assertEquals(-1.2d, range.getLowerBound(), EPSILON);
    assertEquals(3.8d, range.getUpperBound(), EPSILON);
}
 

/**
 * Some checks for the indexOf(Comparable) method.
 */
public void testIndexOf() {
    DefaultXYZDataset d = createSampleDataset1();
    assertEquals(0, d.indexOf("S1"));
    assertEquals(1, d.indexOf("S2"));
    assertEquals(-1, d.indexOf("Green Eggs and Ham"));
    assertEquals(-1, d.indexOf(null));
}
 

private XYZDataset createXYZDataset() {
  final DefaultXYZDataset xyzDataset = new DefaultXYZDataset();
  final double bs = 3;
  xyzDataset
    .addSeries( "First", new double[][] { { 1, 2, 3 }, { 2, 1, 3 }, { 0.1 * bs, 0.2 * bs, 0.1 * bs } } );// NON-NLS
  xyzDataset
    .addSeries( "Second", new double[][] { { 1, 2, 3 }, { 3, 0, 1 }, { 0.2 * bs, 0.1 * bs, 0.15 * bs } } );// NON-NLS
  return xyzDataset;
}
 

/**
 * A simple test for bug 1766646.
 */
@Test
public void testBug1766646A() {
    XYBlockRenderer r = new XYBlockRenderer();
    Range range = r.findDomainBounds(null);
    assertTrue(range == null);
    DefaultXYZDataset emptyDataset = new DefaultXYZDataset();
    range = r.findDomainBounds(emptyDataset);
    assertTrue(range == null);
}
 

/**
 * A simple test for bug 1766646.
 */
@Test
public void testBug1766646B() {
    XYBlockRenderer r = new XYBlockRenderer();
    Range range = r.findRangeBounds(null);
    assertTrue(range == null);
    DefaultXYZDataset emptyDataset = new DefaultXYZDataset();
    range = r.findRangeBounds(emptyDataset);
    assertTrue(range == null);
}
 

/**
 * Check if finding the bounds in z-dimension of an XYZDataset works. 
 */
public void testFindZBounds() {
    XYShapeRenderer r = new XYShapeRenderer();
    assertNull(r.findZBounds(null));

    DefaultXYZDataset dataset = new DefaultXYZDataset();
    Range range;

    double data1[][] = { {1,1,1}, {1,1,1}, {1,2,3} };
    dataset.addSeries("series1", data1);
    range = r.findZBounds(dataset);
    assertNotNull(range);
    assertEquals(1d, range.getLowerBound(), EPSILON);
    assertEquals(3d, range.getUpperBound(), EPSILON);

    double data2[][] = { {1,1,1}, {1,1,1}, {-1,-2,-3} };
    dataset.removeSeries("series1");
    dataset.addSeries("series2", data2);
    range = r.findZBounds(dataset);
    assertNotNull(range);
    assertEquals(-3d, range.getLowerBound(), EPSILON);
    assertEquals(-1d, range.getUpperBound(), EPSILON);

    double data3[][] = { {1,1,1}, {1,1,1}, {-1.2,2.9,3.8} };
    dataset.removeSeries("series2");
    dataset.addSeries("series3", data3);
    range = r.findZBounds(dataset);
    assertNotNull(range);
    assertEquals(-1.2d, range.getLowerBound(), EPSILON);
    assertEquals(3.8d, range.getUpperBound(), EPSILON);
}
 

/**
 * A simple test for bug 1766646.
 */
@Test
public void testBug1766646B() {
    XYBlockRenderer r = new XYBlockRenderer();
    Range range = r.findRangeBounds(null);
    assertTrue(range == null);
    DefaultXYZDataset emptyDataset = new DefaultXYZDataset();
    range = r.findRangeBounds(emptyDataset);
    assertTrue(range == null);
}
 

/**
 * A simple test for bug 1766646.
 */
@Test
public void testBug1766646B() {
    XYBlockRenderer r = new XYBlockRenderer();
    Range range = r.findRangeBounds(null);
    assertTrue(range == null);
    DefaultXYZDataset emptyDataset = new DefaultXYZDataset();
    range = r.findRangeBounds(emptyDataset);
    assertTrue(range == null);
}
 

/**
 * Check if finding the bounds in z-dimension of an XYZDataset works. 
 */
public void testFindZBounds() {
    XYShapeRenderer r = new XYShapeRenderer();
    assertNull(r.findZBounds(null));

    DefaultXYZDataset dataset = new DefaultXYZDataset();
    Range range;

    double data1[][] = { {1,1,1}, {1,1,1}, {1,2,3} };
    dataset.addSeries("series1", data1);
    range = r.findZBounds(dataset);
    assertNotNull(range);
    assertEquals(1d, range.getLowerBound(), EPSILON);
    assertEquals(3d, range.getUpperBound(), EPSILON);

    double data2[][] = { {1,1,1}, {1,1,1}, {-1,-2,-3} };
    dataset.removeSeries("series1");
    dataset.addSeries("series2", data2);
    range = r.findZBounds(dataset);
    assertNotNull(range);
    assertEquals(-3d, range.getLowerBound(), EPSILON);
    assertEquals(-1d, range.getUpperBound(), EPSILON);

    double data3[][] = { {1,1,1}, {1,1,1}, {-1.2,2.9,3.8} };
    dataset.removeSeries("series2");
    dataset.addSeries("series3", data3);
    range = r.findZBounds(dataset);
    assertNotNull(range);
    assertEquals(-1.2d, range.getLowerBound(), EPSILON);
    assertEquals(3.8d, range.getUpperBound(), EPSILON);
}
 

/**
 * A simple test for bug 1766646.
 */
@Test
public void testBug1766646A() {
    XYBlockRenderer r = new XYBlockRenderer();
    Range range = r.findDomainBounds(null);
    assertTrue(range == null);
    DefaultXYZDataset emptyDataset = new DefaultXYZDataset();
    range = r.findDomainBounds(emptyDataset);
    assertTrue(range == null);
}
 

/**
 * A simple test for bug 1766646.
 */
@Test
public void testBug1766646B() {
    XYBlockRenderer r = new XYBlockRenderer();
    Range range = r.findRangeBounds(null);
    assertTrue(range == null);
    DefaultXYZDataset emptyDataset = new DefaultXYZDataset();
    range = r.findRangeBounds(emptyDataset);
    assertTrue(range == null);
}
 

/**
 * Check if finding the bounds in z-dimension of an XYZDataset works. 
 */
public void testFindZBounds() {
    XYShapeRenderer r = new XYShapeRenderer();
    assertNull(r.findZBounds(null));

    DefaultXYZDataset dataset = new DefaultXYZDataset();
    Range range;

    double data1[][] = { {1,1,1}, {1,1,1}, {1,2,3} };
    dataset.addSeries("series1", data1);
    range = r.findZBounds(dataset);
    assertNotNull(range);
    assertEquals(1d, range.getLowerBound(), EPSILON);
    assertEquals(3d, range.getUpperBound(), EPSILON);

    double data2[][] = { {1,1,1}, {1,1,1}, {-1,-2,-3} };
    dataset.removeSeries("series1");
    dataset.addSeries("series2", data2);
    range = r.findZBounds(dataset);
    assertNotNull(range);
    assertEquals(-3d, range.getLowerBound(), EPSILON);
    assertEquals(-1d, range.getUpperBound(), EPSILON);

    double data3[][] = { {1,1,1}, {1,1,1}, {-1.2,2.9,3.8} };
    dataset.removeSeries("series2");
    dataset.addSeries("series3", data3);
    range = r.findZBounds(dataset);
    assertNotNull(range);
    assertEquals(-1.2d, range.getLowerBound(), EPSILON);
    assertEquals(3.8d, range.getUpperBound(), EPSILON);
}
 

/**
 * A simple test for bug 1766646.
 */
@Test
public void testBug1766646A() {
    XYBlockRenderer r = new XYBlockRenderer();
    Range range = r.findDomainBounds(null);
    assertTrue(range == null);
    DefaultXYZDataset emptyDataset = new DefaultXYZDataset();
    range = r.findDomainBounds(emptyDataset);
    assertTrue(range == null);
}
 

/**
 * A simple test for bug 1766646.
 */
@Test
public void testBug1766646B() {
    XYBlockRenderer r = new XYBlockRenderer();
    Range range = r.findRangeBounds(null);
    assertTrue(range == null);
    DefaultXYZDataset emptyDataset = new DefaultXYZDataset();
    range = r.findRangeBounds(emptyDataset);
    assertTrue(range == null);
}
 

/**
 * Check if finding the bounds in z-dimension of an XYZDataset works. 
 */
public void testFindZBounds() {
    XYShapeRenderer r = new XYShapeRenderer();
    assertNull(r.findZBounds(null));

    DefaultXYZDataset dataset = new DefaultXYZDataset();
    Range range;

    double data1[][] = { {1,1,1}, {1,1,1}, {1,2,3} };
    dataset.addSeries("series1", data1);
    range = r.findZBounds(dataset);
    assertNotNull(range);
    assertEquals(1d, range.getLowerBound(), EPSILON);
    assertEquals(3d, range.getUpperBound(), EPSILON);

    double data2[][] = { {1,1,1}, {1,1,1}, {-1,-2,-3} };
    dataset.removeSeries("series1");
    dataset.addSeries("series2", data2);
    range = r.findZBounds(dataset);
    assertNotNull(range);
    assertEquals(-3d, range.getLowerBound(), EPSILON);
    assertEquals(-1d, range.getUpperBound(), EPSILON);

    double data3[][] = { {1,1,1}, {1,1,1}, {-1.2,2.9,3.8} };
    dataset.removeSeries("series2");
    dataset.addSeries("series3", data3);
    range = r.findZBounds(dataset);
    assertNotNull(range);
    assertEquals(-1.2d, range.getLowerBound(), EPSILON);
    assertEquals(3.8d, range.getUpperBound(), EPSILON);
}
 

/**
 * A simple test for bug 1766646.
 */
@Test
public void testBug1766646A() {
    XYBlockRenderer r = new XYBlockRenderer();
    Range range = r.findDomainBounds(null);
    assertTrue(range == null);
    DefaultXYZDataset emptyDataset = new DefaultXYZDataset();
    range = r.findDomainBounds(emptyDataset);
    assertTrue(range == null);
}
 

/**
 * A check for the datasetIndex and seriesIndex fields in the LegendItem
 * returned by the getLegendItem() method.
 */
@Test
public void testGetLegendItemSeriesIndex() {
    DefaultXYZDataset d1 = new DefaultXYZDataset();
    double[] x = {2.1, 2.3, 2.3, 2.2, 2.2, 1.8, 1.8, 1.9, 2.3, 3.8};
    double[] y = {14.1, 11.1, 10.0, 8.8, 8.7, 8.4, 5.4, 4.1, 4.1, 25};
    double[] z = {2.4, 2.7, 2.7, 2.2, 2.2, 2.2, 2.1, 2.2, 1.6, 4};
    double[][] s1 = new double[][] {x, y, z};
    d1.addSeries("S1", s1);
    x = new double[] {2.1};
    y = new double[] {14.1};
    z = new double[] {2.4};
    double[][] s2 = new double[][] {x, y, z};
    d1.addSeries("S2", s2);

    DefaultXYZDataset d2 = new DefaultXYZDataset();
    x = new double[] {2.1};
    y = new double[] {14.1};
    z = new double[] {2.4};
    double[][] s3 = new double[][] {x, y, z};
    d2.addSeries("S3", s3);
    x = new double[] {2.1};
    y = new double[] {14.1};
    z = new double[] {2.4};
    double[][] s4 = new double[][] {x, y, z};
    d2.addSeries("S4", s4);
    x = new double[] {2.1};
    y = new double[] {14.1};
    z = new double[] {2.4};
    double[][] s5 = new double[][] {x, y, z};
    d2.addSeries("S5", s5);

    XYBubbleRenderer r = new XYBubbleRenderer();
    XYPlot plot = new XYPlot(d1, new NumberAxis("x"),
            new NumberAxis("y"), r);
    plot.setDataset(1, d2);
    /*JFreeChart chart =*/ new JFreeChart(plot);
    LegendItem li = r.getLegendItem(1, 2);
    assertEquals("S5", li.getLabel());
    assertEquals(1, li.getDatasetIndex());
    assertEquals(2, li.getSeriesIndex());
}