223031012
/
DAMASK_EICMD
mirror of https://github.com/achalhp/DAMASK_EICMD.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

finish h5table class

This commit is contained in:
chen 2016-10-07 17:43:52 -04:00
parent 859e3584ca
commit 012b702f7a
1 changed files with 21 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
lib/damask/h5table.py
View File

@ -48,15 +48,17 @@ class H5Table(object):
Absolute path the HDF5 file Absolute path the HDF5 file
METHOD METHOD
------ ------
del_entry()
get_attr() get_attr()
add_attr() add_attr()
get_data() get_data()
add_data() add_data()
NOTE NOTE
---- ----
1. As an interface class, it uses the lazy evaluation design 1. As an interface class, it uses the lazy evaluation design
that read the data only when its absolutely necessary. that read the data only when its absolutely necessary.
2. The command line used to generate new feature is stored with
each dataset as dataset attribute.
""" """
def __init__(self, h5f_path): def __init__(self, h5f_path):
@ -64,20 +66,24 @@ class H5Table(object):
""" """
self.h5f_path = h5f_path self.h5f_path = h5f_path
def __del__(self, feature_name=None): def del_entry(self, feature_name):
""" delete entry in HDF5 table """
dataType, h5f_path = lables_to_path(feature_name) dataType, h5f_path = lables_to_path(feature_name)
h5f = h5py.File(self.h5f_path, 'a') h5f = h5py.File(self.h5f_path, 'a')
del h5f[h5f_path] del h5f[h5f_path]
def get_attr(self, attr_name=None): def get_attr(self, attr_name):
"""
"""
h5f = h5py.File(self.h5f_path, 'r') h5f = h5py.File(self.h5f_path, 'r')
dataType, h5f_path = lables_to_path(attr_name)
return h5f[h5f_path].attrs[attr_name]
def add_attr(self, ): def add_attr(self, attr_name, attr_data):
""" h5f = h5py.File(self.h5f_path, 'a')
""" dataType, h5f_path = lables_to_path(attr_name)
pass if dataType == "attr":
h5f[h5f_path].attrs[attr_name] = attr_data
else:
raise ValueError("Unspported attr: {}".format(attr_name))
def get_data(self, feature_name=None): def get_data(self, feature_name=None):
""" extract dataset from HDF5 table and return it in a numpy array """ """ extract dataset from HDF5 table and return it in a numpy array """
@ -94,3 +100,9 @@ class H5Table(object):
h5f.create_dataset(h5f_path, data=dataset) h5f.create_dataset(h5f_path, data=dataset)
else: else:
raise ValueError("feature {} isn't valid".format(feature_name)) raise ValueError("feature {} isn't valid".format(feature_name))
def get_log(self, feature_name):
""" get cmd history used to generate the data"""
dataType, h5f_path = lables_to_path(feature_name)
h5f = ht5py.File(self.h5f_path, 'r')
return h5f[h5f_path].attrs['log']