,xTc@ sddlmZddlmZddlmZmZmZmZm Z m Z ddl m Z m Z ddlmZddlZddlmZdZdd Zd Zd Zd Zd dZdZedkrendS(i(tdivision(tflex(t covariancetcrystaltgeometrytsgtbxtuctbxtxray(t approx_equalt show_diff(tmatrixN(tStringIOc C sRtjdtjdddtjtjddddftjd dgS(Ntcrystal_symmetryg ףp= @gzG@iZixtP6222t scattererstSiig@g@tOgK7?gK7ɿg_?(g ףp= @g ףp= @gzG@iZiZixg?g?gUUUUUU?(gK7?gK7ɿg_?(Rt structureRtsymmetryRtxray_scatterert scatterer(((sO/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry/tests/tst_geometry.pytquartz s   cC sW|dk r!tjd|}ntjdd}tjdtjd|d d d tjtj d d ddddtj d d ddddtj d d ddddtj d dddddtj d ddd ddtj d ddd!ddtj d ddd"ddtj d ddd#ddtj d ddd$ddf S(%Ntmetrical_matrixt parametersg ףp= @gzG@iZixR t unit_celltspace_group_symbols hall: P 1RtlabelRtsiteg?gF]kSU?tugg]QJVU?RgK7?gV-?g?gK7?gtE)!XU?g1Zd?(g ףp= @g ףp= @gzG@iZiZix(g?g?gF]kSU?(gg?g]QJVU?(g?gg(gK7?gV-?g?(gK7?gK7?gtE)!XU?(gV-?g1Zd?g?(gK7?gK7?g?(g1Zd?gV-?gtE)!XU?(gV-?gK7?g?( tNoneRRRRRRRRR(RR((sO/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry/tests/tst_geometry.pyt quartz_p1sX             cC st}|j}|j}x|D]}|jtq(W|j|tjd}tjd}|j}t j |||}t j |j }t j |j} |j} |j} tjd } | d |j| | df} tj| }t|jds$tt j| dt j| d }|t j | jjj| }|j}t j|d t|t j|d}|t j |j |j!}t|j"||| |d ddsttd|j"|||| s#ttjd} | d |j| | d f} tj| }t|j|j#d stt|j$d |j"|||| sttjd} tjd }|t j | jjj| }|t j |jjj| }t j%tj&dd dg||}|j| | d| d |j|| df} tj'| }t|j(dst|j)}t jt|t j|d |dt|t j|d}|t j |j |j!}t|j"||| tj|f|d dds]ttd|j"|||| tj|fstdS(Ng:0yE>g& .>g& .!>gA:)>g& .1>g:0y5>g:0yU>g+i)+`>g:0ye>gQ?ig{Gz?s -y,x-y,z-1/3iggd/?tepsgV瞯g& .>g& .!>gA:)>g& .1>g:0y5>g:0yU>g& .>g& .!>gA:)>g& .1>g+i)+`>g& .>g& .!>gA:)>g:0yU>g& .>g& .!>g+i)+`>g& .>g:0ye>(gQ?gQ?iiiigQ?iiiig{Gz?iiiiiiiii(*RRtscatterer_flagst set_grad_sitetTruetset_scatterer_flagsRtdoublet parameter_mapRtorthogonalize_covariance_matrixR tsqrtorthogonalization_matrixtfractionalization_matrixt sites_cartt sites_fracRtrt_mxt orthogonalizeRtdistanceRtdistance_modeltAssertionErrortcoltrtinverset as_doubletd_distance_d_sitestrowttupletmatrix_packed_u_as_symmetrict transposetvarianceRtmatrix_packed_u_diagonalt#extract_covariance_matrix_for_sitestsize_ttanglet angle_modeltd_angle_d_sites(txstuctflagstftcovtcell_vcvt param_maptcov_cartRtFR*R+trt_mx_jitsitestdtvt r_inv_carttgtrt_mx_kit r_inv_cart_jit r_inv_cart_kitcov_ata((sO/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry/tests/tst_geometry.pytexercise_geometryAsx          !$) .#.$!"!))' (!#:c C sdd}dd}t}|j}|j}|d|df}tj|}|j|}t||j}t||st |d|d|df}tj |} | j |}t||j}t||st dS(NcS sQtd|}|j}|j}|d|df}tj|}|jS(NRii(RRR*RR.R/(RRARBR*RKRL((sO/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry/tests/tst_geometry.pyt calc_distances   cS sctd|}|j}|j}|d|d|df}tj|}|jtjdS(NRiiii(RRR*RR>R?tmathtpi(RRARBR*RKRT((sO/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry/tests/tst_geometry.pyt calc_angles   iii( RRRR*RR.td_distance_d_metrical_matrixtfinite_differencesRR0R>td_angle_d_metrical_matrix( RVRYRARBR*RKRLtgradstfd_gradsRT((sO/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry/tests/tst_geometry.pytexercise_grad_metrical_matrixs     cOC stjdtjddQdddtjtjd d d dRddSddddtjd dd dTddUddddtjd d d dVddWddddtjd d(d dXddYdd-dd.f}|j}x|D]}|jt qW|j ||j }|j }tj d/d/d/d/d/d/d/d/d/d/d/d/d0d1d/d2d3d4d/d/d/d/d/d5d/d6d7d8d/d/d/d/d/d/d/d/d/d/d/d/d/d/d9d:d;d/d/d/d/d/d<d=d/d/d/d/d/d>d/d/d/d/d/d/d/d/d/d/d/d/d/d/d/d/d/d/d/d/gN}tj d?d?d/d/d/d/d?d/d/d/d/d@d/d/d/d/d/d/d/d/d/g}|j}tj|||}tjdZ} tj| ||} x:tjtjtjftjdDtjtjftjtjtjdDftjdDtjtjdDffD]} | d/|| d/} | dC|| dC} | dA|| dA}tj|j| |j| |j|f}|j| ||| }t|d/dEdFs5tq5W|jdGdA}dHdIl}|jjdJrdHdIl}t }||j!j"||j#j$dK|j dL|dM|dN|j%IJt&|j'dO stt }||j!j(||j#j$dK|j dL|dM|dN|j%IJt&|j'dP stndIS([sK BF6 sitting on inversion centre in high symmetry space group. The angles formed between F20, B17 and either F18 or F19 are necessarily exactly equal to 90 degrees, and hence their variance should be exactly zero. The angles formed by F18, B17 and F19 are free to refine, and hence should have an associated variance. R Rg^I 6@g9#A@iZRshall: -I 4bd 2RRtF18Rg?g@-?g+}RghY0?g_vOF?g$D"?gg6w\tfpgձJ^?tfdpghW!'?tF19g|_?gI?gX|[T'?g(֩=#Q?g:`$?g?̔"?tF20g p?g?ggRG.?gMb@?grQ-"?g}r?tB17gLn!?g}rD?g! ?gTqs* ?g 2&W?g}F?igk=[>goK~RgL^1AVg-!_qgW_q&gAK~>g.as%cg]z?>g|Qwe=gA~P>g;@>gt\@ghB>g"oC>g/n>gƐ>gv(>iiis 1-x,2-y,-zRgؗҬR-R:RR0tpair_asu_tabletlibtbx.load_envtenvt has_modulet iotbx.cifR tciftdistances_as_cif_loopRtextract_labelstloopR tgetvaluetangles_as_cif_loop(RARCRDR+RREtcell_covRGRHti_seqst cov_i_seqstsym_opst site_frac_jit site_frac_it site_frac_kiRTtvarRjtlibtbxRgts((sO/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry/tests/tst_geometry.pytexercise_high_symmetry_cases            -0H6 !!.             gư>c C sg}xtdD]}}t|j}||cd|7<|d|}||cd|8<|d|}||d|}|j|qW|S(NiiR(trangetlistRtappend( tfuncRRt gradientstitmmtqmtqptdq((sO/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry/tests/tst_geometry.pyR[%scC stttdGHdS(NtOK(RR_RU(((sO/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry/tests/tst_geometry.pytrun1st__main__(t __future__Rtcctbx.array_familyRtcctbxRRRRRRtlibtbx.test_utilsRR tscitbxR RWt cStringIOR RRRRUR_RR[Rt__name__(((sO/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry/tests/tst_geometry.pyts.   0 @