U f{jA@sddlZddlmZddlmZddlmZmZddlZ ddl m Z ddl m Z ddlmZddlmZd d lmZd d lmZmZmZmZmZedd dZeGddde ZddZddZddZdS)N) OrderedDict)datetimetimezone)BaseRaw)FIFF) create_info) _mult_cal_one) Annotations)loggerverbosefill_docwarn _check_fnameFcCs t|||S)aReader for a Persyst (.lay/.dat) recording. Parameters ---------- fname : str Path to the Persyst header (.lay) file. %(preload)s %(verbose)s Returns ------- raw : instance of RawPersyst A Raw object containing Persyst data. See Also -------- mne.io.Raw : Documentation of attribute and methods. Notes ----- It is assumed that the ``.lay`` and ``.dat`` file are in the same directory. To get the correct file path to the ``.dat`` file, ``read_raw_persyst`` will get the corresponding dat filename from the lay file, and look for that file inside the same directory as the lay file. ) RawPersyst)fnamepreloadr rU/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/io/persyst/persyst.pyread_raw_persystsrcs.eZdZdZedfdd ZddZZS) rzRaw object from a Persyst file. Parameters ---------- fname : str Path to the Persyst header (.lay) file. %(preload)s %(verbose)s See Also -------- mne.io.Raw : Documentation of attribute and methods. FNc- s*t|ddd}td||ds.|d}t|t|}}t|s^td|dt |\}}}t } t } t } t } d} t |||D]\}}}|d krq|dk r|d kr| }|d kr|d krt|}t |t|}d |d}t|st||| |<q|dkr.|| |<q|dkrB|| |<q|dkr| |t|g| |<| d7} q| d}t| d}t| d}t| }| d}|dk rHd|krt|d}n.d|krt|d}nd|krt|d}t|tstd|d}n6t| dd}t|j|j|j|j|j|jtjd }t| }|t |krjt!d!d"d#|D}| d$}d%}t"|||d&}|j#|d'|$2t%|D]}|d(|d)|d*<q||d+<W5QRXt&|d,} t|d-krt'(d.}!n(t|dkr t'(d/}!nt!d0|| )dt*j+| ,}"|"|!j-|}"t.d1|"d2|d3W5QRX|!||"d4}#t/t0|j1|||g|"dg|#g|d5t'2| t}$t'2| t}%d g| }&d}'| 3D]:\}(})|)D]*\}*}+|*|$|'<|+|%|'<|(|&|'<|'d7}'qqt4|$|%|&},|5|,dS)6NreadTrz Loading %sz.layzThe path you specified, "z",does not exist.r) channelmapcommentsZfileinfofilez=The data path you specified does not exist for the lay path, zj. Make sure the dat file is in the same directory as the lay file, and the specified dat filename matches.rZpatientrdatatypeZ calibrationZ waveformcounttestdate/z%m/%d/%Y-z%d-%m-%Y.z%Y.%m.%dz[Cannot read in the measurement date due to incompatible format. Please set manually for %s testtimez%H:%M:%S)yearmonthdayhourminutesecondtzinfozJChannels in lay file do not match the number of channels in the .dat file.cSsg|]}|ddqS)z-REFr)uppersplit).0chrrr sz'RawPersyst.__init__..Z samplingrateZeeg)ch_types) subject_infogư>Zchscal meas_daterbi4i2zUnknown format: zLoaded z samples for z channels.)dtypen_chs n_samples) filenamesZ last_samps raw_extrasr )6rr infoendswithopdirnamebasenameexistsFileNotFoundError_read_lay_contentsrziplowerjoingetlistfloatint_get_subjectinforstrptime isinstancerr#r$r%r&r'r(rutckeyslen RuntimeErrorrupdateZ_unlockrangeopennpr7seekosSEEK_ENDtellitemsizedebugsuperr__init__Zzerositemsr Zset_annotations)-selfrrr Z curr_pathZ lay_fnamerOdatasectionsZ fileinfo_dictZchannelmap_dict patient_dictZ comments_dictZ num_commentskeyvalsection dat_fnameZ dat_fpath error_msgrr1r8r0rr2r"Zch_namesZsfreqr/r<idxfr7r9r;Zonsetduration descriptionZt_idxZ _descriptionZ event_tuplesZ_onsetZ _durationZannot __class__rrr]Bs                              "      zRawPersyst.__init__c Cs|j|d}|j|d} |j|} ||} | | } t| d,} | | |j|dtj| || d}W5QRXt|| dfdtj }t |||||dS) zRead a segment of data from a file. The Persyst software records raw data in either 16 or 32 bit binary files. In addition, it stores the calibration to convert data to uV in the lay file. r7r8r3r)r7countFN) Z _raw_extrasZ _filenamesrTrVrZrUfromfileZreshapeZastypeZfloat32r )r_r`rhfistartstopZcalsZmultr7r8rfZtime_length_sampsrnZ dat_file_IDrecordrrr_read_segment_files  zRawPersyst._read_segment_file)FN)__name__ __module__ __qualname____doc__r r]rv __classcell__rrrlrr2s 5rc CsL|d}d|krJzt|d}Wqtk rFd}td|YqXn>d|krzt|d}Wn$tk rd}td|YnX|d|d|d |d |d |d |d }ttjtjtjtjd}||d tj |d <ttj tj tj tj tj tj tj d}z||d |d <Wn t k rF|d YnX|S)N birthdaterz%m/%d/%yz"Unable to process birthdate of %s r z%d-%m-%yfirstmiddlelastsexhandid)Z first_nameZ middle_name last_namerrZhis_idbirthday)mZmaleriZfemale)rrightlleftaZ ambidextrousZambi)rGrrL ValueErrorprintdictrZFIFFV_SUBJ_SEX_MALEZFIFFV_SUBJ_SEX_FEMALEZFIFFV_SUBJ_SEX_UNKNOWNZFIFFV_SUBJ_HAND_RIGHTZFIFFV_SUBJ_HAND_LEFTZFIFFV_SUBJ_HAND_AMBIKeyErrorpop)rbr|r0Zsex_dictZ hand_dictrrrrKsX    rKc Cs|g}gg}}d}t|dP}|D]D}t||\}}} |dkrH| }q"|||| ||q"W5QRX|||fS)z&Lay file are laid out like a INI file.rrr)rT_process_lay_linerEappend) rrarOr`reZfinlinestatusrcrdrrrrCRs    rCc Csd}|}t|d}|s2d}d}d}|||fS|ddkrl||dkrl|ddkrld}|d|}n|dkr|dd \}}}} } d }| }||f}nRd |krtd ||d } d }|d| }||| d|d}||||fS) aProcess a line read from the Lay (INI) file. Each line in the .lay file will be processed into a structured ``status``, ``key`` and ``value``. Parameters ---------- line : str The actual line in the Lay file. section : str The section in the Lay file. Returns ------- status : int Returns the following integers based on status. -1 => unknown string found 0 => empty line found 1 => section found 2 => key-value pair found key : str The string before the ``'='`` character. If section is "Comments", then returns the text comment description. value : str The string from the line after the ``'='`` character. If section is "Comments", then returns the onset and duration as a tuple. Notes ----- The lay file comprises of multiple "sections" that are documented with bracket ``[]`` characters. For example, ``[FileInfo]`` and the lines afterward indicate metadata about the data file itself. Within each section, there are multiple lines in the format of ``=``. For ``FileInfo``, ``Patient`` and ``ChannelMap`` each line will be denoted with a ``key`` and a ``value`` that can be represented as a dictionary. The keys describe what sort of data that line holds, while the values contain the corresponding value. In some cases, the ``value``. For ``SampleTimes``, the ``key`` and ``value`` pair indicate the start and end time in seconds of the original data file. For ``Comments`` section, this denotes an area where users through Persyst actually annotate data in time. These are instead represented as 5 data points that are ``,`` delimited. These data points are ordered as: 1. time (in seconds) of the annotation 2. duration (in seconds) of the annotation 3. state (unused) 4. variable type (unused) 5. free-form text describing the annotation rrr[]r r,r=zJThe line %s does not conform to the standards. Please check the .lay file.)striprPrEr+rQindex) rrercZend_idxrvalueZtime_secrjstateZvar_typetextposrrrrns:8      r)FN)rWos.pathpathr> collectionsrrrnumpyrUbaser constantsrZ meas_inforutilsr annotationsr r r rrrrrrKrCrrrrrs$        h8