ó ò™xTc@ sìddlmZddljjZddlmZddlZddlm Z ddl m Z ddl m Z ddlZdefd„ƒYZd efd „ƒYZd efd „ƒYZd efd„ƒYZdefd„ƒYZdS(iÿÿÿÿ(tdivisionN(tInvalidConstraint(t superpose(tmatrix(tflextrigid_pivoted_rotatable_groupcB s/eZdZd„Zed„ƒZd„ZRS(s a set of atoms (rigid body) rides on a pivot atom and rotates around the pivot-pivot_neighbour bond, the original geometry is not altered cC sRt|ƒdkr!tdƒ‚n||_||_||_||_||_dS(Nisat least one atom is expected(tlenRtpivottpivot_neighbourtindicestsizeablet rotatable(tselfRRt ind_sequenceR R ((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pyt__init__s    cC std„|jDƒƒS(Ncs s|]}|dfVqdS(tsiteN((t.0tidx((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pys s(ttupleR (R ((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pytconstrained_parameterssc C s|jjƒ}|j|jƒ}|j|jƒ}|jtjddd|jƒ}|jtjddd|j ƒ}t g|j D]}||^q‚ƒ}|jtj d|d|d|d|d |ƒ}x@t |j ƒD]/\}} |j||| ƒ||j| _qÛWdS( NtvalueitvariableiRRtazimuthtsizet scatterers(t structureRtadd_new_site_parameterRRtaddt_tindependent_scalar_parameterR R RR Rt enumeratetadd_new_site_proxy_parametertasu_scatterer_parametersR( R treparametrisationRtpivot_spt pivot_n_spRRtitparamtj((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pytadd_tos$&  (t__name__t __module__t__doc__RtpropertyRR'(((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pyR s t rigid_rotatable_expandable_groupcB s/eZdZd„Zed„ƒZd„ZRS(sV a set of atoms rides on a pivot atom, rotates and uniformly expands or contracts cC sIt|ƒdkr!tdƒ‚n||_||_||_||_dS(Nisat least one atom is expected(RRRR R R (R tcenterR R R ((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pyR5s    cC std„|jDƒƒS(Ncs s|]}|dfVqdS(RN((RR((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pys ?s(RR (R ((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pyR=sc C sH|jjƒ}|j|jƒ}|jtjddd|jƒ}|jtjddd|jƒ}|jtjddd|jƒ}|jtjddd|jƒ}t g|j D]}||^q²ƒ}|jtj d|d|d|d|d |d |ƒ} x@t |j ƒD]/\}} |j | || ƒ| |j| _qWdS( NRiRiRRtalphatbetatgammaR(RRRRRRRR R RR R,RRR R( R R!RR"RR.R/R0R$R%R&((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pyR'As,&  (R(R)R*RR+RR'(((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pyR,0s trigid_riding_expandable_groupcB s/eZdZd„Zed„ƒZd„ZRS(sT a set of atoms rides on a pivot atom, rotates and uniformly expands or shrinks cC s@t|ƒdkr!tdƒ‚n||_||_||_dS(Nisat least one atom is expected(RRRR R (R R-R R ((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pyR_s   cC std„|jDƒƒS(Ncs s|]}|dfVqdS(RN((RR((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pys hs(RR (R ((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pyRfscC sÓ|jjƒ}|j|jƒ}|jtjddd|jƒ}tg|j D]}||^qOƒ}|jtj d|d|d|ƒ}x@t |j ƒD]/\}}|j |||ƒ||j |_qœWdS(NRiRRRR(RRRRRRRR RR R1RRR R(R R!RR"RR$R%R&((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pyR'js&  (R(R)R*RR+RR'(((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pyR1Zs tidealised_fragmentcB sKeZdZd„Zddd„ƒYZd„Zdd„Zdd„ZRS( s| ported from olex2 xlib/fragment.h generates parameterised coordinates for four framents: Cp, Ph, Cp* and naphthalene cC s)idd6d d6d d6d d6|_dS( Ng¸…ëQ¸ö?tCpg5^ºI ñ?sCp*g= ×£p=ö?tPht Naphthalene(g¸…ëQ¸ö?(g¸…ëQ¸ö?g5^ºI ñ?(g= ×£p=ö?(g= ×£p=ö?(tdefault_lengths(R ((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pyR~s tpointcB s5eZd„Zd„Zd„Zd„Zd„ZRS(cC s||_||_dS(N(txty(R R8R9((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pyR†s cC stj|j||j|ƒS(N(R2R7R8R9(R tk((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pyt__mul__‰scC s$tj|j|j|j|jƒS(N(R2R7R8R9(R tp((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pyt__add__‹scC s%tj|j|j|j|jƒS(N(tmathtsqrtR8R9(R ((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pytlengthscC s&dt|jƒdt|jƒdS(Nt(t,t)(tstrR8R9(R ((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pyt__repr__s(R(R)RR;R=R@RE(((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pyR7…s     c C s°tjd|}tj|ƒ}tj|ƒ}g}tj|| ƒ}xatd|ƒD]P}|j||ƒ|j} || ||j |_||j || |_ qXW|S(s7 generates coordinates of a ring with given radius ii( R>tpitcostsinR2R7txrangetappendR8R9( R tedgestrtangletcatsatresultR<R$R8((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pyt generate_ring’s cC sî|dkr|j|}n|dkrX|jdd|dtjdtjdƒƒS|dkrx|jd|dƒS|d kròd|dtjdtjdƒ}|jd|ƒ}x/|jd||d ƒD]}|j|ƒq×W|S|d krê|jd|dƒ}|j|dƒx(td dƒD]}|j||ƒq5W|d |d}||ddtjtjdƒ|jƒ}x/tdt |ƒƒD]}|||||RGRFRJRIR@R(R tfragmenttlengthsRLtresR$R<((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pytgenerate_fragment¡s2  0  &! 0  cC s»|stdt|ƒƒ}ng|D]#}||j||jdf^q%}tjtj|ƒtj|ƒƒ}tjg|D]}|j|jdf^qƒ}|jj ||j j S(s fits given fragment to given sites, if control_points indices are not given - all points are fit, otherwise only control points are fit and the result is propagated to the rest of the fragment coordinates. returns coordinates of the trasformed fragment i( trangeRR8R9Rtleast_squares_fitRt vec3_doubleRLtelemstt(R RStreference_sitestcontrol_point_indicesR$tto_fittlsf((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pytfit¿s01(N( R(R)R*RR7RQRRRVR`(((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pyR2ys    t same_groupcB s8eZdZdeed„Zed„ƒZd„ZRS(s, non-crystallographic symmetry constraint icC st|ƒdkr!tdƒ‚nt|dƒ}x3|dD]'}t|ƒ|kr<tdƒ‚q<q<W||_||_||_||_dS(s» fix_xyz and fix_u are to be used for debugging purposes only: if the coordinates are not fixed, then the refined angles should be stored externally and passed to this object is#at least two atom sets are expectediisatoms sets differ in sizeN(RRtgroupstfix_xyztfix_utangles(R RbReRcRdtltg((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pyRÓs   cC syd}xltj|jdƒD]U}xL|D]D}|jrN||dff7}n|jr)||dff7}q)q)WqW|S(NiRtU((t itertoolstisliceRbRcRd(R RPRgR$((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pyRãs   c C s/|j r|j rdS|jjƒ}g}g}g}g}g}g}|jjƒ} x¢|jdD]“} |j| j|| jƒƒ|jr­|j|j | ƒƒn|jrh|| j j ƒrâ|j|j | ƒƒqû|j|j | ƒƒqhqhWx)|jdD]} t | ƒt |jdƒkrAtdƒ‚ng} g} g}g}xÀt| ƒD]²\}} || j j ƒ||jd|j j ƒkr²tdƒ‚n| j|| ƒ|j| j|| jƒƒ|| j j ƒr| j|| ƒqf|j|| ƒqfWtjtj|ƒtj|ƒƒ}t |ƒdkr£xNtdt | ƒƒD]4} |jdtj|jƒtj|| ƒƒqhWn|j}tj|jƒtj|jƒ}|jƒ}d}xMt|ƒD]?\} }|tjtj|ƒtj|| ƒƒjƒ7}qíWtjtj|ƒtj|ƒƒ}|jƒ}d}xMt|ƒD]?\} }|tjtj|ƒtj|| ƒƒjƒ7}qsW||krÏ|j }n|jƒ}|jrb|jtjd|d|d|ddddfdt ƒ}t |ƒdkrÂ|jtj!d|d |d |d |ƒ}qÂn`t |ƒdkrÂ|jtj"d|j#dt ƒ}|jtj!d|d |d |d |ƒ}n|jrø|jtj$d| d |d |d |ƒ}nt |ƒdkr+|jtj%d| d|ƒ}nd}d}d}xç| D]ß} |jrƒ||j&| _|j'||| ƒ|d7}n|jrD|| j j ƒrá||j&| _(|jtj)d|d|ƒ|j*| <|d7}q#||j&| _(|jtj+d|d|ƒ|j*| <|d7}qDqDWq WdS(NiisGroup size mismatchs+Mixing isotropic and anisotropic parametersiRRRtu_starstalignment_matrixtshifts_and_anglesRetsitestu_isostparenttindex(,RdRcRRt unit_cellRbRJt orthogonalizeRRtflagst use_u_isot&add_new_thermal_displacement_parameterRRRRRXRRYRIRtcolt other_shiftRLtreference_shifttother_sites_best_fitt length_sqt transposeRRt$independent_small_6_vector_parametertTruetsame_group_u_start$independent_small_3_vector_parameterRetsame_group_xyztsame_group_u_isoR t'add_new_same_group_site_proxy_parametertutsame_group_u_iso_proxyt shared_Ustsame_group_u_star_proxy( R R!Rt ref_sitest ref_u_isost ref_u_starstref_adpstsrc_crdst inv_src_crdstucR$Rgt g_scattererstg_u_iso_scattererstg_u_star_scattererstcrdsRR_trmR[tnew_crdtdtctd_invRmt u_star_paramRet site_paramt u_iso_paramtsite_proxy_indextu_star_proxy_indextu_iso_proxy_index((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pyR'îsÞ  !/ " 7 7    -                (iii(R(R)R*R~RR+RR'(((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pyRaÏs (t __future__Rtsmtbx.refinement.constraintst refinementt constraintsRRR>t scitbx.mathRtscitbxRtscitbx.array_familyRRitobjectRR,R1R2Ra(((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/rigid.pyts  &*V