# This file is part of Pymepix
#
# In all scientific work using Pymepix, please reference it as
#
# A. F. Al-Refaie, M. Johny, J. Correa, D. Pennicard, P. Svihra, A. Nomerotski, S. Trippel, and J. Küpper:
# "PymePix: a python library for SPIDR readout of Timepix3", J. Inst. 14, P10003 (2019)
# https://doi.org/10.1088/1748-0221/14/10/P10003
# https://arxiv.org/abs/1905.07999
#
# Pymepix is free software: you can redistribute it and/or modify it under the terms of the GNU
# General Public License as published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without
# even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License along with this program. If not,
# see <https://www.gnu.org/licenses/>.
"""Useful functions to store data"""
import numpy as np
[docs]def open_output_file(filename, ext, index=0):
import logging
import os
file_format = "{}_{:06d}.{}"
raw_filename = file_format.format(filename, index, ext)
while os.path.isfile(raw_filename):
index += 1
raw_filename = file_format.format(filename, index, ext)
logging.info("Opening output file {}".format(filename))
return open(raw_filename, "wb")
[docs]def store_raw(f, data):
raw, longtime = data
f.write(raw.tostring())
[docs]def store_toa(f, data):
x, y, toa, tot = data
np.save(f, x)
np.save(f, y)
np.save(f, toa)
np.save(f, tot)
[docs]def store_tof(f, data):
counter, x, y, tof, tot = data
np.save(f, counter)
np.save(f, x)
np.save(f, y)
np.save(f, tof)
np.save(f, tot)
[docs]def store_centroid(f, data):
cluster_shot, cluster_x, cluster_y, cluster_tof, cluster_tot = data
np.save(f, cluster_shot)
np.save(f, cluster_x)
np.save(f, cluster_y)
np.save(f, cluster_tof)
np.save(f, cluster_tot)