From 62370df548a52b0ce6f8c0ddb7ea6c2c34804447 Mon Sep 17 00:00:00 2001 From: Peter Madigan Date: Tue, 24 Aug 2021 19:23:34 -0500 Subject: [PATCH] Speed up dumpTree.py for larger files --- cli/dumpTree.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/cli/dumpTree.py b/cli/dumpTree.py index 042b20ef..4f43f54c 100644 --- a/cli/dumpTree.py +++ b/cli/dumpTree.py @@ -123,7 +123,7 @@ def dump(input_file, output_file): ('t_end','f4'),('dEdx','f4'),('dE','f4'),('t','f4'), ('y','f4'),('x','f4'),('z','f4')]) - segments = np.array([], dtype=dtype) + segments = [] for jentry in range(entries): print(jentry) @@ -179,7 +179,8 @@ def dump(input_file, output_file): segment['long_diff'] = 0 segment['tran_diff'] = 0 segment['pixel_plane'] = 0 - segments = np.hstack((segments, segment)) + segments.append(segment) + segments = np.concatenate(segments, axis=0) with h5py.File(output_file, 'w') as f: f.create_dataset("segments", data=segments)