xTc@ sddlmZdefdYZdefdYZdefdYZdefd YZd efd YZd efd YZdefdYZ defdYZ dS(i(tdivisiont DataBlockcB seZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZRS(s: High level container for blocks of sweeps and imagesets. cC sCd|_g|_|dk r?x|D]}|j|q%WndS(s' Instantiate from a list of imagesets. N(tNonet _format_classt _imagesetstappend(tselft imagesetstiset((s=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pyt__init__s     cC st|jdkr?y|jj|_Wq`tk r;q`Xn!|j|jjks`t|jj|dS(s Add an imageset to the block. N(RRtreadertget_format_classt ExceptiontAssertionErrorRR(Rtimageset((s=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pyRs !cC s"x|D]}|j|qWdS(s Add two datablocks. N(R(Rt datablockR((s=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pytextend's cC s|jS(s Return the format class. (R(R((s=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pyt format_class,scC st|jS(s! Extract all the still imagesets (tlistt iter_stills(R((s=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pytextract_stills0scC st|jS(s( Extract all the sweeps from the block. (Rt iter_sweeps(R((s=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pytextract_sweeps4scC s t|jS(s Extract all imagesets. (RR(R((s=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pytextract_imagesets8scC s&tg|jD]}t|^q S(s Get the number of images. (tsumRtlen(RR((s=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pyt num_images<scC s t|jS(s The number of image sets. (RR(R((s=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pyt__len__@scC s"|j|jko!|j|jkS(s# Check if two blocks are the same. (RR(Rtrhs((s=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pyt__eq__DscC s|j| S(s$ Check if two blocks are not equal. (R(RR((s=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pyt__ne__Iscc sx|jD] }|Vq WdS(s Iterate through the imagesets. N(R(RR((s=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pyt__iter__Mscc s?ddlm}x(|jD]}t||r|VqqWdS(s Iterate over sweep groups. i(t ImageSweepN(tdxtbx.imagesetR Rt isinstance(RR R((s=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pyRRscc s?ddlm}x(|jD]}t||s|VqqWdS(s Iterate over still groups. i(R N(R!R RR"(RR R((s=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pyRYscC sddlm}ddlm}|}xc|jD]X}t||r[d||jsN(R>R2ReRdR?R3t unhandledRoRTRlRRRAt_create_single_file_imagesett_extract_file_metadataRt_create_multi_file_imageset(RRiRVR2RdR3RpRhRYRR;Rtrecordstitems((RRVs=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pyR s*       cC sddlm}dtfdY}g}d}x|D]{}||}y|j} Wntk rtd} nXy|j} Wntk rd} nXy|j} Wntk rd} nXy|j} Wntk rd} nX| dk r0t | j ddkr0||\} }n d\} }t |dkrt|d}t t t g}|j | kr|j } t|ds(tcPickleRtloadR"RR}R R(R;tpickleRR(((s=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pytfrom_pickle_files cC st|}|jS(s9 Load a datablock from an imageset of list of imagesets. (RRo(RR((s=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pyt from_imagesets cC s,ddlm}|j|}tj|S(s% Load a datablock from a sweep file. i(R(tdxtbx.serializeRRRR(R;RR((s=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pytfrom_imageset_json_filescC s\ytj||SWntk r)}nXytj|SWntk rNnXtj|S(s+ Load a datablock from serialized formats. (RRR RR(R;RR((s=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pyRs cC s&ddlm}t|||gS(s= Function to instantiate data block from in memory imageset. i(t MemImageSet(R!RR(R=tindicesR((s=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pytfrom_in_memorysN(RQRRRSt staticmethodR`RRRRbRRRRRRRR(((s=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pyRs$   tDataBlockDumpercB s;eZdZdZdedZddZdZRS(s- Class to help in dumping datablock objects. cC s+t|tr|g|_n ||_dS(s% Initialise the list of data blocks. N(R"Rt _datablocks(RRo((s=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pyR sc C sddlm}ddl}ddl}||d}g|jD]}|j^qB}|r|j|t|ddd d tn%|j|t|dd d d tdS( s Dump datablock as json. i(tsplitextNitwt separatorst,t:t ensure_asciitindenti(RR( R@RRRRREtdumpRRb( RR;tcompactRRRtextRt dictionary((s=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pytas_jsons  "cK scddl}|j|jd|j}|dk r[t|d}|j|WdQXn|SdS(s Dump datablock as pickle. iNtprotocoltwb(RtdumpsRtHIGHEST_PROTOCOLRRtwrite(RR;tkwargsRttexttoutfile((s=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pyt as_pickles   cK sddlm}||d}dg}dddg}|j|krZ|j||S|j|kr||j||Sdj||}td ||fd S( s Dump datablocks as file. i(Ris.jsons.ps.pkls.picklet|sexpected extension {%s}, got %sN(R@RtlowerRRtjoinRs(RR;RRRtj_exttp_exttext_str((s=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pytas_file#s N( RQRRRSR RR`RRR(((s=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pyRs   N( t __future__RR_RRTRmRRRRR(((s=/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/datablock.pyt sPNSm