easier to use for single output label

This commit is contained in:
Martin Diehl 2020-01-03 15:41:15 +01:00
parent b5e4e42505
commit e692118ec5
1 changed files with 3 additions and 3 deletions

View File

@ -855,7 +855,7 @@ class DADF5():
Parameters Parameters
---------- ----------
labels : list of str labels : str or list of
Labels of the datasets to be exported. Labels of the datasets to be exported.
mode : str, either 'Cell' or 'Point' mode : str, either 'Cell' or 'Point'
Export in cell format or point format. Export in cell format or point format.
@ -908,7 +908,7 @@ class DADF5():
materialpoints_backup = self.visible['materialpoints'].copy() materialpoints_backup = self.visible['materialpoints'].copy()
self.set_visible('materialpoints',False) self.set_visible('materialpoints',False)
for label in labels: for label in (labels if isinstance(labels,list) else [labels]):
for p in self.iter_visible('con_physics'): for p in self.iter_visible('con_physics'):
if p != 'generic': if p != 'generic':
for c in self.iter_visible('constituents'): for c in self.iter_visible('constituents'):
@ -939,7 +939,7 @@ class DADF5():
constituents_backup = self.visible['constituents'].copy() constituents_backup = self.visible['constituents'].copy()
self.set_visible('constituents',False) self.set_visible('constituents',False)
for label in labels: for label in (labels if isinstance(labels,list) else [labels]):
for p in self.iter_visible('mat_physics'): for p in self.iter_visible('mat_physics'):
if p != 'generic': if p != 'generic':
for m in self.iter_visible('materialpoints'): for m in self.iter_visible('materialpoints'):