ó ü™xTc@ s½ddlmZddlZddlmZddlmZddlmZmZddl m Z defd„ƒYZ e d kr¹ddl Z x#e jd D]Ze jeƒGHqžWndS( iÿÿÿÿ(tdivisionN(tFormatCBFMultiTile(tHierarchicalDetector(tsqrtcol(tSorrytFormatCBFMultiTileHierarchycB sheZdZed„ƒZd„Zd„Zd„Zd„Zd„Z d„Z d„Z d d „Z RS( s+An image reading class multi-tile CBF filescC s‰tjƒ}|j|tjƒ|jƒdkr5tSy|jdƒ|jdƒWn/tk r„}dt |ƒkr{tS|‚nXt S(sVCheck to see if this looks like an CBF format image, i.e. we can make sense of it.itaxistequipment_componentt CBF_NOTFOUND( tpycbftcbf_handle_structt read_widefilet MSG_DIGESTtcount_elementstFalset find_categoryt find_columnt ExceptiontstrtTrue(t image_filet cbf_handlete((sV/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/FormatCBFMultiTileHierarchy.pyt understands   cC s)|j|ƒst‚tj||ƒdS(s3Initialise the image structure from the given file.N(RtAssertionErrorRt__init__(tselfR((sV/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/FormatCBFMultiTileHierarchy.pyR-scC stj|ƒdS(scParent class will open the image file as a cbf file handle, and keep the handle somewhere safe.N(Rt_start(R((sV/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/FormatCBFMultiTileHierarchy.pyR6sc C st|j}|j|ƒ}t|j|ƒƒ}t|j|ƒƒjƒ}|j|ƒ\}}d}|dkr÷|j||dt ƒ} t | d| d| d|d| d| d| d|d| d | d | d |dddddfƒ}ny|d kr`||||} t ddd| dddd| dddd| dddddfƒ}nt d |ƒ‚|S(s™ Get the 4x4 homogenous coordinate matrix for a given axis. Assumes the cbf handle has been intialized @param axis_id axis name of basis to get trotationtdegiiiiiiiiit translationsUnrecognized vector type: %dN( t _cbf_handlet get_axis_typeRtget_axis_offsettget_axis_vectort normalizetget_axis_settingtNonet$axis_and_angle_as_r3_rotation_matrixRRR( Rtaxis_idtcbft axis_typetoffsettvectortsettingt incrementtcobtr3R((sV/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/FormatCBFMultiTileHierarchy.pyt_get_change_of_basis;s(   c C s”|j}|j|ƒ}|j|ƒ}|dkr=d|fS|j|ƒ}|j|ƒ}||krŠ|j|ƒ\}}|||fS||fS(s× Get the 4x4 homogenous coordinate matrix for a given axis, combining it with the change of basis matrices of parent axes with the same equipment component as the given axis. Assumes the cbf handle has been intialized @param axis_id axis name of basis to get @return (parent, change of basis matrix), where parent is None if the parent in the cbf file is ".". Parent is the axis that the top level axis in this chain of dependent axis depends on t.N(R R1tget_axis_depends_onR&tget_axis_equipment_componentt _get_cummulative_change_of_basis( RR(R)R/t parent_idteq_comptparent_eq_comptnon_matching_parentt parent_cob((sV/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/FormatCBFMultiTileHierarchy.pyR5Zs    c C sm|dkst‚|}x0|jƒjƒD]}|jƒ|kr+|Sq+W|j|ƒ\}}|dkrÃ|jƒ}y|jƒWn+tk r³}dt|ƒksÀt‚qáXt sát‚n|j ||ƒ} | j ƒ}t |t dƒdd!ƒ} t |t dƒdd!ƒ| } t |t d ƒdd!ƒ| } |j | j| j| jƒ|j|ƒ|S( só Adds a panel group to the detector d. If the group's parent hasn't been added yet, recursively add parents to the detector until the detector itself is reached. @param group_id name of a cbf axis @param d detector object R2sDXTBX_ASSERT(D_)iiiN(iiii(iiii(iiii(Rt hierarchyt iter_preordertget_nameR5R&t get_D_matrixt RuntimeErrorRRt_add_panel_groupt add_groupRtset_local_frametelemstset_name( Rtgroup_idtdtnametsubobjtparentR/tpgRt parent_pgtorigintfasttslow((sV/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/dxtbx/format/FormatCBFMultiTileHierarchy.pyR@ss0    !!  cC s‹|j}tƒ}|jdƒy|jdƒWn;tk rm}dt|ƒkr]|‚n|jdƒnXg}xgt|jƒƒD]S}|j|ƒ|j ƒdkr²q‡n|j ƒ}||kr‡|j |ƒq‡q‡Wg}xEtt |ƒƒD]1}|j |ƒ}|jdƒ} |j | ƒq÷Wg} |jdƒxŠ|D]‚} |jdƒ|j| ƒy|jdƒWn;tk r´}dt|ƒkr¤|‚n|jdƒnX| j |j ƒƒqFWx¸|D]°} |j | j| ƒƒ}|jdƒ|jd ƒf} |jdƒ} |jdƒ}|j| ƒ|ksHt‚yt|j|ƒƒ}Wnätk rG}dt|ƒkr>ddg}|jdƒxt| |gD]f}|jdƒ|j|ƒ|jd ƒt|j ƒƒd}|jd ƒt|j ƒƒ|| s ÿQ