diff-grok 1.0.8

package/src/pipeline/loops-pipeline-creator.ts

@@ -0,0 +1,138 @@
+import {IVP} from '../scripting-tools';
+import {MATH_CONSTS, MATH_FUNCS, POW_IDX} from '../scripting-tools/scripting-tools';
+import {getOutputCode} from './output';
+import {Pipeline, Wrapper} from './pipeline';
+import {PipelineCreator} from './pipeline-creator';
+
+import {argName2IdxMap, ARG, ARG_COL_IDX, CORRECTION_FACTOR, DEFAULT_CORRECTION, LOOP_MIN_COUNT} from './constants';
+
+/** Pipeline creator for cyclic model */
+export class CyclicModelPipelineCreator extends PipelineCreator {
+  constructor(ivp: IVP) {
+    super(ivp);
+
+    if (ivp.loop === null)
+      throw new Error('Incorrect use of CyclicModelPipelineCreator: model has no loop');
+  }
+
+  /** Return computations pipeline */
+  public getPipeline(inputs: Float64Array): Pipeline {
+    if (this.ivp.loop === null)
+      throw new Error('Incorrect use of CyclicModelPipelineCreator: model has no loop');
+
+    const argSize = argName2IdxMap.size;
+    const restricted = inputs.length - 1;
+    const repetitions = inputs[restricted];
+
+    if (repetitions < LOOP_MIN_COUNT)
+      throw new Error(`Incorrect repetitions count: ${repetitions}, expected: > 0`);
+
+    const preProcLines: string[] = [];
+
+    // Used math funcs
+    this.ivp.usedMathFuncs.forEach((idx) => {
+      if (idx !== POW_IDX)
+        preProcLines.push(`const ${MATH_FUNCS[idx]} = (x) => Math.${MATH_FUNCS[idx]}(x);`);
+      else
+        preProcLines.push(`const pow = (x, y) => Math.pow(x, y);`);
+    });
+
+    // Used math consts
+    this.ivp.usedMathConsts.forEach((idx) => {
+      preProcLines.push(`const ${MATH_CONSTS[idx]} = Math.${MATH_CONSTS[idx]};`);
+    });
+
+    // Model constants
+    if (this.ivp.consts !== null)
+      this.ivp.consts.forEach((input, name) => preProcLines.push(`const ${name} = ${input.value};`));
+
+    preProcLines.push(`const inputs = arguments[1].slice(0, ${restricted});`);
+
+    // Extract inputs
+    argName2IdxMap.forEach((idx, name) => preProcLines.push(`let ${name} = inputs[${idx}];`));
+
+    this.ivp.deqs.solutionNames.forEach((name, idx) => {
+      preProcLines.push(`let ${name} = inputs[${idx + argSize}];`);
+    });
+
+    if (this.ivp.params !== null) {
+      const shift = argSize + this.ivp.deqs.solutionNames.length;
+      let idx = 0;
+
+      this.ivp.params.forEach((_, name) => {
+        preProcLines.push(`let ${name} = inputs[${idx + shift}];`);
+        ++idx;
+      });
+    }
+
+    // Updates
+    this.ivp.loop.updates.forEach((expr) => preProcLines.push(`${expr};`));
+
+    // Input vector update
+    argName2IdxMap.forEach((idx, name) => preProcLines.push(`inputs[${idx}] = ${name};`));
+
+    this.ivp.deqs.solutionNames.forEach((name, idx) => {
+      preProcLines.push(`inputs[${idx + argSize}] = ${name};`);
+    });
+
+    if (this.ivp.params !== null) {
+      const shift = argSize + this.ivp.deqs.solutionNames.length;
+      let idx = 0;
+
+      this.ivp.params.forEach((_, name) => {
+        preProcLines.push(`inputs[${idx + shift}] = ${name};`);
+        ++idx;
+      });
+    }
+
+    preProcLines.push(`return inputs;`);
+
+    const postProcLines = [
+      'const solution = arguments[0];',
+      'const lastIdx = solution[0].length - 1;',
+      'const inputs = new Float64Array(arguments[1]);',
+      `const duration = inputs[${argName2IdxMap.get(ARG.FINISH)}] - inputs[${argName2IdxMap.get(ARG.START)}];`,
+      `inputs[${argName2IdxMap.get(ARG.START)}] = inputs[${argName2IdxMap.get(ARG.FINISH)}];`,
+      `inputs[${argName2IdxMap.get(ARG.FINISH)}] += duration;`,
+    ];
+
+    this.ivp.deqs.solutionNames.forEach((_, idx) => {
+      postProcLines.push(`inputs[${idx + argSize}] = solution[${idx + 1}][lastIdx];`);
+    });
+
+    postProcLines.push('return inputs;');
+
+    const preCode = preProcLines.join('\n');
+    const postCode = postProcLines.join('\n');
+
+    const wrappers = new Array<Wrapper>(repetitions);
+
+    for (let i = 0; i < repetitions; ++i) {
+      wrappers[i] = {
+        preproc: preCode,
+        out: null,
+        postproc: postCode,
+      };
+    }
+
+    wrappers[repetitions - 1].postproc = this.getArgumentCorrectionCode();
+
+    return {
+      wrappers: wrappers,
+      out: getOutputCode(this.ivp),
+    };
+  } // getPipeline
+
+  /** Return code for correction of an argument */
+  private getArgumentCorrectionCode(): string {
+    return `const arg = arguments[0][${ARG_COL_IDX}];
+    const length = arg.length;
+    const step = arg[1] - arg[0];
+    const correction = Math.min(step * ${CORRECTION_FACTOR}, ${DEFAULT_CORRECTION});
+    for (let i = 0; i < length - 1; ++i) {
+      if (arg[i + 1] - arg[i] < step)
+        arg[i] -= correction;
+    }
+    return arguments[0];`;
+  } // getArgumentCorrectionCode
+} // CyclicModelPipelineCreator

