@datagrok-libraries/utils 0.0.3 → 0.0.7

package/src/spe.js

@@ -0,0 +1,151 @@
+import { calculateEuclideanDistance, calcDistanceMatrix, fillRandomMatrix, vectorAdd, randomInt } from './operations';
+/**
+ * Implements stochastic proximity embedding.
+ *
+ * @export
+ * @class SPEBase
+ * @link doi:10.1016/S1093-3263(03)00155-4
+ */
+export class SPEBase {
+    /**
+     * Creates an instance of SPEBase.
+     * @param {Options} [options] Options to pass to the constructor.
+     * @memberof SPEBase
+     */
+    constructor(options) {
+        var _a, _b, _c, _d, _e, _f, _g;
+        this.steps = (_a = options === null || options === void 0 ? void 0 : options.steps) !== null && _a !== void 0 ? _a : 0;
+        this.cycles = (_b = options === null || options === void 0 ? void 0 : options.cycles) !== null && _b !== void 0 ? _b : 1e6;
+        // Select a cutoff distance {cutoff} and ...
+        this.cutoff = (_c = options === null || options === void 0 ? void 0 : options.cutoff) !== null && _c !== void 0 ? _c : 0;
+        // ... an initial learning rate {lambda} > 0
+        this.lambda = (_d = options === null || options === void 0 ? void 0 : options.lambda) !== null && _d !== void 0 ? _d : 2.0;
+        this.dlambda = (_e = options === null || options === void 0 ? void 0 : options.dlambda) !== null && _e !== void 0 ? _e : 0.01;
+        this.lambda2 = this.lambda / 2.;
+        this.dlambda2 = this.dlambda / 2.;
+        this.epsilon = (_f = options === null || options === void 0 ? void 0 : options.epsilon) !== null && _f !== void 0 ? _f : 1e-10;
+        // eslint-disable-next-line brace-style
+        this.distanceFunction = (_g = options === null || options === void 0 ? void 0 : options.distance) !== null && _g !== void 0 ? _g : calculateEuclideanDistance;
+        this.distance = [];
+    }
+    /**
+     * Initializes distance matrix.
+     *
+     * @protected
+     * @param {Vectors} vectors Input vectors to calculate distance between.
+     * @memberof SPEBase
+     */
+    initDistance(vectors) {
+        this.distance = calcDistanceMatrix(vectors, this.distanceFunction);
+    }
+    /**
+     * Calculates distance between the two vectors given.
+     *
+     * @param {Vectors} vectors Set of vectors to calculate distances between.
+     * @param {number} index1 Index of the first vector of the pair.
+     * @param {number} index2 Index of the second vector of the pair.
+     * @return {number} Distance between these two vectors.
+     */
+    calcDistance(vectors, index1, index2) {
+        return this.distance[index1][index2];
+    }
+    /**
+     * Embeds the vectors given into a two-dimensional space.
+     *
+     * @param {Vectors} vectors D-dimensional coordinates.
+     * @return {Coordinates} SPE coordinates in D space.
+     */
+    embed(vectors) {
+        const nItems = vectors.length;
+        const areaWidth = 40;
+        // Initialize the D-dimensional coordinates of the N points.
+        const coordinates = fillRandomMatrix(nItems, SPEBase.dimension, areaWidth);
+        let lambda2 = this.lambda2;
+        if (this.steps == 0) {
+            this.steps = vectors.length - 1;
+        }
+        this.initDistance(vectors);
+        for (let cycle = 0; cycle < this.cycles; ++cycle) {
+            for (let step = 0; step < this.steps; ++step) {
+                // Select two points, i and j, at random, ...
+                const i = randomInt(nItems);
+                let j = randomInt(nItems);
+                while (i == j)
+                    j = randomInt(nItems);
+                const rowi = coordinates[i];
+                const rowj = coordinates[j];
+                // ... retrieve (or evaluate) their proximity in the input space, rij and ...
+                const r = this.calcDistance(vectors, i, j);
+                // ... compute their Euclidean distance on the D-dimensional map, dij.
+                const d = calculateEuclideanDistance(rowi, rowj);
+                // If rij <= rc, or if rij > rc and dij < rij ...
+                if ((this.cutoff == 0) || (r <= this.cutoff) || (d < r)) {
+                    const multiplier = lambda2 * (r - d) / (d + this.epsilon);
+                    // ... update the coordinates xi and xj.
+                    const diffIJ = vectorAdd(rowi, rowj, -1);
+                    coordinates[i] = vectorAdd(rowi, diffIJ, multiplier);
+                    coordinates[j] = vectorAdd(rowj, diffIJ, -multiplier);
+                }
+            }
+            // Decrease the learning rate {lambda} by a prescribed {dlambda}.
+            lambda2 -= this.dlambda2;
+            if (lambda2 <= 0.) {
+                break;
+            }
+        }
+        return coordinates;
+    }
+}
+SPEBase.dimension = 2;
+/**
+ * Implements modified stochastic proximity embedding.
+ *
+ * @export
+ * @class PSPEBase
+ * @link doi:10.1016/S1093-3263(03)00155-4
+ */
+export class PSPEBase extends SPEBase {
+    /**
+     * Embeds the vectors given into a two-dimensional space using a modified update rule.
+     *
+     * @param {Vectors} vectors D-dimensional coordinates.
+     * @return {Coordinates} SPE coordinates in D space.
+     */
+    embed(vectors) {
+        const nItems = vectors.length;
+        const areaWidth = 40;
+        //  Initialize the D-dimensional coordinates of the N points.
+        const coordinates = fillRandomMatrix(nItems, PSPEBase.dimension, areaWidth);
+        let lambda = this.lambda;
+        this.initDistance(vectors);
+        for (let cycle = 0; cycle < this.cycles; ++cycle) {
+            // Select a point, i, at random (pivot).
+            const i = randomInt(nItems);
+            const rowi = coordinates[i];
+            // For every point j != i ...
+            for (let j = 0; j < nItems; ++j) {
+                if (i == j)
+                    continue;
+                const rowj = coordinates[j];
+                // ... retrieve (or evaluate) its proximity to i in the input space, rij ...
+                const r = this.calcDistance(vectors, i, j);
+                // ... and compute their Euclidean distance on the D-dimensional map, dij.
+                const d = calculateEuclideanDistance(rowi, rowj);
+                // If rij <= rc, or if rij > rc and dij < rij ...
+                if ((this.cutoff == 0) || (r <= this.cutoff) || (d < r)) {
+                    const multiplier = lambda * (r - d) / (d + this.epsilon);
+                    const diffIJ = vectorAdd(rowi, rowj, -1);
+                    // ... update the coordinates xj.
+                    coordinates[j] = vectorAdd(rowj, diffIJ, -multiplier);
+                }
+            }
+            // Decrease the learning rate {lambda} by a prescribed {dlambda}.
+            lambda -= this.dlambda;
+            if (lambda <= 0.) {
+                break;
+            }
+        }
+        return coordinates;
+    }
+}
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoic3BlLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsic3BlLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUNBLE9BQU8sRUFBQywwQkFBMEIsRUFBRSxrQkFBa0IsRUFBRSxnQkFBZ0IsRUFBRSxTQUFTLEVBQUUsU0FBUyxFQUFDLE1BQU0sY0FBYyxDQUFDO0FBRXBIOzs7Ozs7R0FNRztBQUNILE1BQU0sT0FBTyxPQUFPO0lBYWxCOzs7O09BSUc7SUFDSCxZQUFZLE9BQWlCOztRQUMzQixJQUFJLENBQUMsS0FBSyxHQUFHLE1BQUEsT0FBTyxhQUFQLE9BQU8sdUJBQVAsT0FBTyxDQUFFLEtBQUssbUNBQUksQ0FBQyxDQUFDO1FBQ2pDLElBQUksQ0FBQyxNQUFNLEdBQUcsTUFBQSxPQUFPLGFBQVAsT0FBTyx1QkFBUCxPQUFPLENBQUUsTUFBTSxtQ0FBSSxHQUFHLENBQUM7UUFDckMsNENBQTRDO1FBQzVDLElBQUksQ0FBQyxNQUFNLEdBQUcsTUFBQSxPQUFPLGFBQVAsT0FBTyx1QkFBUCxPQUFPLENBQUUsTUFBTSxtQ0FBSSxDQUFDLENBQUM7UUFDbkMsNENBQTRDO1FBQzVDLElBQUksQ0FBQyxNQUFNLEdBQUcsTUFBQSxPQUFPLGFBQVAsT0FBTyx1QkFBUCxPQUFPLENBQUUsTUFBTSxtQ0FBSSxHQUFHLENBQUM7UUFDckMsSUFBSSxDQUFDLE9BQU8sR0FBRyxNQUFBLE9BQU8sYUFBUCxPQUFPLHVCQUFQLE9BQU8sQ0FBRSxPQUFPLG1DQUFJLElBQUksQ0FBQztRQUN4QyxJQUFJLENBQUMsT0FBTyxHQUFHLElBQUksQ0FBQyxNQUFNLEdBQUMsRUFBRSxDQUFDO1FBQzlCLElBQUksQ0FBQyxRQUFRLEdBQUcsSUFBSSxDQUFDLE9BQU8sR0FBQyxFQUFFLENBQUM7UUFDaEMsSUFBSSxDQUFDLE9BQU8sR0FBRyxNQUFBLE9BQU8sYUFBUCxPQUFPLHVCQUFQLE9BQU8sQ0FBRSxPQUFPLG1DQUFJLEtBQUssQ0FBQztRQUN6Qyx1Q0FBdUM7UUFDdkMsSUFBSSxDQUFDLGdCQUFnQixHQUFHLE1BQUEsT0FBTyxhQUFQLE9BQU8sdUJBQVAsT0FBTyxDQUFFLFFBQVEsbUNBQUksMEJBQTBCLENBQUM7UUFDeEUsSUFBSSxDQUFDLFFBQVEsR0FBRyxFQUFFLENBQUM7SUFDckIsQ0FBQztJQUVEOzs7Ozs7T0FNRztJQUNPLFlBQVksQ0FBQyxPQUFnQjtRQUNyQyxJQUFJLENBQUMsUUFBUSxHQUFHLGtCQUFrQixDQUFDLE9BQU8sRUFBRSxJQUFJLENBQUMsZ0JBQWdCLENBQUMsQ0FBQztJQUNyRSxDQUFDO0lBRUQ7Ozs7Ozs7T0FPRztJQUNPLFlBQVksQ0FBQyxPQUFnQixFQUFFLE1BQWMsRUFBRSxNQUFjO1FBQ3JFLE9BQU8sSUFBSSxDQUFDLFFBQVEsQ0FBQyxNQUFNLENBQUMsQ0FBQyxNQUFNLENBQUMsQ0FBQztJQUN2QyxDQUFDO0lBRUQ7Ozs7O09BS0c7SUFDSSxLQUFLLENBQUMsT0FBZ0I7UUFDM0IsTUFBTSxNQUFNLEdBQUcsT0FBTyxDQUFDLE1BQU0sQ0FBQztRQUM5QixNQUFNLFNBQVMsR0FBRyxFQUFFLENBQUM7UUFDckIsNERBQTREO1FBQzVELE1BQU0sV0FBVyxHQUFHLGdCQUFnQixDQUFDLE1BQU0sRUFBRSxPQUFPLENBQUMsU0FBUyxFQUFFLFNBQVMsQ0FBQyxDQUFDO1FBRTNFLElBQUksT0FBTyxHQUFHLElBQUksQ0FBQyxPQUFPLENBQUM7UUFFM0IsSUFBSSxJQUFJLENBQUMsS0FBSyxJQUFJLENBQUMsRUFBRTtZQUNuQixJQUFJLENBQUMsS0FBSyxHQUFHLE9BQU8sQ0FBQyxNQUFNLEdBQUMsQ0FBQyxDQUFDO1NBQy9CO1FBRUQsSUFBSSxDQUFDLFlBQVksQ0FBQyxPQUFPLENBQUMsQ0FBQztRQUUzQixLQUFLLElBQUksS0FBSyxHQUFHLENBQUMsRUFBRSxLQUFLLEdBQUcsSUFBSSxDQUFDLE1BQU0sRUFBRSxFQUFFLEtBQUssRUFBRTtZQUNoRCxLQUFLLElBQUksSUFBSSxHQUFHLENBQUMsRUFBRSxJQUFJLEdBQUcsSUFBSSxDQUFDLEtBQUssRUFBRSxFQUFFLElBQUksRUFBRTtnQkFDNUMsNkNBQTZDO2dCQUM3QyxNQUFNLENBQUMsR0FBRyxTQUFTLENBQUMsTUFBTSxDQUFDLENBQUM7Z0JBQUMsSUFBSSxDQUFDLEdBQUcsU0FBUyxDQUFDLE1BQU0sQ0FBQyxDQUFDO2dCQUN2RCxPQUFPLENBQUMsSUFBSSxDQUFDO29CQUFFLENBQUMsR0FBRyxTQUFTLENBQUMsTUFBTSxDQUFDLENBQUM7Z0JBRXJDLE1BQU0sSUFBSSxHQUFHLFdBQVcsQ0FBQyxDQUFDLENBQUMsQ0FBQztnQkFBQyxNQUFNLElBQUksR0FBRyxXQUFXLENBQUMsQ0FBQyxDQUFDLENBQUM7Z0JBRXpELDZFQUE2RTtnQkFDN0UsTUFBTSxDQUFDLEdBQUcsSUFBSSxDQUFDLFlBQVksQ0FBQyxPQUFPLEVBQUUsQ0FBQyxFQUFFLENBQUMsQ0FBQyxDQUFDO2dCQUMzQyxzRUFBc0U7Z0JBQ3RFLE1BQU0sQ0FBQyxHQUFHLDBCQUEwQixDQUFDLElBQUksRUFBRSxJQUFJLENBQUMsQ0FBQztnQkFFakQsaURBQWlEO2dCQUNqRCxJQUFJLENBQUMsSUFBSSxDQUFDLE1BQU0sSUFBSSxDQUFDLENBQUMsSUFBSSxDQUFDLENBQUMsSUFBSSxJQUFJLENBQUMsTUFBTSxDQUFDLElBQUksQ0FBQyxDQUFDLEdBQUcsQ0FBQyxDQUFDLEVBQUU7b0JBQ3ZELE1BQU0sVUFBVSxHQUFHLE9BQU8sR0FBQyxDQUFDLENBQUMsR0FBQyxDQUFDLENBQUMsR0FBQyxDQUFDLENBQUMsR0FBQyxJQUFJLENBQUMsT0FBTyxDQUFDLENBQUM7b0JBQ2xELHdDQUF3QztvQkFDeEMsTUFBTSxNQUFNLEdBQUcsU0FBUyxDQUFDLElBQUksRUFBRSxJQUFJLEVBQUUsQ0FBQyxDQUFDLENBQUMsQ0FBQztvQkFDekMsV0FBVyxDQUFDLENBQUMsQ0FBQyxHQUFHLFNBQVMsQ0FBQyxJQUFJLEVBQUUsTUFBTSxFQUFFLFVBQVUsQ0FBQyxDQUFDO29CQUNyRCxXQUFXLENBQUMsQ0FBQyxDQUFDLEdBQUcsU0FBUyxDQUFDLElBQUksRUFBRSxNQUFNLEVBQUUsQ0FBQyxVQUFVLENBQUMsQ0FBQztpQkFDdkQ7YUFDRjtZQUNELGlFQUFpRTtZQUNqRSxPQUFPLElBQUksSUFBSSxDQUFDLFFBQVEsQ0FBQztZQUN6QixJQUFJLE9BQU8sSUFBSSxFQUFFLEVBQUU7Z0JBQ2pCLE1BQU07YUFDUDtTQUNGO1FBQ0QsT0FBTyxXQUFXLENBQUM7SUFDckIsQ0FBQzs7QUF6R2dCLGlCQUFTLEdBQUcsQ0FBQyxDQUFDO0FBNEdqQzs7Ozs7O0dBTUc7QUFDRixNQUFNLE9BQU8sUUFBUyxTQUFRLE9BQU87SUFDcEM7Ozs7O09BS0c7SUFDSSxLQUFLLENBQUMsT0FBZ0I7UUFDM0IsTUFBTSxNQUFNLEdBQUcsT0FBTyxDQUFDLE1BQU0sQ0FBQztRQUM5QixNQUFNLFNBQVMsR0FBRyxFQUFFLENBQUM7UUFDckIsNkRBQTZEO1FBQzdELE1BQU0sV0FBVyxHQUFHLGdCQUFnQixDQUFDLE1BQU0sRUFBRSxRQUFRLENBQUMsU0FBUyxFQUFFLFNBQVMsQ0FBQyxDQUFDO1FBQzVFLElBQUksTUFBTSxHQUFHLElBQUksQ0FBQyxNQUFNLENBQUM7UUFFekIsSUFBSSxDQUFDLFlBQVksQ0FBQyxPQUFPLENBQUMsQ0FBQztRQUUzQixLQUFLLElBQUksS0FBSyxHQUFHLENBQUMsRUFBRSxLQUFLLEdBQUcsSUFBSSxDQUFDLE1BQU0sRUFBRSxFQUFFLEtBQUssRUFBRTtZQUNoRCx3Q0FBd0M7WUFDeEMsTUFBTSxDQUFDLEdBQVcsU0FBUyxDQUFDLE1BQU0sQ0FBQyxDQUFDO1lBQ3BDLE1BQU0sSUFBSSxHQUFHLFdBQVcsQ0FBQyxDQUFDLENBQUMsQ0FBQztZQUU1Qiw2QkFBNkI7WUFDN0IsS0FBSyxJQUFJLENBQUMsR0FBRyxDQUFDLEVBQUUsQ0FBQyxHQUFHLE1BQU0sRUFBRSxFQUFFLENBQUMsRUFBRTtnQkFDL0IsSUFBSSxDQUFDLElBQUksQ0FBQztvQkFBRSxTQUFTO2dCQUNyQixNQUFNLElBQUksR0FBRyxXQUFXLENBQUMsQ0FBQyxDQUFDLENBQUM7Z0JBQzVCLDRFQUE0RTtnQkFDNUUsTUFBTSxDQUFDLEdBQUcsSUFBSSxDQUFDLFlBQVksQ0FBQyxPQUFPLEVBQUUsQ0FBQyxFQUFFLENBQUMsQ0FBQyxDQUFDO2dCQUMzQywwRUFBMEU7Z0JBQzFFLE1BQU0sQ0FBQyxHQUFHLDBCQUEwQixDQUFDLElBQUksRUFBRSxJQUFJLENBQUMsQ0FBQztnQkFDakQsaURBQWlEO2dCQUNqRCxJQUFJLENBQUMsSUFBSSxDQUFDLE1BQU0sSUFBSSxDQUFDLENBQUMsSUFBSSxDQUFDLENBQUMsSUFBSSxJQUFJLENBQUMsTUFBTSxDQUFDLElBQUksQ0FBQyxDQUFDLEdBQUcsQ0FBQyxDQUFDLEVBQUU7b0JBQ3ZELE1BQU0sVUFBVSxHQUFHLE1BQU0sR0FBQyxDQUFDLENBQUMsR0FBQyxDQUFDLENBQUMsR0FBQyxDQUFDLENBQUMsR0FBQyxJQUFJLENBQUMsT0FBTyxDQUFDLENBQUM7b0JBQ2pELE1BQU0sTUFBTSxHQUFHLFNBQVMsQ0FBQyxJQUFJLEVBQUUsSUFBSSxFQUFFLENBQUMsQ0FBQyxDQUFDLENBQUM7b0JBQ3pDLGlDQUFpQztvQkFDakMsV0FBVyxDQUFDLENBQUMsQ0FBQyxHQUFHLFNBQVMsQ0FBQyxJQUFJLEVBQUUsTUFBTSxFQUFFLENBQUMsVUFBVSxDQUFDLENBQUM7aUJBQ3ZEO2FBQ0Y7WUFDRCxpRUFBaUU7WUFDakUsTUFBTSxJQUFJLElBQUksQ0FBQyxPQUFPLENBQUM7WUFDdkIsSUFBSSxNQUFNLElBQUksRUFBRSxFQUFFO2dCQUNoQixNQUFNO2FBQ1A7U0FDRjtRQUNELE9BQU8sV0FBVyxDQUFDO0lBQ3JCLENBQUM7Q0FDRiIsInNvdXJjZXNDb250ZW50IjpbImltcG9ydCB7T3B0aW9ucywgQ29vcmRpbmF0ZXMsIFZlY3RvcnMsIERpc3RhbmNlTWV0cmljfSBmcm9tICcuL3R5cGVfZGVjbGFyYXRpb25zJztcclxuaW1wb3J0IHtjYWxjdWxhdGVFdWNsaWRlYW5EaXN0YW5jZSwgY2FsY0Rpc3RhbmNlTWF0cml4LCBmaWxsUmFuZG9tTWF0cml4LCB2ZWN0b3JBZGQsIHJhbmRvbUludH0gZnJvbSAnLi9vcGVyYXRpb25zJztcclxuXHJcbi8qKlxyXG4gKiBJbXBsZW1lbnRzIHN0b2NoYXN0aWMgcHJveGltaXR5IGVtYmVkZGluZy5cclxuICpcclxuICogQGV4cG9ydFxyXG4gKiBAY2xhc3MgU1BFQmFzZVxyXG4gKiBAbGluayBkb2k6MTAuMTAxNi9TMTA5My0zMjYzKDAzKTAwMTU1LTRcclxuICovXHJcbmV4cG9ydCBjbGFzcyBTUEVCYXNlIHtcclxuICBwcm90ZWN0ZWQgc3RhdGljIGRpbWVuc2lvbiA9IDI7XHJcbiAgcHJvdGVjdGVkIHN0ZXBzOiBudW1iZXI7XHJcbiAgcHJvdGVjdGVkIGN5Y2xlczogbnVtYmVyO1xyXG4gIHByb3RlY3RlZCBjdXRvZmY6IG51bWJlcjtcclxuICBwcm90ZWN0ZWQgbGFtYmRhOiBudW1iZXI7XHJcbiAgcHJvdGVjdGVkIGRsYW1iZGE6IG51bWJlcjtcclxuICBwcm90ZWN0ZWQgbGFtYmRhMjogbnVtYmVyO1xyXG4gIHByb3RlY3RlZCBkbGFtYmRhMjogbnVtYmVyO1xyXG4gIHByb3RlY3RlZCBlcHNpbG9uOiBudW1iZXI7XHJcbiAgcHJvdGVjdGVkIGRpc3RhbmNlRnVuY3Rpb246IERpc3RhbmNlTWV0cmljO1xyXG4gIHByb3RlY3RlZCBkaXN0YW5jZTogQ29vcmRpbmF0ZXM7XHJcblxyXG4gIC8qKlxyXG4gICAqIENyZWF0ZXMgYW4gaW5zdGFuY2Ugb2YgU1BFQmFzZS5cclxuICAgKiBAcGFyYW0ge09wdGlvbnN9IFtvcHRpb25zXSBPcHRpb25zIHRvIHBhc3MgdG8gdGhlIGNvbnN0cnVjdG9yLlxyXG4gICAqIEBtZW1iZXJvZiBTUEVCYXNlXHJcbiAgICovXHJcbiAgY29uc3RydWN0b3Iob3B0aW9ucz86IE9wdGlvbnMpIHtcclxuICAgIHRoaXMuc3RlcHMgPSBvcHRpb25zPy5zdGVwcyA/PyAwO1xyXG4gICAgdGhpcy5jeWNsZXMgPSBvcHRpb25zPy5jeWNsZXMgPz8gMWU2O1xyXG4gICAgLy8gU2VsZWN0IGEgY3V0b2ZmIGRpc3RhbmNlIHtjdXRvZmZ9IGFuZCAuLi5cclxuICAgIHRoaXMuY3V0b2ZmID0gb3B0aW9ucz8uY3V0b2ZmID8/IDA7XHJcbiAgICAvLyAuLi4gYW4gaW5pdGlhbCBsZWFybmluZyByYXRlIHtsYW1iZGF9ID4gMFxyXG4gICAgdGhpcy5sYW1iZGEgPSBvcHRpb25zPy5sYW1iZGEgPz8gMi4wO1xyXG4gICAgdGhpcy5kbGFtYmRhID0gb3B0aW9ucz8uZGxhbWJkYSA/PyAwLjAxO1xyXG4gICAgdGhpcy5sYW1iZGEyID0gdGhpcy5sYW1iZGEvMi47XHJcbiAgICB0aGlzLmRsYW1iZGEyID0gdGhpcy5kbGFtYmRhLzIuO1xyXG4gICAgdGhpcy5lcHNpbG9uID0gb3B0aW9ucz8uZXBzaWxvbiA/PyAxZS0xMDtcclxuICAgIC8vIGVzbGludC1kaXNhYmxlLW5leHQtbGluZSBicmFjZS1zdHlsZVxyXG4gICAgdGhpcy5kaXN0YW5jZUZ1bmN0aW9uID0gb3B0aW9ucz8uZGlzdGFuY2UgPz8gY2FsY3VsYXRlRXVjbGlkZWFuRGlzdGFuY2U7XHJcbiAgICB0aGlzLmRpc3RhbmNlID0gW107XHJcbiAgfVxyXG5cclxuICAvKipcclxuICAgKiBJbml0aWFsaXplcyBkaXN0YW5jZSBtYXRyaXguXHJcbiAgICpcclxuICAgKiBAcHJvdGVjdGVkXHJcbiAgICogQHBhcmFtIHtWZWN0b3JzfSB2ZWN0b3JzIElucHV0IHZlY3RvcnMgdG8gY2FsY3VsYXRlIGRpc3RhbmNlIGJldHdlZW4uXHJcbiAgICogQG1lbWJlcm9mIFNQRUJhc2VcclxuICAgKi9cclxuICBwcm90ZWN0ZWQgaW5pdERpc3RhbmNlKHZlY3RvcnM6IFZlY3RvcnMpIHtcclxuICAgIHRoaXMuZGlzdGFuY2UgPSBjYWxjRGlzdGFuY2VNYXRyaXgodmVjdG9ycywgdGhpcy5kaXN0YW5jZUZ1bmN0aW9uKTtcclxuICB9XHJcblxyXG4gIC8qKlxyXG4gICAqIENhbGN1bGF0ZXMgZGlzdGFuY2UgYmV0d2VlbiB0aGUgdHdvIHZlY3RvcnMgZ2l2ZW4uXHJcbiAgICogXHJcbiAgICogQHBhcmFtIHtWZWN0b3JzfSB2ZWN0b3JzIFNldCBvZiB2ZWN0b3JzIHRvIGNhbGN1bGF0ZSBkaXN0YW5jZXMgYmV0d2Vlbi5cclxuICAgKiBAcGFyYW0ge251bWJlcn0gaW5kZXgxIEluZGV4IG9mIHRoZSBmaXJzdCB2ZWN0b3Igb2YgdGhlIHBhaXIuXHJcbiAgICogQHBhcmFtIHtudW1iZXJ9IGluZGV4MiBJbmRleCBvZiB0aGUgc2Vjb25kIHZlY3RvciBvZiB0aGUgcGFpci5cclxuICAgKiBAcmV0dXJuIHtudW1iZXJ9IERpc3RhbmNlIGJldHdlZW4gdGhlc2UgdHdvIHZlY3RvcnMuXHJcbiAgICovXHJcbiAgcHJvdGVjdGVkIGNhbGNEaXN0YW5jZSh2ZWN0b3JzOiBWZWN0b3JzLCBpbmRleDE6IG51bWJlciwgaW5kZXgyOiBudW1iZXIpOiBudW1iZXIge1xyXG4gICAgcmV0dXJuIHRoaXMuZGlzdGFuY2VbaW5kZXgxXVtpbmRleDJdO1xyXG4gIH1cclxuXHJcbiAgLyoqXHJcbiAgICogRW1iZWRzIHRoZSB2ZWN0b3JzIGdpdmVuIGludG8gYSB0d28tZGltZW5zaW9uYWwgc3BhY2UuXHJcbiAgICogXHJcbiAgICogQHBhcmFtIHtWZWN0b3JzfSB2ZWN0b3JzIEQtZGltZW5zaW9uYWwgY29vcmRpbmF0ZXMuXHJcbiAgICogQHJldHVybiB7Q29vcmRpbmF0ZXN9IFNQRSBjb29yZGluYXRlcyBpbiBEIHNwYWNlLlxyXG4gICAqL1xyXG4gIHB1YmxpYyBlbWJlZCh2ZWN0b3JzOiBWZWN0b3JzKTogQ29vcmRpbmF0ZXMge1xyXG4gICAgY29uc3Qgbkl0ZW1zID0gdmVjdG9ycy5sZW5ndGg7XHJcbiAgICBjb25zdCBhcmVhV2lkdGggPSA0MDtcclxuICAgIC8vIEluaXRpYWxpemUgdGhlIEQtZGltZW5zaW9uYWwgY29vcmRpbmF0ZXMgb2YgdGhlIE4gcG9pbnRzLlxyXG4gICAgY29uc3QgY29vcmRpbmF0ZXMgPSBmaWxsUmFuZG9tTWF0cml4KG5JdGVtcywgU1BFQmFzZS5kaW1lbnNpb24sIGFyZWFXaWR0aCk7XHJcblxyXG4gICAgbGV0IGxhbWJkYTIgPSB0aGlzLmxhbWJkYTI7XHJcblxyXG4gICAgaWYgKHRoaXMuc3RlcHMgPT0gMCkge1xyXG4gICAgICB0aGlzLnN0ZXBzID0gdmVjdG9ycy5sZW5ndGgtMTtcclxuICAgIH1cclxuXHJcbiAgICB0aGlzLmluaXREaXN0YW5jZSh2ZWN0b3JzKTtcclxuXHJcbiAgICBmb3IgKGxldCBjeWNsZSA9IDA7IGN5Y2xlIDwgdGhpcy5jeWNsZXM7ICsrY3ljbGUpIHtcclxuICAgICAgZm9yIChsZXQgc3RlcCA9IDA7IHN0ZXAgPCB0aGlzLnN0ZXBzOyArK3N0ZXApIHtcclxuICAgICAgICAvLyBTZWxlY3QgdHdvIHBvaW50cywgaSBhbmQgaiwgYXQgcmFuZG9tLCAuLi5cclxuICAgICAgICBjb25zdCBpID0gcmFuZG9tSW50KG5JdGVtcyk7IGxldCBqID0gcmFuZG9tSW50KG5JdGVtcyk7XHJcbiAgICAgICAgd2hpbGUgKGkgPT0gaikgaiA9IHJhbmRvbUludChuSXRlbXMpO1xyXG5cclxuICAgICAgICBjb25zdCByb3dpID0gY29vcmRpbmF0ZXNbaV07IGNvbnN0IHJvd2ogPSBjb29yZGluYXRlc1tqXTtcclxuXHJcbiAgICAgICAgLy8gLi4uIHJldHJpZXZlIChvciBldmFsdWF0ZSkgdGhlaXIgcHJveGltaXR5IGluIHRoZSBpbnB1dCBzcGFjZSwgcmlqIGFuZCAuLi5cclxuICAgICAgICBjb25zdCByID0gdGhpcy5jYWxjRGlzdGFuY2UodmVjdG9ycywgaSwgaik7XHJcbiAgICAgICAgLy8gLi4uIGNvbXB1dGUgdGhlaXIgRXVjbGlkZWFuIGRpc3RhbmNlIG9uIHRoZSBELWRpbWVuc2lvbmFsIG1hcCwgZGlqLlxyXG4gICAgICAgIGNvbnN0IGQgPSBjYWxjdWxhdGVFdWNsaWRlYW5EaXN0YW5jZShyb3dpLCByb3dqKTtcclxuXHJcbiAgICAgICAgLy8gSWYgcmlqIDw9IHJjLCBvciBpZiByaWogPiByYyBhbmQgZGlqIDwgcmlqIC4uLlxyXG4gICAgICAgIGlmICgodGhpcy5jdXRvZmYgPT0gMCkgfHwgKHIgPD0gdGhpcy5jdXRvZmYpIHx8IChkIDwgcikpIHtcclxuICAgICAgICAgIGNvbnN0IG11bHRpcGxpZXIgPSBsYW1iZGEyKihyLWQpLyhkK3RoaXMuZXBzaWxvbik7XHJcbiAgICAgICAgICAvLyAuLi4gdXBkYXRlIHRoZSBjb29yZGluYXRlcyB4aSBhbmQgeGouXHJcbiAgICAgICAgICBjb25zdCBkaWZmSUogPSB2ZWN0b3JBZGQocm93aSwgcm93aiwgLTEpO1xyXG4gICAgICAgICAgY29vcmRpbmF0ZXNbaV0gPSB2ZWN0b3JBZGQocm93aSwgZGlmZklKLCBtdWx0aXBsaWVyKTtcclxuICAgICAgICAgIGNvb3JkaW5hdGVzW2pdID0gdmVjdG9yQWRkKHJvd2osIGRpZmZJSiwgLW11bHRpcGxpZXIpO1xyXG4gICAgICAgIH1cclxuICAgICAgfVxyXG4gICAgICAvLyBEZWNyZWFzZSB0aGUgbGVhcm5pbmcgcmF0ZSB7bGFtYmRhfSBieSBhIHByZXNjcmliZWQge2RsYW1iZGF9LlxyXG4gICAgICBsYW1iZGEyIC09IHRoaXMuZGxhbWJkYTI7XHJcbiAgICAgIGlmIChsYW1iZGEyIDw9IDAuKSB7XHJcbiAgICAgICAgYnJlYWs7XHJcbiAgICAgIH1cclxuICAgIH1cclxuICAgIHJldHVybiBjb29yZGluYXRlcztcclxuICB9XHJcbn1cclxuXHJcbi8qKlxyXG4gKiBJbXBsZW1lbnRzIG1vZGlmaWVkIHN0b2NoYXN0aWMgcHJveGltaXR5IGVtYmVkZGluZy5cclxuICpcclxuICogQGV4cG9ydFxyXG4gKiBAY2xhc3MgUFNQRUJhc2VcclxuICogQGxpbmsgZG9pOjEwLjEwMTYvUzEwOTMtMzI2MygwMykwMDE1NS00XHJcbiAqL1xyXG4gZXhwb3J0IGNsYXNzIFBTUEVCYXNlIGV4dGVuZHMgU1BFQmFzZSB7XHJcbiAgLyoqXHJcbiAgICogRW1iZWRzIHRoZSB2ZWN0b3JzIGdpdmVuIGludG8gYSB0d28tZGltZW5zaW9uYWwgc3BhY2UgdXNpbmcgYSBtb2RpZmllZCB1cGRhdGUgcnVsZS5cclxuICAgKiBcclxuICAgKiBAcGFyYW0ge1ZlY3RvcnN9IHZlY3RvcnMgRC1kaW1lbnNpb25hbCBjb29yZGluYXRlcy5cclxuICAgKiBAcmV0dXJuIHtDb29yZGluYXRlc30gU1BFIGNvb3JkaW5hdGVzIGluIEQgc3BhY2UuXHJcbiAgICovXHJcbiAgcHVibGljIGVtYmVkKHZlY3RvcnM6IFZlY3RvcnMpOiBDb29yZGluYXRlcyB7XHJcbiAgICBjb25zdCBuSXRlbXMgPSB2ZWN0b3JzLmxlbmd0aDtcclxuICAgIGNvbnN0IGFyZWFXaWR0aCA9IDQwO1xyXG4gICAgLy8gIEluaXRpYWxpemUgdGhlIEQtZGltZW5zaW9uYWwgY29vcmRpbmF0ZXMgb2YgdGhlIE4gcG9pbnRzLlxyXG4gICAgY29uc3QgY29vcmRpbmF0ZXMgPSBmaWxsUmFuZG9tTWF0cml4KG5JdGVtcywgUFNQRUJhc2UuZGltZW5zaW9uLCBhcmVhV2lkdGgpO1xyXG4gICAgbGV0IGxhbWJkYSA9IHRoaXMubGFtYmRhO1xyXG5cclxuICAgIHRoaXMuaW5pdERpc3RhbmNlKHZlY3RvcnMpO1xyXG5cclxuICAgIGZvciAobGV0IGN5Y2xlID0gMDsgY3ljbGUgPCB0aGlzLmN5Y2xlczsgKytjeWNsZSkge1xyXG4gICAgICAvLyBTZWxlY3QgYSBwb2ludCwgaSwgYXQgcmFuZG9tIChwaXZvdCkuXHJcbiAgICAgIGNvbnN0IGk6IG51bWJlciA9IHJhbmRvbUludChuSXRlbXMpO1xyXG4gICAgICBjb25zdCByb3dpID0gY29vcmRpbmF0ZXNbaV07XHJcblxyXG4gICAgICAvLyBGb3IgZXZlcnkgcG9pbnQgaiAhPSBpIC4uLlxyXG4gICAgICBmb3IgKGxldCBqID0gMDsgaiA8IG5JdGVtczsgKytqKSB7XHJcbiAgICAgICAgaWYgKGkgPT0gaikgY29udGludWU7XHJcbiAgICAgICAgY29uc3Qgcm93aiA9IGNvb3JkaW5hdGVzW2pdO1xyXG4gICAgICAgIC8vIC4uLiByZXRyaWV2ZSAob3IgZXZhbHVhdGUpIGl0cyBwcm94aW1pdHkgdG8gaSBpbiB0aGUgaW5wdXQgc3BhY2UsIHJpaiAuLi5cclxuICAgICAgICBjb25zdCByID0gdGhpcy5jYWxjRGlzdGFuY2UodmVjdG9ycywgaSwgaik7XHJcbiAgICAgICAgLy8gLi4uIGFuZCBjb21wdXRlIHRoZWlyIEV1Y2xpZGVhbiBkaXN0YW5jZSBvbiB0aGUgRC1kaW1lbnNpb25hbCBtYXAsIGRpai5cclxuICAgICAgICBjb25zdCBkID0gY2FsY3VsYXRlRXVjbGlkZWFuRGlzdGFuY2Uocm93aSwgcm93aik7XHJcbiAgICAgICAgLy8gSWYgcmlqIDw9IHJjLCBvciBpZiByaWogPiByYyBhbmQgZGlqIDwgcmlqIC4uLlxyXG4gICAgICAgIGlmICgodGhpcy5jdXRvZmYgPT0gMCkgfHwgKHIgPD0gdGhpcy5jdXRvZmYpIHx8IChkIDwgcikpIHtcclxuICAgICAgICAgIGNvbnN0IG11bHRpcGxpZXIgPSBsYW1iZGEqKHItZCkvKGQrdGhpcy5lcHNpbG9uKTtcclxuICAgICAgICAgIGNvbnN0IGRpZmZJSiA9IHZlY3RvckFkZChyb3dpLCByb3dqLCAtMSk7XHJcbiAgICAgICAgICAvLyAuLi4gdXBkYXRlIHRoZSBjb29yZGluYXRlcyB4ai5cclxuICAgICAgICAgIGNvb3JkaW5hdGVzW2pdID0gdmVjdG9yQWRkKHJvd2osIGRpZmZJSiwgLW11bHRpcGxpZXIpO1xyXG4gICAgICAgIH1cclxuICAgICAgfVxyXG4gICAgICAvLyBEZWNyZWFzZSB0aGUgbGVhcm5pbmcgcmF0ZSB7bGFtYmRhfSBieSBhIHByZXNjcmliZWQge2RsYW1iZGF9LlxyXG4gICAgICBsYW1iZGEgLT0gdGhpcy5kbGFtYmRhO1xyXG4gICAgICBpZiAobGFtYmRhIDw9IDAuKSB7XHJcbiAgICAgICAgYnJlYWs7XHJcbiAgICAgIH1cclxuICAgIH1cclxuICAgIHJldHVybiBjb29yZGluYXRlcztcclxuICB9XHJcbn1cclxuIl19

