def test_sliding_window_max_sum(toymodel):
waveforms, subarray, telid, selected_gain_channel, true_charge, true_time = toymodel
extractor = SlidingWindowMaxSum(subarray=subarray)
charge, peak_time = extractor(waveforms, telid, selected_gain_channel)
print(true_charge, charge, true_time, peak_time)
assert_allclose(charge, true_charge, rtol=0.1)
assert_allclose(peak_time, true_time, rtol=0.1)
def test_sw_pulse_lst():
"""
Test function of sliding window extractor for LST camera pulse shape with
the correction for the integration window completeness
"""
# prepare array with 1 LST
subarray = SubarrayDescription(
"LST1",
tel_positions={1: np.zeros(3) * u.m},
tel_descriptions={
1:
TelescopeDescription.from_name(optics_name="LST",
camera_name="LSTCam")
},
)
telid = list(subarray.tel.keys())[0]
n_pixels = subarray.tel[telid].camera.geometry.n_pixels
n_samples = 40
readout = subarray.tel[telid].camera.readout
random = np.random.RandomState(1)
min_charge = 100
max_charge = 1000
charge_true = random.uniform(min_charge, max_charge, n_pixels)
time_true = random.uniform(n_samples // 2 - 1, n_samples // 2 + 1,
n_pixels) / readout.sampling_rate.to_value(
u.GHz)
waveform_model = WaveformModel.from_camera_readout(readout)
waveform = waveform_model.get_waveform(charge_true, time_true, n_samples)
selected_gain_channel = np.zeros(charge_true.size, dtype=np.int8)
# define extractor
config = Config({"SlidingWindowMaxSum": {"window_width": 8}})
extractor = SlidingWindowMaxSum(subarray=subarray)
extractor = ImageExtractor.from_name("SlidingWindowMaxSum",
subarray=subarray,
config=config)
dl1: DL1CameraContainer = extractor(waveform, telid, selected_gain_channel)
print(dl1.image / charge_true)
assert_allclose(dl1.image, charge_true, rtol=0.02)
assert dl1.is_valid