public boolean hasRegistryTotalLines() {
Iterator<Column> it = registryConfig.getColumns().iterator();
boolean found = false;
while (it.hasNext() && !found) {
Column column = it.next();
String sfO = column.getSummaryFunction();
if (sfO != null && sfO.equalsIgnoreCase("sum")) {
found = true;
}
}
return found;
}
private void addSumRows(IDataStore dataStore) {
logger.debug("IN");
summaryColorCellsArray = new JSONArray();
summaryCellsArray = new JSONArray();
ArrayList<Integer> columnsIndexToMerge = new ArrayList<Integer>();
ArrayList<Integer> columnsIndexToEmpty = new ArrayList<Integer>();
ArrayList<Integer> columnsIndexToAfter = new ArrayList<Integer>();
HashMap<Integer, Object> columnsIndexToSum2Counter = new HashMap<Integer, Object>();
// collect columns to merge and columns to sum and colummsn to empty:
// -- columns to merge have merge attributes until a columns with summaryFunc is found
// then other columns that have merge attribute but no
List<Column> columns = registryConfig.getColumns();
Integer index = 0;
boolean summaryFuncFound = false;
boolean measureFound = false;
for (Iterator iterator = columns.iterator(); iterator.hasNext(); ) {
Column column = (Column) iterator.next();
if (column.isMerge() && summaryFuncFound == false) {
columnsIndexToMerge.add(index);
} else if (summaryFuncFound == true && !column.isMeasure() && !measureFound) {
columnsIndexToEmpty.add(index);
} else if (summaryFuncFound == true && !column.isMeasure() && measureFound) {
columnsIndexToAfter.add(index);
} else if (column.isMeasure()) {
columnsIndexToSum2Counter.put(index, 0);
measureFound = true;
}
if (column.getSummaryFunction() != null && column.getSummaryFunction().equals("sum"))
summaryFuncFound = true;
index++;
}
// Map to store previous merge values on iteration
HashMap<Integer, Object> previousMergeValues = new HashMap<Integer, Object>();
for (Iterator iterator = columnsIndexToMerge.iterator(); iterator.hasNext(); ) {
Integer columnIndex = (Integer) iterator.next();
previousMergeValues.put(columnIndex, null);
}
TreeMap<Integer, Record> recordsToAddMap = new TreeMap<Integer, Record>();
int sumCounter = 0; // add total row only if grouping has more than one member
// iterate on each store row
for (int i = 0; i < dataStore.getRecordsCount(); i++) {
IRecord record = dataStore.getRecordAt(i);
// get current values of column to merge
HashMap<Integer, Object> currentMergeValues = new HashMap<Integer, Object>();
// iterate on each column to merge and store values
for (Iterator iterator = columnsIndexToMerge.iterator(); iterator.hasNext(); ) {
Integer columnIndex = (Integer) iterator.next();
Object value = record.getFieldAt(columnIndex).getValue();
currentMergeValues.put(columnIndex, value);
}
// compare current values with previous ones
boolean isEqual = compareValuesMaps(previousMergeValues, currentMergeValues);
// if merging goes on update counters else add summarization line
if (isEqual) {
sumCounter++;
for (Iterator iterator = columnsIndexToSum2Counter.keySet().iterator();
iterator.hasNext(); ) {
Integer indexMeasure = (Integer) iterator.next();
Object value = record.getFieldAt(indexMeasure).getValue();
// TODO treat the case this is not a number, should keep it to null
if (value != null) {
// get previous value
Object result = operateWithNumbers(columnsIndexToSum2Counter.get(indexMeasure), value);
columnsIndexToSum2Counter.put(indexMeasure, result);
} else {
columnsIndexToSum2Counter.put(indexMeasure, null);
}
}
} else {
// breaking point, add summarization lines at previous index; i-1
// add a new record only if sumCounter > 0
if (sumCounter > 0) {
addTotalRecord(
dataStore,
i,
columnsIndexToMerge,
columnsIndexToEmpty,
columnsIndexToAfter,
columnsIndexToSum2Counter,
previousMergeValues,
recordsToAddMap);
}
// put the counters to actual values
for (Iterator iterator = columnsIndexToSum2Counter.keySet().iterator();
iterator.hasNext(); ) {
Integer columnInd = (Integer) iterator.next();
Object v = record.getFieldAt(columnInd).getValue();
columnsIndexToSum2Counter.put(columnInd, v);
}
sumCounter = 0;
}
// update previousValues
previousMergeValues = currentMergeValues;
}
// add final total if last records were merged
if (sumCounter > 0) {
addTotalRecord(
dataStore,
null,
columnsIndexToMerge,
columnsIndexToEmpty,
columnsIndexToAfter,
columnsIndexToSum2Counter,
previousMergeValues,
recordsToAddMap);
}
// finally add the record (could not add them while cycling the store)
for (Iterator iterator = recordsToAddMap.keySet().iterator(); iterator.hasNext(); ) {
Integer indexR = (Integer) iterator.next();
Record rec = recordsToAddMap.get(indexR);
if (indexR == -1) {
dataStore.appendRecord(rec);
} else {
dataStore.insertRecord(indexR, rec);
}
}
logger.debug("OUT");
}