U md2i@stdgZddlZddlZddlZddlZddlZddddddd 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%dWdVdZ&dS)XreadsavNz>u1z>i2z>i4z>f4z>f8z>c8z|Oz>c16z>u2z>u4z>i8z>u8) Z START_MARKERZCOMMON_VARIABLEVARIABLESYSTEM_VARIABLE END_MARKER TIMESTAMPZCOMPILEDZIDENTIFICATIONVERSION HEAP_HEADER HEAP_DATAZ PROMOTE64NOTICE DESCRIPTION)rrrrrr rrrrcCs.|}|ddkr*||d|ddS)z+Align to the next 32-bit position in a filerrN)tellseek)fposr#F/home/sam/Atlas/atlas_env/lib/python3.8/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_bytesZsr+cCs$ttd|ddddS)zRead a single bytez>BrNrr)npZuint8structunpackr'r!r#r#r$ _read_byte_sr0cCsttd|ddS)Read a signed 32-bit integer>lrrr,Zint32r-r.r'r/r#r#r$ _read_longdsr4cCs$ttd|ddddS)zRead a signed 16-bit integerz>hrrr)r,Zint16r-r.r'r/r#r#r$ _read_int16isr5cCsttd|ddS)r1z>irrr3r/r#r#r$ _read_int32nsr6cCsttd|ddS)zRead a signed 64-bit integerz>qr r)r,int64r-r.r'r/r#r#r$ _read_int64ssr8cCs$ttd|ddddS)zRead an unsigned 16-bit integerz>Hrrr)r,Zuint16r-r.r'r/r#r#r$ _read_uint16xsr9cCsttd|ddS)zRead an unsigned 32-bit integer>Irr)r,Zuint32r-r.r'r/r#r#r$ _read_uint32}sr;cCsttd|ddS)zRead an unsigned 64-bit integerz>Qr r)r,Zuint64r-r.r'r/r#r#r$ _read_uint64sr<cCsttd|ddS)zRead a 32-bit floatz>frr)r,Zfloat32r-r.r'r/r#r#r$ _read_float32sr=cCsttd|ddS)zRead a 64-bit floatz>dr r)r,Zfloat64r-r.r'r/r#r#r$ _read_float64sr>c@seZdZdZddZdS)PointerzClass used to define pointerscCs ||_dSN)index)selfrAr#r#r$__init__szPointer.__init__N)__name__ __module__ __qualname____doc__rCr#r#r#r$r?sr?c@seZdZdZdS) ObjectPointerz$Class used to define object pointersN)rDrErFrGr#r#r#r$rHsrHcCs2t|}|dkr*t||d}t|nd}|S)z Read a stringrlatin1)r4r+decoder%)r!lengthcharsr#r#r$ _read_strings  rNcCs4t|}|dkr,t|}t||}t|nd}|S)z.Read a data string (length is specified twice)rrJ)r4r+r%)r!rLZ string_datar#r#r$_read_string_datas  rOcCs^|dkr$t|dkrtdt|S|dkr4t|S|dkrDt|S|dkrTt|S|dkrdt|S|dkrt|}t|}t||dS|d krt|S|d krtd n|d krt|}t|}t ||dS|d krt t|S|dkrt t|S|dkrt |S|dkr*t |S|dkrr,Z complex64rOZ complex128r?rHr9r;r8r<)r!dtyperealimagr#r#r$ _read_datasJ          rTc Cs|d}|d}g}|D]z}|ds,|drN||d|dftjfq|dtkr||d|dft|dfqtd|dqtj|f|d}t|D]}|D]}|d}|drt||d |d|d |d||d|<q|dr*t |||d |d||d|<qt ||||d|<qq|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 implementedrQarrtable structtablendimsrdimsN)appendlowerr,object_ DTYPE_DICTrPrecarrayrange_read_structure _read_arrayrTintreversereshape) r! array_desc struct_descZnrowscolumnsrQcolrWir_r#r#r$rfs@"     rfc Cs|dkrL|dkr.t|}||dkr.tdtj||dt|d}nr|dkrtj||ddt|dddd}n>g}t|d D]}|}t||}| |qtj |tj d}|d dkr|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 headerr[)rrrNrUr^r_)r6warningswarnr,Z frombufferr'rcrerTr`rXrbrhrirjr%) r!rZrkrprXrorQdatar_r#r#r$rg s8      rgcs dti}t}|td7}td|dtkrJtd|dt|d|d<|ddkrF|ddkrt|d<nt|d<tdt}|d d kr|krd |d <ntd nvt}|dkrtd|drt |d|d|d <n:|dr,t |d |d|d <n|d }t ||d <n|ddkrtdt|d<t|d<t|d<n|ddkrt|d<t|d<t|d<t|d<nJ|ddkrt|d<t|d <t|d!<n|dd"krt|d#<n|dd$kr6t |d%<n|dd&krpt|d'<fd(d)t |d'D|d*<n|dd+krt|d,<t|d-<fd.d)t |d,D|d/<n\|dd0krd1|d2<nD|dd3krtd4n*|dd5krtd6ntd7|d||S)8z!Function to read in a full recordrectyperzUnknown RECTYPE: %i)rrrvarname heap_indexrZrNrszUnexpected type code: 0r zVARSTART is not 7rWrkrlrXridateuserhostrformatarchosrelease IDENTIFICATONauthortitleidcodernoticer descriptionrZnvaluescsg|] }tqSr#r4.0_r/r#r$ sz _read_record..indicesZ COMMONBLOCKZnvarsrYcsg|] }tqSr#rNrr/r#r$rsvarnamesrTendUNKNOWNzSkipping UNKNOWN recordrzSkipping SYSTEM_VARIABLE recordz$record['rectype']=%s not implemented)r4r;r* RECTYPE_DICTrPrN_read_typedescr ValueErrorrfrgrTrOrerqrrr )r!recordnextrecZ rectypedescZvarstartrQr#r/r$ _read_record;s                                 rcCst|t|d}|dd@dkr*td|dd@dk|d<|dd@dk|d<|drtt||d <t||d <n|drt||d <|S) z%Function to read in a type descriptor)rZvarflagsrrz System variables not implementedrrX rWrkrl)r4rP_read_arraydesc_read_structdesc)r!Ztypedescr#r#r$rs  rcs0dti}|ddkr|tdt|d<t|d<t|d<tdt|d<fdd t|dD|d <n|dd krtd tdt|d<t|d<t|d<tdd|d<g|d <t|dD]0}t}|d krtd|d tqntd|d|S)z'Function to read in an array descriptorZarrstartr rrprUr^Znmaxcsg|] }tqSr#rrr/r#r$rsz#_read_arraydesc..r_z$Using experimental 64-bit array readrzExpected a zero in ARRAY_DESCzUnknown ARRSTART: %i)r4r*rerqrrr<rPr`)r!Z arraydescdvr#r/r$rs2                rcs|i}t}|dkrtdt|d<t}t|d<t|d<|d@|d<|d@|d <|d @|d <|dsVfd d t|dD|d<|dD]}t|d<qfdd|dD|d<fdd|dD|d<|d s|d rHt|d<t|d<fdd t|dD|d<fdd t|dD|d<|t|d<n"|dtkrltdt|d}|S)z*Function to read in a structure descriptorr zSTRUCTSTART should be 9rYZntagsrprpredefrZinheritsrZis_supercsg|] }tqSr#) _read_tagdescrr/r#r$rsz$_read_structdesc..rVcs"i|]}|dr|dtqS)rXrY)rrtagr/r#r$ sz$_read_structdesc..r\cs"i|]}|dr|dtqS)rWrYrrr/r#r$r sr] classnameZ nsupclassescsg|] }tqSr#rrr/r#r$rsZ supclassnamescsg|] }tqSr#rrr/r#r$rsZ supclasstablez"PREDEF=1 but can't find definition)r4rPrNre STRUCT_DICT)r!Z structdescZ structstartrrr#r/r$rsJ                     rcCsldt|i}|ddkr$t||d<t||d<t|}|d@dk|d<|d@dk|d<|dtk|d<|S) z$Function to read in a tag descriptoroffsetrZrrXrrWZscalar)r4r<rc)r!ZtagdescZtagflagsr#r#r$r!s    rcCsZt|trht|trJ|jdkr$d}q |j|kr:||j}q tdd}q t||\}}|r`|}d|fSt|tjjj rt |D]"\}}t||\}}|r|||<qd|fSt|tjjj rt |D]"\}}t||\}}|r|||<qd|fSt|tj rN|j jtjkrFt|jD],}t|||\}}|r|||qd|fSd|fSdS)NrzNVariable referenced by pointer not found in heap: variable will be set to NoneTF) isinstancer?rArqrr _replace_heapr,corerecordsrd enumeraterZndarrayrQtyperbresizeitemZitemset)variableheapreplacenewZirrZivvaluer#r#r$r4s@        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||dSr@)dictrC)rBinitr#r#r$rCszAttrDict.__init__cst|Sr@)super __getitem__rarBrY __class__r#r$rszAttrDict.__getitem__cst||Sr@)r __setitem__ra)rBkeyrrr#r$rszAttrDict.__setitem__c CsBz ||WStk r<tdt|d|ddYnXdS)N'z' object has no attribute ')rKeyErrorAttributeErrorrrr#r#r$ __getattr__s zAttrDict.__getattr__) rDrErFrGrCrrr __setattr____call__ __classcell__r#r#rr$rvs   rFc Csg}|s |ri}nt}t|d}t|d}|dkr@td|t|d} | dkrVn| dkr|rltd|r|t|d} n tjd d } |rtd | j| d t |} | t d t | t |} | t |d7} |d} t| dkrFtd}| t dt | || t dt | | ||| | q|}t|| |}| t|d} | t dt | d| t dt | | dd| | | |q|| }|dn td| t|}||d|kr|drqq|i}|D]$}|ddkr*|d||d<q*|D]H}|ddkrTt|d|\}}|r||d<|d||d<qT|r|D]R}|ddkrtdtd|dtd|d td!|d"qq|D]b}|dd#krtdtd$|d%td&|d'td(|d)td*|d+qdq|D]R}|dd,krhtdtd-|d.td/|d0td1|d2qqh|D]2}|dd3krtdtd4|d5qqtdtd6t|d7d8|D}t|D]&}|dkr"td9|||fq"tdd|krtd:|D]}td;|t||fqhtd|r|D]}||||<q|S|Sd>> 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 %ssSRr2rurrr:rzInvalid RECFMT: %srrtrrsrwrrvrz2--------------------------------------------------zDate: %srxzUser: %sryzHost: %srzrz Format: %sr{zArchitecture: %sr|zOperating System: %sr}zIDL Version: %sr~rz Author: %srz Title: %srz ID Code: %srrzDescription: %srz&Successfully read %i records of which:cSsg|] }|dqS)rtr#)rrr#r#r$rszreadsav..z - %i are of type %szAvailable variables:z - %s [%s]N)ropenr+rPprinttempfileNamedTemporaryFilerYwriter4r-packrhr;r'rr,r7rzlib decompresslencloser rr`rrasetcountr) file_nameZidictZ python_dictZuncompressed_file_nameverboser variablesr! signatureZrecfmtZfoutrtrunknownZmodvalr"Z rec_stringrrrrrZrectypesrtvarr#r#r$rsF          " "              )NFNF)'__all__r-numpyr,rrrqrcrrr%r*r+r0r4r5r6r8r9r;r<r=r>r?rHrNrOrTrfrgrrrrrrrrrr#r#r#r$s   *-/m/6B-