OpenMRS

Timestamp:

05/09/08 10:53:35 (2 months ago)

Author:

bwolfe

Message:

Merging report-api-refactoring to trunk [2696]:[4157]

Files:

• openmrs/trunk/src/api/org/openmrs/api/impl/CohortServiceImpl.java (modified) (2 diffs)

openmrs/trunk/src/api/org/openmrs/api/impl/CohortServiceImpl.java

@@ -14 +14 @@
-package org.openmrs.api.impl;
-
+import java.util.ArrayList;
+import java.util.Iterator;
+import java.util.LinkedHashMap;
-import java.util.List;
-
+
+
+
+
-import org.openmrs.Cohort;
-import org.openmrs.Patient;
+import org.openmrs.api.APIException;
-import org.openmrs.api.CohortService;
-import org.openmrs.api.context.Context;
-import org.openmrs.api.db.CohortDAO;
-
+
+
+
+
+
+
+
+
+
-public class CohortServiceImpl implements CohortService {
-
+        protected Log log = LogFactory.getLog(this.getClass());
-        private CohortDAO dao;
-        
-
+
+
+
+
-        
-        private CohortDAO getCohortDAO() {
@@ -104 +123 @@
-    }
-
+    /**
+     * Auto generated method comment
+     * 
+     * @param definitionClass
+     * @return
+     * @throws APIException
+     */
+    private CohortDefinitionProvider getCohortDefinitionProvider(Class<? extends CohortDefinition> definitionClass) throws APIException {
+        CohortDefinitionProvider ret = cohortDefinitionProviders.get(definitionClass);
+        if (ret == null)
+                throw new APIException("No CohortDefinitionProvider registered for " + definitionClass);
+        else
+                return ret;
+    }
+        
+        /**
+     * @see org.openmrs.api.CohortService#evaluate(org.openmrs.cohort.CohortDefinition, org.openmrs.report.EvaluationContext)
+     */
+    public Cohort evaluate(CohortDefinition definition, EvaluationContext evalContext) throws APIException {
+            CohortDefinitionProvider provider = getCohortDefinitionProvider(definition.getClass());
+            return provider.evaluate(definition, evalContext);
+    }
+
+
+
+        /**
+     * @see org.openmrs.api.CohortService#getAllPatientsCohortDefinition()
+     */
+    public CohortDefinition getAllPatientsCohortDefinition() {
+            PatientSearch ps = new PatientSearch();
+            ps.setFilterClass(PatientCharacteristicFilter.class);           
+            return ps;
+    }
+
+        /**
+     * @see org.openmrs.api.CohortService#getCohortDefinition(java.lang.Class, java.lang.Integer)
+     */
+    public CohortDefinition getCohortDefinition(Class<CohortDefinition> clazz, Integer id) {
+        CohortDefinitionProvider provider = getCohortDefinitionProvider(clazz);
+        return provider.getCohortDefinition(id);
+    }
+
+        /**
+     * @see org.openmrs.api.CohortService#getCohortDefinitions()
+     */
+    public List<CohortDefinition> getAllCohortDefinitions() {
+            List<CohortDefinition> ret = new ArrayList<CohortDefinition>();
+            for (CohortDefinitionProvider p : cohortDefinitionProviders.values()) {
+                ret.addAll(p.getAllCohortDefinitions());
+            }
+            return ret;
+    }
+
+        /**
+     * @see org.openmrs.api.CohortService#purgeCohortDefinition(org.openmrs.cohort.CohortDefinition)
+     */
+    public void purgeCohortDefinition(CohortDefinition definition) {
+        CohortDefinitionProvider provider = getCohortDefinitionProvider(definition.getClass());
+            provider.purgeCohortDefinition(definition);
+    }
+    
+    /**
+     * @see org.openmrs.api.CohortService#setCohortDefinitionProviders(Map)
+     */
+    public void setCohortDefinitionProviders(Map<Class<? extends CohortDefinition>, CohortDefinitionProvider> providerClassMap) {
+        for (Map.Entry<Class<? extends CohortDefinition>, CohortDefinitionProvider> entry : providerClassMap.entrySet()) {
+                registerCohortDefinitionProvider(entry.getKey(), entry.getValue());
+        }
+    }
+    
+    /**
+     * @see org.openmrs.api.CohortService#getCohortDefinitionProviders()
+     */
+    public Map<Class<? extends CohortDefinition>, CohortDefinitionProvider> getCohortDefinitionProviders() {
+        if (cohortDefinitionProviders == null)
+                cohortDefinitionProviders = new LinkedHashMap<Class<? extends CohortDefinition>, CohortDefinitionProvider>();
+        
+        return cohortDefinitionProviders;
+    }
+    
+    /**
+     * @see org.openmrs.api.CohortService#registerCohortDefinitionProvider(java.lang.Class, org.openmrs.api.CohortDefinitionService)
+     */
+    public void registerCohortDefinitionProvider(Class<? extends CohortDefinition> defClass, CohortDefinitionProvider cohortDefProvider) throws APIException {
+        getCohortDefinitionProviders().put(defClass, cohortDefProvider);
+    }
+    
+        /**
+     * @see org.openmrs.api.CohortService#removeCohortDefinitionProvider(java.lang.Class)
+     */
+    public void removeCohortDefinitionProvider(Class<? extends CohortDefinitionProvider> providerClass) {
+        
+        // TODO: should this be looking through the values or the keys?
+        for (Iterator<CohortDefinitionProvider> i = cohortDefinitionProviders.values().iterator(); i.hasNext(); ) {
+                if (i.next().getClass().equals(providerClass))
+                        i.remove();
+        }
+    }
+
+        /**
+     * @see org.openmrs.api.CohortService#saveCohortDefinition(org.openmrs.cohort.CohortDefinition)
+     */
+    public CohortDefinition saveCohortDefinition(CohortDefinition definition) throws APIException {
+        CohortDefinitionProvider provider = getCohortDefinitionProvider(definition.getClass());
+        return provider.saveCohortDefinition(definition);
+    }
+
+        /**
+     * @see org.openmrs.api.CohortService#getCohortDefinitions(java.lang.Class)
+     */
+    public List<CohortDefinition> getCohortDefinitions(Class providerClass) {
+        CohortDefinitionProvider provider = getCohortDefinitionProvider(providerClass);
+            return provider.getAllCohortDefinitions();
+    }
-}