xTc@ sddlmZdefdYZdefdYZdefdYZdefd YZd efd YZd S( i(tdivisiont ExperimentcB sVeZdZdddddddZdZdZdZdZdZ RS(s) A class to represent what's in an experiment. Contains: - imageset Access to the image data - beam The beam model - detector The detector model - goniometer The goniometer model - scan The scan model - crystal The crystal model Some of these may be set to "None" cC s:||_||_||_||_||_||_dS(s2 Initialise the experiment with the given models. N(timagesettbeamtdetectort goniometertscantcrystal(tselfRRRRRR((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyt__init__s      cC sX||jkpW||jkpW||jkpW||jkpW||jkpW||jkS(s- Check if the experiment contains the model. (RRRRRR(Rtitem((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyt __contains__&s cC s}t|tstS|j|jko||j|jko||j|jko||j|jko||j|jko||j|jkS(s0 Check if an experiment is the same as another. ( t isinstanceRtFalseRRRRRR(Rtother((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyt__eq__/scC s|j| S(s. Check if an experiment not equal to another. (R(RR((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyt__ne__:scC stS(sD If a scan is present, check that it makes sense with the imageset. (tTrue(R((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pytassert_is_consistent>scC s+y|jtSWntk r&tSXdS(sD If a scan is present, check that it makes sense with the imageset. N(RRt ExceptionR (R((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyt is_consistentKs   N( t__name__t __module__t__doc__tNoneR R RRRR(((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyR s    tExperimentListcB seZdZddZdZdZdZdZdZ dZ dZ d Z d Z d Zd Zd ZdZdZdZdZdZdZdZdZdZdZdZRS(s The experiment list class. This class is used to manage all the experiments and contains methods to get groups of models etc. cC s.|dk rt||_n t|_dS(s Initialise the list. N(Rtlistt_data(RR ((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyR Ws cC s9t|tr||j|{s(Rtany(RR ((R sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyR xscC s|jj|S(s! Get the index of an experiment. (RR(RR ((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyR}scC s<t|tr"|jj|ntdt|dS(s# Add a new experiment to the list. s expected type Experiment, got %sN(R RRtappendRR(RR ((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyR)scC s?t|tr%|jj|jntdttdS(s* Add another experiment list to this one. s$expected type ExperimentList, got %sN(R RRtextendRRR (RR((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyR*scC sx|j|D]}|j|}|j|kr>||_q|j|krY||_q|j|krt||_q|j|kr||_q|j|kr||_q|j|kr||_qtd|qWdS(s% Replace all occurances of a with b. sunidentified model %sN( tindicesRRRRRRRt ValueError(Rtatbtitexp((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pytreplaces       cC s|j|ddS(s% Remove all occurances of the model. N(R1R(Rtmodel((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pytremovescC st|tst|trSttjg|D]}t|j|^q.Sgt|D]\}}||kr`|^q`SdS(s> Get the indices of the experiments which contains the model. N(R Rttupletsett intersectionR+t enumerate(RR2tmR/R%((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyR+s5cC s<ddlm}|g|D]}|jdf^qjS(s1 Get a list of the unique beams (includes None). i(t OrderedDictN(tlibtbx.containersR9RRtkeys(RR9R%((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pytbeamsscC s<ddlm}|g|D]}|jdf^qjS(s5 Get a list of the unique detectors (includes None). i(R9N(R:R9RRR;(RR9R%((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyt detectorsscC s<ddlm}|g|D]}|jdf^qjS(s7 Get a list of the unique goniometers (includes None). i(R9N(R:R9RRR;(RR9R%((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyt goniometersscC s<ddlm}|g|D]}|jdf^qjS(s1 Get a list of the unique scans (includes None). i(R9N(R:R9RRR;(RR9R%((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pytscansscC s<ddlm}|g|D]}|jdf^qjS(s4 Get a list of the unique crystals (includes None). i(R9N(R:R9RRR;(RR9R%((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pytcrystalsscC sKddlm}|g|D]$}|jdk r|jdf^qjS(su Get a list of the unique imagesets (includes None). This returns unique complete sets rather than partial. i(R9N(R:R9RRR;(RR9R%((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyt imagesetsscC s#tg|D]}|j^q S(s& Check all the models are consistent. (tallR(RR%((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyRscC s[t|dkstt}x6|D].}|jdk r%|jdk r%t}Pq%q%W|S(s) Check if all the experiments are stills iN(R#tAssertionErrorRRRRR (RtresultR%((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyt all_stillss cC s[t|dkstt}x6|D].}|jdksI|jdkr%t}Pq%q%W|S(s. Check if all the experiments are from sweeps iN(R#RCRRRRR (RRDR%((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyt all_sweepss c s1ddlm}ddlm}m}m}ddlm}ddlm |j s^t |j }|j }|j}|j} |j} |j} |} d| dR?R@RARRRRRRRRR R#t complete_setR+R)tpathsRR(RR9RGRHRItcrystal_to_dicttblisttdlisttglisttslisttclisttilistRDRPR%tobjRTRWRXR.tdtgtstc((RKsT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyRJsp             "$&      /////cC sddlm}|j}d|dR?R@RARRERFRJRp(((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyRSs2                   `tExperimentListDictcB seZdZedZdZdZdZdZdZ dZ dZ e d Z e d Ze d Ze d Ze d Ze dZe dZRS(sx A helper class for serializing the experiment list to dictionary (needed to save the experiment list to JSON format. cC s,ddlm}|||_||_dS(s" Initialise. Copy the dictionary. i(tdeepcopyN(tcopyRrt_objt _check_format(RRft check_formatRr((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyR escC ss|jd|_|jd|_|jd|_|jd|_|jd|_|j|_|jS(s3 Decode the dictionary into a list of experiments. RRRRR( t_extract_modelst_blistt_dlistt_glistt_slistt_clistt_extract_imagesetst_ilistt_extract_experiments(R((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pytdecodekscC st|d|}|jj|g}g|D]}||^q/}i}x|jdD]}|j|d}|dkrq[q[t|tr||krt|||<|j|tj |n||||eZdZdZdeedZddZdZRS(s% A class to help writing JSON files. cC s ||_dS(s Initialise N(t_experiment_list(Rtexperiment_list((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyR bscC sddl}ddlm}ddlm}|jj}|dk r|r||d}gt|dD]"\} } d|| f| f^qn} gt|dD]"\} } d || f| f^q} gt|d D]"\} } d || f| f^q} gt|d D]"\} } d || f| f^q}gt|dD]"\} } d|| f| f^qR}gt|dD]"\} } d|| f| f^q}|dd|dfg}x|dD]}d|kr| |dd|dsRiiRRRiRiRN(t itertoolsRRR#R)RRRRRRtzipRRR( RRRRRtgonioRR/RR8R+((RsT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyRs8     1  "cC st}t|trFx'|D]}|jtj||qW|Sx-|jD]}|jtj||qSW|jst |S(s+ Load an experiment list from a datablock. ( RR RR*Rtfrom_datablock_and_crystaltextract_imagesetsRRRC(t datablockRRtdbR((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyRs    cC s+t||j}|js't|S(s, Load an experiment list from a dictionary. (RqRRRC(RfRvR((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyRn6scC s;ddlm}ddl}tj|j|d||S(s$ Load an experiment list from JSON. i(RNR(RRRRRnR(RRvRR((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyt from_jsonCs  c C s}ddlm}ddlm}m}||}|||3t|d}tj|j|SWdQXWdQXdS(s+ Load an experiment list from a json file. i(R(RtabspathRN( RRRRRRRRR(RRvRRRR((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pytfrom_json_fileLs  cC sPddl}t|d/}|j|}t|tsBt|SWdQXdS(s/ Decode an experiment list from a pickle file. iNtrb(RRtloadR RRC(RRRRf((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pytfrom_pickle_fileVs  cC soddlm}ddlm}|j||}|j|}tj||}t|dkskt |S(s- Generate an experiment list from XDS files. i(txds(Rki( tdxtbx.serializeRRmRkt to_imagesett to_crystalRRR#RC(txds_inpt xds_otherRRktsweepRR((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pytfrom_xds_s cC s7ytj||SWntk r)}nXtj|S(s; Try to load the experiment list from a serialized format. (RRRR(RRvR%((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyRus N(RRRRR RRRRRRRRnRRRRR(((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyRs"  &   N(t __future__RtobjectRRRqRR(((sT/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/model/experiment/experiment_list.pyt s Fg