示例#1
0
def add_report_to_tracker(tracker_id, report):
    logger.info('Adding report to tracker_id "{}": {}'.format(tracker_id, report))
    day = set_tracker_day(tracker_id, report)
    
    # update train position
    if report.get_my_loc():
        update_coords(report, tracker_id)
    
    is_updated_stop_time = stop_detector.add_report(tracker_id, report)
    
    if is_updated_stop_time:
        logger.info('stop_time updated')
        stop_times = stop_detector.get_detected_stop_times(tracker_id)
        trip_delays_ids_list_of_lists = update_trips(tracker_id, day, stop_times)
        trip_ids = get_trusted_trips(trip_delays_ids_list_of_lists)
        for trip_id in trip_ids:
            trip_delays_ids_list = [x for x in trip_delays_ids_list_of_lists if x[0][1] == trip_id][0]
            if (len(stop_times)-1) in trip_delays_ids_list[0][2]:
                if len(trip_delays_ids_list[0][2]) == 2:  # Need to add the first station
                    if (len(stop_times)-2) in trip_delays_ids_list[0][2]:
                        stop_time = stop_times[trip_delays_ids_list[0][2][-2]]
                        logger.debug(stop_time)
                        save_stop_times_to_db(tracker_id, stop_time, trip_id) 
                    else:
                        logger.error('Two stops were detected for trip, last detected stop for tracker belongs to trip, but one before last does not, while it most probably should.')
                stop_time = stop_times[trip_delays_ids_list[0][2][-1]]
                logger.debug(stop_time)
                save_stop_times_to_db(tracker_id, stop_time, trip_id)
示例#2
0
    def _stop_detector_on_real_trip(self, device_id='1cb87f1e', do_preload_reports=True, set_reports_to_same_weekday_last_week=True, do_show_fig=False):
        remove_from_redis([device_id])
        now = ot_utils.get_localtime_now()
        reports_queryset = get_device_id_reports(device_id)
        tracker_id = device_id
        if do_show_fig:
            display_utils.draw_map()

        fps_period_start = time.clock()
        fps_period_length = 100
        if do_preload_reports:
            reports_queryset = list(reports_queryset)
        count = len(reports_queryset) if isinstance(
            reports_queryset, list) else reports_queryset.count()
        for i in xrange(count):
            if i % fps_period_length == 0:
                elapsed = (time.clock() - fps_period_start)
                if elapsed > 0:
                    logger.debug('%d\t%.1f qps' %
                                 (i, fps_period_length / elapsed))
                else:
                    logger.debug(
                        'Elapsed time should be positive but is %d' % (elapsed))
                fps_period_start = time.clock()

            report = reports_queryset[i]

            if set_reports_to_same_weekday_last_week:
                # fix finding same weekday last week by
                # http://stackoverflow.com/questions/6172782/find-the-friday-of-previous-last-week-in-python
                day_fix = (now.weekday() - report.timestamp.weekday()) % 7
                day = now + datetime.timedelta(days=-day_fix)
                # move day and correct for DST (daylight savings time)
                dst_before = report.get_timestamp_israel_time().dst()
                report.timestamp = report.timestamp.replace(
                    year=day.year, month=day.month, day=day.day)
                dst_after = report.get_timestamp_israel_time().dst()
                report.timestamp -= dst_after - dst_before
            if do_show_fig:
                plt.scatter(report.my_loc.lat, report.my_loc.lon)
                plt.show()
            #print i, ot_utils.get_localtime(report.timestamp)
            is_updated_stop_time = add_report(tracker_id, report)
            if is_updated_stop_time:
                logger.debug('stop_time updated')

        stop_detector.print_tracked_stop_times(device_id)
        detected_stop_times = stop_detector.get_detected_stop_times(device_id)
        ground_truth_stops = stop_detector_ground_truth.data[device_id]
        for x, y in zip(detected_stop_times, ground_truth_stops):
            self.assertEquals(x.__str__(), y.__str__())

        remove_from_redis([device_id])
        print 'done'
        return tracker_id
示例#3
0
 def evaluate_detected_stop_times(self, device_id, trip_id):
     detected_stop_times = stop_detector.get_detected_stop_times(tracker_id=device_id)
     gtfs_stop_times = gtfs.models.StopTime.objects.filter(trip = trip_id).order_by('arrival_time').values_list('stop', 'arrival_time', 'departure_time')
     acceptible_time_delta = 60 # one minute
     for detected_stop_time, gtfs_stop_time in zip(detected_stop_times, gtfs_stop_times):
         gtfs_stop_id = gtfs_stop_time[0]
         gtfs_arrival = gtfs_stop_time[1]
         gtfs_departure = gtfs_stop_time[2]
         msg = str(detected_stop_time)
         detected_arrival = ot_utils.datetime_to_db_time(detected_stop_time.arrival)
         self.assertAlmostEquals(detected_arrival, gtfs_arrival, msg=msg, delta=acceptible_time_delta)   
         is_last_detected_stop = detected_stop_time == detected_stop_times[-1]
         # allow for last stop to not have departure
         if detected_stop_time.departure or not is_last_detected_stop:
             detected_departure = ot_utils.datetime_to_db_time(detected_stop_time.departure)
             self.assertAlmostEquals(detected_departure, gtfs_departure, msg=msg, delta=acceptible_time_delta)
示例#4
0
 def evaluate_detected_stop_times(self, device_id, trip_id):
     detected_stop_times = stop_detector.get_detected_stop_times(
         tracker_id=device_id)
     trip = timetable.services.get_trip(trip_id)
     gtfs_stop_times = trip.get_stop_times().values_list('stop', 'exp_arrival', 'exp_departure')
     acceptible_time_delta = datetime.timedelta(minutes = 1)
     for detected_stop_time, gtfs_stop_time in zip(detected_stop_times, gtfs_stop_times):
         gtfs_arrival = gtfs_stop_time[1]
         gtfs_departure = gtfs_stop_time[2]
         msg = str(detected_stop_time)
         self.assertAlmostEquals(
             detected_stop_time.arrival, gtfs_arrival, msg=msg, delta=acceptible_time_delta)
         is_last_detected_stop = detected_stop_time == detected_stop_times[
             -1]
         # allow for last stop to not have departure
         if detected_stop_time.departure or not is_last_detected_stop:
             self.assertAlmostEquals(
                 detected_stop_time.departure, gtfs_departure, msg=msg, delta=acceptible_time_delta)