updating snapshot and loading datasets

This commit is contained in:
reiche committed 2024-09-11 17:06:10 +02:00
1 parent 24cbccb6a6
commit 26eebc650b
5 files changed
+24 -5

No files matched your search

+14 -2
View File
@@ -3,10 +3,22 @@ import h5py
def loadDataset(filename):
hid = h5py.File(filename, "r")
icount = 0
snap = loadSnap(hid)
data = loadData(hid)
act = loadActuator(hid)
data =[]
act = []
for key in hid.keys():
if 'scan' in key:
ID = int(key.split('_')[1])
print('Reading scan_%d' % ID)
data.append(loadData(hid,ID))
act.append(loadActuator(hid,ID))
icount +=1
hid.close()
if icount == 0:
return None, None, None
elif icount == 1:
return data[0], act[0],snap
return data,act,snap
def loadSnap(hid):