ZH: rit flag to disable pol_and_bp in reconstruction

This commit is contained in:
Eric Teunis de Boone 2023-01-20 14:35:12 +01:00
parent 7008e9e8ce
commit a58c34f843
2 changed files with 4 additions and 3 deletions

View file

@ -160,7 +160,7 @@ if __name__ == "__main__":
N_X, Xlow, Xhigh = 23, 100, 1200 N_X, Xlow, Xhigh = 23, 100, 1200
with joblib.parallel_backend("loky"): with joblib.parallel_backend("loky"):
res = rit.reconstruction(ev, outfile=fig_subdir+'/fig.pdf', slice_outdir=fig_subdir+'/', Xlow=Xlow, N_X=N_X, Xhigh=Xhigh) res = rit.reconstruction(ev, outfile=fig_subdir+'/fig.pdf', slice_outdir=fig_subdir+'/', Xlow=Xlow, N_X=N_X, Xhigh=Xhigh, disable_pol_and_bp=True)
## Save a pickle ## Save a pickle
with open(pickle_fname, 'wb') as fp: with open(pickle_fname, 'wb') as fp:

View file

@ -335,14 +335,15 @@ def fill_stations_propeties(e,res):
res.station_ids.append(ids) res.station_ids.append(ids)
#res.has_pulse.append(has_pulse) #res.has_pulse.append(has_pulse)
def reconstruction(e,outfile='', slice_outdir=None, Xlow=300, Xhigh=1000, N_X=15): def reconstruction(e,outfile='', slice_outdir=None, Xlow=300, Xhigh=1000, N_X=15, disable_pol_and_bp=False):
res = RITResult() res = RITResult()
res.isMC.append(True) res.isMC.append(True)
res.zenith_ini.append(e.zenith) res.zenith_ini.append(e.zenith)
res.azimuth_ini.append(e.azimuth) res.azimuth_ini.append(e.azimuth)
res.core_ini.append(e.core) res.core_ini.append(e.core)
set_pol_and_bp(e) if not disable_pol_and_bp:
set_pol_and_bp(e, 0.03, 0.08)
#only use signal that have a signal in data #only use signal that have a signal in data
fill_stations_propeties(e,res) fill_stations_propeties(e,res)