o h,i@sxdgZddlZddlZddlZddlZddlZddddddd d d d d d d d ddZdddddddddddddddZiZ dd Z d!d"Z d#d$Z d%d&Z d'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6Zd7d8ZGd9d:d:ZGd;d<dZd?d@ZdAdBZdCdDZdEdFZdGdHZdIdJZdKdLZ dMdNZ!dOdPZ"dQdRZ#GdSdTdTe$Z% U UdWdVdZ&dS)XreadsavNz>u1z>i2z>i4z>f4z>f8z>c8z|Oz>c16z>u2z>u4z>i8z>u8)  START_MARKERCOMMON_VARIABLEVARIABLESYSTEM_VARIABLE END_MARKER TIMESTAMPCOMPILEDIDENTIFICATIONVERSION HEAP_HEADER HEAP_DATA PROMOTE64NOTICE DESCRIPTION)rrrrrr rrrrcCs.|}|ddkr||d|ddS)z+Align to the next 32-bit position in a filerrN)tellseek)fposr(H/var/www/html/scripts/venv/lib/python3.10/site-packages/scipy/io/_idl.py _align_32Ks r*cCs||dS)zSkip `n` bytesNreadr&nr(r(r) _skip_bytesTs r/cCs ||S)zRead the next `n` bytesr+r-r(r(r) _read_bytesZs r0cCs$ttd|ddddS)zRead a single bytez>BrNrr)npuint8structunpackr,r&r(r(r) _read_byte_$r6cCttd|ddS)Read a signed 32-bit integer>lrrr1int32r3r4r,r5r(r(r) _read_longdr=cC$ttd|ddddS)zRead a signed 16-bit integerz>hrrr)r1int16r3r4r,r5r(r(r) _read_int16ir7rAcCr8)r9z>irrr;r5r(r(r) _read_int32nr>rBcCr8)zRead a signed 64-bit integerz>qr r)r1int64r3r4r,r5r(r(r) _read_int64sr>rDcCr?)zRead an unsigned 16-bit integerz>Hrrr)r1uint16r3r4r,r5r(r(r) _read_uint16xr7rFcCr8)zRead an unsigned 32-bit integer>Irr)r1uint32r3r4r,r5r(r(r) _read_uint32}r>rIcCr8)zRead an unsigned 64-bit integerz>Qr r)r1uint64r3r4r,r5r(r(r) _read_uint64r>rKcCr8)zRead a 32-bit floatz>frr)r1float32r3r4r,r5r(r(r) _read_float32r>rMcCr8)zRead a 64-bit floatz>dr r)r1float64r3r4r,r5r(r(r) _read_float64r>rOc@seZdZdZddZdS)PointerzClass used to define pointerscCs ||_dSN)index)selfrRr(r(r)__init__szPointer.__init__N)__name__ __module__ __qualname____doc__rTr(r(r(r)rPs rPc@seZdZdZdS) ObjectPointerz$Class used to define object pointersN)rUrVrWrXr(r(r(r)rYsrYcCs4t|}|dkrt||d}t||Sd}|S)z Read a stringrlatin1)r=r0decoder*)r&lengthcharsr(r(r) _read_stringsr_cCs6t|}|dkrt|}t||}t||Sd}|S)z.Read a data string (length is specified twice)rr[)r=r0r*)r&r] string_datar(r(r)_read_string_datas racCsL|dkrt|dkrtdt|S|dkrt|S|dkr"t|S|dkr*t|S|dkr2t|S|dkrGt|}t|}t||dS|d krOt|S|d krWtd |d krlt|}t|}t ||dS|d krvt t|S|dkrt t|S|dkrt |S|dkrt |S|dkrt|S|dkrt|Std|)z*Read a variable with a specified data typerz*Error occurred while reading byte variablerrrrry?r r z'Should not be here - please report thisr r r rrrrz)Unknown IDL type: %i - please report this)rB Exceptionr6rArMrOr1 complex64ra complex128rPrYrFrIrDrK)r&dtyperealimagr(r(r) _read_datasJ    rhc Csz|d}|d}g}|D]<}|ds|dr'||d|dftjfq |dtvrA||d|dft|dfq td|dtj|f|d}t|D]J}|D]E}|d}|dr|t||d |d|d |d||d|<qY|drt |||d |d||d|<qYt ||||d|<qYqU|d d kr|d dt |d } | | | }|S)z Read a structure, with the array and structure descriptors given as `array_desc` and `structure_desc` respectively. nelementstagtable structurearraynametypecodez Variable type %i not implementedrearrtable structtablendimsrdimsN)appendlowerr1object_ DTYPE_DICTrbrecarrayrange_read_structure _read_arrayrhintreversereshape) r& array_desc struct_descnrowscolumnsrecolrkirsr(r(r)rzsB"     rzc Cs|dvr&|dkrt|}||dkrtdtj||dt|d}n9|dvr@tj||ddt|dddd}ng}t|d D]}|}t||}| |qHtj |tj d}|d dkrz|d dt |d }| ||}t||S) z` Read an array of type `typecode`, with the array descriptor given as `array_desc`. ) rrrrrr rrrrnbytesz.Not able to verify number of bytes from headerro)rrrNrirrrs)rBwarningswarnr1 frombufferr,rwryrhrtrlrvr|r}r~r*) r&rnrrrlrredatarsr(r(r)r{ s4      r{cs dti}t}|td7}td|dtvr%td|dt|d|d<|ddvr|ddkr@t|d<n t|d<tdt}|d d kre|krad |d <ntd t}|dkrqtd|drt |d|d|d <n|drt |d |d|d <n|d }t ||d <n|ddkrtdt|d<t|d<t|d<n|ddkrt|d<t|d<t|d<t|d<n|ddkrt|d<t|d <t|d!<n|dd"krt|d#<n||dd$krt |d%<nn|dd&kr.t|d'<fd(d)t |d'D|d*<nQ|dd+krQt|d,<t|d-<fd.d)t |d,D|d/<n.|dd0kr]d1|d2<n"|dd3krjtd4n|dd5krwtd6ntd7|d||S)8z!Function to read in a full recordrectyperzUnknown RECTYPE: %i)rrrvarname heap_indexrnrNrzUnexpected type code: 0r zVARSTART is not 7rkrrrlridateuserhostrformatarchosrelease IDENTIFICATONauthortitleidcodernoticer descriptionrnvaluescg|]}tqSr(r=.0_r5r(r) z _read_record..indices COMMONBLOCKnvarsrmcrr(r_rr5r(r)rrvarnamesrTendUNKNOWNzSkipping UNKNOWN recordrzSkipping SYSTEM_VARIABLE recordz$record['rectype']=%s not implemented)r=rIr/ RECTYPE_DICTrbr__read_typedescr$ ValueErrorrzr{rhraryrrr%)r&recordnextrec rectypedescvarstartrer(r5r) _read_record;s                                  rcCst|t|d}|dd@dkrtd|dd@dk|d<|dd@dk|d<|dr;t||d <t||d <|S|drEt||d <|S) z%Function to read in a type descriptor)rnvarflagsrrz System variables not implementedrrl rkrr)r=rb_read_arraydesc_read_structdesc)r&typedescr(r(r)rs   rcs,dti}|ddkr?tdt|d<t|d<t|d<tdt|d<fdd t|dD|d <|S|dd krtd tdt|d<t|d<t|d<tdd|d<g|d <t|dD]}t}|d krtd|d tqt|Std|d)z'Function to read in an array descriptorarrstartr rrrirrnmaxcrr(rrr5r(r)rrz#_read_arraydesc..rsz$Using experimental 64-bit array readrzExpected a zero in ARRAY_DESCzUnknown ARRSTART: %i)r=r/ryrrrKrbrt)r& arraydescdvr(r5r)rs4              rcsxi}t}|dkrtdt|d<t}t|d<t|d<|d@|d<|d@|d <|d @|d <|dsfd d t|dD|d<|dD]}t|d<qMfdd|dD|d<fdd|dD|d<|d sx|d rt|d<t|d<fdd t|dD|d<fdd t|dD|d<|t|d<|S|dtvrtdt|d}|S)z*Function to read in a structure descriptorr zSTRUCTSTART should be 9rmntagsrrpredefrinheritsris_supercrr() _read_tagdescrr5r(r)rs z$_read_structdesc..rjc"i|] }|dr|dtqS)rlrm)rrtagr5r(r) z$_read_structdesc..rpcr)rkrmrrr5r(r)r rrq classname nsupclassescrr(rrr5r(r)r supclassnamescrr(rrr5r(r)rr supclasstablez"PREDEF=1 but can't find definition)r=rbr_ry STRUCT_DICT)r& structdesc structstartrrr(r5r)rsL                     rcCsldt|i}|ddkrt||d<t||d<t|}|d@dk|d<|d@dk|d<|dtv|d<|S) z$Function to read in a tag descriptoroffsetrnrrlrrkscalar)r=rKrw)r&tagdesctagflagsr(r(r)r!s    rcCsVt|tr8t|tr)|jdkrd}n|j|vr||j}ntdd}t|ts t||\}}|r4|}d|fSt|tjjj rZt |D]\}}t||\}}|rU|||<qDd|fSt|tjjj r|t |D]\}}t||\}}|rw|||<qfd|fSt|tj r|j jtjurt|jD]}t|||\}}|r|||qd|fSd|fS)NrzNVariable referenced by pointer not found in heap: variable will be set to NoneTF) isinstancerPrRrr _replace_heapr1corerecordsrx enumeraterndarrayretypervrysizeitemitemset)variableheapreplacenewirrivvaluer(r(r)r4sH          rcsHeZdZdZifddZfddZfddZdd ZeZeZ Z S) AttrDicta. A case-insensitive dictionary with access via item, attribute, and call notations: >>> d = AttrDict() >>> d['Variable'] = 123 >>> d['Variable'] 123 >>> d.Variable 123 >>> d.variable 123 >>> d('VARIABLE') 123 >>> d['missing'] Traceback (most recent error last): ... KeyError: 'missing' >>> d.missing Traceback (most recent error last): ... AttributeError: 'AttrDict' object has no attribute 'missing' cCst||dSrQ)dictrT)rSinitr(r(r)rTzAttrDict.__init__cst|SrQ)super __getitem__rurSrm __class__r(r)rrzAttrDict.__getitem__cst||SrQ)r __setitem__ru)rSkeyrrr(r)rszAttrDict.__setitem__c Cs8z||WStytdt|d|ddw)N'z' object has no attribute ')rKeyErrorAttributeErrorrrr(r(r) __getattr__s  zAttrDict.__getattr__) rUrVrWrXrTrrr __setattr____call__ __classcell__r(r(rr)rvs    rFc Csg}|s|r i}nt}t|d}t|d}|dkr td|t|d} | dkr*n| dkr|r4td|rq,td|rT|D] }||||<qH|S|S)?a Read an IDL .sav file. Parameters ---------- file_name : str Name of the IDL save file. idict : dict, optional Dictionary in which to insert .sav file variables. python_dict : bool, optional By default, the object return is not a Python dictionary, but a case-insensitive dictionary with item, attribute, and call access to variables. To get a standard Python dictionary, set this option to True. uncompressed_file_name : str, optional This option only has an effect for .sav files written with the /compress option. If a file name is specified, compressed .sav files are uncompressed to this file. Otherwise, readsav will use the `tempfile` module to determine a temporary filename automatically, and will remove the temporary file upon successfully reading it in. verbose : bool, optional Whether to print out information about the save file, including the records read, and available variables. Returns ------- idl_dict : AttrDict or dict If `python_dict` is set to False (default), this function returns a case-insensitive dictionary with item, attribute, and call access to variables. If `python_dict` is set to True, this function returns a Python dictionary with all variable names in lowercase. If `idict` was specified, then variables are written to the dictionary specified, and the updated dictionary is returned. Examples -------- >>> from os.path import dirname, join as pjoin >>> import scipy.io as sio >>> from scipy.io import readsav Get the filename for an example .sav file from the tests/data directory. >>> data_dir = pjoin(dirname(sio.__file__), 'tests', 'data') >>> sav_fname = pjoin(data_dir, 'array_float32_1d.sav') Load the .sav file contents. >>> sav_data = readsav(sav_fname) Get keys of the .sav file contents. >>> print(sav_data.keys()) dict_keys(['array1d']) Access a content with a key. >>> print(sav_data['array1d']) [0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0.] rbrsSRzInvalid SIGNATURE: %ssszIDL Save file is compressedzw+bz.sav)suffixz -> expanding to %ssSRTr:rrrrGrzInvalid RECFMT: %srrrrrrrrz2--------------------------------------------------zDate: %srzUser: %srzHost: %srrz Format: %srzArchitecture: %srzOperating System: %srzIDL Version: %srrz Author: %srz Title: %srz ID Code: %srrzDescription: %srz&Successfully read %i records of which:cSsg|]}|dqS)rr()rrr(r(r)rrzreadsav..z - %i are of type %szAvailable variables:z - z [])ropenr0rbprinttempfileNamedTemporaryFilermwriter=r3packr|rIr,rr1rCr$zlib decompresslencloser%rrtrrusetcountr) file_nameidict python_dictuncompressed_file_nameverboser variablesr& signaturerecfmtfoutrrunknownmodvalr' rec_stringrrrrrrectypesrtvarr(r(r)rsF          " "  %         ")NFNF)'__all__r3numpyr1rrrrwrrr*r/r0r6r=rArBrDrFrIrKrMrOrPrYr_rarhrzr{rrrrrrrrrr(r(r(r)s   *-/m/6B-