node-red-contrib-self-healing/kalman-noise-filter/kalman.js

SHEN: Self-healing extensions for Node-RED.

const kalmanjs = require("kalmanjs");

module.exports = function (RED) {
  function KalmanNoiseFilter(config) {
    RED.nodes.createNode(this, config);
    const filter = new kalmanjs({
      R: Number.parseFloat(config.r) || 0.01,
      Q: Number.parseFloat(config.q) || 3,
    });
    this.on("input", function (msg) {
      if (Array.isArray(msg.payload)) {
        msg.payload = msg.payload.map((val) => filter.filter(val));
      } else {
        msg.payload = filter.filter(msg.payload);
      }
      this.send(msg);
    });
  }
  RED.nodes.registerType("kalman-noise-filter", KalmanNoiseFilter);
};