Skip to content

Commit

Permalink
fix window function transform (#1691)
Browse files Browse the repository at this point in the history
  • Loading branch information
Fil committed Aug 21, 2023
1 parent 9099312 commit bfa0049
Show file tree
Hide file tree
Showing 4 changed files with 125 additions and 8 deletions.
16 changes: 8 additions & 8 deletions src/transforms/window.js
Original file line number Diff line number Diff line change
Expand Up @@ -96,24 +96,24 @@ function reduceAccessor(f) {
strict
? {
mapIndex(I, S, T) {
const s = (i) => (S[i] == null ? NaN : +S[i]);
const v = (i) => (S[i] == null ? NaN : +S[i]);
let nans = 0;
for (let i = 0; i < k - 1; ++i) if (isNaN(s(i))) ++nans;
for (let i = 0; i < k - 1; ++i) if (isNaN(v(i))) ++nans;
for (let i = 0, n = I.length - k + 1; i < n; ++i) {
if (isNaN(s(i + k - 1))) ++nans;
T[I[i + s]] = nans === 0 ? f(slice(I, i, i + k), s) : NaN;
if (isNaN(s(i))) --nans;
if (isNaN(v(i + k - 1))) ++nans;
T[I[i + s]] = nans === 0 ? f(slice(I, i, i + k), v) : NaN;
if (isNaN(v(i))) --nans;
}
}
}
: {
mapIndex(I, S, T) {
const s = (i) => (S[i] == null ? NaN : +S[i]);
const v = (i) => (S[i] == null ? NaN : +S[i]);
for (let i = -s; i < 0; ++i) {
T[I[i + s]] = f(slice(I, 0, i + k), s);
T[I[i + s]] = f(slice(I, 0, i + k), v);
}
for (let i = 0, n = I.length - s; i < n; ++i) {
T[I[i + s]] = f(slice(I, i, i + k), s);
T[I[i + s]] = f(slice(I, i, i + k), v);
}
}
};
Expand Down
Loading

0 comments on commit bfa0049

Please sign in to comment.