nmr-processing 9.3.6 → 9.3.8-pre.1681135907

package/lib-esm/xyz/util/fft2d/fftDirectDimension.js

@@ -0,0 +1,50 @@
+import Matrix from 'ml-matrix';
+import { reimAutoPhaseCorrection, reimFFT, reimPhaseCorrection, } from 'ml-spectra-processing';
+import { apodization, } from '../../../apodization/apodization';
+import { digitalFilter } from './digitalFilter';
+import { removeDCOffset } from './removeDCOffset';
+import { zeroFilling } from './zeroFilling';
+export function fftDirectDimension(data, options) {
+    const { re: reData, im: imData } = data;
+    const nbRows = reData.length;
+    const nbColumns = reData[0].length;
+    const { zeroFilling: zeroFillingF2, digitalFilterValue = 0 } = options;
+    const { factor: f2Factor = 2 } = zeroFillingF2 || {};
+    const nbPointsF2 = (zeroFillingF2 === null || zeroFillingF2 === void 0 ? void 0 : zeroFillingF2.nbPoints) || 2 ** Math.round(Math.log2(nbColumns * f2Factor));
+    const reFFT = new Matrix(nbRows, nbPointsF2);
+    const imFFT = new Matrix(nbRows, nbPointsF2);
+    const { apodization: apodizationF2 } = options;
+    let { ph0: ph0F2, ph1: ph1F2, mode: pcModeF2 = 'no', } = (options === null || options === void 0 ? void 0 : options.phaseCorrection) || {};
+    for (let i = 0; i < nbRows; i++) {
+        const newData = digitalFilter({ re: reData[i], im: imData[i] }, { digitalFilterValue });
+        Object.assign(newData, removeDCOffset(newData, { digitalFilterValue }));
+        if (apodizationF2) {
+            const { re: aRe, im: aIm } = apodization(newData, apodizationF2);
+            Object.assign(newData, { re: aRe, im: aIm });
+        }
+        Object.assign(newData, zeroFilling(newData, { digitalFilterValue, nbPoints: nbPointsF2 }));
+        Object.assign(newData, reimFFT(newData, { applyZeroShift: true }));
+        const dfResidual = (digitalFilterValue - Math.round(digitalFilterValue)) * Math.PI * 2;
+        if (dfResidual > 0) {
+            Object.assign(newData, reimPhaseCorrection(newData, 0, -dfResidual));
+        }
+        if (i === 0 &&
+            pcModeF2 === 'pk' &&
+            ph0F2 === undefined &&
+            ph1F2 === undefined) {
+            const phased = reimAutoPhaseCorrection(newData, { magnitudeMode: false });
+            ph0F2 = phased.ph0;
+            ph1F2 = phased.ph1;
+        }
+        if (pcModeF2 === 'pk' && ph0F2 !== undefined && ph1F2 !== undefined) {
+            Object.assign(newData, reimPhaseCorrection(newData, (ph0F2 * Math.PI) / 180, (ph1F2 * Math.PI) / 180));
+        }
+        reFFT.setRow(i, newData.re);
+        imFFT.setRow(i, newData.im);
+    }
+    return {
+        re: reFFT,
+        im: imFFT,
+    };
+}
+//# sourceMappingURL=fftDirectDimension.js.map

package/lib-esm/xyz/util/fft2d/fftDirectDimension.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"fftDirectDimension.js","sourceRoot":"","sources":["../../../../src/xyz/util/fft2d/fftDirectDimension.ts"],"names":[],"mappings":"AACA,OAAO,MAAM,MAAM,WAAW,CAAC;AAC/B,OAAO,EACL,uBAAuB,EACvB,OAAO,EACP,mBAAmB,GACpB,MAAM,uBAAuB,CAAC;AAE/B,OAAO,EACL,WAAW,GAEZ,MAAM,kCAAkC,CAAC;AAE1C,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,WAAW,EAAsB,MAAM,eAAe,CAAC;AAgBhE,MAAM,UAAU,kBAAkB,CAChC,IAA8C,EAC9C,OAA+B;IAE/B,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;IACxC,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;IAC7B,MAAM,SAAS,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;IAEnC,MAAM,EAAE,WAAW,EAAE,aAAa,EAAE,kBAAkB,GAAG,CAAC,EAAE,GAAG,OAAO,CAAC;IAEvE,MAAM,EAAE,MAAM,EAAE,QAAQ,GAAG,CAAC,EAAE,GAAG,aAAa,IAAI,EAAE,CAAC;IACrD,MAAM,UAAU,GACd,CAAA,aAAa,aAAb,aAAa,uBAAb,aAAa,CAAE,QAAQ,KAAI,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC,CAAC,CAAC;IAE9E,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IAC7C,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IAE7C,MAAM,EAAE,WAAW,EAAE,aAAa,EAAE,GAAG,OAAO,CAAC;IAC/C,IAAI,EACF,GAAG,EAAE,KAAK,EACV,GAAG,EAAE,KAAK,EACV,IAAI,EAAE,QAAQ,GAAG,IAAI,GACtB,GAAG,CAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,eAAe,KAAI,EAAE,CAAC;IAEnC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,EAAE,CAAC,EAAE,EAAE;QAC/B,MAAM,OAAO,GAAG,aAAa,CAC3B,EAAE,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,EAChC,EAAE,kBAAkB,EAAE,CACvB,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,cAAc,CAAC,OAAO,EAAE,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC;QAExE,IAAI,aAAa,EAAE;YACjB,MAAM,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,GAAG,EAAE,GAAG,WAAW,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;YACjE,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC;SAC9C;QAED,MAAM,CAAC,MAAM,CACX,OAAO,EACP,WAAW,CAAC,OAAO,EAAE,EAAE,kBAAkB,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC,CACnE,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAEnE,MAAM,UAAU,GACd,CAAC,kBAAkB,GAAG,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC,GAAG,IAAI,CAAC,EAAE,GAAG,CAAC,CAAC;QACtE,IAAI,UAAU,GAAG,CAAC,EAAE;YAClB,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,mBAAmB,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;SACtE;QAED,IACE,CAAC,KAAK,CAAC;YACP,QAAQ,KAAK,IAAI;YACjB,KAAK,KAAK,SAAS;YACnB,KAAK,KAAK,SAAS,EACnB;YACA,MAAM,MAAM,GAAG,uBAAuB,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE,CAAC,CAAC;YAC1E,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC;YACnB,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC;SACpB;QACD,IAAI,QAAQ,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,SAAS,EAAE;YACnE,MAAM,CAAC,MAAM,CACX,OAAO,EACP,mBAAmB,CACjB,OAAO,EACP,CAAC,KAAK,GAAG,IAAI,CAAC,EAAE,CAAC,GAAG,GAAG,EACvB,CAAC,KAAK,GAAG,IAAI,CAAC,EAAE,CAAC,GAAG,GAAG,CACxB,CACF,CAAC;SACH;QAED,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC;QAC5B,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC;KAC7B;IAED,OAAO;QACL,EAAE,EAAE,KAAK;QACT,EAAE,EAAE,KAAK;KACV,CAAC;AACJ,CAAC"}

