1import json
 2from obspy.taup import TauPyModel
 3import numpy as np
 4from tqdm import tqdm
 5
 6model = TauPyModel(model='iasp91')
 7
 8computed_depths = 70
 9computed_angles = 180
10max_angle = 90 # deg
11max_depth = 700 # km
12
13depth_list = np.linspace(0,max_depth,computed_depths)
14angle_list = np.linspace(0,max_angle,computed_angles)
15
16p_table = []
17for depth in tqdm(range(len(depth_list))):
18    arrivals = []
19    for angle in angle_list:
20        arrival = model.get_travel_times(source_depth_in_km=depth_list[depth],
21                                    distance_in_degree=angle,phase_list=["p","P"])[0].time
22        arrivals.append(arrival)
23    p_table.append(arrivals)
24
25s_table = []
26for depth in tqdm(range(len(depth_list))):
27    arrivals = []
28    for angle in angle_list:
29        arrival = model.get_travel_times(source_depth_in_km=depth_list[depth],
30                                    distance_in_degree=angle,phase_list=["s","S"])[0].time
31        arrivals.append(arrival)
32    s_table.append(arrivals)
33
34data = {"angles": list(angle_list), "depths":list(depth_list), "p_table":p_table,"s_table":s_table}
35
36print(data)
37
38f = open("computed_tables.json", "w")
39f.write(json.dumps(data))
40f.close()