package/src/pipeline/output.js

@@ -0,0 +1,146 @@
+// Features for creating output code for a pipeline
+import { MATH_CONSTS, POW_IDX, MATH_FUNCS } from '../scripting-tools/scripting-tools';
+import { ARG_INP_COUNT } from './constants';
+const SHIFT = 1;
+/** Return output for the model WITHOUT expressions */
+function getOutputNoExprs(ivp) {
+    const outputs = ivp.outputs;
+    if (outputs === null)
+        throw new Error('Grok Lib issue: no outputs in the initial value problem');
+    if (exprsInOutputs(ivp))
+        throw new Error('Grok Lib issue: no expressions are allowed');
+    const lines = ['const solution = [];'];
+    outputs.forEach((_, name) => {
+        if (name === ivp.arg.name)
+            lines.push('solution.push(arguments[0][0]);');
+        else {
+            const idx = ivp.deqs.solutionNames.indexOf(name);
+            if (idx > -1)
+                lines.push(`solution.push(arguments[0][${idx + 1}]);`);
+        }
+    });
+    lines.push('return solution;');
+    return lines.join('\n');
+} // getOutputNoExprs
+/** Return output for the model WITH expressions */
+function getOutputWithExprs(ivp) {
+    const outputs = ivp.outputs;
+    if (outputs === null)
+        throw new Error('Grok Lib issue: no outputs in the initial value problem');
+    const exprs = ivp.exprs;
+    if (!exprsInOutputs(ivp) || (exprs === null))
+        throw new Error('Grok Lib issue: expressions must be given');
+    const argName = ivp.arg.name;
+    const lines = ['const solution = [];'];
+    // Used math funcs
+    ivp.usedMathFuncs.forEach((idx) => {
+        if (idx !== POW_IDX)
+            lines.push(`const ${MATH_FUNCS[idx]} = (x) => Math.${MATH_FUNCS[idx]}(x);`);
+        else
+            lines.push(`const pow = (x, y) => Math.pow(x, y);`);
+    });
+    // Used math consts
+    ivp.usedMathConsts.forEach((idx) => lines.push(`const ${MATH_CONSTS[idx]} = Math.${MATH_CONSTS[idx]};`));
+    // Model constants
+    if (ivp.consts !== null)
+        ivp.consts.forEach((input, name) => lines.push(`const ${name} = ${input.value};`));
+    // Model parameters
+    if (ivp.params !== null) {
+        let idx = ARG_INP_COUNT + ivp.deqs.solutionNames.length;
+        ivp.params.forEach((_, name) => {
+            lines.push(`const ${name} = arguments[1][${idx}];`);
+            ++idx;
+        });
+    }
+    lines.push('');
+    lines.push('const length = arguments[0][0].length;');
+    const funcNames = ivp.deqs.solutionNames;
+    lines.push(`let ${argName} = 0;`);
+    funcNames.forEach((name) => lines.push(`let ${name} = 0;`));
+    lines.push('');
+    exprs.forEach((_, name) => {
+        lines.push(`let ${name} = 0;`);
+        if (outputs.has(name))
+            lines.push(`let ${name}Raw = new Float64Array(length);`);
+    });
+    lines.push('');
+    lines.push('for (let k = 0; k < length; ++k) {');
+    lines.push(`  ${argName} = arguments[0][0][k];`);
+    funcNames.forEach((name, idx) => lines.push(`  ${name} = arguments[0][${SHIFT + idx}][k];`));
+    lines.push('');
+    exprs.forEach((expr, name) => {
+        lines.push(`  ${name} = ${expr};`);
+        if (outputs.has(name))
+            lines.push(`  ${name}Raw[k] = ${name};`);
+    });
+    lines.push('}');
+    lines.push('');
+    outputs.forEach((_, name) => {
+        if (name === argName)
+            lines.push('solution.push(arguments[0][0]);');
+        else {
+            const idx = ivp.deqs.solutionNames.indexOf(name);
+            if (idx > -1)
+                lines.push(`solution.push(arguments[0][${idx + 1}]);`);
+            else if (exprs.has(name))
+                lines.push(`solution.push(${name}Raw);`);
+        }
+    });
+    lines.push('return solution;');
+    return lines.join('\n');
+} // getOutputWithExprs
+/** Return a code for output extraction */
+export function getOutputCode(ivp) {
+    const outputs = ivp.outputs;
+    if (outputs === null)
+        return null;
+    checkApplicability(ivp);
+    if (exprsInOutputs(ivp))
+        return getOutputWithExprs(ivp);
+    else
+        return getOutputNoExprs(ivp);
+} // getOutputCode
+/** Check whether outputs have items from expressions */
+function exprsInOutputs(ivp) {
+    const outputs = ivp.outputs;
+    if (outputs === null)
+        return false;
+    const exprs = ivp.exprs;
+    if (exprs === null)
+        return false;
+    let flag = false;
+    exprs.forEach((_, name) => flag || (flag = outputs.has(name)));
+    return flag;
+}
+/** Check applicability of model outputs */
+function checkApplicability(ivp) {
+    const outputs = ivp.outputs;
+    if (outputs === null)
+        throw new Error('Model has no outputs');
+    const exprs = ivp.exprs;
+    if (exprs === null)
+        return;
+    if (ivp.loop !== null) {
+        if (exprsInOutputs(ivp))
+            throw new Error('Non-supported model: expressions in output & loops');
+        return;
+    }
+    if (ivp.updates !== null) {
+        if (exprsInOutputs(ivp))
+            throw new Error('Non-supported model: expressions in output and updates');
+    }
+} // checkApplicability
+/** Return a code for output extraction */
+export function getOutputNames(ivp) {
+    const outputs = ivp.outputs;
+    if (outputs === null)
+        return [ivp.arg.name].concat(ivp.deqs.solutionNames);
+    const names = new Array(outputs.size);
+    let idx = 0;
+    outputs.forEach((output) => {
+        names[idx] = output.caption;
+        ++idx;
+    });
+    return names;
+} // getOutputNames
+//# sourceMappingURL=output.js.map