package/lib-esm/xyz/util/fft2d/fftIndirectDimension.js

@@ -0,0 +1,65 @@
+import Matrix from 'ml-matrix';
+import { reimAbsolute, reimAutoPhaseCorrection, reimFFT, reimPhaseCorrection, } from 'ml-spectra-processing';
+import { apodization, } from '../../../apodization/apodization';
+import { zeroFilling } from './zeroFilling';
+export function fftIndirectDimension(data, options) {
+    var _a;
+    const reData = Matrix.checkMatrix(data.re);
+    const imData = Matrix.checkMatrix(data.im);
+    const nbRows = reData.rows;
+    const nbColumns = reData.columns;
+    const { zeroFilling: zeroFillingF1 } = options;
+    const { factor: f1Factor = 2 } = zeroFillingF1 || {};
+    const nbPointsF1 = (zeroFillingF1 === null || zeroFillingF1 === void 0 ? void 0 : zeroFillingF1.nbPoints) || 2 ** Math.round(Math.log2(nbRows * f1Factor));
+    const { apodization: apodizationF1 } = options;
+    let { ph0: ph0F1, ph1: ph1F1, mode: pcModeF1, } = options.phaseCorrection || {};
+    const reResult = new Matrix(nbPointsF1, nbColumns);
+    const imResult = new Matrix(nbPointsF1, nbColumns);
+    for (let i = 0; i < nbColumns; i++) {
+        const { re: aRe, im: aIm } = apodization({
+            re: reData.getColumn(i),
+            im: imData.getColumn(i),
+        }, apodizationF1);
+        const newRe = [];
+        const newIm = [];
+        if (options.fnMode === 'echo') {
+            for (let j = 0; j < aRe.length; j += 2) {
+                newRe.push(aRe[j]);
+                newIm.push(aRe[j + 1]);
+            }
+        }
+        else {
+            for (let j = 0; j < aRe.length; j++) {
+                newRe.push(aRe[j]);
+                newIm.push(-aIm[j]);
+            }
+        }
+        let transformed = reimFFT(zeroFilling({
+            re: newRe,
+            im: newIm,
+        }, { digitalFilterValue: 0, nbPoints: nbPointsF1 }), { applyZeroShift: true });
+        let { re, im } = transformed;
+        if (((_a = options.phaseCorrection) === null || _a === void 0 ? void 0 : _a.mode) === 'mc') {
+            re = reimAbsolute({ re, im });
+            im = new Float64Array(im.length);
+        }
+        if (pcModeF1 === 'pk') {
+            if (i === 0 && ph0F1 === undefined && ph1F1 === undefined) {
+                //here we can improve it by checking if the experiment is phase sensitive.
+                const phased = reimAutoPhaseCorrection({ re, im }, { magnitudeMode: false });
+                ph0F1 = phased.ph0;
+                ph1F1 = phased.ph1;
+            }
+            if (ph0F1 !== undefined && ph1F1 !== undefined) {
+                Object.assign(transformed, reimPhaseCorrection(transformed, (ph0F1 * Math.PI) / 180, (ph1F1 * Math.PI) / 180));
+            }
+        }
+        reResult.setColumn(i, re);
+        imResult.setColumn(i, im);
+    }
+    return {
+        re: reResult,
+        im: imResult,
+    };
+}
+//# sourceMappingURL=fftIndirectDimension.js.map

