def walk_files(dir):
for root, dirs, files in os.walk(dir):
for d in dirs:
pos_dir = os.path.join(root, d)
print(pos_dir)
for rs, ds, fs in os.walk(pos_dir):
for f in fs:
pos_path = os.path.join(pos_dir, f)
# print(pos_path)
if pos_path[-4:] == '.pos':
site = pos_path[-8:-4]
data = parse_sol(pos_dir, site)
if ENU_PLOT:
enu_plot(pos_dir, site, data, d)
if DOP_PLOT:
dop_plot(pos_dir, site, data)
if TRP_PLOT:
trp_plot(pos_dir, site, data)
if SAT_PLOT:
sat_plot(pos_dir, site, data, RES_P1, SAT_ID)
sat_plot(pos_dir, site, data, RES_L1, SAT_ID)
sat_plot(pos_dir, site, data, AMB1, SAT_ID)
sat_plot(pos_dir, site, data, MW_AMB, SAT_ID)
else:
continue
def walk_files(dir, ana_dir):
if not os.path.exists(ana_dir):
os.makedirs(ana_dir)
for root, dirs, files in os.walk(dir):
for d in dirs:
if d == 'analysis':
continue
if len(d) == 4:
continue
pos_dir = os.path.join(root, d)
ana_path = os.path.join(ana_dir, d + '_analysis.txt')
ana_handle = open(ana_path, mode='w')
print(pos_dir)
for rs, ds, fs in os.walk(pos_dir):
for f in fs:
pos_path = os.path.join(pos_dir, f)
# print(pos_path)
if pos_path[-4:] == '.pos':
site = pos_path[-8:-4]
data = parse_sol(pos_dir, site)
analysis_pos(pos_dir, site, data, ana_handle)
else:
continue
ana_handle.close()
DOP_PLOT = 0 # 绘制卫星数量和pdop
ENU_PLOT = 1 # 绘制ENU位置误差
TRP_PLOT = 0 # 绘制对流层延迟
# 绘制卫星有关，指定卫星id后绘制单颗卫星，不指定则不绘制
SAT_PLOT = 0
SAT_ID = 'G16'
if BATCH_PLOT:
walk_files(sol_dir)
else:
pos_dir = ''
site = ''
if sol_dir[-4:] == '.pos':
pos_dir = sol_dir[:-8]
site = sol_dir[-8:-4]
data = parse_sol(pos_dir, site)
if ENU_PLOT:
enu_plot(pos_dir, site, data, '')
if DOP_PLOT:
dop_plot(pos_dir, site, data)
if TRP_PLOT:
trp_plot(pos_dir, site, data)
if SAT_PLOT:
sat_plot(pos_dir, site, data, RES_P1, SAT_ID)
sat_plot(pos_dir, site, data, RES_L1, SAT_ID)
sat_plot(pos_dir, site, data, AMB1, SAT_ID)
sat_plot(pos_dir, site, data, MW_AMB, SAT_ID)
fig = plt.figure(figsize=(4, 3))
ax = fig.add_subplot(111)
# ax.xaxis.set_major_formatter(mdate.DateFormatter('%H'))
# ax.grid(axis='x', linestyle='-.', linewidth=0.2)
# ax.grid(axis='y', linestyle='-.', linewidth=0.2)
plt.plot_date(ts, ions)
plt.show()
DOP_PLOT = 1
ENU_PLOT = 1
ISB_PLOT = 0
TRP_PLOT = 0
ION_PLOT = 0
AMB_PLOT = 0
SAT_ID = 'G10'
if __name__ == '__main__':
sol_path = '../../example/result_wum/PPP_STATIC/G_DF_IF/faa1.pos'
data = parse_sol(sol_path)
if ENU_PLOT:
enu_plot(sol_path, data)
if DOP_PLOT:
dop_plot(sol_path, data)
if ISB_PLOT:
a = 1
if TRP_PLOT:
trp_plot(sol_path, data)
if ION_PLOT:
ion_plot(sol_path, data, SAT_ID)
comp_dir1 = os.path.join(main_dir, sub_dir1)
comp_dir2 = os.path.join(main_dir, sub_dir2)
comp_save_dir = os.path.join(main_dir, 'compare')
if not os.path.exists(comp_save_dir):
os.makedirs(comp_save_dir)
save_path = os.path.join(comp_save_dir,
sub_dir1 + '-' + sub_dir2 + '-' + site)
for root, dirs, files in os.walk(comp_dir1):
pos_dir1 = comp_dir1
for f in files:
if site in f:
pos_path1 = os.path.join(comp_dir1, f)
data1 = parse_sol(pos_dir1, site)
n_data += 1
data.append(data1)
break
else:
continue
for root, dirs, files in os.walk(comp_dir2):
pos_dir2 = comp_dir2
for f in files:
if site in f:
pos_path2 = os.path.join(comp_dir2, f)
data2 = parse_sol(pos_dir2, site)
n_data += 1
data.append(data2)
break
