#j7 dddlZddlmZmZddlmZddlmZmZm Z m Z m Z m Z ddl ZddlmZddlmZddlmZdd lmZmZmZdd lmZmZer ddlZd d lmZej rddlZGd dej!j"Z#eGddZ$dddde%de&e'e e e'e(fffdZ)dS)N) dataclassfield)Path) TYPE_CHECKINGAnyClassVarDictOptionalUnion)config)DownloadConfig) array_cast) is_local_path is_remote_urlxopen)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 a/lsinfo/ai/hellotax_ai/base_platform/venv/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_nvr0volumes 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"rB __classcell__)r%s@r&rrsb   + (> + + + + + +       r'rc eZdZUdZdZeed<eddZe e ed<dZ e e ed <e je je jd Ze eed <eddd Ze ed <dZdee eeedfdefdZddeddfdZ dde jdedede 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.Nifti1Imager bytespathpa_type)rNinitrO_typec|jSN)rT)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 rNrSrRrSrQrRzTA 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 isinstancestrrabsoluterR bytearray spatialimages SpatialImageencode_nibabel_imagedictgetosrSisfile ValueErrortype)r$rZrGs 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'.rSrRzBA nifti should have one of 'path' or 'bytes' but both are None in r_zgzip://::repo_idtokenrbdownload_configr s)r r`ra ImportErrorrm startswithrsplitload HF_ENDPOINTHUB_DATASETS_URLHUB_DATASETS_HFFS_URLrrjrrgzip decompressrI from_bytesr)r$rZtoken_per_repo_idrGrSr?nifti source_urlpatternsource_url_fieldsrurxfrs 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 storage local_files remote_filescitfdtjfd|Dtj}tjfd|dDtj}tj||gddg| }t||j S) a7Embed NifTI files into the Arrow array. Args: storage (`pa.StructArray`): PyArrow array to embed. token_per_repo_id (`dict`, optional): Dictionary repo_id -> token to fetch the files bytes. local_files (`bool`, defaults to `True`) Whether to embed local files data in the array remote_files (`bool`, defaults to `True`) Whether to embed remote files data in the array. E.g. files with paths that start with hf:// or https:// 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)Nrqrrrsrtrvrw) r{rzr r}r~rrrjrrread)rSrrrrurxrrs 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 cg|]U}|O|d?rt|dsr&t|dr|dn |dndVS)NrRrS)rr).0xrrrs r& z'Nifti.embed_storage..s   = z)%**7& *B*B*HT*YfghiogpYqYq*"M!F),,,7   r'rncg|]I}|Crt|sr.t|rtj|n|ndJSrX)rrrkrSbasename)rrSrrs r&rz'Nifti.embed_storage.. s   #$(5d(;(;AMR_`dReReBG$$T***   r'rSrRmask) rpaarray to_pylistbinaryrstring StructArray from_arraysis_nullrrT)r$rrrr bytes_array path_arrayrs ``` @r& embed_storagezNifti.embed_storagesI0  $ "     h      !**,,       X     $MM&11;;==       .,,k:-FRXHY`g`o`o`q`q,rr'4<000r'rcNddlm}|jr|n|d|ddS)zfIf in the decodable state, return the feature itself, otherwise flatten the feature into a dictionary.r)ValuerrrQ)featuresrrM)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()})`. NrrRrSrr)rtypes is_stringrnrlenrrrr is_binaryr is_structget_field_indexrrrT)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'rX)NTT)"rCrDrErFrMbool__annotations__rrPr rcr rrstructrrrTrrVrYr rRrerirorrrr r StringArray BinaryArrayrr'r&rLrL?s  DFDd777B 7778E8C=777&RYibikk'R'RSSGXc]SSSwU???E3???&E#uiGX*X$Y&^b&&&&P=)=)D=)EY=)=)=)=)@osC1C1~C1LPC1gkC1 C1C1C1C1J  }d3 3E.FFG     "1E".".".*X$Y"1^`^l"1"1"1"1"1"1r'rLFimgr[ 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/r4)rrr/ bytes_datas r&rhrhLsZsJ1CL$<[$<<(1 4000J: . ..r')F)*rk dataclassesrrpathlibrtypingrrrr r r pyarrowrr download.download_configrtablerutils.file_utilsrrrutils.py_utilsrrrarGrrr`rHrIrrLrrircrRrhrr'r&rs ((((((((FFFFFFFFFFFFFFFF555555BBBBBBBBBBCCCCCCCC&%%%%%% (%%%%%SZ3%%%P I1I1I1I1I1I1I1 I1X////d/tTWYabghkmrhrbsYtTtOu//////r'