Merge branch 'even-more-HDF5-postprocessing' into MiscImprovements

This commit is contained in:
Martin Diehl 2019-09-19 10:49:14 -07:00
commit bf7db26680
1 changed files with 44 additions and 67 deletions

View File

@ -45,18 +45,17 @@ class DADF5():
self.materialpoints = [m.decode() for m in np.unique(f['mapping/cellResults/materialpoint']['Name'])] self.materialpoints = [m.decode() for m in np.unique(f['mapping/cellResults/materialpoint']['Name'])]
self.constituents = [c.decode() for c in np.unique(f['mapping/cellResults/constituent'] ['Name'])] self.constituents = [c.decode() for c in np.unique(f['mapping/cellResults/constituent'] ['Name'])]
self.con_physics = [] self.con_physics = []
for c in self.constituents: for c in self.constituents:
self.con_physics += f['inc00000/constituent/{}'.format(c)].keys() self.con_physics += f['/'.join([self.increments[0],'constituent',c])].keys()
self.con_physics = list(set(self.con_physics)) # make unique self.con_physics = list(set(self.con_physics)) # make unique
self.mat_physics = [] self.mat_physics = []
for m in self.materialpoints: for m in self.materialpoints:
self.mat_physics += f['inc00000/materialpoint/{}'.format(m)].keys() self.mat_physics += f['/'.join([self.increments[0],'materialpoint',m])].keys()
self.mat_physics = list(set(self.mat_physics)) # make unique self.mat_physics = list(set(self.mat_physics)) # make unique
self.visible= {'increments': self.increments, # ToDo:simplify, activity only positions that translate into (no complex types) self.visible= {'increments': self.increments,
'constituents': self.constituents, 'constituents': self.constituents,
'materialpoints': self.materialpoints, 'materialpoints': self.materialpoints,
'constituent': range(self.Nconstituents), # ToDo: stupid naming 'constituent': range(self.Nconstituents), # ToDo: stupid naming
@ -163,18 +162,11 @@ class DADF5():
groups = [] groups = []
with h5py.File(self.filename,'r') as f: with h5py.File(self.filename,'r') as f:
for i in self.iter_visible('increments'): #ToDo: Merge path only once at the end '/'.join(listE) for i in self.iter_visible('increments'):
for c in self.iter_visible('constituents'): for o,p in zip(['constituents','materialpoints'],['con_physics','mat_physics']):
for p in self.iter_visible('con_physics'): for oo in self.iter_visible(o):
group = '/'.join([i,'constituent',c,p]) for pp in self.iter_visible(p):
if sets is True: group = '/'.join([i,o[:-1],oo,pp]) # o[:-1]: plural/singular issue
groups.append(group)
else:
match = [e for e_ in [glob.fnmatch.filter(f[group].keys(),s) for s in sets] for e in e_]
if len(set(match)) == len(sets) : groups.append(group)
for m in self.iter_visible('materialpoints'):
for p in self.iter_visible('mat_physics'):
group = '/'.join([i,'materialpoint',m,p])
if sets is True: if sets is True:
groups.append(group) groups.append(group)
else: else:
@ -184,29 +176,23 @@ class DADF5():
def list_data(self): def list_data(self):
"""Shows information on all active datasets in the file.""" """Gives information on all active datasets in the file."""
message = ''
with h5py.File(self.filename,'r') as f: with h5py.File(self.filename,'r') as f:
i = 'inc{:05}'.format(0) for i in self.iter_visible('increments'):
for c in self.iter_visible('constituents'): message+='\n{}\n'.format(i)
print('{}'.format(c)) for o,p in zip(['constituents','materialpoints'],['con_physics','mat_physics']):
for p in self.iter_visible('con_physics'): for oo in self.iter_visible(o):
print(' {}'.format(p)) message+=' {}\n'.format(oo)
for pp in self.iter_visible(p):
message+=' {}\n'.format(pp)
group = '/'.join([i,o[:-1],oo,pp]) # o[:-1]: plural/singular issue
for d in f[group].keys():
try: try:
k = '/'.join([i,'constituent',c,p]) message+=' {} ({})\n'.format(d,f['/'.join([group,d])].attrs['Description'].decode())
for d in f[k].keys():
print(' {} ({})'.format(d,f[k+'/'+d].attrs['Description'].decode()))
except KeyError:
pass
for m in self.iter_visible('materialpoints'):
print('{}'.format(m))
for p in self.iter_visible('mat_physics'):
print(' {}'.format(p))
try:
k = '/'.join([i,'materialpoint',m,p])
for d in f[k].keys():
print(' {} ({})'.format(d,f[k+'/'+d].attrs['Description'].decode()))
except KeyError: except KeyError:
pass pass
return message
def get_dataset_location(self,label): def get_dataset_location(self,label):
@ -214,24 +200,15 @@ class DADF5():
path = [] path = []
with h5py.File(self.filename,'r') as f: with h5py.File(self.filename,'r') as f:
for i in self.iter_visible('increments'): for i in self.iter_visible('increments'):
for c in self.iter_visible('constituents'): for o,p in zip(['constituents','materialpoints'],['con_physics','mat_physics']):
for p in self.iter_visible('con_physics'): for oo in self.iter_visible(o):
for pp in self.iter_visible(p):
k = '/'.join([i,o[:-1],oo,pp,label])
try: try:
k = '/'.join([i,'constituent',c,p,label])
f[k] f[k]
path.append(k) path.append(k)
except KeyError as e: except KeyError as e:
print('unable to locate constituents dataset: '+ str(e)) print('unable to locate constituents dataset: '+ str(e))
for m in self.iter_visible('materialpoints'):
for p in self.iter_visible('mat_physics'):
try:
k = '/'.join([i,'materialpoint',m,p,label])
f[k]
path.append(k)
except KeyError as e:
print('unable to locate materialpoints dataset: '+ str(e))
return path return path