UNPKG

mingo

Version:

MongoDB query language for in-memory objects

24 lines (23 loc) 822 B
import { assert, isNumber } from "../../util"; import { $push } from "../accumulator/push"; import { TIMEUNIT_IN_MILLIS } from "../expression/date/_internal"; const $integral = (_, coll, expr, options) => { const { input, unit } = expr.inputExpr; const sortKey = "$" + Object.keys(expr.parentExpr.sortBy)[0]; const points = $push(coll, [sortKey, input], options).filter( (([x, y]) => isNumber(+x) && isNumber(+y)) ); const size = points.length; assert(coll.length === size, "$integral expects an array of numeric values"); let result = 0; for (let k = 1; k < size; k++) { const [x1, y1] = points[k - 1]; const [x2, y2] = points[k]; const deltaX = (x2 - x1) / TIMEUNIT_IN_MILLIS[unit ?? "millisecond"]; result += 0.5 * (y1 + y2) * deltaX; } return result; }; export { $integral };