DAMASK_EICMD/python/damask/_configmaterial.py

330 lines
11 KiB
Python
Raw Normal View History

import numpy as np
from . import Config
from . import Rotation
from . import Orientation
from . import util
class ConfigMaterial(Config):
"""Material configuration."""
_defaults = {'material': [],
'homogenization': {},
'phase': {}}
def __init__(self,d=_defaults):
"""Initialize object with default dictionary keys."""
super().__init__(d)
def save(self,fname='material.yaml',**kwargs):
2020-09-30 12:19:55 +05:30
"""
Save to yaml file.
Parameters
----------
fname : file, str, or pathlib.Path, optional
Filename or file for writing. Defaults to 'material.yaml'.
2020-10-09 17:49:19 +05:30
**kwargs
Keyword arguments parsed to yaml.dump.
2020-09-30 12:19:55 +05:30
"""
super().save(fname,**kwargs)
2020-09-30 12:19:55 +05:30
2020-10-09 11:15:20 +05:30
@classmethod
def load(cls,fname='material.yaml'):
"""
Load from yaml file.
Parameters
----------
fname : file, str, or pathlib.Path, optional
Filename or file for writing. Defaults to 'material.yaml'.
"""
return super(ConfigMaterial,cls).load(fname)
2020-10-09 11:15:20 +05:30
@staticmethod
def from_table(table,**kwargs):
2020-10-09 17:49:19 +05:30
"""
Load from an ASCII table.
Parameters
----------
table : damask.Table
2020-10-09 17:49:19 +05:30
Table that contains material information.
**kwargs
Keyword arguments where the key is the name and the value specifies
the label of the data column in the table.
2020-10-09 17:49:19 +05:30
Examples
--------
>>> import damask
>>> import damask.ConfigMaterial as cm
>>> t = damask.Table.load('small.txt')
>>> t
2020-10-09 17:49:19 +05:30
pos pos pos qu qu qu qu phase homog
0 0 0 0 0.19 0.8 0.24 -0.51 Aluminum SX
1 1 0 0 0.8 0.19 0.24 -0.51 Steel SX
1 1 1 0 0.8 0.19 0.24 -0.51 Steel SX
>>> cm.from_table(t,O='qu',phase='phase',homogenization='homog')
2020-10-09 17:49:19 +05:30
material:
- constituents:
- O: [0.19, 0.8, 0.24, -0.51]
2021-02-04 18:16:01 +05:30
v: 1.0
2020-10-09 17:49:19 +05:30
phase: Aluminum
homogenization: SX
- constituents:
- O: [0.8, 0.19, 0.24, -0.51]
2021-02-04 18:16:01 +05:30
v: 1.0
2020-10-09 17:49:19 +05:30
phase: Steel
homogenization: SX
2020-12-18 11:39:05 +05:30
homogenization: {}
phase: {}
2020-10-09 17:49:19 +05:30
"""
kwargs_ = {k:table.get(v) for k,v in kwargs.items()}
2020-10-09 11:15:20 +05:30
_,idx = np.unique(np.hstack(list(kwargs_.values())),return_index=True,axis=0)
idx = np.sort(idx)
kwargs_ = {k:np.atleast_1d(v[idx].squeeze()) for k,v in kwargs_.items()}
2020-10-09 11:15:20 +05:30
return ConfigMaterial().material_add(**kwargs_)
2020-10-09 11:15:20 +05:30
@property
def is_complete(self):
"""Check for completeness."""
ok = True
2020-10-02 21:21:33 +05:30
for top_level in ['homogenization','phase','material']:
ok &= top_level in self
if top_level not in self: print(f'{top_level} entry missing')
if ok:
2020-10-02 21:21:33 +05:30
ok &= len(self['material']) > 0
if len(self['material']) < 1: print('Incomplete material definition')
if ok:
homogenization = set()
phase = set()
2020-10-02 21:21:33 +05:30
for i,v in enumerate(self['material']):
if 'homogenization' in v:
homogenization.add(v['homogenization'])
else:
2020-10-02 21:21:33 +05:30
print(f'No homogenization specified in material {i}')
ok = False
if 'constituents' in v:
for ii,vv in enumerate(v['constituents']):
2020-10-02 21:21:33 +05:30
if 'O' not in vv:
print('No orientation specified in constituent {ii} of material {i}')
ok = False
if 'phase' in vv:
phase.add(vv['phase'])
else:
2020-10-02 21:21:33 +05:30
print(f'No phase specified in constituent {ii} of material {i}')
ok = False
for k,v in self['phase'].items():
if 'lattice' not in v:
print(f'No lattice specified in phase {k}')
ok = False
for k,v in self['homogenization'].items():
if 'N_constituents' not in v:
print(f'No. of constituents not specified in homogenization {k}')
ok = False
if phase - set(self['phase']):
print(f'Phase(s) {phase-set(self["phase"])} missing')
ok = False
if homogenization - set(self['homogenization']):
print(f'Homogenization(s) {homogenization-set(self["homogenization"])} missing')
ok = False
return ok
@property
def is_valid(self):
2021-02-24 05:24:55 +05:30
"""Check for valid content."""
ok = True
if 'phase' in self:
for k,v in self['phase'].items():
if 'lattice' in v:
try:
Orientation(lattice=v['lattice'])
except KeyError:
2021-02-24 05:24:55 +05:30
print(f"Invalid lattice '{v['lattice']}' in phase '{k}'")
ok = False
2020-10-02 21:21:33 +05:30
if 'material' in self:
for i,m in enumerate(self['material']):
if 'constituents' in m:
v = 0.0
for c in m['constituents']:
2021-02-24 05:24:55 +05:30
v += float(c['v'])
2020-10-02 21:21:33 +05:30
if 'O' in c:
try:
2020-10-02 21:21:33 +05:30
Rotation.from_quaternion(c['O'])
except ValueError:
2021-02-24 05:24:55 +05:30
print(f"Invalid orientation '{c['O']}' in material '{i}'")
ok = False
if not np.isclose(v,1.0):
2021-02-24 05:24:55 +05:30
print(f"Total fraction v = {v} ≠ 1 in material '{i}'")
ok = False
return ok
2020-10-02 21:21:33 +05:30
def material_rename_phase(self,mapping,ID=None,constituent=None):
"""
2020-10-02 21:21:33 +05:30
Change phase name in material.
Parameters
----------
mapping: dictionary
Mapping from old name to new name
ID: list of ints, optional
2020-10-02 21:21:33 +05:30
Limit renaming to selected material IDs.
constituent: list of ints, optional
Limit renaming to selected constituents.
2021-02-24 05:24:55 +05:30
Returns
-------
cfg : damask.ConfigMaterial
Updated material configuration.
"""
2021-01-03 16:33:40 +05:30
dup = self.copy()
2020-10-02 21:21:33 +05:30
for i,m in enumerate(dup['material']):
2021-02-19 21:04:28 +05:30
if ID is not None and i not in ID: continue
for c in m['constituents']:
if constituent is not None and c not in constituent: continue
try:
c['phase'] = mapping[c['phase']]
except KeyError:
continue
return dup
2020-10-02 21:21:33 +05:30
def material_rename_homogenization(self,mapping,ID=None):
"""
2020-10-02 21:21:33 +05:30
Change homogenization name in material.
Parameters
----------
mapping: dictionary
Mapping from old name to new name
ID: list of ints, optional
Limit renaming to selected homogenization IDs.
2021-02-24 05:24:55 +05:30
Returns
-------
cfg : damask.ConfigMaterial
Updated material configuration.
"""
2021-01-03 16:33:40 +05:30
dup = self.copy()
2020-10-02 21:21:33 +05:30
for i,m in enumerate(dup['material']):
2021-02-19 21:04:28 +05:30
if ID is not None and i not in ID: continue
try:
m['homogenization'] = mapping[m['homogenization']]
except KeyError:
continue
return dup
def material_add(self,**kwargs):
"""
Add material entries.
Parameters
----------
2020-10-09 17:49:19 +05:30
**kwargs
2020-10-29 02:22:51 +05:30
Key-value pairs.
2021-02-24 05:10:32 +05:30
Returns
-------
cfg : damask.ConfigMaterial
Updated material configuration.
Examples
--------
2021-02-24 05:10:32 +05:30
>>> import numpy as np
2020-10-09 17:49:19 +05:30
>>> import damask
2021-02-24 05:10:32 +05:30
>>> m = damask.ConfigMaterial().material_add(phase = ['Aluminum','Steel'],
... O = damask.Rotation.from_random(2),
... homogenization = 'SX')
2020-10-30 00:39:13 +05:30
>>> m
2020-10-09 17:49:19 +05:30
material:
- constituents:
- O: [0.577764, -0.146299, -0.617669, 0.513010]
2021-02-04 18:16:01 +05:30
v: 1.0
2020-10-09 17:49:19 +05:30
phase: Aluminum
homogenization: SX
- constituents:
- O: [0.184176, 0.340305, 0.737247, 0.553840]
2021-02-04 18:16:01 +05:30
v: 1.0
2020-10-09 17:49:19 +05:30
phase: Steel
homogenization: SX
2021-02-24 05:10:32 +05:30
homogenization: {}
phase: {}
>>> m = damask.ConfigMaterial().material_add(phase = np.array(['Austenite','Martensite']).reshape(1,2),
... O = damask.Rotation.from_random((2,2)),
... v = np.array([0.2,0.8]).reshape(1,2),
... homogenization = ['A','B'])
>>> m
material:
2020-10-09 17:49:19 +05:30
- constituents:
2021-02-24 05:10:32 +05:30
- phase: Austenite
O: [0.659802978293224, 0.6953785848195171, 0.22426295326327111, -0.17554139512785227]
v: 0.2
- phase: Martensite
O: [0.49356745891301596, 0.2841806579193434, -0.7487679215072818, -0.339085707289975]
v: 0.8
homogenization: A
- constituents:
- phase: Austenite
O: [0.26542221365204055, 0.7268854930702071, 0.4474726435701472, -0.44828201137283735]
v: 0.2
- phase: Martensite
O: [0.6545817158479885, -0.08004812803625233, -0.6226561293931374, 0.4212059104577611]
v: 0.8
homogenization: B
2020-12-18 11:39:05 +05:30
homogenization: {}
phase: {}
"""
N,n,shaped = 1,1,{}
for k,v in kwargs.items():
shaped[k] = np.array(v)
s = shaped[k].shape[:-1] if k=='O' else shaped[k].shape
N = max(N,s[0]) if len(s)>0 else N
n = max(n,s[1]) if len(s)>1 else n
mat = [{'constituents':[{} for _ in range(n)]} for _ in range(N)]
if 'v' not in kwargs:
shaped['v'] = np.broadcast_to(1/n,(N,n))
for k,v in shaped.items():
2021-02-24 05:31:10 +05:30
target = (N,n,4) if k=='O' else (N,n)
obj = np.broadcast_to(v.reshape(util.shapeshifter(v.shape,target,mode='right')),target)
for i in range(N):
if k in ['phase','O','v']:
for j in range(n):
mat[i]['constituents'][j][k] = obj[i,j].item() if isinstance(obj[i,j],np.generic) else obj[i,j]
else:
mat[i][k] = obj[i,0].item() if isinstance(obj[i,0],np.generic) else obj[i,0]
dup = self.copy()
dup['material'] = dup['material'] + mat if 'material' in dup else mat
return dup