R1i-dZddlmZmZmZddlmZddlmZddl m Z ddl m Z dZ d Zd ZGd d eZd ZGddeZGddeZGddeZdZeeedZy)zCompute positions from a NASA SPICE SPK ephemeris kernel file. http://naif.jpl.nasa.gov/pub/naif/toolkit_docs/FORTRAN/req/spk.html )arrayinterprollaxis)DAF)reify)OutOfRangeError) target_namesg,BAg@c"t|tz zS)z9Convert a number of seconds since J2000 to a Julian Date.)T0 S_PER_DAY)secondss P/home/cursorai/projects/iching/venv/lib/python3.12/site-packages/jplephem/spk.pyjdrs )# ##cJeZdZdZdZedZdZdZdZ dZ dZ d Z y ) SPKaA JPL SPK ephemeris kernel for computing positions and velocities. You can load an SPK by specifying its filename:: kernel = SPK.open('de431.bsp') Run ``print(kernel)`` see which segments are inside. You can also loop across all of the segments in the list ``kernel.segments`` or, as a convenience, you can select a particular segment by providing a center and target integer in square brackets. So ``kernel[3,399]`` will select the segment that computes the distance between the Earth-Moon barycenter (3) and the Earth itself (399). To extract the text comments from the SPK use ``kernel.comments()``. c ||_|jjDcgc]\}}t|j||c}}|_t d|jD|_ycc}}w)Nc3PK|]}|j|jf|f ywN)centertarget).0ss r zSPK.__init__..*s#K]AHHahh/3]s$&)daf summaries build_segmentsegmentsdictpairsselfrsource descriptors r__init__z SPK.__init__$sf'+hh&8&8&: &:"  $((FJ 7&:  KT]]KK  s!A1c8|tt|dS)z3Open the file at `path` and return an SPK instance.rb)ropen)clspaths rr)zSPK.open,s3tD$'())rc|jjj|jD]}d|jvs|`d|j_d|j_y)zClose this SPK file._dataN)rfilecloser__dict__r-_array_map)r#segments rr/z SPK.close1sO  }}G'***M% rc|j}d}dj||j||jt |j g}|j d|j Ddj|S)Nc$|jdS)Nzlatin-1)decode)bs rzSPK.__str__..<s ahhy)rz/File type {0} and format {1} with {2} segments:c32K|]}t|ywr)str)rr3s rrzSPK.__str__..As?gS\s )rformatlocidwlocfmtlenrextendjoin)r#rdliness r__str__z SPK.__str__:shhh ) = VAcjjM1SZZ=#dmm2D E   ???yyrc |j|S)zBGiven (center, target) integers, return the last matching segment.)r!)r#keys r __getitem__zSPK.__getitem__Dszz#rc6|jjS)z&Return the file comments, as a string.)rcommentsr#s rrIz SPK.commentsHsxx  ""rc|SrrJs r __enter__z SPK.__enter__Ls rc$|jyr)r/)r#exc_typeexc_valexc_tbs r__exit__z SPK.__exit__Os  rN) __name__ __module__ __qualname____doc__r& classmethodr)r/rDrGrIrMrRrLrrrrs? L** #rrcT|d}tj|t}||||S)N)_segment_classesget BaseSegment)rr$r% data_typer*s rrrRs,1 I   y+ 6C sFJ ''rc8eZdZdZdZdZdZddZd dZd dZ y) r\aA single segment of an SPK file. There are several items of information about each segment that are loaded from the underlying SPK file, and made available as object attributes: segment.source - official ephemeris name, like 'DE-0430LE-0430' segment.start_second - initial epoch, as seconds from J2000 segment.end_second - final epoch, as seconds from J2000 segment.start_jd - start_second, converted to a Julian Date segment.end_jd - end_second, converted to a Julian Date segment.center - integer center identifier segment.target - integer target identifier segment.frame - integer frame identifier segment.data_type - integer data type identifier segment.start_i - index where segment starts segment.end_i - index where segment ends Nc ||_||_|\|_|_|_|_|_|_|_|_ t|j|_ t|j|_ yr) rr$ start_second end_secondrrframer]start_iend_irstart_jdend_jdr"s rr&zBaseSegment.__init__ms\ AK ? DOT[$+ T^T\4:4,,- ) rc&|jdS)NF)verbose)describerJs rrDzBaseSegment.__str__us}}U}++rc0ttj|jd}ttj|jd}dj |||}|r.|dj ||j jdz }|S)z,Return a textual description of the segment.zUnknown centerzUnknown targetzZ{0.start_jd:.2f}..{0.end_jd:.2f} Type {0.data_type} {1} ({0.center}) -> {2} ({0.target})z frame={0.frame} source={1}ascii) titlecaser r[rrr<r$r6)r#rhrrtexts rrizBaseSegment.describexs<++DKK9IJK<++DKK9IJK9ff-   5fT4;;#5#5g#>? AD rcJtdj|j)>Q O ^^q ONO O!%!4!4TZZ!^TZZ!PfeQ No=xx))$,, QG !!fc%j1 #AabD) !!fo7HI  a0  a0 #DbD) V\))rc|j\}}}t|}|tz }|ddd}t|d}t|d}|||fS)Nrr)r-rr r)r#rrr initial_epochinterval_lengths r load_arrayzSegment.load_arraysX%)ZZ"fl4  9,#DbD)  a0  a0 o|;;rc#Kt|dd xrt|dd }|r t|f}|j\}}}|j\}}} t |t z t z|z |\} } t |t z|\} } t | | z|\}}| | z|zjt}|dkjs|| kDjr/td|j|jfz|dk|| kDz|| k(}||xxdzcc<||xx|z cc<|dddd|f}d|z|z dz }d|z}d x}}g}|dd D]"}|}|}|||z|z z}|j|$|d ||z|z z}|r |dddf}|d x}}t|dd |D]\}}|}|}d|z||zz|z }|||zz|z }||z}|dz}|t z}|r |dddf}|yw) aGenerate components and differentials for time `tdb` plus `tdb2`. Most uses will simply want to call the `compute()` method or the `compute_differentials()` method, for convenience. But in those cases (see Skyfield) where you want to compute a position and examine it before deciding whether to proceed with the velocity, but without losing all of the work that it took to get to that point, this generator lets you get them as two separate steps. rrz+segment only covers dates %.1f through %.1f)out_of_range_timesrNg@g?rzr)getattrrr-rdivmodr r astyperanyr rerfappendzip)r#rsrtscalarrrrrrrindex1offset1index2offset2index3offsetindexomegasrs2w0w1wlist coefficientw2 componentsdw0dw1dw2ratess rr~zSegment.generatesfS'1--OgdGQ6O2O -C%)ZZ"fl0<0B0B-*A !#(i!7$!>G  !16:' 16:&6)11#6 AI??  1!===$++./$)AI%!)#<  1* f  v& #AaI.  &L6 !C ' 1W R',KBBR" -B LL  - ""%R"5 #AaCJc"<#4e6i@ifS&!,i@AA@5c t|j\}}}t|Dcgc]}t|||c}Scc}w)zInterpolate [x y z dx dy dz] at time `tdb` plus `tdb2`. A standard JPL Type 9 ephemerides will return kilometers and kilometers per second. rrs rrxz&Type9Segment.compute_and_differentiate9s:-1JJ) >6nEnfS&!,nEFFErNry) rSrTrUrVrrr-rurxrLrrrrs*5$" 1 1BGrrcH|jdr|S|jS)z3Title-case target `name` if it looks safe to do so.)1zC/zDSS-) startswithtitle)names rrlrlCs??#674ITZZ\Ir)rr N)rVnumpyrrrrr descriptorlibr exceptionsr namesr r r robjectrrr\r|rrlrZrLrrrs *) '  $=&=~( ; &; |zkzx0G;0GdJ r