package/src/spe.ts

@@ -0,0 +1,172 @@
+import {Options, Coordinates, Vectors, DistanceMetric} from './type_declarations';
+import {calculateEuclideanDistance, calcDistanceMatrix, fillRandomMatrix, vectorAdd, randomInt} from './operations';
+
+/**
+ * Implements stochastic proximity embedding.
+ *
+ * @export
+ * @class SPEBase
+ * @link doi:10.1016/S1093-3263(03)00155-4
+ */
+export class SPEBase {
+  protected static dimension = 2;
+  protected steps: number;
+  protected cycles: number;
+  protected cutoff: number;
+  protected lambda: number;
+  protected dlambda: number;
+  protected lambda2: number;
+  protected dlambda2: number;
+  protected epsilon: number;
+  protected distanceFunction: DistanceMetric;
+  protected distance: Coordinates;
+
+  /**
+   * Creates an instance of SPEBase.
+   * @param {Options} [options] Options to pass to the constructor.
+   * @memberof SPEBase
+   */
+  constructor(options?: Options) {
+    this.steps = options?.steps ?? 0;
+    this.cycles = options?.cycles ?? 1e6;
+    // Select a cutoff distance {cutoff} and ...
+    this.cutoff = options?.cutoff ?? 0;
+    // ... an initial learning rate {lambda} > 0
+    this.lambda = options?.lambda ?? 2.0;
+    this.dlambda = options?.dlambda ?? 0.01;
+    this.lambda2 = this.lambda/2.;
+    this.dlambda2 = this.dlambda/2.;
+    this.epsilon = options?.epsilon ?? 1e-10;
+    // eslint-disable-next-line brace-style
+    this.distanceFunction = options?.distance ?? calculateEuclideanDistance;
+    this.distance = [];
+  }
+
+  /**
+   * Initializes distance matrix.
+   *
+   * @protected
+   * @param {Vectors} vectors Input vectors to calculate distance between.
+   * @memberof SPEBase
+   */
+  protected initDistance(vectors: Vectors) {
+    this.distance = calcDistanceMatrix(vectors, this.distanceFunction);
+  }
+
+  /**
+   * Calculates distance between the two vectors given.
+   * 
+   * @param {Vectors} vectors Set of vectors to calculate distances between.
+   * @param {number} index1 Index of the first vector of the pair.
+   * @param {number} index2 Index of the second vector of the pair.
+   * @return {number} Distance between these two vectors.
+   */
+  protected calcDistance(vectors: Vectors, index1: number, index2: number): number {
+    return this.distance[index1][index2];
+  }
+
+  /**
+   * Embeds the vectors given into a two-dimensional space.
+   * 
+   * @param {Vectors} vectors D-dimensional coordinates.
+   * @return {Coordinates} SPE coordinates in D space.
+   */
+  public embed(vectors: Vectors): Coordinates {
+    const nItems = vectors.length;
+    const areaWidth = 40;
+    // Initialize the D-dimensional coordinates of the N points.
+    const coordinates = fillRandomMatrix(nItems, SPEBase.dimension, areaWidth);
+
+    let lambda2 = this.lambda2;
+
+    if (this.steps == 0) {
+      this.steps = vectors.length-1;
+    }
+
+    this.initDistance(vectors);
+
+    for (let cycle = 0; cycle < this.cycles; ++cycle) {
+      for (let step = 0; step < this.steps; ++step) {
+        // Select two points, i and j, at random, ...
+        const i = randomInt(nItems); let j = randomInt(nItems);
+        while (i == j) j = randomInt(nItems);
+
+        const rowi = coordinates[i]; const rowj = coordinates[j];
+
+        // ... retrieve (or evaluate) their proximity in the input space, rij and ...
+        const r = this.calcDistance(vectors, i, j);
+        // ... compute their Euclidean distance on the D-dimensional map, dij.
+        const d = calculateEuclideanDistance(rowi, rowj);
+
+        // If rij <= rc, or if rij > rc and dij < rij ...
+        if ((this.cutoff == 0) || (r <= this.cutoff) || (d < r)) {
+          const multiplier = lambda2*(r-d)/(d+this.epsilon);
+          // ... update the coordinates xi and xj.
+          const diffIJ = vectorAdd(rowi, rowj, -1);
+          coordinates[i] = vectorAdd(rowi, diffIJ, multiplier);
+          coordinates[j] = vectorAdd(rowj, diffIJ, -multiplier);
+        }
+      }
+      // Decrease the learning rate {lambda} by a prescribed {dlambda}.
+      lambda2 -= this.dlambda2;
+      if (lambda2 <= 0.) {
+        break;
+      }
+    }
+    return coordinates;
+  }
+}
+
+/**
+ * Implements modified stochastic proximity embedding.
+ *
+ * @export
+ * @class PSPEBase
+ * @link doi:10.1016/S1093-3263(03)00155-4
+ */
+ export class PSPEBase extends SPEBase {
+  /**
+   * Embeds the vectors given into a two-dimensional space using a modified update rule.
+   * 
+   * @param {Vectors} vectors D-dimensional coordinates.
+   * @return {Coordinates} SPE coordinates in D space.
+   */
+  public embed(vectors: Vectors): Coordinates {
+    const nItems = vectors.length;
+    const areaWidth = 40;
+    //  Initialize the D-dimensional coordinates of the N points.
+    const coordinates = fillRandomMatrix(nItems, PSPEBase.dimension, areaWidth);
+    let lambda = this.lambda;
+
+    this.initDistance(vectors);
+
+    for (let cycle = 0; cycle < this.cycles; ++cycle) {
+      // Select a point, i, at random (pivot).
+      const i: number = randomInt(nItems);
+      const rowi = coordinates[i];
+
+      // For every point j != i ...
+      for (let j = 0; j < nItems; ++j) {
+        if (i == j) continue;
+        const rowj = coordinates[j];
+        // ... retrieve (or evaluate) its proximity to i in the input space, rij ...
+        const r = this.calcDistance(vectors, i, j);
+        // ... and compute their Euclidean distance on the D-dimensional map, dij.
+        const d = calculateEuclideanDistance(rowi, rowj);
+        // If rij <= rc, or if rij > rc and dij < rij ...
+        if ((this.cutoff == 0) || (r <= this.cutoff) || (d < r)) {
+          const multiplier = lambda*(r-d)/(d+this.epsilon);
+          const diffIJ = vectorAdd(rowi, rowj, -1);
+          // ... update the coordinates xj.
+          coordinates[j] = vectorAdd(rowj, diffIJ, -multiplier);
+        }
+      }
+      // Decrease the learning rate {lambda} by a prescribed {dlambda}.
+      lambda -= this.dlambda;
+      if (lambda <= 0.) {
+        break;
+      }
+    }
+    return coordinates;
+  }
+}