package/src/pipeline/output.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"output.js","sourceRoot":"","sources":["output.ts"],"names":[],"mappings":"AAAA,mDAAmD;AAEnD,OAAO,EAAM,WAAW,EAAE,OAAO,EAAE,UAAU,EAAC,MAAM,oCAAoC,CAAC;AACzF,OAAO,EAAC,aAAa,EAAC,MAAM,aAAa,CAAC;AAE1C,MAAM,KAAK,GAAG,CAAC,CAAC;AAEhB,sDAAsD;AACtD,SAAS,gBAAgB,CAAC,GAAQ;IAChC,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;IAC5B,IAAI,OAAO,KAAK,IAAI;QAClB,MAAM,IAAI,KAAK,CAAC,yDAAyD,CAAC,CAAC;IAE7E,IAAI,cAAc,CAAC,GAAG,CAAC;QACrB,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;IAGhE,MAAM,KAAK,GAAa,CAAC,sBAAsB,CAAC,CAAC;IAEjD,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE;QAC1B,IAAI,IAAI,KAAK,GAAG,CAAC,GAAG,CAAC,IAAI;YACvB,KAAK,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;aAC3C,CAAC;YACJ,MAAM,GAAG,GAAG,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAEjD,IAAI,GAAG,GAAG,CAAC,CAAC;gBACV,KAAK,CAAC,IAAI,CAAC,8BAA8B,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC;QAC3D,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,KAAK,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;IAE/B,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC,CAAC,mBAAmB;AAErB,mDAAmD;AACnD,SAAS,kBAAkB,CAAC,GAAQ;IAClC,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;IAC5B,IAAI,OAAO,KAAK,IAAI;QAClB,MAAM,IAAI,KAAK,CAAC,yDAAyD,CAAC,CAAC;IAE7E,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC;IAExB,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,KAAK,IAAI,CAAC;QAC1C,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;IAG/D,MAAM,OAAO,GAAG,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC;IAC7B,MAAM,KAAK,GAAa,CAAC,sBAAsB,CAAC,CAAC;IAEjD,kBAAkB;IAClB,GAAG,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;QAChC,IAAI,GAAG,KAAK,OAAO;YACjB,KAAK,CAAC,IAAI,CAAC,SAAS,UAAU,CAAC,GAAG,CAAC,kBAAkB,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;;YAE5E,KAAK,CAAC,IAAI,CAAC,uCAAuC,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,mBAAmB;IACnB,GAAG,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,WAAW,CAAC,GAAG,CAAC,WAAW,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;IAEzG,kBAAkB;IAClB,IAAI,GAAG,CAAC,MAAM,KAAK,IAAI;QACrB,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,IAAI,MAAM,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;IAErF,mBAAmB;IACnB,IAAI,GAAG,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC;QACxB,IAAI,GAAG,GAAG,aAAa,GAAG,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC;QACxD,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE;YAC7B,KAAK,CAAC,IAAI,CAAC,SAAS,IAAI,mBAAmB,GAAG,IAAI,CAAC,CAAC;YACpD,EAAE,GAAG,CAAC;QACR,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAEf,KAAK,CAAC,IAAI,CAAC,wCAAwC,CAAC,CAAC;IACrD,MAAM,SAAS,GAAG,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC;IAEzC,KAAK,CAAC,IAAI,CAAC,OAAO,OAAO,OAAO,CAAC,CAAC;IAClC,SAAS,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,CAAC,CAAC;IAE5D,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAEf,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE;QACxB,KAAK,CAAC,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,CAAC;QAE/B,IAAI,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC;YACnB,KAAK,CAAC,IAAI,CAAC,OAAO,IAAI,iCAAiC,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAEf,KAAK,CAAC,IAAI,CAAC,oCAAoC,CAAC,CAAC;IACjD,KAAK,CAAC,IAAI,CAAC,KAAK,OAAO,wBAAwB,CAAC,CAAC;IACjD,SAAS,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,IAAI,mBAAmB,KAAK,GAAG,GAAG,OAAO,CAAC,CAAC,CAAC;IAC7F,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACf,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE;QAC3B,KAAK,CAAC,IAAI,CAAC,KAAK,IAAI,MAAM,IAAI,GAAG,CAAC,CAAC;QAEnC,IAAI,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC;YACnB,KAAK,CAAC,IAAI,CAAC,KAAK,IAAI,YAAY,IAAI,GAAG,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IACH,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAEhB,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAEf,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE;QAC1B,IAAI,IAAI,KAAK,OAAO;YAClB,KAAK,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;aAC3C,CAAC;YACJ,MAAM,GAAG,GAAG,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAEjD,IAAI,GAAG,GAAG,CAAC,CAAC;gBACV,KAAK,CAAC,IAAI,CAAC,8BAA8B,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC;iBACpD,IAAI,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC;gBACtB,KAAK,CAAC,IAAI,CAAC,iBAAiB,IAAI,OAAO,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,KAAK,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;IAE/B,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC,CAAC,qBAAqB;AAEvB,0CAA0C;AAC1C,MAAM,UAAU,aAAa,CAAC,GAAQ;IACpC,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;IAC5B,IAAI,OAAO,KAAK,IAAI;QAClB,OAAO,IAAI,CAAC;IAEd,kBAAkB,CAAC,GAAG,CAAC,CAAC;IAExB,IAAI,cAAc,CAAC,GAAG,CAAC;QACrB,OAAO,kBAAkB,CAAC,GAAG,CAAC,CAAC;;QAE/B,OAAO,gBAAgB,CAAC,GAAG,CAAC,CAAC;AACjC,CAAC,CAAC,gBAAgB;AAElB,wDAAwD;AACxD,SAAS,cAAc,CAAC,GAAQ;IAC9B,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;IAC5B,IAAI,OAAO,KAAK,IAAI;QAClB,OAAO,KAAK,CAAC;IAEf,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC;IACxB,IAAI,KAAK,KAAK,IAAI;QAChB,OAAO,KAAK,CAAC;IAEf,IAAI,IAAI,GAAG,KAAK,CAAC;IAEjB,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,KAAJ,IAAI,GAAK,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA,CAAC,CAAC;IAEvD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,2CAA2C;AAC3C,SAAS,kBAAkB,CAAC,GAAQ;IAClC,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;IAE5B,IAAI,OAAO,KAAK,IAAI;QAClB,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;IAE1C,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC;IACxB,IAAI,KAAK,KAAK,IAAI;QAChB,OAAO;IAET,IAAI,GAAG,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;QACtB,IAAI,cAAc,CAAC,GAAG,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,oDAAoD,CAAC,CAAC;QAExE,OAAO;IACT,CAAC;IAED,IAAI,GAAG,CAAC,OAAO,KAAK,IAAI,EAAE,CAAC;QACzB,IAAI,cAAc,CAAC,GAAG,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,wDAAwD,CAAC,CAAC;IAC9E,CAAC;AACH,CAAC,CAAC,qBAAqB;AAEvB,0CAA0C;AAC1C,MAAM,UAAU,cAAc,CAAC,GAAQ;IACrC,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;IAC5B,IAAI,OAAO,KAAK,IAAI;QAClB,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAEvD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAS,OAAO,CAAC,IAAI,CAAC,CAAC;IAE9C,IAAI,GAAG,GAAG,CAAC,CAAC;IACZ,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;QACzB,KAAK,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC;QAC5B,EAAE,GAAG,CAAC;IACR,CAAC,CAAC,CAAC;IAEH,OAAO,KAAK,CAAC;AACf,CAAC,CAAC,iBAAiB"}