package/lib-esm/xyz/util/fft2d/fftIndirectDimension.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"fftIndirectDimension.js","sourceRoot":"","sources":["../../../../src/xyz/util/fft2d/fftIndirectDimension.ts"],"names":[],"mappings":"AACA,OAAO,MAAM,MAAM,WAAW,CAAC;AAC/B,OAAO,EACL,YAAY,EACZ,uBAAuB,EACvB,OAAO,EACP,mBAAmB,GACpB,MAAM,uBAAuB,CAAC;AAE/B,OAAO,EACL,WAAW,GAEZ,MAAM,kCAAkC,CAAC;AAE1C,OAAO,EAAE,WAAW,EAAsB,MAAM,eAAe,CAAC;AAehE,MAAM,UAAU,oBAAoB,CAClC,IAAgE,EAChE,OAAiC;;IAEjC,MAAM,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC3C,MAAM,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAE3C,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC;IAC3B,MAAM,SAAS,GAAG,MAAM,CAAC,OAAO,CAAC;IAEjC,MAAM,EAAE,WAAW,EAAE,aAAa,EAAE,GAAG,OAAO,CAAC;IAC/C,MAAM,EAAE,MAAM,EAAE,QAAQ,GAAG,CAAC,EAAE,GAAG,aAAa,IAAI,EAAE,CAAC;IACrD,MAAM,UAAU,GACd,CAAA,aAAa,aAAb,aAAa,uBAAb,aAAa,CAAE,QAAQ,KAAI,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,QAAQ,CAAC,CAAC,CAAC;IAC3E,MAAM,EAAE,WAAW,EAAE,aAAa,EAAE,GAAG,OAAO,CAAC;IAC/C,IAAI,EACF,GAAG,EAAE,KAAK,EACV,GAAG,EAAE,KAAK,EACV,IAAI,EAAE,QAAQ,GACf,GAAG,OAAO,CAAC,eAAe,IAAI,EAAE,CAAC;IAElC,MAAM,QAAQ,GAAG,IAAI,MAAM,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;IACnD,MAAM,QAAQ,GAAG,IAAI,MAAM,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;IACnD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,EAAE,CAAC,EAAE,EAAE;QAClC,MAAM,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,GAAG,EAAE,GAAG,WAAW,CACtC;YACE,EAAE,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;YACvB,EAAE,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;SACxB,EACD,aAAa,CACd,CAAC;QAEF,MAAM,KAAK,GAAG,EAAE,CAAC;QACjB,MAAM,KAAK,GAAG,EAAE,CAAC;QACjB,IAAI,OAAO,CAAC,MAAM,KAAK,MAAM,EAAE;YAC7B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE;gBACtC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;gBACnB,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;aACxB;SACF;aAAM;YACL,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;gBACnC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;gBACnB,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;aACrB;SACF;QAED,IAAI,WAAW,GAAG,OAAO,CACvB,WAAW,CACT;YACE,EAAE,EAAE,KAAK;YACT,EAAE,EAAE,KAAK;SACV,EACD,EAAE,kBAAkB,EAAE,CAAC,EAAE,QAAQ,EAAE,UAAU,EAAE,CAChD,EACD,EAAE,cAAc,EAAE,IAAI,EAAE,CACzB,CAAC;QACF,IAAI,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,WAAW,CAAC;QAE7B,IAAI,CAAA,MAAA,OAAO,CAAC,eAAe,0CAAE,IAAI,MAAK,IAAI,EAAE;YAC1C,EAAE,GAAG,YAAY,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;YAC9B,EAAE,GAAG,IAAI,YAAY,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC;SAClC;QAED,IAAI,QAAQ,KAAK,IAAI,EAAE;YACrB,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,SAAS,EAAE;gBACzD,0EAA0E;gBAC1E,MAAM,MAAM,GAAG,uBAAuB,CACpC,EAAE,EAAE,EAAE,EAAE,EAAE,EACV,EAAE,aAAa,EAAE,KAAK,EAAE,CACzB,CAAC;gBACF,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC;gBACnB,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC;aACpB;YACD,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,SAAS,EAAE;gBAC9C,MAAM,CAAC,MAAM,CACX,WAAW,EACX,mBAAmB,CACjB,WAAW,EACX,CAAC,KAAK,GAAG,IAAI,CAAC,EAAE,CAAC,GAAG,GAAG,EACvB,CAAC,KAAK,GAAG,IAAI,CAAC,EAAE,CAAC,GAAG,GAAG,CACxB,CACF,CAAC;aACH;SACF;QAED,QAAQ,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QAC1B,QAAQ,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;KAC3B;IAED,OAAO;QACL,EAAE,EAAE,QAAQ;QACZ,EAAE,EAAE,QAAQ;KACb,CAAC;AACJ,CAAC"}

package/lib-esm/xyz/util/fft2d/removeDCOffset.js

@@ -0,0 +1,15 @@
+import { xMean } from 'ml-spectra-processing';
+export function removeDCOffset(data, options) {
+    let { digitalFilterValue = 0 } = options;
+    const nbPoints = data.re.length;
+    const newRe = new Float64Array(data.re);
+    const newIm = new Float64Array(data.im);
+    const averageRe = xMean(data.re.slice((nbPoints * 0.75) >> 0, nbPoints - digitalFilterValue));
+    const averageIm = xMean(data.im.slice((nbPoints * 0.75) >> 0, nbPoints - digitalFilterValue));
+    for (let i = digitalFilterValue; i < nbPoints - digitalFilterValue; i++) {
+        newRe[i] -= averageRe;
+        newIm[i] -= averageIm;
+    }
+    return { re: newRe, im: newIm };
+}
+//# sourceMappingURL=removeDCOffset.js.map

