package org.pentaho.plugin.jfreereport.reportcharts;
import org.jfree.chart.urls.CategoryURLGenerator;
import org.jfree.data.category.CategoryDataset;
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 FormulaCategoryURLGenerator implements CategoryURLGenerator
{
private FormulaExpression formulaExpression;
private ExpressionRuntime runtime;
private static final String[] ADDITIONAL_COLUMN_KEYS = new String[]{
"chart::series-key", "chart::category-key",
"chart::series-index", "chart::category-index",
"chart::series-keys", "chart::category-keys",
"chart::value"
};
public FormulaCategoryURLGenerator(final ExpressionRuntime runtime,
final String formula)
{
this.runtime = runtime;
this.formulaExpression = new FormulaExpression();
this.formulaExpression.setFormula(formula);
}
/**
* Returns a URL for one item in a dataset. As a guideline, the URL
* should be valid within the context of an XHTML 1.0 document. Classes
* that implement this interface are responsible for correctly escaping
* any text that is derived from the dataset, as this may be user-specified
* and could pose a security risk.
*
* @param dataset the dataset.
* @param series the series (zero-based index).
* @param category the category.
*
* @return A string containing the URL.
*/
public String generateURL(final CategoryDataset dataset, final int series, final int category)
{
try
{
final Comparable seriesKey = dataset.getRowKey(series);
final Comparable categoryKey = dataset.getColumnKey(category);
final Object[] categoryKeys = dataset.getColumnKeys().toArray();
final Object[] seriesKeys = dataset.getRowKeys().toArray();
final Object value = dataset.getValue(series, category);
final Object[] values = new Object[]{
seriesKey, categoryKey,
IntegerCache.getInteger(series), IntegerCache.getInteger(category),
seriesKeys, categoryKeys,
value
};
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);
}
}
}