package er.bugtracker.components.reporting;
import java.text.Format;
import java.util.Enumeration;
import org.apache.log4j.Logger;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.axis.CategoryLabelPositions;
import org.jfree.data.category.CategoryDataset;
import org.jfree.data.category.DefaultCategoryDataset;
import org.jfree.data.general.Dataset;
import org.jfree.data.general.DefaultPieDataset;
import org.jfree.data.general.PieDataset;
import com.webobjects.appserver.WOContext;
import com.webobjects.foundation.NSArray;
import com.webobjects.foundation.NSDictionary;
import com.webobjects.foundation.NSMutableDictionary;
import com.webobjects.foundation.NSTimestamp;
import er.extensions.foundation.ERXDictionaryUtilities;
import er.grouping.DRCriteria;
import er.grouping.DRGroup;
import er.grouping.DRRecordGroup;
import er.grouping.DRSubMasterCriteria;
import er.grouping.DRValue;
import er.plot.ERPCategoryChart;
import er.plot.ERPPieChart;
public class ChartReport extends Report {
/** logging support */
private static final Logger log = Logger.getLogger(ChartReport.class);
public String currentType;
public ChartConfiguration selectedConfiguration = (ChartConfiguration) CHART_CONFIGURATIONS.lastObject();
public ChartConfiguration currentConfiguration;
public String selectedType = selectedConfiguration.defaultType();
private static NSArray CHART_CONFIGURATIONS = new NSArray(new Object[] {
new PieChartConfiguration(),
new AreaChartConfiguration()
});
public NSArray chartConfigurations() {
return CHART_CONFIGURATIONS;
}
public static interface ChartConfiguration {
public NSArray supportedTypes();
public NSDictionary configuration();
public String componentName();
public Dataset createDataset();
public String defaultType();
}
public static class PieChartConfiguration implements ChartConfiguration {
public NSArray supportedTypes() {
return ERPPieChart.SUPPORTED_TYPES;
}
public NSDictionary configuration() {
return null;
}
public String componentName() {
return ERPPieChart.class.getName();
}
public Dataset createDataset() {
return new ERPPieChart.AccumulatingPieDataset();
}
public String defaultType() {
return "PieChart";
}
}
public static class AreaChartConfiguration implements ChartConfiguration {
public NSArray supportedTypes() {
return ERPCategoryChart.SUPPORTED_TYPES;
}
public NSDictionary configuration() {
return ERXDictionaryUtilities.dictionaryWithObjectsAndKeys(new Object[] {
CategoryLabelPositions.UP_45, "categoryPlot.domainAxis.categoryLabelPositions"
});
}
public String componentName() {
return ERPCategoryChart.class.getName();
}
public Dataset createDataset() {
return new DefaultCategoryDataset();
}
public String defaultType() {
return "StackedAreaChart";
}
}
/**
* Public constructor
* @param context the context
*/
public ChartReport(WOContext context) {
super(context);
}
@Override
public void awake() {
super.awake();
}
public boolean isCurrentTypeSelected() {
return currentType == selectedType;
}
public void selectType() {
selectedType = currentType;
selectedConfiguration = currentConfiguration;
}
public JFreeChart chart() {
// NOTE: AK this is a dummy implementation, because ERPChart.chart will setValueForBinding("chart"), which will
// try to validate the binding, which in turn calls valueForKey("chart"), which would finally choke and call
// handleQueryForUnboundKey if we didn't return null here. Duh.
return null;
}
public void setChart(JFreeChart chart) {
// chart.getCategoryPlot().getDomainAxis().setCategoryLabelPositions();
}
public Dataset dataset(Integer attribTotalKey) {
Dataset dataset = selectedConfiguration.createDataset();
NSArray dataList = model().vList();
NSArray valueList = model().hList();
if(true) {
dataList = model().hList();
valueList = model().vList();
}
DRGroup dayGroup = (DRGroup) dataList.objectAtIndex(0);
DRSubMasterCriteria daySubMasterCriteria = (DRSubMasterCriteria) dayGroup.masterCriteria().subCriteriaList().objectAtIndex(0);
String dayKey = daySubMasterCriteria.keyDesc();
NSArray dayCriterias = dayGroup.sortedCriteriaList();
Format dayFormat = DRCriteria.formatterForFormat(daySubMasterCriteria.format());
NSMutableDictionary coordinates = new NSMutableDictionary();
for(Enumeration en = valueList.objectEnumerator(); en.hasMoreElements();) {
DRGroup mainGroup = (DRGroup)en.nextElement();
DRSubMasterCriteria mainSubMasterCriteria = (DRSubMasterCriteria) mainGroup.masterCriteria().subCriteriaList().lastObject();
String mainKey = mainSubMasterCriteria.keyDesc();
NSArray mainCriterias = mainGroup.sortedCriteriaList();
for (Enumeration values = mainCriterias.objectEnumerator(); values.hasMoreElements();) {
DRCriteria mainCriteria = (DRCriteria)values.nextElement();
String label = (String) mainCriteria.valueDict().objectForKey(mainKey);
coordinates.setObjectForKey(mainCriteria, mainGroup.masterCriteria().keyDesc());
if(!mainCriteria.isTotal()) {
for (Enumeration days = dayCriterias.objectEnumerator(); days.hasMoreElements();) {
DRCriteria dayCriteria = (DRCriteria)days.nextElement();
Object o = dayCriteria.valueDict().objectForKey(dayKey);
if(!dayCriteria.isTotal()) {
// we just want the actual times, not the totals
if(o instanceof NSTimestamp) {
NSTimestamp ts = (NSTimestamp)o;
if(ts != null) {
double value = 0.0D;
String dayLabel = dayFormat.format(ts);
coordinates.setObjectForKey(dayCriteria, dayGroup.masterCriteria().keyDesc());
DRRecordGroup recordGroup = model().recordGroupForCoordinates(coordinates);
if(attribTotalKey != null) {
DRValue drValue = (DRValue) recordGroup.totals().objectForKey(attribTotalKey);
if(drValue != null) {
value = drValue.total();
}
} else {
value = ((Number) recordGroup.rawRecordList().valueForKeyPath(recordGroupTotalToShow())).doubleValue();
}
if(dataset instanceof CategoryDataset) {
((DefaultCategoryDataset)dataset).setValue(value, label, dayLabel);
} else if (dataset instanceof PieDataset) {
((DefaultPieDataset) dataset).setValue(label, value);
} else {
throw new IllegalArgumentException("Neither a Pie nor CategoryDataset");
}
}
} else {
double value = 0.0D;
coordinates.setObjectForKey(dayCriteria, dayGroup.masterCriteria().keyDesc());
DRRecordGroup recordGroup = model().recordGroupForCoordinates(coordinates);
if(attribTotalKey != null) {
DRValue drValue = (DRValue) recordGroup.totals().objectForKey(attribTotalKey);
if(drValue != null) {
value = drValue.total();
}
} else {
value = ((Number) recordGroup.rawRecordList().valueForKeyPath(recordGroupTotalToShow())).doubleValue();
}
if(dataset instanceof CategoryDataset) {
((DefaultCategoryDataset)dataset).setValue(value, label, o.toString());
} else if (dataset instanceof PieDataset) {
((DefaultPieDataset) dataset).setValue(label, value);
} else {
throw new IllegalArgumentException("Neither a Pie nor CategoryDataset");
}
}
}
}
}
}
}
return dataset;
}
public Dataset datasetForAttribute() {
Integer attribTotalKey = Integer.valueOf(model().flatAttributeList().indexOfObject(attrib));
return dataset(attribTotalKey);
}
public Dataset dataset() {
return dataset(null);
}
}