package/lib-esm/xyz/util/fft2d/removeDCOffset.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"removeDCOffset.js","sourceRoot":"","sources":["../../../../src/xyz/util/fft2d/removeDCOffset.ts"],"names":[],"mappings":"AAAA,OAAO,EAAY,KAAK,EAAE,MAAM,uBAAuB,CAAC;AAExD,MAAM,UAAU,cAAc,CAAC,IAAc,EAAE,OAAY;IACzD,IAAI,EAAE,kBAAkB,GAAG,CAAC,EAAE,GAAG,OAAO,CAAC;IACzC,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC;IAChC,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACxC,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACxC,MAAM,SAAS,GAAG,KAAK,CACrB,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,QAAQ,GAAG,kBAAkB,CAAC,CACrE,CAAC;IACF,MAAM,SAAS,GAAG,KAAK,CACrB,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,QAAQ,GAAG,kBAAkB,CAAC,CACrE,CAAC;IACF,KAAK,IAAI,CAAC,GAAG,kBAAkB,EAAE,CAAC,GAAG,QAAQ,GAAG,kBAAkB,EAAE,CAAC,EAAE,EAAE;QACvE,KAAK,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC;QACtB,KAAK,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC;KACvB;IACD,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC;AAClC,CAAC"}

package/lib-esm/xyz/util/fft2d/zeroFilling.js

@@ -0,0 +1,25 @@
+export function zeroFilling(data, options) {
+    let { nbPoints, factor = 2, digitalFilterValue: grpdly = 0 } = options;
+    if (!nbPoints) {
+        nbPoints = 2 ** Math.round(Math.log2(data.re.length * factor));
+    }
+    let pointsToShift;
+    if (grpdly > 0) {
+        pointsToShift = Math.floor(grpdly);
+    }
+    else {
+        pointsToShift = 0;
+    }
+    const { re, im } = data;
+    let newRE = new Float64Array(nbPoints);
+    let newIM = new Float64Array(nbPoints);
+    const length = Math.min(nbPoints, re.length);
+    newRE.set(re.slice(0, length - pointsToShift));
+    newIM.set(im.slice(0, length - pointsToShift));
+    if (pointsToShift > 0 && pointsToShift < nbPoints) {
+        newRE.set(re.slice(re.length - pointsToShift), nbPoints - pointsToShift);
+        newIM.set(im.slice(re.length - pointsToShift), nbPoints - pointsToShift);
+    }
+    return { re: newRE, im: newIM };
+}
+//# sourceMappingURL=zeroFilling.js.map

package/lib-esm/xyz/util/fft2d/zeroFilling.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"zeroFilling.js","sourceRoot":"","sources":["../../../../src/xyz/util/fft2d/zeroFilling.ts"],"names":[],"mappings":"AAQA,MAAM,UAAU,WAAW,CAAC,IAAc,EAAE,OAA2B;IACrE,IAAI,EAAE,QAAQ,EAAE,MAAM,GAAG,CAAC,EAAE,kBAAkB,EAAE,MAAM,GAAG,CAAC,EAAE,GAAG,OAAO,CAAC;IAEvE,IAAI,CAAC,QAAQ,EAAE;QACb,QAAQ,GAAG,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC;KAChE;IAED,IAAI,aAAa,CAAC;IAClB,IAAI,MAAM,GAAG,CAAC,EAAE;QACd,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;KACpC;SAAM;QACL,aAAa,GAAG,CAAC,CAAC;KACnB;IAED,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,IAAI,CAAC;IAExB,IAAI,KAAK,GAAG,IAAI,YAAY,CAAC,QAAQ,CAAC,CAAC;IACvC,IAAI,KAAK,GAAG,IAAI,YAAY,CAAC,QAAQ,CAAC,CAAC;IAEvC,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC;IAE7C,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,GAAG,aAAa,CAAC,CAAC,CAAC;IAC/C,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,GAAG,aAAa,CAAC,CAAC,CAAC;IAE/C,IAAI,aAAa,GAAG,CAAC,IAAI,aAAa,GAAG,QAAQ,EAAE;QACjD,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,MAAM,GAAG,aAAa,CAAC,EAAE,QAAQ,GAAG,aAAa,CAAC,CAAC;QACzE,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,MAAM,GAAG,aAAa,CAAC,EAAE,QAAQ,GAAG,aAAa,CAAC,CAAC;KAC1E;IAED,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC;AAClC,CAAC"}

package/lib-esm/xyz/xyzBidimensionalFFT.js

