xTc@ sddlmZddlZyddlZWn dZnXyddlZWn dZnXddlZddlmZm Z ddl m Z m Z ddl mZmZddlmZmZdefdYZd efd YZdS( i(tdivisionN(t Goniometertgoniometer_factory(tDetectortdetector_factory(tBeamt beam_factory(tScant scan_factoryt _MetaFormatcB seZdZdZRS(s~A metaclass for the Format base class (and hence all format classes) to allow autoregistration of the class implementations.cC stt|j|||ytjtjWntk rBdSXg|_t|kryddl m }|j |dSx|D]}|jj |qWdS(Ni(tRegistry( tsuperR t__init__tsystmodulestFormatt __module__t NameErrort _childrentdxtbx.format.RegistryR taddtappend(tselftnametbasest attributesR tbase((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pyR *s     (t__name__Rt__doc__R (((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pyR &sRcB seZdZeZedZdZdZdZ dZ dZ dZ dZ d Zd Zd Zd Zd ZdZdZdZedZedZedZededZRS(sWA base class for the representation and interrogation of diffraction image formats, from which all classes for reading the header should be inherited. This includes: autoregistration of implementation classes, stubs which need to be overridden and links to static factory methods which will prove to be useful in other implementations.cC stS(s<Overload this to publish whether this class instance understands a given file. N.B. to say that we understand it, return True. If a subclass also understands the image (because, for example, its detector serial number takes a certain value), it will by definition understand it better than its superclass. Thus, the preferred class will be the deepest subclass in the inheritance hierarchy. Finally, if you are writing this subclass for a given instrument and you are given a different example return False. Implementing understand() in a subclass, one can safely assume that the superclasss understand() function returned True. The understand() function of two different classes directly derived from the same base should never both return True for the same input image.(tFalse(t image_file((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pyt understandJscC s_||_d|_d|_d|_d|_t|_t|_ t |_ t |_ |jdS(s/Initialize a class instance from an image file.N(t _image_filetNonet_goniometer_instancet_detector_instancet_beam_instancet_scan_instanceRt_goniometer_factoryRt_detector_factoryRt _beam_factoryRt _scan_factorytsetup(RR((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pyR ^s          cC s|jzy|j}t|ts1t||_|j}t|ts[t||_|j }t|t st||_ |j }t|t st||_Wntjk r}nXWd|jXdS(sjRead the image file, construct the information which we will be wanting about the experiment from this. N.B. in your implementation of this you will probably want to make use of the static methods below and probably add some format parsing code too. Please also keep in mind that your implementation may be further subclassed by someone else.N(t_startt _goniometert isinstanceRtAssertionErrorR"t _detectorRR#t_beamRR$t_scanRR%t exceptionst Exceptiont_end(Rtgoniometer_instancetdetector_instancet beam_instancet scan_instancete((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pyR*ps&          cC s|jS(sRGet the standard goniometer instance which was derived from the image headers.(R"(R((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pytget_goniometerscC s|jS(sPGet the standard detector instance which was derived from the image headers.(R#(R((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pyt get_detectorscC s|jS(sLGet the standard beam instance which was derived from the image headers.(R$(R((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pytget_beamscC s|jS(sLGet the standard scan instance which was derived from the image headers.(R%(R((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pytget_scanscC s#|j}|j|j}|S(sNGet the pixel intensities (i.e. read the image and return as a flex array.(t detectorbasetreadt get_raw_data(Rtimagetraw_data((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pyR@s   cC s dtfdY}||S(s%Return the instance of detector base.t_detectorbase_proxycB s#eZdZdZdZRS(cS s ||_dS(N(t_fi(Rtformat_instance((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pyR scS s&|dkr|jjStj||S(Nt __class__(RDRFtobjectt__getattribute__(RR((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pyRHs  cS sgy|jj|SWnLtk rby|jjj|SWqctk r^|jjj|SXnXdS(N(RDRHtAttributeErrorR>t __getattr__(RR((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pyRJs  (RRR RHRJ(((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pyRCs  (RG(RRC((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pytget_detectorbasescC s|jS(s/Get the image file provided to the constructor.(R (R((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pytget_image_filescC sdS(sStart code for handling this image file, which may open a link to it once, say, and pass this around within the implementation. How you use this is up to you, though you probably want to overload it...N((R((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pyR+scC sdS(siClean up things - keeping in mind that this should run even in the case of an exception being raised.N((R((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pyR4scC s tddS(shOverload this method to read the image file however you like so long as the result is an goniometer.s overload meN(t RuntimeError(R((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pyR,scC s tddS(sfOverload this method to read the image file however you like so long as the result is an detector.s overload meN(RM(R((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pyR/scC s tddS(sbOverload this method to read the image file however you like so long as the result is an beam.s overload meN(RM(R((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pyR0scC s tddS(sbOverload this method to read the image file however you like so long as the result is an scan.s overload meN(RM(R((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pyR1scC s?ddlm}||j}|r;t|dkr;tStS(sSee if the file is a URL.i(turlparsei(RNtschemetlentTrueR(tpathRNRO((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pytis_urls cC s0d|dkrtSdt|djdkS(s7Check if a file pointed at by filename is bzip2 format.s.bz2itBZhtrbi(RtopenR?(tfilename((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pytis_bz2scC sVd|dkrtSt|djd}t|ddkoUt|ddkS( s:Check if a file pointed at by filename is gzip compressed.s.gziRUiiiii(RRVR?tord(RWtmagic((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pytis_gzipsRUcC s|r1tj|r1ddl}|j|Sntj|rhtdkrXtdntj||Stj |rt dkrtdnt j ||St ||S(s;Open file for reading, decompressing silently if necessary.iNs$bz2 file provided without bz2 modules$gz file provided without gzip module( RRSturllib2turlopenRXtbz2R!RMtBZ2FileR[tgziptGzipFileRV(RWtmodeturlR\((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pyt open_file(s     (RRRR t __metaclass__t staticmethodRR R*R:R;R<R=R@RKRLR+R4R,R/R0R1RSRXR[RRd(((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pyRAs.  #              (t __future__RR R^R!R`R2tdxtbx.model.goniometerRRtdxtbx.model.detectorRRtdxtbx.model.beamRRtdxtbx.model.scanRRttypeR RGR(((sA/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/Format.pyts