package/src/pipeline/output.ts

@@ -0,0 +1,196 @@
+// Features for creating output code for a pipeline
+
+import {IVP, MATH_CONSTS, POW_IDX, MATH_FUNCS} from '../scripting-tools/scripting-tools';
+import {ARG_INP_COUNT} from './constants';
+
+const SHIFT = 1;
+
+/** Return output for the model WITHOUT expressions */
+function getOutputNoExprs(ivp: IVP): string {
+  const outputs = ivp.outputs;
+  if (outputs === null)
+    throw new Error('Grok Lib issue: no outputs in the initial value problem');
+
+  if (exprsInOutputs(ivp))
+    throw new Error('Grok Lib issue: no expressions are allowed');
+
+
+  const lines: string[] = ['const solution = [];'];
+
+  outputs.forEach((_, name) => {
+    if (name === ivp.arg.name)
+      lines.push('solution.push(arguments[0][0]);');
+    else {
+      const idx = ivp.deqs.solutionNames.indexOf(name);
+
+      if (idx > -1)
+        lines.push(`solution.push(arguments[0][${idx + 1}]);`);
+    }
+  });
+
+  lines.push('return solution;');
+
+  return lines.join('\n');
+} // getOutputNoExprs
+
+/** Return output for the model WITH expressions */
+function getOutputWithExprs(ivp: IVP): string {
+  const outputs = ivp.outputs;
+  if (outputs === null)
+    throw new Error('Grok Lib issue: no outputs in the initial value problem');
+
+  const exprs = ivp.exprs;
+
+  if (!exprsInOutputs(ivp) || (exprs === null))
+    throw new Error('Grok Lib issue: expressions must be given');
+
+
+  const argName = ivp.arg.name;
+  const lines: string[] = ['const solution = [];'];
+
+  // Used math funcs
+  ivp.usedMathFuncs.forEach((idx) => {
+    if (idx !== POW_IDX)
+      lines.push(`const ${MATH_FUNCS[idx]} = (x) => Math.${MATH_FUNCS[idx]}(x);`);
+    else
+      lines.push(`const pow = (x, y) => Math.pow(x, y);`);
+  });
+
+  // Used math consts
+  ivp.usedMathConsts.forEach((idx) => lines.push(`const ${MATH_CONSTS[idx]} = Math.${MATH_CONSTS[idx]};`));
+
+  // Model constants
+  if (ivp.consts !== null)
+    ivp.consts.forEach((input, name) => lines.push(`const ${name} = ${input.value};`));
+
+  // Model parameters
+  if (ivp.params !== null) {
+    let idx = ARG_INP_COUNT + ivp.deqs.solutionNames.length;
+    ivp.params.forEach((_, name) => {
+      lines.push(`const ${name} = arguments[1][${idx}];`);
+      ++idx;
+    });
+  }
+
+  lines.push('');
+
+  lines.push('const length = arguments[0][0].length;');
+  const funcNames = ivp.deqs.solutionNames;
+
+  lines.push(`let ${argName} = 0;`);
+  funcNames.forEach((name) => lines.push(`let ${name} = 0;`));
+
+  lines.push('');
+
+  exprs.forEach((_, name) => {
+    lines.push(`let ${name} = 0;`);
+
+    if (outputs.has(name))
+      lines.push(`let ${name}Raw = new Float64Array(length);`);
+  });
+
+  lines.push('');
+
+  lines.push('for (let k = 0; k < length; ++k) {');
+  lines.push(`  ${argName} = arguments[0][0][k];`);
+  funcNames.forEach((name, idx) => lines.push(`  ${name} = arguments[0][${SHIFT + idx}][k];`));
+  lines.push('');
+  exprs.forEach((expr, name) => {
+    lines.push(`  ${name} = ${expr};`);
+
+    if (outputs.has(name))
+      lines.push(`  ${name}Raw[k] = ${name};`);
+  });
+  lines.push('}');
+
+  lines.push('');
+
+  outputs.forEach((_, name) => {
+    if (name === argName)
+      lines.push('solution.push(arguments[0][0]);');
+    else {
+      const idx = ivp.deqs.solutionNames.indexOf(name);
+
+      if (idx > -1)
+        lines.push(`solution.push(arguments[0][${idx + 1}]);`);
+      else if (exprs.has(name))
+        lines.push(`solution.push(${name}Raw);`);
+    }
+  });
+
+  lines.push('return solution;');
+
+  return lines.join('\n');
+} // getOutputWithExprs
+
+/** Return a code for output extraction */
+export function getOutputCode(ivp: IVP): string | null {
+  const outputs = ivp.outputs;
+  if (outputs === null)
+    return null;
+
+  checkApplicability(ivp);
+
+  if (exprsInOutputs(ivp))
+    return getOutputWithExprs(ivp);
+  else
+    return getOutputNoExprs(ivp);
+} // getOutputCode
+
+/** Check whether outputs have items from expressions */
+function exprsInOutputs(ivp: IVP): boolean {
+  const outputs = ivp.outputs;
+  if (outputs === null)
+    return false;
+
+  const exprs = ivp.exprs;
+  if (exprs === null)
+    return false;
+
+  let flag = false;
+
+  exprs.forEach((_, name) => flag ||= outputs.has(name));
+
+  return flag;
+}
+
+/** Check applicability of model outputs */
+function checkApplicability(ivp: IVP): void {
+  const outputs = ivp.outputs;
+
+  if (outputs === null)
+    throw new Error('Model has no outputs');
+
+  const exprs = ivp.exprs;
+  if (exprs === null)
+    return;
+
+  if (ivp.loop !== null) {
+    if (exprsInOutputs(ivp))
+      throw new Error('Non-supported model: expressions in output & loops');
+
+    return;
+  }
+
+  if (ivp.updates !== null) {
+    if (exprsInOutputs(ivp))
+      throw new Error('Non-supported model: expressions in output and updates');
+  }
+} // checkApplicability
+
+/** Return a code for output extraction */
+export function getOutputNames(ivp: IVP): string[] {
+  const outputs = ivp.outputs;
+  if (outputs === null)
+    return [ivp.arg.name].concat(ivp.deqs.solutionNames);
+
+  const names = new Array<string>(outputs.size);
+
+  let idx = 0;
+  outputs.forEach((output) => {
+    names[idx] = output.caption;
+    ++idx;
+  });
+
+  return names;
+} // getOutputNames

