B \0@sdZddlmZddlZddlZddlZy ddlmZmZm Z m Z m Z Wn0e k rxddl mZmZm Z m Z m Z YnXddlmZmZddlmZmZmZmZmZmZmZdd lmZmZd d Zd d ZddZddZeZ eZ!ddZ"Gddde#Z$Gddde#Z%Gddde$Z&Gddde Z'Gddde Z(Gddde Z)Gd d!d!eZ*Gd"d#d#e*eZ+Gd$d%d%e#Z,dS)&zJ Implements operations common to all high-level objects (File, etc.). )absolute_importN)MappingMutableMappingKeysView ValuesView ItemsView)fspathfilename_encode)h5dh5ih5rh5ph5fh5th5s)phil with_philc Cs>t0tjt|}tj|r0tt|SdSQRXdS)z@ Determine if a file is valid HDF5 (False if it doesn't exist). FN) rospathabspathr isfileris_hdf5r )fnamer,lib/python3.7/site-packages/h5py/_hl/base.pyr&s  rc Cs|tnt|tjr tjtjdSt|tjr:tjtjdSt|tkrRtjtdSt|t j krntjt j dSdSQRXdS)z Attempt to guess an appropriate dtype for the object, returning None if nothing is appropriate (or if it should be left up the the array constructor to figure out) )ref)ZvlenN) r isinstancerZRegionReferencerZ special_dtypeZ ReferencetypebytessixZ text_type)datarrr guess_dtype0s    r#cCs2ttj}ttj}|tj|||S)z# Default link access property list )rcreateZ LINK_ACCESSZ FILE_ACCESSZset_fclose_degreerZ CLOSE_STRONGZset_elink_fapl)ZlaplZfaplrrr default_laplBs     r%cCsttj}|d|S)z% Default link creation property list T)rr$Z LINK_CREATEZset_create_intermediate_group)lcplrrr default_lcplKs  r'cCs|tjkrdSdS)z) Check if an object's dataspace is empty TF)Z get_spaceZget_simple_extent_typerZNULL)objrrris_empty_dataspaceUsr)c@s:eZdZdZeddZeddZd ddZd d ZdS) CommonStateObjecta  Mixin class that allows sharing information between objects which reside in the same HDF5 file. Requires that the host class have a ".id" attribute which returns a low-level ObjectID subclass. Also implements Unicode operations. cCstS)zI Fetch the link access property list appropriate for this object )dlapl)selfrrr_laplfszCommonStateObject._laplcCstS)zK Fetch the link creation property list appropriate for this object )dlcpl)r,rrr_lcpllszCommonStateObject._lcplNcsfdd}|dkr |rdSdSt|tr2tj}n:y|d}tj}Wn$tk rj|d}tj}YnX|r||||fS|S)aG Encode a name according to the current file settings. Returns name, or 2-tuple (name, lcpl) if lcpl is True - Binary strings are always passed as-is, h5t.CSET_ASCII - Unicode strings are encoded utf8, h5t.CSET_UTF8 If name is None, returns either None or (None, None) appropriately. csj}|||S)z3 Create an appropriate link creation property list )r/copyZset_char_encoding)codingr&)r,rrget_lcpl|s  z&CommonStateObject._e..get_lcplN)NNasciiutf8)rr rZ CSET_ASCIIencodeUnicodeEncodeErrorZ CSET_UTF8)r,namer&r2r1r)r,r_ers        zCommonStateObject._ecCs0|dkr dSy |dStk r*YnX|S)z Decode a name according to the current file settings. - Try to decode utf8 - Failing that, return the byte string If name is None, returns None. Nr4)decodeUnicodeDecodeError)r,r7rrr_ds zCommonStateObject._d)N) __name__ __module__ __qualname____doc__propertyr-r/r8r;rrrrr*\s    !r*c@s0eZdZdZddZddZddZdd Zd S) _RegionProxya  Proxy object which handles region references. To create a new region reference (datasets only), use slicing syntax: >>> newref = obj.regionref[0:10:2] To determine the target dataset shape from an existing reference: >>> shape = obj.regionref.shape(existingref) where may be any object in the file. To determine the shape of the selection in use on the target dataset: >>> selection_shape = obj.regionref.selection(existingref) cCs |j|_dS)N)id)r,r(rrr__init__sz_RegionProxy.__init__c Cs^t|jtjstdddlm}t.|j|jj ||jd}t |jdt j |jSQRXdS)Nz.Region references can only be made to datasetsr) selections)Zdsid.) rrBr Z DatasetID TypeErrorrDrZselectshaperr$ZDATASET_REGION)r,argsrD selectionrrr __getitem__s  z_RegionProxy.__getitem__c Cs$tt||j}|jSQRXdS)z= Get the shape of the target dataspace referred to by *ref*. N)rr get_regionrBrH)r,rsidrrrrHsz_RegionProxy.shapec Cs4ddlm}tt||j}||SQRXdS)zN Get the shape of the target dataspace selection referred to by *ref* r)rDN)rGrDrrrLrBZ guess_shape)r,rrDrMrrrrJs z_RegionProxy.selectionN)r<r=r>r?rCrKrHrJrrrrrAs rAc@seZdZdZeddZeeddZeeddZeedd Z eed d Z eed d Z eddZ eddZ eddZeddZeddZddZeZdS)HLObjectz: Base class for high-level interface objects. c Cs(ddlm}t||jSQRXdS)z4 Return a File instance associated with this object r)filesN)rGrOrZFilerB)r,rOrrrfiles z HLObject.filecCs|t|jS)z: Return the full name of this object. None if anonymous. )r;r Zget_namerB)r,rrrr7sz HLObject.namecCs$|jdkrtd|jt|jS)zReturn the parent group of this object. This is always equivalent to obj.file[posixpath.dirname(obj.name)]. ValueError if this object is anonymous. Nz*Parent of an anonymous object is undefined)r7 ValueErrorrP posixpathdirname)r,rrrparents zHLObject.parentcCs|jS)z2 Low-level identifier appropriate for this object )_id)r,rrrrBsz HLObject.idcCst|jdtjS)z+ An (opaque) HDF5 reference to this object rE)rr$rBZOBJECT)r,rrrrsz HLObject.refcCst|S)azCreate a region reference (Datasets only). The syntax is regionref[]. For example, dset.regionref[...] creates a region reference in which the whole dataset is selected. Can also be used to determine the shape of the referenced dataset (via .shape property), or the shape of the selection (via the .selection property). )rA)r,rrr regionrefs zHLObject.regionrefc Cs&ddlm}t ||SQRXdS)z$ Attributes attached to this object r)attrsN)rGrWrZAttributeManager)r,rWrrrrW s zHLObject.attrscCs ||_dS)z3 Setup this object, given its low-level identifier N)rU)r,ZoidrrrrCszHLObject.__init__cCs t|jS)N)hashrB)r,rrr__hash__szHLObject.__hash__cCst|dr|j|jkSdS)NrBF)hasattrrB)r,otherrrr__eq__s  zHLObject.__eq__cCs || S)N)r\)r,r[rrr__ne__!szHLObject.__ne__c Cst t|jSQRXdS)N)rboolrB)r,rrr__bool__%szHLObject.__bool__N)r<r=r>r?r@rPrr7rTrBrrVrWrCrYr\r]r_Z __nonzero__rrrrrNs&       rNc@seZdZddZeZdS) KeysViewHDF5cCsdt|S)Nz)formatlist)r,rrr__str__6szKeysViewHDF5.__str__N)r<r=r>rc__repr__rrrrr`5sr`c@s eZdZdZddZddZdS)ValuesViewHDF5z Wraps e.g. a Group or AttributeManager to provide a value view. Note that __contains__ will have poor performance as it has to scan all the links or attributes. c Cs8t*x"|jD]}||j|krdSqWdSQRXdS)NTF)r_mappingget)r,valuekeyrrr __contains__Ds  zValuesViewHDF5.__contains__c cs2t$x|jD]}|j|VqWWdQRXdS)N)rrfrg)r,rirrr__iter__Ks zValuesViewHDF5.__iter__N)r<r=r>r?rjrkrrrrre;srec@s eZdZdZddZddZdS) ItemsViewHDF5zN Wraps e.g. a Group or AttributeManager to provide an items view. c Cs6t(|\}}||jkr(||j|kSdSQRXdS)NF)rrfrg)r,itemrivalrrrrjWs  zItemsViewHDF5.__contains__c cs6t(x |jD]}||j|fVqWWdQRXdS)N)rrfrg)r,rirrrrk^s zItemsViewHDF5.__iter__N)r<r=r>r?rjrkrrrrrlQsrlc@sXeZdZdZejrvsz&MappingHDF5.values..N)r)r,r)r,rvaluessszMappingHDF5.valuesccsx|D]}||VqWdS)z% Get an iterator over member objects N)rg)r,rrrrr itervaluesxs zMappingHDF5.itervaluesc s"tfddDSQRXdS)z6 Get a list of tuples containing (name, object) pairs csg|]}||fqSr)rg)rqrr)r,rrrssz%MappingHDF5.items..N)r)r,r)r,ritems}szMappingHDF5.itemsccs"x|D]}|||fVqWdS)z+ Get an iterator over (name, object) pairs N)rg)r,rrrrr iteritemss zMappingHDF5.iteritemscCst|S)z# Get a view object on member names )r`)r,rrrrpscCst|S)z% Get a view object on member objects )re)r,rrrrtscCst|S)z# Get a view object on member items )rl)r,rrrrvsN) r<r=r>r?r!ZPY2rprtrurvrwrrrrrods roc@seZdZdZdS)MutableMappingHDF5z Wraps a Group or AttributeManager object to provide a mutable mapping interface, in contrast to the read-only mapping of MappingHDF5. N)r<r=r>r?rrrrrxsrxc@s0eZdZdZdZdZddZddZddZdS) Emptyz Proxy object to represent empty/null dataspaces (a.k.a H5S_NULL). This can have an associated dtype, but has no shape or data. This is not the same as an array with shape (0,). NcCs ||_dS)N)dtype)r,rzrrrrCszEmpty.__init__cCst|tr|j|jkrdSdS)NTF)rryrz)r,r[rrrr\sz Empty.__eq__cCs d|jS)NzEmpty(dtype={0!r}))rarz)r,rrrrdszEmpty.__repr__) r<r=r>r?rHsizerCr\rdrrrrrys ry)-r?Z __future__rrRrr!Zcollections.abcrrrrr ImportError collectionscompatr r rGr r rrrrrZ_objectsrrrr#r%r'r+r.r)objectr*rArNr`rerlrorxryrrrr s6  "$  I-c1