lisa
353268b816
Displays the PAD for a channel of a spacecraft. It plots both the raw data points and their mean values, including error bars.
261 lines
5.6 KiB
Python
261 lines
5.6 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
Created on Fri Apr 17 10:05:13 2026
|
|
|
|
@author: vieweg
|
|
"""
|
|
|
|
from rapid_fgm import *
|
|
from datetime import datetime, timedelta
|
|
import matplotlib.pyplot as plt
|
|
|
|
#--------------USER INPUT----------------------
|
|
TIME_PERIOD = (datetime(2002,6,27,00), datetime(2002,6,29,0))
|
|
|
|
TIME_POINT = datetime(2002,6,28,19,15)
|
|
|
|
SC_ID = 'C3'
|
|
|
|
e_idx = 6 #ch 0, ...,7
|
|
|
|
PARTICLE_TYPE = "electron" #'electron' oder "ion"
|
|
|
|
#----------------------------------------
|
|
instrument = "IES" if PARTICLE_TYPE == "electron" else "IIMS"
|
|
|
|
r = RAPID(sc_id=SC_ID, time_period=TIME_PERIOD, particle_type=PARTICLE_TYPE)
|
|
|
|
data_rapid = r.load_rapid()
|
|
r.get_rapid_var(r.data_rapid)
|
|
print('rapid time:', r.time_rapid[0], r.time_rapid[-1])
|
|
|
|
|
|
|
|
|
|
flow = FLOW(
|
|
sc_id=SC_ID,
|
|
time_period=TIME_PERIOD,
|
|
particle_type=PARTICLE_TYPE
|
|
)
|
|
|
|
flow.get_flow_var(flow.data_flow)
|
|
flow.load_flow()
|
|
|
|
|
|
f = FGM(SC_ID, time_period=TIME_PERIOD)
|
|
data_fgm = f.load_fgm()
|
|
f.get_fgm_var(f.data_fgm)
|
|
|
|
p = PITCH_CSA(SC_ID, time_period= TIME_PERIOD)
|
|
data_pad = p.load_pad()
|
|
pad_data = p.get_pad_var(p.data_pad)
|
|
|
|
|
|
w = WORKSPACE()
|
|
sync = w.sync_all_by_resolution(r, f, flow)
|
|
|
|
t_sync = sync['time']
|
|
rapid_sync = sync['rapid']
|
|
fgm_sync = sync['fgm']
|
|
sc2gse = sync['sc2gse']
|
|
|
|
sc_data = w.spacecrafts(
|
|
sc_id=SC_ID,
|
|
t_sync=t_sync,
|
|
rapid_sync=rapid_sync,
|
|
fgm_sync=fgm_sync,
|
|
sc2gse=sc2gse,
|
|
time_period=None)
|
|
print("len SC time:", len(w.SC_data['C3']['time']))
|
|
|
|
|
|
|
|
#font in Times New Roman
|
|
plt.rcParams['font.family'] = 'Times New Roman'
|
|
plt.rcParams['mathtext.fontset'] = 'stix'
|
|
|
|
|
|
|
|
|
|
w.rapid_n_sc2gse(SC_ID)
|
|
|
|
|
|
pitch_angle = w.calc_pitch_angle('C3')
|
|
|
|
|
|
|
|
|
|
def channel_label(e_idx):
|
|
channel_phys = e_idx + 1
|
|
energy = r.energy_table[e_idx]
|
|
return energy, channel_phys
|
|
|
|
color1 ='#0070C0'
|
|
color2 = '#63A375'
|
|
|
|
|
|
IES_energy, channel_phys = channel_label(e_idx)
|
|
|
|
pitch, flux, flux_sd = w.plot_flux_of_pitch(SC_ID, TIME_POINT, e_idx)
|
|
flux = np.ma.masked_where(flux <= 0, flux)
|
|
plt.figure(figsize=(10, 9))
|
|
plt.errorbar(
|
|
pitch,
|
|
flux,
|
|
yerr=flux_sd,
|
|
linestyle='none',
|
|
marker='o',
|
|
color='gray',
|
|
ecolor='gray',
|
|
label=r'Flux $\pm$ SD', alpha = 0.7
|
|
)
|
|
# Reference xerror bar
|
|
ref_x = 15
|
|
|
|
if len(flux) > 0:
|
|
ref_y = np.nanmax(flux) * 0.9
|
|
else:
|
|
ref_y = 1
|
|
|
|
|
|
plt.errorbar(
|
|
ref_x,
|
|
ref_y,
|
|
xerr=10,
|
|
fmt='o',
|
|
color='black',
|
|
capsize=4,
|
|
label='10° Sector Width'
|
|
)
|
|
|
|
plt.xticks([10, 30, 50, 70, 90, 110, 130, 150, 170])
|
|
plt.xlabel('Pitch Angle / °')
|
|
plt.ylabel(r'Flux / $(\mathrm{cm^{2}\,s\,sr\,keV})^{-1}$')
|
|
plt.ylim(0,4000)
|
|
|
|
plt.title(
|
|
f'{instrument} Pitch Angle Distribution - {SC_ID} - '
|
|
f'Ch{channel_phys} ({IES_energy:.1f} keV) - {TIME_POINT}'
|
|
)
|
|
|
|
plt.legend()
|
|
plt.grid(True)
|
|
|
|
|
|
# Histogram Mean Flux
|
|
|
|
IES_energy, channel_phys = channel_label(e_idx)
|
|
|
|
bins, histo, shist, nhist = w.histo_flux_pitch(SC_ID, TIME_POINT, e_idx)
|
|
|
|
|
|
|
|
print(w.SC_data[SC_ID]['time'][0], w.SC_data[SC_ID]['time'][-1])
|
|
|
|
|
|
#plot mean flux with errorbar
|
|
#plt.figure()
|
|
plt.errorbar(
|
|
bins,
|
|
histo,
|
|
yerr=shist,
|
|
color=color1,
|
|
ecolor=color2,
|
|
linestyle='none',
|
|
linewidth = 4,
|
|
markersize = 10,
|
|
marker='^',
|
|
capsize=6,
|
|
label='Mean Flux ± SD'
|
|
)
|
|
|
|
plt.legend(fontsize = 14)
|
|
|
|
plt.title(
|
|
f"Pitch Angle Distribution - {instrument} - {SC_ID} - "
|
|
f"Ch{channel_phys} ({IES_energy:.1f} keV) - {TIME_POINT}", fontsize = 16
|
|
)
|
|
|
|
plt.grid(True)
|
|
plt.xticks([10, 30, 50, 70, 90, 110, 130, 150, 170], fontsize = 14)
|
|
plt.yticks(fontsize = 14)
|
|
plt.ylim(0,4000)
|
|
plt.ylabel(r'Flux / $(\mathrm{cm^{2}\,s\,sr\,keV})^{-1}$', fontsize = 16)
|
|
plt.xlabel(r'Pitch Angle / °', fontsize = 16)
|
|
|
|
timestamp = TIME_POINT.strftime("%Y%m%d_%H%M%S")
|
|
filename = 'C:/Users/lview/OneDrive/Dokumente/Desktop/C3_RAPID_20020628/191500/Flux_Pitch/'+ f'{SC_ID}_{instrument}_Ch'+f'{channel_phys}_'+f'{timestamp}'+'_FP_v2.pdf'
|
|
print(filename)
|
|
plt.savefig(filename)
|
|
plt.show()
|
|
|
|
|
|
idx_pad = w.time_to_index(pad_data['time_pad'], TIME_POINT)
|
|
idx_sc = w.time_to_index(w.SC_data[SC_ID]['time'], TIME_POINT)
|
|
|
|
#CSA PAD
|
|
|
|
flux_all = p.data_pad['PAD_Electron_Dif_flux__C3_CP_RAP_PAD_E3DD'][:]
|
|
flux_sd_all = p.data_pad['PAD_Electron_Dif_flux_SD__C3_CP_RAP_PAD_E3DD'][:]
|
|
|
|
pitch_csa = p.data_pad['Dimension_Pitch__C3_CP_RAP_PAD_E3DD'][:]
|
|
time_pad = p.data_pad['Time_tags__C3_CP_RAP_PAD_E3DD'][:]
|
|
|
|
import cdflib
|
|
time_pad_dt = cdflib.cdfepoch.to_datetime(time_pad)
|
|
|
|
idx_pad = w.time_to_index(time_pad_dt, TIME_POINT)
|
|
|
|
flux_csa = flux_all[idx_pad, e_idx, :]
|
|
flux_sd_csa = flux_sd_all[idx_pad, e_idx, :]
|
|
flux_sd_csa = flux_sd_all[idx_pad, e_idx, :]
|
|
|
|
plt.figure(figsize=(10,9))
|
|
|
|
|
|
plt.errorbar(
|
|
pitch_csa,
|
|
flux_csa,
|
|
yerr=flux_sd_csa,
|
|
fmt='o',
|
|
color=color1,
|
|
ecolor='green',
|
|
capsize=5,
|
|
markersize = 10,
|
|
linewidth = 4,
|
|
label='Flux ± SD'
|
|
)
|
|
|
|
# xerror separat
|
|
plt.errorbar(
|
|
pitch_csa,
|
|
flux_csa,
|
|
xerr=10,
|
|
fmt='none',
|
|
ecolor='black',
|
|
capsize=5,
|
|
linewidth = 4,
|
|
label = '10° Sector Width')
|
|
|
|
|
|
plt.xticks([10, 30, 50, 70, 90, 110, 130, 150, 170], fontsize = 14)
|
|
plt.yticks(fontsize = 14)
|
|
plt.xlabel('Pitch Angle / °', fontsize = 16)
|
|
plt.ylabel(r'Flux / $(\mathrm{cm^{2}\,s\,sr\,keV})^{-1}$', fontsize = 16)
|
|
plt.ylim(0,4000)
|
|
|
|
plt.title(
|
|
f'CSA {instrument} Pitch Angle Distribution - {SC_ID} - '
|
|
f'Ch{channel_phys} ({IES_energy:.1f} keV) - {TIME_POINT}', fontsize = 16
|
|
)
|
|
|
|
plt.grid(True)
|
|
plt.legend(fontsize = 14)
|
|
timestamp = TIME_POINT.strftime("%Y%m%d_%H%M%S")
|
|
filename = 'C:/Users/lview/OneDrive/Dokumente/Desktop/C3_RAPID_20020628/191500/Flux_Pitch/'+ f'{SC_ID}_{instrument}_Ch'+f'{channel_phys}_'+f'{timestamp}'+'_FP_CSA.pdf'
|
|
print(filename)
|
|
plt.savefig(filename)
|
|
plt.show()
|
|
|
|
|