package/src/pipeline/pipeline-creator.js

@@ -0,0 +1,10 @@
+import { getOutputNames } from './output';
+/** Creator of a pipeline for in-webworker computations */
+export class PipelineCreator {
+    constructor(ivp) {
+        this.ivp = ivp;
+        this.outputNames = getOutputNames(ivp);
+    }
+    ;
+}
+//# sourceMappingURL=pipeline-creator.js.map

package/src/pipeline/pipeline-creator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pipeline-creator.js","sourceRoot":"","sources":["pipeline-creator.ts"],"names":[],"mappings":"AACA,OAAO,EAAC,cAAc,EAAC,MAAM,UAAU,CAAC;AAGxC,0DAA0D;AAC1D,MAAM,OAAgB,eAAe;IAInC,YAAY,GAAQ;QAClB,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,WAAW,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;IACzC,CAAC;IAAA,CAAC;CAGH"}

package/src/pipeline/pipeline-creator.ts

@@ -0,0 +1,16 @@
+import {IVP} from '../scripting-tools';
+import {getOutputNames} from './output';
+import {Pipeline} from './pipeline';
+
+/** Creator of a pipeline for in-webworker computations */
+export abstract class PipelineCreator {
+  protected ivp: IVP;
+  protected outputNames: string[];
+
+  constructor(ivp: IVP) {
+    this.ivp = ivp;
+    this.outputNames = getOutputNames(ivp);
+  };
+
+  public abstract getPipeline(inputs: Float64Array): Pipeline;
+}

