diff options
| author | Ethan Werner <mail@ethanwerner.com> | 2026-08-09 15:02:21 +0400 |
|---|---|---|
| committer | Ethan Werner <mail@ethanwerner.com> | 2026-08-09 15:02:21 +0400 |
| commit | 3d92926a902017e7cf1f5738e99ff62da596ce12 (patch) | |
| tree | 2b55422c26a48759d56fa91f06df24449e9dfa88 /lib/atr_bands.c | |
| parent | 07e18043145f3d91c8ea7ec3ad41b3c905585277 (diff) | |
Diffstat (limited to 'lib/atr_bands.c')
| -rw-r--r-- | lib/atr_bands.c | 15 |
1 files changed, 12 insertions, 3 deletions
diff --git a/lib/atr_bands.c b/lib/atr_bands.c index 73b7d44..1aa7464 100644 --- a/lib/atr_bands.c +++ b/lib/atr_bands.c @@ -25,7 +25,7 @@ ts_parameter_t const ts_atr_bands_p = { .n_inputs = 4, .n_outputs = 3, .n_consta uint_t ts_atr_bands_offset( ts_constant_t const *c ) { ( void )c; - return ( c ? ((ts_atr_bands_params_t *)c)->k : 0 ); + return ( c ? ((ts_atr_bands_params_t const *)c)->k : 0 ); } int ts_atr_bands( @@ -36,11 +36,20 @@ int ts_atr_bands( { TS_PROLOGUE( ts_atr_bands_p ); if ( !c ) return -1; - uint_t k = ((ts_atr_bands_params_t *)c)->k; + TS_CHECK_INPUT(x, x_n, n); + uint_t k = ((ts_atr_bands_params_t const *)c)->k; if (k == 0 || n < k) return -1; if (n < 2) return -1; - fp_t mult = ((ts_atr_bands_params_t *)c)->mult; + fp_t mult = ((ts_atr_bands_params_t const *)c)->mult; + if (mult < 0 || isnan(mult) || isinf(mult)) return -1; + + for ( uint_t i = 0; i < k - 1; ++i ) + { + y[0][i] = NAN; + y[1][i] = NAN; + y[2][i] = NAN; + } fp_t tr_sum = 0; fp_t atr = 0; |