1use crate::devices::Device;
49use crate::disto::DistoSetupCtx;
50use crate::integ::TerrCtx;
51use crate::ltrafit::{
52 ExpFit, RC_WINDOW_FRACTION, fit_rc, fit_rlc, i0e, i1e, i1e_over_x, rc_integrals, segment,
53};
54use crate::matrix::{
55 AcCtx, AcceptCtx, InitMode, LoadCtx, MatrixBuilder, PeriodicPast, Slot, TRASH, Var,
56};
57
58use libm::{atan, cos, exp, expm1, sin};
71
72#[derive(Clone, Copy, Debug, PartialEq, Eq)]
76pub enum LtraInterp {
77 Linear,
79 Quadratic,
81 Mixed,
84}
85
86#[derive(Clone, Debug)]
90pub struct LtraModel {
91 pub r: Option<f64>,
93 pub l: Option<f64>,
95 pub g: Option<f64>,
97 pub c: Option<f64>,
100 pub len: Option<f64>,
102 pub rel: Option<f64>,
106 pub abs: Option<f64>,
109 pub steplimit: bool,
112 pub interp: Option<LtraInterp>,
115 pub truncnr: bool,
119 pub truncdontcut: bool,
122 pub compactrel: f64,
125 pub compactabs: f64,
128}
129
130impl Default for LtraModel {
131 fn default() -> Self {
132 LtraModel {
133 r: None,
134 l: None,
135 g: None,
136 c: None,
137 len: None,
138 rel: None,
139 abs: None,
140 steplimit: true,
141 interp: None,
142 truncnr: false,
143 truncdontcut: false,
144 compactrel: 0.0,
145 compactabs: 0.0,
146 }
147 }
148}
149
150#[derive(Clone, Copy, Debug, PartialEq, Eq)]
152pub(crate) enum LtraCase {
153 Lc,
154 Rlc,
155 Rc,
156 Rg,
157}
158
159#[derive(Clone, Debug)]
163pub(crate) struct LtraParams {
164 pub name: String,
166 pub case: LtraCase,
167 resist: f64,
168 induct: f64,
169 capac: f64,
170 length: f64,
171 pub td: f64,
173 imped: f64,
174 admit: f64,
175 alpha: f64,
176 beta: f64,
177 attenuation: f64,
179 cbyr: f64,
180 rclsqr: f64,
181 int_h1dash: f64,
182 int_h2: f64,
183 int_h3dash: f64,
184 cosh_lroot_gr: f64,
185 r_rs_lr_gr_or_g: f64,
186 r_gs_lr_gr_or_r: f64,
187 reltol: f64,
188 abstol: f64,
189 chop_reltol: f64,
192 step_limit: bool,
193 interp: LtraInterp,
194 trunc_dont_cut: bool,
195 max_safe_step: f64,
196 st_line_reltol: f64,
200 st_line_abstol: f64,
201 interp_given: bool,
204 pub notes: Vec<String>,
206}
207
208impl LtraParams {
209 pub fn new(name: &str, m: &LtraModel, reltol: f64, abstol: f64) -> Result<Self, String> {
213 let mut notes = Vec::new();
214 for (key, v) in [
215 ("r", m.r),
216 ("l", m.l),
217 ("g", m.g),
218 ("c", m.c),
219 ("len", m.len),
220 ("rel", m.rel),
221 ("abs", m.abs),
222 ] {
223 if let Some(v) = v
224 && !v.is_finite()
225 {
226 return Err(format!(
227 "`{name}`: lossy line parameter {key} is not finite"
228 ));
229 }
230 }
231 let resist = match m.r {
232 Some(v) => v,
233 None => {
234 notes.push(format!(
235 "`{name}`: lossy line series resistance not given, assumed zero"
236 ));
237 0.0
238 }
239 };
240 let st_line_reltol = if m.compactrel == 0.0 {
241 reltol
242 } else {
243 m.compactrel
244 };
245 let st_line_abstol = if m.compactabs == 0.0 {
246 abstol
247 } else {
248 m.compactabs
249 };
250 let interp = m.interp.unwrap_or(LtraInterp::Quadratic);
251 let conduct = m.g.unwrap_or(0.0);
252 let induct = match m.l {
253 Some(v) => v,
254 None => {
255 notes.push(format!(
256 "`{name}`: lossy line series inductance not given, assumed zero"
257 ));
258 0.0
259 }
260 };
261 let capac = match m.c {
264 Some(v) => v,
265 None => {
266 notes.push(format!(
267 "`{name}`: lossy line parallel capacitance not given, assumed zero"
268 ));
269 0.0
270 }
271 };
272 let Some(length) = m.len else {
273 return Err(format!("`{name}`: lossy line length must be given"));
274 };
275
276 let (r, g, l, c) = (resist, conduct, induct, capac);
279 let mut case = None;
280 if r == 0.0 && g == 0.0 && c != 0.0 && l != 0.0 {
281 case = Some(LtraCase::Lc);
282 }
283 if r != 0.0 && g == 0.0 && c != 0.0 && l != 0.0 {
284 case = Some(LtraCase::Rlc);
285 }
286 if r != 0.0 && g == 0.0 && c != 0.0 && l == 0.0 {
287 case = Some(LtraCase::Rc);
288 }
289 if r != 0.0 && g == 0.0 && c == 0.0 && l != 0.0 {
290 return Err(format!("`{name}`: RL line not supported yet"));
291 }
292 if r != 0.0 && g != 0.0 && c == 0.0 && l == 0.0 {
293 case = Some(LtraCase::Rg);
294 }
295 if g != 0.0 && (c != 0.0 || l != 0.0) {
296 return Err(format!(
297 "`{name}`: Nonzero G (except RG) line not supported yet"
298 ));
299 }
300 let nonzero = [r, g, l, c].iter().filter(|v| **v != 0.0).count();
301 let Some(case) = case.filter(|_| nonzero > 1) else {
302 return Err(format!(
303 "`{name}`: At least two of R,L,G,C must be specified and nonzero"
304 ));
305 };
306
307 let mut p = LtraParams {
308 name: name.to_string(),
309 case,
310 resist,
311 induct,
312 capac,
313 length,
314 td: 0.0,
315 imped: 0.0,
316 admit: 0.0,
317 alpha: 0.0,
318 beta: 0.0,
319 attenuation: 0.0,
320 cbyr: 0.0,
321 rclsqr: 0.0,
322 int_h1dash: 0.0,
323 int_h2: 0.0,
324 int_h3dash: 0.0,
325 cosh_lroot_gr: 0.0,
326 r_rs_lr_gr_or_g: 0.0,
327 r_gs_lr_gr_or_r: 0.0,
328 reltol: m.rel.unwrap_or(1.0),
329 abstol: m.abs.unwrap_or(1.0),
330 chop_reltol: 0.0,
331 step_limit: m.steplimit,
332 interp,
333 trunc_dont_cut: m.truncdontcut,
334 max_safe_step: 0.0,
335 st_line_reltol,
336 st_line_abstol,
337 interp_given: m.interp.is_some(),
338 notes,
339 };
340
341 match case {
343 LtraCase::Lc => {
344 p.imped = (l / c).sqrt();
345 p.admit = 1.0 / p.imped;
346 p.td = (l * c).sqrt() * length;
347 p.attenuation = 1.0;
348 }
349 LtraCase::Rlc => {
350 p.imped = (l / c).sqrt();
351 p.admit = 1.0 / p.imped;
352 p.td = (l * c).sqrt() * length;
353 p.alpha = 0.5 * (r / l);
354 p.beta = p.alpha;
355 p.attenuation = exp(-p.beta * p.td);
356 if p.alpha > 0.0 {
357 p.int_h1dash = -1.0;
358 p.int_h2 = 1.0 - p.attenuation;
359 p.int_h3dash = -p.attenuation;
360 }
361 if !p.trunc_dont_cut {
364 p.max_safe_step =
365 max_safe_step(p.td, p.alpha, p.beta, st_line_reltol, st_line_abstol);
366 }
367 }
368 LtraCase::Rc => {
369 p.cbyr = c / r;
370 p.rclsqr = r * c * length * length;
371 p.int_h1dash = 0.0;
372 p.int_h2 = 1.0;
373 p.int_h3dash = 0.0;
374 }
375 LtraCase::Rg => {
376 let x = length * (r * g).sqrt();
378 let em = exp(-x);
379 let ep = exp(x); p.cosh_lroot_gr = 0.5 * (ep + em);
381 p.r_rs_lr_gr_or_g = if g <= 1.0e-10 {
382 length * r
384 } else {
385 0.5 * (ep - em) * (r / g).sqrt()
386 };
387 p.r_gs_lr_gr_or_r = if r <= 1.0e-10 {
388 length * g
390 } else {
391 0.5 * (ep - em) * (g / r).sqrt()
392 };
393 }
394 }
395 Ok(p)
396 }
397
398 pub fn default_linear(&mut self) {
406 if !self.interp_given {
407 self.interp = LtraInterp::Linear;
408 }
409 }
410
411 pub fn try_to_compact(&mut self) {
415 if !self.interp_given {
416 self.interp = LtraInterp::Linear;
417 self.notes.push(format!(
418 "{}: using linear interpolation because trytocompact option specified",
419 self.name
420 ));
421 }
422 }
423}
424
425fn max_safe_step(td: f64, alpha: f64, beta: f64, reltol: f64, abstol: f64) -> f64 {
432 let mut xbig = td + 9.0 * td; let xsmall = td;
434 let mut xmid = 0.5 * (xbig + xsmall);
435 let y1small = rlc_h2(xsmall, td, alpha, beta);
436 let maxiter = 50;
437 let mut iters = 0;
438 loop {
439 iters += 1;
440 let y1big = rlc_h2(xbig, td, alpha, beta);
441 let y1mid = rlc_h2(xmid, td, alpha, beta);
442 let one = straight_line_check(xbig, y1big, xmid, y1mid, xsmall, y1small, reltol, abstol);
443 let done = one as u32 + one as u32;
444 if done == 2 || iters > maxiter {
445 break;
446 }
447 xbig = xmid;
448 xmid = 0.5 * (xbig + xsmall);
449 }
450 xbig - td
451}
452
453fn quad_interp(t: f64, t1: f64, t2: f64, t3: f64) -> Option<(f64, f64, f64)> {
461 if t == t1 {
462 return Some((1.0, 0.0, 0.0));
463 }
464 if t == t2 {
465 return Some((0.0, 1.0, 0.0));
466 }
467 if t == t3 {
468 return Some((0.0, 0.0, 1.0));
469 }
470 if (t2 - t1) == 0.0 || (t3 - t2) == 0.0 || (t1 - t3) == 0.0 {
471 return None;
472 }
473 let mut f1 = (t - t2) * (t - t3);
474 let mut f2 = (t - t1) * (t - t3);
475 let mut f3 = (t - t1) * (t - t2);
476 f1 /= t1 - t2;
477 f2 /= t2 - t1;
478 f2 /= t2 - t3;
479 f3 /= t2 - t3;
480 f1 /= t1 - t3;
481 f3 /= t1 - t3;
482 Some((f1, f2, f3))
483}
484
485fn lin_interp(t: f64, t1: f64, t2: f64) -> Option<(f64, f64)> {
487 if t1 == t2 {
488 return None;
489 }
490 if t == t1 {
491 return Some((1.0, 0.0));
492 }
493 if t == t2 {
494 return Some((0.0, 1.0));
495 }
496 let temp = (t - t1) / (t2 - t1);
497 Some((1.0 - temp, temp))
498}
499
500fn intlinfunc(lolimit: f64, hilimit: f64, lovalue: f64, hivalue: f64, t1: f64, t2: f64) -> f64 {
503 let width = t2 - t1;
504 if width == 0.0 {
505 return 0.0;
506 }
507 let m = (hivalue - lovalue) / width;
508 (hilimit - lolimit) * lovalue
509 + 0.5 * m * ((hilimit - t1) * (hilimit - t1) - (lolimit - t1) * (lolimit - t1))
510}
511
512fn twiceintlinfunc(
515 lolimit: f64,
516 hilimit: f64,
517 otherlolimit: f64,
518 lovalue: f64,
519 hivalue: f64,
520 t1: f64,
521 t2: f64,
522) -> f64 {
523 let width = t2 - t1;
524 if width == 0.0 {
525 return 0.0;
526 }
527 let m = (hivalue - lovalue) / width;
528 let temp1 = hilimit - t1;
529 let temp2 = lolimit - t1;
530 let temp3 = otherlolimit - t1;
531 let mut dummy = lovalue
532 * ((hilimit - otherlolimit) * (hilimit - otherlolimit)
533 - (lolimit - otherlolimit) * (lolimit - otherlolimit));
534 dummy += m
535 * ((temp1 * temp1 * temp1 - temp2 * temp2 * temp2) / 3.0
536 - temp3 * temp3 * (hilimit - lolimit));
537 dummy * 0.5
538}
539
540fn bess_i0(x: f64) -> f64 {
542 let ax = x.abs();
543 if ax < 3.75 {
544 let y = (x / 3.75) * (x / 3.75);
545 1.0 + y
546 * (3.5156229
547 + y * (3.0899424
548 + y * (1.2067492 + y * (0.2659732 + y * (0.360768e-1 + y * 0.45813e-2)))))
549 } else {
550 let y = 3.75 / ax;
551 (exp(ax) / ax.sqrt())
552 * (0.39894228
553 + y * (0.1328592e-1
554 + y * (0.225319e-2
555 + y * (-0.157565e-2
556 + y * (0.916281e-2
557 + y * (-0.2057706e-1
558 + y * (0.2635537e-1
559 + y * (-0.1647633e-1 + y * 0.392377e-2))))))))
560 }
561}
562
563fn bess_i1(x: f64) -> f64 {
565 let ax = x.abs();
566 let ans = if ax < 3.75 {
567 let y = (x / 3.75) * (x / 3.75);
568 ax * (0.5
569 + y * (0.87890594
570 + y * (0.51498869
571 + y * (0.15084934 + y * (0.2658733e-1 + y * (0.301532e-2 + y * 0.32411e-3))))))
572 } else {
573 let y = 3.75 / ax;
574 let a = 0.2282967e-1 + y * (-0.2895312e-1 + y * (0.1787654e-1 - y * 0.420059e-2));
575 let a = 0.39894228
576 + y * (-0.3988024e-1
577 + y * (-0.362018e-2 + y * (0.163801e-2 + y * (-0.1031555e-1 + y * a))));
578 a * (exp(ax) / ax.sqrt())
579 };
580 if x < 0.0 { -ans } else { ans }
581}
582
583fn bess_i1_x_over_x(x: f64) -> f64 {
585 let ax = x.abs();
586 if ax < 3.75 {
587 let y = (x / 3.75) * (x / 3.75);
588 0.5 + y
589 * (0.87890594
590 + y * (0.51498869
591 + y * (0.15084934 + y * (0.2658733e-1 + y * (0.301532e-2 + y * 0.32411e-3)))))
592 } else {
593 let y = 3.75 / ax;
594 let a = 0.2282967e-1 + y * (-0.2895312e-1 + y * (0.1787654e-1 - y * 0.420059e-2));
595 let a = 0.39894228
596 + y * (-0.3988024e-1
597 + y * (-0.362018e-2 + y * (0.163801e-2 + y * (-0.1031555e-1 + y * a))));
598 a * (exp(ax) / (ax * ax.sqrt()))
599 }
600}
601
602fn rlc_h2(time: f64, t: f64, alpha: f64, beta: f64) -> f64 {
605 if alpha == 0.0 {
606 return 0.0;
607 }
608 if time < t {
609 return 0.0;
610 }
611 let besselarg = if time != t {
612 alpha * (time * time - t * t).sqrt()
613 } else {
614 0.0
615 };
616 let exparg = -beta * time;
617 alpha * alpha * t * exp(exparg) * bess_i1_x_over_x(besselarg)
618}
619
620#[allow(clippy::too_many_arguments)]
624fn straight_line_check(
625 x1: f64,
626 y1: f64,
627 x2: f64,
628 y2: f64,
629 x3: f64,
630 y3: f64,
631 reltol: f64,
632 abstol: f64,
633) -> bool {
634 let quad1 = (y2.abs() + y1.abs()) * 0.5 * (x2 - x1).abs();
635 let quad2 = (y3.abs() + y2.abs()) * 0.5 * (x3 - x2).abs();
636 let quad3 = (y3.abs() + y1.abs()) * 0.5 * (x3 - x1).abs();
637 let tr_area = (quad3 - quad1 - quad2).abs();
638 let area = quad1 + quad2;
639 area * reltol + abstol > tr_area
640}
641
642#[derive(Clone, Debug, Default)]
645struct Coeffs {
646 h1_first: f64,
647 h2_first: f64,
648 h3_first: f64,
649 h1: Vec<f64>,
650 h2: Vec<f64>,
651 h3: Vec<f64>,
652 aux_index: usize,
653}
654
655impl Coeffs {
656 fn ensure(&mut self, n: usize) {
657 if self.h1.len() < n {
658 self.h1.resize(n, 0.0);
659 self.h2.resize(n, 0.0);
660 self.h3.resize(n, 0.0);
661 }
662 }
663}
664
665#[allow(clippy::too_many_arguments)]
677fn rlc_coeffs(
678 co: &mut Coeffs,
679 t: f64,
680 alpha: f64,
681 beta: f64,
682 curtime: f64,
683 timelist: &[f64],
684 timeindex: usize,
685 reltol: f64,
686) {
687 co.ensure(timeindex + 1);
688
689 let auxindex: usize = if t == 0.0 {
690 timeindex
691 } else if curtime - t <= 0.0 {
692 0
693 } else {
694 let mut exact = false;
695 let mut found: isize = -1;
696 for i in (0..=timeindex).rev() {
697 if curtime - timelist[i] == t {
698 exact = true;
699 found = i as isize;
700 break;
701 }
702 if curtime - timelist[i] > t {
703 found = i as isize;
704 break;
705 }
706 }
707 let a = if exact { found - 1 } else { found };
711 a.max(0) as usize
712 };
713
714 let alphasq_t = alpha * alpha * t;
715 let expbeta_t = exp(-beta * t);
716
717 let (mut h2lovalue1, mut h2hivalue1, mut h2dummy1, mut h2relval) = (0.0, 0.0, 0.0, 0.0);
718 let (mut h3hivalue1, mut h3dummy1, mut h3relval) = (0.0, 0.0, 0.0);
719
720 if auxindex != 0 {
722 let lolimit1 = t;
723 let hilimit1 = curtime - timelist[auxindex];
724 let delta1 = hilimit1 - lolimit1;
725
726 h2lovalue1 = rlc_h2(t, t, alpha, beta);
727 let besselarg = if hilimit1 > t {
728 alpha * (hilimit1 * hilimit1 - t * t).sqrt()
729 } else {
730 0.0
731 };
732 let expterm = exp(-beta * hilimit1);
733 let bessi1overxterm = bess_i1_x_over_x(besselarg);
734 h2hivalue1 = if alpha == 0.0 || hilimit1 < t {
735 0.0
736 } else {
737 alphasq_t * expterm * bessi1overxterm
738 };
739 h2dummy1 = twiceintlinfunc(
740 lolimit1, hilimit1, lolimit1, h2lovalue1, h2hivalue1, lolimit1, hilimit1,
741 ) / delta1;
742 co.h2_first = h2dummy1;
743 h2relval = (reltol * h2dummy1).abs();
744
745 let h3lovalue1 = 0.0; let bessi0term = bess_i0(besselarg);
747 h3hivalue1 = if hilimit1 <= t || beta == 0.0 {
748 0.0
749 } else {
750 expterm * bessi0term - expbeta_t
751 };
752 h3dummy1 = intlinfunc(
753 lolimit1, hilimit1, h3lovalue1, h3hivalue1, lolimit1, hilimit1,
754 ) / delta1;
755 co.h3_first = h3dummy1;
756 h3relval = (h3dummy1 * reltol).abs();
757 } else {
758 co.h2_first = 0.0;
759 co.h3_first = 0.0;
760 }
761
762 let mut lolimit1 = 0.0;
763 let mut hilimit1 = curtime - timelist[timeindex];
764 let mut delta1 = hilimit1 - lolimit1;
765 let mut exparg = -beta * hilimit1;
766 let mut expterm = exp(exparg);
767
768 let h1twice = |hilimit1: f64, exparg: f64, expterm: f64| -> f64 {
769 if beta == 0.0 {
771 hilimit1
772 } else if hilimit1 == 0.0 {
773 0.0
774 } else {
775 (bess_i1(-exparg) + bess_i0(-exparg)) * hilimit1 * expterm - hilimit1
776 }
777 };
778 let mut h1hivalue1 = h1twice(hilimit1, exparg, expterm);
779 let mut h1dummy1 = h1hivalue1 / delta1;
780 co.h1_first = h1dummy1;
781 let h1relval = (h1dummy1 * reltol).abs();
782
783 let (mut doh1, mut doh2, mut doh3) = (true, true, true);
784 let (mut lolimit2, mut hilimit2) = (0.0, 0.0);
785
786 for i in (1..=timeindex).rev() {
787 if doh1 || doh2 || doh3 {
788 lolimit2 = lolimit1;
789 hilimit2 = hilimit1;
790 lolimit1 = hilimit2;
791 hilimit1 = curtime - timelist[i - 1];
792 delta1 = timelist[i] - timelist[i - 1];
793 exparg = -beta * hilimit1;
794 expterm = exp(exparg);
795 }
796
797 if doh1 {
798 let h1hivalue2 = h1hivalue1;
799 let h1dummy2 = h1dummy1;
800 let h1lovalue1 = h1hivalue2;
801 h1hivalue1 = h1twice(hilimit1, exparg, expterm);
802 h1dummy1 = (h1hivalue1 - h1lovalue1) / delta1;
803 co.h1[i] = h1dummy1 - h1dummy2;
804 if co.h1[i].abs() <= h1relval {
805 doh1 = false;
806 }
807 } else {
808 co.h1[i] = 0.0;
809 }
810
811 if i <= auxindex {
812 let mut besselarg = 0.0;
813 if doh2 || doh3 {
814 besselarg = if hilimit1 > t {
815 alpha * (hilimit1 * hilimit1 - t * t).sqrt()
816 } else {
817 0.0
818 };
819 }
820
821 if doh2 {
822 let h2lovalue2 = h2lovalue1;
823 let h2hivalue2 = h2hivalue1;
824 let h2dummy2 = h2dummy1;
825 h2lovalue1 = h2hivalue2;
826 let bessi1overxterm = bess_i1_x_over_x(besselarg);
827 h2hivalue1 = if alpha == 0.0 || hilimit1 < t {
828 0.0
829 } else {
830 alphasq_t * expterm * bessi1overxterm
831 };
832 h2dummy1 = twiceintlinfunc(
833 lolimit1, hilimit1, lolimit1, h2lovalue1, h2hivalue1, lolimit1, hilimit1,
834 ) / delta1;
835 co.h2[i] = h2dummy1 - h2dummy2
836 + intlinfunc(
837 lolimit2, hilimit2, h2lovalue2, h2hivalue2, lolimit2, hilimit2,
838 );
839 if co.h2[i].abs() <= h2relval {
840 doh2 = false;
841 }
842 } else {
843 co.h2[i] = 0.0;
844 }
845
846 if doh3 {
847 let h3hivalue2 = h3hivalue1;
848 let h3dummy2 = h3dummy1;
849 let h3lovalue1 = h3hivalue2;
850 let bessi0term = bess_i0(besselarg);
851 h3hivalue1 = if hilimit1 <= t || beta == 0.0 {
852 0.0
853 } else {
854 expterm * bessi0term - expbeta_t
855 };
856 h3dummy1 = intlinfunc(
857 lolimit1, hilimit1, h3lovalue1, h3hivalue1, lolimit1, hilimit1,
858 ) / delta1;
859 co.h3[i] = h3dummy1 - h3dummy2;
860 if co.h3[i].abs() <= h3relval {
861 doh3 = false;
862 }
863 } else {
864 co.h3[i] = 0.0;
865 }
866 }
867 }
868 co.aux_index = auxindex;
869}
870
871fn erfc(x: f64) -> f64 {
874 libm::erfc(x)
875}
876
877fn rc_coeffs(
884 co: &mut Coeffs,
885 cbyr: f64,
886 rclsqr: f64,
887 curtime: f64,
888 timelist: &[f64],
889 timeindex: usize,
890 reltol: f64,
891) {
892 use std::f64::consts::PI;
893 co.ensure(timeindex + 1);
894 let auxindex = timeindex;
895
896 let mut delta1 = curtime - timelist[auxindex];
897 let mut hilimit1 = delta1;
898
899 let mut h1hivalue1 = (4.0 * cbyr * hilimit1 / PI).sqrt();
900 let mut h1dummy1 = h1hivalue1 / delta1;
901 co.h1_first = h1dummy1;
902 let h1relval = (h1dummy1 * reltol).abs();
903
904 let mut temp = rclsqr / (4.0 * hilimit1);
905 let mut temp2 = if temp >= 100.0 {
906 0.0
907 } else {
908 erfc(temp.sqrt())
909 };
910 let mut temp3 = exp(-temp);
911 let temp4 = rclsqr.sqrt();
912 let temp5 = cbyr.sqrt();
913
914 let mut h2hivalue1 = if hilimit1 != 0.0 {
915 (hilimit1 + rclsqr * 0.5) * temp2 - (hilimit1 * rclsqr / PI).sqrt() * temp3
916 } else {
917 0.0
918 };
919 let mut h2dummy1 = h2hivalue1 / delta1;
920 co.h2_first = h2dummy1;
921 let h2relval = (h2dummy1 * reltol).abs();
922
923 let h3of = |hilimit1: f64, temp2: f64, temp3: f64| -> f64 {
924 if hilimit1 != 0.0 {
925 let temp = 2.0 * (hilimit1 / PI).sqrt() * temp3 - temp4 * temp2;
926 temp5 * temp
927 } else {
928 0.0
929 }
930 };
931 let mut h3hivalue1 = h3of(hilimit1, temp2, temp3);
932 let mut h3dummy1 = h3hivalue1 / delta1;
933 co.h3_first = h3dummy1;
934 let h3relval = (h3dummy1 * reltol).abs();
935
936 let (mut doh1, mut doh2, mut doh3) = (true, true, true);
937 for i in (1..=auxindex).rev() {
938 delta1 = timelist[i] - timelist[i - 1];
939 hilimit1 = curtime - timelist[i - 1];
940
941 if doh1 {
942 let h1hivalue2 = h1hivalue1;
943 let h1dummy2 = h1dummy1;
944 let h1lovalue1 = h1hivalue2;
945 h1hivalue1 = (4.0 * cbyr * hilimit1 / PI).sqrt();
946 h1dummy1 = (h1hivalue1 - h1lovalue1) / delta1;
947 co.h1[i] = h1dummy1 - h1dummy2;
948 if co.h1[i].abs() < h1relval {
949 doh1 = false;
950 }
951 } else {
952 co.h1[i] = 0.0;
953 }
954
955 if doh2 || doh3 {
956 temp = rclsqr / (4.0 * hilimit1);
957 temp2 = if temp >= 100.0 {
958 0.0
959 } else {
960 erfc(temp.sqrt())
961 };
962 temp3 = exp(-temp);
963 }
964
965 if doh2 {
966 let h2hivalue2 = h2hivalue1;
967 let h2dummy2 = h2dummy1;
968 let h2lovalue1 = h2hivalue2;
969 h2hivalue1 = if hilimit1 != 0.0 {
970 (hilimit1 + rclsqr * 0.5) * temp2 - (hilimit1 * rclsqr / PI).sqrt() * temp3
971 } else {
972 0.0
973 };
974 h2dummy1 = (h2hivalue1 - h2lovalue1) / delta1;
975 co.h2[i] = h2dummy1 - h2dummy2;
976 if co.h2[i].abs() < h2relval {
977 doh2 = false;
978 }
979 } else {
980 co.h2[i] = 0.0;
981 }
982
983 if doh3 {
984 let h3hivalue2 = h3hivalue1;
985 let h3dummy2 = h3dummy1;
986 let h3lovalue1 = h3hivalue2;
987 h3hivalue1 = h3of(hilimit1, temp2, temp3);
988 h3dummy1 = (h3hivalue1 - h3lovalue1) / delta1;
989 co.h3[i] = h3dummy1 - h3dummy2;
990 if co.h3[i].abs() < h3relval {
991 doh3 = false;
992 }
993 } else {
994 co.h3[i] = 0.0;
995 }
996 }
997 co.aux_index = auxindex;
998}
999
1000#[derive(Clone, Copy)]
1006struct Slots {
1007 ibr1_pos1: Slot,
1008 ibr1_neg1: Slot,
1009 ibr1_pos2: Slot,
1010 ibr1_neg2: Slot,
1011 ibr1_ibr1: Slot,
1012 ibr1_ibr2: Slot,
1013 ibr2_pos1: Slot,
1014 ibr2_neg1: Slot,
1015 ibr2_pos2: Slot,
1016 ibr2_neg2: Slot,
1017 ibr2_ibr1: Slot,
1018 ibr2_ibr2: Slot,
1019 pos1_ibr1: Slot,
1020 neg1_ibr1: Slot,
1021 pos2_ibr2: Slot,
1022 neg2_ibr2: Slot,
1023}
1024
1025const NO_SLOTS: Slots = Slots {
1026 ibr1_pos1: TRASH,
1027 ibr1_neg1: TRASH,
1028 ibr1_pos2: TRASH,
1029 ibr1_neg2: TRASH,
1030 ibr1_ibr1: TRASH,
1031 ibr1_ibr2: TRASH,
1032 ibr2_pos1: TRASH,
1033 ibr2_neg1: TRASH,
1034 ibr2_pos2: TRASH,
1035 ibr2_neg2: TRASH,
1036 ibr2_ibr1: TRASH,
1037 ibr2_ibr2: TRASH,
1038 pos1_ibr1: TRASH,
1039 neg1_ibr1: TRASH,
1040 pos2_ibr2: TRASH,
1041 neg2_ibr2: TRASH,
1042};
1043
1044#[derive(Clone, Copy, Default)]
1046struct Delayed {
1047 v1: f64,
1048 i1: f64,
1049 v2: f64,
1050 i2: f64,
1051}
1052
1053pub(crate) struct LtraI {
1055 p: LtraParams,
1056 pos1: Var,
1057 neg1: Var,
1058 pos2: Var,
1059 neg2: Var,
1060 br1: Var,
1061 br2: Var,
1062 slots: Slots,
1063 input1: f64,
1065 input2: f64,
1066 init_v1: f64,
1068 init_i1: f64,
1069 init_v2: f64,
1070 init_i2: f64,
1071 times: Vec<f64>,
1074 v1: Vec<f64>,
1075 i1: Vec<f64>,
1076 v2: Vec<f64>,
1077 i2: Vec<f64>,
1078 co: Coeffs,
1079 gmin: f64,
1081 fit_on: bool,
1084 slope_floor: bool,
1089 interp_control: bool,
1092 corner: bool,
1095 wave_peak: [f64; 2],
1098 fit: Option<FitState>,
1101 fit_skip: Option<&'static str>,
1104 periodic: bool,
1107 periodic_fault: Option<&'static str>,
1110}
1111
1112const PERIODIC_HORIZON_PERIODS: f64 = 1000.0;
1117
1118const V1: usize = 0;
1120const V2: usize = 1;
1121const I1: usize = 2;
1122const I2: usize = 3;
1123
1124struct FitState {
1136 fit: ExpFit,
1137 snaps: Vec<f64>,
1139 base: usize,
1140 ew: Vec<f64>,
1142}
1143
1144impl FitState {
1145 fn new(fit: ExpFit) -> Self {
1146 let n = fit.rates.len();
1147 FitState {
1148 fit,
1149 snaps: vec![0.0; 4 * n],
1150 base: 0,
1151 ew: vec![0.0; n],
1152 }
1153 }
1154
1155 fn n(&self) -> usize {
1156 self.fit.rates.len()
1157 }
1158
1159 fn snap(&self, k: usize) -> Option<&[f64]> {
1161 let n4 = 4 * self.n();
1162 let off = k.checked_sub(self.base)? * n4;
1163 self.snaps.get(off..off + n4)
1164 }
1165
1166 fn exps(&mut self, lag: f64) {
1168 for (e, x) in self.ew.iter_mut().zip(&self.fit.rates) {
1169 *e = exp(-x * lag);
1170 }
1171 }
1172
1173 fn tail(&self, kernel: usize, k: usize, wave: usize) -> Option<f64> {
1176 let n = self.n();
1177 let s = &self.snap(k)?[wave * n..(wave + 1) * n];
1178 Some(
1179 self.fit.res[kernel]
1180 .iter()
1181 .zip(&self.ew)
1182 .zip(s)
1183 .map(|((r, e), s)| r * e * s)
1184 .sum(),
1185 )
1186 }
1187
1188 fn periodic_start(&mut self, period: f64, segs: &[(f64, [f64; 4], [f64; 4])]) {
1195 let n = self.n();
1196 let mut s = vec![0.0; 4 * n];
1197 for &(h, dn, dp) in segs {
1198 for j in 0..n {
1199 let (e, a, b) = segment(self.fit.rates[j] * h);
1200 for w in 0..4 {
1201 let v = &mut s[w * n + j];
1202 *v = e * *v + h * (a * dn[w] + b * dp[w]);
1203 }
1204 }
1205 }
1206 for j in 0..n {
1207 let g = -expm1(-self.fit.rates[j] * period);
1208 for w in 0..4 {
1209 s[w * n + j] /= g;
1210 }
1211 }
1212 self.snaps = s;
1213 self.base = 0;
1214 }
1215
1216 fn advance(&mut self, h: f64, dn: [f64; 4], dp: [f64; 4]) {
1219 let n = self.n();
1220 let start = self.snaps.len() - 4 * n;
1221 self.snaps.extend_from_within(start..);
1222 let new = start + 4 * n;
1223 for j in 0..n {
1224 let (e, a, b) = segment(self.fit.rates[j] * h);
1225 for w in 0..4 {
1226 let s = &mut self.snaps[new + w * n + j];
1227 *s = e * *s + h * (a * dn[w] + b * dp[w]);
1228 }
1229 }
1230 }
1231
1232 fn prune(&mut self, keep: usize) {
1234 let dead = keep.saturating_sub(self.base);
1235 if dead >= 1024 {
1236 self.snaps.drain(..dead * 4 * self.n());
1237 self.base += dead;
1238 }
1239 }
1240}
1241
1242impl LtraI {
1243 pub fn new(p: LtraParams, nodes: [Var; 4], br: [Var; 2]) -> Self {
1246 LtraI {
1247 p,
1248 pos1: nodes[0],
1249 neg1: nodes[1],
1250 pos2: nodes[2],
1251 neg2: nodes[3],
1252 br1: br[0],
1253 br2: br[1],
1254 slots: NO_SLOTS,
1255 input1: 0.0,
1256 input2: 0.0,
1257 init_v1: 0.0,
1258 init_i1: 0.0,
1259 init_v2: 0.0,
1260 init_i2: 0.0,
1261 times: Vec::new(),
1262 v1: Vec::new(),
1263 i1: Vec::new(),
1264 v2: Vec::new(),
1265 i2: Vec::new(),
1266 co: Coeffs::default(),
1267 gmin: 0.0,
1268 fit_on: false,
1269 slope_floor: false,
1270 corner: false,
1271 interp_control: false,
1272 wave_peak: [0.0; 2],
1273 fit: None,
1274 fit_skip: None,
1275 periodic: false,
1276 periodic_fault: None,
1277 }
1278 }
1279
1280 pub fn with_interp_control(mut self, on: bool) -> Self {
1283 self.interp_control = on;
1284 self
1285 }
1286
1287 pub fn with_slope_floor(mut self, on: bool) -> Self {
1289 self.slope_floor = on;
1290 self
1291 }
1292
1293 pub fn with_fit(mut self, on: bool) -> Self {
1295 self.fit_on = on;
1296 self
1297 }
1298
1299 fn start_fit(&mut self, final_time: f64) {
1304 self.build_fit(final_time.max(f64::MIN_POSITIVE));
1305 }
1306
1307 fn build_fit(&mut self, horizon: f64) {
1309 self.fit = None;
1310 self.fit_skip = None;
1311 let p = &self.p;
1312 match p.case {
1313 LtraCase::Lc | LtraCase::Rg => {
1314 self.fit_skip = Some("no history convolution in this regime");
1315 }
1316 LtraCase::Rlc if !(p.alpha > 0.0 && p.alpha.is_finite()) => {
1317 self.fit_skip = Some("needs R/L > 0");
1318 }
1319 LtraCase::Rlc => {
1320 self.fit = Some(FitState::new(fit_rlc(p.alpha, p.td, horizon)));
1321 }
1322 LtraCase::Rc => {
1323 let w = RC_WINDOW_FRACTION * p.rclsqr;
1324 if !(w > 0.0 && w.is_finite() && p.cbyr > 0.0 && p.cbyr.is_finite()) {
1325 self.fit_skip = Some("needs R, C > 0");
1326 } else if w >= horizon {
1327 self.fit_skip =
1328 Some("the diffusion time exceeds the run: exact window covers it");
1329 } else {
1330 self.fit = Some(FitState::new(fit_rc(p.cbyr, p.rclsqr, w, horizon)));
1331 }
1332 }
1333 }
1334 }
1335
1336 fn dev(&self, wave: usize, k: usize) -> f64 {
1339 if k == 0 && !self.periodic {
1340 return 0.0;
1341 }
1342 match wave {
1343 V1 => self.v1[k] - self.init_v1,
1344 V2 => self.v2[k] - self.init_v2,
1345 I1 => self.i1[k] - self.init_i1,
1346 _ => self.i2[k] - self.init_i2,
1347 }
1348 }
1349
1350 fn aux_index(&self, time: f64) -> usize {
1353 let td = self.p.td;
1354 if td == 0.0 {
1355 return self.times.len() - 1;
1356 }
1357 if time - td <= 0.0 {
1358 return 0;
1359 }
1360 let n = self.times.partition_point(|&t| time - t >= td);
1361 let Some(found) = n.checked_sub(1) else {
1362 return 0;
1363 };
1364 if time - self.times[found] == td {
1365 found.saturating_sub(1)
1366 } else {
1367 found
1368 }
1369 }
1370
1371 fn oldest_needed(&self) -> usize {
1374 let n = self.times.len() - 1;
1375 let t = self.times[n];
1376 let k = match self.p.case {
1377 LtraCase::Rc => {
1378 let w = self.fit.as_ref().map_or(0.0, |f| f.fit.window);
1379 self.times
1380 .partition_point(|&ti| t - ti >= w)
1381 .saturating_sub(1)
1382 }
1383 _ => self.aux_index(t),
1384 };
1385 k.saturating_sub(1).min(n)
1386 }
1387
1388 fn fit_inputs_lc_rlc(&mut self, time: f64, del: Option<Delayed>) -> Result<(), &'static str> {
1394 const PRUNED: &str = "LTRA: recursive history state pruned";
1395 let n = self.times.len() - 1;
1396 let (mut in1, mut in2) = (0.0, 0.0);
1397 if self.p.case == LtraCase::Rlc {
1398 let (alpha, beta, td, admit) = (self.p.alpha, self.p.beta, self.p.td, self.p.admit);
1399 let fs = self.fit.as_mut().ok_or(PRUNED)?;
1400 let dlt = time - self.times[n];
1403 let bd = beta * dlt;
1404 let h1_first = if dlt == 0.0 {
1405 0.0
1406 } else {
1407 (i1e(bd) + i0e(bd)) - 1.0
1408 };
1409 let c_n = (i0e(bd) - 1.0) - h1_first;
1410 fs.exps(dlt);
1411 let t1 = fs.tail(0, n, V1).ok_or(PRUNED)?;
1412 let t2 = fs.tail(0, n, V2).ok_or(PRUNED)?;
1413 self.co.h1_first = h1_first;
1414 let p = &self.p;
1415 let mut d1 = c_n * self.dev(V1, n) + t1;
1416 let mut d2 = c_n * self.dev(V2, n) + t2;
1417 d1 += self.init_v1 * p.int_h1dash - self.init_v1 * h1_first;
1418 d2 += self.init_v2 * p.int_h1dash - self.init_v2 * h1_first;
1419 in1 -= d1 * admit;
1420 in2 -= d2 * admit;
1421
1422 let (mut e1, mut e2, mut f1, mut f2) = (0.0, 0.0, 0.0, 0.0);
1423 if let Some(dl) = del {
1424 let aux = self.aux_index(time);
1425 let hi = time - self.times[aux];
1426 let dp = hi - td;
1427 let h2lo = rlc_h2(td, td, alpha, beta);
1428 let z = if hi > td {
1429 alpha * (hi * hi - td * td).sqrt()
1430 } else {
1431 0.0
1432 };
1433 let damp = exp(-(beta * hi - z));
1434 let h2hi = if hi < td {
1435 0.0
1436 } else {
1437 alpha * alpha * td * damp * i1e_over_x(z)
1438 };
1439 let h2_first = twiceintlinfunc(td, hi, td, h2lo, h2hi, td, hi) / dp;
1440 let c2 = intlinfunc(td, hi, h2lo, h2hi, td, hi) - h2_first;
1441 let h3hi = if hi <= td || beta == 0.0 {
1442 0.0
1443 } else {
1444 damp * i0e(z) - exp(-beta * td)
1445 };
1446 let h3_first = intlinfunc(td, hi, 0.0, h3hi, td, hi) / dp;
1447 let c3 = h3hi - h3_first;
1448 let fs = self.fit.as_mut().ok_or(PRUNED)?;
1449 fs.exps(dp);
1450 let (s2a, s2b) = (
1451 fs.tail(1, aux, I2).ok_or(PRUNED)?,
1452 fs.tail(1, aux, I1).ok_or(PRUNED)?,
1453 );
1454 let (s3a, s3b) = (
1455 fs.tail(2, aux, V2).ok_or(PRUNED)?,
1456 fs.tail(2, aux, V1).ok_or(PRUNED)?,
1457 );
1458 e1 = (dl.i2 - self.init_i2) * h2_first + c2 * self.dev(I2, aux) + s2a;
1459 e2 = (dl.i1 - self.init_i1) * h2_first + c2 * self.dev(I1, aux) + s2b;
1460 f1 = (dl.v2 - self.init_v2) * h3_first + c3 * self.dev(V2, aux) + s3a;
1461 f2 = (dl.v1 - self.init_v1) * h3_first + c3 * self.dev(V1, aux) + s3b;
1462 }
1463 let p = &self.p;
1464 e1 += self.init_i2 * p.int_h2;
1465 e2 += self.init_i1 * p.int_h2;
1466 in1 += e1;
1467 in2 += e2;
1468 f1 += self.init_v2 * p.int_h3dash;
1469 f2 += self.init_v1 * p.int_h3dash;
1470 in1 += admit * f1;
1471 in2 += admit * f2;
1472 }
1473 let p = &self.p;
1474 match del {
1475 None => {
1476 in1 += p.attenuation * (self.init_v2 * p.admit + self.init_i2);
1477 in2 += p.attenuation * (self.init_v1 * p.admit + self.init_i1);
1478 }
1479 Some(dl) => {
1480 in1 += p.attenuation * (dl.v2 * p.admit + dl.i2);
1481 in2 += p.attenuation * (dl.v1 * p.admit + dl.i1);
1482 }
1483 }
1484 self.input1 = in1;
1485 self.input2 = in2;
1486 Ok(())
1487 }
1488
1489 fn fit_inputs_rc(&mut self, time: f64) -> Result<(), &'static str> {
1493 const PRUNED: &str = "LTRA: recursive history state pruned";
1494 let n = self.times.len() - 1;
1495 let (cbyr, rclsqr) = (self.p.cbyr, self.p.rclsqr);
1496 let w = self.fit.as_ref().ok_or(PRUNED)?.fit.window;
1497 let b = self
1498 .times
1499 .partition_point(|&t| time - t >= w)
1500 .saturating_sub(1)
1501 .min(n);
1502 let lb = time - self.times[b];
1503 let fs = self.fit.as_mut().ok_or(PRUNED)?;
1504 fs.exps(lb);
1505 let waves = [[V1, V2], [I2, I1], [V2, V1]];
1508 let mut tails = [[0.0; 2]; 3];
1509 for k in 0..3 {
1510 for q in 0..2 {
1511 tails[k][q] = fs.tail(k, b, waves[k][q]).ok_or(PRUNED)?;
1512 }
1513 }
1514 let mut firsts = [0.0; 3];
1515 let mut sums = tails;
1516 for k in 0..3 {
1517 let ln = time - self.times[n];
1519 let mut hi_val = rc_integrals(k, cbyr, rclsqr, ln).1;
1520 let mut dd_new = hi_val / ln; firsts[k] = dd_new;
1522 for i in (b + 1..=n).rev() {
1523 let lag = time - self.times[i - 1];
1524 let v = rc_integrals(k, cbyr, rclsqr, lag).1;
1525 let dd_old = (v - hi_val) / (self.times[i] - self.times[i - 1]);
1526 let c = dd_old - dd_new;
1527 for q in 0..2 {
1528 sums[k][q] += c * self.dev(waves[k][q], i);
1529 }
1530 hi_val = v;
1531 dd_new = dd_old;
1532 }
1533 let c_b = rc_integrals(k, cbyr, rclsqr, lb).0 - dd_new;
1535 for q in 0..2 {
1536 sums[k][q] += c_b * self.dev(waves[k][q], b);
1537 }
1538 }
1539 self.co.h1_first = firsts[0];
1540 self.co.h2_first = firsts[1];
1541 self.co.h3_first = firsts[2];
1542 let p = &self.p;
1543 let init = |wave: usize| match wave {
1544 V1 => self.init_v1,
1545 V2 => self.init_v2,
1546 I1 => self.init_i1,
1547 _ => self.init_i2,
1548 };
1549 let ints = [p.int_h1dash, p.int_h2, p.int_h3dash];
1550 let mut inp = [0.0; 2];
1551 for q in 0..2 {
1552 for k in 0..3 {
1553 let x0 = init(waves[k][q]);
1554 let d = sums[k][q] + x0 * ints[k] - x0 * firsts[k];
1555 if k == 0 {
1556 inp[q] -= d;
1557 } else {
1558 inp[q] += d;
1559 }
1560 }
1561 }
1562 self.input1 = inp[0];
1563 self.input2 = inp[1];
1564 Ok(())
1565 }
1566
1567 fn fit_accept(&mut self) {
1570 let k = self.times.len() - 1;
1571 let h = self.times[k] - self.times[k - 1];
1572 let dn = [V1, V2, I1, I2].map(|w| self.dev(w, k));
1573 let dp = [V1, V2, I1, I2].map(|w| self.dev(w, k - 1));
1574 let keep = self.oldest_needed();
1575 if let Some(fs) = self.fit.as_mut() {
1576 fs.advance(h, dn, dp);
1577 fs.prune(keep);
1578 }
1579 }
1580
1581 fn load_dc(&mut self, ctx: &mut LoadCtx) {
1584 let s = self.slots;
1585 let p = &self.p;
1586 match p.case {
1587 LtraCase::Rg => {
1588 ctx.add(s.ibr1_pos1, 1.0);
1589 ctx.add(s.ibr1_neg1, -1.0);
1590 ctx.add(s.ibr1_pos2, -p.cosh_lroot_gr);
1591 ctx.add(s.ibr1_neg2, p.cosh_lroot_gr);
1592 ctx.add(s.ibr1_ibr2, (1.0 + ctx.gmin) * p.r_rs_lr_gr_or_g);
1593
1594 ctx.add(s.ibr2_ibr2, p.cosh_lroot_gr);
1595 ctx.add(s.ibr2_pos2, -(1.0 + ctx.gmin) * p.r_gs_lr_gr_or_r);
1596 ctx.add(s.ibr2_neg2, (1.0 + ctx.gmin) * p.r_gs_lr_gr_or_r);
1597 ctx.add(s.ibr2_ibr1, 1.0);
1598
1599 ctx.add(s.pos1_ibr1, 1.0);
1600 ctx.add(s.neg1_ibr1, -1.0);
1601 ctx.add(s.pos2_ibr2, 1.0);
1602 ctx.add(s.neg2_ibr2, -1.0);
1603 }
1604 LtraCase::Lc | LtraCase::Rlc | LtraCase::Rc => {
1605 ctx.add(s.pos1_ibr1, 1.0);
1609 ctx.add(s.neg1_ibr1, -1.0);
1610 ctx.add(s.pos2_ibr2, 1.0);
1611 ctx.add(s.neg2_ibr2, -1.0);
1612
1613 ctx.add(s.ibr1_ibr1, 1.0);
1614 ctx.add(s.ibr1_ibr2, 1.0);
1615 ctx.add(s.ibr2_pos1, 1.0);
1616 ctx.add(s.ibr2_pos2, -1.0);
1617 ctx.add(s.ibr2_ibr1, -p.resist * p.length);
1618 }
1619 }
1620 self.input1 = 0.0;
1621 self.input2 = 0.0;
1622 }
1623
1624 fn delayed(&self, time: f64) -> Option<Delayed> {
1629 let td = self.p.td;
1630 let time_index = self.times.len() - 1;
1631 let target = time - td;
1632 let mut i: isize = time_index as isize;
1633 while i >= 0 {
1634 if self.times[i as usize] < target {
1635 break;
1636 }
1637 i -= 1;
1638 }
1639 if i == time_index as isize {
1640 i -= 1;
1641 }
1642 if i < 0 {
1643 return None;
1644 }
1645 Some(self.interp_delayed(i as usize, target))
1646 }
1647
1648 fn delayed_bisect(&self, time: f64) -> Option<Delayed> {
1653 let target = time - self.p.td;
1654 let time_index = self.times.len() - 1;
1655 let below = self.times.partition_point(|&t| t < target);
1656 if below == 0 {
1657 return None;
1658 }
1659 let i = (below - 1).min(time_index.checked_sub(1)?);
1660 Some(self.interp_delayed(i, target))
1661 }
1662
1663 fn interp_delayed(&self, isaved: usize, target: f64) -> Delayed {
1666 let t2 = self.times[isaved];
1667 let t3 = self.times[isaved + 1];
1668 let interp = self.p.interp;
1669 let quad_ok = isaved != 0 && matches!(interp, LtraInterp::Quadratic | LtraInterp::Mixed);
1670 let mut qf = (0.0, 0.0, 0.0);
1671 if quad_ok {
1672 let t1 = self.times[isaved - 1];
1673 qf = quad_interp(target, t1, t2, t3).unwrap_or((0.0, 0.0, 0.0));
1674 }
1675 let mut lf = (0.0, 0.0);
1676 if isaved == 0 || matches!(interp, LtraInterp::Mixed | LtraInterp::Linear) {
1677 lf = lin_interp(target, t2, t3).unwrap_or((0.0, 0.0));
1678 }
1679 let one = |h: &[f64]| -> f64 {
1680 let mut d = 0.0;
1681 let (mut max, mut min) = (0.0, 0.0);
1682 if quad_ok {
1683 let (a, b, c) = (h[isaved - 1], h[isaved], h[isaved + 1]);
1684 d = a * qf.0 + b * qf.1 + c * qf.2;
1685 max = f64::max(f64::max(a, b), c);
1686 min = f64::min(f64::min(a, b), c);
1687 }
1688 if interp == LtraInterp::Linear || isaved == 0 || (quad_ok && (d > max || d < min)) {
1689 if !(quad_ok && interp == LtraInterp::Quadratic) {
1692 d = h[isaved] * lf.0 + h[isaved + 1] * lf.1;
1693 }
1694 }
1695 d
1696 };
1697 Delayed {
1698 v1: one(&self.v1),
1699 i1: one(&self.i1),
1700 v2: one(&self.v2),
1701 i2: one(&self.i2),
1702 }
1703 }
1704
1705 fn inputs_lc_rlc(&mut self, del: Option<Delayed>) {
1708 let p = &self.p;
1709 let time_index = self.times.len() - 1;
1710 let (mut in1, mut in2) = (0.0, 0.0);
1711 if p.case == LtraCase::Rlc {
1712 let co = &self.co;
1713 let (mut d1, mut d2) = (0.0, 0.0);
1715 for i in (1..=time_index).rev() {
1716 if co.h1[i] != 0.0 {
1717 d1 += co.h1[i] * (self.v1[i] - self.init_v1);
1718 d2 += co.h1[i] * (self.v2[i] - self.init_v2);
1719 }
1720 }
1721 d1 += self.init_v1 * p.int_h1dash;
1722 d2 += self.init_v2 * p.int_h1dash;
1723 d1 -= self.init_v1 * co.h1_first;
1724 d2 -= self.init_v2 * co.h1_first;
1725 in1 -= d1 * p.admit;
1726 in2 -= d2 * p.admit;
1727
1728 let (mut d1, mut d2) = (0.0, 0.0);
1730 if let Some(dl) = del {
1731 d1 = (dl.i2 - self.init_i2) * co.h2_first;
1732 d2 = (dl.i1 - self.init_i1) * co.h2_first;
1733 for i in (1..=co.aux_index).rev() {
1734 if co.h2[i] != 0.0 {
1735 d1 += co.h2[i] * (self.i2[i] - self.init_i2);
1736 d2 += co.h2[i] * (self.i1[i] - self.init_i1);
1737 }
1738 }
1739 }
1740 d1 += self.init_i2 * p.int_h2;
1741 d2 += self.init_i1 * p.int_h2;
1742 in1 += d1;
1743 in2 += d2;
1744
1745 let (mut d1, mut d2) = (0.0, 0.0);
1747 if let Some(dl) = del {
1748 d1 = (dl.v2 - self.init_v2) * co.h3_first;
1749 d2 = (dl.v1 - self.init_v1) * co.h3_first;
1750 for i in (1..=co.aux_index).rev() {
1751 if co.h3[i] != 0.0 {
1752 d1 += co.h3[i] * (self.v2[i] - self.init_v2);
1753 d2 += co.h3[i] * (self.v1[i] - self.init_v1);
1754 }
1755 }
1756 }
1757 d1 += self.init_v2 * p.int_h3dash;
1758 d2 += self.init_v1 * p.int_h3dash;
1759 in1 += p.admit * d1;
1760 in2 += p.admit * d2;
1761 }
1762 match del {
1764 None => {
1765 in1 += p.attenuation * (self.init_v2 * p.admit + self.init_i2);
1766 in2 += p.attenuation * (self.init_v1 * p.admit + self.init_i1);
1767 }
1768 Some(dl) => {
1769 in1 += p.attenuation * (dl.v2 * p.admit + dl.i2);
1770 in2 += p.attenuation * (dl.v1 * p.admit + dl.i1);
1771 }
1772 }
1773 self.input1 = in1;
1774 self.input2 = in2;
1775 }
1776
1777 fn inputs_rc(&mut self) {
1779 let p = &self.p;
1780 let co = &self.co;
1781 let time_index = self.times.len() - 1;
1782 let (mut in1, mut in2) = (0.0, 0.0);
1783
1784 let (mut d1, mut d2) = (0.0, 0.0);
1785 for i in (1..=time_index).rev() {
1786 if co.h1[i] != 0.0 {
1787 d1 += co.h1[i] * (self.v1[i] - self.init_v1);
1788 d2 += co.h1[i] * (self.v2[i] - self.init_v2);
1789 }
1790 }
1791 d1 += self.init_v1 * p.int_h1dash;
1792 d2 += self.init_v2 * p.int_h1dash;
1793 d1 -= self.init_v1 * co.h1_first;
1794 d2 -= self.init_v2 * co.h1_first;
1795 in1 -= d1;
1796 in2 -= d2;
1797
1798 let (mut d1, mut d2) = (0.0, 0.0);
1799 for i in (1..=time_index).rev() {
1800 if co.h2[i] != 0.0 {
1801 d1 += co.h2[i] * (self.i2[i] - self.init_i2);
1802 d2 += co.h2[i] * (self.i1[i] - self.init_i1);
1803 }
1804 }
1805 d1 += self.init_i2 * p.int_h2;
1806 d2 += self.init_i1 * p.int_h2;
1807 d1 -= self.init_i2 * co.h2_first;
1808 d2 -= self.init_i1 * co.h2_first;
1809 in1 += d1;
1810 in2 += d2;
1811
1812 let (mut d1, mut d2) = (0.0, 0.0);
1813 for i in (1..=time_index).rev() {
1814 if co.h3[i] != 0.0 {
1815 d1 += co.h3[i] * (self.v2[i] - self.init_v2);
1816 d2 += co.h3[i] * (self.v1[i] - self.init_v1);
1817 }
1818 }
1819 d1 += self.init_v2 * p.int_h3dash;
1820 d2 += self.init_v1 * p.int_h3dash;
1821 d1 -= self.init_v2 * co.h3_first;
1822 d2 -= self.init_v1 * co.h3_first;
1823 in1 += d1;
1824 in2 += d2;
1825
1826 self.input1 = in1;
1827 self.input2 = in2;
1828 }
1829}
1830
1831impl Device for LtraI {
1832 fn par_load(&self) -> bool {
1833 true
1834 }
1835
1836 fn disto_setup(&mut self, _ctx: &DistoSetupCtx) -> Result<(), &'static str> {
1838 Ok(())
1839 }
1840
1841 fn tran_notes(&self) -> Vec<String> {
1845 if !self.fit_on {
1846 return Vec::new();
1847 }
1848 let name = &self.p.name;
1849 let Some(fs) = &self.fit else {
1850 let why = self.fit_skip.unwrap_or("not started");
1851 return vec![format!(
1852 "`{name}`: ltrafit not applied ({why}); exact convolution"
1853 )];
1854 };
1855 let f = &fs.fit;
1856 let (labels, delayed) = match self.p.case {
1857 LtraCase::Rc => (["h1", "h2", "h3"], false),
1858 _ => (["h1'", "h2", "h3'"], true),
1859 };
1860 let mut hist = [0.0; 3];
1861 if let Some(&te) = self.times.last() {
1862 let n = self.times.len() - 1;
1863 let w = f.window;
1864 let lags0 = self.times[..n].iter().map(|t| te - t).filter(|l| *l >= w);
1865 hist[0] = f.err_at(0, lags0);
1866 for (k, h) in hist.iter_mut().enumerate().skip(1) {
1867 *h = if delayed {
1868 let td = self.p.td;
1869 let lags = self.times.iter().map(|t| te - td - t).filter(|l| *l > 0.0);
1870 f.err_at(k, lags)
1871 } else {
1872 f.err_at(
1873 k,
1874 self.times[..n].iter().map(|t| te - t).filter(|l| *l >= w),
1875 )
1876 };
1877 }
1878 }
1879 let errs: Vec<String> = (0..3)
1880 .map(|k| {
1881 format!(
1882 "{} {:.1e}/{:.1e}/{:.1e}",
1883 labels[k], f.err[k].max_rel, f.err[k].l1_rel, hist[k]
1884 )
1885 })
1886 .collect();
1887 let regime = match self.p.case {
1888 LtraCase::Rc => format!(
1889 "RC, RCl²={:.3e}s, exact window {:.3e}s",
1890 self.p.rclsqr, f.window
1891 ),
1892 _ => format!("RLC, T={:.3e}s, R/2L={:.3e}/s", self.p.td, self.p.alpha),
1893 };
1894 vec![format!(
1895 "`{name}`: ltrafit {regime}: {} real poles ({} panels x {} Gauss points); \
1896 kernel error max/L1 on [{:.3e}s, {:.3e}s] / max at the {} history lags: {}",
1897 f.rates.len(),
1898 f.panels,
1899 f.order,
1900 f.window,
1901 f.horizon,
1902 self.times.len(),
1903 errs.join(", ")
1904 )]
1905 }
1906
1907 fn alloc(&mut self, mb: &mut MatrixBuilder) {
1908 let (p1, n1, p2, n2, b1, b2) = (
1909 self.pos1, self.neg1, self.pos2, self.neg2, self.br1, self.br2,
1910 );
1911 self.slots = Slots {
1912 ibr1_pos1: mb.slot(b1, p1),
1913 ibr1_neg1: mb.slot(b1, n1),
1914 ibr1_pos2: mb.slot(b1, p2),
1915 ibr1_neg2: mb.slot(b1, n2),
1916 ibr1_ibr1: mb.slot(b1, b1),
1917 ibr1_ibr2: mb.slot(b1, b2),
1918 ibr2_pos1: mb.slot(b2, p1),
1919 ibr2_neg1: mb.slot(b2, n1),
1920 ibr2_pos2: mb.slot(b2, p2),
1921 ibr2_neg2: mb.slot(b2, n2),
1922 ibr2_ibr1: mb.slot(b2, b1),
1923 ibr2_ibr2: mb.slot(b2, b2),
1924 pos1_ibr1: mb.slot(p1, b1),
1925 neg1_ibr1: mb.slot(n1, b1),
1926 pos2_ibr2: mb.slot(p2, b2),
1927 neg2_ibr2: mb.slot(n2, b2),
1928 };
1929 mb.slot(p1, p1);
1932 mb.slot(n1, n1);
1933 mb.slot(p2, p2);
1934 mb.slot(n2, n2);
1935 }
1936
1937 fn load(&mut self, ctx: &mut LoadCtx) {
1938 if ctx.is_dc() {
1939 self.gmin = ctx.gmin;
1940 self.load_dc(ctx);
1941 ctx.add_rhs(self.br1, self.input1);
1942 ctx.add_rhs(self.br2, self.input2);
1943 return;
1944 }
1945 if self.times.is_empty() {
1946 ctx.fault = Some("LTRA: transient load before the operating point was stored");
1949 return;
1950 }
1951 if self.periodic {
1952 if let Some(f) = self.periodic_fault {
1953 ctx.fault = Some(f);
1954 return;
1955 }
1956 if self.fit.is_none() && matches!(self.p.case, LtraCase::Rlc | LtraCase::Rc) {
1957 ctx.fault = Some(
1960 "LTRA: a lossy (RLC/RC) line in .pss needs the recursive convolution \
1961 (.options ltrafit)",
1962 );
1963 return;
1964 }
1965 }
1966 let first = matches!(ctx.mode, InitMode::Tran | InitMode::Pred);
1967 let time_index = self.times.len() - 1;
1968 let tdover = matches!(self.p.case, LtraCase::Lc | LtraCase::Rlc)
1970 && (ctx.time > self.p.td || self.periodic);
1971 let mut del = None;
1972 if first && self.fit.is_some() {
1973 if tdover {
1975 match self.delayed_bisect(ctx.time) {
1976 Some(d) => del = Some(d),
1977 None => {
1978 ctx.fault = Some("LTRA: cannot find delayed timepoint");
1979 return;
1980 }
1981 }
1982 }
1983 } else if first {
1984 match self.p.case {
1985 LtraCase::Rlc => rlc_coeffs(
1986 &mut self.co,
1987 self.p.td,
1988 self.p.alpha,
1989 self.p.beta,
1990 ctx.time,
1991 &self.times,
1992 time_index,
1993 self.p.chop_reltol,
1994 ),
1995 LtraCase::Rc => rc_coeffs(
1996 &mut self.co,
1997 self.p.cbyr,
1998 self.p.rclsqr,
1999 ctx.time,
2000 &self.times,
2001 time_index,
2002 self.p.chop_reltol,
2003 ),
2004 LtraCase::Lc | LtraCase::Rg => {}
2005 }
2006 if tdover {
2007 match self.delayed(ctx.time) {
2008 Some(d) => del = Some(d),
2009 None => {
2010 ctx.fault = Some("LTRA: cannot find delayed timepoint");
2011 return;
2012 }
2013 }
2014 }
2015 }
2016
2017 if self.p.case == LtraCase::Rg {
2018 self.load_dc(ctx);
2019 ctx.add_rhs(self.br1, self.input1);
2020 ctx.add_rhs(self.br2, self.input2);
2021 return;
2022 }
2023
2024 if ctx.mode == InitMode::Tran && !ctx.uic && !self.periodic {
2026 self.init_v1 = ctx.sol[self.pos1] - ctx.sol[self.neg1];
2027 self.init_v2 = ctx.sol[self.pos2] - ctx.sol[self.neg2];
2028 self.init_i1 = ctx.sol[self.br1];
2029 self.init_i2 = ctx.sol[self.br2];
2030 }
2031
2032 if first && self.fit.is_some() {
2035 let r = match self.p.case {
2036 LtraCase::Lc | LtraCase::Rlc => self.fit_inputs_lc_rlc(ctx.time, del),
2037 LtraCase::Rc => self.fit_inputs_rc(ctx.time),
2038 LtraCase::Rg => Ok(()),
2039 };
2040 if let Err(e) = r {
2041 ctx.fault = Some(e);
2042 return;
2043 }
2044 }
2045
2046 let s = self.slots;
2048 let p = &self.p;
2049 match p.case {
2050 LtraCase::Rlc | LtraCase::Lc => {
2051 if p.case == LtraCase::Rlc {
2052 let d = p.admit * self.co.h1_first;
2053 ctx.add(s.ibr1_pos1, d);
2054 ctx.add(s.ibr1_neg1, -d);
2055 ctx.add(s.ibr2_pos2, d);
2056 ctx.add(s.ibr2_neg2, -d);
2057 }
2058 ctx.add(s.ibr1_pos1, p.admit);
2059 ctx.add(s.ibr1_neg1, -p.admit);
2060 ctx.add(s.ibr1_ibr1, -1.0);
2061 ctx.add(s.pos1_ibr1, 1.0);
2062 ctx.add(s.neg1_ibr1, -1.0);
2063
2064 ctx.add(s.ibr2_pos2, p.admit);
2065 ctx.add(s.ibr2_neg2, -p.admit);
2066 ctx.add(s.ibr2_ibr2, -1.0);
2067 ctx.add(s.pos2_ibr2, 1.0);
2068 ctx.add(s.neg2_ibr2, -1.0);
2069 }
2070 LtraCase::Rc => {
2071 ctx.add(s.ibr1_ibr1, -1.0);
2072 ctx.add(s.pos1_ibr1, 1.0);
2073 ctx.add(s.neg1_ibr1, -1.0);
2074
2075 ctx.add(s.ibr2_ibr2, -1.0);
2076 ctx.add(s.pos2_ibr2, 1.0);
2077 ctx.add(s.neg2_ibr2, -1.0);
2078
2079 let d = self.co.h1_first;
2080 ctx.add(s.ibr1_pos1, d);
2081 ctx.add(s.ibr1_neg1, -d);
2082 ctx.add(s.ibr2_pos2, d);
2083 ctx.add(s.ibr2_neg2, -d);
2084
2085 let d = self.co.h2_first;
2086 ctx.add(s.ibr1_ibr2, -d);
2087 ctx.add(s.ibr2_ibr1, -d);
2088
2089 let d = self.co.h3_first;
2090 ctx.add(s.ibr1_pos2, -d);
2091 ctx.add(s.ibr1_neg2, d);
2092 ctx.add(s.ibr2_pos1, -d);
2093 ctx.add(s.ibr2_neg1, d);
2094 }
2095 LtraCase::Rg => {} }
2097
2098 if first && self.fit.is_none() {
2099 match self.p.case {
2100 LtraCase::Lc | LtraCase::Rlc => self.inputs_lc_rlc(del),
2101 LtraCase::Rc => self.inputs_rc(),
2102 LtraCase::Rg => {}
2103 }
2104 }
2105
2106 ctx.add_rhs(self.br1, self.input1);
2107 ctx.add_rhs(self.br2, self.input2);
2108 }
2109
2110 fn trunc(&self, t: &TerrCtx, time_step: &mut f64) {
2115 let p = &self.p;
2116 if matches!(p.case, LtraCase::Lc | LtraCase::Rlc) {
2117 if p.step_limit {
2118 *time_step = time_step.min(p.td);
2119 } else if self.times.len() >= 2 {
2120 let idx = self.times.len() - 1;
2121 let sol = t.sol;
2122 let a = p.admit;
2123 let att = p.attenuation;
2124 let i1 = ((sol[self.pos2] - sol[self.neg2]) * a + sol[self.br2]) * att;
2125 let i2 = (self.v2[idx] * a + self.i2[idx]) * att;
2126 let i3 = (self.v2[idx - 1] * a + self.i2[idx - 1]) * att;
2127 let i4 = ((sol[self.pos1] - sol[self.neg1]) * a + sol[self.br1]) * att;
2128 let i5 = (self.v1[idx] * a + self.i1[idx]) * att;
2129 let i6 = (self.v1[idx - 1] * a + self.i1[idx - 1]) * att;
2130 let h_now = t.time - self.times[idx];
2131 let h_old = self.times[idx] - self.times[idx - 1];
2132 let d1 = (i1 - i2) / h_now;
2133 let d2 = (i2 - i3) / h_old;
2134 let d3 = (i4 - i5) / h_now;
2135 let d4 = (i5 - i6) / h_old;
2136 if (d1 - d2).abs() >= p.reltol * f64::max(d1.abs(), d2.abs()) + p.abstol
2137 || (d3 - d4).abs() >= p.reltol * f64::max(d3.abs(), d4.abs()) + p.abstol
2138 {
2139 *time_step = time_step.min(p.td);
2142 }
2143 }
2144 }
2145 if p.case == LtraCase::Rlc && !p.trunc_dont_cut {
2146 *time_step = time_step.min(p.max_safe_step);
2147 }
2148 if self.interp_control
2149 && matches!(p.case, LtraCase::Rlc | LtraCase::Lc)
2150 && let Some(err) = self.interp_error(t)
2151 {
2152 let peak = f64::max(self.wave_peak[0], self.wave_peak[1]);
2155 let tol = t.trtol * (t.reltol * peak + t.abstol);
2156 let h = t.time - self.times[self.times.len() - 1];
2157 if err > 0.0 && h > 0.0 {
2158 *time_step = time_step.min(h * (tol / err).sqrt());
2159 }
2160 }
2161 }
2162
2163 fn load_ac(&mut self, ctx: &mut AcCtx) {
2167 let p = &self.p;
2168 let s = self.slots;
2169 let w = ctx.omega;
2170 let (y0_r, y0_i, lambda_r, lambda_i) = match p.case {
2171 LtraCase::Lc => (p.admit, 0.0, 0.0, p.td * w),
2172 LtraCase::Rlc => {
2173 let theta = 0.5 * atan(p.resist / (w * p.induct));
2174 let zmag = (p.resist * p.resist + w * w * p.induct * p.induct).sqrt();
2175 let mag = (w * p.capac / zmag).sqrt();
2176 let y0_r = mag * cos(theta);
2177 let y0_i = mag * sin(theta);
2178 let theta = std::f64::consts::FRAC_PI_2 - theta;
2179 let mag = mag * zmag;
2180 (y0_r, y0_i, mag * cos(theta), mag * sin(theta))
2181 }
2182 LtraCase::Rc => {
2183 let y = (0.5 * w * p.cbyr).sqrt();
2184 let l = (0.5 * w * p.resist * p.capac).sqrt();
2185 (y, y, l, l)
2186 }
2187 LtraCase::Rg => {
2188 let g1 = (1.0 + self.gmin) * p.r_rs_lr_gr_or_g;
2189 let g2 = (1.0 + self.gmin) * p.r_gs_lr_gr_or_r;
2190 ctx.add_g(s.ibr1_pos1, 1.0);
2191 ctx.add_g(s.ibr1_neg1, -1.0);
2192 ctx.add_g(s.ibr1_pos2, -p.cosh_lroot_gr);
2193 ctx.add_g(s.ibr1_neg2, p.cosh_lroot_gr);
2194 ctx.add_g(s.ibr1_ibr2, g1);
2195 ctx.add_g(s.ibr2_ibr2, p.cosh_lroot_gr);
2196 ctx.add_g(s.ibr2_pos2, -g2);
2197 ctx.add_g(s.ibr2_neg2, g2);
2198 ctx.add_g(s.ibr2_ibr1, 1.0);
2199 ctx.add_g(s.pos1_ibr1, 1.0);
2200 ctx.add_g(s.neg1_ibr1, -1.0);
2201 ctx.add_g(s.pos2_ibr2, 1.0);
2202 ctx.add_g(s.neg2_ibr2, -1.0);
2203 return;
2204 }
2205 };
2206 let exparg_r = -lambda_r * p.length;
2207 let exparg_i = -lambda_i * p.length;
2208 let explambda_r = exp(exparg_r) * cos(exparg_i);
2209 let explambda_i = exp(exparg_r) * sin(exparg_i);
2210 let y0exp_r = y0_r * explambda_r - y0_i * explambda_i;
2211 let y0exp_i = y0_r * explambda_i + y0_i * explambda_r;
2212
2213 let both = |ctx: &mut AcCtx, sl: Slot, re: f64, im: f64| {
2214 ctx.add_g(sl, re);
2215 ctx.add_b(sl, im);
2216 };
2217 both(ctx, s.ibr1_pos1, y0_r, y0_i);
2218 both(ctx, s.ibr1_neg1, -y0_r, -y0_i);
2219 ctx.add_g(s.ibr1_ibr1, -1.0);
2220 both(ctx, s.ibr1_pos2, -y0exp_r, -y0exp_i);
2221 both(ctx, s.ibr1_neg2, y0exp_r, y0exp_i);
2222 both(ctx, s.ibr1_ibr2, -explambda_r, -explambda_i);
2223
2224 both(ctx, s.ibr2_pos2, y0_r, y0_i);
2225 both(ctx, s.ibr2_neg2, -y0_r, -y0_i);
2226 ctx.add_g(s.ibr2_ibr2, -1.0);
2227 both(ctx, s.ibr2_pos1, -y0exp_r, -y0exp_i);
2228 both(ctx, s.ibr2_neg1, y0exp_r, y0exp_i);
2229 both(ctx, s.ibr2_ibr1, -explambda_r, -explambda_i);
2230
2231 ctx.add_g(s.pos1_ibr1, 1.0);
2232 ctx.add_g(s.neg1_ibr1, -1.0);
2233 ctx.add_g(s.pos2_ibr2, 1.0);
2234 ctx.add_g(s.neg2_ibr2, -1.0);
2235 }
2236
2237 fn compact_ready(&self) -> Option<bool> {
2238 self.straight_tail()
2239 }
2240
2241 fn compact(&mut self) {
2242 self.drop_middle();
2243 }
2244
2245 fn history_len(&self) -> Option<usize> {
2246 Some(self.times.len())
2247 }
2248
2249 fn accept(&mut self, ctx: &mut AcceptCtx) {
2254 if ctx.first {
2255 self.wave_peak = [0.0; 2];
2257 self.times.clear();
2258 self.v1.clear();
2259 self.i1.clear();
2260 self.v2.clear();
2261 self.i2.clear();
2262 self.periodic = false;
2263 self.periodic_fault = None;
2264 if let Some(past) = ctx.periodic {
2265 self.seed_periodic(past, ctx);
2266 return;
2267 }
2268 }
2269 let sol = ctx.sol;
2270 self.times.push(ctx.time);
2271 self.v1.push(sol[self.pos1] - sol[self.neg1]);
2272 self.v2.push(sol[self.pos2] - sol[self.neg2]);
2273 self.i1.push(sol[self.br1]);
2274 self.i2.push(sol[self.br2]);
2275
2276 let idx = self.times.len() - 1;
2277 if self.fit_on {
2278 if ctx.first {
2279 self.start_fit(ctx.final_time);
2280 } else if idx > 0 && self.fit.is_some() {
2281 self.fit_accept();
2282 }
2283 }
2284 let p = &self.p;
2285 let wave = |v: f64, i: f64| ((v + i * p.imped) * p.attenuation).abs();
2286 self.wave_peak[0] = self.wave_peak[0].max(wave(self.v1[idx], self.i1[idx]));
2287 self.wave_peak[1] = self.wave_peak[1].max(wave(self.v2[idx], self.i2[idx]));
2288 if idx == 0 {
2289 return;
2290 }
2291 self.corner = self.slope_changed(idx, ctx.reltol, ctx.abstol);
2292 if self.corner {
2293 let t = self.times[idx - 1] + self.p.td;
2294 ctx.set_slope_break(t);
2295 }
2296 }
2297}
2298
2299impl LtraI {
2300 fn interp_error(&self, t: &TerrCtx) -> Option<f64> {
2313 let p = &self.p;
2314 let idx = self.times.len().checked_sub(1)?;
2315 if idx < 1 {
2316 return None;
2317 }
2318 let wave = |v: f64, i: f64| (v + i * p.imped) * p.attenuation;
2319 let (t2, t1, t0) = (self.times[idx - 1], self.times[idx], t.time);
2320 let h = t0 - t1;
2321 let second = |a: f64, b: f64, c: f64| ((c - b) / h - (b - a) / (t1 - t2)) / (t0 - t2);
2322 let d1 = second(
2323 wave(self.v1[idx - 1], self.i1[idx - 1]),
2324 wave(self.v1[idx], self.i1[idx]),
2325 wave(t.sol[self.pos1] - t.sol[self.neg1], t.sol[self.br1]),
2326 );
2327 let d2 = second(
2328 wave(self.v2[idx - 1], self.i2[idx - 1]),
2329 wave(self.v2[idx], self.i2[idx]),
2330 wave(t.sol[self.pos2] - t.sol[self.neg2], t.sol[self.br2]),
2331 );
2332 let err = 0.125 * h * h * f64::max(d1.abs(), d2.abs());
2333 err.is_finite().then_some(err)
2334 }
2335
2336 fn straight_tail(&self) -> Option<bool> {
2343 let n = self.times.len();
2344 if n < 3 || self.fit_on || self.periodic {
2345 return None;
2346 }
2347 if self.slope_floor && self.corner {
2356 return Some(false);
2357 }
2358 let (t1, t2, t3) = (self.times[n - 3], self.times[n - 2], self.times[n - 1]);
2359 let (rt, at) = (self.p.st_line_reltol, self.p.st_line_abstol);
2360 let straight =
2361 |y: &[f64]| straight_line_check(t1, y[n - 3], t2, y[n - 2], t3, y[n - 1], rt, at);
2362 Some(straight(&self.v1) && straight(&self.v2) && straight(&self.i1) && straight(&self.i2))
2363 }
2364
2365 fn drop_middle(&mut self) {
2368 let n = self.times.len();
2369 debug_assert!(n >= 3);
2370 for h in [
2371 &mut self.times,
2372 &mut self.v1,
2373 &mut self.v2,
2374 &mut self.i1,
2375 &mut self.i2,
2376 ] {
2377 h.swap_remove(n - 2);
2378 }
2379 }
2380
2381 fn slope_changed(&self, idx: usize, reltol: f64, abstol: f64) -> bool {
2386 let p = &self.p;
2387 let wave1 = |k: usize| (self.v1[k] + self.i1[k] * p.imped) * p.attenuation;
2388 let wave2 = |k: usize| (self.v2[k] + self.i2[k] * p.imped) * p.attenuation;
2389 let v1 = wave1(idx);
2390 let v2 = wave1(idx - 1);
2391 let v3 = if idx < 2 { v2 } else { wave1(idx - 2) };
2392 let v4 = wave2(idx);
2393 let v5 = wave2(idx - 1);
2394 let v6 = if idx < 2 { v5 } else { wave2(idx - 2) };
2395 let h1 = self.times[idx] - self.times[idx - 1];
2396 let d1 = (v1 - v2) / h1;
2397 let d3 = (v4 - v5) / h1;
2398 let (d2, d4) = if idx < 2 {
2402 (0.0, 0.0)
2403 } else {
2404 let h2 = self.times[idx - 1] - self.times[idx - 2];
2405 ((v2 - v3) / h2, (v5 - v6) / h2)
2406 };
2407 let peak = f64::max(self.wave_peak[0], self.wave_peak[1]);
2415 let check = |a: f64, b: f64, c: f64| {
2416 let spread = f64::max(f64::max(a, b), c) - f64::min(f64::min(a, b), c);
2417 spread >= (50.0 * (reltol / 3.0 * (a + b + c) + abstol)).abs()
2418 && (!self.slope_floor || spread >= reltol * peak)
2419 };
2420 let test1 = (d1 - d2).abs() >= p.reltol * f64::max(d1.abs(), d2.abs()) + p.abstol
2421 && check(v1, v2, v3);
2422 let test2 = (d3 - d4).abs() >= p.reltol * f64::max(d3.abs(), d4.abs()) + p.abstol
2423 && check(v4, v5, v6);
2424 test1 || test2
2425 }
2426
2427 fn seed_periodic(&mut self, past: &PeriodicPast, ctx: &mut AcceptCtx) {
2440 self.periodic = true;
2441 let t0 = ctx.time;
2442 let period = past.period;
2443 let mut reach = self.p.td;
2444 if self.fit_on && self.p.case == LtraCase::Rc {
2445 reach = reach.max(RC_WINDOW_FRACTION * self.p.rclsqr);
2446 }
2447 let from = t0 - reach - 1e-6 * period;
2450 let Some(samples) = past.samples(t0, from, ctx.sol) else {
2451 self.periodic_fault = Some("LTRA: the delay spans too many periods for .pss");
2452 return;
2453 };
2454 let (p1, n1, p2, n2, b1, b2) = (
2455 self.pos1, self.neg1, self.pos2, self.neg2, self.br1, self.br2,
2456 );
2457 for (t, x) in samples.into_iter().chain([(t0, ctx.sol)]) {
2458 self.times.push(t);
2459 self.v1.push(x[p1] - x[n1]);
2460 self.v2.push(x[p2] - x[n2]);
2461 self.i1.push(x[b1]);
2462 self.i2.push(x[b2]);
2463 }
2464 let per = past.phase.len() + 1;
2466 let mean = |w: &[f64]| {
2467 let mut acc = 0.0;
2468 for k in 1..=per {
2469 acc += 0.5 * (w[k] + w[k - 1]) * (self.times[k] - self.times[k - 1]);
2470 }
2471 acc / (self.times[per] - self.times[0])
2472 };
2473 self.init_v1 = mean(&self.v1);
2474 self.init_v2 = mean(&self.v2);
2475 self.init_i1 = mean(&self.i1);
2476 self.init_i2 = mean(&self.i2);
2477 if matches!(self.p.case, LtraCase::Rlc | LtraCase::Rc) {
2478 let i = 0.5 * (self.init_i1 - self.init_i2);
2497 let drop = self.p.resist * self.p.length * i;
2498 let avg = 0.5 * (self.init_v1 + self.init_v2);
2499 let (s1, s2) = (
2500 avg + 0.5 * drop - self.init_v1,
2501 avg - 0.5 * drop - self.init_v2,
2502 );
2503 let last = self.times.len() - 1;
2504 for k in 0..last {
2505 self.v1[k] += s1;
2506 self.v2[k] += s2;
2507 }
2508 self.init_v1 += s1;
2509 self.init_v2 += s2;
2510 }
2511 if self.fit_on {
2512 self.build_fit(ctx.final_time.max(PERIODIC_HORIZON_PERIODS * period));
2513 let seg = |k: usize| {
2514 (
2515 self.times[k] - self.times[k - 1],
2516 [V1, V2, I1, I2].map(|w| self.dev(w, k)),
2517 [V1, V2, I1, I2].map(|w| self.dev(w, k - 1)),
2518 )
2519 };
2520 let first: Vec<_> = (1..=per).map(seg).collect();
2521 let rest: Vec<_> = (1..self.times.len()).map(seg).collect();
2522 let keep = self.oldest_needed();
2523 let span = self.times[per] - self.times[0];
2524 if let Some(fs) = self.fit.as_mut() {
2525 fs.periodic_start(span, &first);
2526 for (h, dn, dp) in rest {
2527 fs.advance(h, dn, dp);
2528 }
2529 fs.prune(keep);
2530 }
2531 }
2532 for idx in 1..self.times.len() {
2533 let t = self.times[idx - 1] + self.p.td;
2534 if t > t0 + ctx.min_break
2535 && t < ctx.final_time - ctx.min_break
2536 && self.slope_changed(idx, ctx.reltol, ctx.abstol)
2537 {
2538 ctx.set_slope_break(t);
2539 }
2540 }
2541 }
2542}
2543
2544#[cfg(test)]
2546pub(crate) fn tests_bess(x: f64) -> (f64, f64) {
2547 (bess_i0(x), bess_i1(x))
2548}
2549
2550#[cfg(test)]
2551mod tests {
2552 use super::*;
2553
2554 fn close(a: f64, b: f64, rel: f64) -> bool {
2555 (a - b).abs() <= rel * f64::max(a.abs(), b.abs())
2556 }
2557
2558 fn rlc_h1dash(time: f64, alpha: f64, beta: f64) -> f64 {
2561 if alpha == 0.0 {
2562 return 0.0;
2563 }
2564 (bess_i1(alpha * time) - bess_i0(alpha * time)) * alpha * (-beta * time).exp()
2565 }
2566
2567 fn rlc_h3dash(time: f64, t: f64, alpha: f64, beta: f64) -> f64 {
2569 if alpha == 0.0 || time < t {
2570 return 0.0;
2571 }
2572 let besselarg = if time != t {
2573 alpha * (time * time - t * t).sqrt()
2574 } else {
2575 0.0
2576 };
2577 (alpha * time * bess_i1_x_over_x(besselarg) - bess_i0(besselarg))
2578 * alpha
2579 * (-beta * time).exp()
2580 }
2581
2582 fn rlc_h1dash_twice_int(time: f64, beta: f64) -> f64 {
2584 if beta == 0.0 {
2585 return time;
2586 }
2587 let arg = beta * time;
2588 if arg == 0.0 {
2589 return 0.0;
2590 }
2591 (bess_i1(arg) + bess_i0(arg)) * time * (-arg).exp() - time
2592 }
2593
2594 fn rlc_h3dash_int(time: f64, t: f64, beta: f64) -> f64 {
2596 if time <= t || beta == 0.0 {
2597 return 0.0;
2598 }
2599 (-beta * time).exp() * bess_i0(beta * (time * time - t * t).sqrt()) - (-beta * t).exp()
2600 }
2601
2602 #[test]
2603 fn bessel_fits_match_reference_values() {
2604 assert!(close(bess_i0(1.0), 1.266_065_877_752_008, 2e-7));
2606 assert!(close(bess_i1(1.0), 0.565_159_103_992_485, 2e-7));
2607 assert!(close(bess_i0(5.0), 27.239_871_823_604_44, 2e-7));
2608 assert!(close(bess_i1(5.0), 24.335_642_142_450_52, 2e-7));
2609 assert!(close(bess_i1(-5.0), -24.335_642_142_450_52, 2e-7));
2610 assert!(close(
2611 bess_i1_x_over_x(5.0),
2612 24.335_642_142_450_52 / 5.0,
2613 2e-7
2614 ));
2615 assert_eq!(bess_i1_x_over_x(0.0), 0.5);
2616 assert!(close(bess_i1_x_over_x(2.0), bess_i1(2.0) / 2.0, 1e-15));
2617 }
2618
2619 #[test]
2620 fn linear_integrals_are_exact() {
2621 assert!(close(intlinfunc(0.0, 2.0, 1.0, 3.0, 0.0, 2.0), 4.0, 1e-15));
2623 assert!(close(intlinfunc(0.5, 1.5, 1.0, 3.0, 0.0, 2.0), 2.0, 1e-15));
2624 let v = twiceintlinfunc(0.0, 2.0, 0.0, 1.0, 3.0, 0.0, 2.0);
2626 assert!(close(v, 2.0 + 4.0 / 3.0, 1e-14), "{v}");
2627 let v = twiceintlinfunc(1.0, 2.0, 1.0, 1.0, 3.0, 0.0, 2.0);
2630 assert!(close(v, 7.0 / 6.0, 1e-14), "{v}");
2631 }
2632
2633 #[test]
2634 fn interpolation_reproduces_polynomials() {
2635 let (c1, c2, c3) = quad_interp(1.7, 0.0, 1.0, 3.0).unwrap();
2636 let f = |t: f64| 2.0 - t + 0.5 * t * t;
2637 let q = c1 * f(0.0) + c2 * f(1.0) + c3 * f(3.0);
2638 assert!(close(q, f(1.7), 1e-14));
2639 assert_eq!(quad_interp(1.0, 0.0, 1.0, 3.0), Some((0.0, 1.0, 0.0)));
2640 assert_eq!(quad_interp(0.5, 1.0, 1.0, 3.0), None);
2641 let (l1, l2) = lin_interp(0.25, 0.0, 1.0).unwrap();
2642 assert_eq!((l1, l2), (0.75, 0.25));
2643 }
2644
2645 #[test]
2646 fn closed_form_integrals_differentiate_to_the_responses() {
2647 let (alpha, t) = (1.1e7, 4.38e-9);
2649 for &time in &[1e-9, 5e-9, 2e-8] {
2650 let h = time * 1e-4;
2651 let f = |x: f64| rlc_h1dash_twice_int(x, alpha);
2652 let second = (f(time + h) - 2.0 * f(time) + f(time - h)) / (h * h);
2653 assert!(
2654 close(second, rlc_h1dash(time, alpha, alpha), 1e-4),
2655 "{second} vs {}",
2656 rlc_h1dash(time, alpha, alpha)
2657 );
2658 if time > t {
2659 let g = |x: f64| rlc_h3dash_int(x, t, alpha);
2660 let first = (g(time + h) - g(time - h)) / (2.0 * h);
2661 assert!(close(first, rlc_h3dash(time, t, alpha, alpha), 1e-4));
2662 }
2663 }
2664 }
2665
2666 #[test]
2667 fn rlc_coefficients_telescope_to_the_last_divided_difference() {
2668 let (alpha, td) = (1.1e7, 4.38e-9);
2671 let times: Vec<f64> = (0..40)
2672 .map(|k| k as f64 * 0.1e-9 + (k * k) as f64 * 1e-12)
2673 .collect();
2674 let n = times.len() - 1;
2675 let cur = times[n] + 0.07e-9;
2676 let mut co = Coeffs::default();
2677 rlc_coeffs(&mut co, td, alpha, alpha, cur, ×, n, 0.0);
2678 let sum: f64 = co.h1_first + co.h1[1..=n].iter().sum::<f64>();
2679 let f = |x: f64| rlc_h1dash_twice_int(x, alpha);
2680 let last = (f(cur - times[0]) - f(cur - times[1])) / (times[1] - times[0]);
2681 assert!(close(sum, last, 1e-9), "{sum} vs {last}");
2682 let d = cur - times[n];
2684 assert!(close(co.h1_first, f(d) / d, 1e-15));
2685 assert!(cur - times[co.aux_index] > td);
2687 assert!(cur - times[co.aux_index + 1] <= td);
2688 assert!(co.h2_first > 0.0);
2689 }
2690
2691 #[test]
2692 fn rc_coefficients_telescope() {
2693 let (cbyr, rclsqr) = (3.65e-12 / 0.2, 0.2 * 3.65e-12 * 24.0 * 24.0);
2694 let times: Vec<f64> = (0..30).map(|k| k as f64 * 0.1e-9).collect();
2695 let n = times.len() - 1;
2696 let cur = times[n] + 0.05e-9;
2697 let mut co = Coeffs::default();
2698 rc_coeffs(&mut co, cbyr, rclsqr, cur, ×, n, 0.0);
2699 let f = |x: f64| (4.0 * cbyr * x / std::f64::consts::PI).sqrt();
2700 let sum: f64 = co.h1_first + co.h1[1..=n].iter().sum::<f64>();
2701 let last = (f(cur - times[0]) - f(cur - times[1])) / (times[1] - times[0]);
2702 assert!(close(sum, last, 1e-9));
2703 }
2704
2705 #[test]
2709 fn matched_lc_line_delays_by_td() {
2710 use crate::circuit::{Circuit, Ltra, Resistor, Vsource};
2711 let (l, c, len) = (250e-9, 100e-12, 1.0); let mut ckt = Circuit::new("matched line");
2713 ckt.add(Vsource {
2714 name: "v1".into(),
2715 nodes: ["1".into(), "0".into()],
2716 dc: None,
2717 wave: Some(crate::srcfn::Wave::Pulse(vec![
2718 0.0, 1.0, 1e-9, 1e-9, 1e-9, 100e-9, 200e-9,
2719 ])),
2720 ac: None,
2721 distof1: None,
2722 distof2: None,
2723 port: None,
2724 })
2725 .unwrap();
2726 for (name, a, b) in [("rs", "1", "2"), ("rl", "3", "0")] {
2727 ckt.add(Resistor {
2728 name: name.into(),
2729 nodes: [a.into(), b.into()],
2730 value: 50.0,
2731 })
2732 .unwrap();
2733 }
2734 ckt.add(Ltra {
2735 name: "o1".into(),
2736 nodes: ["2".into(), "0".into(), "3".into(), "0".into()],
2737 model: LtraModel {
2738 r: Some(0.0),
2739 l: Some(l),
2740 c: Some(c),
2741 len: Some(len),
2742 ..LtraModel::default()
2743 },
2744 })
2745 .unwrap();
2746 let opts = crate::options::Options::default();
2747 let tr = ckt.tran(&opts, 0.1e-9, 20e-9, 0.0, None, false).unwrap();
2748 let t = tr.solution.column("time").unwrap();
2749 let v2 = tr.solution.column("v(2)").unwrap();
2750 let v3 = tr.solution.column("v(3)").unwrap();
2751 assert!(tr.solution.column("v(o1#i1)").is_some());
2752 let at = |col: &[f64], when: f64| {
2753 let k = t.iter().position(|&x| x >= when).unwrap();
2754 let f = (when - t[k - 1]) / (t[k] - t[k - 1]);
2755 col[k - 1] + f * (col[k] - col[k - 1])
2756 };
2757 assert!(at(v3, 5.5e-9).abs() < 1e-6, "{}", at(v3, 5.5e-9));
2759 assert!((at(v2, 1.5e-9) - 0.25).abs() < 2e-3, "{}", at(v2, 1.5e-9));
2761 assert!((at(v3, 6.5e-9) - 0.25).abs() < 2e-3, "{}", at(v3, 6.5e-9));
2762 assert!((at(v3, 12e-9) - 0.5).abs() < 1e-6);
2763 assert!((at(v2, 12e-9) - 0.5).abs() < 1e-6);
2764 }
2765
2766 #[test]
2767 fn setup_classifies_and_rejects_like_ltraset() {
2768 let base = LtraModel {
2769 r: Some(0.2),
2770 l: Some(9.13e-9),
2771 c: Some(3.65e-12),
2772 len: Some(24.0),
2773 ..LtraModel::default()
2774 };
2775 let p = LtraParams::new("o1", &base, 1e-3, 1e-12).unwrap();
2776 assert_eq!(p.case, LtraCase::Rlc);
2777 assert!(close(p.td, (9.13e-9f64 * 3.65e-12).sqrt() * 24.0, 1e-15));
2778 assert!(p.max_safe_step > 0.0 && p.max_safe_step <= 9.0 * p.td);
2779 assert!(p.notes.is_empty());
2780
2781 let lc = LtraModel {
2782 r: Some(0.0),
2783 ..base.clone()
2784 };
2785 assert_eq!(
2786 LtraParams::new("o1", &lc, 1e-3, 1e-12).unwrap().case,
2787 LtraCase::Lc
2788 );
2789 let rc = LtraModel {
2790 l: Some(0.0),
2791 ..base.clone()
2792 };
2793 assert_eq!(
2794 LtraParams::new("o1", &rc, 1e-3, 1e-12).unwrap().case,
2795 LtraCase::Rc
2796 );
2797 let rl = LtraModel {
2798 c: Some(0.0),
2799 ..base.clone()
2800 };
2801 assert!(
2802 LtraParams::new("o1", &rl, 1e-3, 1e-12)
2803 .unwrap_err()
2804 .contains("RL line")
2805 );
2806 let rlcg = LtraModel {
2807 g: Some(1e-3),
2808 ..base.clone()
2809 };
2810 assert!(
2811 LtraParams::new("o1", &rlcg, 1e-3, 1e-12)
2812 .unwrap_err()
2813 .contains("Nonzero G")
2814 );
2815 let rg = LtraModel {
2816 g: Some(1e-3),
2817 l: Some(0.0),
2818 c: Some(0.0),
2819 ..base.clone()
2820 };
2821 let p = LtraParams::new("o1", &rg, 1e-3, 1e-12).unwrap();
2822 assert_eq!(p.case, LtraCase::Rg);
2823 assert!(p.cosh_lroot_gr > 1.0);
2824 let nolen = LtraModel {
2825 len: None,
2826 ..base.clone()
2827 };
2828 assert!(
2829 LtraParams::new("o1", &nolen, 1e-3, 1e-12)
2830 .unwrap_err()
2831 .contains("length")
2832 );
2833 let one = LtraModel {
2834 r: None,
2835 l: None,
2836 ..base.clone()
2837 };
2838 let e = LtraParams::new("o1", &one, 1e-3, 1e-12).unwrap_err();
2839 assert!(e.contains("At least two"), "{e}");
2840 let nan = LtraModel {
2841 r: Some(f64::NAN),
2842 ..base
2843 };
2844 assert!(LtraParams::new("o1", &nan, 1e-3, 1e-12).is_err());
2845 }
2846}