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

directly test the internal conversions 

some work to do ...
This commit is contained in:
Martin Diehl 2020-05-19 09:55:23 +02:00
parent 065c624f94
commit de79a5af43
3 changed files with 54 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
python/damask/_rotation.py
View File

@ -468,8 +468,8 @@ class Rotation:
weights = np.ones(N,dtype='i')
for i,(r,n) in enumerate(zip(rotations,weights)):
M = r.asM() * n if i == 0 \
else M + r.asM() * n # noqa add (multiples) of this rotation to average noqa
M = r.M() * n if i == 0 \
else M + r.M() * n # noqa add (multiples) of this rotation to average noqa
eig, vec = np.linalg.eig(M/N)
return Rotation.from_quaternion(np.real(vec.T[eig.argmax()]),accept_homomorph = True)
@ -661,6 +661,7 @@ class Rotation:
)
)
)*np.array([1,_P,_P,_P])
qu[qu[...,0]<0] *=-1
return qu
@staticmethod

1
python/tests/rotation_conversion.py
View File

@ -129,6 +129,7 @@ def om2qu(a):
else:
s = 2.0 * np.sqrt( 1.0 + a[2,2] - a[0,0] - a[1,1] )
qu = np.array([ (a[1,0] - a[0,1]) / s,(a[0,2] + a[2,0]) / s,(a[1,2] + a[2,1]) / s,0.25 * s])
if qu[0]<0: qu*=-1
return qu*np.array([1.,_P,_P,_P])
def om2eu(om):

50
python/tests/test_Rotation.py
View File

@ -91,6 +91,56 @@ def reference_dir(reference_dir_base):
class TestRotation:
@pytest.mark.parametrize('forward,backward',[(Rotation.qu2om,Rotation.om2qu),
(Rotation.qu2eu,Rotation.eu2qu),
(Rotation.qu2ax,Rotation.ax2qu),
(Rotation.qu2ro,Rotation.ro2qu),
(Rotation.qu2ho,Rotation.ho2qu),
(Rotation.qu2cu,Rotation.cu2qu)])
def test_quaternion_internal(self,default,forward,backward):
for rot in default:
m = rot.as_quaternion()
o = backward(forward(m))
ok = np.allclose(m,o,atol=atol)
if np.isclose(rot.as_quaternion()[0],0.0,atol=atol):
ok = ok or np.allclose(m*-1.,o,atol=atol)
print(m,o,rot.as_quaternion())
assert ok and np.isclose(np.linalg.norm(o),1.0)
@pytest.mark.parametrize('forward,backward',[(Rotation.om2qu,Rotation.qu2om)])
#(Rotation.om2eu,Rotation.eu2om),
#(Rotation.om2ax,Rotation.ax2om),
#(Rotation.om2ro,Rotation.ro2om),
#(Rotation.om2ho,Rotation.ho2om),
#(Rotation.om2cu,Rotation.cu2om)])
def test_matrix_internal(self,default,forward,backward):
for rot in default:
m = rot.as_matrix()
o = backward(forward(m))
ok = np.allclose(m,o,atol=atol)
print(m,o,rot.as_quaternion())
assert ok and np.isclose(np.linalg.det(o),1.0)
@pytest.mark.parametrize('forward,backward',[(Rotation.eu2qu,Rotation.qu2eu)])
#(Rotation.eu2om,Rotation.om2eu),
#(Rotation.eu2ax,Rotation.ax2eu),
#(Rotation.eu2ro,Rotation.ro2eu),
#(Rotation.eu2ho,Rotation.ho2eu),
#(Rotation.eu2cu,Rotation.cu2eu)])
def test_Eulers_internal(self,default,forward,backward):
for rot in default:
m = rot.as_Eulers()
o = backward(forward(m))
u = np.array([np.pi*2,np.pi,np.pi*2])
ok = np.allclose(m,o,atol=atol)
ok = ok or np.allclose(np.where(np.isclose(m,u),m-u,m),np.where(np.isclose(o,u),o-u,o),atol=atol)
if np.isclose(m[1],0.0,atol=atol) or np.isclose(m[1],np.pi,atol=atol):
sum_phi = np.unwrap([m[0]+m[2],o[0]+o[2]])
ok = ok or np.isclose(sum_phi[0],sum_phi[1],atol=atol)
print(m,o,rot.as_quaternion())
assert ok and (np.zeros(3)-1.e-9 <= o).all() and (o <= np.array([np.pi*2.,np.pi,np.pi*2.])+1.e-9).all()
@pytest.mark.parametrize('degrees',[True,False])
def test_Eulers(self,default,degrees):
for rot in default: