package org.pentaho.plugin.jfreereport.reportcharts;
import org.jfree.chart.urls.PieURLGenerator;
import org.jfree.data.general.PieDataset;
import org.pentaho.reporting.engine.classic.core.function.ExpressionRuntime;
import org.pentaho.reporting.engine.classic.core.function.FormulaExpression;
import org.pentaho.reporting.engine.classic.core.function.WrapperExpressionRuntime;
import org.pentaho.reporting.engine.classic.core.states.datarow.StaticDataRow;
import org.pentaho.reporting.engine.classic.core.util.IntegerCache;
/**
* Todo: Document me!
*
* @author Thomas Morgner.
*/
public class FormulaPieURLGenerator implements PieURLGenerator
{
private FormulaExpression formulaExpression;
private ExpressionRuntime runtime;
private static final String[] ADDITIONAL_COLUMN_KEYS = new String[]{
"chart::key", "chart::keys",
"chart::item", "chart::items",
"chart::pie-index"
};
public FormulaPieURLGenerator(final ExpressionRuntime runtime,
final String formula)
{
this.runtime = runtime;
this.formulaExpression = new FormulaExpression();
this.formulaExpression.setFormula(formula);
}
/**
* Generates a URL for one item in a {@link org.jfree.data.general.PieDataset}. As a guideline,
* the URL should be valid within the context of an XHTML 1.0 document.
*
* @param dataset the dataset (<code>null</code> not permitted).
* @param key the item key (<code>null</code> not permitted).
* @param pieIndex the pie index (differentiates between pies in a
* 'multi' pie chart).
* @return A string containing the URL.
*/
public String generateURL(final PieDataset dataset, final Comparable key, final int pieIndex)
{
try
{
final Object[] keys = dataset.getKeys().toArray();
final Object[] items = new Object[keys.length];
for (int i = 0; i < keys.length; i++)
{
items[i] = dataset.getValue(key);
}
final Object[] values = new Object[]{
key, keys,
IntegerCache.getInteger(dataset.getIndex(key)), items,
IntegerCache.getInteger(pieIndex)
};
formulaExpression.setRuntime(new WrapperExpressionRuntime
(new StaticDataRow(ADDITIONAL_COLUMN_KEYS, values), runtime));
final Object o = formulaExpression.getValue();
if (o == null)
{
return null;
}
return String.valueOf(o);
}
finally
{
formulaExpression.setRuntime(null);
}
}
}