123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302 |
- /* ===========================================================
- * JFreeChart : a free chart library for the Java(tm) platform
- * ===========================================================
- *
- * (C) Copyright 2000-2013, by Object Refinery Limited 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.]
- *
- * -------------------
- * GanttChartTest.java
- * -------------------
- * (C) Copyright 2005-2013, by Object Refinery Limited.
- *
- * Original Author: David Gilbert (for Object Refinery Limited);
- * Contributor(s): -;
- *
- * Changes:
- * --------
- * 12-Apr-2005 : Version 1 (DG);
- *
- */
- package org.jfree.chart;
- import java.awt.Graphics2D;
- import java.awt.geom.Rectangle2D;
- import java.awt.image.BufferedImage;
- import java.util.Calendar;
- import java.util.Date;
- import org.jfree.chart.event.ChartChangeEvent;
- import org.jfree.chart.event.ChartChangeListener;
- import org.jfree.chart.labels.CategoryToolTipGenerator;
- import org.jfree.chart.labels.StandardCategoryToolTipGenerator;
- import org.jfree.chart.plot.CategoryPlot;
- import org.jfree.chart.renderer.category.CategoryItemRenderer;
- import org.jfree.chart.urls.CategoryURLGenerator;
- import org.jfree.chart.urls.StandardCategoryURLGenerator;
- import org.jfree.data.category.IntervalCategoryDataset;
- import org.jfree.data.gantt.Task;
- import org.jfree.data.gantt.TaskSeries;
- import org.jfree.data.gantt.TaskSeriesCollection;
- import org.jfree.data.time.SimpleTimePeriod;
- import org.junit.Before;
- import org.junit.Test;
- import static org.junit.Assert.fail;
- import static org.junit.Assert.assertEquals;
- import static org.junit.Assert.assertSame;
- /**
- * Some tests for a Gantt chart.
- */
- public class GanttChartTest {
- /** A chart. */
- private JFreeChart chart;
- /**
- * Common test setup.
- */
- @Before
- public void setUp() {
- this.chart = createGanttChart();
- }
- /**
- * Draws the chart with a <code>null</code> info object to make sure that
- * no exceptions are thrown (a problem that was occurring at one point).
- */
- @Test
- public void testDrawWithNullInfo() {
- try {
- BufferedImage image = new BufferedImage(200 , 100,
- BufferedImage.TYPE_INT_RGB);
- Graphics2D g2 = image.createGraphics();
- this.chart.draw(g2, new Rectangle2D.Double(0, 0, 200, 100), null,
- null);
- g2.dispose();
- }
- catch (Exception e) {
- fail("There should be no exception.");
- }
- }
- /**
- * Draws the chart with a <code>null</code> info object to make sure that
- * no exceptions are thrown (a problem that was occurring at one point).
- */
- @Test
- public void testDrawWithNullInfo2() {
- JFreeChart chart = createGanttChart();
- CategoryPlot plot = (CategoryPlot) chart.getPlot();
- plot.setDataset(createDataset());
- /* BufferedImage img =*/ chart.createBufferedImage(300, 200, null);
- //FIXME we should really assert a value
- }
- /**
- * Replaces the chart's dataset and then checks that the new dataset is OK.
- */
- @Test
- public void testReplaceDataset() {
- LocalListener l = new LocalListener();
- this.chart.addChangeListener(l);
- CategoryPlot plot = (CategoryPlot) this.chart.getPlot();
- plot.setDataset(null);
- assertEquals(true, l.flag);
- }
- /**
- * Check that setting a tool tip generator for a series does override the
- * default generator.
- */
- @Test
- public void testSetSeriesToolTipGenerator() {
- CategoryPlot plot = (CategoryPlot) this.chart.getPlot();
- CategoryItemRenderer renderer = plot.getRenderer();
- StandardCategoryToolTipGenerator tt
- = new StandardCategoryToolTipGenerator();
- renderer.setSeriesToolTipGenerator(0, tt);
- CategoryToolTipGenerator tt2 = renderer.getToolTipGenerator(0, 0);
- assertSame(tt2, tt);
- }
- /**
- * Check that setting a URL generator for a series does override the
- * default generator.
- */
- @Test
- public void testSetSeriesURLGenerator() {
- CategoryPlot plot = (CategoryPlot) this.chart.getPlot();
- CategoryItemRenderer renderer = plot.getRenderer();
- StandardCategoryURLGenerator url1
- = new StandardCategoryURLGenerator();
- renderer.setSeriesItemURLGenerator(0, url1);
- CategoryURLGenerator url2 = renderer.getItemURLGenerator(0, 0);
- assertSame(url2, url1);
- }
- /**
- * Create a Gantt chart.
- *
- * @return The chart.
- */
- private static JFreeChart createGanttChart() {
- return ChartFactory.createGanttChart(
- "Gantt Chart",
- "Domain", "Range",
- null,
- true, // include legend
- true,
- true
- );
- }
- /**
- * Creates a sample dataset for a Gantt chart.
- *
- * @return The dataset.
- */
- public static IntervalCategoryDataset createDataset() {
- TaskSeries s1 = new TaskSeries("Scheduled");
- s1.add(new Task("Write Proposal",
- new SimpleTimePeriod(date(1, Calendar.APRIL, 2001),
- date(5, Calendar.APRIL, 2001))));
- s1.add(new Task("Obtain Approval",
- new SimpleTimePeriod(date(9, Calendar.APRIL, 2001),
- date(9, Calendar.APRIL, 2001))));
- s1.add(new Task("Requirements Analysis",
- new SimpleTimePeriod(date(10, Calendar.APRIL, 2001),
- date(5, Calendar.MAY, 2001))));
- s1.add(new Task("Design Phase",
- new SimpleTimePeriod(date(6, Calendar.MAY, 2001),
- date(30, Calendar.MAY, 2001))));
- s1.add(new Task("Design Signoff",
- new SimpleTimePeriod(date(2, Calendar.JUNE, 2001),
- date(2, Calendar.JUNE, 2001))));
- s1.add(new Task("Alpha Implementation",
- new SimpleTimePeriod(date(3, Calendar.JUNE, 2001),
- date(31, Calendar.JULY, 2001))));
- s1.add(new Task("Design Review",
- new SimpleTimePeriod(date(1, Calendar.AUGUST, 2001),
- date(8, Calendar.AUGUST, 2001))));
- s1.add(new Task("Revised Design Signoff",
- new SimpleTimePeriod(date(10, Calendar.AUGUST, 2001),
- date(10, Calendar.AUGUST, 2001))));
- s1.add(new Task("Beta Implementation",
- new SimpleTimePeriod(date(12, Calendar.AUGUST, 2001),
- date(12, Calendar.SEPTEMBER, 2001))));
- s1.add(new Task("Testing",
- new SimpleTimePeriod(date(13, Calendar.SEPTEMBER, 2001),
- date(31, Calendar.OCTOBER, 2001))));
- s1.add(new Task("Final Implementation",
- new SimpleTimePeriod(date(1, Calendar.NOVEMBER, 2001),
- date(15, Calendar.NOVEMBER, 2001))));
- s1.add(new Task("Signoff",
- new SimpleTimePeriod(date(28, Calendar.NOVEMBER, 2001),
- date(30, Calendar.NOVEMBER, 2001))));
- TaskSeries s2 = new TaskSeries("Actual");
- s2.add(new Task("Write Proposal",
- new SimpleTimePeriod(date(1, Calendar.APRIL, 2001),
- date(5, Calendar.APRIL, 2001))));
- s2.add(new Task("Obtain Approval",
- new SimpleTimePeriod(date(9, Calendar.APRIL, 2001),
- date(9, Calendar.APRIL, 2001))));
- s2.add(new Task("Requirements Analysis",
- new SimpleTimePeriod(date(10, Calendar.APRIL, 2001),
- date(15, Calendar.MAY, 2001))));
- s2.add(new Task("Design Phase",
- new SimpleTimePeriod(date(15, Calendar.MAY, 2001),
- date(17, Calendar.JUNE, 2001))));
- s2.add(new Task("Design Signoff",
- new SimpleTimePeriod(date(30, Calendar.JUNE, 2001),
- date(30, Calendar.JUNE, 2001))));
- s2.add(new Task("Alpha Implementation",
- new SimpleTimePeriod(date(1, Calendar.JULY, 2001),
- date(12, Calendar.SEPTEMBER, 2001))));
- s2.add(new Task("Design Review",
- new SimpleTimePeriod(date(12, Calendar.SEPTEMBER, 2001),
- date(22, Calendar.SEPTEMBER, 2001))));
- s2.add(new Task("Revised Design Signoff",
- new SimpleTimePeriod(date(25, Calendar.SEPTEMBER, 2001),
- date(27, Calendar.SEPTEMBER, 2001))));
- s2.add(new Task("Beta Implementation",
- new SimpleTimePeriod(date(27, Calendar.SEPTEMBER, 2001),
- date(30, Calendar.OCTOBER, 2001))));
- s2.add(new Task("Testing",
- new SimpleTimePeriod(date(31, Calendar.OCTOBER, 2001),
- date(17, Calendar.NOVEMBER, 2001))));
- s2.add(new Task("Final Implementation",
- new SimpleTimePeriod(date(18, Calendar.NOVEMBER, 2001),
- date(5, Calendar.DECEMBER, 2001))));
- s2.add(new Task("Signoff",
- new SimpleTimePeriod(date(10, Calendar.DECEMBER, 2001),
- date(11, Calendar.DECEMBER, 2001))));
- TaskSeriesCollection collection = new TaskSeriesCollection();
- collection.add(s1);
- collection.add(s2);
- return collection;
- }
- /**
- * Utility method for creating <code>Date</code> objects.
- *
- * @param day the date.
- * @param month the month.
- * @param year the year.
- *
- * @return a date.
- */
- private static Date date(int day, int month, int year) {
- Calendar calendar = Calendar.getInstance();
- calendar.set(year, month, day);
- Date result = calendar.getTime();
- return result;
- } /**
- * A chart change listener.
- *
- */
- static class LocalListener implements ChartChangeListener {
- /** A flag. */
- private boolean flag;
- /**
- * Event handler.
- *
- * @param event the event.
- */
- @Override
- public void chartChanged(ChartChangeEvent event) {
- this.flag = true;
- }
- }
- }
|