package/src/string-measure.js

@@ -0,0 +1,46 @@
+import * as fl from 'fastest-levenshtein';
+import { jaroWinkler } from 'jaro-winkler-typescript';
+import { assert } from './operations';
+/**
+ * Unified class implementing the different string measures.
+ *
+ * @export
+ * @class Measurer
+ */
+export class Measurer {
+    /**
+     * Creates an instance of Measurer.
+     * @param {string} method Method to calculate distance between strings.
+     * @memberof Measurer
+     */
+    constructor(method) {
+        assert(Measurer.availableMeasures.includes(method), 'The ${method} was not found.');
+        this.method = method;
+    }
+    /**
+     * Returns custom string distance function specified.
+     * @return {DistanceMetric} Callback of the measure chosen.
+     */
+    getMeasure() {
+        return Measurer.receipt[this.method];
+    }
+    /**
+     * Returns available string distance metrics.
+     *
+     * @readonly
+     * @type {string[]}
+     * @memberof Measurer
+     */
+    static get availableMeasures() {
+        let keys = [];
+        for (let key in Measurer.receipt) {
+            keys.push(key);
+        }
+        return keys;
+    }
+}
+Measurer.receipt = {
+    'Levenshtein': fl.distance,
+    'Jaro-Winkler': jaroWinkler,
+};
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoic3RyaW5nLW1lYXN1cmUuanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyJzdHJpbmctbWVhc3VyZS50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiQUFBQSxPQUFPLEtBQUssRUFBRSxNQUFNLHFCQUFxQixDQUFDO0FBQzFDLE9BQU8sRUFBQyxXQUFXLEVBQUMsTUFBTSx5QkFBeUIsQ0FBQztBQUdwRCxPQUFPLEVBQUMsTUFBTSxFQUFDLE1BQU0sY0FBYyxDQUFDO0FBRXBDOzs7OztHQUtHO0FBQ0gsTUFBTSxPQUFPLFFBQVE7SUFPakI7Ozs7T0FJRztJQUNILFlBQVksTUFBYztRQUN4QixNQUFNLENBQUMsUUFBUSxDQUFDLGlCQUFpQixDQUFDLFFBQVEsQ0FBQyxNQUFNLENBQUMsRUFBRSw4QkFBOEIsQ0FBQyxDQUFBO1FBQ25GLElBQUksQ0FBQyxNQUFNLEdBQUcsTUFBTSxDQUFDO0lBQ3ZCLENBQUM7SUFFRDs7O09BR0c7SUFDSSxVQUFVO1FBQ2YsT0FBTyxRQUFRLENBQUMsT0FBTyxDQUFDLElBQUksQ0FBQyxNQUFNLENBQUMsQ0FBQztJQUN2QyxDQUFDO0lBRUQ7Ozs7OztPQU1HO0lBQ0ksTUFBTSxLQUFLLGlCQUFpQjtRQUNqQyxJQUFJLElBQUksR0FBRyxFQUFFLENBQUM7UUFFZCxLQUFLLElBQUksR0FBRyxJQUFJLFFBQVEsQ0FBQyxPQUFPLEVBQUU7WUFDaEMsSUFBSSxDQUFDLElBQUksQ0FBQyxHQUFHLENBQUMsQ0FBQztTQUNoQjtRQUNELE9BQU8sSUFBSSxDQUFDO0lBQ2QsQ0FBQzs7QUFyQ2EsZ0JBQU8sR0FBcUM7SUFDeEQsYUFBYSxFQUFFLEVBQUUsQ0FBQyxRQUFRO0lBQzFCLGNBQWMsRUFBRSxXQUFXO0NBQzVCLENBQUMiLCJzb3VyY2VzQ29udGVudCI6WyJpbXBvcnQgKiBhcyBmbCBmcm9tICdmYXN0ZXN0LWxldmVuc2h0ZWluJztcclxuaW1wb3J0IHtqYXJvV2lua2xlcn0gZnJvbSAnamFyby13aW5rbGVyLXR5cGVzY3JpcHQnO1xyXG5cclxuaW1wb3J0IHtEaXN0YW5jZU1ldHJpY30gZnJvbSAnLi90eXBlX2RlY2xhcmF0aW9ucyc7XHJcbmltcG9ydCB7YXNzZXJ0fSBmcm9tICcuL29wZXJhdGlvbnMnO1xyXG5cclxuLyoqXHJcbiAqIFVuaWZpZWQgY2xhc3MgaW1wbGVtZW50aW5nIHRoZSBkaWZmZXJlbnQgc3RyaW5nIG1lYXN1cmVzLlxyXG4gKlxyXG4gKiBAZXhwb3J0XHJcbiAqIEBjbGFzcyBNZWFzdXJlclxyXG4gKi9cclxuZXhwb3J0IGNsYXNzIE1lYXN1cmVyIHtcclxuICAgIHByb3RlY3RlZCBtZXRob2Q6IHN0cmluZztcclxuICAgIHB1YmxpYyBzdGF0aWMgcmVjZWlwdDoge1tuYW1lOiBzdHJpbmddOiBEaXN0YW5jZU1ldHJpY30gPSB7XHJcbiAgICAgICdMZXZlbnNodGVpbic6IGZsLmRpc3RhbmNlLFxyXG4gICAgICAnSmFyby1XaW5rbGVyJzogamFyb1dpbmtsZXIsXHJcbiAgICB9O1xyXG5cclxuICAgIC8qKlxyXG4gICAgICogQ3JlYXRlcyBhbiBpbnN0YW5jZSBvZiBNZWFzdXJlci5cclxuICAgICAqIEBwYXJhbSB7c3RyaW5nfSBtZXRob2QgTWV0aG9kIHRvIGNhbGN1bGF0ZSBkaXN0YW5jZSBiZXR3ZWVuIHN0cmluZ3MuXHJcbiAgICAgKiBAbWVtYmVyb2YgTWVhc3VyZXJcclxuICAgICAqL1xyXG4gICAgY29uc3RydWN0b3IobWV0aG9kOiBzdHJpbmcpIHtcclxuICAgICAgYXNzZXJ0KE1lYXN1cmVyLmF2YWlsYWJsZU1lYXN1cmVzLmluY2x1ZGVzKG1ldGhvZCksICdUaGUgJHttZXRob2R9IHdhcyBub3QgZm91bmQuJylcclxuICAgICAgdGhpcy5tZXRob2QgPSBtZXRob2Q7XHJcbiAgICB9XHJcblxyXG4gICAgLyoqXHJcbiAgICAgKiBSZXR1cm5zIGN1c3RvbSBzdHJpbmcgZGlzdGFuY2UgZnVuY3Rpb24gc3BlY2lmaWVkLlxyXG4gICAgICogQHJldHVybiB7RGlzdGFuY2VNZXRyaWN9IENhbGxiYWNrIG9mIHRoZSBtZWFzdXJlIGNob3Nlbi5cclxuICAgICAqL1xyXG4gICAgcHVibGljIGdldE1lYXN1cmUoKTogRGlzdGFuY2VNZXRyaWMge1xyXG4gICAgICByZXR1cm4gTWVhc3VyZXIucmVjZWlwdFt0aGlzLm1ldGhvZF07XHJcbiAgICB9XHJcblxyXG4gICAgLyoqXHJcbiAgICAgKiBSZXR1cm5zIGF2YWlsYWJsZSBzdHJpbmcgZGlzdGFuY2UgbWV0cmljcy5cclxuICAgICAqXHJcbiAgICAgKiBAcmVhZG9ubHlcclxuICAgICAqIEB0eXBlIHtzdHJpbmdbXX1cclxuICAgICAqIEBtZW1iZXJvZiBNZWFzdXJlclxyXG4gICAgICovXHJcbiAgICBwdWJsaWMgc3RhdGljIGdldCBhdmFpbGFibGVNZWFzdXJlcygpIDogc3RyaW5nW10ge1xyXG4gICAgICBsZXQga2V5cyA9IFtdO1xyXG5cclxuICAgICAgZm9yIChsZXQga2V5IGluIE1lYXN1cmVyLnJlY2VpcHQpIHtcclxuICAgICAgICBrZXlzLnB1c2goa2V5KTtcclxuICAgICAgfVxyXG4gICAgICByZXR1cm4ga2V5cztcclxuICAgIH1cclxuICAgIFxyXG59XHJcbiJdfQ==

