ó *™xTc@ sˆddlmZddlmZddlmZddlmZddlZ ddl m Z ddl Z de j jfd„ƒYZdS( iÿÿÿÿ(tdivision(tadopt_init_args(tgeometry_restraints(tflexN(tmathtenergiescB sÚeZddddddddddddddedeegdd„Zd„Zd„Zd„Zd„Z d„Z d„Z d„Z d„Z d „Zd „Zd „Zd „Zd „Zd„Zddd„ZRS(c C set|tƒƒtjjj|d|d|d|jƒdtjd|ƒd|_ d|_ |dk r}|dk s}t ‚n|rÝ|j dkr¹tj|jƒdddgƒ|_ qÝ|j jƒ|jƒksÝt ‚n|dkrþd|_d|_n]|jƒ|_tjd|d|d |j d |ƒ|_|j|j7_|j|j7_|dkr|d|_d|_nc|jƒ|_tjd|d|d |j d |d tƒ|_|j|j7_|j|j7_|dkrd|_d|_nt|ƒ|_|dkrBtjd|d |d |j ƒ|_n*tjd |d|d |d |j ƒ|_|j|j7_|j|j7_|dkr±d|_d|_nt|ƒ|_|dkrótjd|d |d |j ƒ|_n*tjd |d|d |d |j ƒ|_|j|j7_|j|j7_|dkrbd|_d|_nt|ƒ|_|dkr¤tjd|d |d |j ƒ|_n*tjd |d|d |d |j ƒ|_|j|j7_|j|j7_| dkrd|_d|_nt| ƒ|_|dkrUtjd|d | d |j ƒ|_n*tjd |d|d | d |j ƒ|_|j|j7_|j|j7_| dkrÄd|_ d|_!nWt| ƒ|_ tj!d|d | d |j ƒ|_!|j|j 7_|j|j!7_| dkr<d|_"d|_#nt| ƒ|_"|dkr~tj#d|d | d |j ƒ|_#n*tj#d |d|d | d |j ƒ|_#|j|j"7_|j|j#7_| dkríd|_$d|_%nt| ƒ|_$|dkr/tj%d|d | d |j ƒ|_%n*tj%d |d|d | d |j ƒ|_%|j|j$7_|j|j%7_| dkržd|_&d|_'nt| ƒ|_&|dkràtj'd|d | d |j ƒ|_'n*tj'd |d|d | d |j ƒ|_'|j|j&7_|j|j'7_|dkrXd|_(d|_)d|_*n–|j+ƒ|_)|j,d|d |j ƒ|_*|j|j)7_|j|j*7_|j-j.ƒ|_(|j(dkrî|j-j/d|d |j ƒ|_0n|dk r*|d|d |j ƒ|_1|j|j17_n d|_1x!|j2D]}|j3d|ƒq=W|j4ƒdS(Ntcompute_gradientst gradientstgradients_sizetgradients_factoryt normalizationit sites_carttsorted_asu_proxiestgradient_arrayt disable_cachetfunctiontproxiest unit_cellt energies_obj(5Rtlocalstscitbxt restraintsRt__init__tsizeRt vec3_doubletNonetn_dihedral_restraintst dihedral_restraints_residual_sumtAssertionErrorRtn_bond_proxiestbond_residual_sumtn_totalRtnumber_of_restraintst residual_sumtn_nonbonded_proxiestnonbonded_residual_sumtFalsetn_angle_proxiestangle_residual_sumtlentn_dihedral_proxiestdihedral_residual_sumtn_reference_dihedral_proxiestreference_dihedral_residual_sumtn_ncs_dihedral_proxiestncs_dihedral_residual_sumtn_chirality_proxiestchirality_residual_sumtn_planarity_proxiestplanarity_residual_sumtn_parallelity_proxiestparallelity_residual_sumtn_bond_similarity_proxiestbond_similarity_residual_sumtn_hbond_proxiestn_generic_proxiestgeneric_restraint_residual_sumt get_n_proxiestrestraints_residual_sumtgeneric_restraints_managert get_n_hbondstget_hbonds_residual_sumthbond_residual_sumtexternal_energytextension_objectst energies_addtfinalize_target_and_gradients(tselfR Rt bond_proxiestnonbonded_proxiestnonbonded_functiont angle_proxiestdihedral_proxiestreference_dihedral_proxiestncs_dihedral_proxiestchirality_proxiestplanarity_proxiestparallelity_proxiestbond_similarity_proxiesR;texternal_energy_functionRRtdisable_asu_cacheR R@thbonds_in_bond_listt extension_obj((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry_restraints/energies.pyR sp     '$                                                                  cC s.|jdk r#|jt|jƒS|jSdS(N(RQRRR'(RC((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry_restraints/energies.pytget_filtered_n_bond_proxiesùsc C sÐ|jdk rÈ|jdkrÁ|jdk rœ|jjddd|jddddd|jƒ\}}|dksxt‚tjg|D]}|d^q…ƒSt j d|jd |jƒ}|SqÌdSndSdS( Nitby_valuetresidualR t site_labelst max_itemstexcludeiR ( RRRQRDt get_sortedR RRtdoubleRt bond_deltas(RCt sorted_tablet n_not_showntxR[((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry_restraints/energies.pytget_filtered_bond_deltasÿs"  $   c C s |jdk r|jjd|jd|jƒ}|dk rþg|jjD]}tj|j ƒ^qI}t j gt ||ƒD]\}}||^q}ƒ}t jt j||dƒƒ}t jt j|ƒdƒ}t jt j|ƒdƒ} | ||fSdSndS(sa Calculate rmsz of bond deviations Compute rmsz, the Root-Mean-Square of the z-scors for a set of data using z_i = {x_i - mu / sigma} and rmsz = sqrt(mean(z*z)) x_i: atcual bond length mu: geometry restraints mean sigma: geometry restraints standard deviation z_i: z-score for bond i z: array of z_i The sigma and the (x_i - mu) are model constrains, geometry restraints. They function extracts from self, not calculated from data. :returns: b_rmsz: rmsz, root mean square of the z-scors of all bonds b_z_min/max: min/max abolute values of z-scors R RXiN(iii(RRRDtget_filtered_deltasR RQtsimpleRtweight_as_sigmatweightRRZtzipRtsqrtt mean_defaultt max_defaulttabst min_default( RCR[R^tsigmast bond_deltatsigmatz_scorestb_rmsztb_z_maxtb_z_min((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry_restraints/energies.pytbond_deviations_zs    +5 cC s¯|jdk r«|jjd|jd|jƒ}|dk r¤||}tjtj |dƒƒ}tjtj |dƒƒ}tjtj |dƒƒ}|||fSdSndS(NR RXi(iii( RRRDR`R RQRReRRfRgRi(RCR[tb_sqtb_avetb_maxtb_min((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry_restraints/energies.pytbond_deviations:s      c C sð|jdk rìtjd|jd|jƒ}g|jD]}tj|jƒ^q7}tj gt ||ƒD]\}}||^qkƒ}t j tj ||dƒƒ}tjtj|ƒdƒ}tjtj|ƒdƒ} | ||fSdS(sã Calculate rmsz of angles deviations Compute rmsz, the Root-Mean-Square of the z-scors for a set of data using z_i = {x_i - mu / sigma} and rmsz = sqrt(mean(z*z)) Compute rmsz, the Root-Mean-Square of the z-scors for a set of data using z_i = {x_i - mu / sigma} and rmsz = sqrt(mean(z*z)) x_i: atcual bond angle mu: geometry restraints mean sigma: geometry restraints standard deviation z_i: z-score for bond i z: array of z_i The sigma and the (x_i - mu) are model constrains, geometry restraints. They function extracts from self, not calculated from data. :returns: a_rmsz: rmsz, root mean square of the z-scors of all angles a_z_min/max: min/max values of z-scors R RiN(R%RRt angle_deltasR RGRbRcRRZRdRReRfRgRhRi( RCRwR^Rjt angle_deltaRlRmta_rmszta_z_maxta_z_min((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry_restraints/energies.pytangle_deviations_zLs   (5cC s™|jdk r•tjd|jd|jƒ}||}tjtj |dƒƒ}tjtj |dƒƒ}tjtj |dƒƒ}|||fSdS(NR Ri( R%RRRwR RGRReRRfRgRi(RCRwta_sqta_aveta_maxta_min((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry_restraints/energies.pytangle_deviationsms    cC stjd|jd|jƒS(NR R (Rtnonbonded_deltasR RE(RC((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry_restraints/energies.pytnonbonded_distancesxs  cC s‡|jdk rƒ|jƒ}||}tjtj|dƒƒ}tjtj|dƒƒ}tjtj|dƒƒ}|||fSdS(Ni( R"RRƒRReRRfRgRi(RCR‚tr_sqtr_avetr_maxtr_min((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry_restraints/energies.pytnonbonded_deviations}s  cC s™|jdk r•tjd|jd|jƒ}||}tjtj |dƒƒ}tjtj |dƒƒ}tjtj |dƒƒ}|||fSdS(NR Ri( R(RRtdihedral_deltasR RHRReRRfRgRi(RCR‰td_sqtd_avetd_maxtd_min((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry_restraints/energies.pytdihedral_deviations†s    cC s™|jdk r•tjd|jd|jƒ}||}tjtj |dƒƒ}tjtj |dƒƒ}tjtj |dƒƒ}|||fSdS(NR Ri( R*RRtreference_dihedral_deltasR RIRReRRfRgRi(RCRRŠR‹RŒR((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry_restraints/energies.pytreference_dihedral_deviations‘s    cC s™|jdk r•tjd|jd|jƒ}||}tjtj |dƒƒ}tjtj |dƒƒ}tjtj |dƒƒ}|||fSdS(NR Ri( R,RRtncs_dihedral_deltasR RJRReRRfRgRi(RCR‘RŠR‹RŒR((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry_restraints/energies.pytncs_dihedral_deviationsœs    cC s™|jdk r•tjd|jd|jƒ}||}tjtj |dƒƒ}tjtj |dƒƒ}tjtj |dƒƒ}|||fSdS(NR Ri( R.RRtchirality_deltasR RKRReRRfRgRi(RCR“tc_sqtc_avetc_maxtc_min((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry_restraints/energies.pytchirality_deviations§s    cC s™|jdk r•tjd|jd|jƒ}||}tjtj |dƒƒ}tjtj |dƒƒ}tjtj |dƒƒ}|||fSdS(NR Ri( R0RRtplanarity_deltas_rmsR RLRReRRfRgRi(RCtplanarity_deltastp_sqtp_avetp_maxtp_min((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry_restraints/energies.pytplanarity_deviations²s    cC s™|jdk r•tjd|jd|jƒ}||}tjtj |dƒƒ}tjtj |dƒƒ}tjtj |dƒƒ}|||fSdS(NR Ri( R2RRtparallelity_deltasR RMRReRRfRgRi(RCR R›RœRRž((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry_restraints/energies.pytparallelity_deviations½s    tcC sž|dkrtj}n||d|jIJ|jdk r]||d|j|jfIJn|jdk r||d|j|jfIJn|jdk r½||d|j|j fIJn|j dk rí||d|j |j fIJn|j dk r||d|j |j fIJn|jdk rM||d|j|jfIJn|jdk r}||d|j|jfIJn|jdk r­||d |j|jfIJn|jdk rÝ||d |j|jfIJn|jd kr ||d |j|jfIJn|jdk r=||d |j|jfIJnx-|jD]"}|jd|d|d|ƒqGW|jdk rš||d|jjƒIJndS(Ns target: %.6gs bond_residual_sum (n=%d): %.6gs% nonbonded_residual_sum (n=%d): %.6gs! angle_residual_sum (n=%d): %.6gs$ dihedral_residual_sum (n=%d): %.6gs. reference_dihedral_residual_sum (n=%d): %.6gs( ncs_dihedral_residual_sum (n=%d): %.6gs% chirality_residual_sum (n=%d): %.6gs% planarity_residual_sum (n=%d): %.6gs' parallelity_residual_sum (n=%d): %.6gis" hbonds_residual_sum (n=%d): %.6gs+ bond_similarity_residual_sum (n=%d): %.6gRtftprefixs norm of gradients: %.6g(RtsyststdoutttargetRRR"R#R%R&R(R)R*R+R,R-R.R/R0R1R2R3R6R>R4R5R@t energies_showRtnorm(RCR£R¤RR((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry_restraints/energies.pytshowÈsP             N(t__name__t __module__RtTrueR$RRSR_RqRvR|RRƒRˆRŽRR’R˜RŸR¡Rª(((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry_restraints/energies.pyR sF Ú   %  !  (t __future__RtlibtbxRtcctbxRtcctbx.array_familyRtscitbx.restraintsRtstdlibRR¥RR(((sP/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/cctbx/geometry_restraints/energies.pyts