source: trunk/GSASIIlattice.py @ 2120

Last change on this file since 2120 was 2120, checked in by vondreele, 6 years ago

Add plotting of 2D modulation vector result (a0g) for monoclinic.
Trap excessive approximations in TOF2dsp
Add a general purpose contour plot
restrict maxH = 1 for modulation search; higher maxH not effective

  • Property svn:eol-style set to native
  • Property svn:keywords set to Date Author Revision URL Id
File size: 62.1 KB
Line 
1# -*- coding: utf-8 -*-
2'''
3*GSASIIlattice: Unit cells*
4---------------------------
5
6Perform lattice-related computations
7
8Note that *g* is the reciprocal lattice tensor, and *G* is its inverse,
9:math:`G = g^{-1}`, where
10
11  .. math::
12
13   G = \\left( \\begin{matrix}
14   a^2 & a b\\cos\gamma & a c\\cos\\beta \\\\
15   a b\\cos\\gamma & b^2 & b c \cos\\alpha \\\\
16   a c\\cos\\beta &  b c \\cos\\alpha & c^2
17   \\end{matrix}\\right)
18
19The "*A* tensor" terms are defined as
20:math:`A = (\\begin{matrix} G_{11} & G_{22} & G_{33} & 2G_{12} & 2G_{13} & 2G_{23}\\end{matrix})` and *A* can be used in this fashion:
21:math:`d^* = \sqrt {A_1 h^2 + A_2 k^2 + A_3 l^2 + A_4 hk + A_5 hl + A_6 kl}`, where
22*d* is the d-spacing, and :math:`d^*` is the reciprocal lattice spacing,
23:math:`Q = 2 \\pi d^* = 2 \\pi / d`
24'''
25########### SVN repository information ###################
26# $Date: 2016-01-11 20:12:00 +0000 (Mon, 11 Jan 2016) $
27# $Author: vondreele $
28# $Revision: 2120 $
29# $URL: trunk/GSASIIlattice.py $
30# $Id: GSASIIlattice.py 2120 2016-01-11 20:12:00Z vondreele $
31########### SVN repository information ###################
32import math
33import numpy as np
34import numpy.linalg as nl
35import GSASIIpath
36import GSASIImath as G2mth
37import GSASIIspc as G2spc
38GSASIIpath.SetVersionNumber("$Revision: 2120 $")
39# trig functions in degrees
40sind = lambda x: np.sin(x*np.pi/180.)
41asind = lambda x: 180.*np.arcsin(x)/np.pi
42tand = lambda x: np.tan(x*np.pi/180.)
43atand = lambda x: 180.*np.arctan(x)/np.pi
44atan2d = lambda y,x: 180.*np.arctan2(y,x)/np.pi
45cosd = lambda x: np.cos(x*np.pi/180.)
46acosd = lambda x: 180.*np.arccos(x)/np.pi
47rdsq2d = lambda x,p: round(1.0/np.sqrt(x),p)
48rpd = np.pi/180.
49RSQ2PI = 1./np.sqrt(2.*np.pi)
50SQ2 = np.sqrt(2.)
51RSQPI = 1./np.sqrt(np.pi)
52R2pisq = 1./(2.*np.pi**2)
53
54def sec2HMS(sec):
55    """Convert time in sec to H:M:S string
56   
57    :param sec: time in seconds
58    :return: H:M:S string (to nearest 100th second)
59   
60    """
61    H = int(sec/3600)
62    M = int(sec/60-H*60)
63    S = sec-3600*H-60*M
64    return '%d:%2d:%.2f'%(H,M,S)
65   
66def rotdMat(angle,axis=0):
67    """Prepare rotation matrix for angle in degrees about axis(=0,1,2)
68
69    :param angle: angle in degrees
70    :param axis:  axis (0,1,2 = x,y,z) about which for the rotation
71    :return: rotation matrix - 3x3 numpy array
72
73    """
74    if axis == 2:
75        return np.array([[cosd(angle),-sind(angle),0],[sind(angle),cosd(angle),0],[0,0,1]])
76    elif axis == 1:
77        return np.array([[cosd(angle),0,-sind(angle)],[0,1,0],[sind(angle),0,cosd(angle)]])
78    else:
79        return np.array([[1,0,0],[0,cosd(angle),-sind(angle)],[0,sind(angle),cosd(angle)]])
80       
81def rotdMat4(angle,axis=0):
82    """Prepare rotation matrix for angle in degrees about axis(=0,1,2) with scaling for OpenGL
83
84    :param angle: angle in degrees
85    :param axis:  axis (0,1,2 = x,y,z) about which for the rotation
86    :return: rotation matrix - 4x4 numpy array (last row/column for openGL scaling)
87
88    """
89    Mat = rotdMat(angle,axis)
90    return np.concatenate((np.concatenate((Mat,[[0],[0],[0]]),axis=1),[[0,0,0,1],]),axis=0)
91   
92def fillgmat(cell):
93    """Compute lattice metric tensor from unit cell constants
94
95    :param cell: tuple with a,b,c,alpha, beta, gamma (degrees)
96    :return: 3x3 numpy array
97
98    """
99    a,b,c,alp,bet,gam = cell
100    g = np.array([
101        [a*a,  a*b*cosd(gam),  a*c*cosd(bet)],
102        [a*b*cosd(gam),  b*b,  b*c*cosd(alp)],
103        [a*c*cosd(bet) ,b*c*cosd(alp),   c*c]])
104    return g
105           
106def cell2Gmat(cell):
107    """Compute real and reciprocal lattice metric tensor from unit cell constants
108
109    :param cell: tuple with a,b,c,alpha, beta, gamma (degrees)
110    :return: reciprocal (G) & real (g) metric tensors (list of two numpy 3x3 arrays)
111
112    """
113    g = fillgmat(cell)
114    G = nl.inv(g)       
115    return G,g
116
117def A2Gmat(A,inverse=True):
118    """Fill real & reciprocal metric tensor (G) from A.
119
120    :param A: reciprocal metric tensor elements as [G11,G22,G33,2*G12,2*G13,2*G23]
121    :param bool inverse: if True return both G and g; else just G
122    :return: reciprocal (G) & real (g) metric tensors (list of two numpy 3x3 arrays)
123
124    """
125    G = np.zeros(shape=(3,3))
126    G = [
127        [A[0],  A[3]/2.,  A[4]/2.], 
128        [A[3]/2.,A[1],    A[5]/2.], 
129        [A[4]/2.,A[5]/2.,    A[2]]]
130    if inverse:
131        g = nl.inv(G)
132        return G,g
133    else:
134        return G
135
136def Gmat2A(G):
137    """Extract A from reciprocal metric tensor (G)
138
139    :param G: reciprocal maetric tensor (3x3 numpy array
140    :return: A = [G11,G22,G33,2*G12,2*G13,2*G23]
141
142    """
143    return [G[0][0],G[1][1],G[2][2],2.*G[0][1],2.*G[0][2],2.*G[1][2]]
144   
145def cell2A(cell):
146    """Obtain A = [G11,G22,G33,2*G12,2*G13,2*G23] from lattice parameters
147
148    :param cell: [a,b,c,alpha,beta,gamma] (degrees)
149    :return: G reciprocal metric tensor as 3x3 numpy array
150
151    """
152    G,g = cell2Gmat(cell)
153    return Gmat2A(G)
154
155def A2cell(A):
156    """Compute unit cell constants from A
157
158    :param A: [G11,G22,G33,2*G12,2*G13,2*G23] G - reciprocal metric tensor
159    :return: a,b,c,alpha, beta, gamma (degrees) - lattice parameters
160
161    """
162    G,g = A2Gmat(A)
163    return Gmat2cell(g)
164
165def Gmat2cell(g):
166    """Compute real/reciprocal lattice parameters from real/reciprocal metric tensor (g/G)
167    The math works the same either way.
168
169    :param g (or G): real (or reciprocal) metric tensor 3x3 array
170    :return: a,b,c,alpha, beta, gamma (degrees) (or a*,b*,c*,alpha*,beta*,gamma* degrees)
171
172    """
173    oldset = np.seterr('raise')
174    a = np.sqrt(max(0,g[0][0]))
175    b = np.sqrt(max(0,g[1][1]))
176    c = np.sqrt(max(0,g[2][2]))
177    alp = acosd(g[2][1]/(b*c))
178    bet = acosd(g[2][0]/(a*c))
179    gam = acosd(g[0][1]/(a*b))
180    np.seterr(**oldset)
181    return a,b,c,alp,bet,gam
182
183def invcell2Gmat(invcell):
184    """Compute real and reciprocal lattice metric tensor from reciprocal
185       unit cell constants
186       
187    :param invcell: [a*,b*,c*,alpha*, beta*, gamma*] (degrees)
188    :return: reciprocal (G) & real (g) metric tensors (list of two 3x3 arrays)
189
190    """
191    G = fillgmat(invcell)
192    g = nl.inv(G)
193    return G,g
194       
195def calc_rVsq(A):
196    """Compute the square of the reciprocal lattice volume (1/V**2) from A'
197
198    """
199    G,g = A2Gmat(A)
200    rVsq = nl.det(G)
201    if rVsq < 0:
202        return 1
203    return rVsq
204   
205def calc_rV(A):
206    """Compute the reciprocal lattice volume (V*) from A
207    """
208    return np.sqrt(calc_rVsq(A))
209   
210def calc_V(A):
211    """Compute the real lattice volume (V) from A
212    """
213    return 1./calc_rV(A)
214
215def A2invcell(A):
216    """Compute reciprocal unit cell constants from A
217    returns tuple with a*,b*,c*,alpha*, beta*, gamma* (degrees)
218    """
219    G,g = A2Gmat(A)
220    return Gmat2cell(G)
221   
222def Gmat2AB(G):
223    """Computes orthogonalization matrix from reciprocal metric tensor G
224
225    :returns: tuple of two 3x3 numpy arrays (A,B)
226
227       * A for crystal to Cartesian transformations A*x = np.inner(A,x) = X
228       * B (= inverse of A) for Cartesian to crystal transformation B*X = np.inner(B,X) = x
229
230    """
231    cellstar = Gmat2cell(G)
232    g = nl.inv(G)
233    cell = Gmat2cell(g)
234    A = np.zeros(shape=(3,3))
235    # from Giacovazzo (Fundamentals 2nd Ed.) p.75
236    A[0][0] = cell[0]                # a
237    A[0][1] = cell[1]*cosd(cell[5])  # b cos(gamma)
238    A[0][2] = cell[2]*cosd(cell[4])  # c cos(beta)
239    A[1][1] = cell[1]*sind(cell[5])  # b sin(gamma)
240    A[1][2] = -cell[2]*cosd(cellstar[3])*sind(cell[4]) # - c cos(alpha*) sin(beta)
241    A[2][2] = 1/cellstar[2]         # 1/c*
242    B = nl.inv(A)
243    return A,B
244   
245
246def cell2AB(cell):
247    """Computes orthogonalization matrix from unit cell constants
248
249    :param tuple cell: a,b,c, alpha, beta, gamma (degrees)
250    :returns: tuple of two 3x3 numpy arrays (A,B)
251       A for crystal to Cartesian transformations A*x = np.inner(A,x) = X
252       B (= inverse of A) for Cartesian to crystal transformation B*X = np.inner(B,X) = x
253    """
254    G,g = cell2Gmat(cell) 
255    cellstar = Gmat2cell(G)
256    A = np.zeros(shape=(3,3))
257    # from Giacovazzo (Fundamentals 2nd Ed.) p.75
258    A[0][0] = cell[0]                # a
259    A[0][1] = cell[1]*cosd(cell[5])  # b cos(gamma)
260    A[0][2] = cell[2]*cosd(cell[4])  # c cos(beta)
261    A[1][1] = cell[1]*sind(cell[5])  # b sin(gamma)
262    A[1][2] = -cell[2]*cosd(cellstar[3])*sind(cell[4]) # - c cos(alpha*) sin(beta)
263    A[2][2] = 1/cellstar[2]         # 1/c*
264    B = nl.inv(A)
265    return A,B
266   
267def U6toUij(U6):
268    """Fill matrix (Uij) from U6 = [U11,U22,U33,U12,U13,U23]
269    NB: there is a non numpy version in GSASIIspc: U2Uij
270
271    :param list U6: 6 terms of u11,u22,...
272    :returns:
273        Uij - numpy [3][3] array of uij
274    """
275    U = np.array([
276        [U6[0],  U6[3],  U6[4]], 
277        [U6[3],  U6[1],  U6[5]], 
278        [U6[4],  U6[5],  U6[2]]])
279    return U
280
281def UijtoU6(U):
282    """Fill vector [U11,U22,U33,U12,U13,U23] from Uij
283    NB: there is a non numpy version in GSASIIspc: Uij2U
284    """
285    U6 = np.array([U[0][0],U[1][1],U[2][2],U[0][1],U[0][2],U[1][2]])
286    return U6
287
288def betaij2Uij(betaij,G):
289    """
290    Convert beta-ij to Uij tensors
291   
292    :param beta-ij - numpy array [beta-ij]
293    :param G: reciprocal metric tensor
294    :returns: Uij: numpy array [Uij]
295    """
296    ast = np.sqrt(np.diag(G))   #a*, b*, c*
297    Mast = np.multiply.outer(ast,ast)   
298    return R2pisq*UijtoU6(U6toUij(betaij)/Mast)
299   
300def Uij2betaij(Uij,G):
301    """
302    Convert Uij to beta-ij tensors -- stub for eventual completion
303   
304    :param Uij: numpy array [Uij]
305    :param G: reciprocal metric tensor
306    :returns: beta-ij - numpy array [beta-ij]
307    """
308    pass
309   
310def cell2GS(cell):
311    ''' returns Uij to betaij conversion matrix'''
312    G,g = cell2Gmat(cell)
313    GS = G
314    GS[0][1] = GS[1][0] = math.sqrt(GS[0][0]*GS[1][1])
315    GS[0][2] = GS[2][0] = math.sqrt(GS[0][0]*GS[2][2])
316    GS[1][2] = GS[2][1] = math.sqrt(GS[1][1]*GS[2][2])
317    return GS   
318   
319def Uij2Ueqv(Uij,GS,Amat):
320    ''' returns 1/3 trace of diagonalized U matrix'''
321    U = np.multiply(U6toUij(Uij),GS)
322    U = np.inner(Amat,np.inner(U,Amat).T)
323    E,R = nl.eigh(U)
324    return np.sum(E)/3.
325       
326def CosAngle(U,V,G):
327    """ calculate cos of angle between U & V in generalized coordinates
328    defined by metric tensor G
329
330    :param U: 3-vectors assume numpy arrays, can be multiple reflections as (N,3) array
331    :param V: 3-vectors assume numpy arrays, only as (3) vector
332    :param G: metric tensor for U & V defined space assume numpy array
333    :returns:
334        cos(phi)
335    """
336    u = (U.T/np.sqrt(np.sum(np.inner(U,G)*U,axis=1))).T
337    v = V/np.sqrt(np.inner(V,np.inner(G,V)))
338    cosP = np.inner(u,np.inner(G,v))
339    return cosP
340   
341def CosSinAngle(U,V,G):
342    """ calculate sin & cos of angle between U & V in generalized coordinates
343    defined by metric tensor G
344
345    :param U: 3-vectors assume numpy arrays
346    :param V: 3-vectors assume numpy arrays
347    :param G: metric tensor for U & V defined space assume numpy array
348    :returns:
349        cos(phi) & sin(phi)
350    """
351    u = U/np.sqrt(np.inner(U,np.inner(G,U)))
352    v = V/np.sqrt(np.inner(V,np.inner(G,V)))
353    cosP = np.inner(u,np.inner(G,v))
354    sinP = np.sqrt(max(0.0,1.0-cosP**2))
355    return cosP,sinP
356   
357def criticalEllipse(prob):
358    """
359    Calculate critical values for probability ellipsoids from probability
360    """
361    if not ( 0.01 <= prob < 1.0):
362        return 1.54 
363    coeff = np.array([6.44988E-09,4.16479E-07,1.11172E-05,1.58767E-04,0.00130554,
364        0.00604091,0.0114921,-0.040301,-0.6337203,1.311582])
365    llpr = math.log(-math.log(prob))
366    return np.polyval(coeff,llpr)
367   
368def CellBlock(nCells):
369    """
370    Generate block of unit cells n*n*n on a side; [0,0,0] centered, n = 2*nCells+1
371    currently only works for nCells = 0 or 1 (not >1)
372    """
373    if nCells:
374        N = 2*nCells+1
375        N2 = N*N
376        N3 = N*N*N
377        cellArray = []
378        A = np.array(range(N3))
379        cellGen = np.array([A/N2-1,A/N%N-1,A%N-1]).T
380        for cell in cellGen:
381            cellArray.append(cell)
382        return cellArray
383    else:
384        return [0,0,0]
385       
386def CellAbsorption(ElList,Volume):
387    '''Compute unit cell absorption
388
389    :param dict ElList: dictionary of element contents including mu and
390      number of atoms be cell
391    :param float Volume: unit cell volume
392    :returns: mu-total/Volume
393    '''
394    muT = 0
395    for El in ElList:
396        muT += ElList[El]['mu']*ElList[El]['FormulaNo']
397    return muT/Volume
398   
399#Permutations and Combinations
400# Four routines: combinations,uniqueCombinations, selections & permutations
401#These taken from Python Cookbook, 2nd Edition. 19.15 p724-726
402#   
403def _combinators(_handle, items, n):
404    """ factored-out common structure of all following combinators """
405    if n==0:
406        yield [ ]
407        return
408    for i, item in enumerate(items):
409        this_one = [ item ]
410        for cc in _combinators(_handle, _handle(items, i), n-1):
411            yield this_one + cc
412def combinations(items, n):
413    """ take n distinct items, order matters """
414    def skipIthItem(items, i):
415        return items[:i] + items[i+1:]
416    return _combinators(skipIthItem, items, n)
417def uniqueCombinations(items, n):
418    """ take n distinct items, order is irrelevant """
419    def afterIthItem(items, i):
420        return items[i+1:]
421    return _combinators(afterIthItem, items, n)
422def selections(items, n):
423    """ take n (not necessarily distinct) items, order matters """
424    def keepAllItems(items, i):
425        return items
426    return _combinators(keepAllItems, items, n)
427def permutations(items):
428    """ take all items, order matters """
429    return combinations(items, len(items))
430
431#reflection generation routines
432#for these: H = [h,k,l]; A is as used in calc_rDsq; G - inv metric tensor, g - metric tensor;
433#           cell - a,b,c,alp,bet,gam in A & deg
434   
435def Pos2dsp(Inst,pos):
436    ''' convert powder pattern position (2-theta or TOF, musec) to d-spacing
437    '''
438    if 'C' in Inst['Type'][0] or 'PKS' in Inst['Type'][0]:
439        wave = G2mth.getWave(Inst)
440        return wave/(2.0*sind((pos-Inst.get('Zero',[0,0])[1])/2.0))
441    else:   #'T'OF - ignore difB
442        return TOF2dsp(Inst,pos)
443       
444def TOF2dsp(Inst,Pos):
445    ''' convert powder pattern TOF, musec to d-spacing by successive approximation
446    Pos can be numpy array
447    '''
448    def func(d,pos,Inst):       
449        return (pos-Inst['difA'][1]*d**2-Inst['Zero'][1]-Inst['difB'][1]/d)/Inst['difC'][1]
450    dsp0 = np.ones_like(Pos)
451    N = 0
452    while True:      #successive approximations
453        dsp = func(dsp0,Pos,Inst)
454        if np.allclose(dsp,dsp0,atol=0.000001):
455            return dsp
456        dsp0 = dsp
457        N += 1
458        if N > 10:
459            return dsp
460   
461def Dsp2pos(Inst,dsp):
462    ''' convert d-spacing to powder pattern position (2-theta or TOF, musec)
463    '''
464    if 'C' in Inst['Type'][0] or 'PKS' in Inst['Type'][0]:
465        wave = G2mth.getWave(Inst)
466        pos = 2.0*asind(wave/(2.*dsp))+Inst.get('Zero',[0,0])[1]             
467    else:   #'T'OF
468        pos = Inst['difC'][1]*dsp+Inst['Zero'][1]+Inst['difA'][1]*dsp**2+Inst.get('difB',[0,0,False])[1]/dsp
469    return pos
470   
471def getPeakPos(dataType,parmdict,dsp):
472    ''' convert d-spacing to powder pattern position (2-theta or TOF, musec)
473    '''
474    if 'C' in dataType:
475        pos = 2.0*asind(parmdict['Lam']/(2.*dsp))+parmdict['Zero']
476    else:   #'T'OF
477        pos = parmdict['difC']*dsp+parmdict['difA']*dsp**2+parmdict['difB']/dsp+parmdict['Zero']
478    return pos
479                   
480def calc_rDsq(H,A):
481    'needs doc string'
482    rdsq = H[0]*H[0]*A[0]+H[1]*H[1]*A[1]+H[2]*H[2]*A[2]+H[0]*H[1]*A[3]+H[0]*H[2]*A[4]+H[1]*H[2]*A[5]
483    return rdsq
484   
485def calc_rDsq2(H,G):
486    'needs doc string'
487    return np.inner(H,np.inner(G,H))
488   
489def calc_rDsqSS(H,A,vec):
490    'needs doc string'
491    rdsq = calc_rDsq(H[:3]+(H[3]*vec).T,A)
492    return rdsq
493       
494def calc_rDsqZ(H,A,Z,tth,lam):
495    'needs doc string'
496    rdsq = calc_rDsq(H,A)+Z*sind(tth)*2.0*rpd/lam**2
497    return rdsq
498       
499def calc_rDsqZSS(H,A,vec,Z,tth,lam):
500    'needs doc string'
501    rdsq = calc_rDsq(H[:3]+(H[3][:,np.newaxis]*vec).T,A)+Z*sind(tth)*2.0*rpd/lam**2
502    return rdsq
503       
504def calc_rDsqT(H,A,Z,tof,difC):
505    'needs doc string'
506    rdsq = calc_rDsq(H,A)+Z/difC
507    return rdsq
508       
509def calc_rDsqTSS(H,A,vec,Z,tof,difC):
510    'needs doc string'
511    rdsq = calc_rDsq(H[:3]+(H[3][:,np.newaxis]*vec).T,A)+Z/difC
512    return rdsq
513       
514def MaxIndex(dmin,A):
515    'needs doc string'
516    Hmax = [0,0,0]
517    try:
518        cell = A2cell(A)
519    except:
520        cell = [1,1,1,90,90,90]
521    for i in range(3):
522        Hmax[i] = int(round(cell[i]/dmin))
523    return Hmax
524   
525def sortHKLd(HKLd,ifreverse,ifdup,ifSS=False):
526    '''needs doc string
527
528    :param HKLd: a list of [h,k,l,d,...];
529    :param ifreverse: True for largest d first
530    :param ifdup: True if duplicate d-spacings allowed
531    '''
532    T = []
533    N = 3
534    if ifSS:
535        N = 4
536    for i,H in enumerate(HKLd):
537        if ifdup:
538            T.append((H[N],i))
539        else:
540            T.append(H[N])           
541    D = dict(zip(T,HKLd))
542    T.sort()
543    if ifreverse:
544        T.reverse()
545    X = []
546    okey = ''
547    for key in T: 
548        if key != okey: X.append(D[key])    #remove duplicate d-spacings
549        okey = key
550    return X
551   
552def SwapIndx(Axis,H):
553    'needs doc string'
554    if Axis in [1,-1]:
555        return H
556    elif Axis in [2,-3]:
557        return [H[1],H[2],H[0]]
558    else:
559        return [H[2],H[0],H[1]]
560       
561def Rh2Hx(Rh):
562    'needs doc string'
563    Hx = [0,0,0]
564    Hx[0] = Rh[0]-Rh[1]
565    Hx[1] = Rh[1]-Rh[2]
566    Hx[2] = np.sum(Rh)
567    return Hx
568   
569def Hx2Rh(Hx):
570    'needs doc string'
571    Rh = [0,0,0]
572    itk = -Hx[0]+Hx[1]+Hx[2]
573    if itk%3 != 0:
574        return 0        #error - not rhombohedral reflection
575    else:
576        Rh[1] = itk/3
577        Rh[0] = Rh[1]+Hx[0]
578        Rh[2] = Rh[1]-Hx[1]
579        if Rh[0] < 0:
580            for i in range(3):
581                Rh[i] = -Rh[i]
582        return Rh
583       
584def CentCheck(Cent,H):
585    'needs doc string'
586    h,k,l = H
587    if Cent == 'A' and (k+l)%2:
588        return False
589    elif Cent == 'B' and (h+l)%2:
590        return False
591    elif Cent == 'C' and (h+k)%2:
592        return False
593    elif Cent == 'I' and (h+k+l)%2:
594        return False
595    elif Cent == 'F' and ((h+k)%2 or (h+l)%2 or (k+l)%2):
596        return False
597    elif Cent == 'R' and (-h+k+l)%3:
598        return False
599    else:
600        return True
601                                   
602def GetBraviasNum(center,system):
603    """Determine the Bravais lattice number, as used in GenHBravais
604   
605    :param center: one of: 'P', 'C', 'I', 'F', 'R' (see SGLatt from GSASIIspc.SpcGroup)
606    :param system: one of 'cubic', 'hexagonal', 'tetragonal', 'orthorhombic', 'trigonal' (for R)
607      'monoclinic', 'triclinic' (see SGSys from GSASIIspc.SpcGroup)
608    :return: a number between 0 and 13
609      or throws a ValueError exception if the combination of center, system is not found (i.e. non-standard)
610
611    """
612    if center.upper() == 'F' and system.lower() == 'cubic':
613        return 0
614    elif center.upper() == 'I' and system.lower() == 'cubic':
615        return 1
616    elif center.upper() == 'P' and system.lower() == 'cubic':
617        return 2
618    elif center.upper() == 'R' and system.lower() == 'trigonal':
619        return 3
620    elif center.upper() == 'P' and system.lower() == 'hexagonal':
621        return 4
622    elif center.upper() == 'I' and system.lower() == 'tetragonal':
623        return 5
624    elif center.upper() == 'P' and system.lower() == 'tetragonal':
625        return 6
626    elif center.upper() == 'F' and system.lower() == 'orthorhombic':
627        return 7
628    elif center.upper() == 'I' and system.lower() == 'orthorhombic':
629        return 8
630    elif center.upper() == 'C' and system.lower() == 'orthorhombic':
631        return 9
632    elif center.upper() == 'P' and system.lower() == 'orthorhombic':
633        return 10
634    elif center.upper() == 'C' and system.lower() == 'monoclinic':
635        return 11
636    elif center.upper() == 'P' and system.lower() == 'monoclinic':
637        return 12
638    elif center.upper() == 'P' and system.lower() == 'triclinic':
639        return 13
640    raise ValueError,'non-standard Bravais lattice center=%s, cell=%s' % (center,system)
641
642def GenHBravais(dmin,Bravais,A):
643    """Generate the positionally unique powder diffraction reflections
644     
645    :param dmin: minimum d-spacing in A
646    :param Bravais: lattice type (see GetBraviasNum). Bravais is one of::
647             0 F cubic
648             1 I cubic
649             2 P cubic
650             3 R hexagonal (trigonal not rhombohedral)
651             4 P hexagonal
652             5 I tetragonal
653             6 P tetragonal
654             7 F orthorhombic
655             8 I orthorhombic
656             9 C orthorhombic
657             10 P orthorhombic
658             11 C monoclinic
659             12 P monoclinic
660             13 P triclinic
661           
662    :param A: reciprocal metric tensor elements as [G11,G22,G33,2*G12,2*G13,2*G23]
663    :return: HKL unique d list of [h,k,l,d,-1] sorted with largest d first
664           
665    """
666    import math
667    if Bravais in [9,11]:
668        Cent = 'C'
669    elif Bravais in [1,5,8]:
670        Cent = 'I'
671    elif Bravais in [0,7]:
672        Cent = 'F'
673    elif Bravais in [3]:
674        Cent = 'R'
675    else:
676        Cent = 'P'
677    Hmax = MaxIndex(dmin,A)
678    dminsq = 1./(dmin**2)
679    HKL = []
680    if Bravais == 13:                       #triclinic
681        for l in range(-Hmax[2],Hmax[2]+1):
682            for k in range(-Hmax[1],Hmax[1]+1):
683                hmin = 0
684                if (k < 0): hmin = 1
685                if (k ==0 and l < 0): hmin = 1
686                for h in range(hmin,Hmax[0]+1):
687                    H=[h,k,l]
688                    rdsq = calc_rDsq(H,A)
689                    if 0 < rdsq <= dminsq:
690                        HKL.append([h,k,l,rdsq2d(rdsq,6),-1])
691    elif Bravais in [11,12]:                #monoclinic - b unique
692        Hmax = SwapIndx(2,Hmax)
693        for h in range(Hmax[0]+1):
694            for k in range(-Hmax[1],Hmax[1]+1):
695                lmin = 0
696                if k < 0:lmin = 1
697                for l in range(lmin,Hmax[2]+1):
698                    [h,k,l] = SwapIndx(-2,[h,k,l])
699                    H = []
700                    if CentCheck(Cent,[h,k,l]): H=[h,k,l]
701                    if H:
702                        rdsq = calc_rDsq(H,A)
703                        if 0 < rdsq <= dminsq:
704                            HKL.append([h,k,l,rdsq2d(rdsq,6),-1])
705                    [h,k,l] = SwapIndx(2,[h,k,l])
706    elif Bravais in [7,8,9,10]:            #orthorhombic
707        for h in range(Hmax[0]+1):
708            for k in range(Hmax[1]+1):
709                for l in range(Hmax[2]+1):
710                    H = []
711                    if CentCheck(Cent,[h,k,l]): H=[h,k,l]
712                    if H:
713                        rdsq = calc_rDsq(H,A)
714                        if 0 < rdsq <= dminsq:
715                            HKL.append([h,k,l,rdsq2d(rdsq,6),-1])
716    elif Bravais in [5,6]:                  #tetragonal
717        for l in range(Hmax[2]+1):
718            for k in range(Hmax[1]+1):
719                for h in range(k,Hmax[0]+1):
720                    H = []
721                    if CentCheck(Cent,[h,k,l]): H=[h,k,l]
722                    if H:
723                        rdsq = calc_rDsq(H,A)
724                        if 0 < rdsq <= dminsq:
725                            HKL.append([h,k,l,rdsq2d(rdsq,6),-1])
726    elif Bravais in [3,4]:
727        lmin = 0
728        if Bravais == 3: lmin = -Hmax[2]                  #hexagonal/trigonal
729        for l in range(lmin,Hmax[2]+1):
730            for k in range(Hmax[1]+1):
731                hmin = k
732                if l < 0: hmin += 1
733                for h in range(hmin,Hmax[0]+1):
734                    H = []
735                    if CentCheck(Cent,[h,k,l]): H=[h,k,l]
736                    if H:
737                        rdsq = calc_rDsq(H,A)
738                        if 0 < rdsq <= dminsq:
739                            HKL.append([h,k,l,rdsq2d(rdsq,6),-1])
740
741    else:                                   #cubic
742        for l in range(Hmax[2]+1):
743            for k in range(l,Hmax[1]+1):
744                for h in range(k,Hmax[0]+1):
745                    H = []
746                    if CentCheck(Cent,[h,k,l]): H=[h,k,l]
747                    if H:
748                        rdsq = calc_rDsq(H,A)
749                        if 0 < rdsq <= dminsq:
750                            HKL.append([h,k,l,rdsq2d(rdsq,6),-1])
751    return sortHKLd(HKL,True,False)
752   
753def getHKLmax(dmin,SGData,A):
754    'finds maximum allowed hkl for given A within dmin'
755    SGLaue = SGData['SGLaue']
756    if SGLaue in ['3R','3mR']:        #Rhombohedral axes
757        Hmax = [0,0,0]
758        cell = A2cell(A)
759        aHx = cell[0]*math.sqrt(2.0*(1.0-cosd(cell[3])))
760        cHx = cell[0]*math.sqrt(3.0*(1.0+2.0*cosd(cell[3])))
761        Hmax[0] = Hmax[1] = int(round(aHx/dmin))
762        Hmax[2] = int(round(cHx/dmin))
763        #print Hmax,aHx,cHx
764    else:                           # all others
765        Hmax = MaxIndex(dmin,A)
766    return Hmax
767   
768def GenHLaue(dmin,SGData,A):
769    """Generate the crystallographically unique powder diffraction reflections
770    for a lattice and Bravais type
771   
772    :param dmin: minimum d-spacing
773    :param SGData: space group dictionary with at least
774   
775        * 'SGLaue': Laue group symbol: one of '-1','2/m','mmm','4/m','6/m','4/mmm','6/mmm', '3m1', '31m', '3', '3R', '3mR', 'm3', 'm3m'
776        * 'SGLatt': lattice centering: one of 'P','A','B','C','I','F'
777        * 'SGUniq': code for unique monoclinic axis one of 'a','b','c' (only if 'SGLaue' is '2/m') otherwise an empty string
778       
779    :param A: reciprocal metric tensor elements as [G11,G22,G33,2*G12,2*G13,2*G23]
780    :return: HKL = list of [h,k,l,d] sorted with largest d first and is unique
781            part of reciprocal space ignoring anomalous dispersion
782           
783    """
784    import math
785    SGLaue = SGData['SGLaue']
786    SGLatt = SGData['SGLatt']
787    SGUniq = SGData['SGUniq']
788    #finds maximum allowed hkl for given A within dmin
789    Hmax = getHKLmax(dmin,SGData,A)
790       
791    dminsq = 1./(dmin**2)
792    HKL = []
793    if SGLaue == '-1':                       #triclinic
794        for l in range(-Hmax[2],Hmax[2]+1):
795            for k in range(-Hmax[1],Hmax[1]+1):
796                hmin = 0
797                if (k < 0) or (k ==0 and l < 0): hmin = 1
798                for h in range(hmin,Hmax[0]+1):
799                    H = []
800                    if CentCheck(SGLatt,[h,k,l]): H=[h,k,l]
801                    if H:
802                        rdsq = calc_rDsq(H,A)
803                        if 0 < rdsq <= dminsq:
804                            HKL.append([h,k,l,1/math.sqrt(rdsq)])
805    elif SGLaue == '2/m':                #monoclinic
806        axisnum = 1 + ['a','b','c'].index(SGUniq)
807        Hmax = SwapIndx(axisnum,Hmax)
808        for h in range(Hmax[0]+1):
809            for k in range(-Hmax[1],Hmax[1]+1):
810                lmin = 0
811                if k < 0:lmin = 1
812                for l in range(lmin,Hmax[2]+1):
813                    [h,k,l] = SwapIndx(-axisnum,[h,k,l])
814                    H = []
815                    if CentCheck(SGLatt,[h,k,l]): H=[h,k,l]
816                    if H:
817                        rdsq = calc_rDsq(H,A)
818                        if 0 < rdsq <= dminsq:
819                            HKL.append([h,k,l,1/math.sqrt(rdsq)])
820                    [h,k,l] = SwapIndx(axisnum,[h,k,l])
821    elif SGLaue in ['mmm','4/m','6/m']:            #orthorhombic
822        for l in range(Hmax[2]+1):
823            for h in range(Hmax[0]+1):
824                kmin = 1
825                if SGLaue == 'mmm' or h ==0: kmin = 0
826                for k in range(kmin,Hmax[1]+1):
827                    H = []
828                    if CentCheck(SGLatt,[h,k,l]): H=[h,k,l]
829                    if H:
830                        rdsq = calc_rDsq(H,A)
831                        if 0 < rdsq <= dminsq:
832                            HKL.append([h,k,l,1/math.sqrt(rdsq)])
833    elif SGLaue in ['4/mmm','6/mmm']:                  #tetragonal & hexagonal
834        for l in range(Hmax[2]+1):
835            for h in range(Hmax[0]+1):
836                for k in range(h+1):
837                    H = []
838                    if CentCheck(SGLatt,[h,k,l]): H=[h,k,l]
839                    if H:
840                        rdsq = calc_rDsq(H,A)
841                        if 0 < rdsq <= dminsq:
842                            HKL.append([h,k,l,1/math.sqrt(rdsq)])
843    elif SGLaue in ['3m1','31m','3','3R','3mR']:                  #trigonals
844        for l in range(-Hmax[2],Hmax[2]+1):
845            hmin = 0
846            if l < 0: hmin = 1
847            for h in range(hmin,Hmax[0]+1):
848                if SGLaue in ['3R','3']:
849                    kmax = h
850                    kmin = -int((h-1.)/2.)
851                else:
852                    kmin = 0
853                    kmax = h
854                    if SGLaue in ['3m1','3mR'] and l < 0: kmax = h-1
855                    if SGLaue == '31m' and l < 0: kmin = 1
856                for k in range(kmin,kmax+1):
857                    H = []
858                    if CentCheck(SGLatt,[h,k,l]): H=[h,k,l]
859                    if SGLaue in ['3R','3mR']:
860                        H = Hx2Rh(H)
861                    if H:
862                        rdsq = calc_rDsq(H,A)
863                        if 0 < rdsq <= dminsq:
864                            HKL.append([H[0],H[1],H[2],1/math.sqrt(rdsq)])
865    else:                                   #cubic
866        for h in range(Hmax[0]+1):
867            for k in range(h+1):
868                lmin = 0
869                lmax = k
870                if SGLaue =='m3':
871                    lmax = h-1
872                    if h == k: lmax += 1
873                for l in range(lmin,lmax+1):
874                    H = []
875                    if CentCheck(SGLatt,[h,k,l]): H=[h,k,l]
876                    if H:
877                        rdsq = calc_rDsq(H,A)
878                        if 0 < rdsq <= dminsq:
879                            HKL.append([h,k,l,1/math.sqrt(rdsq)])
880    return sortHKLd(HKL,True,True)
881   
882def GenPfHKLs(nMax,SGData,A):   
883    """Generate the unique pole figure reflections for a lattice and Bravais type.
884    Min d-spacing=1.0A & no more than nMax returned
885   
886    :param nMax: maximum number of hkls returned
887    :param SGData: space group dictionary with at least
888   
889        * 'SGLaue': Laue group symbol: one of '-1','2/m','mmm','4/m','6/m','4/mmm','6/mmm', '3m1', '31m', '3', '3R', '3mR', 'm3', 'm3m'
890        * 'SGLatt': lattice centering: one of 'P','A','B','C','I','F'
891        * 'SGUniq': code for unique monoclinic axis one of 'a','b','c' (only if 'SGLaue' is '2/m') otherwise an empty string
892       
893    :param A: reciprocal metric tensor elements as [G11,G22,G33,2*G12,2*G13,2*G23]
894    :return: HKL = list of 'h k l' strings sorted with largest d first; no duplicate zones
895           
896    """
897    HKL = np.array(GenHLaue(1.0,SGData,A)).T[:3].T     #strip d-spacings
898    N = min(nMax,len(HKL))
899    return ['%d %d %d'%(h[0],h[1],h[2]) for h in HKL[:N]]       
900       
901
902def GenSSHLaue(dmin,SGData,SSGData,Vec,maxH,A):
903    'needs a doc string'
904    HKLs = []
905    vec = np.array(Vec)
906    vstar = np.sqrt(calc_rDsq(vec,A))     #find extra needed for -n SS reflections
907    dvec = 1./(maxH*vstar+1./dmin)
908    HKL = GenHLaue(dvec,SGData,A)       
909    SSdH = [vec*h for h in range(-maxH,maxH+1)]
910    SSdH = dict(zip(range(-maxH,maxH+1),SSdH))
911    for h,k,l,d in HKL:
912        ext = G2spc.GenHKLf([h,k,l],SGData)[0]  #h,k,l must be integral values here
913        if not ext and d >= dmin:
914            HKLs.append([h,k,l,0,d])
915        for dH in SSdH:
916            if dH:
917                DH = SSdH[dH]
918                H = [h+DH[0],k+DH[1],l+DH[2]]
919                d = 1/np.sqrt(calc_rDsq(H,A))
920                if d >= dmin:
921                    HKLM = np.array([h,k,l,dH])
922                    if G2spc.checkSSLaue([h,k,l,dH],SGData,SSGData) and G2spc.checkSSextc(HKLM,SSGData):
923                        HKLs.append([h,k,l,dH,d])   
924    return HKLs
925
926#Spherical harmonics routines
927def OdfChk(SGLaue,L,M):
928    'needs doc string'
929    if not L%2 and abs(M) <= L:
930        if SGLaue == '0':                      #cylindrical symmetry
931            if M == 0: return True
932        elif SGLaue == '-1':
933            return True
934        elif SGLaue == '2/m':
935            if not abs(M)%2: return True
936        elif SGLaue == 'mmm':
937            if not abs(M)%2 and M >= 0: return True
938        elif SGLaue == '4/m':
939            if not abs(M)%4: return True
940        elif SGLaue == '4/mmm':
941            if not abs(M)%4 and M >= 0: return True
942        elif SGLaue in ['3R','3']:
943            if not abs(M)%3: return True
944        elif SGLaue in ['3mR','3m1','31m']:
945            if not abs(M)%3 and M >= 0: return True
946        elif SGLaue == '6/m':
947            if not abs(M)%6: return True
948        elif SGLaue == '6/mmm':
949            if not abs(M)%6 and M >= 0: return True
950        elif SGLaue == 'm3':
951            if M > 0:
952                if L%12 == 2:
953                    if M <= L/12: return True
954                else:
955                    if M <= L/12+1: return True
956        elif SGLaue == 'm3m':
957            if M > 0:
958                if L%12 == 2:
959                    if M <= L/12: return True
960                else:
961                    if M <= L/12+1: return True
962    return False
963       
964def GenSHCoeff(SGLaue,SamSym,L,IfLMN=True):
965    'needs doc string'
966    coeffNames = []
967    for iord in [2*i+2 for i in range(L/2)]:
968        for m in [i-iord for i in range(2*iord+1)]:
969            if OdfChk(SamSym,iord,m):
970                for n in [i-iord for i in range(2*iord+1)]:
971                    if OdfChk(SGLaue,iord,n):
972                        if IfLMN:
973                            coeffNames.append('C(%d,%d,%d)'%(iord,m,n))
974                        else:
975                            coeffNames.append('C(%d,%d)'%(iord,n))
976    return coeffNames
977   
978def CrsAng(H,cell,SGData):
979    'needs doc string'
980    a,b,c,al,be,ga = cell
981    SQ3 = 1.732050807569
982    H1 = np.array([1,0,0])
983    H2 = np.array([0,1,0])
984    H3 = np.array([0,0,1])
985    H4 = np.array([1,1,1])
986    G,g = cell2Gmat(cell)
987    Laue = SGData['SGLaue']
988    Naxis = SGData['SGUniq']
989    if len(H.shape) == 1:
990        DH = np.inner(H,np.inner(G,H))
991    else:
992        DH = np.array([np.inner(h,np.inner(G,h)) for h in H])
993    if Laue == '2/m':
994        if Naxis == 'a':
995            DR = np.inner(H1,np.inner(G,H1))
996            DHR = np.inner(H,np.inner(G,H1))
997        elif Naxis == 'b':
998            DR = np.inner(H2,np.inner(G,H2))
999            DHR = np.inner(H,np.inner(G,H2))
1000        else:
1001            DR = np.inner(H3,np.inner(G,H3))
1002            DHR = np.inner(H,np.inner(G,H3))
1003    elif Laue in ['R3','R3m']:
1004        DR = np.inner(H4,np.inner(G,H4))
1005        DHR = np.inner(H,np.inner(G,H4))
1006    else:
1007        DR = np.inner(H3,np.inner(G,H3))
1008        DHR = np.inner(H,np.inner(G,H3))
1009    DHR /= np.sqrt(DR*DH)
1010    phi = np.where(DHR <= 1.0,acosd(DHR),0.0)
1011    if Laue == '-1':
1012        BA = H.T[1]*a/(b-H.T[0]*cosd(ga))
1013        BB = H.T[0]*sind(ga)**2
1014    elif Laue == '2/m':
1015        if Naxis == 'a':
1016            BA = H.T[2]*b/(c-H.T[1]*cosd(al))
1017            BB = H.T[1]*sind(al)**2
1018        elif Naxis == 'b':
1019            BA = H.T[0]*c/(a-H.T[2]*cosd(be))
1020            BB = H.T[2]*sind(be)**2
1021        else:
1022            BA = H.T[1]*a/(b-H.T[0]*cosd(ga))
1023            BB = H.T[0]*sind(ga)**2
1024    elif Laue in ['mmm','4/m','4/mmm']:
1025        BA = H.T[1]*a
1026        BB = H.T[0]*b
1027    elif Laue in ['3R','3mR']:
1028        BA = H.T[0]+H.T[1]-2.0*H.T[2]
1029        BB = SQ3*(H.T[0]-H.T[1])
1030    elif Laue in ['m3','m3m']:
1031        BA = H.T[1]
1032        BB = H.T[0]
1033    else:
1034        BA = H.T[0]+2.0*H.T[1]
1035        BB = SQ3*H.T[0]
1036    beta = atan2d(BA,BB)
1037    return phi,beta
1038   
1039def SamAng(Tth,Gangls,Sangl,IFCoup):
1040    """Compute sample orientation angles vs laboratory coord. system
1041
1042    :param Tth:        Signed theta                                   
1043    :param Gangls:     Sample goniometer angles phi,chi,omega,azmuth 
1044    :param Sangl:      Sample angle zeros om-0, chi-0, phi-0         
1045    :param IFCoup:     True if omega & 2-theta coupled in CW scan
1046    :returns: 
1047        psi,gam:    Sample odf angles                             
1048        dPSdA,dGMdA:    Angle zero derivatives
1049    """                         
1050   
1051    if IFCoup:
1052        GSomeg = sind(Gangls[2]+Tth)
1053        GComeg = cosd(Gangls[2]+Tth)
1054    else:
1055        GSomeg = sind(Gangls[2])
1056        GComeg = cosd(Gangls[2])
1057    GSTth = sind(Tth)
1058    GCTth = cosd(Tth)     
1059    GSazm = sind(Gangls[3])
1060    GCazm = cosd(Gangls[3])
1061    GSchi = sind(Gangls[1])
1062    GCchi = cosd(Gangls[1])
1063    GSphi = sind(Gangls[0]+Sangl[2])
1064    GCphi = cosd(Gangls[0]+Sangl[2])
1065    SSomeg = sind(Sangl[0])
1066    SComeg = cosd(Sangl[0])
1067    SSchi = sind(Sangl[1])
1068    SCchi = cosd(Sangl[1])
1069    AT = -GSTth*GComeg+GCTth*GCazm*GSomeg
1070    BT = GSTth*GSomeg+GCTth*GCazm*GComeg
1071    CT = -GCTth*GSazm*GSchi
1072    DT = -GCTth*GSazm*GCchi
1073   
1074    BC1 = -AT*GSphi+(CT+BT*GCchi)*GCphi
1075    BC2 = DT-BT*GSchi
1076    BC3 = AT*GCphi+(CT+BT*GCchi)*GSphi
1077     
1078    BC = BC1*SComeg*SCchi+BC2*SComeg*SSchi-BC3*SSomeg     
1079    psi = acosd(BC)
1080   
1081    BD = 1.0-BC**2
1082    C = np.where(BD>1.e-6,rpd/np.sqrt(BD),0.)
1083    dPSdA = [-C*(-BC1*SSomeg*SCchi-BC2*SSomeg*SSchi-BC3*SComeg),
1084        -C*(-BC1*SComeg*SSchi+BC2*SComeg*SCchi),
1085        -C*(-BC1*SSomeg-BC3*SComeg*SCchi)]
1086     
1087    BA = -BC1*SSchi+BC2*SCchi
1088    BB = BC1*SSomeg*SCchi+BC2*SSomeg*SSchi+BC3*SComeg
1089    gam = atan2d(BB,BA)
1090
1091    BD = (BA**2+BB**2)/rpd
1092
1093    dBAdO = 0
1094    dBAdC = -BC1*SCchi-BC2*SSchi
1095    dBAdF = BC3*SSchi
1096   
1097    dBBdO = BC1*SComeg*SCchi+BC2*SComeg*SSchi-BC3*SSomeg
1098    dBBdC = -BC1*SSomeg*SSchi+BC2*SSomeg*SCchi
1099    dBBdF = BC1*SComeg-BC3*SSomeg*SCchi
1100   
1101    dGMdA = np.where(BD > 1.e-6,[(BA*dBBdO-BB*dBAdO)/BD,(BA*dBBdC-BB*dBAdC)/BD, \
1102        (BA*dBBdF-BB*dBAdF)/BD],[np.zeros_like(BD),np.zeros_like(BD),np.zeros_like(BD)])
1103       
1104    return psi,gam,dPSdA,dGMdA
1105
1106BOH = {
1107'L=2':[[],[],[]],
1108'L=4':[[0.30469720,0.36418281],[],[]],
1109'L=6':[[-0.14104740,0.52775103],[],[]],
1110'L=8':[[0.28646862,0.21545346,0.32826995],[],[]],
1111'L=10':[[-0.16413497,0.33078546,0.39371345],[],[]],
1112'L=12':[[0.26141975,0.27266871,0.03277460,0.32589402],
1113    [0.09298802,-0.23773812,0.49446631,0.0],[]],
1114'L=14':[[-0.17557309,0.25821932,0.27709173,0.33645360],[],[]],
1115'L=16':[[0.24370673,0.29873515,0.06447688,0.00377,0.32574495],
1116    [0.12039646,-0.25330128,0.23950998,0.40962508,0.0],[]],
1117'L=18':[[-0.16914245,0.17017340,0.34598142,0.07433932,0.32696037],
1118    [-0.06901768,0.16006562,-0.24743528,0.47110273,0.0],[]],
1119'L=20':[[0.23067026,0.31151832,0.09287682,0.01089683,0.00037564,0.32573563],
1120    [0.13615420,-0.25048007,0.12882081,0.28642879,0.34620433,0.0],[]],
1121'L=22':[[-0.16109560,0.10244188,0.36285175,0.13377513,0.01314399,0.32585583],
1122    [-0.09620055,0.20244115,-0.22389483,0.17928946,0.42017231,0.0],[]],
1123'L=24':[[0.22050742,0.31770654,0.11661736,0.02049853,0.00150861,0.00003426,0.32573505],
1124    [0.13651722,-0.21386648,0.00522051,0.33939435,0.10837396,0.32914497,0.0],
1125    [0.05378596,-0.11945819,0.16272298,-0.26449730,0.44923956,0.0,0.0]],
1126'L=26':[[-0.15435003,0.05261630,0.35524646,0.18578869,0.03259103,0.00186197,0.32574594],
1127    [-0.11306511,0.22072681,-0.18706142,0.05439948,0.28122966,0.35634355,0.0],[]],
1128'L=28':[[0.21225019,0.32031716,0.13604702,0.03132468,0.00362703,0.00018294,0.00000294,0.32573501],
1129    [0.13219496,-0.17206256,-0.08742608,0.32671661,0.17973107,0.02567515,0.32619598,0.0],
1130    [0.07989184,-0.16735346,0.18839770,-0.20705337,0.12926808,0.42715602,0.0,0.0]],
1131'L=30':[[-0.14878368,0.01524973,0.33628434,0.22632587,0.05790047,0.00609812,0.00022898,0.32573594],
1132    [-0.11721726,0.20915005,-0.11723436,-0.07815329,0.31318947,0.13655742,0.33241385,0.0],
1133    [-0.04297703,0.09317876,-0.11831248,0.17355132,-0.28164031,0.42719361,0.0,0.0]],
1134'L=32':[[0.20533892,0.32087437,0.15187897,0.04249238,0.00670516,0.00054977,0.00002018,0.00000024,0.32573501],
1135    [0.12775091,-0.13523423,-0.14935701,0.28227378,0.23670434,0.05661270,0.00469819,0.32578978,0.0],
1136    [0.09703829,-0.19373733,0.18610682,-0.14407046,0.00220535,0.26897090,0.36633402,0.0,0.0]],
1137'L=34':[[-0.14409234,-0.01343681,0.31248977,0.25557722,0.08571889,0.01351208,0.00095792,0.00002550,0.32573508],
1138    [-0.11527834,0.18472133,-0.04403280,-0.16908618,0.27227021,0.21086614,0.04041752,0.32688152,0.0],
1139    [-0.06773139,0.14120811,-0.15835721,0.18357456,-0.19364673,0.08377174,0.43116318,0.0,0.0]]
1140}
1141
1142Lnorm = lambda L: 4.*np.pi/(2.0*L+1.)
1143
1144def GetKcl(L,N,SGLaue,phi,beta):
1145    'needs doc string'
1146    import pytexture as ptx
1147    if SGLaue in ['m3','m3m']:
1148        if 'array' in str(type(phi)) and np.any(phi.shape):
1149            Kcl = np.zeros_like(phi)
1150        else:
1151            Kcl = 0.
1152        for j in range(0,L+1,4):
1153            im = j/4
1154            if 'array' in str(type(phi)) and np.any(phi.shape):
1155                pcrs = ptx.pyplmpsi(L,j,len(phi),phi)[0]
1156            else:
1157                pcrs = ptx.pyplmpsi(L,j,1,phi)[0]
1158            Kcl += BOH['L=%d'%(L)][N-1][im]*pcrs*cosd(j*beta)       
1159    else:
1160        if 'array' in str(type(phi)) and np.any(phi.shape):
1161            pcrs = ptx.pyplmpsi(L,N,len(phi),phi)[0]
1162        else:
1163            pcrs = ptx.pyplmpsi(L,N,1,phi)[0]
1164        pcrs *= RSQ2PI
1165        if N:
1166            pcrs *= SQ2
1167        if SGLaue in ['mmm','4/mmm','6/mmm','R3mR','3m1','31m']:
1168            if SGLaue in ['3mR','3m1','31m']: 
1169                if N%6 == 3:
1170                    Kcl = pcrs*sind(N*beta)
1171                else:
1172                    Kcl = pcrs*cosd(N*beta)
1173            else:
1174                Kcl = pcrs*cosd(N*beta)
1175        else:
1176            Kcl = pcrs*(cosd(N*beta)+sind(N*beta))
1177    return Kcl
1178   
1179def GetKsl(L,M,SamSym,psi,gam):
1180    'needs doc string'
1181    import pytexture as ptx
1182    if 'array' in str(type(psi)) and np.any(psi.shape):
1183        psrs,dpdps = ptx.pyplmpsi(L,M,len(psi),psi)
1184    else:
1185        psrs,dpdps = ptx.pyplmpsi(L,M,1,psi)
1186    psrs *= RSQ2PI
1187    dpdps *= RSQ2PI
1188    if M:
1189        psrs *= SQ2
1190        dpdps *= SQ2
1191    if SamSym in ['mmm',]:
1192        dum = cosd(M*gam)
1193        Ksl = psrs*dum
1194        dKsdp = dpdps*dum
1195        dKsdg = -psrs*M*sind(M*gam)
1196    else:
1197        dum = cosd(M*gam)+sind(M*gam)
1198        Ksl = psrs*dum
1199        dKsdp = dpdps*dum
1200        dKsdg = psrs*M*(-sind(M*gam)+cosd(M*gam))
1201    return Ksl,dKsdp,dKsdg
1202   
1203def GetKclKsl(L,N,SGLaue,psi,phi,beta):
1204    """
1205    This is used for spherical harmonics description of preferred orientation;
1206        cylindrical symmetry only (M=0) and no sample angle derivatives returned
1207    """
1208    import pytexture as ptx
1209    Ksl,x = ptx.pyplmpsi(L,0,1,psi)
1210    Ksl *= RSQ2PI
1211    if SGLaue in ['m3','m3m']:
1212        Kcl = 0.0
1213        for j in range(0,L+1,4):
1214            im = j/4
1215            pcrs,dum = ptx.pyplmpsi(L,j,1,phi)
1216            Kcl += BOH['L=%d'%(L)][N-1][im]*pcrs*cosd(j*beta)       
1217    else:
1218        pcrs,dum = ptx.pyplmpsi(L,N,1,phi)
1219        pcrs *= RSQ2PI
1220        if N:
1221            pcrs *= SQ2
1222        if SGLaue in ['mmm','4/mmm','6/mmm','R3mR','3m1','31m']:
1223            if SGLaue in ['3mR','3m1','31m']: 
1224                if N%6 == 3:
1225                    Kcl = pcrs*sind(N*beta)
1226                else:
1227                    Kcl = pcrs*cosd(N*beta)
1228            else:
1229                Kcl = pcrs*cosd(N*beta)
1230        else:
1231            Kcl = pcrs*(cosd(N*beta)+sind(N*beta))
1232    return Kcl*Ksl,Lnorm(L)
1233   
1234def Glnh(Start,SHCoef,psi,gam,SamSym):
1235    'needs doc string'
1236    import pytexture as ptx
1237
1238    if Start:
1239        ptx.pyqlmninit()
1240        Start = False
1241    Fln = np.zeros(len(SHCoef))
1242    for i,term in enumerate(SHCoef):
1243        l,m,n = eval(term.strip('C'))
1244        pcrs,dum = ptx.pyplmpsi(l,m,1,psi)
1245        pcrs *= RSQPI
1246        if m == 0:
1247            pcrs /= SQ2
1248        if SamSym in ['mmm',]:
1249            Ksl = pcrs*cosd(m*gam)
1250        else:
1251            Ksl = pcrs*(cosd(m*gam)+sind(m*gam))
1252        Fln[i] = SHCoef[term]*Ksl*Lnorm(l)
1253    ODFln = dict(zip(SHCoef.keys(),list(zip(SHCoef.values(),Fln))))
1254    return ODFln
1255
1256def Flnh(Start,SHCoef,phi,beta,SGData):
1257    'needs doc string'
1258    import pytexture as ptx
1259   
1260    if Start:
1261        ptx.pyqlmninit()
1262        Start = False
1263    Fln = np.zeros(len(SHCoef))
1264    for i,term in enumerate(SHCoef):
1265        l,m,n = eval(term.strip('C'))
1266        if SGData['SGLaue'] in ['m3','m3m']:
1267            Kcl = 0.0
1268            for j in range(0,l+1,4):
1269                im = j/4
1270                pcrs,dum = ptx.pyplmpsi(l,j,1,phi)
1271                Kcl += BOH['L='+str(l)][n-1][im]*pcrs*cosd(j*beta)       
1272        else:                #all but cubic
1273            pcrs,dum = ptx.pyplmpsi(l,n,1,phi)
1274            pcrs *= RSQPI
1275            if n == 0:
1276                pcrs /= SQ2
1277            if SGData['SGLaue'] in ['mmm','4/mmm','6/mmm','R3mR','3m1','31m']:
1278               if SGData['SGLaue'] in ['3mR','3m1','31m']: 
1279                   if n%6 == 3:
1280                       Kcl = pcrs*sind(n*beta)
1281                   else:
1282                       Kcl = pcrs*cosd(n*beta)
1283               else:
1284                   Kcl = pcrs*cosd(n*beta)
1285            else:
1286                Kcl = pcrs*(cosd(n*beta)+sind(n*beta))
1287        Fln[i] = SHCoef[term]*Kcl*Lnorm(l)
1288    ODFln = dict(zip(SHCoef.keys(),list(zip(SHCoef.values(),Fln))))
1289    return ODFln
1290   
1291def polfcal(ODFln,SamSym,psi,gam):
1292    '''Perform a pole figure computation.
1293    Note that the the number of gam values must either be 1 or must
1294    match psi. Updated for numpy 1.8.0
1295    '''
1296    import pytexture as ptx
1297    PolVal = np.ones_like(psi)
1298    for term in ODFln:
1299        if abs(ODFln[term][1]) > 1.e-3:
1300            l,m,n = eval(term.strip('C'))
1301            psrs,dum = ptx.pyplmpsi(l,m,len(psi),psi)
1302            if SamSym in ['-1','2/m']:
1303                if m:
1304                    Ksl = RSQPI*psrs*(cosd(m*gam)+sind(m*gam))
1305                else:
1306                    Ksl = RSQPI*psrs/SQ2
1307            else:
1308                if m:
1309                    Ksl = RSQPI*psrs*cosd(m*gam)
1310                else:
1311                    Ksl = RSQPI*psrs/SQ2
1312            PolVal += ODFln[term][1]*Ksl
1313    return PolVal
1314   
1315def invpolfcal(ODFln,SGData,phi,beta):
1316    'needs doc string'
1317    import pytexture as ptx
1318   
1319    invPolVal = np.ones_like(beta)
1320    for term in ODFln:
1321        if abs(ODFln[term][1]) > 1.e-3:
1322            l,m,n = eval(term.strip('C'))
1323            if SGData['SGLaue'] in ['m3','m3m']:
1324                Kcl = 0.0
1325                for j in range(0,l+1,4):
1326                    im = j/4
1327                    pcrs,dum = ptx.pyplmpsi(l,j,len(beta),phi)
1328                    Kcl += BOH['L=%d'%(l)][n-1][im]*pcrs*cosd(j*beta)       
1329            else:                #all but cubic
1330                pcrs,dum = ptx.pyplmpsi(l,n,len(beta),phi)
1331                pcrs *= RSQPI
1332                if n == 0:
1333                    pcrs /= SQ2
1334                if SGData['SGLaue'] in ['mmm','4/mmm','6/mmm','R3mR','3m1','31m']:
1335                   if SGData['SGLaue'] in ['3mR','3m1','31m']: 
1336                       if n%6 == 3:
1337                           Kcl = pcrs*sind(n*beta)
1338                       else:
1339                           Kcl = pcrs*cosd(n*beta)
1340                   else:
1341                       Kcl = pcrs*cosd(n*beta)
1342                else:
1343                    Kcl = pcrs*(cosd(n*beta)+sind(n*beta))
1344            invPolVal += ODFln[term][1]*Kcl
1345    return invPolVal
1346   
1347   
1348def textureIndex(SHCoef):
1349    'needs doc string'
1350    Tindx = 1.0
1351    for term in SHCoef:
1352        l = eval(term.strip('C'))[0]
1353        Tindx += SHCoef[term]**2/(2.0*l+1.)
1354    return Tindx
1355   
1356# self-test materials follow.
1357selftestlist = []
1358'''Defines a list of self-tests'''
1359selftestquiet = True
1360def _ReportTest():
1361    'Report name and doc string of current routine when ``selftestquiet`` is False'
1362    if not selftestquiet:
1363        import inspect
1364        caller = inspect.stack()[1][3]
1365        doc = eval(caller).__doc__
1366        if doc is not None:
1367            print('testing '+__file__+' with '+caller+' ('+doc+')')
1368        else:
1369            print('testing '+__file__()+" with "+caller)
1370NeedTestData = True
1371def TestData():
1372    array = np.array
1373    global NeedTestData
1374    NeedTestData = False
1375    global CellTestData
1376    # output from uctbx computed on platform darwin on 2010-05-28
1377    CellTestData = [
1378# cell, g, G, cell*, V, V*
1379  [(4, 4, 4, 90, 90, 90), 
1380   array([[  1.60000000e+01,   9.79717439e-16,   9.79717439e-16],
1381       [  9.79717439e-16,   1.60000000e+01,   9.79717439e-16],
1382       [  9.79717439e-16,   9.79717439e-16,   1.60000000e+01]]), array([[  6.25000000e-02,   3.82702125e-18,   3.82702125e-18],
1383       [  3.82702125e-18,   6.25000000e-02,   3.82702125e-18],
1384       [  3.82702125e-18,   3.82702125e-18,   6.25000000e-02]]), (0.25, 0.25, 0.25, 90.0, 90.0, 90.0), 64.0, 0.015625],
1385# cell, g, G, cell*, V, V*
1386  [(4.0999999999999996, 5.2000000000000002, 6.2999999999999998, 100, 80, 130), 
1387   array([[ 16.81      , -13.70423184,   4.48533243],
1388       [-13.70423184,  27.04      ,  -5.6887143 ],
1389       [  4.48533243,  -5.6887143 ,  39.69      ]]), array([[ 0.10206349,  0.05083339, -0.00424823],
1390       [ 0.05083339,  0.06344997,  0.00334956],
1391       [-0.00424823,  0.00334956,  0.02615544]]), (0.31947376387537696, 0.25189277536327803, 0.16172643497798223, 85.283666420376008, 94.716333579624006, 50.825714168082683), 100.98576357983838, 0.0099023858863968445],
1392# cell, g, G, cell*, V, V*
1393  [(3.5, 3.5, 6, 90, 90, 120), 
1394   array([[  1.22500000e+01,  -6.12500000e+00,   1.28587914e-15],
1395       [ -6.12500000e+00,   1.22500000e+01,   1.28587914e-15],
1396       [  1.28587914e-15,   1.28587914e-15,   3.60000000e+01]]), array([[  1.08843537e-01,   5.44217687e-02,   3.36690552e-18],
1397       [  5.44217687e-02,   1.08843537e-01,   3.36690552e-18],
1398       [  3.36690552e-18,   3.36690552e-18,   2.77777778e-02]]), (0.32991443953692895, 0.32991443953692895, 0.16666666666666669, 90.0, 90.0, 60.000000000000021), 63.652867178156257, 0.015710211406520427],
1399  ]
1400    global CoordTestData
1401    CoordTestData = [
1402# cell, ((frac, ortho),...)
1403  ((4,4,4,90,90,90,), [
1404 ((0.10000000000000001, 0.0, 0.0),(0.40000000000000002, 0.0, 0.0)),
1405 ((0.0, 0.10000000000000001, 0.0),(2.4492935982947065e-17, 0.40000000000000002, 0.0)),
1406 ((0.0, 0.0, 0.10000000000000001),(2.4492935982947065e-17, -2.4492935982947065e-17, 0.40000000000000002)),
1407 ((0.10000000000000001, 0.20000000000000001, 0.29999999999999999),(0.40000000000000013, 0.79999999999999993, 1.2)),
1408 ((0.20000000000000001, 0.29999999999999999, 0.10000000000000001),(0.80000000000000016, 1.2, 0.40000000000000002)),
1409 ((0.29999999999999999, 0.20000000000000001, 0.10000000000000001),(1.2, 0.80000000000000004, 0.40000000000000002)),
1410 ((0.5, 0.5, 0.5),(2.0, 1.9999999999999998, 2.0)),
1411]),
1412# cell, ((frac, ortho),...)
1413  ((4.1,5.2,6.3,100,80,130,), [
1414 ((0.10000000000000001, 0.0, 0.0),(0.40999999999999998, 0.0, 0.0)),
1415 ((0.0, 0.10000000000000001, 0.0),(-0.33424955703700043, 0.39834311042186865, 0.0)),
1416 ((0.0, 0.0, 0.10000000000000001),(0.10939835193016617, -0.051013289294572106, 0.6183281045774256)),
1417 ((0.10000000000000001, 0.20000000000000001, 0.29999999999999999),(0.069695941716497567, 0.64364635296002093, 1.8549843137322766)),
1418 ((0.20000000000000001, 0.29999999999999999, 0.10000000000000001),(-0.073350319180835066, 1.1440160419710339, 0.6183281045774256)),
1419 ((0.29999999999999999, 0.20000000000000001, 0.10000000000000001),(0.67089923785616512, 0.74567293154916525, 0.6183281045774256)),
1420 ((0.5, 0.5, 0.5),(0.92574397446582857, 1.7366491056364828, 3.0916405228871278)),
1421]),
1422# cell, ((frac, ortho),...)
1423  ((3.5,3.5,6,90,90,120,), [
1424 ((0.10000000000000001, 0.0, 0.0),(0.35000000000000003, 0.0, 0.0)),
1425 ((0.0, 0.10000000000000001, 0.0),(-0.17499999999999993, 0.3031088913245536, 0.0)),
1426 ((0.0, 0.0, 0.10000000000000001),(3.6739403974420595e-17, -3.6739403974420595e-17, 0.60000000000000009)),
1427 ((0.10000000000000001, 0.20000000000000001, 0.29999999999999999),(2.7675166561703527e-16, 0.60621778264910708, 1.7999999999999998)),
1428 ((0.20000000000000001, 0.29999999999999999, 0.10000000000000001),(0.17500000000000041, 0.90932667397366063, 0.60000000000000009)),
1429 ((0.29999999999999999, 0.20000000000000001, 0.10000000000000001),(0.70000000000000018, 0.6062177826491072, 0.60000000000000009)),
1430 ((0.5, 0.5, 0.5),(0.87500000000000067, 1.5155444566227676, 3.0)),
1431]),
1432]
1433    global LaueTestData             #generated by GSAS
1434    LaueTestData = {
1435    'R 3 m':[(4.,4.,6.,90.,90.,120.),((1,0,1,6),(1,0,-2,6),(0,0,3,2),(1,1,0,6),(2,0,-1,6),(2,0,2,6),
1436        (1,1,3,12),(1,0,4,6),(2,1,1,12),(2,1,-2,12),(3,0,0,6),(1,0,-5,6),(2,0,-4,6),(3,0,-3,6),(3,0,3,6),
1437        (0,0,6,2),(2,2,0,6),(2,1,4,12),(2,0,5,6),(3,1,-1,12),(3,1,2,12),(1,1,6,12),(2,2,3,12),(2,1,-5,12))],
1438    'R 3':[(4.,4.,6.,90.,90.,120.),((1,0,1,6),(1,0,-2,6),(0,0,3,2),(1,1,0,6),(2,0,-1,6),(2,0,2,6),(1,1,3,6),
1439        (1,1,-3,6),(1,0,4,6),(3,-1,1,6),(2,1,1,6),(3,-1,-2,6),(2,1,-2,6),(3,0,0,6),(1,0,-5,6),(2,0,-4,6),
1440        (2,2,0,6),(3,0,3,6),(3,0,-3,6),(0,0,6,2),(3,-1,4,6),(2,0,5,6),(2,1,4,6),(4,-1,-1,6),(3,1,-1,6),
1441        (3,1,2,6),(4,-1,2,6),(2,2,-3,6),(1,1,-6,6),(1,1,6,6),(2,2,3,6),(2,1,-5,6),(3,-1,-5,6))],
1442    'P 3':[(4.,4.,6.,90.,90.,120.),((0,0,1,2),(1,0,0,6),(1,0,1,6),(0,0,2,2),(1,0,-1,6),(1,0,2,6),(1,0,-2,6),
1443        (1,1,0,6),(0,0,3,2),(1,1,1,6),(1,1,-1,6),(1,0,3,6),(1,0,-3,6),(2,0,0,6),(2,0,-1,6),(1,1,-2,6),
1444        (1,1,2,6),(2,0,1,6),(2,0,-2,6),(2,0,2,6),(0,0,4,2),(1,1,-3,6),(1,1,3,6),(1,0,-4,6),(1,0,4,6),
1445        (2,0,-3,6),(2,1,0,6),(2,0,3,6),(3,-1,0,6),(2,1,1,6),(3,-1,-1,6),(2,1,-1,6),(3,-1,1,6),(1,1,4,6),
1446        (3,-1,2,6),(3,-1,-2,6),(1,1,-4,6),(0,0,5,2),(2,1,2,6),(2,1,-2,6),(3,0,0,6),(3,0,1,6),(2,0,4,6),
1447        (2,0,-4,6),(3,0,-1,6),(1,0,-5,6),(1,0,5,6),(3,-1,-3,6),(2,1,-3,6),(2,1,3,6),(3,-1,3,6),(3,0,-2,6),
1448        (3,0,2,6),(1,1,5,6),(1,1,-5,6),(2,2,0,6),(3,0,3,6),(3,0,-3,6),(0,0,6,2),(2,0,-5,6),(2,1,-4,6),
1449        (2,2,-1,6),(3,-1,-4,6),(2,2,1,6),(3,-1,4,6),(2,1,4,6),(2,0,5,6),(1,0,-6,6),(1,0,6,6),(4,-1,0,6),
1450        (3,1,0,6),(3,1,-1,6),(3,1,1,6),(4,-1,-1,6),(2,2,2,6),(4,-1,1,6),(2,2,-2,6),(3,1,2,6),(3,1,-2,6),
1451        (3,0,4,6),(3,0,-4,6),(4,-1,-2,6),(4,-1,2,6),(2,2,-3,6),(1,1,6,6),(1,1,-6,6),(2,2,3,6),(3,-1,5,6),
1452        (2,1,5,6),(2,1,-5,6),(3,-1,-5,6))],
1453    'P 3 m 1':[(4.,4.,6.,90.,90.,120.),((0,0,1,2),(1,0,0,6),(1,0,-1,6),(1,0,1,6),(0,0,2,2),(1,0,-2,6),
1454        (1,0,2,6),(1,1,0,6),(0,0,3,2),(1,1,1,12),(1,0,-3,6),(1,0,3,6),(2,0,0,6),(1,1,2,12),(2,0,1,6),
1455        (2,0,-1,6),(0,0,4,2),(2,0,-2,6),(2,0,2,6),(1,1,3,12),(1,0,-4,6),(1,0,4,6),(2,0,3,6),(2,1,0,12),
1456        (2,0,-3,6),(2,1,1,12),(2,1,-1,12),(1,1,4,12),(2,1,2,12),(0,0,5,2),(2,1,-2,12),(3,0,0,6),(1,0,-5,6),
1457        (3,0,1,6),(3,0,-1,6),(1,0,5,6),(2,0,4,6),(2,0,-4,6),(2,1,3,12),(2,1,-3,12),(3,0,-2,6),(3,0,2,6),
1458        (1,1,5,12),(3,0,-3,6),(0,0,6,2),(2,2,0,6),(3,0,3,6),(2,1,4,12),(2,2,1,12),(2,0,5,6),(2,1,-4,12),
1459        (2,0,-5,6),(1,0,-6,6),(1,0,6,6),(3,1,0,12),(3,1,-1,12),(3,1,1,12),(2,2,2,12),(3,1,2,12),
1460        (3,0,4,6),(3,1,-2,12),(3,0,-4,6),(1,1,6,12),(2,2,3,12))],
1461    'P 3 1 m':[(4.,4.,6.,90.,90.,120.),((0,0,1,2),(1,0,0,6),(0,0,2,2),(1,0,1,12),(1,0,2,12),(1,1,0,6),
1462        (0,0,3,2),(1,1,-1,6),(1,1,1,6),(1,0,3,12),(2,0,0,6),(2,0,1,12),(1,1,2,6),(1,1,-2,6),(2,0,2,12),
1463        (0,0,4,2),(1,1,-3,6),(1,1,3,6),(1,0,4,12),(2,1,0,12),(2,0,3,12),(2,1,1,12),(2,1,-1,12),(1,1,-4,6),
1464        (1,1,4,6),(0,0,5,2),(2,1,-2,12),(2,1,2,12),(3,0,0,6),(1,0,5,12),(2,0,4,12),(3,0,1,12),(2,1,-3,12),
1465        (2,1,3,12),(3,0,2,12),(1,1,5,6),(1,1,-5,6),(3,0,3,12),(0,0,6,2),(2,2,0,6),(2,1,-4,12),(2,0,5,12),
1466        (2,2,-1,6),(2,2,1,6),(2,1,4,12),(3,1,0,12),(1,0,6,12),(2,2,2,6),(3,1,-1,12),(2,2,-2,6),(3,1,1,12),
1467        (3,1,-2,12),(3,0,4,12),(3,1,2,12),(1,1,-6,6),(2,2,3,6),(2,2,-3,6),(1,1,6,6))],
1468    }
1469   
1470    global FLnhTestData
1471    FLnhTestData = [{
1472    'C(4,0,0)': (0.965, 0.42760447),
1473    'C(2,0,0)': (1.0122, -0.80233610),
1474    'C(2,0,2)': (0.0061, 8.37491546E-03),
1475    'C(6,0,4)': (-0.0898, 4.37985696E-02),
1476    'C(6,0,6)': (-0.1369, -9.04081762E-02),
1477    'C(6,0,0)': (0.5935, -0.18234928),
1478    'C(4,0,4)': (0.1872, 0.16358127),
1479    'C(6,0,2)': (0.6193, 0.27573633),
1480    'C(4,0,2)': (-0.1897, 0.12530720)},[1,0,0]]
1481def test0():
1482    if NeedTestData: TestData()
1483    msg = 'test cell2Gmat, fillgmat, Gmat2cell'
1484    for (cell, tg, tG, trcell, tV, trV) in CellTestData:
1485        G, g = cell2Gmat(cell)
1486        assert np.allclose(G,tG),msg
1487        assert np.allclose(g,tg),msg
1488        tcell = Gmat2cell(g)
1489        assert np.allclose(cell,tcell),msg
1490        tcell = Gmat2cell(G)
1491        assert np.allclose(tcell,trcell),msg
1492selftestlist.append(test0)
1493
1494def test1():
1495    'test cell2A and A2Gmat'
1496    _ReportTest()
1497    if NeedTestData: TestData()
1498    msg = 'test cell2A and A2Gmat'
1499    for (cell, tg, tG, trcell, tV, trV) in CellTestData:
1500        G, g = A2Gmat(cell2A(cell))
1501        assert np.allclose(G,tG),msg
1502        assert np.allclose(g,tg),msg
1503selftestlist.append(test1)
1504
1505def test2():
1506    'test Gmat2A, A2cell, A2Gmat, Gmat2cell'
1507    _ReportTest()
1508    if NeedTestData: TestData()
1509    msg = 'test Gmat2A, A2cell, A2Gmat, Gmat2cell'
1510    for (cell, tg, tG, trcell, tV, trV) in CellTestData:
1511        G, g = cell2Gmat(cell)
1512        tcell = A2cell(Gmat2A(G))
1513        assert np.allclose(cell,tcell),msg
1514selftestlist.append(test2)
1515
1516def test3():
1517    'test invcell2Gmat'
1518    _ReportTest()
1519    if NeedTestData: TestData()
1520    msg = 'test invcell2Gmat'
1521    for (cell, tg, tG, trcell, tV, trV) in CellTestData:
1522        G, g = invcell2Gmat(trcell)
1523        assert np.allclose(G,tG),msg
1524        assert np.allclose(g,tg),msg
1525selftestlist.append(test3)
1526
1527def test4():
1528    'test calc_rVsq, calc_rV, calc_V'
1529    _ReportTest()
1530    if NeedTestData: TestData()
1531    msg = 'test calc_rVsq, calc_rV, calc_V'
1532    for (cell, tg, tG, trcell, tV, trV) in CellTestData:
1533        assert np.allclose(calc_rV(cell2A(cell)),trV), msg
1534        assert np.allclose(calc_V(cell2A(cell)),tV), msg
1535selftestlist.append(test4)
1536
1537def test5():
1538    'test A2invcell'
1539    _ReportTest()
1540    if NeedTestData: TestData()
1541    msg = 'test A2invcell'
1542    for (cell, tg, tG, trcell, tV, trV) in CellTestData:
1543        rcell = A2invcell(cell2A(cell))
1544        assert np.allclose(rcell,trcell),msg
1545selftestlist.append(test5)
1546
1547def test6():
1548    'test cell2AB'
1549    _ReportTest()
1550    if NeedTestData: TestData()
1551    msg = 'test cell2AB'
1552    for (cell,coordlist) in CoordTestData:
1553        A,B = cell2AB(cell)
1554        for (frac,ortho) in coordlist:
1555            to = np.inner(A,frac)
1556            tf = np.inner(B,to)
1557            assert np.allclose(ortho,to), msg
1558            assert np.allclose(frac,tf), msg
1559            to = np.sum(A*frac,axis=1)
1560            tf = np.sum(B*to,axis=1)
1561            assert np.allclose(ortho,to), msg
1562            assert np.allclose(frac,tf), msg
1563selftestlist.append(test6)
1564
1565def test7():
1566    'test GetBraviasNum(...) and GenHBravais(...)'
1567    _ReportTest()
1568    import os.path
1569    import sys
1570    import GSASIIspc as spc
1571    testdir = os.path.join(os.path.split(os.path.abspath( __file__ ))[0],'testinp')
1572    if os.path.exists(testdir):
1573        if testdir not in sys.path: sys.path.insert(0,testdir)
1574    import sgtbxlattinp
1575    derror = 1e-4
1576    def indexmatch(hklin, hkllist, system):
1577        for hklref in hkllist:
1578            hklref = list(hklref)
1579            # these permutations are far from complete, but are sufficient to
1580            # allow the test to complete
1581            if system == 'cubic':
1582                permlist = [(1,2,3),(1,3,2),(2,1,3),(2,3,1),(3,1,2),(3,2,1),]
1583            elif system == 'monoclinic':
1584                permlist = [(1,2,3),(-1,2,-3)]
1585            else:
1586                permlist = [(1,2,3)]
1587
1588            for perm in permlist:
1589                hkl = [abs(i) * hklin[abs(i)-1] / i for i in perm]
1590                if hkl == hklref: return True
1591                if [-i for i in hkl] == hklref: return True
1592        else:
1593            return False
1594
1595    for key in sgtbxlattinp.sgtbx7:
1596        spdict = spc.SpcGroup(key)
1597        cell = sgtbxlattinp.sgtbx7[key][0]
1598        system = spdict[1]['SGSys']
1599        center = spdict[1]['SGLatt']
1600
1601        bravcode = GetBraviasNum(center, system)
1602
1603        g2list = GenHBravais(sgtbxlattinp.dmin, bravcode, cell2A(cell))
1604
1605        assert len(sgtbxlattinp.sgtbx7[key][1]) == len(g2list), 'Reflection lists differ for %s' % key
1606        for h,k,l,d,num in g2list:
1607            for hkllist,dref in sgtbxlattinp.sgtbx7[key][1]: 
1608                if abs(d-dref) < derror:
1609                    if indexmatch((h,k,l,), hkllist, system):
1610                        break
1611            else:
1612                assert 0,'No match for %s at %s (%s)' % ((h,k,l),d,key)
1613selftestlist.append(test7)
1614
1615def test8():
1616    'test GenHLaue'
1617    _ReportTest()
1618    import GSASIIspc as spc
1619    import sgtbxlattinp
1620    derror = 1e-4
1621    dmin = sgtbxlattinp.dmin
1622
1623    def indexmatch(hklin, hklref, system, axis):
1624        # these permutations are far from complete, but are sufficient to
1625        # allow the test to complete
1626        if system == 'cubic':
1627            permlist = [(1,2,3),(1,3,2),(2,1,3),(2,3,1),(3,1,2),(3,2,1),]
1628        elif system == 'monoclinic' and axis=='b':
1629            permlist = [(1,2,3),(-1,2,-3)]
1630        elif system == 'monoclinic' and axis=='a':
1631            permlist = [(1,2,3),(1,-2,-3)]
1632        elif system == 'monoclinic' and axis=='c':
1633            permlist = [(1,2,3),(-1,-2,3)]
1634        elif system == 'trigonal':
1635            permlist = [(1,2,3),(2,1,3),(-1,-2,3),(-2,-1,3)]
1636        elif system == 'rhombohedral':
1637            permlist = [(1,2,3),(2,3,1),(3,1,2)]
1638        else:
1639            permlist = [(1,2,3)]
1640
1641        hklref = list(hklref)
1642        for perm in permlist:
1643            hkl = [abs(i) * hklin[abs(i)-1] / i for i in perm]
1644            if hkl == hklref: return True
1645            if [-i for i in hkl] == hklref: return True
1646        return False
1647
1648    for key in sgtbxlattinp.sgtbx8:
1649        spdict = spc.SpcGroup(key)[1]
1650        cell = sgtbxlattinp.sgtbx8[key][0]
1651        center = spdict['SGLatt']
1652        Laue = spdict['SGLaue']
1653        Axis = spdict['SGUniq']
1654        system = spdict['SGSys']
1655
1656        g2list = GenHLaue(dmin,spdict,cell2A(cell))
1657        #if len(g2list) != len(sgtbxlattinp.sgtbx8[key][1]):
1658        #    print 'failed',key,':' ,len(g2list),'vs',len(sgtbxlattinp.sgtbx8[key][1])
1659        #    print 'GSAS-II:'
1660        #    for h,k,l,d in g2list: print '  ',(h,k,l),d
1661        #    print 'SGTBX:'
1662        #    for hkllist,dref in sgtbxlattinp.sgtbx8[key][1]: print '  ',hkllist,dref
1663        assert len(g2list) == len(sgtbxlattinp.sgtbx8[key][1]), (
1664            'Reflection lists differ for %s' % key
1665            )
1666        #match = True
1667        for h,k,l,d in g2list:
1668            for hkllist,dref in sgtbxlattinp.sgtbx8[key][1]: 
1669                if abs(d-dref) < derror:
1670                    if indexmatch((h,k,l,), hkllist, system, Axis): break
1671            else:
1672                assert 0,'No match for %s at %s (%s)' % ((h,k,l),d,key)
1673                #match = False
1674        #if not match:
1675            #for hkllist,dref in sgtbxlattinp.sgtbx8[key][1]: print '  ',hkllist,dref
1676            #print center, Laue, Axis, system
1677selftestlist.append(test8)
1678           
1679def test9():
1680    'test GenHLaue'
1681    _ReportTest()
1682    import GSASIIspc as G2spc
1683    if NeedTestData: TestData()
1684    for spc in LaueTestData:
1685        data = LaueTestData[spc]
1686        cell = data[0]
1687        hklm = np.array(data[1])
1688        H = hklm[-1][:3]
1689        hklO = hklm.T[:3].T
1690        A = cell2A(cell)
1691        dmin = 1./np.sqrt(calc_rDsq(H,A))
1692        SGData = G2spc.SpcGroup(spc)[1]
1693        hkls = np.array(GenHLaue(dmin,SGData,A))
1694        hklN = hkls.T[:3].T
1695        #print spc,hklO.shape,hklN.shape
1696        err = True
1697        for H in hklO:
1698            if H not in hklN:
1699                print H,' missing from hkl from GSASII'
1700                err = False
1701        assert(err)
1702selftestlist.append(test9)
1703       
1704       
1705   
1706
1707if __name__ == '__main__':
1708    # run self-tests
1709    selftestquiet = False
1710    for test in selftestlist:
1711        test()
1712    print "OK"
Note: See TracBrowser for help on using the repository browser.