pingu_98/cara
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

rework concentration distribution calculation

Browse source
This commit is contained in:
markus 2021-02-04 13:51:57 +01:00
parent bdb2185678
commit 57839595fc
1 changed files with 14 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
cara/montecarlo.py
View file

@ -158,15 +158,26 @@ class MCInfectedPopulation(MCPopulation):
return lognormal(*br_params) return lognormal(*br_params)
@functools.lru_cache() @functools.lru_cache()
def _concentration_distribution(self) -> typing.Callable: def _concentration_distribution_without_mask(self) -> typing.Callable:
if self.expiratory_activity == 1: if self.expiratory_activity == 1:
return concentration_vs_diameter[0] return concentration_vs_diameter[0]
if self.expiratory_activity == 2: if self.expiratory_activity == 2:
return np.vectorize(lambda d: concentration_vs_diameter[1](d) + concentration_vs_diameter[2](d)) return np.vectorize(lambda d: sum(f(d) for f in concentration_vs_diameter))
if self.expiratory_activity == 3: if self.expiratory_activity == 3:
return np.vectorize(lambda d: 5 * (concentration_vs_diameter[1](d) + concentration_vs_diameter[2](d))) return np.vectorize(lambda d: 5 * sum(f(d) for f in concentration_vs_diameter))
@functools.lru_cache()
def _concentration_distribution_with_mask(self) -> typing.Callable:
function = self._concentration_distribution_without_mask()
return np.vectorize(lambda d: function(d) * mask_leak_out(d))
def calculate_emission_concentration(self) -> float:
function = self._concentration_distribution_with_mask if self.masked else \
self._concentration_distribution_without_mask()
function = np.vectorize(lambda d: function(d) * np.pi * d ** 3 / 6.0)
return quad_vec(function, 0, 30)[0]
def emission_rate_when_present(self) -> np.ndarray: def emission_rate_when_present(self) -> np.ndarray:
""" """