package/src/pipeline/pipeline.js

@@ -0,0 +1,71 @@
+import { solveIvp } from '../worker-tools';
+import { BasicModelPipelineCreator } from './basic-pipeline-creator';
+import { UpdatesModelPipelineCreator } from './updates-pipeline-creator';
+import { CyclicModelPipelineCreator } from './loops-pipeline-creator';
+/** Return concatenated arrays */
+function concat(arr1, arr2) {
+    const result = new Float64Array(arr1.length + arr2.length);
+    result.set(arr1, 0);
+    result.set(arr2, arr1.length);
+    return result;
+}
+/** Apply pipeline to initial value problem with the source inputs, and return a solution */
+export function applyPipeline(pipeline, ivp, sourceInputs) {
+    let solution = [];
+    let currentSolution;
+    let func;
+    let inputs = new Float64Array(sourceInputs);
+    const wrappers = pipeline.wrappers;
+    let step;
+    let isFirstSolverCall = true;
+    const stepsCount = wrappers.length;
+    if (stepsCount < 1)
+        throw new Error(`Incorrect solution pipeline: ${stepsCount} steps.`);
+    // Perform pipeline steps
+    for (let idx = 0; idx < stepsCount; ++idx) {
+        step = wrappers[idx];
+        // Pre-processing
+        if (step.preproc !== null) {
+            func = new Function(step.preproc);
+            inputs = func(solution, inputs);
+            //console.log('Pre');
+        }
+        //console.log(inputs);
+        // Solving a problem
+        currentSolution = solveIvp(ivp, inputs);
+        // Output from solution
+        if (step.out !== null) {
+            func = new Function(step.out);
+            currentSolution = func(currentSolution, inputs);
+        }
+        if (isFirstSolverCall) {
+            solution = currentSolution;
+            isFirstSolverCall = false;
+        }
+        else
+            currentSolution.forEach((arr, kdx) => solution[kdx] = concat(solution[kdx], arr));
+        // Post-processing
+        if (step.postproc !== null) {
+            func = new Function(step.postproc);
+            inputs = func(solution, inputs);
+            //console.log('Pro');
+        }
+        //console.log(inputs);
+    }
+    // Compute final output
+    const out = pipeline.out;
+    if (out !== null) {
+        func = new Function(out);
+        return func(solution, sourceInputs);
+    }
+    return solution;
+} // performPipeline
+/** Return pipeline creator specified by the initial value problem */
+export function getPipelineCreator(ivp) {
+    if (ivp.updates !== null)
+        return new UpdatesModelPipelineCreator(ivp);
+    if (ivp.loop !== null)
+        return new CyclicModelPipelineCreator(ivp);
+    return new BasicModelPipelineCreator(ivp);
+}
+//# sourceMappingURL=pipeline.js.map

