Pi3 `ddlZddlmZmZddlmZddlmZmZm Z m Z m Z m Z ddl ZddlmZddlmZddlmZdd lmZmZdd lmZmZer ddlZd d lmZejrddlZGd dej j!Z"eGddZ#dddde$de%e&e e e&e'fffdZ(dS)N) dataclassfield)Path) TYPE_CHECKINGAnyClassVarDictOptionalUnion)config)DownloadConfig) array_cast) is_local_pathxopen)no_op_if_value_is_nullstring_to_dict) FeatureTypecBeZdZdZdejjffd ZdZxZ S)Nifti1ImageWrapperzY A wrapper around nibabel's Nifti1Image to customize its representation. nifti_imagec t|j|j|j|j|j|||_dS)N)dataobjaffineheaderextrafile_mapdtype) super__init__rrrrrget_data_dtyper)selfr __class__s k/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/datasets/features/nifti.pyr!zNifti1ImageWrapper.__init__sa GG  #+")")!'$-!0022     +D   cddlm}m}m}m}ddlm}|j}|}| |j |j |j _ d|j _d|j _d}t!|jdrFd|jjvr8t%|jjddd|jjdj}|d}||| } || g||dS) Nr)NiiVue ShowRender SliceTypeVolume)displayTrimagefilenamez volume.nii.gz)namedata) ipyniivuer(r)r*r+IPython.displayr,rto_bytesset_slice_type MULTIPLANARALWAYSoptsmultiplanar_show_rendershow_3d_crosshairmultiplanar_force_renderhasattrrgetattrr. load_volumes) r#r(r)r*r+r,bytes_nvr/volumes r% _repr_html_zNifti1ImageWrapper._repr_html_'s/ G G G G G G G G G G G G / / / / / /%..00FB   i3 4 4 4.8.?BG +(,BG %/3BG ,Dt'44 Gt/888 0 9' BJPTUUa+4W=FD|&VF333F OOVH % % % GBKKKKKr&) __name__ __module__ __qualname____doc__nibnifti1 Nifti1Imager!rA __classcell__)r$s@r%rrsb   + (> + + + + + +       r&rceZdZUdZdZeed<eddZe e ed<dZ e e ed <e je je jd Ze eed <eddd Ze ed <dZdee eeedfdefdZddeddfdZdde jde jfdZdedee dfffdZdee je je j fde jfdZ!dS)Niftia **Experimental.** Nifti [`Feature`] to read NIfTI neuroimaging files. Input: The Nifti feature accepts as input: - A `str`: Absolute path to the NIfTI file (i.e. random access is allowed). - A `pathlib.Path`: path to the NIfTI file (i.e. random access is allowed). - A `dict` with the keys: - `path`: String with relative path of the NIfTI file in a dataset repository. - `bytes`: Bytes of the NIfTI file. This is useful for archived files with sequential access. - A `nibabel` image object (e.g., `nibabel.nifti1.Nifti1Image`). Args: decode (`bool`, defaults to `True`): Whether to decode the NIfTI data. If `False` a string with the bytes is returned. `decode=False` is not supported when decoding examples. Examples: ```py >>> from datasets import Dataset, Nifti >>> ds = Dataset.from_dict({"nifti": ["path/to/file.nii.gz"]}).cast_column("nifti", Nifti()) >>> ds.features["nifti"] Nifti(decode=True, id=None) >>> ds[0]["nifti"] >>> ds = ds.cast_column("nifti", Nifti(decode=False)) >>> ds[0]["nifti"] {'bytes': None, 'path': 'path/to/file.nii.gz'} ``` TdecodeNF)defaultrepridznibabel.nifti1.Nifti1Imagerbytespathpa_type)rMinitrN_typec|jSN)rS)r#s r%__call__zNifti.__call__ks |r&valuenib.Nifti1ImagereturncDtjrddl}nd}t|tr|ddSt|t r$t |ddSt|ttfrd|dS|)t||j j rt|St|tr| d=tj|drd| ddS| d| d+| d| ddSt#d|dt#d t%|d) zEncode example into a format for Arrow. Args: value (`str`, `bytes`, `nibabel.Nifti1Image` or `dict`): Data passed as input to Nifti feature. Returns: `dict` with "path" and "bytes" fields rNrRrQrRrPrQzTA nifti sample should have one of 'path' or 'bytes' but they are missing or None in .zPA nifti sample should be a string, bytes, Path, nibabel image, or dict, but got )r NIBABEL_AVAILABLEnibabel isinstancestrrabsoluterQ bytearray spatialimages SpatialImageencode_nibabel_imagedictgetosrRisfile ValueErrortype)r#rYrFs r%encode_examplezNifti.encode_examplens  #  ! ! ! ! !C eS ! ! !D11 1 t $ $  0 011DAA A y1 2 2  511 1 _E33D3Q!R!R_'.. . t $ $ yy  ,f 1N1N,!%uyy/@/@AAA7##/599V3D3D3P!&7!3!3UYYv=N=NOOO skpsssqcghmcncnqqq r&rctjrddl}ntd|i}|d|d}}|Z|t d|d|drCt |d d r|d d }t |r|j|}n |d d }|tj r tj n tj }t||} | | | d nd} t| } t|d | 5} |j| }dddn #1swxYwYnAddl} |dddkr| |}|j|}t)|S)aDecode example NIfTI file into nibabel image object. Args: value (`str` or `dict`): A string with the absolute NIfTI file path, a dictionary with keys: - `path`: String with absolute or relative NIfTI file path. - `bytes`: The bytes of the NIfTI file. token_per_repo_id (`dict`, *optional*): To access and decode NIfTI files from private repositories on the Hub, you can pass a dictionary repo_id (`str`) -> token (`bool` or `str`). Returns: `nibabel.Nifti1Image` objects rNz:To support decoding NIfTI files, please install 'nibabel'.rRrQzBA nifti should have one of 'path' or 'bytes' but both are None in r^zgzip://::repo_idtokenrbdownload_configr s)r r_r` ImportErrorrl startswithrsplitload HF_ENDPOINTHUB_DATASETS_URLHUB_DATASETS_HFFS_URLrrirrgzip decompressrH from_bytesr)r#rYtoken_per_repo_idrFrRr>nifti source_urlpatternsource_url_fieldsrtrwfrs r%decode_examplezNifti.decode_examples9&  # \ ! ! ! ! !Z[[ [  $ " V}eGnf >| !nfk!n!n!nooo??9--0- 4@P@PQS@T2U2U0::d++B/D &&,$CHTNNEE!%D!1!1"!5J&001CDD://#9 )7z7(K(K%O`Ol)--.? .JKKKrv'55&A&A&AOtT?KKK,q ( ,,,,,,,,,,,,,,, KKKrr k))00O..v66E!%(((s)FF  F storagecitfdtjfd|Dtj}tjd|dDtj}tj||gddg| }t||j S) a8Embed NifTI files into the Arrow array. Args: storage (`pa.StructArray`): PyArrow array to embed. Returns: `pa.StructArray`: Array in the NifTI arrow storage type, that is `pa.struct({"bytes": pa.binary(), "path": pa.string()})`. Nc|dd}|tjr tjn tj}t ||}||dnd}t|}t|d|5}| cdddS#1swxYwYdS)Nrprqrrrsrurv) rzryr r|r}r~rrirrread)rRrrrrtrwrrs r% path_to_bytesz*Nifti.embed_storage..path_to_bytess D))"-J+5+@+@AS+T+Tv''Z`Zv !/z7 C C K\Kh%))*;I*FGGGnrE,5999OtT?CCC qvvxx                  s%CC  C cZg|]'}|!|d|dn |dnd(S)NrQrR).0xrs r% z'Nifti.embed_storage..sU   UVTaQwZ-?qy)))QwZZgk   r&rmcTg|]%}|tj|nd&SrW)rjrRbasename)rrRs r%rz'Nifti.embed_storage..s2 p p pdt'7RW  d # # #T p p pr&rRrQmask) rpaarray to_pylistbinaryrstring StructArray from_arraysis_nullrrS)r#rr bytes_array path_arrayrs ` @r% embed_storagezNifti.embed_storages  $ "     h     **,,       X p pgmm\bNcNcNmNmNoNo p p p   .,,k:-FRXHY`k`s`s`u`u,vv'4<000r&rcNddlm}|jr|n|d|ddS)zfIf in the decodable state, return the feature itself, otherwise flatten the feature into a dictionary.r)ValuerrrP)featuresrrL)r#rs r%flattenz Nifti.flattensK######{ DDxh r&ctj|jrrtjdgt |ztj}tj||gddg| }ntj |jrrtjdgt |ztj }tj||gddg| }n5tj |jr|j ddkr|d}n8tjdgt |ztj}|j ddkr|d}n8tjdgt |ztj }tj||gddg| }t||jS)aCast an Arrow array to the Nifti arrow storage type. The Arrow types that can be converted to the Nifti pyarrow storage type are: - `pa.string()` - it must contain the "path" data - `pa.binary()` - it must contain the NIfTI bytes - `pa.struct({"bytes": pa.binary()})` - `pa.struct({"path": pa.string()})` - `pa.struct({"bytes": pa.binary(), "path": pa.string()})` - order doesn't matter Args: storage (`Union[pa.StringArray, pa.StructArray, pa.BinaryArray]`): PyArrow array to cast. Returns: `pa.StructArray`: Array in the Nifti arrow storage type, that is `pa.struct({"bytes": pa.binary(), "path": pa.string()})`. NrrQrRrr)rtypes is_stringrmrlenrrrr is_binaryr is_structget_field_indexrrrS)r#rrrs r% cast_storagezNifti.cast_storage s$ 8  gl + + w(D6CLL#8ry{{KKKKn00+w1G'SYIZahapaparar0ssGG X   - - w4&3w<<"7bikkJJJJn00':1FRXHY`g`o`o`q`q0rrGG X   - - w|++G4499%mmG44  hvG '<29;;OOO |++F33q88$]]622 Xtfs7||&;")++NNN n00+z1JWV\L]dkdsdsdudu0vvG'4<000r&rW)"rBrCrDrErLbool__annotations__rrOr rbrrrstructrrrSrrUrXr rQrdrhrnrrrr r StringArray BinaryArrayrrr&r%rKrK?s  DFDd777B 7778E8C=777&RYibikk'R'RSSGXc]SSSwU???E3???&E#uiGX*X$Y&^b&&&&P=)=)D=)EY=)=)=)=)~&1&1R^&1PRP^&1&1&1&1P  }d3 3E.FFG     "1E".".".*X$Y"1^`^l"1"1"1"1"1"1r&rKFimgrZ force_bytesr[ct|dr |j|s|jdj}|ddS|}d|dS)a Encode a nibabel image object into a dictionary. If the image has an associated file path, returns the path. Otherwise, serializes the image content into bytes. Args: img: A nibabel image object (e.g., Nifti1Image). force_bytes: If `True`, always serialize to bytes even if a file path exists. Needed to upload bytes properly. Returns: dict: A dictionary with "path" or "bytes" field. rNr-r])r;rr.r3)rrr. bytes_datas r%rgrg/sZsJ1CL$<[$<<(1 4000J: . ..r&)F))rj dataclassesrrpathlibrtypingrrrr r r pyarrowrr download.download_configrtablerutils.file_utilsrrutils.py_utilsrrr`rFrrr_rGrHrrKrrhrbrQrgrr&r%rs ((((((((FFFFFFFFFFFFFFFF55555533333333CCCCCCCC&%%%%%% (%%%%%SZ3%%%P l1l1l1l1l1l1l1 l1^////d/tTWYabghkmrhrbsYtTtOu//////r&