shorter code

This commit is contained in:
Philip Eisenlohr 2022-03-09 12:00:17 -05:00
parent f848007878
commit 3d9ac817bb
1 changed files with 5 additions and 10 deletions

View File

@ -473,10 +473,7 @@ class Crystal():
@property @property
def orientation_relationships(self): def orientation_relationships(self):
"""Return labels of orientation relationships.""" """Return labels of orientation relationships."""
labels = [] return [k for k,v in orientation_relationships.items() if self.lattice in v]
for k,v in orientation_relationships.items():
if list(v.items())[1][0] == self.lattice: labels.append(k)
return labels
@property @property
@ -1006,12 +1003,10 @@ class Crystal():
} }
} }
master = _kinematics[self.lattice][mode] master = _kinematics[self.lattice][mode]
if self.lattice == 'hP': return {'direction':[util.Bravais_to_Miller(uvtw=m[:,0:4]) if self.lattice == 'hP'
return {'direction':[util.Bravais_to_Miller(uvtw=m[:,0:4]) for m in master], else m[:,0:3] for m in master],
'plane': [util.Bravais_to_Miller(hkil=m[:,4:8]) for m in master]} 'plane': [util.Bravais_to_Miller(hkil=m[:,4:8]) if self.lattice == 'hP'
else: else m[:,3:6] for m in master]}
return {'direction':[m[:,0:3] for m in master],
'plane': [m[:,3:6] for m in master]}
def relation_operations(self, def relation_operations(self,