mirror of
https://github.com/slsdetectorgroup/aare.git
synced 2025-04-19 21:30:02 +02:00
89 lines
2.7 KiB
Python
89 lines
2.7 KiB
Python
import matplotlib.pyplot as plt
|
|
import numpy as np
|
|
plt.ion()
|
|
|
|
import aare
|
|
from aare import CtbRawFile
|
|
print('aare imported')
|
|
from aare import transform
|
|
print('transform imported')
|
|
from pathlib import Path
|
|
|
|
import json
|
|
|
|
def decode(frames, rawdata):
|
|
# rawdata = np.fromfile(f, dtype = np.uint16)
|
|
counters = int((np.shape(rawdata)[0]/frames-56)/(48*48))
|
|
print('Counters:', counters)
|
|
rawdata = rawdata.reshape(frames,-1)[:,56:]
|
|
rawdata = rawdata.reshape(frames,576*counters,4) #Data come in "blocks" of 4 pixels/receiver
|
|
tr1 = rawdata[:,0:576*counters:2] #Transceiver1
|
|
tr1=tr1.reshape((frames,48*counters,24))
|
|
|
|
tr2 = rawdata[:,1:576*counters:2] #Transceiver2
|
|
tr2=tr2.reshape((frames,48*counters,24))
|
|
|
|
data = np.append(tr1,tr2,axis=2)
|
|
return data
|
|
|
|
def get_Mh02_frames(fname):
|
|
# this function gives you the data from a file that is not a scan
|
|
# it returns a (frames,48*counters,48)
|
|
|
|
jsonf = open(fname)
|
|
jsonpar = json.load(jsonf)
|
|
jsonf.close()
|
|
|
|
frames=jsonpar["Frames in File"]
|
|
print('Frames:', frames)
|
|
|
|
rawf = fname.replace('master','d0_f0')
|
|
rawf = rawf.replace('.json','.raw')
|
|
|
|
with open(rawf, 'rb') as f:
|
|
rawdata = np.fromfile(f, dtype = np.uint16)
|
|
data = decode(frames, rawdata)
|
|
print('Data:', np.shape(data))
|
|
|
|
return data
|
|
|
|
|
|
#target format
|
|
# [frame, counter, row, col]
|
|
# plt.imshow(data[0,0])
|
|
|
|
|
|
base = Path('/mnt/sls_det_storage/matterhorn_data/aare_test_data/ci/aare_test_data')
|
|
# p = Path(base / 'jungfrau/jungfrau_single_master_0.json')
|
|
|
|
# f = aare.File(p)
|
|
# for i in range(10):
|
|
# frame = f.read_frame()
|
|
|
|
|
|
|
|
# # f2 = aare.CtbRawFile(fpath, transform=transform.matterhorn02)
|
|
# # header, data = f2.read()
|
|
# # plt.plot(data[:,0,20,20])
|
|
# from aare import RawMasterFile, File, RawSubFile, DetectorType, RawFile
|
|
# base = Path('/mnt/sls_det_storage/matterhorn_data/aare_test_data/Jungfrau10/Jungfrau_DoubleModule_1UDP_ROI/SideBySide/')
|
|
# fpath = Path('241019_JF_12keV_Si_FF_GaAs_FF_7p88mmFilter_PedestalStart_ZPos_5.5_master_0.json')
|
|
# raw = Path('241019_JF_12keV_Si_FF_GaAs_FF_7p88mmFilter_PedestalStart_ZPos_5.5_d0_f0_0.raw')
|
|
|
|
|
|
|
|
# m = RawMasterFile(base / fpath)
|
|
# # roi = m.roi
|
|
# # rows = roi.ymax-roi.ymin+1
|
|
# # cols = 1024-roi.xmin
|
|
# # sf = RawSubFile(base / raw, DetectorType.Jungfrau, rows, cols, 16)
|
|
|
|
from aare import RawFile
|
|
|
|
base = Path('/mnt/sls_det_storage/matterhorn_data/aare_test_data/Jungfrau10/Jungfrau_DoubleModule_1UDP_ROI/SideBySide/')
|
|
fname = base / Path('241019_JF_12keV_Si_FF_GaAs_FF_7p88mmFilter_PedestalStart_ZPos_5.5_master_0.json')
|
|
# fname = Path(base / 'jungfrau/jungfrau_single_master_0.json')
|
|
f = RawFile(fname)
|
|
h,img = f.read_frame()
|
|
print(f'{h["frameNumber"]}')
|