UNPKG

mingo

Version:

MongoDB query language for in-memory objects

24 lines (23 loc) 796 B
import { isNumber } from "../../util"; import { $push } from "../accumulator/push"; import { TIMEUNIT_IN_MILLIS } from "../expression/date/_internal"; const $integral = (_, collection, expr, options) => { const { input, unit } = expr.inputExpr; const sortKey = "$" + Object.keys(expr.parentExpr.sortBy)[0]; const points = $push(collection, [sortKey, input], options).filter( (([x, y]) => isNumber(+x) && isNumber(+y)) ); if (points.length !== collection.length) return null; let result = 0; const size = collection.length; 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] || 1); result += 0.5 * (y1 + y2) * deltaX; } return result; }; export { $integral };