def generate_donors(d, patients, weight_dists):
donors = []
d_id = 0
for patient in patients:
n_donors = patient.draw_number_of_donors()
for i in range(n_donors):
if random.random() < 0.5:
sex = "F"
age_d = d["donor_age_dist_f"]
else:
sex = "M"
age_d = d["donor_age_dist_m"]
age_dist = DiscreteDist(age_d["ages"], age_d["probs"])
age = age_dist.sample1()
bt_d = d["donor_bt_dist"]
bt_dist = DiscreteDist(bt_d["bts"], bt_d["probs"])
bt = bt_dist.sample1()
weight = weight_dists.draw_weight(sex, age)
donor = Donor(sex, age, bt, weight, d_id, patient)
donors.append(donor)
d_id += 1
return donors
def generate_patients(d, n_patients, weight_dists):
patients = []
for p_id in range(n_patients):
if rand() < d["prob_cystic_fibrosis"]:
disease = "CF"
dd = d["cystic_f_dists"]
else:
disease = "PH"
dd = d["pulm_h_dists"]
if rand() < dd["prob_female"]:
gender = "F"
age_d = dd["age_dist_f"]
else:
gender = "M"
age_d = dd["age_dist_m"]
age_dist = DiscreteDist(age_d["age_bands"], age_d["probs"])
age_band = age_dist.sample1()
age = randint(age_band["min"], age_band["max"])
bt_d = dd["bt_dist"]
bt_dist = DiscreteDist(bt_d["bts"], bt_d["probs"])
bt = bt_dist.sample1()
weight = weight_dists.draw_weight(gender, age)
patient = Patient(disease, gender, age, bt, weight, p_id)
patients.append(patient)
return patients