@@ -0,0 +1,75 @@
+import { quadrature } from './quadrature';
+import { fftDirectDimension, } from './util/fft2d/fftDirectDimension';
+import { fftIndirectDimension, } from './util/fft2d/fftIndirectDimension';
+export function xyzBidimensionalFFT(data, options) {
+    ensureData2DFid(data);
+    const { info, fnMode, digitalFilterValue = 0, zeroFilling: zeroFillingGeneral, } = options;
+    const orderedData = quadrature(data, { fnMode });
+    const { re: reData } = orderedData;
+    const nbRows = reData.length;
+    const nbColumns = reData[0].length;
+    const { direct = {}, indirect = {} } = options;
+    const { zeroFilling: zeroFillingF2 = zeroFillingGeneral } = direct;
+    const { factor: f2Factor = 2 } = zeroFillingF2 || {};
+    const nbPointsF2 = (zeroFillingF2 === null || zeroFillingF2 === void 0 ? void 0 : zeroFillingF2.nbPoints) || 2 ** Math.round(Math.log2(nbColumns * f2Factor));
+    const transformed = fftDirectDimension(orderedData, {
+        ...direct,
+        fnMode,
+        digitalFilterValue,
+        zeroFilling: { nbPoints: nbPointsF2 },
+    });
+    const { zeroFilling: zeroFillingF1 } = options;
+    const { factor: f1Factor = 2 } = zeroFillingF1 || {};
+    const nbPointsF1 = (zeroFillingF1 === null || zeroFillingF1 === void 0 ? void 0 : zeroFillingF1.nbPoints) || 2 ** Math.round(Math.log2(nbRows * f1Factor));
+    const { re: reResult, im: imResult } = fftIndirectDimension(transformed, {
+        ...indirect,
+        fnMode,
+        zeroFilling: { nbPoints: nbPointsF1 },
+    });
+    const minMaxXY = getMinMaxXY(info);
+    return {
+        re: {
+            z: convertToDoubleArray(reResult),
+            maxZ: reResult.max(),
+            minZ: reResult.min(),
+            ...minMaxXY,
+        },
+        im: {
+            z: convertToDoubleArray(imResult),
+            maxZ: imResult.max(),
+            minZ: imResult.min(),
+            ...minMaxXY,
+        },
+    };
+}
+function ensureData2DFid(data) {
+    if (!('re' in data)) {
+        throw new Error('bidimensional FFT only can be applied to raw data');
+    }
+}
+function convertToDoubleArray(data) {
+    const result = new Array(data.rows);
+    for (let i = 0; i < data.rows; i++) {
+        result[i] = new Float64Array(data.getRow(i));
+    }
+    return result;
+}
+function getMinMaxXY(info) {
+    const min = [];
+    const max = [];
+    for (let i = 0; i < info.baseFrequency.length; i++) {
+        const baseFrequency = Number.parseFloat(info.baseFrequency[i]);
+        const frequencyOffset = Number.parseFloat(info.frequencyOffset[i]);
+        const spectralWidth = Number.parseFloat(info.spectralWidth[i]);
+        const offset = frequencyOffset / baseFrequency;
+        min.push(offset - spectralWidth * 0.5);
+        max.push(offset + spectralWidth * 0.5);
+    }
+    return {
+        minX: min[0],
+        maxX: max[0],
+        minY: min[1],
+        maxY: max[1],
+    };
+}
+//# sourceMappingURL=xyzBidimensionalFFT.js.map