package/src/string-measure.ts

@@ -0,0 +1,54 @@
+import * as fl from 'fastest-levenshtein';
+import {jaroWinkler} from 'jaro-winkler-typescript';
+
+import {DistanceMetric} from './type_declarations';
+import {assert} from './operations';
+
+/**
+ * Unified class implementing the different string measures.
+ *
+ * @export
+ * @class Measurer
+ */
+export class Measurer {
+    protected method: string;
+    public static receipt: {[name: string]: DistanceMetric} = {
+      'Levenshtein': fl.distance,
+      'Jaro-Winkler': jaroWinkler,
+    };
+
+    /**
+     * Creates an instance of Measurer.
+     * @param {string} method Method to calculate distance between strings.
+     * @memberof Measurer
+     */
+    constructor(method: string) {
+      assert(Measurer.availableMeasures.includes(method), 'The ${method} was not found.')
+      this.method = method;
+    }
+
+    /**
+     * Returns custom string distance function specified.
+     * @return {DistanceMetric} Callback of the measure chosen.
+     */
+    public getMeasure(): DistanceMetric {
+      return Measurer.receipt[this.method];
+    }
+
+    /**
+     * Returns available string distance metrics.
+     *
+     * @readonly
+     * @type {string[]}
+     * @memberof Measurer
+     */
+    public static get availableMeasures() : string[] {
+      let keys = [];
+
+      for (let key in Measurer.receipt) {
+        keys.push(key);
+      }
+      return keys;
+    }
+    
+}

