U f{jI@sddlZddlmZddlZddlmZddlZddl m Z m Z m Z m Z mZmZddlmZmZddlmZddlmZdd lmZmZdd lmZmZmZmZmZm Z m!Z!m"Z"dd l#m$Z$ej%ej&ej'ej(d Z)e*e+e),Z-d dddddZ.dde)/DZ0dddddZ1dAddZ2ddZ3dd Z4d!d"d#d$hZ5Gd%d&d&e6Z7d'd(Z8edBddd)d*d+Z9ej:d,ej;d-ejdDd2d3Z?d4d5Z@d6d7ZAdEd9d:ZBdFd;d<ZCd=d>ZDdGd?d@ZEdS)HN)Counter)loggerwarnBunch_validate_type _check_fnameverbose)FIFF_coord_frame_named) dir_tree_find)read_tag)start_and_end_filewrite_dig_points) apply_trans Transformget_ras_to_neuromag_transcombine_transformsinvert_transform _to_const _str_to_frame_coord_frame_name) __version__)cardinalhpieegextraZCardinalZHPIZEEGZExtraUnknown)rrrrunknowncCsi|]\}}||qSr ).0keyvalr r S/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/io/_digitization.py +sr%ZLPAZNasionZRPAZInion)r rFc CsX|r<|dk rZsz&_format_dig_points..) enumerater FIFFV_POINT_CARDINALheapqheappushFIFFV_POINT_HPIFIFFV_POINT_EEGFIFFV_POINT_EXTRAZFIFFV_POINT_HEADsortappend) digZ enforce_orderZfids_digpointsZ hpi_digpointsZ eeg_digpointsZextra_digpointsZhead_digpointsidxZdigpointr(r)Znew_digr-r r r$_format_dig_points/sB     r:cCsdd|DS)NcSsg|]}|dtjkr|qSr))r r4r,r r r$r.^sz _get_dig_eeg..r )r8r r r$ _get_dig_eeg]sr<cCs:tdd|D}t|tj|tj|tj|tjdS)z&Get the number of points of each type.cSsg|] }|dqSr;r r,r r r$r.csz)_count_points_by_type..)fidrrr)rdictr r0r3r4r5)r8Z occurrencesr r r$_count_points_by_typeasr?r)r(r coord_framec@s(eZdZdZddZddZddZdS) r+axContainer for a digitization point. This is a simple subclass of the standard dict type designed to provide a readable string representation. Parameters ---------- kind : int The kind of channel, e.g. ``FIFFV_POINT_EEG``, ``FIFFV_POINT_CARDINAL``. r : array, shape (3,) 3D position in m. and coord_frame. ident : int Number specifying the identity of the point. e.g. ``FIFFV_POINT_NASION`` if kind is ``FIFFV_POINT_CARDINAL``, or 42 if kind is ``FIFFV_POINT_EEG``. coord_frame : int The coordinate frame used, e.g. ``FIFFV_COORD_HEAD``. cCs|dtjkr t|dd}n$tt|dd}d||df}|d}t|d}d|krzd t|d  d }nd td |d  d }d|||fS)Nr)r(zUnknown cardinalrz%s #%s rAZvoxelz(%0.1f, %0.1f, %0.1f)r@z(%0.1f, %0.1f, %0.1f) mmiz) r r0_cardinal_kind_revget_dig_kind_proper _dig_kind_revrjustrtupleljust)selfZid_cfposr r r$__repr__s   zDigPoint.__repr__cCs$t|d|d|d|ddS)zMake a deepcopy.r)r@r(rA)r)r@r(rA)r+copy)rKZmemodictr r r$ __deepcopy__s  zDigPoint.__deepcopy__csZdddg}ttkr&dStfdd|DrBdStddSdS) zCompare two DigPoints. Two digpoints are equal if they are the same kind, share the same coordinate frame and position. r)r(rAFc3s|]}||kVqdSNr )r!_otherrKr r$ sz"DigPoint.__eq__..r@N)setkeysanynpZallclose)rKrTZmy_keysr rSr$__eq__s  zDigPoint.__eq__N)__name__ __module__ __qualname____doc__rNrPrZr r r r$r+osr+c Cst|tj}d}t|dkr(tdnt|dkr>tdn|d}tj}g}t|dD]f}|d|j }|d|j }|tj krt ||}| |jq\|tjkr\t ||}tt|j}q\|D] } || d<qt|S) z%Read digitizer data from a FIFF file.NrzIsotrak not foundr zMultiple Isotrak foundZnent directoryrA)r r Z FIFFB_ISOTRAKlenrinforZFIFFV_COORD_HEADranger)rMZFIFF_DIG_POINTrr7dataZFIFF_MNE_COORD_FRAMEr rEintr:) r=Z meas_infoZisotrakr8rAkr)rMtagr-r r r$ _read_dig_fifs*          rg) overwriter c st||d}dk r\tfdd|D}|h}t|dkr\tdtt|ft|}t||ddW5QRXdS) a=Write digitization data to a FIF file. Parameters ---------- fname : path-like Destination file name. pts : iterator of dict Iterator through digitizer points. Each point is a dictionary with the keys 'kind', 'ident' and 'r'. coord_frame : int | str | None If all the points have the same coordinate frame, specify the type here. Can be None (default) if the points could have varying coordinate frames. %(overwrite)s .. versionadded:: 1.0 %(verbose)s .. versionadded:: 1.0 )rhNcsh|]}|dqSrA)rE)r!ptrir r$ szwrite_dig..rzNPoints have coord_frame entries that are incompatible with coord_frame=%i: %s.T)blockrA)rrr` ValueErrorstrrIrr)fnameZptsrArhr Z pts_framesZ bad_framesr=r rir$ write_digs    rpnasionlparpaTc Cstttt}}}tt}}|D]}|dtjkrP|d|t|d<q(|dtjkr|||d||dq(|dtjkr||dq(|dtjkr(|ddks|s(||dq(t dd|D}t |dkrt d|t |}d |_t|d d |d d |d d t |r8t |nd t |rNt |nd t |rdt |nd ||dS)aObtain coordinate data from a Dig. Parameters ---------- dig : list of dicts A container of DigPoints to be added to the info['dig']. Returns ------- ch_pos : dict The container of all relevant channel positions inside dig. r)r@r(rcSsg|] }|dqSrir r,r r r$r.sz._get_data_as_dict_from_dig..r z6Only single coordinate frame in dig is supported, got )r*r&rqNrrrs)rqrrrshsprelpdig_ch_pos_locationrA)listr>r r0_cardinal_ident_mappingr3r7r5r4rVr` RuntimeErrorrYarrayshaperrEpop) r8Zexclude_ref_channelrtrruZfidsrvr-Zdig_coord_framesr r r$_get_data_as_dict_from_digs8     r}cCstdddd}t}|D]6}|dtjkrt|d}|d||<|d||<qt|dkr|rt|tddd gkrtd |tt| d krtd t||r| d nd}||fS) N)rqrrrsr)r(r@rArrqrrrsz*Some fiducial points are missing (got %s).r zBAll fiducial points must be in the same coordinate system (got %s)) rr>r r0rxr`rVrWrmvaluespopitem)r8Z raise_errorZ fid_coordsZfid_coord_framesr-r"rAr r r$_get_fid_coords"s(  rc Cst|\}}t|}|jdks0|jddkrDd|jf}t||dkrt|dX}t}t j d}| d| | f| d t|tj||d d d W5QRXnd |}t|dS)a!Write points to text file. Parameters ---------- fname : str Path to the file to write. The kind of file to write is determined based on the extension: '.txt' for tab separated text file. dig_points : numpy.ndarray, shape (n_points, 3) Points. rr r&z-Points must be of shape (n_points, 3), not %sz.txtwbz%I:%M%p on %B %d, %Ys?%% Ascii 3D points file created by mne-python version %s at %s s %% %d 3D points, x y z per line   ) delimiternewlinez(Unrecognized extension: %r. Need '.txt'.N)opsplitextrYasarrayndimr{rmopenrdatetimenowstrftimewriteencoder`Zsavetxt) ro dig_pointsrRexterrr=versionrmsgr r r$_write_dig_points<s$   rcCs2t|tr|tkr*tdtt|ft|S)Nz%coord_frame must be one of %s, got %s) isinstancernrrmsortedrWrir r r$_coord_frame_const[s rheadc Cst|}g}|dk rPt|}|jdkr8td|jf||tjtj|d|dk rt|}|jdkr|td|jf||tj tj|d|dk rt|}|jdkrtd|jf||tj tj|d|dk rFt|}|j dks|jdd krtd |jft |D]$\}} || |dtj |dq |dk rt|}t|r|jdd krtd |jft |D]$\}} || |dtj|dq|dk rg} d } |D]\} } t| td z| t| ddWntk rd} YnXt| tjttfd tj| td} | || <| jdkrtd| jdq| r|tdt|d} t|| D](\} }||| t|tj|dqt|S)aConstruct digitizer info for the info. Parameters ---------- nasion : array-like | numpy.ndarray, shape (3,) | None Point designated as the nasion point. lpa : array-like | numpy.ndarray, shape (3,) | None Point designated as the left auricular point. rpa : array-like | numpy.ndarray, shape (3,) | None Point designated as the right auricular point. hpi : array-like | numpy.ndarray, shape (n_points, 3) | None Points designated as head position indicator points. extra_points : array-like | numpy.ndarray, shape (n_points, 3) Points designed as the headshape points. dig_ch_pos : dict Dict of EEG channel positions. coord_frame : str The coordinate frame of the points. Usually this is "unknown" for native digitizer space. Defaults to "head". Returns ------- dig : list of dicts A container of DigPoints to be added to the info['dig']. N)r&z,LPA should have the shape (3,) instead of %s)r@r(r)rAz/Nasion should have the shape (3,) instead of %sz,RPA should have the shape (3,) instead of %srr r&z5HPI should have the shape (n_points, 3) instead of %sz8Points should have the shape (n_points, 3) instead of %sF dig_ch_posT)Zdtypez>The position should be a 1D array of 3 floats. Provided shape .)rrYrr{rmr7r FIFFV_POINT_LPAr0FIFFV_POINT_NASIONFIFFV_POINT_RPArr/r3r`r5itemsrrnrdZndarrayrwrIrzfloatryZarangezipr4r:)rqrrrsr extra_pointsrrAr8r9pointZidentsZ use_aranger"valuer(r r r$_make_dig_pointsbs                       rc Cs|rTt|||}t||}t||}t||}|dk r@t||}t||tj}nd}tdd|d}t|||||d}||fS)Nctf_headr)Zfrototrans)rqrrrsrr)rrZastyperYZfloat32rr) rqrrrsrrconvertZneuromag_trans ctf_head_tinfo_digr r r$_call_make_dig_pointss"     rcCst|||||\}}|SrQ)r)ZnasrrrsrrrrRr r r$_artemis123_read_possrc Cs`|r|nd} t|||| ||\} } |rHtt||dd} t| | dd} ntdddd} | | | fS)NZmegrr)r)rrrr)rqrrrsrrrZuse_hpiZ bti_dev_tZ dev_ctf_tZ_hpirrtZ dev_head_tr r r$_make_bti_dig_pointss   r)F)N)T)T)NNNNNNr)T)FFFF)Fr1 collectionsrros.pathpathrnumpyrYutilsrrrrrr constantsr r treer rfrrrrZ transformsrrrrrrrrrr0r3r4r5Z_dig_kind_dictrIrr~Z_dig_kind_intsrFrrGrDr:r<r?Z _dig_keysr>r+rgrprrrrxr}rrrrrrrr r r r$ sp     (  . ;$ /  c