From f50ce6d15b3dcba6869b5592ecf5923fa3de989d Mon Sep 17 00:00:00 2001
From: Luis Aleixo <luis.aleixo@cern.ch>
Date: Wed, 15 Sep 2021 10:51:34 +0200
Subject: [PATCH] Updated fraction_deposited value according to the diameter

---
 cara/models.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/cara/models.py b/cara/models.py
index 367866f9..8c5ceee1 100644
--- a/cara/models.py
+++ b/cara/models.py
@@ -915,7 +915,10 @@ class ExposureModel:
     repeats: int = 1
 
     #: The fraction of viruses actually deposited in the respiratory tract
-    fraction_deposited: _VectorisedFloat = 0.6
+    d = 1.37 # To be updated in the future.
+    IF = 1 - 0.5 * (1 - (1 / (1 + (0.00076*(d**2.8)))))
+    DF = IF * (0.0587 + (0.911/(1 + np.exp(4.77 + 1.485 * np.log(d)))) + (0.943/(1 + np.exp(0.508 - 2.58 * np.log(d)))))
+    fraction_deposited: _VectorisedFloat = DF
 
     def _normed_exposure(self) -> _VectorisedFloat:
         """
@@ -936,7 +939,6 @@ class ExposureModel:
 
     def infection_probability(self) -> _VectorisedFloat:
         exposure = self.exposure()
-        print('oi', self.concentration_model.infected.virus.viable_to_RNA)
 
         # Dose
         inf_aero = (