package/src/string-utils.js

@@ -0,0 +1,9 @@
+export function tryParseJson(s) {
+    try {
+        return JSON.parse(s);
+    }
+    catch (_) {
+        return null;
+    }
+}
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoic3RyaW5nLXV0aWxzLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsic3RyaW5nLXV0aWxzLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUFBLE1BQU0sVUFBVSxZQUFZLENBQUMsQ0FBUztJQUNsQyxJQUFJO1FBQ0EsT0FBTyxJQUFJLENBQUMsS0FBSyxDQUFDLENBQUMsQ0FBQyxDQUFDO0tBQ3hCO0lBQ0QsT0FBTyxDQUFDLEVBQUU7UUFDTixPQUFPLElBQUksQ0FBQztLQUNmO0FBQ0wsQ0FBQyIsInNvdXJjZXNDb250ZW50IjpbImV4cG9ydCBmdW5jdGlvbiB0cnlQYXJzZUpzb24oczogc3RyaW5nKSA6IGFueSB7XHJcbiAgICB0cnkge1xyXG4gICAgICAgIHJldHVybiBKU09OLnBhcnNlKHMpO1xyXG4gICAgfVxyXG4gICAgY2F0Y2ggKF8pIHtcclxuICAgICAgICByZXR1cm4gbnVsbDtcclxuICAgIH1cclxufSJdfQ==

