Allow for agents to die directly (without hospitalization)

AMReX-Agent · Jan 28, 2025 · 8ccb8f1 · 8ccb8f1
1 parent 933109a
commit 8ccb8f1
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/src/DiseaseParm.H b/src/DiseaseParm.H
@@ -115,6 +115,10 @@ struct DiseaseParm
         {Real(0), Real(0), Real(0), Real(0), Real(0), Real(0.26)},
         {Real(0.20), Real(0.20), Real(0.20), Real(0.45), Real(0.45), Real(1.0)}};
 
+    /*! Probabilites of dying while *not* in hospital for age groups */
+    Real m_unhospToDeath[AgeGroups::total] = {
+        Real(0), Real(0), Real(0), Real(0), Real(0), Real(0)};
+
 
     DiseaseParm(const std::string& dname) {
         memset(disease_name, 0, 50);

diff --git a/src/DiseaseParm.cpp b/src/DiseaseParm.cpp
@@ -94,6 +94,7 @@ void DiseaseParm::readInputs ( const std::string& a_pp_str /*!< Parmparse string
     queryArray(pp, "CHR", m_CHR, AgeGroups::total);
     queryArray(pp, "CIC", m_CIC, AgeGroups::total);
     queryArray(pp, "CVE", m_CVE, AgeGroups::total);
+    queryArray(pp, "unhospCVF", m_unhospToDeath, AgeGroups::total);
     queryArray(pp, "hospCVF", m_hospToDeath[DiseaseStats::hospitalization], AgeGroups::total);
     queryArray(pp, "icuCVF", m_hospToDeath[DiseaseStats::ICU], AgeGroups::total);
     queryArray(pp, "ventCVF", m_hospToDeath[DiseaseStats::ventilator], AgeGroups::total);

diff --git a/src/HospitalModel.H b/src/HospitalModel.H
@@ -132,6 +132,12 @@ void HospitalModel<PCType, PTDType, PType>::treatAgents(PCType& a_agents, /*!< A
                 {
                     if ( !inHospital(i, ptd) )  {
                         // agent is not in hospital
+                        // determine if agent dies anyway.
+                        if (Random(engine) < disease_parm_d->m_unhospToDeath[age_group_ptr[i]]) {
+                            is_alive_ptr[i] = 0;
+                            flag_status_ptr[i] *= -1;
+                            status_ptrs[d][i] = Status::dead;
+                        }
                         return;
                     }
                     if (counter_ptrs[d][i] == Math::floor(incubation_per_ptrs[d][i])) {