R1i'hdZddlZddlZddlZddlmZddlmZm Z dZ dddZ d Z Gd d e ZeZy) zAccess a NASA JPL SPICE Double Precision Array File (DAF). http://naif.jpl.nasa.gov/pub/naif/toolkit_docs/FORTRAN/req/daf.html N)Struct)arrayndarraysFTPSTR: : : : :::ENDFTP><)sBIG-IEEEsLTL-IEEEcXeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd Zy)DAFa2Access to NASA SPICE Double Precision Array Files (DAF). Provide the constructor with a ``file_object`` for full access to both the segment summaries and to the numeric arrays. If you pass a ``StringIO`` instead, then you can fetch the summary information but not access the arrays. ct|ddr td|_d_d_j dfd}ddj j_jdk(rItjD]'\_ _ |jdk(s'ntdjjd rd d jd t k7r td dd_ tj#j_ j$tdj%j|n$tdj%jj&j_djzdj*zz}t-jdz_t-j|zx_}|j2x_}|| dzz_d j6z_y)Nencodingz-file_object must be opened in binary "b" modec jdz}t|_jj\ }______ }_ _ _ y)N8sII60sIII8s603s28s297s) endianrfile_record_structunpackndnilocifnfwardbwardfreeprenulftpstrpstnul)fmtlocidwlocfmt file_recordselfs P/home/cursorai/projects/iching/venv/lib/python3.12/site-packages/jplephem/daf.pyrzDAF.__init__..unpack#sb++ 99C&,SkD #''..{; VTWdgt{DJ Y T[$+sNAIF/DAFzOneither a big- nor a little-endian scan of this file produces the expected ND=2sDAF/izthis SPK file has been damagedX`zunknown format {0!r}z0file starts with {0!r}, not "NAIF/DAF" or "DAF/"diddd)getattr ValueErrorfile_map_array read_recordupperrstriprLOCFMTitemsrrr startswithstripFTPSTRgetformatr locifn_textrrsummary_control_structsummary_structsizesummary_length summary_stepsummaries_per_record)r file_objectrsummary_formatstructlengthrs` @r!__init__z DAF.__init__s ; D 1LM M   &&q)  <""1o++-446 ;;+ %,2LLN( T[77a<-; !"LMM [[ # #G ,3t$**51V; !ABB%b,DK **T[[1DK{{" !7!>!>t{{!KLL HO$fT[[13 3 ;;--/twwtww6&,T[[5-@&A#'-dkkN.J'KKf'-{{2f"vgk2%1d6G6G$G!r"c|jj|tztz |jjtS)z=Return record `n` as 1,024 bytes; records are indexed from 1.)r.seekKread)r ns r!r1zDAF.read_recordJs- q1uqy!yy~~a  r"c|jj|tztz |jj|S)zzDAF.comments..s$L^((+Ad3^s!Nascii z-DAF file comment area is missing its EOT bytez'DAF file comment area is not ASCII text) rangerjoinfinddecodereplace IndexErrorr-UnicodeDecodeError)r record_numbersrNs` r!commentsz DAF.commentszsq$**-xxL^LL H)5)*11':BB4N N NLM M! HFG G Hs 2A))(Bc|j}|jd|dz zd|z|z }|jd|z}t||jdz|S)aReturn floats from `start` to `end` inclusive, indexed from 1. The entire range of floats is immediately read into memory from the file, making this efficient for small sequences of floats whose values are all needed immediately. r#r r))r.rHrJrr)r rdrefrErNs r! read_arrayzDAF.read_arraysX II qEAIS5vva&j!vt{{S0$77r"c|jd|jd|jdz \|_}|dk(sJt |jdz |j dz|j|_|j|dz |S)aReturn floats from `start` to `end` inclusive, indexed from 1. Instead of pausing to load all of the floats into RAM, this routine creates a memory map which will load data from the file only as it is accessed, and then will let it expire back out to disk later. This is very efficient for large data sets to which you need random access. r rr))r0rjrr/rr)r rdrergs r! map_arrayz DAF.map_arraysr ;; "nnQ A >ODIt19 9!$))a-s1BDIINDK{{519s++r"c#K|j}|jj}|r6|j|}||dd\}}}|||ft |}|r5yyw)zYield (record_number, n_summaries, record_data) for each record. Readers will only use the second two values in each tuple. Writers can update the record using the `record_number`. N)rr<rr1int)r record_numberrrN next_numberprevious_number n_summariess r!summary_recordszDAF.summary_recordssh ,,33##M2D8>tCRy8I 5K+d2 2 ,M s AAAc#K|j}|j}|jD]\}}}|j|dz}t dt ||z|D]Z}|j j|z}||||zj} ||||z} |jj| } | | f\yw)z?Yield (name, (value, value, ...)) for each summary in the file.r rN) r?r@rr1rtrr<r>r7r=r) r rEsteprr summary_data name_datar*rfnamerNvaluess r! summariesz DAF.summariess$$  8<8L8L8N 4M; (():;I1c+.5t<//44q8 1T6*002#Aah/,,33D9Fl" =9OsB=B?c2|j|d|dS)azReturn the array of floats described by a summary. Instead of pausing to load all of the floats into RAM, this routine creates a memory map which will load data from the file only as it is accessed, and then will let it expire back out to disk later. This is very efficient for large data sets to which you need random access. )r)r summary_valuess r!mapzDAF.maps~~nR0.2DEEr"c`|j}|j}|j}t|j |}|j |dd\}} } | |j kr3|} | dz} |j|| | dz|dd|j| |n|jdz dzdzdzdz} | dz} | dz} d} |j| | | |dd|j|||jd|djdd }d }|j| ||j| || |_| dz dzdzdz|_ t||jd z}|j}|j|dz dz|j|j|j!dz}|dz|_ |j#|d|j$|j&zdz ||fz}d| dz z}t)| |j*z}|j||j,z|z|j|j.j||j|dz|z|j|d|j0j|j*d y) zAdd a new array to the DAF file. The summary will be initialized with the `name` and `values`, and will have its start word and end word fields set to point to where the `array` of floats has been appended to the file. Nrr r#irr$rr&sf8rQ)r.r<r bytearrayr1rrArRrOrrS numpy_arrayrrHrMviewtellrTrrrr@r>r=r?)r rrrr~scsrrN next_recordprevious_recordrsummary_record name_record free_recordrnames start_wordend_wordbaserWs r! add_arrayz DAF.add_arrays II)) ))-8947JJtCRy4I1 _k 22 2*N(1,Ko{QOD"I   nd 3#yy1}1D8TAAEN(1,K(1,KK+ND"I   mT 2M15;;D%HI E   ni 8   k5 1'DJ$qD0A59DIE4;;#56YY   Q!#$  668q=qL   .477*Q./:x2HH~)*[!D$5$55 tchh'( (##((&12 td{V#$ )d))*001B1BDIJr"N)__name__ __module__ __qualname____doc__rFr1rOrTrjr|rrrrrrr"r!r r sH/Hb! % #< H 8, - # F6Kr"r )rr_rZrarDrnumpyrrrr8r4rIobjectr NAIF_DAFrr"r!rsG  / 9 -sK&sKl r"