package/src/string-utils.ts

@@ -0,0 +1,8 @@
+export function tryParseJson(s: string) : any {
+    try {
+        return JSON.parse(s);
+    }
+    catch (_) {
+        return null;
+    }
+}

package/src/table-validation.js

@@ -0,0 +1,47 @@
+import * as DG from "datagrok-api/dg";
+// @ts-ignore
+let ruleSet = {
+    name: 'SDTM DM rules',
+    rules: [
+        {
+            column: 'age',
+            type: 'int',
+            pattern: '18-90'
+        },
+        {
+            column: 'height',
+            type: 'numerical',
+            pattern: '18-90'
+        },
+        {
+            column: 'sex',
+            type: 'string',
+            pattern: 'in (F, M)'
+        },
+        {
+            column: 'started',
+            type: 'datetime',
+            pattern: '> 1/1/1900'
+        }
+    ]
+};
+/** Returns error message, or null if valid. */
+function validateCell(cell, ruleset) {
+    var _a;
+    let name = cell.column.name.toLowerCase();
+    for (let rule of ruleset.rules)
+        if (rule.column == name) {
+            rule.matcher = (_a = rule.matcher) !== null && _a !== void 0 ? _a : DG.ValueMatcher.forColumn(cell.column, rule.pattern);
+            let error = rule.matcher.validate(cell.value);
+            if (error !== null)
+                return error;
+        }
+    return null;
+}
+function main() {
+    console.log('foo');
+}
+let cell = { column: { name: 'age' }, value: 999 };
+validateCell(cell, ruleSet);
+console.log('ddd');
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidGFibGUtdmFsaWRhdGlvbi5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbInRhYmxlLXZhbGlkYXRpb24udHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6IkFBR0EsT0FBTyxLQUFLLEVBQUUsTUFBTSxpQkFBaUIsQ0FBQztBQWN0QyxhQUFhO0FBQ2IsSUFBSSxPQUFPLEdBQVk7SUFDckIsSUFBSSxFQUFFLGVBQWU7SUFDckIsS0FBSyxFQUFFO1FBQ0w7WUFDRSxNQUFNLEVBQUUsS0FBSztZQUNiLElBQUksRUFBRSxLQUFLO1lBQ1gsT0FBTyxFQUFFLE9BQU87U0FDakI7UUFDRDtZQUNFLE1BQU0sRUFBRSxRQUFRO1lBQ2hCLElBQUksRUFBRSxXQUFXO1lBQ2pCLE9BQU8sRUFBRSxPQUFPO1NBQ2pCO1FBQ0Q7WUFDRSxNQUFNLEVBQUUsS0FBSztZQUNiLElBQUksRUFBRSxRQUFRO1lBQ2QsT0FBTyxFQUFFLFdBQVc7U0FDckI7UUFDRDtZQUNFLE1BQU0sRUFBRSxTQUFTO1lBQ2pCLElBQUksRUFBRSxVQUFVO1lBQ2hCLE9BQU8sRUFBRSxZQUFZO1NBQ3RCO0tBQ0Y7Q0FDRixDQUFDO0FBRUYsK0NBQStDO0FBQy9DLFNBQVMsWUFBWSxDQUFDLElBQVMsRUFBRSxPQUFnQjs7SUFDL0MsSUFBSSxJQUFJLEdBQUcsSUFBSSxDQUFDLE1BQU0sQ0FBQyxJQUFJLENBQUMsV0FBVyxFQUFFLENBQUM7SUFFMUMsS0FBSyxJQUFJLElBQUksSUFBSSxPQUFPLENBQUMsS0FBSztRQUM1QixJQUFJLElBQUksQ0FBQyxNQUFNLElBQUksSUFBSSxFQUFFO1lBQ3ZCLElBQUksQ0FBQyxPQUFPLEdBQUcsTUFBQSxJQUFJLENBQUMsT0FBTyxtQ0FBSSxFQUFFLENBQUMsWUFBWSxDQUFDLFNBQVMsQ0FBQyxJQUFJLENBQUMsTUFBTSxFQUFFLElBQUksQ0FBQyxPQUFPLENBQUMsQ0FBQztZQUNwRixJQUFJLEtBQUssR0FBRyxJQUFJLENBQUMsT0FBTyxDQUFDLFFBQVEsQ0FBQyxJQUFJLENBQUMsS0FBSyxDQUFDLENBQUM7WUFDOUMsSUFBSSxLQUFLLEtBQUssSUFBSTtnQkFDaEIsT0FBTyxLQUFLLENBQUM7U0FDaEI7SUFFSCxPQUFPLElBQUksQ0FBQztBQUNkLENBQUM7QUFFRCxTQUFTLElBQUk7SUFDWCxPQUFPLENBQUMsR0FBRyxDQUFDLEtBQUssQ0FBQyxDQUFDO0FBQ3JCLENBQUM7QUFFRCxJQUFJLElBQUksR0FBRyxFQUFDLE1BQU0sRUFBRSxFQUFFLElBQUksRUFBRSxLQUFLLEVBQUUsRUFBRSxLQUFLLEVBQUUsR0FBRyxFQUFFLENBQUM7QUFDbEQsWUFBWSxDQUFDLElBQUksRUFBRSxPQUFPLENBQUMsQ0FBQztBQUU1QixPQUFPLENBQUMsR0FBRyxDQUFDLEtBQUssQ0FBQyxDQUFDIiwic291cmNlc0NvbnRlbnQiOlsiLyogRG8gbm90IGNoYW5nZSB0aGVzZSBpbXBvcnQgbGluZXMuIERhdGFncm9rIHdpbGwgaW1wb3J0IEFQSSBsaWJyYXJ5IGluIGV4YWN0bHkgdGhlIHNhbWUgbWFubmVyICovXHJcbmltcG9ydCAqIGFzIGdyb2sgZnJvbSAnZGF0YWdyb2stYXBpL2dyb2snO1xyXG5pbXBvcnQgKiBhcyB1aSBmcm9tICdkYXRhZ3Jvay1hcGkvdWknO1xyXG5pbXBvcnQgKiBhcyBERyBmcm9tIFwiZGF0YWdyb2stYXBpL2RnXCI7XHJcblxyXG5pbnRlcmZhY2UgUnVsZSB7XHJcbiAgY29sdW1uOiBzdHJpbmc7XHJcbiAgdHlwZTogc3RyaW5nO1xyXG4gIHBhdHRlcm46IHN0cmluZztcclxuICBtYXRjaGVyPzogYW55O1xyXG59XHJcblxyXG5pbnRlcmZhY2UgUnVsZVNldCB7XHJcbiAgbmFtZTogc3RyaW5nO1xyXG4gIHJ1bGVzOiBSdWxlW107XHJcbn1cclxuXHJcbi8vIEB0cy1pZ25vcmVcclxubGV0IHJ1bGVTZXQ6IFJ1bGVTZXQgPSB7XHJcbiAgbmFtZTogJ1NEVE0gRE0gcnVsZXMnLFxyXG4gIHJ1bGVzOiBbXHJcbiAgICB7XHJcbiAgICAgIGNvbHVtbjogJ2FnZScsXHJcbiAgICAgIHR5cGU6ICdpbnQnLFxyXG4gICAgICBwYXR0ZXJuOiAnMTgtOTAnXHJcbiAgICB9LFxyXG4gICAge1xyXG4gICAgICBjb2x1bW46ICdoZWlnaHQnLFxyXG4gICAgICB0eXBlOiAnbnVtZXJpY2FsJyxcclxuICAgICAgcGF0dGVybjogJzE4LTkwJ1xyXG4gICAgfSxcclxuICAgIHtcclxuICAgICAgY29sdW1uOiAnc2V4JyxcclxuICAgICAgdHlwZTogJ3N0cmluZycsXHJcbiAgICAgIHBhdHRlcm46ICdpbiAoRiwgTSknXHJcbiAgICB9LFxyXG4gICAge1xyXG4gICAgICBjb2x1bW46ICdzdGFydGVkJyxcclxuICAgICAgdHlwZTogJ2RhdGV0aW1lJyxcclxuICAgICAgcGF0dGVybjogJz4gMS8xLzE5MDAnXHJcbiAgICB9XHJcbiAgXVxyXG59O1xyXG5cclxuLyoqIFJldHVybnMgZXJyb3IgbWVzc2FnZSwgb3IgbnVsbCBpZiB2YWxpZC4gKi9cclxuZnVuY3Rpb24gdmFsaWRhdGVDZWxsKGNlbGw6IGFueSwgcnVsZXNldDogUnVsZVNldCkgOiBzdHJpbmcgfCBudWxsIHtcclxuICBsZXQgbmFtZSA9IGNlbGwuY29sdW1uLm5hbWUudG9Mb3dlckNhc2UoKTtcclxuXHJcbiAgZm9yIChsZXQgcnVsZSBvZiBydWxlc2V0LnJ1bGVzKVxyXG4gICAgaWYgKHJ1bGUuY29sdW1uID09IG5hbWUpIHtcclxuICAgICAgcnVsZS5tYXRjaGVyID0gcnVsZS5tYXRjaGVyID8/IERHLlZhbHVlTWF0Y2hlci5mb3JDb2x1bW4oY2VsbC5jb2x1bW4sIHJ1bGUucGF0dGVybik7XHJcbiAgICAgIGxldCBlcnJvciA9IHJ1bGUubWF0Y2hlci52YWxpZGF0ZShjZWxsLnZhbHVlKTtcclxuICAgICAgaWYgKGVycm9yICE9PSBudWxsKVxyXG4gICAgICAgIHJldHVybiBlcnJvcjtcclxuICAgIH1cclxuXHJcbiAgcmV0dXJuIG51bGw7XHJcbn1cclxuXHJcbmZ1bmN0aW9uIG1haW4oKSA6IHZvaWQge1xyXG4gIGNvbnNvbGUubG9nKCdmb28nKTtcclxufVxyXG5cclxubGV0IGNlbGwgPSB7Y29sdW1uOiB7IG5hbWU6ICdhZ2UnIH0sIHZhbHVlOiA5OTkgfTtcclxudmFsaWRhdGVDZWxsKGNlbGwsIHJ1bGVTZXQpO1xyXG5cclxuY29uc29sZS5sb2coJ2RkZCcpOyJdfQ==

