1use super::{PI, SQRT_PI, TWO_129, TWO_M27};
2
3const R00: f64 = -6.25000000000000000000e-02; const R01: f64 = 1.40705666955189706048e-03; const R02: f64 = -1.59955631084035597520e-05; const R03: f64 = 4.96727999609584448412e-08; const S01: f64 = 1.91537599538363460805e-02; const S02: f64 = 1.85946785588630915560e-04; const S03: f64 = 1.17718464042623683263e-06; const S04: f64 = 5.04636257076217042715e-09; const S05: f64 = 1.23542274426137913908e-11; pub fn bessel_j1(x: f64) -> f64 {
59 if f64::is_nan(x) {
89 return f64::NAN;
90 } else if f64::is_infinite(x) {
91 return 0.0;
92 } else if x == 0.0 {
93 return 0.0;
94 }
95
96 let (xx, negative) = if x < 0.0 { (-x, true) } else { (x, false) };
97
98 if xx >= 2.0 {
99 let (s, c) = f64::sin_cos(xx);
100 let mut ss = -s - c;
101 let mut cc = s - c;
102
103 if xx < f64::MAX / 2.0 {
105 let z = f64::cos(xx + xx);
106 if s * c > 0.0 {
107 cc = z / ss;
108 } else {
109 ss = z / cc;
110 }
111 }
112
113 let z = if xx > TWO_129 {
114 (1.0 / SQRT_PI) * cc / f64::sqrt(xx)
115 } else {
116 let u = pone(xx);
117 let v = qone(xx);
118 (1.0 / SQRT_PI) * (u * cc - v * ss) / f64::sqrt(xx)
119 };
120
121 if negative {
122 return -z;
123 } else {
124 return z;
125 }
126 }
127
128 if xx < TWO_M27 {
129 return 0.5 * xx;
130 }
131
132 let mut z = xx * xx;
133 let mut r = z * (R00 + z * (R01 + z * (R02 + z * R03)));
134 let s = 1.0 + z * (S01 + z * (S02 + z * (S03 + z * (S04 + z * S05))));
135 r *= xx;
136 z = 0.5 * xx + r / s;
137
138 if negative {
139 return -z;
140 } else {
141 return z;
142 }
143}
144
145const TWO_M54: f64 = 5.5511151231257827021181583404541015625000000000000e-17; const U00: f64 = -1.96057090646238940668e-01; const U01: f64 = 5.04438716639811282616e-02; const U02: f64 = -1.91256895875763547298e-03; const U03: f64 = 2.35252600561610495928e-05; const U04: f64 = -9.19099158039878874504e-08; const V00: f64 = 1.99167318236649903973e-02; const V01: f64 = 2.02552581025135171496e-04; const V02: f64 = 1.35608801097516229404e-06; const V03: f64 = 6.22741452364621501295e-09; const V04: f64 = 1.66559246207992079114e-11; pub fn bessel_y1(x: f64) -> f64 {
182 if x < 0.0 || f64::is_nan(x) {
212 return f64::NAN;
213 } else if f64::is_infinite(x) {
214 return 0.0;
216 } else if x == 0.0 {
217 return f64::NEG_INFINITY;
218 }
219
220 if x >= 2.0 {
221 let (s, c) = f64::sin_cos(x);
222 let mut ss = -s - c;
223 let mut cc = s - c;
224
225 if x < f64::MAX / 2.0 {
227 let z = f64::cos(x + x);
228 if s * c > 0.0 {
229 cc = z / ss;
230 } else {
231 ss = z / cc;
232 }
233 }
234
235 let z = if x > TWO_129 {
236 (1.0 / SQRT_PI) * ss / f64::sqrt(x)
237 } else {
238 let u = pone(x);
239 let v = qone(x);
240 (1.0 / SQRT_PI) * (u * ss + v * cc) / f64::sqrt(x)
241 };
242
243 return z;
244 }
245
246 if x <= TWO_M54 {
247 return -(2.0 / PI) / x;
248 }
249
250 let z = x * x;
251 let u = U00 + z * (U01 + z * (U02 + z * (U03 + z * U04)));
252 let v = 1.0 + z * (V00 + z * (V01 + z * (V02 + z * (V03 + z * V04))));
253 x * (u / v) + (2.0 / PI) * (bessel_j1(x) * f64::ln(x) - 1.0 / x)
254}
255
256const P1R8: [f64; 6] = [
274 0.00000000000000000000e+00, 1.17187499999988647970e-01, 1.32394806593073575129e+01, 4.12051854307378562225e+02, 3.87474538913960532227e+03, 7.91447954031891731574e+03, ];
281const P1S8: [f64; 5] = [
282 1.14207370375678408436e+02, 3.65093083420853463394e+03, 3.69562060269033463555e+04, 9.76027935934950801311e+04, 3.08042720627888811578e+04, ];
288
289const P1R5: [f64; 6] = [
291 1.31990519556243522749e-11, 1.17187493190614097638e-01, 6.80275127868432871736e+00, 1.08308182990189109773e+02, 5.17636139533199752805e+02, 5.28715201363337541807e+02, ];
298const P1S5: [f64; 5] = [
299 5.92805987221131331921e+01, 9.91401418733614377743e+02, 5.35326695291487976647e+03, 7.84469031749551231769e+03, 1.50404688810361062679e+03, ];
305
306const P1R3: [f64; 6] = [
308 3.02503916137373618024e-09, 1.17186865567253592491e-01, 3.93297750033315640650e+00, 3.51194035591636932736e+01, 9.10550110750781271918e+01, 4.85590685197364919645e+01, ];
315const P1S3: [f64; 5] = [
316 3.47913095001251519989e+01, 3.36762458747825746741e+02, 1.04687139975775130551e+03, 8.90811346398256432622e+02, 1.03787932439639277504e+02, ];
322
323const P1R2: [f64; 6] = [
325 1.07710830106873743082e-07, 1.17176219462683348094e-01, 2.36851496667608785174e+00, 1.22426109148261232917e+01, 1.76939711271687727390e+01, 5.07352312588818499250e+00, ];
332const P1S2: [f64; 5] = [
333 2.14364859363821409488e+01, 1.25290227168402751090e+02, 2.32276469057162813669e+02, 1.17679373287147100768e+02, 8.36463893371618283368e+00, ];
339
340fn pone(x: f64) -> f64 {
341 let (p, q) = if x >= 8.0 {
342 (&P1R8, &P1S8)
343 } else if x >= 4.5454 {
344 (&P1R5, &P1S5)
345 } else if x >= 2.8571 {
346 (&P1R3, &P1S3)
347 } else if x >= 2.0 {
348 (&P1R2, &P1S2)
349 } else {
350 panic!("INTERNAL ERROR: x must be ≥ 2.0 for pone");
351 };
352 let z = 1.0 / (x * x);
353 let r = p[0] + z * (p[1] + z * (p[2] + z * (p[3] + z * (p[4] + z * p[5]))));
354 let s = 1.0 + z * (q[0] + z * (q[1] + z * (q[2] + z * (q[3] + z * q[4]))));
355 1.0 + r / s
356}
357
358const Q1R8: [f64; 6] = [
377 0.00000000000000000000e+00, -1.02539062499992714161e-01, -1.62717534544589987888e+01, -7.59601722513950107896e+02, -1.18498066702429587167e+04, -4.84385124285750353010e+04, ];
384const Q1S8: [f64; 6] = [
385 1.61395369700722909556e+02, 7.82538599923348465381e+03, 1.33875336287249578163e+05, 7.19657723683240939863e+05, 6.66601232617776375264e+05, -2.94490264303834643215e+05, ];
392
393const Q1R5: [f64; 6] = [
395 -2.08979931141764104297e-11, -1.02539050241375426231e-01, -8.05644828123936029840e+00, -1.83669607474888380239e+02, -1.37319376065508163265e+03, -2.61244440453215656817e+03, ];
402const Q1S5: [f64; 6] = [
403 8.12765501384335777857e+01, 1.99179873460485964642e+03, 1.74684851924908907677e+04, 4.98514270910352279316e+04, 2.79480751638918118260e+04, -4.71918354795128470869e+03, ];
410
411const Q1R3: [f64; 6] = [
413 -5.07831226461766561369e-09, -1.02537829820837089745e-01, -4.61011581139473403113e+00, -5.78472216562783643212e+01, -2.28244540737631695038e+02, -2.19210128478909325622e+02, ];
420const Q1S3: [f64; 6] = [
421 4.76651550323729509273e+01, 6.73865112676699709482e+02, 3.38015286679526343505e+03, 5.54772909720722782367e+03, 1.90311919338810798763e+03, -1.35201191444307340817e+02, ];
428
429const Q1R2: [f64; 6] = [
431 -1.78381727510958865572e-07, -1.02517042607985553460e-01, -2.75220568278187460720e+00, -1.96636162643703720221e+01, -4.23253133372830490089e+01, -2.13719211703704061733e+01, ];
438const Q1S2: [f64; 6] = [
439 2.95333629060523854548e+01, 2.52981549982190529136e+02, 7.57502834868645436472e+02, 7.39393205320467245656e+02, 1.55949003336666123687e+02, -4.95949898822628210127e+00, ];
446
447fn qone(x: f64) -> f64 {
448 let (p, q) = if x >= 8.0 {
449 (&Q1R8, &Q1S8)
450 } else if x >= 4.5454 {
451 (&Q1R5, &Q1S5)
452 } else if x >= 2.8571 {
453 (&Q1R3, &Q1S3)
454 } else if x >= 2.0 {
455 (&Q1R2, &Q1S2)
456 } else {
457 panic!("INTERNAL ERROR: x must be ≥ 2.0 for qone");
458 };
459 let z = 1.0 / (x * x);
460 let r = p[0] + z * (p[1] + z * (p[2] + z * (p[3] + z * (p[4] + z * p[5]))));
461 let s = 1.0 + z * (q[0] + z * (q[1] + z * (q[2] + z * (q[3] + z * (q[4] + z * q[5])))));
462 (0.375 + r / s) / x
463}
464
465#[cfg(test)]
468mod tests {
469 use super::{bessel_j1, bessel_y1, pone, qone, TWO_129, TWO_M54};
470 use crate::{approx_eq, assert_alike};
471
472 #[test]
473 #[should_panic(expected = "INTERNAL ERROR: x must be ≥ 2.0 for pone")]
474 fn pone_panics_on_wrong_input() {
475 pone(1.99);
476 }
477
478 #[test]
479 #[should_panic(expected = "INTERNAL ERROR: x must be ≥ 2.0 for qone")]
480 fn qone_panics_on_wrong_input() {
481 qone(1.99);
482 }
483
484 #[test]
485 fn bessel_j1_handles_special_cases() {
486 assert!(bessel_j1(f64::NAN).is_nan());
487 assert_eq!(bessel_j1(f64::NEG_INFINITY), 0.0);
488 assert_eq!(bessel_j1(f64::INFINITY), 0.0);
489 assert_eq!(bessel_j1(0.0), 0.0);
490 }
491
492 #[test]
493 fn bessel_y1_handles_special_cases() {
494 assert!(bessel_y1(f64::NEG_INFINITY).is_nan());
495 assert!(bessel_y1(-0.01).is_nan());
496 assert!(bessel_y1(f64::NAN).is_nan());
497 assert_eq!(bessel_y1(f64::INFINITY), 0.0);
498 assert_eq!(bessel_y1(0.0), f64::NEG_INFINITY);
499 }
500
501 #[test]
502 fn bessel_j1_works() {
503 approx_eq(
505 bessel_j1(-123.0),
506 -0.02156735149890660940086328069036361009670280158394365333670828669644907165480535175323397489229874718,
507 1e-17,
508 );
509
510 assert_eq!(
512 bessel_j1(-5.0),
513 0.3275791375914652220377343219101691327608499046240540186864806450648753089914574086157808718640701942
514 );
515
516 approx_eq(
518 bessel_j1(-2.0),
519 -0.5767248077568733872024482422691370869203026897196754401211390207640871162896121849483995433063402923,
520 1e-15,
521 );
522
523 assert_eq!(
525 bessel_j1(-1.0),
526 -0.4400505857449335159596822037189149131273723019927652511367581717801382224780155479307965923811982542
527 );
528
529 assert_eq!(
531 bessel_j1(1e-9),
532 4.999999999999999999375000000000000000026041666666666666666124131944444444444451226128472222222222166e-10
533 );
534
535 assert_eq!(
537 bessel_j1(1.0),
538 0.4400505857449335159596822037189149131273723019927652511367581717801382224780155479307965923811982542
539 );
540
541 approx_eq(
543 bessel_j1(2.0),
544 0.5767248077568733872024482422691370869203026897196754401211390207640871162896121849483995433063402923,
545 1e-15,
546 );
547
548 assert_eq!(
550 bessel_j1(5.0),
551 -0.3275791375914652220377343219101691327608499046240540186864806450648753089914574086157808718640701942
552 );
553
554 approx_eq(
556 bessel_j1(123.0),
557 0.02156735149890660940086328069036361009670280158394365333670828669644907165480535175323397489229874718,
558 1e-17,
559 );
560 }
561
562 #[test]
563 fn bessel_y1_works() {
564 approx_eq(
566 bessel_y1(1e-9),
567 -6.366197723675813498680125340511460635200799230620073016920009919302213806667149128128951250692787669e8,
568 1e-6,
569 );
570
571 approx_eq(
573 bessel_y1(1.0),
574 -0.7812128213002887165471500000479648205499063907164446078438332461277843915385602167276292380048056346,
575 1e-17,
576 );
577
578 approx_eq(
580 bessel_y1(2.0),
581 -0.1070324315409375468883707722774766366874808982350538605257945572313199774994906724960750929874561422,
582 1e-16,
583 );
584
585 approx_eq(
587 bessel_y1(5.0),
588 0.1478631433912268448010506754880372053734652184104756133006976595086882659262171067506925531074677045,
589 1e-16,
590 );
591
592 approx_eq(
594 bessel_y1(123.0),
595 0.06863489010254926652096999352892516176822669293542798648532852667873692449570610866356639175413101604,
596 1e-17,
597 );
598 }
599
600 #[test]
601 fn bessel_j1_edge_cases_work() {
602 approx_eq(bessel_j1(f64::MAX / 2.0), 5.936112522662019e-155, 1e-310);
609
610 approx_eq(
615 bessel_j1(2.0 * TWO_129),
616 -2.148812412212001397545607680138144226557017053953123779067461809570040430645106137686812136124982505e-20,
617 1e-100,
618 );
619 }
620
621 #[test]
622 fn bessel_y1_edge_cases_work() {
623 approx_eq(bessel_y1(f64::MAX / 2.0), -5.965640685080747e-155, 1e-310);
630
631 approx_eq(
635 bessel_y1(TWO_M54),
636 -1.146832227844531729100095246803011708838780681972379423086390888013625612852380226098068118499203402e16,
637 1e-100,
638 );
639 }
640
641 const VALUES: [f64; 10] = [
644 4.9790119248836735e+00,
645 7.7388724745781045e+00,
646 -2.7688005719200159e-01,
647 -5.0106036182710749e+00,
648 9.6362937071984173e+00,
649 2.9263772392439646e+00,
650 5.2290834314593066e+00,
651 2.7279399104360102e+00,
652 1.8253080916808550e+00,
653 -8.6859247685756013e+00,
654 ];
655
656 const SOLUTION_J1: [f64; 10] = [
657 -3.251526395295203422162967e-01,
658 1.893581711430515718062564e-01,
659 -1.3711761352467242914491514e-01,
660 3.287486536269617297529617e-01,
661 1.3133899188830978473849215e-01,
662 3.660243417832986825301766e-01,
663 -3.4436769271848174665420672e-01,
664 4.329481396640773768835036e-01,
665 5.8181350531954794639333955e-01,
666 -2.7030574577733036112996607e-01,
667 ];
668
669 const SOLUTION_Y1: [f64; 10] = [
670 0.15494213737457922210218611,
671 -0.2165955142081145245075746,
672 -2.4644949631241895201032829,
673 0.1442740489541836405154505,
674 0.2215379960518984777080163,
675 0.3038800915160754150565448,
676 0.0691107642452362383808547,
677 0.2380116417809914424860165,
678 -0.20849492979459761009678934,
679 0.0242503179793232308250804,
680 ];
681
682 const SC_VALUES: [f64; 4] = [f64::NEG_INFINITY, 0.0, f64::INFINITY, f64::NAN];
683
684 const SC_SOLUTION_J1: [f64; 4] = [0.0, 0.0, 0.0, f64::NAN];
685
686 const SC_SOLUTION_Y1: [f64; 5] = [f64::NAN, f64::NEG_INFINITY, 0.0, f64::NAN, f64::NAN];
687
688 #[test]
689 fn test_bessel_j1() {
690 for (i, v) in VALUES.iter().enumerate() {
691 let f = bessel_j1(*v);
692 approx_eq(SOLUTION_J1[i], f, 1e-15);
693 }
694 for (i, v) in SC_VALUES.iter().enumerate() {
695 let f = bessel_j1(*v);
696 assert_alike(SC_SOLUTION_J1[i], f);
697 }
698 }
699
700 #[test]
701 fn test_bessel_y1() {
702 for (i, v) in VALUES.iter().enumerate() {
703 let a = f64::abs(*v);
704 let f = bessel_y1(a);
705 approx_eq(SOLUTION_Y1[i], f, 1e-15);
706 }
707 for (i, v) in SC_VALUES.iter().enumerate() {
708 let f = bessel_y1(*v);
709 assert_alike(SC_SOLUTION_Y1[i], f);
710 }
711 }
712}