a Sic$@sdZddlmZddlmZddlZddlmZddl m Z ddl m Z m Z d d lmZmZmZmZmZd d lmZGd d d ed dZdZejddZeejkrdZddZGdddZGdddZdS)z= High-level interface for creating HDF5 virtual datasets )deepcopy) namedtupleNfilename_encode)Datatype)SimpleSelectionselect)h5dh5ph5sh5th5)versionc@seZdZdZdS)VDSmapzNDefines a region in a virtual dataset mapping to part of a source dataset N)__name__ __module__ __qualname____doc__rrH/var/www/html/django/DPS/env/lib/python3.9/site-packages/h5py/_hl/vds.pyrsr)vspace file_nameZ dset_name src_spaceFTcCst|tr|n|f}|}|tjkr|r|}t|}d}|\}}}} t |D]R\} } t| t r| j tj krt |} | |} tj | | <t| }qZ| turZ||}qZ||||| dS)zz Converts the space with the given key. Mainly used to allow unlimited dimensions in virtual space selection. rN) isinstancetupleget_select_typer SEL_HYPERSLABSis_regular_hyperslabget_simple_extent_ndimslenget_regular_hyperslab enumerateslicestop UNLIMITEDlistEllipsisselect_hyperslab)spacekey type_coderanknargsZ idx_offsetstartstridecountblockislcountsidxrrr_convert_space_for_key's"     r8c@s.eZdZdZd ddZeddZddZdS) VirtualSourceaSource definition for virtual data sets. Instantiate this class to represent an entire source dataset, and then slice it to indicate which regions should be used in the virtual dataset. path_or_dataset The path to a file, or an h5py dataset. If a dataset is given, no other parameters are allowed, as the relevant values are taken from the dataset instead. name The name of the source dataset within the file. shape A tuple giving the shape of the dataset. dtype Numpy dtype or string. maxshape The source dataset is resizable up to this shape. Use None for axes you want to be unlimited. Nc Csddlm}t||rldd||||dD}|rFtdj|d|}|jj} |j}|j }|j }|j }nF|} |durtd|durtd nt|t r|f}t|t r|f}| |_ ||_||_ |dur||_ ntd d |D|_ t||_dS) Nr)DatasetcSsi|]\}}|dur||qSNr).0kvrrr `sz*VirtualSource.__init__..)nameshapedtypemaxshapezhIf a Dataset is passed as the first argument then no other arguments may be passed. You passed {failed})failedz?The name parameter is required when specifying a source by pathz@The shape parameter is required when specifying a source by pathcSsg|]}|durtjn|qSr;r r')r<ixrrr sz*VirtualSource.__init__..)datasetr:ritems TypeErrorformatfilefilenamer@rArBrCintpathrrsel) selfZpath_or_datasetr@rArBrCr:rDdsrOrrr__init__\sF      zVirtualSource.__init__cCs|jjSr;)rP array_shape)rQrrrrAszVirtualSource.shapecCs,t|}t|j|dd|_t|jj||S)NrH)copyr rArPr8id)rQr,tmprrr __getitem__szVirtualSource.__getitem__)NNNN)rrrrrSpropertyrArYrrrrr9Hs , r9c@sFeZdZdZdddZddZeeddd Zd d Z dd d Z dS) VirtualLayouta!Object for building a virtual dataset. Instantiate this class to define a virtual dataset, assign to slices of it (using VirtualSource objects), and then pass it to group.create_virtual_dataset() to add the virtual dataset to a file. This class does not allow access to the data; the virtual dataset must be created in a file before it can be used. shape A tuple giving the shape of the dataset. dtype Numpy dtype or string. maxshape The virtual dataset is resizable up to this shape. Use None for axes you want to be unlimited. filename The name of the destination file, if known in advance. Mappings from data in the same file will be stored with filename '.', allowing the file to be renamed later. NcCsRt|tr|fn||_||_t|tr,|fn||_||_t|_t tj |_ dSr;) rrNrArBrC _filenameset_src_filenamesr createDATASET_CREATEdcpl)rQrArBrCrMrrrrSs zVirtualLayout.__init__cCsft|j|dd}t|j|||j|j}|j|j||j d|j j|jdurb|j |dS)NrUutf-8)r rAr8rW_source_file_namerOr\ra set_virtualr@encoderPr^add)rQr,sourcerP src_filenamerrr __setitem__s  zVirtualLayout.__setitem__)returncCs$t|}|r|t|krdSt|S)N.r)rh dst_filenamerrrrcszVirtualLayout._source_file_namec Cst|}|jdur:|t|jkr4t|d|j|jS||jvrttj}t|j D]F}|j |}| |j || |||j|d|j|q^|S|jSdS)zGet the property list containing virtual dataset mappings If the destination filename wasn't known when the VirtualLayout was created, it is handled here. Nz != rb)rr\ Exceptionrar^r r_r`rangeget_virtual_countget_virtual_filenamerdget_virtual_vspacercget_virtual_dsetnamereget_virtual_srcspace)rQrlZnew_dcplr4rhrrr _get_dcpls"       zVirtualLayout._get_dcplc Cs||jj}|dur(|t|g|j}|durHtdd|D}t |j |}t |j t rl|j j}nt |j }tj|dd}tj|j||||dS)zA Return a new low-level dataset identifier for a virtual dataset Ncss |]}|dur|ntjVqdSr;rE)r<mrrr z-VirtualLayout.make_dataset..r)logical)r@tidr+ra)rtrLrMset_fill_valuenparrayrCrr create_simplerArrBrrWr py_creater r_) rQparentr@ fillvaluerarCZ virt_dspaceryrBrrr make_datasets   zVirtualLayout.make_dataset)NN)N) rrrrrSri staticmethodbytesrcrtrrrrrr[s   r[)rrVr collectionsrnumpyr{compatrdatatyper selectionsrr r r r rrrr vds_supporthdf5_version_tuple hdf5_version get_configvds_min_hdf5_versionr8r9r[rrrr s      !J