/*
* JasperReports - Free Java Reporting Library.
* Copyright (C) 2001 - 2009 Jaspersoft Corporation. All rights reserved.
* http://www.jaspersoft.com
*
* Unless you have purchased a commercial license agreement from Jaspersoft,
* the following license terms apply:
*
* This program is part of JasperReports.
*
* JasperReports 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 3 of the License, or
* (at your option) any later version.
*
* JasperReports 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 JasperReports. If not, see <http://www.gnu.org/licenses/>.
*/
package net.sf.jasperreports.charts.fill;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import net.sf.jasperreports.charts.JRGanttDataset;
import net.sf.jasperreports.charts.JRGanttSeries;
import net.sf.jasperreports.charts.util.CategoryLabelGenerator;
import net.sf.jasperreports.engine.JRChartDataset;
import net.sf.jasperreports.engine.JRExpressionCollector;
import net.sf.jasperreports.engine.design.JRVerifier;
import net.sf.jasperreports.engine.fill.JRCalculator;
import net.sf.jasperreports.engine.fill.JRExpressionEvalException;
import net.sf.jasperreports.engine.fill.JRFillChartDataset;
import net.sf.jasperreports.engine.fill.JRFillObjectFactory;
import net.sf.jasperreports.engine.util.Pair;
import org.jfree.data.gantt.Task;
import org.jfree.data.gantt.TaskSeries;
import org.jfree.data.gantt.TaskSeriesCollection;
import org.jfree.data.general.Dataset;
import org.jfree.data.time.SimpleTimePeriod;
/**
* @author Peter Risko (peter@risko.hu)
* @version $Id: JRFillGanttDataset.java 3938 2010-08-19 14:59:36Z teodord $
*/
public class JRFillGanttDataset extends JRFillChartDataset implements JRGanttDataset {
/**
*
*/
protected JRFillGanttSeries[] ganttSeries;
private List seriesNames;
private Map seriesMap;
private Map labelsMap;
private Map itemHyperlinks;
/**
*
*/
public JRFillGanttDataset(
JRGanttDataset ganttDataset,
JRFillObjectFactory factory
)
{
super(ganttDataset, factory);
/* */
JRGanttSeries[] srcGanttSeries = ganttDataset.getSeries();
if (srcGanttSeries != null && srcGanttSeries.length > 0)
{
ganttSeries = new JRFillGanttSeries[srcGanttSeries.length];
for(int i = 0; i < ganttSeries.length; i++)
{
ganttSeries[i] = (JRFillGanttSeries)factory.getGanttSeries(srcGanttSeries[i]);
}
}
}
/**
*
*/
public JRGanttSeries[] getSeries()
{
return ganttSeries;
}
/**
*
*/
protected void customInitialize()
{
seriesNames = null;
seriesMap = null;
labelsMap = null;
itemHyperlinks = null;
}
/**
*
*/
protected void customEvaluate(JRCalculator calculator) throws JRExpressionEvalException
{
if (ganttSeries != null && ganttSeries.length > 0)
{
for(int i = 0; i < ganttSeries.length; i++)
{
ganttSeries[i].evaluate(calculator);
}
}
}
/**
*
*/
protected void customIncrement()
{
if (ganttSeries != null && ganttSeries.length > 0)
{
if (seriesNames == null)
{
seriesNames = new ArrayList();
seriesMap = new HashMap();
labelsMap = new HashMap();
itemHyperlinks = new HashMap();
}
for(int i = 0; i < ganttSeries.length; i++)
{
JRFillGanttSeries crtGanttSeries = ganttSeries[i];
Comparable seriesName = crtGanttSeries.getSeries();
TaskSeries taskSrs = (TaskSeries)seriesMap.get(seriesName);
if (taskSrs == null)
{
taskSrs = new TaskSeries((String)seriesName);
seriesNames.add(seriesName);
seriesMap.put(seriesName, taskSrs);
}
// create task
Task task = taskSrs.get(crtGanttSeries.getTask());
if(task == null) {
task = new Task(crtGanttSeries.getTask(),
crtGanttSeries.getStartDate(),
crtGanttSeries.getEndDate());
taskSrs.add(task);
}
// create subtask
Task subtask = new Task(crtGanttSeries.getSubtask(),
crtGanttSeries.getStartDate(),
crtGanttSeries.getEndDate());
// NOTE: For correct scaling/plotting JFreeChart needs the subtasks
// to be 'inside' of the containing task.
// Therefore the earliest subtask startvalue
// is set as startvalue for the whole task, and the
// latest subtask endvalue set as endvalue for the
// whole task.
if(subtask.getDuration().getStart().before(task.getDuration().getStart())) {
task.setDuration(new SimpleTimePeriod(subtask.getDuration().getStart(), task.getDuration().getEnd()));
}
if(subtask.getDuration().getEnd().after(task.getDuration().getEnd())) {
task.setDuration(new SimpleTimePeriod(task.getDuration().getStart(), subtask.getDuration().getEnd()));
}
Number percent = crtGanttSeries.getPercent();
if (percent != null)
{
subtask.setPercentComplete(percent.doubleValue());
}
task.addSubtask(subtask);
if (crtGanttSeries.getLabelExpression() != null)
{
Map seriesLabels = (Map)labelsMap.get(seriesName);
if (seriesLabels == null)
{
seriesLabels = new HashMap();
labelsMap.put(seriesName, seriesLabels);
}
// TODO: is it OK like this?
//seriesLabels.put(crtXySeries.getXValue(), crtXySeries.getLabel());
seriesLabels.put(crtGanttSeries.getTask(), crtGanttSeries.getLabel());
}
if (crtGanttSeries.hasItemHyperlinks())
{
Map seriesLinks = (Map) itemHyperlinks.get(seriesName);
if (seriesLinks == null)
{
seriesLinks = new HashMap();
itemHyperlinks.put(seriesName, seriesLinks);
}
// TODO: ?? not sure how to do
//Pair xyKey = new Pair(crtXySeries.getXValue(), crtXySeries.getYValue());
//seriesLinks.put(xyKey, crtXySeries.getPrintItemHyperlink());
Pair taskSubtaskKey = new Pair(crtGanttSeries.getTask(), crtGanttSeries.getSubtask());
seriesLinks.put(taskSubtaskKey, crtGanttSeries.getPrintItemHyperlink());
}
}
}
}
/**
*
*/
public Dataset getCustomDataset()
{
TaskSeriesCollection dataset = new TaskSeriesCollection();
if (seriesNames != null)
{
for(int i = 0; i < seriesNames.size(); i++)
{
Comparable seriesName = (Comparable)seriesNames.get(i);
dataset.add((TaskSeries)seriesMap.get(seriesName));
}
}
return dataset;
}
/**
*
*/
public byte getDatasetType() {
return JRChartDataset.GANTT_DATASET;
}
/**
*
*/
public Object getLabelGenerator(){
return new CategoryLabelGenerator(labelsMap);
}
/**
*
*/
public void collectExpressions(JRExpressionCollector collector)
{
collector.collect(this);
}
public Map getItemHyperlinks()
{
return itemHyperlinks;
}
public boolean hasItemHyperlinks()
{
boolean foundLinks = false;
if (ganttSeries != null && ganttSeries.length > 0)
{
for (int i = 0; i < ganttSeries.length && !foundLinks; i++)
{
JRFillGanttSeries series = ganttSeries[i];
foundLinks = series.hasItemHyperlinks();
}
}
return foundLinks;
}
public void validate(JRVerifier verifier)
{
verifier.verify(this);
}
}