From 3d92926a902017e7cf1f5738e99ff62da596ce12 Mon Sep 17 00:00:00 2001 From: Ethan Werner Date: Sun, 9 Aug 2026 15:02:21 +0400 Subject: refactor --- lib/atr_bands.c | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) (limited to 'lib/atr_bands.c') 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; -- cgit v1.3.1