ó ð™xTc@ sddlmZddlZddlZejjdƒZddlTddl Z ddl m Z ddl ZddlmZdejjfd„ƒYZd efd „ƒYZd Zd ejjejfd „ƒYZd ejjejfd„ƒYZdejfd„ƒYZdS(iÿÿÿÿ(tdivisionNt smtbx_refinement_constraints_ext(t*(tflex(txraytInvalidConstraintcB seZejZRS((t__name__t __module__t Exception(((sS/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/__init__.pyRstConflictingConstraintWarningcB s eZdZd„Zd„ZRS(su Attempt to constrain some scatterer parameters that have already been constrained: this attempt will be ignored. cC s||_||_||_dS(N(t conflictstconstraint_typet scatterers(tselfR R R ((sS/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/__init__.pyt__init__s  cC s\djg|jD])\}}d||j|j|f^qƒ}d|jj|j|fS(Ns s * %s of scatterer %s (#%i)sj%s The attempted constraint is of type '%s' and the list of already constrained scatterer parameters is %s(tjoinR R tlabelt __class__t__doc__R (R t scatt_idxt param_tagtconflicts_printout((sS/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/__init__.pyt__str__s 9(RRRRR(((sS/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/__init__.pyR s s4Invalid %s constraint involving %s: bad connectivityt_cB seZd„Zd„ZRS(cc s,x%t|jƒD]}|j|ƒVqWdS(s An iterator over its arguments N(txranget n_argumentstargument(R ti((sS/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/__init__.pyt arguments,scC sºy6djg|jD]}|j^qƒ}d|}Wntk rOd}nXg}|jso|jdƒn|r‹ddj|ƒ}nd}d|j|jj|||jf}|S(s7 String representation using the graphviz DOT language s, s (%s)ttcsts [%s]s%i [label="%s%s%s #%s"]( RR RtAttributeErrort is_variabletappendtindexRR(R tsctscatttinfotlbl((sS/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/__init__.pyR1s"(    (RRRR(((sS/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/__init__.pyR*s cB seZd„ZRS(cC sÕ|jƒg}xH|jƒD]:}x1|jƒD]#}|jd|j|jfƒq0WqWg}x0|jƒD]"}|j|jt|ƒfƒqnW|jƒ|jg|D]\}}|^q«ƒddj|ƒS(s7 String representation using the graphviz DOT language s%i -> %isdigraph dependencies { %s }s; ( tfinaliset parametersRR!R"tstrtsorttextendR(R tbitstptqtdsu_bitsR((sS/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/__init__.pyREs %  &(RRR(((sS/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/__init__.pyRCstreparametrisationcB sÃeZdZdZdZdZdZd„Zd„Z d„Z de j fd„ƒYZ d„Zd„Zdd „Zd „Zd „Zd „Zd „Zed„Zd„Zd„Zd„ZRS(s? Enhance the C++ level reparametrisation class for ease of use icK stt|ƒj|jƒƒi|_i|_i|_i|_i|_i|_ ||_ }||_ |j j dtdtƒ|_|jƒ}|j jƒ|_tj||ƒt|jƒƒ|_tƒ|_|jd k rài}xç|jD]Ü}|ddkr(tj|dƒ||dt constraintsR?tkwdstxsR RMtdRtsitest constrainttc_paramst warned_abouttuniquesR-ti_sctfraction((sS/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/__init__.pyRksr            **            cC sStt|ƒjƒ|jjƒ|_|jjƒ|_|jjƒ|_|jj |jƒd}|j dk ržx*|j D]}|j r{|d7}q{q{Wn|j dk rÆ|j j rÆ|d7}n|jj|}|j dk rx3|j D]%}|j rï||_|d7}qïqïWn|j dk rO|j j rO||j _|d7}ndS(Nii(R6R0R'RJtmapping_to_grad_fcRLt&mapping_to_grad_fc_independent_scalarst deep_copytmapping_to_grad_fc_allR+R]RNR^R`tjacobian_transposetn_rowst grad_index(R tindependent_grad_cntRltindependent_grad_i((sS/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/__init__.pyR'Ãs*     cC stjj||ƒdS(N(ROR0t apply_shifts(R tshifts((sS/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/__init__.pyRvßstcomponent_annotationscB seZd„ZRS(cC s)|jjd|jjƒjdƒd ƒS(Nt_component_annotationst,iÿÿÿÿ(t__dict__t setdefaultRJRxtsplit(R ((sS/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/__init__.pytfgetãs (RRR~(((sS/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/__init__.pyRxâscC s|j|jƒS(s The columns of self.jacobian_transpose corresponding to crystallographic parameters for the scatterers, in the same order as the derivatives in grad Fc. In this class, the latter is assumed to follow the convention of smtbx.structure_factors (tjacobian_transpose_matchingRm(R ((sS/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/__init__.pyt#jacobian_transpose_matching_grad_fcèscC su|jj|ƒr|j|S|j|j}|dkrq|jjƒ|}|jt|ƒ}||j|_n|S(N( R:thas_keyRJt occupancyRNR>R RVtindependent_occupancy_parameter(R RktoccR#((sS/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/__init__.pyR\ðs  cC sÜ|jj|ƒr|j|S|j|j}|dkr§|jj|ƒ}|jjƒ|}|j ƒr|j t |ƒ}n|j t ||ƒ}||j|_n|dk rØ|j ƒ rØ|j t||ƒ}n|S(N(R9RRJtsiteRNRFtgetR>R tis_point_group_1RVtindependent_site_parametertspecial_position_site_parametert is_unit_mxt"symmetry_equivalent_site_parameter(R t i_scatterertsymm_optst site_symmR#((sS/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/__init__.pyRQús   cC s=|jj|ƒr|j|S|jt||ƒ|j|R tflagst use_u_isot use_u_anisotAssertionErrorRVtindependent_u_iso_parameterRFR†R‡tindependent_u_star_parametert!special_position_u_star_parameter(R RŒR•R#R((sS/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/__init__.pyR[s   %   cC s&|jt|ƒ}|jj|ƒ|S(N(RVttwin_fraction_parameterRLR!(R t twin_fractionR-((sS/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/__init__.pyR_(scC s/|jtd|d|ƒ}|jj|ƒ|S(Ntvaluetvariable(RVtindependent_scalar_parameterRLR!(R RŸR R-((sS/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/__init__.pyt$add_new_independent_scalar_parameter-scC sNd}|jdk r-|jj|dƒ}n|dkrJd|d‚n|S(NsUndefined direction: 't'(RNRMR†(R tid_tres((sS/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/__init__.pytfind_direction2s  cC sK|jjƒ}g}x&|D]}|jd||jƒqWdj|ƒS(Ns%st (R>R R!RR(R tslR trvR((sS/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/__init__.pytformat_scatter_list:s  cC s€tj|jjƒƒ}|jdk rTx*|jD]}|jr1|jƒq1q1Wn|jdk r||jjr||jƒn|S(N( Rt parameter_mapR>R R]RNR^tadd_independent_scalarR`(R R©Rl((sS/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/__init__.pyR«As  N(RRRt temperatureRNR]R`RMRR'RvRGtpropertyRxR€R\RQR’R”R[R_RCR¢R¦RªR«(((sS/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/__init__.pyR0cs( X           (t __future__RRWt boost.pythontboosttpythont import_extRORt scitbx.sparsetscitbxtscitbx.array_familyRt libtbx.utilsRGtcctbxRtutilstSorryRt UserWarningR tbad_connectivity_msgtinjectort parameterRR0(((sS/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/smtbx/refinement/constraints/__init__.pyts     ""