ó ö™xTc@ sOddlmZddlZddlZddlmZddd„ƒYZdS(iÿÿÿÿ(tdivisionN(tBeamt beam_factoryc B sªeZdZd„Zed d d d d d d„ƒZed d d d d d d d d„ƒZed„ƒZed„ƒZ ed„ƒZ ed„ƒZ ed„ƒZ RS( sÅA factory class for beam objects, which encapsulate standard beam models. In cases where a full cbf description is available this will be used, otherwise simplified descriptions can be applied.cC sdS(N((tself((s>/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/beam.pyt__init__scC sè|dks|dkr'd}d}n|rm|s9t‚tttt|ƒƒt|ƒt|ƒt|ƒƒS|r¿|st‚tttd„tt|ƒƒƒt|ƒt|ƒt|ƒƒS|sËt‚tttt|ƒƒƒSdS(NgcS s| S(N((tx((s>/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/beam.pyt,s(tNonetAssertionErrorRttupletmaptfloat(tsample_to_sourcet wavelengthts0tunit_s0t divergencetsigma_divergence((s>/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/beam.pyt make_beams&          c C sc|s t‚|st‚|dks0|dkr?d}d}n|r |sQt‚tttt|ƒƒt|ƒt|ƒt|ƒttt|ƒƒt|ƒƒS|r |s²t‚tttd„tt|ƒƒƒt|ƒt|ƒt|ƒttt|ƒƒt|ƒƒS|st‚tttt|ƒƒt|ƒt|ƒttt|ƒƒt|ƒƒSdS(NgcS s| S(N((R((s>/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/beam.pyRLs(RRRR R R (R R RRt polarizationtpolarization_fractionRR((s>/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/beam.pytmake_polarized_beam4s<                cC stjddd|ƒS(sÇConstruct a beam object on the principle that the beam is aligned with the +z axis, as is quite normal. Also assume the beam has polarization fraction 0.999 and is polarized in the x-z plane.R ggð?R (gggð?(RR(R ((s>/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/beam.pytsimple[s cC stjd|d|ƒS(s/Construct a beam with direction and wavelength.R R (RR(R R ((s>/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/beam.pytsimple_directionales c C s"tjd|d|d|d|ƒS(sXFull access to the constructor for cases where we do know everything that we need...R R RR(RR(R Rtpolarization_plane_normalR ((s>/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/beam.pytcomplexms  c C stjd}tjƒ}|j|tjƒ|jdƒ|jdƒ|jdƒg}x<t dƒD].}|jd|dƒ|j |j ƒƒqfW|j ƒ}y|j ƒ\}}Wnd}d }nXtj||ƒtj||ƒd f}tjd |d |d |d |ƒS(sáInitialize a detector model from an imgCIF file. N.B. the definition of the polarization plane is not completely helpful in this - it is the angle between the polarization plane and the +Y laboratory frame vector.g€f@taxist equipmenttsourceis vector[%d]ig+‡ÙÎ÷ï?gR R RR(tmathtpitpycbftcbf_handle_structt read_filet MSG_DIGESTt find_categoryt find_columntfind_rowtrangetappendtget_doublevaluetget_wavelengthtget_polarizationtsintcosRR( tcif_filetd2rt cbf_handlet directiontjR tpolar_fractiont polar_angletpolar_plane_normal((s>/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/beam.pytimgCIFys,       ) c C sütjd}|jdƒ|jdƒ|jdƒg}x<tdƒD].}|jd|dƒ|j|jƒƒqGW|jƒ}y|j ƒ\}}Wnd}d }nXtj ||ƒtj ||ƒd f}t j d |d |d |d |ƒS(s(Initialize a detector model from an imgCIF file. N.B. the definition of the polarization plane is not completely helpful in this - it is the angle between the polarization plane and the +Y laboratory frame vector. This example works from a cbf_handle, which is already configured.g€f@RRRis vector[%d]ig+‡ÙÎ÷ï?gR R RR(RRR#R$R%R&R'R(R)R*R+R,RR(R/R.R0R1R R2R3R4((s>/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/beam.pytimgCIF_H§s(      ) N( t__name__t __module__t__doc__Rt staticmethodRRRRRRR5R6(((s>/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/beam.pyRs  #  .((t __future__RRRtdxtbx_model_extRR(((s>/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/beam.pyts