base/func.mjs

  1. import { isStr, clTF2, clTF3 } from '../core.mjs';
  2. import * as jsroot_math from './math.mjs';
  4. /** @summary Assign `evalPar` function for TF1 object
  5.   * @private */
  7. function proivdeEvalPar(obj, check_save) {
  8.    obj.$math = jsroot_math;
  10.    let _func = obj.fTitle, isformula = false, pprefix = '[';
  11.    if (_func === 'gaus') _func = 'gaus(0)';
  12.    if (isStr(obj.fFormula?.fFormula)) {
  13.      if (obj.fFormula.fFormula.indexOf('[](double*x,double*p)') === 0) {
  14.         isformula = true; pprefix = 'p[';
  15.         _func = obj.fFormula.fFormula.slice(21);
  16.      } else {
  17.         _func = obj.fFormula.fFormula;
  18.         pprefix = '[p';
  19.      }
  21.      if (obj.fFormula.fClingParameters && obj.fFormula.fParams) {
  22.         obj.fFormula.fParams.forEach(pair => {
  23.            const regex = new RegExp(`(\\[${pair.first}\\])`, 'g'),
  24.                parvalue = obj.fFormula.fClingParameters[pair.second];
  25.            _func = _func.replace(regex, (parvalue < 0) ? `(${parvalue})` : parvalue);
  26.         });
  27.       }
  28.    }
  30.    if (!_func)
  31.       return !check_save || (obj.fSave?.length > 2);
  33.    obj.formulas?.forEach(entry => {
  34.       _func = _func.replaceAll(entry.fName, entry.fTitle);
  35.    });
  37.    _func = _func.replace(/\b(TMath::SinH)\b/g, 'Math.sinh')
  38.                 .replace(/\b(TMath::CosH)\b/g, 'Math.cosh')
  39.                 .replace(/\b(TMath::TanH)\b/g, 'Math.tanh')
  40.                 .replace(/\b(TMath::ASinH)\b/g, 'Math.asinh')
  41.                 .replace(/\b(TMath::ACosH)\b/g, 'Math.acosh')
  42.                 .replace(/\b(TMath::ATanH)\b/g, 'Math.atanh')
  43.                 .replace(/\b(TMath::ASin)\b/g, 'Math.asin')
  44.                 .replace(/\b(TMath::ACos)\b/g, 'Math.acos')
  45.                 .replace(/\b(TMath::Atan)\b/g, 'Math.atan')
  46.                 .replace(/\b(TMath::ATan2)\b/g, 'Math.atan2')
  47.                 .replace(/\b(sin|SIN|TMath::Sin)\b/g, 'Math.sin')
  48.                 .replace(/\b(cos|COS|TMath::Cos)\b/g, 'Math.cos')
  49.                 .replace(/\b(tan|TAN|TMath::Tan)\b/g, 'Math.tan')
  50.                 .replace(/\b(exp|EXP|TMath::Exp)\b/g, 'Math.exp')
  51.                 .replace(/\b(log|LOG|TMath::Log)\b/g, 'Math.log')
  52.                 .replace(/\b(log10|LOG10|TMath::Log10)\b/g, 'Math.log10')
  53.                 .replace(/\b(pow|POW|TMath::Power)\b/g, 'Math.pow')
  54.                 .replace(/\b(pi|PI)\b/g, 'Math.PI')
  55.                 .replace(/\b(abs|ABS|TMath::Abs)\b/g, 'Math.abs')
  56.                 .replace(/\bsqrt\(/g, 'Math.sqrt(')
  57.                 .replace(/\bxygaus\(/g, 'this.$math.gausxy(this, x, y, ')
  58.                 .replace(/\bgaus\(/g, 'this.$math.gaus(this, x, ')
  59.                 .replace(/\bgausn\(/g, 'this.$math.gausn(this, x, ')
  60.                 .replace(/\bexpo\(/g, 'this.$math.expo(this, x, ')
  61.                 .replace(/\blandau\(/g, 'this.$math.landau(this, x, ')
  62.                 .replace(/\blandaun\(/g, 'this.$math.landaun(this, x, ')
  63.                 .replace(/\b(TMath::|ROOT::Math::)/g, 'this.$math.');
  65.    if (_func.match(/^pol[0-9]$/) && (parseInt(_func[3]) === obj.fNpar - 1)) {
  66.       _func = '[0]';
  67.       for (let k = 1; k < obj.fNpar; ++k)
  68.          _func += ` + [${k}] * `+ ((k === 1) ? 'x' : `Math.pow(x,${k})`);
  69.    }
  71.    if (_func.match(/^chebyshev[0-9]$/) && (parseInt(_func[9]) === obj.fNpar - 1)) {
  72.       _func = `this.$math.ChebyshevN(${obj.fNpar-1}, x, `;
  73.       for (let k = 0; k < obj.fNpar; ++k)
  74.          _func += (k === 0 ? '[' : ', ') + `[${k}]`;
  75.       _func += '])';
  76.    }
  78.    for (let i = 0; i < obj.fNpar; ++i)
  79.       _func = _func.replaceAll(pprefix + i + ']', `(${obj.GetParValue(i)})`);
  81.    for (let n = 2; n < 10; ++n)
  82.       _func = _func.replaceAll(`x^${n}`, `Math.pow(x,${n})`);
  84.    if (isformula) {
  85.       _func = _func.replace(/x\[0\]/g, 'x');
  86.       if (obj._typename === clTF3) {
  87.          _func = _func.replace(/x\[1\]/g, 'y');
  88.          _func = _func.replace(/x\[2\]/g, 'z');
  89.          obj.evalPar = new Function('x', 'y', 'z', _func).bind(obj);
  90.       } else if (obj._typename === clTF2) {
  91.          _func = _func.replace(/x\[1\]/g, 'y');
  92.          obj.evalPar = new Function('x', 'y', _func).bind(obj);
  93.       } else
  94.          obj.evalPar = new Function('x', _func).bind(obj);
  95.    } else if (obj._typename === clTF3)
  96.       obj.evalPar = new Function('x', 'y', 'z', 'return ' + _func).bind(obj);
  97.    else if (obj._typename === clTF2)
  98.       obj.evalPar = new Function('x', 'y', 'return ' + _func).bind(obj);
  99.    else
  100.       obj.evalPar = new Function('x', 'return ' + _func).bind(obj);
  102.    return true;
  103. }
  106. /** @summary Get interpolation in saved buffer
  107.   * @desc Several checks must be done before function can be used
  108.   * @private */
  109. function _getTF1Save(func, x) {
  110.    const np = func.fSave.length - 3,
  111.          xmin = func.fSave[np + 1],
  112.         xmax = func.fSave[np + 2],
  113.         dx = (xmax - xmin) / np;
  114.     if (x < xmin)
  115.        return func.fSave[0];
  116.     if (x > xmax)
  117.        return func.fSave[np];
  119.     const bin = Math.min(np - 1, Math.floor((x - xmin) / dx));
  120.     let xlow = xmin + bin * dx,
  121.         xup = xlow + dx,
  122.         ylow = func.fSave[bin],
  123.         yup = func.fSave[bin + 1];
  125.     if (!Number.isFinite(ylow) && (bin < np - 1)) {
  126.        xlow += dx; xup += dx;
  127.        ylow = yup; yup = func.fSave[bin + 2];
  128.     } else if (!Number.isFinite(yup) && (bin > 0)) {
  129.        xup -= dx; xlow -= dx;
  130.        yup = ylow; ylow = func.fSave[bin - 1];
  131.     }
  133.     return ((xup * ylow - xlow * yup) + x * (yup - ylow)) / dx;
  134. }
  136. /** @summary Provide TF1 value
  137.   * @desc First try evaluate, if not possible - check saved buffer
  138.   * @private */
  139. function getTF1Value(func, x, skip_eval = undefined) {
  140.    if (!func)
  141.       return 0;
  143.    let iserr = false;
  145.    if (!skip_eval && !func.evalPar) {
  146.       try {
  147.          if (!proivdeEvalPar(func))
  148.             iserr = true;
  149.       } catch {
  150.          iserr = true;
  151.       }
  152.    }
  154.    if (func.evalPar && !iserr) {
  155.       try {
  156.          return func.evalPar(x);
  157.       } catch {
  158.          /* eslint-disable-next-line  no-useless-assignment */
  159.          iserr = true;
  160.       }
  161.    }
  163.    const np = func.fSave.length - 3;
  164.    return (np < 2) || (func.fSave[np + 1] === func.fSave[np + 2]) ? 0 : _getTF1Save(func, x);
  165. }
  167. export { proivdeEvalPar, getTF1Value, _getTF1Save };