diagnostics.py

import constants
from locations import School, Student
import matplotlib.pyplot as plt
import numpy as np
import pickle
from utils import stud_trav_time_array

#geocodes = open("data//all_geocodes.csv", "r")
#codes = []
#for code in geocodes.readlines():
#    codes.append(code.strip())

#Takes an index in the travel time matrix and returns the link with the
#latitude-longitude pair appended            
def append_to_link(link, tt_ind, slash=True):
    latlong = constants.GEOCODE_STRINGS[tt_ind].split(";")
    if slash:
        link += "/"
    link += latlong[0]
    link += ","
    link += latlong[1]
    return link

def printout(route):
    stops = route.stops
    schools = route.schools
    print("Estimated time: " + str(route.length/60) + " minutes.")
    stop_types = set()
    for s in stops:
        if s.e > 0:
            stop_types.add('E')
        if s.m > 0:
            stop_types.add('M')
        if s.h > 0:
            stop_types.add('H')
    type_printout = "Types of stops picked up: "
    if "E" in stop_types:
        type_printout += "elementary, "
    if "M" in stop_types:
        type_printout += "middle, "
    if "H" in stop_types:
        type_printout += "high, "
    type_printout = type_printout[:-2]
    print(type_printout)
    if route.bus != None:
        print("Bus capacity: " + str(route.bus.capacity))
    print("Number of assigned students: " + str(route.occupants))
    
    for i in range(len(stops)):
        if i == 0 or (i > 0 and stops[i].tt_ind != stops[i - 1].tt_ind):
            print(append_to_link("Go to latitude-longitude ", stops[i].tt_ind,
                                 slash=False))
        if stops[i].occs > 1:
            print("Pick up " + str(stops[i].occs) +
                  " students who go to " + stops[i].school.school_name)
        else:
            print("Pick up " + str(stops[i].occs) +
                  " student who goes to " + stops[i].school.school_name)
        for stud in stops[i].special_ed_students:
            print(stud)
    for i in range(len(schools)):
        if i == 0 or (i > 0 and schools[i].tt_ind != schools[i - 1].tt_ind):
            print(append_to_link("Go to latitude-longitude ", schools[i].tt_ind,
                                 slash=False))
        print("Drop off at " + schools[i].school_name)
            
    print("Google maps link: ")        
    printout_google_maps(route)
    print("Ending printout of route.")
    print("*********************************************************")
    print("*********************************************************")
    for i in range(3):
        print()
    
def google_maps_strings(route):
    maps_strings = []
    locs = route.stops + route.schools
    link = "https://www.google.com/maps/dir"
    appended = 0
    for i in range(len(locs)):
        if i == 0 or locs[i].tt_ind != locs[i-1].tt_ind:
            link = append_to_link(link, locs[i].tt_ind)
            appended += 1
            #Can only have 10 locations in a Maps link, so need to do another
            #link
            if appended == 10 and i < len(locs) - 1:
                maps_strings.append(link)
                link = "https://www.google.com/maps/dir"
                link = append_to_link(link, locs[i].tt_ind)
    maps_strings.append(link)
    return maps_strings
            
def printout_google_maps(route):
    strings_to_print = google_maps_strings(route)
    for (i, maps_string) in enumerate(strings_to_print):
        if i > 0:
            print("Need to start an additional link")
        print(maps_string)
    
def print_all(route_iter):
    for r in route_iter:
        printout(r)