package/src/table-validation.ts

@@ -0,0 +1,67 @@
+/* Do not change these import lines. Datagrok will import API library in exactly the same manner */
+import * as grok from 'datagrok-api/grok';
+import * as ui from 'datagrok-api/ui';
+import * as DG from "datagrok-api/dg";
+
+interface Rule {
+  column: string;
+  type: string;
+  pattern: string;
+  matcher?: any;
+}
+
+interface RuleSet {
+  name: string;
+  rules: Rule[];
+}
+
+// @ts-ignore
+let ruleSet: RuleSet = {
+  name: 'SDTM DM rules',
+  rules: [
+    {
+      column: 'age',
+      type: 'int',
+      pattern: '18-90'
+    },
+    {
+      column: 'height',
+      type: 'numerical',
+      pattern: '18-90'
+    },
+    {
+      column: 'sex',
+      type: 'string',
+      pattern: 'in (F, M)'
+    },
+    {
+      column: 'started',
+      type: 'datetime',
+      pattern: '> 1/1/1900'
+    }
+  ]
+};
+
+/** Returns error message, or null if valid. */
+function validateCell(cell: any, ruleset: RuleSet) : string | null {
+  let name = cell.column.name.toLowerCase();
+
+  for (let rule of ruleset.rules)
+    if (rule.column == name) {
+      rule.matcher = rule.matcher ?? DG.ValueMatcher.forColumn(cell.column, rule.pattern);
+      let error = rule.matcher.validate(cell.value);
+      if (error !== null)
+        return error;
+    }
+
+  return null;
+}
+
+function main() : void {
+  console.log('foo');
+}
+
+let cell = {column: { name: 'age' }, value: 999 };
+validateCell(cell, ruleSet);
+
+console.log('ddd');