able to take phaseID into account
This commit is contained in:
parent
dbab3c3a83
commit
2951617e28
|
@ -104,7 +104,7 @@ class ConfigMaterial(Config):
|
||||||
|
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def load_from_Dream3D(fname,base_group,grain_data,phase_name):
|
def load_from_Dream3D(fname,base_group,grain_data,phase_namei,phase_id):
|
||||||
"""
|
"""
|
||||||
Load material data from DREAM3D file.
|
Load material data from DREAM3D file.
|
||||||
The parts of homogenization and phase need to be added by the user.
|
The parts of homogenization and phase need to be added by the user.
|
||||||
|
@ -121,6 +121,9 @@ class ConfigMaterial(Config):
|
||||||
for example 'EulerAngles'.
|
for example 'EulerAngles'.
|
||||||
phase_name : list
|
phase_name : list
|
||||||
List with name of the phases.
|
List with name of the phases.
|
||||||
|
phase_id : str
|
||||||
|
Name of the dataset containing phase IDs for each grain,
|
||||||
|
for example 'Phases'.
|
||||||
|
|
||||||
Examples
|
Examples
|
||||||
--------
|
--------
|
||||||
|
@ -149,11 +152,19 @@ class ConfigMaterial(Config):
|
||||||
"""
|
"""
|
||||||
root_dir = 'DataContainers'
|
root_dir = 'DataContainers'
|
||||||
hdf = h5py.File(fname,'r')
|
hdf = h5py.File(fname,'r')
|
||||||
|
|
||||||
config_info = ConfigMaterial() # empty yaml dictionary
|
config_info = ConfigMaterial() # empty yaml dictionary
|
||||||
|
|
||||||
orientation_path = path.join(root_dir,base_group,grain_data)
|
orientation_path = path.join(root_dir,base_group,grain_data)
|
||||||
grain_orientations = np.array(hdf[orientation_path])[1:]
|
grain_orientations = np.array(hdf[orientation_path])[1:]
|
||||||
grain_quats = Rotation.from_Euler_angles(grain_orientations).as_quaternion()
|
grain_quats = Rotation.from_Euler_angles(grain_orientations).as_quaternion()
|
||||||
material_dict = config_info.material_add(constituents={'phase':phase_name[0],'O':grain_quats},homogenization='SX')
|
|
||||||
|
phase_path = path.join(root_dir,base_group,phase_id)
|
||||||
|
grain_phase = np.array(hdf[phase_path])[1:]
|
||||||
|
grain_phase = grain_phase.reshape(len(grain_phase),)
|
||||||
|
phase_name_list = [phase_name[i - 1] for i in grain_phase]
|
||||||
|
|
||||||
|
material_dict = config_info.material_add(constituents={'phase':phase_name_list,'O':grain_quats},homogenization='SX')
|
||||||
material_dict.save()
|
material_dict.save()
|
||||||
|
|
||||||
@property
|
@property
|
||||||
|
|
Loading…
Reference in New Issue