package/src/pipeline/pipeline.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pipeline.js","sourceRoot":"","sources":["pipeline.ts"],"names":[],"mappings":"AAAA,OAAO,EAAgB,QAAQ,EAAC,MAAM,iBAAiB,CAAC;AAExD,OAAO,EAAC,yBAAyB,EAAC,MAAM,0BAA0B,CAAC;AAEnE,OAAO,EAAC,2BAA2B,EAAC,MAAM,4BAA4B,CAAC;AACvE,OAAO,EAAC,0BAA0B,EAAC,MAAM,0BAA0B,CAAC;AAgBpE,iCAAiC;AACjC,SAAS,MAAM,CAAC,IAAkB,EAAE,IAAkB;IACpD,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC;IAC3D,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IACpB,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IAC9B,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,4FAA4F;AAC5F,MAAM,UAAU,aAAa,CAAC,QAAkB,EAAE,GAAkB,EAAE,YAA0B;IAC9F,IAAI,QAAQ,GAAmB,EAAE,CAAC;IAClC,IAAI,eAA+B,CAAC;IACpC,IAAI,IAAc,CAAC;IACnB,IAAI,MAAM,GAAG,IAAI,YAAY,CAAC,YAAY,CAAC,CAAC;IAC5C,MAAM,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC;IACnC,IAAI,IAAa,CAAC;IAClB,IAAI,iBAAiB,GAAG,IAAI,CAAC;IAE7B,MAAM,UAAU,GAAG,QAAQ,CAAC,MAAM,CAAC;IAEnC,IAAI,UAAU,GAAG,CAAC;QAChB,MAAM,IAAI,KAAK,CAAC,gCAAgC,UAAU,SAAS,CAAC,CAAC;IAEvE,yBAAyB;IACzB,KAAK,IAAI,GAAG,GAAG,CAAC,EAAE,GAAG,GAAG,UAAU,EAAE,EAAE,GAAG,EAAE,CAAC;QAC1C,IAAI,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC;QAErB,iBAAiB;QACjB,IAAI,IAAI,CAAC,OAAO,KAAK,IAAI,EAAE,CAAC;YAC1B,IAAI,GAAG,IAAI,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAClC,MAAM,GAAG,IAAI,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAChC,qBAAqB;QACvB,CAAC;QAED,sBAAsB;QAEtB,oBAAoB;QACpB,eAAe,GAAG,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAExC,uBAAuB;QACvB,IAAI,IAAI,CAAC,GAAG,KAAK,IAAI,EAAE,CAAC;YACtB,IAAI,GAAG,IAAI,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC9B,eAAe,GAAG,IAAI,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;QAClD,CAAC;QAED,IAAI,iBAAiB,EAAE,CAAC;YACtB,QAAQ,GAAG,eAAe,CAAC;YAC3B,iBAAiB,GAAG,KAAK,CAAC;QAC5B,CAAC;;YACC,eAAe,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;QAEpF,kBAAkB;QAClB,IAAI,IAAI,CAAC,QAAQ,KAAK,IAAI,EAAE,CAAC;YAC3B,IAAI,GAAG,IAAI,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACnC,MAAM,GAAG,IAAI,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAChC,qBAAqB;QACvB,CAAC;QAED,sBAAsB;IACxB,CAAC;IAED,uBAAuB;IACvB,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,CAAC;IAEzB,IAAI,GAAG,KAAK,IAAI,EAAE,CAAC;QACjB,IAAI,GAAG,IAAI,QAAQ,CAAC,GAAG,CAAC,CAAC;QAEzB,OAAO,IAAI,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;IACtC,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC,CAAC,kBAAkB;AAEpB,qEAAqE;AACrE,MAAM,UAAU,kBAAkB,CAAC,GAAQ;IACzC,IAAI,GAAG,CAAC,OAAO,KAAK,IAAI;QACtB,OAAO,IAAI,2BAA2B,CAAC,GAAG,CAAC,CAAC;IAE9C,IAAI,GAAG,CAAC,IAAI,KAAK,IAAI;QACnB,OAAO,IAAI,0BAA0B,CAAC,GAAG,CAAC,CAAC;IAE7C,OAAO,IAAI,yBAAyB,CAAC,GAAG,CAAC,CAAC;AAC5C,CAAC"}