use of org.activityinfo.model.query.ColumnModel in project activityinfo by bedatadriven.
the class PivotAdapter method executeIndicatorValuesQuery.
private void executeIndicatorValuesQuery(final Activity activity, LinkedActivity linkedActivity) throws SQLException {
// Double check that this activity has not been deleted
if (isDeleted(activity, linkedActivity)) {
return;
}
// Query the SOURCE form tree
FormTree formTree = formTrees.get(linkedActivity.getLeafFormClassId());
Preconditions.checkNotNull(formTree, "No form tree for form " + linkedActivity.getLeafFormClassId());
QueryModel queryModel = new QueryModel(linkedActivity.getLeafFormClassId());
List<ActivityField> indicators = selectedIndicators(activity);
// Add Indicators to the query
// Keep track of alias to destination map
final Multimap<String, ActivityField> aliasToIndicator = HashMultimap.create();
for (ActivityField indicator : indicators) {
Collection<Integer> sourceIndicatorIds = linkedActivity.getSourceIndicatorIdsFor(indicator.getId());
for (Integer sourceIndicatorId : sourceIndicatorIds) {
String alias = "I" + sourceIndicatorId;
queryModel.selectExpr(fieldExpression(sourceIndicatorId)).as(alias);
aliasToIndicator.put(alias, indicator);
}
}
// These are the columns we will use for grouping
for (DimBinding dimension : groupBy) {
for (ColumnModel columnModel : dimension.getColumnQuery(formTree)) {
queryModel.addColumn(columnModel);
}
}
// if any of the indicators are "site count" indicators, then we need
// to query the site id as well
addSiteIdToQuery(activity, queryModel);
// declare the filter
queryModel.setFilter(composeFilter(formTree));
// Query the table
enqueueQuery(queryModel, new Function<ColumnSet, Void>() {
@Nullable
@Override
public Void apply(@Nullable ColumnSet columnSet) {
// Now add the results to the buckets
int[] siteIds = extractSiteIds(columnSet);
DimensionCategory[][] categories = extractCategories(activity, columnSet);
for (String sourceAlias : aliasToIndicator.keySet()) {
ColumnView measureView = columnSet.getColumnView(sourceAlias);
// A single source indicator may be mapped to multiple destination Indicators
for (ActivityField destinationIndicator : aliasToIndicator.get(sourceAlias)) {
DimensionCategory indicatorCategory = null;
if (indicatorDimension.isPresent()) {
indicatorCategory = indicatorDimension.get().category(destinationIndicator);
}
for (int i = 0; i < columnSet.getNumRows(); i++) {
if (destinationIndicator.getAggregation() == IndicatorDTO.AGGREGATE_SITE_COUNT) {
Map<Dimension, DimensionCategory> key = bucketKey(i, categories, indicatorCategory);
Accumulator bucket = bucketForKey(key, destinationIndicator.getAggregation());
bucket.addSite(siteIds[i]);
} else {
double value = measureView.getDouble(i);
if (!Double.isNaN(value) && !Double.isInfinite(value)) {
Map<Dimension, DimensionCategory> key = bucketKey(i, categories, indicatorCategory);
if (command.getValueType() == PivotSites.ValueType.INDICATOR) {
Accumulator bucket = bucketForKey(key, destinationIndicator.getAggregation());
bucket.addValue(value);
} else {
Accumulator bucket = bucketForKey(key, IndicatorDTO.AGGREGATE_SITE_COUNT);
bucket.addSite(siteIds[i]);
}
}
}
}
}
}
return null;
}
});
}
Aggregations