)xTc<@ s0ddlmZddlmZmZddlmZddlmZddlm Z ddlm Z ddl m Z ddl mZdd lZdd lZdd lZdd lZdd lmZdd lZd d d d d d Zd ddddgd ddddgdd dddgdddddgdddddgdddddgddd d!d"gddd d"d!gdd#d$d%d&gd#dd$d&d%gd#d'dd(d)gd#d*d+d,d-gd'd#dd)d(gd'dd.d/d0gdddddgdd'd.d0d/gd1d2d3d4dgd5d6d7d8d9gd5d:d;d<d=gd>d?d@dAdBgd>d:dCdDdEgd?d>d@dBdAgd?dFddGdHgd?dId;dJdBgdKdId@dLdMgdNdOd3dPdQgdNdRdSdTdUgdOdNd3dQdPgdOdVdWdXdYgdVdOdWdYdXgdVdZd[d\d]gdZdVd[d]d\gdZd^d$d<d_gd^dZd$d_d<gd^dRd@d`dagdRdNdSdUdTgdRd^d@dad`gdbdcdddedfgdgdhddidjgdhdgddjdigdkdlddmdngdldkddndmgdodpdSdqd)gdodrddsdBgd*d#d+d-d,gd2d1d3dd4gd6d5d7d9d8gdFd?ddHdGgdpdodSd)dqgdrdoddBdsgdd dddgd:d5d;d=d<gd:d>dCdEdDgdId?d;dBdJgdIdKd@dMdLgdcdbdddfdegg8ZdtZduZdvZdwZdxZdyZe dzkr,end S({i(tdivision(t approx_equalt show_diff(tflex(tadptbx(tuctbx(tadp_restraints(tadp_restraint_params(tmatrixN(tStringIOg:0yE>c C sIddddd}d gt|}dgt|} |dk rht|t|kshtnxtt|D]} |dkrg} xtdD]} ddddddg} || | fstt|j d?stt|j dstd@}dA}dB}dC}tj d |d |} tj d |d |} tj d |d |} tj d |d |} | j |ksEt| j|ksZttjdDdEdFdGf}tjdH}tjt ttt f}x | | | | fD]} td|d!|d"|}tj|d#| }t|j|sttj|jdI}tj|jd}tj| | g}tj|d$|}t||j|jfsttj|d$|}t||j |j fsttj|d$|d%|d&|}t|d|jsttd'tjd#| d|d!|d"|\}}x?t||D].\}}t|tj|ds=tq=Wx6t||D]%\}}t||dstqWqWtjd(dJ}tjd(dK}|j|jf}dL}ttf}tj|d d}|j}tj}xtd3D]}tj |j!dM}|||j"}|||j"} |j| jf}tj|d d}t|j|s4tq4WdS(NNiiiiiiiR RviDggg g4@g@gg @g4gg@g?i@ii|iii iiiigZڱ @RuR RR RxRzRRRg?g?g?gQ?g{Gz?g{Gz?gzG?g{Gz?g)\(?gQ?i(iiiiii(iiiiii(ii(gggg g g4@(g@g@gg @g @g4(gggggg@(iiiiii(iiiiii(ii(iiiii i(iiiiii(iiiiii(ii(ii(ii(ii(iiiiii(iiiiii(iiiiii(iiiiii(iiii(iiiiii(g?g?g?gQ?g{Gz?g{Gz?(gzG?g{Gz?g)\(?g{Gz?g{Gz?gQ?(ii(ii(#tTrueRtadp_similarityRRRRvRt gradients2Rt rms_deltastFalsetadp_similarity_proxyRuRRR^tboolRRtshared_adp_similarity_proxytadp_similarity_residualstadp_similarity_deltas_rmstadp_similarity_residual_sumR$RYRRRRRRRRR(!R R RRvRt i_seqs_aat i_seqs_ait i_seqs_iat i_seqs_iitp_aatp_aitp_iatp_iiRrRyRzRRxRt deltas_rmsRRRRRRRRRRRRR((sJ/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/adp_restraints/tst_ext.pytexercise_adp_similaritys          ''  &!   c C sud!}d}d#}d$}tf}tjd|d|}|j|ksNtt|j|sfttjd |dd|}d%}d&}t|j|stt|j|stt|j dstt|j dstt|j |stt j dd'} t jdd} tj||g} t j |}t j|}t j|}td |d|d|} tj| d| } t| |j |j fsttj| d| }t||j |j fs ttj| d| d| }t|ds;ttdtjd|d |d|d|\}}x?t| |D].\}}t|tj|ds{tq{Wtjdd(}tjd |jdd}|j }t j}xytd D]k}tj|jd)}|||j}tjd |jdd}t|j |stqWdS(*NiiiiiiRuRvR iiii(ii@gbz H@gw@R RRxRzg@RR Rg?g?g?gQ?g{Gz?g{Gz?i(i(iiiiii((iiiiii(i(iiiiii(iiii(ii@(iiiiii(g?g?g?gQ?g{Gz?g{Gz?(ii( RRtisotropic_adp_proxyRuRRRvt isotropic_adpRRRR}RRR^tshared_isotropic_adp_proxyRRtisotropic_adp_residualstisotropic_adp_deltas_rmstisotropic_adp_residual_sumR$RYRRRRRRRRR(RuRvR R RRrRtexpected_deltasRRzRRxRyRRRRRRRRRRRt u_cart_rot((sJ/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/adp_restraints/tst_ext.pytexercise_isotropic_adpcsj   ''  &    cC stjjdr.tjd dRkr.dGHdStjdSdTdUdVf}dW}tjdXdYdZd[f}tjd\}tjt t t t f}t j }t }|j d*d+d,tjd-tjd.tjd/|t|jd0 stt j t jd1d2d3gd4d5t jd1dd6gd4d7g}t }|j d*d+d,|d-|d.|d/|d8d9t|jd: stt }|j d*d;d<|d,|d-tjd]d.tjttttfd/|d8dAt|jdB stt j}t }|j d*d+d,tjd-|d.|d/|t|jdC smtt jt jd1d^d4d5t jd1d_d4d7g}t }|j d*d+d<|d,|d-|d.|d/|d8dDt|jdE stt }|j d*d;d,|d-|d.|d/|d8dFt|jdG sRtt j}t }|j d*d+dHtjd,tjd/|t|jdI stt jt jd1d`d4d5t jd1dad4dJt jd1dbd4d5t jd1dcd4dKg}t }|j d*d+dH|d<|d,|d/|d8dLt|jdM sgtt }|j d*dNdH|d,|d/|d8dOdPdt|jdQ stdS(dNtwiniisSkipping exercise_proxy_show()g1%d g*ԩ%@gB@gͪUg3ı.n#@g/n@g- gn4@!@gjM@gʡE g}гY#@gjM@R%R&tO16tN8gK=U?g0*?ggj+?gy&1l?g_LugZӼtgl?g46s $  @ ' S \ <