added missing utils folder

This commit is contained in:
Erik Frojdh 2023-06-01 17:59:15 +02:00
parent 5eeeaf681a
commit 96452488bf

43
utils/write_test_data.py Normal file
View File

@ -0,0 +1,43 @@
import numpy as np
from pathlib import Path
from creader import cluster_dt
#[int32 frame_number][int32 n_clusters][clusters....]
# typedef struct {
# int16_t x;
# int16_t y;
# int32_t data[9];
# } Cluster ;
path = Path('/mnt/sls_det_storage/moench_data/cluster_reader_test/')
data = np.zeros(1, cluster_dt())
data['x'] = 1
data['y'] = 200
data['data'] = np.arange(9)
#Write 10 frames with increadsing number of clusters
header = np.array((135,97), dtype = np.int32)
with open(path/'single_frame_97_clustrers.clust', 'wb') as f:
header.tofile(f)
for i in range(97):
print(data)
data.tofile(f)
data['x'] += 1
data['y'] += 1
data['data'] = np.arange((i+1)*9,(i+2)*9)
header = np.array((135,5), dtype = np.int32)
with open(path/'37frames_with_5_clusters.clust', 'wb') as f:
for i in range(37):
header.tofile(f)
header[0] += 1
for j in range(5):
data['x'] = j+1
data['y'] = j+1
data['data'] = np.arange(j,j+9)
print(data)
data.tofile(f)