From 96452488bff96ba4df8528a66d34686c4d321313 Mon Sep 17 00:00:00 2001 From: Erik Frojdh Date: Thu, 1 Jun 2023 17:59:15 +0200 Subject: [PATCH] added missing utils folder --- utils/write_test_data.py | 43 ++++++++++++++++++++++++++++++++++++++++ 1 file changed, 43 insertions(+) create mode 100644 utils/write_test_data.py diff --git a/utils/write_test_data.py b/utils/write_test_data.py new file mode 100644 index 0000000..7242ac4 --- /dev/null +++ b/utils/write_test_data.py @@ -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) + \ No newline at end of file