diff options
| -rw-r--r-- | scripts/process_route.py | 59 |
1 files changed, 59 insertions, 0 deletions
diff --git a/scripts/process_route.py b/scripts/process_route.py new file mode 100644 index 0000000..5242a82 --- /dev/null +++ b/scripts/process_route.py @@ -0,0 +1,59 @@ +import json +import math + +def calculate_bearing(lat1, lon1, lat2, lon2): + d_lon = lon2 - lon1 + x = math.cos(math.radians(lat2)) * math.sin(math.radians(d_lon)) + y = math.cos(math.radians(lat1)) * math.sin(math.radians(lat2)) - \ + math.sin(math.radians(lat1)) * math.cos(math.radians(lat2)) * math.cos(math.radians(d_lon)) + bearing = math.atan2(x, y) + bearing = math.degrees(bearing) + bearing = (bearing + 360) % 360 + return bearing + +def calculate_distance(lat1, lon1, lat2, lon2): + R = 6371000 # Earth's radius in meters + d_lat = math.radians(lat2 - lat1) + d_lon = math.radians(lon2 - lon1) + a = math.sin(d_lat / 2) * math.sin(d_lat / 2) + \ + math.cos(math.radians(lat1)) * math.cos(math.radians(lat2)) * \ + math.sin(d_lon / 2) * math.sin(d_lon / 2) + c = 2 * math.atan2(math.sqrt(a), math.sqrt(1 - a)) + distance = R * c + return distance + +def calculate_timestamp(data, current_index): + if current_index == 0: + return 0 + else: + prev_timestamp = calculate_timestamp(data, current_index - 1) + return prev_timestamp + data[current_index-1]['distance'] / data[current_index]['speed'] + +def process_route(input_filename, output_filename): + with open(input_filename, 'r') as infile: + data = json.load(infile) + + output_data = [] + + for i in range(len(data)): + if i == len(data) - 1: + data[i]['bearing'] = calculate_bearing(data[i]['latitude'], data[i]['longitude'], + data[i-1]['latitude'], data[i-1]['longitude']) + else: + bearing = calculate_bearing(data[i]['latitude'], data[i]['longitude'], + data[i+1]['latitude'], data[i+1]['longitude']) + data[i]['bearing'] = bearing + + data[i]['distance'] = calculate_distance(data[i]['latitude'], data[i]['longitude'], + data[i+1]['latitude'], data[i+1]['longitude']) + + data[i]['timestamp'] = calculate_timestamp(data, i) + + output_data.append(data[i]) + + for d in output_data: + d.pop("distance", None) + + with open(output_filename, 'w') as outfile: + json.dump(output_data, outfile, indent=2) + |