ó ø™xTc@ soddlmZddlZddlZddlmZddlmZddlmZddd„ƒYZ dS( iÿÿÿÿ(tdivisionN(tScan(tscan_helper_image_files(tscan_helper_image_formatst scan_factorycB szeZdZeed„ƒZed„ƒZed„ƒZed„ƒZed„ƒZ ed„ƒZ ed„ƒZ RS(sgA factory for scan instances, to help with constructing the classes in a set of common circumstances.c C smddlm}t|tƒsW|d|dd}gt|ƒD] }|^qB}n|d|dd}t|ƒ}||kræ|dkr³gt|ƒD] }d^qž}qæ|jgt||ƒD]}|d^qʃ}ngt|ƒD]} || ^qó} tt t t |ƒƒt t t |ƒƒ|j tt t |ƒƒƒ|j tt t | ƒƒƒ|ƒS(Niÿÿÿÿ(tflexii(tscitbx.array_familyRt isinstancetlisttrangetlentextendtsortedRttupletmaptinttfloattdouble( t image_rangetexposure_timest oscillationtepochstdegRt num_imagestitnum_exptjt epoch_list((s>/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/scan.pyt make_scans$"   "-#cC siddl}tj|jj|ƒdƒ}|dkr@d}ntj||f|||fi||6ƒS(s.Construct an scan instance for a single image.iÿÿÿÿNg(tosRtimage_to_indextpathtsplittNoneRR(tfilenametformatRt osc_startt osc_widthtepochRtindex((s>/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/scan.pytsingle3s   cC s/tjƒ}|j|tjƒtj||ƒS(s,Initialize a scan model from an imgCIF file.(tpycbftcbf_handle_structt read_filet MSG_DIGESTRtimgCIF_H(tcif_filet cbf_handle((s>/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/scan.pytimgCIF?s cC sy|jƒ}|jƒd}|jƒ}t|jƒƒ}tj|ƒ}|j|ƒtj ||f||i||6ƒS(spInitialize a scan model from an imgCIF file handle, where it is assumed that the file has already been read.i( tget_integration_timet get_timestamptconstruct_goniometerR tget_rotation_rangeRRt__swig_destroy__RR(R.R/texposuret timestamptgoniotanglesR'((s>/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/scan.pyR-Hs   cC st|d|dƒS(s_Sum a list of scans wrapping the sligtly clumsy idiomatic method: sum(scans[1:], scans[0]).ii(tsum(tscans((s>/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/scan.pytaddZscC sMtj|ƒ\}}tj||ƒ}g|D]}tj|||ƒ^q.S(sÎGet a list of files which appear to match the template and directory implied by the input filename. This could well be used to get a list of image headers to read and hence construct scans from.(Rtimage_to_template_directoryttemplate_directory_to_indicest!template_directory_index_to_image(R"ttemplatet directorytindicesR'((s>/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/scan.pytsearch`s  cC s |jƒdkrtjS|jƒdkr2tjS|jƒdksV|jƒdkr]tjS|jƒdkrvtjS|jƒdkrtjStd|‚dS( sìReturn the correct format token for a given name, for example: cbf, CBF smv, SMV tiff, tif, TIFF raxis, RAXIS mar, MAR to the appropriate static token which will be used as a handle everywhere else in this.tCBFtSMVtTIFtTIFFtRAXIStMARsname %s not knownN(tupperRt FORMAT_CBFt FORMAT_SMVt FORMAT_TIFFt FORMAT_RAXISt FORMAT_MARt RuntimeError(tname((s>/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/scan.pyR#ps $( t__name__t __module__t__doc__t staticmethodtTrueRR(R0R-R<RCR#(((s>/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/scan.pyRs  (( t __future__RR)tcopytdxtbx_model_extRt scan_helpersRRR(((s>/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/scan.pyts