org.openmrs.module.kenyaemr.calculation.library.mchms.LateEnrollmentCalculation.java Source code

Introduction

Here is the source code for org.openmrs.module.kenyaemr.calculation.library.mchms.LateEnrollmentCalculation.java

Source

/**
 * The contents of this file are subject to the OpenMRS Public License
 * Version 1.0 (the "License"); you may not use this file except in
 * compliance with the License. You may obtain a copy of the License at
 * http://license.openmrs.org
 *
 * Software distributed under the License is distributed on an "AS IS"
 * basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
 * License for the specific language governing rights and limitations
 * under the License.
 *
 * Copyright (C) OpenMRS, LLC.  All Rights Reserved.
 */

package org.openmrs.module.kenyaemr.calculation.library.mchms;

import org.joda.time.DateTime;
import org.joda.time.Weeks;
import org.openmrs.Encounter;
import org.openmrs.EncounterType;
import org.openmrs.Obs;
import org.openmrs.Program;
import org.openmrs.calculation.patient.PatientCalculationContext;
import org.openmrs.calculation.result.CalculationResultMap;
import org.openmrs.module.kenyacore.calculation.AbstractPatientCalculation;
import org.openmrs.module.kenyacore.calculation.BooleanResult;
import org.openmrs.module.kenyacore.calculation.Calculations;
import org.openmrs.module.kenyacore.calculation.Filters;
import org.openmrs.module.kenyaemr.wrapper.EncounterWrapper;
import org.openmrs.module.metadatadeploy.MetadataUtils;
import org.openmrs.module.kenyaemr.Dictionary;
import org.openmrs.module.kenyaemr.metadata.MchMetadata;

import java.util.Collection;
import java.util.Map;
import java.util.Set;

/**
 * Calculates whether a mother enrolled into the program at gestation greater than 28 weeks.
 * Calculation returns true if mother is alive, enrolled in the MCH program and had gestation
 * greater than 28 weeks at enrollment.
 */
public class LateEnrollmentCalculation extends AbstractPatientCalculation {

    /**
     * @see org.openmrs.calculation.patient.PatientCalculation#evaluate(java.util.Collection, java.util.Map, org.openmrs.calculation.patient.PatientCalculationContext)
     */
    @Override
    public CalculationResultMap evaluate(Collection<Integer> cohort, Map<String, Object> parameterValues,
            PatientCalculationContext context) {

        Program mchmsProgram = MetadataUtils.existing(Program.class, MchMetadata._Program.MCHMS);

        // Get all patients who are alive and in MCH-MS program
        Set<Integer> alive = Filters.alive(cohort, context);
        Set<Integer> inMchmsProgram = Filters.inProgram(mchmsProgram, alive, context);

        CalculationResultMap ret = new CalculationResultMap();
        CalculationResultMap crm = Calculations.lastEncounter(
                MetadataUtils.existing(EncounterType.class, MchMetadata._EncounterType.MCHMS_ENROLLMENT), cohort,
                context);
        for (Integer ptId : cohort) {
            // Is patient alive and in MCH program?
            boolean lateEnrollment = false;
            if (inMchmsProgram.contains(ptId)) {
                lateEnrollment = gestationAtEnrollmentWasGreaterThan28Weeks(ptId, crm);
            }
            ret.put(ptId, new BooleanResult(lateEnrollment, this, context));
        }
        return ret;
    }

    /**
     * @return true if the given patient's gestation at enrollment was greater than 28 weeks at enrollment and false
     * otherwise.
     * */
    protected boolean gestationAtEnrollmentWasGreaterThan28Weeks(Integer patientId, CalculationResultMap crm) {
        Encounter lastMchEnrollment = (Encounter) crm.get(patientId).getValue();
        EncounterWrapper wrapper = new EncounterWrapper(lastMchEnrollment);

        Obs lmpObs = wrapper.firstObs(Dictionary.getConcept(Dictionary.LAST_MONTHLY_PERIOD));
        if (lmpObs != null) {
            Weeks weeks = Weeks.weeksBetween(new DateTime(lmpObs.getValueDate()),
                    new DateTime(lastMchEnrollment.getDateCreated()));
            if (weeks.getWeeks() > 28) {
                return true;
            }
        }
        return false;
    }
}