def diagnostics(route_iter):
    route_list = []
    total_time = 0
    num_students = 0
    num_mixedload = 0
    num_singleload = 0
    buses_used = dict()
    trav_times = []
    for route in route_iter:
        trav_times.extend(route.student_travel_times())
        route_list.append(route)
        total_time += route.length
        if len(route.schools) > 1:
            num_mixedload += 1
        else:
            num_singleload += 1
        num_students += route.occupants
        if route.bus != None and route.bus.capacity in buses_used:
            buses_used[route.bus.capacity] += 1
        elif route.bus != None:
            buses_used[route.bus.capacity] = 1
    print("Number of routes: " + str(len(route_list)))
    print("Number of students picked up: " + str(num_students))
    for cap in sorted(buses_used.keys()):
        print("Used " + str(buses_used[cap]) + " buses of capacity " + str(cap))
    print("Number of mixed-load routes: " + str(num_mixedload))
    print("Number of single-load routes: " + str(num_singleload))
    print("Average estimated time per route: " +
          str(total_time/len(route_list)/60) + " minutes.")
    print("Mean student travel time: " + str(np.mean(np.array(trav_times))/60) + " minutes.")
    print("Ending printout of summary statistics.")
    print("*********************************************************")
    print("*********************************************************")
    for i in range(3):
        print()
        
    
    #visits most schools
    most_schools = 0
    most_schools_route = None
    for route in route_list:
        cur_schools = len(route.schools)
        if cur_schools > most_schools:
            most_schools = cur_schools
            most_schools_route = route
    print("Most schools on one route: " + str(most_schools))
    printout(most_schools_route)
    
    #visits most distinct locations
    most_locs = 0
    most_locs_route = None
    for route in route_list:
        cur_locs = route.stops + route.schools
        num_locs = 0
        for i in range(len(cur_locs)):
            if i == 0 or (cur_locs[i].tt_ind != cur_locs[i - 1].tt_ind):
                num_locs += 1
        if num_locs > most_locs:
            most_locs = num_locs
            most_locs_route = route
    print("Most distinct locations on one route: " + str(most_locs))
    printout(most_locs_route)
    
    #most riders while having mixed loads
    most_occs = 0
    most_occs_route = None
    for route in route_list:
        cur_occs = route.occupants
        cur_schools = len(route.schools)
        if cur_occs >= most_occs and cur_schools > 1:
            most_occs = cur_occs
            most_occs_route = route
            for j in range(len(route_list)):
                if route == route_list[j]:
                    print(j)
    print("Most occupants on a mixed-loads route: " + str(most_occs))
    printout(most_occs_route)
    
    #do all Vintage/Balboa routes
    #for r in route_list:
    #    to_print = False
    #    for school in r.schools:
    #        if ("VINTAGE" in school.school_name or
    #            ("BALBOA" in school.school_name and "LAKE" not in school.school_name)):
    #            to_print = True
    #    if to_print:
    #        print("Route that goes to Vintage/Balboa")
    #        printout(r)
    
def metrics(route_plan):
    print("Number of routes: " + str(len(route_plan)))
    lengths = np.array([r.length for r in route_plan])
    print("Mean route length: " + str(np.mean(lengths)))
    print("Stdev of route length: " + str(np.std(lengths)))
    trav_times = stud_trav_time_array(route_plan)
    print("Mean student travel time: " + str(np.mean(trav_times)))
    print("Stdev of student travel time: " + str(np.std(trav_times)))
    e_and_h = 0
    for r in route_plan:
        e_no_h = False
        h_no_e = False
        for stop in r.stops:
            if stop.e > 0 and stop.h == 0:
                e_no_h = True
            if stop.h > 0 and stop.e == 0:
                h_no_e = True
        if h_no_e and e_no_h:
            e_and_h += 1
    print("Number of routes which are not age-feasible: " + str(e_and_h))
    
def display_trav_times(route_plan, filename, rplanname):
    trav_times = stud_trav_time_array(route_plan)/60
    fig = plt.hist(trav_times,
                   bins = np.arange(0, np.max(trav_times)+5, 5))
    plt.title("Student travel time distribution (" + rplanname + ")")
    plt.xlabel("Travel time (m)")
    plt.ylabel("Number of students")
    plt.savefig(filename)
    
def display_utilization(route_plan, filename, rplanname):
    utilizations = []
    for r in route_plan:
        e = 0
        m = 0
        h = 0
        for stop in r.stops:
            for s in stop.students:
                if s.type == "E":
                    e += stop.ridership_probability()
                if s.type == "M":
                    m += stop.ridership_probability()
                if s.type == "H":
                    h += stop.ridership_probability()
        mod_caps = constants.CAPACITY_MODIFIED_MAP[r.unmodified_bus_capacity]
        utilizations.append(e/mod_caps[0] + m/mod_caps[1] + h/mod_caps[2])
    utilizations = np.array(utilizations)*100
    fig = plt.hist(utilizations,
                   bins = range(0, 110, 10))
    plt.title("Bus utilization percentage distribution (" + rplanname + ")")
    plt.xlabel("Percentage of capacity occupied")
    plt.ylabel("Number of buses")
    plt.savefig(filename)
    
#loading = open("output//0mdist.obj", "rb")
#obj = pickle.load(loading)
#diagnostics(obj)
#loading.close()