Skip to content

Commit

Permalink
feat: allow NumberArray rather that DoubleArray
Browse files Browse the repository at this point in the history
  • Loading branch information
lpatiny committed Apr 15, 2024
1 parent 5c67b8e commit 078a0f9
Showing 1 changed file with 7 additions and 7 deletions.
14 changes: 7 additions & 7 deletions src/index.ts
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
import { DoubleArray } from 'cheminfo-types';
import { NumberArray } from 'cheminfo-types';
import { isAnyArray } from 'is-any-array';

export interface SGGOptions {
Expand All @@ -24,8 +24,8 @@ export interface SGGOptions {
*/

export function sgg(
ys: DoubleArray,
xs: DoubleArray | number,
ys: NumberArray,
xs: NumberArray | number,
options: SGGOptions = {},
): Float64Array {
const { windowSize = 9, derivative = 0, polynomial = 3 } = options;
Expand Down Expand Up @@ -86,9 +86,9 @@ export function sgg(
ans[half - i - 1] = d1 / hs;
ans[np - half + i] = d2 / hs;
} else {
hs = getHs(xs as DoubleArray, half - i - 1, half, derivative);
hs = getHs(xs as NumberArray, half - i - 1, half, derivative);
ans[half - i - 1] = d1 / hs;
hs = getHs(xs as DoubleArray, np - half + i, half, derivative);
hs = getHs(xs as NumberArray, np - half + i, half, derivative);
ans[np - half + i] = d2 / hs;
}
}
Expand All @@ -99,15 +99,15 @@ export function sgg(
let d = 0;
for (let l = 0; l < windowSize; l++) d += wg[l] * ys[l + i - windowSize];
if (!constantH) {
hs = getHs(xs as DoubleArray, i - half - 1, half, derivative);
hs = getHs(xs as NumberArray, i - half - 1, half, derivative);
}
ans[i - half - 1] = d / hs;
}
return ans;
}

function getHs(
h: DoubleArray,
h: NumberArray,
center: number,
half: number,
derivative: number,
Expand Down

0 comments on commit 078a0f9

Please sign in to comment.