def make_graph_edges(list_of_cluster_data, tour_model):
for cd in list_of_cluster_data:
start_loc = cd['start']
end_loc = cd['end']
start_loc_temp = tm.Location(start_loc, tour_model)
start_loc_temp = tour_model.get_location(start_loc_temp)
end_loc_temp = tm.Location(end_loc, tour_model)
end_loc_temp = tour_model.get_location(end_loc_temp)
e = make_graph_edge(start_loc_temp, end_loc_temp, tour_model)
logging.debug("making edge %s" % e)
for trip_entry in cd["sections"]:
e.add_trip(trip_entry)
def populate_prob_field_for_locatons(list_of_cluster_data, tour_model):
for cd in list_of_cluster_data:
start_loc = cd['start']
end_loc = cd['end']
for model_trip in cd["sections"]:
start_loc_temp = tm.Location(start_loc, tour_model)
start_loc_temp = tour_model.get_location(start_loc_temp)
end_loc_temp = tm.Location(end_loc, tour_model)
end_loc_temp = tour_model.get_location(end_loc_temp)
com = tm.Commute(start_loc_temp, end_loc_temp)
tour_model.add_start_hour(start_loc_temp, model_trip.start_time)
start_loc_temp.increment_successor(end_loc_temp,
get_start_hour(model_trip),
get_day(model_trip))
def set_up(list_of_cluster_data, user_name):
time0 = datetime.datetime(1900, 1, 1, hour=0)
our_tour_model = tm.TourModel(user_name, 0, time0)
for dct in list_of_cluster_data:
start_name = dct['start']
end_name = dct['end']
start_coords = dct['start_coords']
end_coords = dct['end_coords']
for sec in dct['sections']:
start_loc = tm.Location(start_name, our_tour_model)
end_loc = tm.Location(end_name, our_tour_model)
our_tour_model.add_location(start_loc, start_coords)
our_tour_model.add_location(end_loc, end_coords)
return our_tour_model