# load code / config
import os,sys;
sys.path.append(os.path.join((lambda r,f:f[0:f.index(r)+len(r)])('code',os.path.abspath(__file__)),'config'));
import config
config.epimodel()
config.numpy()
config.plot()
# external module
import re
import numpy as np
import matplotlib.pyplot as plt
from copy import deepcopy
# epi-model modules
import utils
import modelutils
from space import Space,Array
from elements import Color
# relative imports
import system
import sensitivity
names = ['S','I','T']
# names = ['S','infected']
lights = [0.6,0.0,0.3]
out = 'X'
n = 1
def make_pie(sim,select,phi):
plt.figure(figsize=(2,2))
selectors = [
sim.model.select[name].union(sim.model.select[select])
import os, sys
sys.path.append(
os.path.join(
(lambda r, f: f[0:f.index(r) + len(r)])('code',
os.path.abspath(__file__)),
'config'))
import config
import numpy as np
config.epimodel()
config.plot(tex=True)
import matplotlib.pyplot as plt
from collections import OrderedDict as odict
import modelutils
import elements
import system
def iterdur(min, max, N):
for i in np.logspace(np.log10(min), np.log10(max), N)[::-1]:
yield i
def get_sim(type, output, dh=None):
model = system.get_model()
if type == 'h**o':
model.collapse(['ii', 'ip'])
name = '$G = 1$ (Homogeneous)'
dt = 0.2
if type == 'zeta':
dhmax = 1.0 / model.params['mu']
model.params['dur'].update(