/**
* Adds the column headers and column data to the DataSet
*
* @param dataSet
* @param encounters
* @param patientIdentifierTypes
* @param optionalColumns
* @param columnDisplayFormat
* @param maxColumnHeaderWidth
* @param allColumns
* @param fieldMap
* @return
*/
public DataSet addData(
SimpleDataSet dataSet,
List<Encounter> encounters,
List<PatientIdentifierType> patientIdentifierTypes,
List<EncounterAndObsDataSetDefinition.ObsOptionalColumn> optionalColumns,
List<EncounterAndObsDataSetDefinition.ColumnDisplayFormat> columnDisplayFormat,
Integer maxColumnHeaderWidth,
Set<ObsColumnDescriptor> allColumns,
Map<Encounter, Map<ObsColumnDescriptor, Obs>> fieldMap) {
for (Encounter encounter : encounters) {
DataSetRow row = new DataSetRow();
List<String> providerNames = new ArrayList<String>();
for (EncounterProvider ep : encounter.getEncounterProviders()) {
providerNames.add(ep.getProvider().getName());
}
// Add the standard columns for encounters
DataSetColumn c1 =
new DataSetColumn(
ObjectUtil.trimStringIfNeeded("ENCOUNTER_ID", maxColumnHeaderWidth),
ObjectUtil.trimStringIfNeeded("ENCOUNTER_ID", maxColumnHeaderWidth),
Integer.class);
row.addColumnValue(c1, encounter.getEncounterId());
DataSetColumn c2 =
new DataSetColumn(
ObjectUtil.trimStringIfNeeded("ENCOUNTER_DATETIME", maxColumnHeaderWidth),
ObjectUtil.trimStringIfNeeded("ENCOUNTER_DATETIME", maxColumnHeaderWidth),
String.class);
row.addColumnValue(c2, encounter.getEncounterDatetime().toString());
DataSetColumn c3 =
new DataSetColumn(
ObjectUtil.trimStringIfNeeded("LOCATION", maxColumnHeaderWidth),
ObjectUtil.trimStringIfNeeded("LOCATION", maxColumnHeaderWidth),
String.class);
row.addColumnValue(
c3, (encounter.getLocation() != null) ? encounter.getLocation().getName() : EMPTY);
DataSetColumn c4 =
new DataSetColumn(
ObjectUtil.trimStringIfNeeded("PROVIDER", maxColumnHeaderWidth),
ObjectUtil.trimStringIfNeeded("PROVIDER", maxColumnHeaderWidth),
String.class);
row.addColumnValue(c4, OpenmrsUtil.join(providerNames, ", "));
DataSetColumn c5 =
new DataSetColumn(
ObjectUtil.trimStringIfNeeded("INTERNAL_PATIENT_ID", maxColumnHeaderWidth),
ObjectUtil.trimStringIfNeeded("INTERNAL_PATIENT_ID", maxColumnHeaderWidth),
Integer.class);
row.addColumnValue(
c5, encounter.getPatient() != null ? encounter.getPatient().getPatientId() : EMPTY);
if (patientIdentifierTypes != null) {
for (PatientIdentifierType pit : patientIdentifierTypes) {
List<PatientIdentifier> patientIdentifiers =
encounter.getPatient().getPatientIdentifiers(pit);
StringBuffer sbPatientIdentifiers = new StringBuffer();
int count = 0;
for (PatientIdentifier patientIdentifier : patientIdentifiers) {
if (count > 0) {
sbPatientIdentifiers.append(", ");
}
sbPatientIdentifiers.append(patientIdentifier.toString());
count++;
}
DataSetColumn c6 =
new DataSetColumn(
pit.getName(),
ObjectUtil.trimStringIfNeeded(pit.getName(), maxColumnHeaderWidth),
String.class);
row.addColumnValue(c6, sbPatientIdentifiers.toString());
}
}
Map<ObsColumnDescriptor, Obs> obsInEncounter = fieldMap.get(encounter);
// Look up all obs for a given encounter based on column headers for all encounters
for (ObsColumnDescriptor columnKey : allColumns) {
Obs obs = obsInEncounter.get(columnKey);
String columnName = columnKey.format(columnDisplayFormat, maxColumnHeaderWidth);
DataSetColumn obsDsc = new DataSetColumn(columnName, columnName, String.class);
StringBuffer columnValue = new StringBuffer();
if (obs != null && obs.getValueCoded() != null) {
if (columnDisplayFormat.contains(
EncounterAndObsDataSetDefinition.ColumnDisplayFormat.ID)) {
columnValue.append(obs.getValueCoded());
}
if (columnDisplayFormat.contains(EncounterAndObsDataSetDefinition.ColumnDisplayFormat.ID)
&& columnDisplayFormat.contains(
EncounterAndObsDataSetDefinition.ColumnDisplayFormat.BEST_SHORT_NAME)) {
columnValue.append("_");
}
if (columnDisplayFormat.contains(
EncounterAndObsDataSetDefinition.ColumnDisplayFormat.BEST_SHORT_NAME)) {
String conceptName = obs.getValueAsString(Context.getLocale());
columnValue.append(
maxColumnHeaderWidth != null
&& conceptName.length() > maxColumnHeaderWidth - columnValue.length()
? conceptName.substring(0, maxColumnHeaderWidth - columnValue.length() - 1)
: conceptName);
}
row.addColumnValue(obsDsc, (obs != null) ? columnValue.toString() : EMPTY);
} else {
row.addColumnValue(
obsDsc, (obs != null) ? obs.getValueAsString(Context.getLocale()) : EMPTY);
}
String dateColumnName =
columnKey.format(
columnDisplayFormat,
maxColumnHeaderWidth != null ? maxColumnHeaderWidth - 5 : null);
DataSetColumn obsDscDate =
new DataSetColumn(dateColumnName + "_DATE", dateColumnName + "_DATE", String.class);
row.addColumnValue(obsDscDate, (obs != null) ? obs.getObsDatetime().toString() : EMPTY);
String parentColumnName =
columnKey.format(
columnDisplayFormat,
maxColumnHeaderWidth != null ? maxColumnHeaderWidth - 7 : null);
DataSetColumn obsDscParent =
new DataSetColumn(
parentColumnName + "_PARENT", parentColumnName + "_PARENT", String.class);
row.addColumnValue(
obsDscParent,
(obs != null && obs.getObsGroup() != null) ? obs.getObsGroup().getId() : EMPTY);
if (optionalColumns != null) {
if (optionalColumns.contains(
EncounterAndObsDataSetDefinition.ObsOptionalColumn.VALUE_MODIFIER)) {
String valModColumnName =
columnKey.format(
columnDisplayFormat,
maxColumnHeaderWidth != null ? maxColumnHeaderWidth - 10 : null);
DataSetColumn obsDscValueModifier =
new DataSetColumn(
valModColumnName + "_VALUE_MOD", valModColumnName + "_VALUE_MOD", String.class);
row.addColumnValue(obsDscValueModifier, (obs != null) ? obs.getValueModifier() : EMPTY);
}
if (optionalColumns.contains(
EncounterAndObsDataSetDefinition.ObsOptionalColumn.ACCESSION_NUMBER)) {
String accessionNumColumnName =
columnKey.format(
columnDisplayFormat,
maxColumnHeaderWidth != null ? maxColumnHeaderWidth - 14 : null);
DataSetColumn obsDscAccessionNumber =
new DataSetColumn(
accessionNumColumnName + "_ACCESSION_NUM",
accessionNumColumnName + "_ACCESSION_NUM",
String.class);
row.addColumnValue(
obsDscAccessionNumber, (obs != null) ? obs.getAccessionNumber() : EMPTY);
}
if (optionalColumns.contains(
EncounterAndObsDataSetDefinition.ObsOptionalColumn.COMMENT)) {
String commentColumnName =
columnKey.format(
columnDisplayFormat,
maxColumnHeaderWidth != null ? maxColumnHeaderWidth - 8 : null);
DataSetColumn obsDscComment =
new DataSetColumn(
commentColumnName + "_COMMENT", commentColumnName + "_COMMENT", String.class);
row.addColumnValue(obsDscComment, (obs != null) ? obs.getComment() : EMPTY);
}
}
}
dataSet.addRow(row);
}
return dataSet;
}
/** @see DataSetEvaluator#evaluate(DataSetDefinition, EvaluationContext) */
@SuppressWarnings("unchecked")
public DataSet evaluate(DataSetDefinition dataSetDefinition, EvaluationContext context)
throws EvaluationException {
PatientDataSetDefinition dsd = (PatientDataSetDefinition) dataSetDefinition;
context = ObjectUtil.nvl(context, new EvaluationContext());
SimpleDataSet dataSet = new SimpleDataSet(dsd, context);
dataSet.setSortCriteria(dsd.getSortCriteria());
// Construct a new EvaluationContext based on the passed filters
Cohort c = context.getBaseCohort();
if (dsd.getRowFilters() != null) {
for (Mapped<? extends CohortDefinition> q : dsd.getRowFilters()) {
Cohort s = Context.getService(CohortDefinitionService.class).evaluate(q, context);
c = CohortUtil.intersectNonNull(c, s);
}
}
if (c == null) {
c =
Context.getService(CohortDefinitionService.class)
.evaluate(new AllPatientsCohortDefinition(), context);
}
EvaluationContext ec = context.shallowCopy();
if (!CohortUtil.areEqual(ec.getBaseCohort(), c)) {
ec.setBaseCohort(c);
}
// Evaluate each specified ColumnDefinition for all of the included rows and add these to the
// dataset
for (RowPerObjectColumnDefinition cd : dsd.getColumnDefinitions()) {
if (log.isDebugEnabled()) {
log.debug("Evaluating column: " + cd.getName());
log.debug(
"With Data Definition: "
+ DefinitionUtil.format(cd.getDataDefinition().getParameterizable()));
log.debug("With Mappings: " + cd.getDataDefinition().getParameterMappings());
log.debug("With Parameters: " + ec.getParameterValues());
}
StopWatch sw = new StopWatch();
sw.start();
MappedData<? extends PatientDataDefinition> dataDef =
(MappedData<? extends PatientDataDefinition>) cd.getDataDefinition();
EvaluatedPatientData data =
Context.getService(PatientDataService.class).evaluate(dataDef, ec);
for (Integer id : c.getMemberIds()) {
for (DataSetColumn column : cd.getDataSetColumns()) {
Object val = data.getData().get(id);
val = DataUtil.convertData(val, dataDef.getConverters());
dataSet.addColumnValue(id, column, val);
}
}
sw.stop();
if (log.isDebugEnabled()) {
log.debug("Evaluated column. Duration: " + sw.toString());
}
}
return dataSet;
}