util.py

import csv
import meta

# read samples from data file
def read_all_data(file_path):
    with open(file_path) as f:
        reader = csv.reader(f, delimiter=',')

        is_header = True
        subjects = []

        for row in reader:
            if (is_header):
                # skip the header
                is_header = False
                continue
            subjects.append(row)

        return subjects

# doses are in weekly basis
def doses_in_same_range(d1, d2):
    thresh_lo = 21
    thresh_hi = 49
    return (
        (d1 < thresh_lo and d2 < thresh_lo) or
        (
            d1 >= thresh_lo and
            d1 <= thresh_hi and
            d2 >= thresh_lo and
            d2 <= thresh_hi
        ) or
        (d1 > thresh_hi and d2 > thresh_hi)
    )

def get_dose_range(dose):
    if (dose < 21):
        return meta.DOSE_LO
    if (21 <= dose and dose <= 49):
        return meta.DOSE_MD
    return meta.DOSE_HI

def has_gold_dose(sample):
    try:
        gold_dose = float(sample[meta.THERAPEUTIC_DOSE])
        return True
    except:
        return False
    return False