Skip to content

Commit 58a3ee1

Browse files
committed
convert h5 to cxi
1 parent 6d28a19 commit 58a3ee1

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

h52cxi.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -48,13 +48,12 @@
4848
peak_x = np.zeros(1024)
4949
peak_y = np.zeros(1024)
5050
peak_intensity = np.zeros(1024)
51-
data = h5['processing/cheetah/peakinfo-raw'].value
52-
peak_mask = (data[:,0] > 0)* (data[:,1] > 0)
53-
n_peak = peak_mask.sum().astype(np.int)
51+
peak_data = h5['/processing/hitfinder/peakinfo'].value
52+
n_peak = peak_data.shape[0]
5453
n_peaks.append(n_peak)
55-
peak_x[0:n_peak] = data[peak_mask,0]
56-
peak_y[0:n_peak] = data[peak_mask,1]
57-
peak_intensity [0:n_peak] = data[peak_mask,2]
54+
peak_x[0:n_peak] = peak_data[:,0]
55+
peak_y[0:n_peak] = peak_data[:,1]
56+
peak_intensity [0:n_peak] = peak_data[:,2]
5857
peaks_x.append(peak_x)
5958
peaks_y.append(peak_y)
6059
peaks_intensity.append(peak_intensity)
@@ -63,15 +62,15 @@
6362
# wavelength
6463
wavelengths.append(h5['/LCLS/photon_energy_eV'].value[0])
6564

66-
pattern = h5['data/rawdata0'].value
65+
pattern = h5['/data/rawdata0'].value
6766
sy, sx = pattern.shape
6867
pattern = pattern.reshape((1, sy, sx))
6968
if i == 0:
70-
cxi.create_dataset('/entry_1/instrument_1/detector_1/detector_corrected/data',
69+
cxi.create_dataset('/entry_1/data_1/data',
7170
data=pattern, maxshape=(None, sy, sx))
7271
else:
73-
cxi['/entry_1/instrument_1/detector_1/detector_corrected/data'].resize(i+1, axis=0)
74-
cxi['/entry_1/instrument_1/detector_1/detector_corrected/data'][i] = pattern
72+
cxi['/entry_1/data_1/data'].resize(i+1, axis=0)
73+
cxi['/entry_1/data_1/data'][i] = pattern
7574

7675
peaks_x = np.asarray(peaks_x)
7776
peaks_y = np.asarray(peaks_y)
@@ -85,4 +84,5 @@
8584
cxi.create_dataset('/LCLS/photon_energy_eV', data=wavelengths)
8685
cxi.create_dataset('/entry_1/result_1/nPeaks', data=n_peaks)
8786
cxi.create_dataset('/entry_1/result_1/peakTotalIntensity', data=peaks_intensity)
87+
cxi.create_dataset('/entry_1/result_1/peakSNR', data=np.zeros_like(peaks_intensity)) # dummy SNR
8888
cxi.close()

0 commit comments

Comments
 (0)