package/lib-esm/xyz/xyzBidimensionalFFT.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"xyzBidimensionalFFT.js","sourceRoot":"","sources":["../../src/xyz/xyzBidimensionalFFT.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAC1C,OAAO,EAEL,kBAAkB,GACnB,MAAM,iCAAiC,CAAC;AACzC,OAAO,EACL,oBAAoB,GAErB,MAAM,mCAAmC,CAAC;AAgB3C,MAAM,UAAU,mBAAmB,CACjC,IAAY,EACZ,OAAgC;IAEhC,eAAe,CAAC,IAAI,CAAC,CAAC;IAEtB,MAAM,EACJ,IAAI,EACJ,MAAM,EACN,kBAAkB,GAAG,CAAC,EACtB,WAAW,EAAE,kBAAkB,GAChC,GAAG,OAAO,CAAC;IAEZ,MAAM,WAAW,GAAG,UAAU,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;IAEjD,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,WAAW,CAAC;IACnC,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;IAC7B,MAAM,SAAS,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;IACnC,MAAM,EAAE,MAAM,GAAG,EAAE,EAAE,QAAQ,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC;IAE/C,MAAM,EAAE,WAAW,EAAE,aAAa,GAAG,kBAAkB,EAAE,GAAG,MAAM,CAAC;IAEnE,MAAM,EAAE,MAAM,EAAE,QAAQ,GAAG,CAAC,EAAE,GAAG,aAAa,IAAI,EAAE,CAAC;IACrD,MAAM,UAAU,GACd,CAAA,aAAa,aAAb,aAAa,uBAAb,aAAa,CAAE,QAAQ,KAAI,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC,CAAC,CAAC;IAE9E,MAAM,WAAW,GAAG,kBAAkB,CAAC,WAAW,EAAE;QAClD,GAAG,MAAM;QACT,MAAM;QACN,kBAAkB;QAClB,WAAW,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE;KACtC,CAAC,CAAC;IAEH,MAAM,EAAE,WAAW,EAAE,aAAa,EAAE,GAAG,OAAO,CAAC;IAC/C,MAAM,EAAE,MAAM,EAAE,QAAQ,GAAG,CAAC,EAAE,GAAG,aAAa,IAAI,EAAE,CAAC;IACrD,MAAM,UAAU,GACd,CAAA,aAAa,aAAb,aAAa,uBAAb,aAAa,CAAE,QAAQ,KAAI,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,QAAQ,CAAC,CAAC,CAAC;IAC3E,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,QAAQ,EAAE,GAAG,oBAAoB,CAAC,WAAW,EAAE;QACvE,GAAG,QAAQ;QACX,MAAM;QACN,WAAW,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE;KACtC,CAAC,CAAC;IAEH,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;IACnC,OAAO;QACL,EAAE,EAAE;YACF,CAAC,EAAE,oBAAoB,CAAC,QAAQ,CAAC;YACjC,IAAI,EAAE,QAAQ,CAAC,GAAG,EAAE;YACpB,IAAI,EAAE,QAAQ,CAAC,GAAG,EAAE;YACpB,GAAG,QAAQ;SACZ;QACD,EAAE,EAAE;YACF,CAAC,EAAE,oBAAoB,CAAC,QAAQ,CAAC;YACjC,IAAI,EAAE,QAAQ,CAAC,GAAG,EAAE;YACpB,IAAI,EAAE,QAAQ,CAAC,GAAG,EAAE;YACpB,GAAG,QAAQ;SACZ;KACF,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,IAAY;IACnC,IAAI,CAAC,CAAC,IAAI,IAAI,IAAI,CAAC,EAAE;QACnB,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAC;KACtE;AACH,CAAC;AAED,SAAS,oBAAoB,CAAC,IAAY;IACxC,MAAM,MAAM,GAAkB,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACnD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE;QAClC,MAAM,CAAC,CAAC,CAAC,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;KAC9C;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,WAAW,CAAC,IAAY;IAC/B,MAAM,GAAG,GAAa,EAAE,CAAC;IACzB,MAAM,GAAG,GAAa,EAAE,CAAC;IACzB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;QAClD,MAAM,aAAa,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC;QAC/D,MAAM,eAAe,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC;QACnE,MAAM,aAAa,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC;QAC/D,MAAM,MAAM,GAAG,eAAe,GAAG,aAAa,CAAC;QAC/C,GAAG,CAAC,IAAI,CAAC,MAAM,GAAG,aAAa,GAAG,GAAG,CAAC,CAAC;QACvC,GAAG,CAAC,IAAI,CAAC,MAAM,GAAG,aAAa,GAAG,GAAG,CAAC,CAAC;KACxC;IAED,OAAO;QACL,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC;QACZ,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC;QACZ,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC;QACZ,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC;KACb,CAAC;AACJ,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "nmr-processing",
-  "version": "9.3.6",
+  "version": "9.3.8-pre.1681135907",
   "description": "Pure functions allowing to process NMR spectra.",
   "main": "./lib/index.js",
   "module": "./lib-esm/index.js",
@@ -38,20 +38,20 @@
   },
   "homepage": "https://github.com/cheminfo/nmr-processing#readme",
   "devDependencies": {
-    "@types/jest": "^29.2.3",
-    "cheminfo-build": "^1.1.11",
+    "@types/jest": "^29.5.0",
+    "cheminfo-build": "^1.2.0",
     "cheminfo-types": "^1.4.0",
-    "eslint": "^8.34.0",
-    "eslint-config-cheminfo-typescript": "^11.2.2",
-    "jest": "^29.3.1",
+    "eslint": "^8.36.0",
+    "eslint-config-cheminfo-typescript": "^11.3.1",
+    "jest": "^29.5.0",
     "jest-matcher-deep-close-to": "^3.0.2",
     "md5": "^2.3.0",
     "nmr-xy-testdata": "^0.5.1",
-    "openchemlib": "^8.0.1",
-    "prettier": "^2.8.4",
-    "rimraf": "^4.1.2",
-    "ts-jest": "^29.0.3",
-    "typescript": "^4.9.3"
+    "openchemlib": "^8.2.0",
+    "prettier": "^2.8.7",
+    "rimraf": "^4.4.1",
+    "ts-jest": "^29.0.5",
+    "typescript": "^5.0.2"
   },
   "dependencies": {
     "@lukeed/uuid": "^2.0.0",
@@ -60,12 +60,12 @@
     "form-data": "^4.0.0",
     "gyromagnetic-ratio": "^1.0.0",
     "is-any-array": "^2.0.0",
-    "linear-sum-assignment": "^1.0.4",
+    "linear-sum-assignment": "^1.0.5",
     "ml-array-mean": "^1.1.6",
     "ml-array-rescale": "^1.3.7",
     "ml-array-sum": "^1.1.6",
-    "ml-direct": "^0.1.1",
-    "ml-gsd": "^12.1.2",
+    "ml-direct": "^0.1.3",
+    "ml-gsd": "^12.1.3",
     "ml-hclust": "^3.1.0",
     "ml-levenberg-marquardt": "^4.1.0",
     "ml-matrix": "^6.10.4",
@@ -74,11 +74,11 @@
     "ml-peak-shape-generator": "^4.1.2",
     "ml-simple-clustering": "^0.1.0",
     "ml-sparse-matrix": "^2.1.0",
-    "ml-spectra-processing": "^11.17.0",
+    "ml-spectra-processing": "^12.0.0",
     "ml-tree-set": "^0.1.1",
     "nmr-correlation": "^2.3.3",
     "nmredata": "^0.9.2",
-    "openchemlib-utils": "^2.1.2",
-    "spectrum-generator": "^8.0.6"
+    "openchemlib-utils": "^2.4.0",
+    "spectrum-generator": "^8.0.7"
   }
 }

package/src/apodization/apodization.ts

@@ -10,23 +10,25 @@ export interface DataReIm {
   im: DoubleArray;
 }
 
-interface ApodizationOptions {
+export interface ApodizationOptions {
   /**
    * number of points at the end of the shape to apply the window function inverted
    * @default 0
    */
   pointsToShift?: number;
-  compose: ShapeOptions;
+  compose?: ShapeOptions;
 }
-export function apodization(data: DataReIm, options: ApodizationOptions) {
+export function apodization(data: DataReIm, options: ApodizationOptions = {}) {
   const { compose: composeOptions, pointsToShift } = options;
+
+  if (!composeOptions) return { ...data, windowData: [] };
+
   const windowData = compose(composeOptions);
 
   const applyWindowOptions = {
     windowData,
     pointsToShift,
   };
-
   const re = applyWindow(data.re, applyWindowOptions);
   const im = applyWindow(data.im, applyWindowOptions);
 

package/src/apodization/applyWindow.ts

@@ -42,7 +42,6 @@ export function applyWindow(data: DoubleArray, options: ApplyWindowOptions) {
   for (let i = start, j = 0; i < firstEndPoint; i++) {
     output[i] *= windowData[j++];
   }
-
   for (let i = dataLength - 1, j = 0; i > dataLength - pointsToShift - 1; i--) {
     output[i] *= windowData[j++];
   }

package/src/apodization/compose.ts

@@ -36,6 +36,7 @@ export function compose(shapeOptions: ShapeOptions) {
       'length' in shapeOptions ? shapeOptions.length : dataLength - start;
 
     const windowData = getData(getFunction(shape), length);
+
     applyWindow(data, {
       windowData,
       length,

package/src/apodization/getFunction.ts

@@ -1,6 +1,7 @@
 import type { WindowFunctions } from './shapes/WindowFunctions';
 import { exponential } from './shapes/exponential';
 import { lorentzToGauss } from './shapes/lorentzToGauss';
+import { sineBell } from './shapes/sineBell';
 
 export function getFunction(shape: WindowFunctions) {
   const { kind, options } = shape;
@@ -9,6 +10,8 @@ export function getFunction(shape: WindowFunctions) {
       return exponential(options);
     case 'lorentzToGauss':
       return lorentzToGauss(options);
+    case 'sineBell':
+      return sineBell(options);
     default:
       throw Error(`Unknown distribution ${kind as string}`);
   }

package/src/apodization/shapes/WindowFunctions.ts

@@ -1,5 +1,6 @@
 import type { ExponentialOptions } from './exponential';
 import type { LorentzToGaussOptions } from './lorentzToGauss';
+import { SineBellOptions } from './sineBell';
 
 interface Exponential {
   kind: 'exponential';
@@ -11,4 +12,9 @@ interface LorentToGauss {
   options: LorentzToGaussOptions;
 }
 
-export type WindowFunctions = Exponential | LorentToGauss;
+interface SineBell {
+  kind: 'sineBell';
+  options: SineBellOptions;
+}
+
+export type WindowFunctions = Exponential | LorentToGauss | SineBell;

package/src/apodization/shapes/sineBell.ts

@@ -0,0 +1,41 @@
+export interface SineBellOptions {
+  /**
+   * Specifies the exponent of the sine-bell; Non-integer values are allowed.
+   * Common values are 1.0 (for ordinary sine-bell) and 2.0 (for squared-bell functions).
+   * @default 1
+   */
+  exponent?: number;
+  /**
+   * Specifies the starting point of the sine-bell in units of pi radians.
+   * Common values are 0.0 (for a sine window which starts height at 0.0) and
+   * 0.5 (for a cosine window, which starts at height 1.0).
+   * @default 0
+   */
+  offset?: number;
+  /**
+   * Specifies the ending point of the sine-bell in units of pi radians.
+   * Common values are 1.0 (for a window which goes to 0.0 height at the last point) and
+   * 0.95 (for a window which doesn't go all the way to 0.0).
+   * @default 1
+   */
+  end?: number;
+  /**
+   * Specifies the number of points in the window function.
+   */
+  length: number;
+}
+
+export function sineBell(options: SineBellOptions) {
+  const { exponent = 1, offset = 0, end = 1, length } = options;
+
+  if (length === undefined) {
+    throw new Error('length options should exists for sineBell shape');
+  }
+
+  if (end < 0 || end > 1) {
+    throw new Error(`the end parameter should be [0-1]`);
+  }
+  const c1 = Math.PI * offset;
+  const c2 = (Math.PI * (end - offset)) / (length - 1);
+  return (i: number) => Math.pow(Math.sin(c1 + c2 * i), exponent);
+}

package/src/apodization/utils/getData.ts

@@ -9,7 +9,6 @@ export function getData(func: (i: number) => number, length: number) {
     data[i] = value;
     if (value > max) max = value;
   }
-
   for (let i = 0; i < length; i++) data[i] /= max;
   return data;
 }

package/src/assignment/utils/buildAssignments.ts

@@ -60,9 +60,6 @@ export async function buildAssignments(props: BuildAssignmentsProps) {
     chemicalShiftRestriction = true,
   } = restrictionByCS;
 
-  let date = new Date();
-  let timeStart = date.getTime();
-
   let store: StoreAssignments = {
     solutions: new TreeSet(comparator),
     nSolutions: 0,
@@ -101,6 +98,8 @@ export async function buildAssignments(props: BuildAssignmentsProps) {
     nSolutions: 0,
   };
 
+  const timeStart = Date.now();
+
   exploreTreeRec(
     {
       nSources,

package/src/assignment/utils/exploreTreeRec.ts

@@ -43,8 +43,7 @@ export function exploreTreeRec(
     diaIDPeerPossibleAssignment,
   } = props;
 
-  const currentDate = new Date();
-  if (currentDate.getTime() - timeStart > timeout) {
+  if (Date.now() - timeStart > timeout) {
     new Error('timeout expired');
     return store;
   }

package/src/assignment/utils/getAssignment/buildAssignments.ts

@@ -130,8 +130,6 @@ export async function buildAssignments(props: BuildAssignmentInput) {
     targets,
   } = props;
 
-  let date = new Date();
-  let timeStart = date.getTime();
   let lowerBoundScore = minScore;
 
   let store: StoreAssignments = {
@@ -208,6 +206,8 @@ export async function buildAssignments(props: BuildAssignmentInput) {
       nSolutions: 0,
     };
 
+    const timeStart = Date.now();
+
     for (let partial of sourceOfPartials) {
       exploreTree(
         {

package/src/assignment/utils/getAssignment/exploreTree.ts

@@ -55,8 +55,7 @@ export function exploreTree(
     diaIDPeerPossibleAssignment,
   } = props;
 
-  const currentDate = new Date();
-  if (currentDate.getTime() - timeStart > timeout) {
+  if (Date.now() - timeStart > timeout) {
     new Error('timeout expired');
     return store;
   }

package/src/utilities/checkData2DFid.ts

@@ -0,0 +1,10 @@
+import { Data2D, Data2DFid } from '../xyz/Data2D';
+
+export type Data2DFidComplete = Required<Data2DFid>;
+export function checkData2DFid(
+  data: Data2D,
+): asserts data is Data2DFidComplete {
+  if (!('re' in data && 'im' in data)) {
+    throw new Error('imaginary (im) data should exists');
+  }
+}

package/src/xyz/Data2D.ts

@@ -0,0 +1,25 @@
+import { DoubleArray } from 'cheminfo-types';
+
+export interface MinMaxContent {
+  z: DoubleArray[];
+  minX: number;
+  minY: number;
+  minZ: number;
+  maxX: number;
+  maxY: number;
+  maxZ: number;
+}
+
+export type Data2D = Data2DFid | Data2DFt;
+
+export interface Data2DFid {
+  re: MinMaxContent;
+  im?: MinMaxContent;
+}
+
+export interface Data2DFt {
+  rr: MinMaxContent;
+  ri?: MinMaxContent;
+  ir?: MinMaxContent;
+  ii?: MinMaxContent;
+}

package/src/xyz/quadrature.ts

@@ -0,0 +1,69 @@
+import { DoubleArray } from 'cheminfo-types';
+import { xAdd, xMultiply, xSubtract } from 'ml-spectra-processing';
+
+import { checkData2DFid, Data2DFidComplete } from '../utilities/checkData2DFid';
+import { Data2D } from './Data2D';
+
+export function quadrature(data: Data2D, options: { fnMode: string }) {
+  checkData2DFid(data);
+
+  const { fnMode } = options;
+
+  switch (fnMode) {
+    case 'ttpi':
+      return ttpiQuadrature(data);
+    case 'states':
+      return shrQuadrature(data);
+    case 'echo':
+      return echoQuadrature(data);
+    default:
+      return {
+        re: data.re.z,
+        im: data.im.z,
+      };
+  }
+}
+
+function echoQuadrature(data: Data2DFidComplete) {
+  const {
+    re: { z: reMatrix },
+    im: { z: imMatrix },
+  } = data;
+
+  const newRe: DoubleArray[] = [];
+  const newIm: DoubleArray[] = [];
+  for (let i = 0; i < reMatrix.length; i += 2) {
+    let reCosine = reMatrix[i];
+    let imCosine = imMatrix[i];
+    let reSine = reMatrix[i + 1];
+    let imSine = imMatrix[i + 1];
+
+    newRe.push(xSubtract(xMultiply(imSine, -1), imCosine));
+    newRe.push(xSubtract(reSine, reCosine));
+    newIm.push(xAdd(reSine, reCosine));
+    newIm.push(xSubtract(imSine, imCosine));
+  }
+
+  return { re: newRe, im: newIm };
+}
+
+function shrQuadrature(data: Data2DFidComplete) {
+  const { re, im } = data;
+  return { re: re.z, im: im.z };
+}
+
+function ttpiQuadrature(data: Data2DFidComplete) {
+  const newRe: DoubleArray[] = [];
+  const newIm: DoubleArray[] = [];
+
+  const {
+    re: { z: reMatrix },
+    im: { z: imMatrix },
+  } = data;
+  for (let i = 0; i < reMatrix.length; i++) {
+    newRe.push(i % 2 ? reMatrix[i].slice() : xMultiply(reMatrix[i], -1));
+    newIm.push(i % 2 ? imMatrix[i].slice() : xMultiply(imMatrix[i], -1));
+  }
+
+  return { re: newRe, im: newIm };
+}