@datagrok-libraries/utils 0.0.1 → 0.0.5

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,eyJ2ZXJzaW9uIjozLCJmaWxlIjoic3BlLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsic3BlLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUNBLE9BQU8sRUFBQywwQkFBMEIsRUFBRSxrQkFBa0IsRUFBRSxnQkFBZ0IsRUFBRSxTQUFTLEVBQUUsU0FBUyxFQUFDLE1BQU0sY0FBYyxDQUFDO0FBRXBIOzs7Ozs7R0FNRztBQUNILE1BQU0sT0FBTyxPQUFPO0lBYWxCOzs7O09BSUc7SUFDSCxZQUFZLE9BQWlCOztRQUMzQixJQUFJLENBQUMsS0FBSyxTQUFHLE9BQU8sYUFBUCxPQUFPLHVCQUFQLE9BQU8sQ0FBRSxLQUFLLG1DQUFJLENBQUMsQ0FBQztRQUNqQyxJQUFJLENBQUMsTUFBTSxTQUFHLE9BQU8sYUFBUCxPQUFPLHVCQUFQLE9BQU8sQ0FBRSxNQUFNLG1DQUFJLEdBQUcsQ0FBQztRQUNyQyw0Q0FBNEM7UUFDNUMsSUFBSSxDQUFDLE1BQU0sU0FBRyxPQUFPLGFBQVAsT0FBTyx1QkFBUCxPQUFPLENBQUUsTUFBTSxtQ0FBSSxDQUFDLENBQUM7UUFDbkMsNENBQTRDO1FBQzVDLElBQUksQ0FBQyxNQUFNLFNBQUcsT0FBTyxhQUFQLE9BQU8sdUJBQVAsT0FBTyxDQUFFLE1BQU0sbUNBQUksR0FBRyxDQUFDO1FBQ3JDLElBQUksQ0FBQyxPQUFPLFNBQUcsT0FBTyxhQUFQLE9BQU8sdUJBQVAsT0FBTyxDQUFFLE9BQU8sbUNBQUksSUFBSSxDQUFDO1FBQ3hDLElBQUksQ0FBQyxPQUFPLEdBQUcsSUFBSSxDQUFDLE1BQU0sR0FBQyxFQUFFLENBQUM7UUFDOUIsSUFBSSxDQUFDLFFBQVEsR0FBRyxJQUFJLENBQUMsT0FBTyxHQUFDLEVBQUUsQ0FBQztRQUNoQyxJQUFJLENBQUMsT0FBTyxTQUFHLE9BQU8sYUFBUCxPQUFPLHVCQUFQLE9BQU8sQ0FBRSxPQUFPLG1DQUFJLEtBQUssQ0FBQztRQUN6Qyx1Q0FBdUM7UUFDdkMsSUFBSSxDQUFDLGdCQUFnQixTQUFHLE9BQU8sYUFBUCxPQUFPLHVCQUFQLE9BQU8sQ0FBRSxRQUFRLG1DQUFJLDBCQUEwQixDQUFDO1FBQ3hFLElBQUksQ0FBQyxRQUFRLEdBQUcsRUFBRSxDQUFDO0lBQ3JCLENBQUM7SUFFRDs7Ozs7O09BTUc7SUFDTyxZQUFZLENBQUMsT0FBZ0I7UUFDckMsSUFBSSxDQUFDLFFBQVEsR0FBRyxrQkFBa0IsQ0FBQyxPQUFPLEVBQUUsSUFBSSxDQUFDLGdCQUFnQixDQUFDLENBQUM7SUFDckUsQ0FBQztJQUVEOzs7Ozs7O09BT0c7SUFDTyxZQUFZLENBQUMsT0FBZ0IsRUFBRSxNQUFjLEVBQUUsTUFBYztRQUNyRSxPQUFPLElBQUksQ0FBQyxRQUFRLENBQUMsTUFBTSxDQUFDLENBQUMsTUFBTSxDQUFDLENBQUM7SUFDdkMsQ0FBQztJQUVEOzs7OztPQUtHO0lBQ0ksS0FBSyxDQUFDLE9BQWdCO1FBQzNCLE1BQU0sTUFBTSxHQUFHLE9BQU8sQ0FBQyxNQUFNLENBQUM7UUFDOUIsTUFBTSxTQUFTLEdBQUcsRUFBRSxDQUFDO1FBQ3JCLDREQUE0RDtRQUM1RCxNQUFNLFdBQVcsR0FBRyxnQkFBZ0IsQ0FBQyxNQUFNLEVBQUUsT0FBTyxDQUFDLFNBQVMsRUFBRSxTQUFTLENBQUMsQ0FBQztRQUUzRSxJQUFJLE9BQU8sR0FBRyxJQUFJLENBQUMsT0FBTyxDQUFDO1FBRTNCLElBQUksSUFBSSxDQUFDLEtBQUssSUFBSSxDQUFDLEVBQUU7WUFDbkIsSUFBSSxDQUFDLEtBQUssR0FBRyxPQUFPLENBQUMsTUFBTSxHQUFDLENBQUMsQ0FBQztTQUMvQjtRQUVELElBQUksQ0FBQyxZQUFZLENBQUMsT0FBTyxDQUFDLENBQUM7UUFFM0IsS0FBSyxJQUFJLEtBQUssR0FBRyxDQUFDLEVBQUUsS0FBSyxHQUFHLElBQUksQ0FBQyxNQUFNLEVBQUUsRUFBRSxLQUFLLEVBQUU7WUFDaEQsS0FBSyxJQUFJLElBQUksR0FBRyxDQUFDLEVBQUUsSUFBSSxHQUFHLElBQUksQ0FBQyxLQUFLLEVBQUUsRUFBRSxJQUFJLEVBQUU7Z0JBQzVDLDZDQUE2QztnQkFDN0MsTUFBTSxDQUFDLEdBQUcsU0FBUyxDQUFDLE1BQU0sQ0FBQyxDQUFDO2dCQUFDLElBQUksQ0FBQyxHQUFHLFNBQVMsQ0FBQyxNQUFNLENBQUMsQ0FBQztnQkFDdkQsT0FBTyxDQUFDLElBQUksQ0FBQztvQkFBRSxDQUFDLEdBQUcsU0FBUyxDQUFDLE1BQU0sQ0FBQyxDQUFDO2dCQUVyQyxNQUFNLElBQUksR0FBRyxXQUFXLENBQUMsQ0FBQyxDQUFDLENBQUM7Z0JBQUMsTUFBTSxJQUFJLEdBQUcsV0FBVyxDQUFDLENBQUMsQ0FBQyxDQUFDO2dCQUV6RCw2RUFBNkU7Z0JBQzdFLE1BQU0sQ0FBQyxHQUFHLElBQUksQ0FBQyxZQUFZLENBQUMsT0FBTyxFQUFFLENBQUMsRUFBRSxDQUFDLENBQUMsQ0FBQztnQkFDM0Msc0VBQXNFO2dCQUN0RSxNQUFNLENBQUMsR0FBRywwQkFBMEIsQ0FBQyxJQUFJLEVBQUUsSUFBSSxDQUFDLENBQUM7Z0JBRWpELGlEQUFpRDtnQkFDakQsSUFBSSxDQUFDLElBQUksQ0FBQyxNQUFNLElBQUksQ0FBQyxDQUFDLElBQUksQ0FBQyxDQUFDLElBQUksSUFBSSxDQUFDLE1BQU0sQ0FBQyxJQUFJLENBQUMsQ0FBQyxHQUFHLENBQUMsQ0FBQyxFQUFFO29CQUN2RCxNQUFNLFVBQVUsR0FBRyxPQUFPLEdBQUMsQ0FBQyxDQUFDLEdBQUMsQ0FBQyxDQUFDLEdBQUMsQ0FBQyxDQUFDLEdBQUMsSUFBSSxDQUFDLE9BQU8sQ0FBQyxDQUFDO29CQUNsRCx3Q0FBd0M7b0JBQ3hDLE1BQU0sTUFBTSxHQUFHLFNBQVMsQ0FBQyxJQUFJLEVBQUUsSUFBSSxFQUFFLENBQUMsQ0FBQyxDQUFDLENBQUM7b0JBQ3pDLFdBQVcsQ0FBQyxDQUFDLENBQUMsR0FBRyxTQUFTLENBQUMsSUFBSSxFQUFFLE1BQU0sRUFBRSxVQUFVLENBQUMsQ0FBQztvQkFDckQsV0FBVyxDQUFDLENBQUMsQ0FBQyxHQUFHLFNBQVMsQ0FBQyxJQUFJLEVBQUUsTUFBTSxFQUFFLENBQUMsVUFBVSxDQUFDLENBQUM7aUJBQ3ZEO2FBQ0Y7WUFDRCxpRUFBaUU7WUFDakUsT0FBTyxJQUFJLElBQUksQ0FBQyxRQUFRLENBQUM7WUFDekIsSUFBSSxPQUFPLElBQUksRUFBRSxFQUFFO2dCQUNqQixNQUFNO2FBQ1A7U0FDRjtRQUNELE9BQU8sV0FBVyxDQUFDO0lBQ3JCLENBQUM7O0FBekdnQixpQkFBUyxHQUFHLENBQUMsQ0FBQztBQTRHakM7Ozs7OztHQU1HO0FBQ0YsTUFBTSxPQUFPLFFBQVMsU0FBUSxPQUFPO0lBQ3BDOzs7OztPQUtHO0lBQ0ksS0FBSyxDQUFDLE9BQWdCO1FBQzNCLE1BQU0sTUFBTSxHQUFHLE9BQU8sQ0FBQyxNQUFNLENBQUM7UUFDOUIsTUFBTSxTQUFTLEdBQUcsRUFBRSxDQUFDO1FBQ3JCLDZEQUE2RDtRQUM3RCxNQUFNLFdBQVcsR0FBRyxnQkFBZ0IsQ0FBQyxNQUFNLEVBQUUsUUFBUSxDQUFDLFNBQVMsRUFBRSxTQUFTLENBQUMsQ0FBQztRQUM1RSxJQUFJLE1BQU0sR0FBRyxJQUFJLENBQUMsTUFBTSxDQUFDO1FBRXpCLElBQUksQ0FBQyxZQUFZLENBQUMsT0FBTyxDQUFDLENBQUM7UUFFM0IsS0FBSyxJQUFJLEtBQUssR0FBRyxDQUFDLEVBQUUsS0FBSyxHQUFHLElBQUksQ0FBQyxNQUFNLEVBQUUsRUFBRSxLQUFLLEVBQUU7WUFDaEQsd0NBQXdDO1lBQ3hDLE1BQU0sQ0FBQyxHQUFXLFNBQVMsQ0FBQyxNQUFNLENBQUMsQ0FBQztZQUNwQyxNQUFNLElBQUksR0FBRyxXQUFXLENBQUMsQ0FBQyxDQUFDLENBQUM7WUFFNUIsNkJBQTZCO1lBQzdCLEtBQUssSUFBSSxDQUFDLEdBQUcsQ0FBQyxFQUFFLENBQUMsR0FBRyxNQUFNLEVBQUUsRUFBRSxDQUFDLEVBQUU7Z0JBQy9CLElBQUksQ0FBQyxJQUFJLENBQUM7b0JBQUUsU0FBUztnQkFDckIsTUFBTSxJQUFJLEdBQUcsV0FBVyxDQUFDLENBQUMsQ0FBQyxDQUFDO2dCQUM1Qiw0RUFBNEU7Z0JBQzVFLE1BQU0sQ0FBQyxHQUFHLElBQUksQ0FBQyxZQUFZLENBQUMsT0FBTyxFQUFFLENBQUMsRUFBRSxDQUFDLENBQUMsQ0FBQztnQkFDM0MsMEVBQTBFO2dCQUMxRSxNQUFNLENBQUMsR0FBRywwQkFBMEIsQ0FBQyxJQUFJLEVBQUUsSUFBSSxDQUFDLENBQUM7Z0JBQ2pELGlEQUFpRDtnQkFDakQsSUFBSSxDQUFDLElBQUksQ0FBQyxNQUFNLElBQUksQ0FBQyxDQUFDLElBQUksQ0FBQyxDQUFDLElBQUksSUFBSSxDQUFDLE1BQU0sQ0FBQyxJQUFJLENBQUMsQ0FBQyxHQUFHLENBQUMsQ0FBQyxFQUFFO29CQUN2RCxNQUFNLFVBQVUsR0FBRyxNQUFNLEdBQUMsQ0FBQyxDQUFDLEdBQUMsQ0FBQyxDQUFDLEdBQUMsQ0FBQyxDQUFDLEdBQUMsSUFBSSxDQUFDLE9BQU8sQ0FBQyxDQUFDO29CQUNqRCxNQUFNLE1BQU0sR0FBRyxTQUFTLENBQUMsSUFBSSxFQUFFLElBQUksRUFBRSxDQUFDLENBQUMsQ0FBQyxDQUFDO29CQUN6QyxpQ0FBaUM7b0JBQ2pDLFdBQVcsQ0FBQyxDQUFDLENBQUMsR0FBRyxTQUFTLENBQUMsSUFBSSxFQUFFLE1BQU0sRUFBRSxDQUFDLFVBQVUsQ0FBQyxDQUFDO2lCQUN2RDthQUNGO1lBQ0QsaUVBQWlFO1lBQ2pFLE1BQU0sSUFBSSxJQUFJLENBQUMsT0FBTyxDQUFDO1lBQ3ZCLElBQUksTUFBTSxJQUFJLEVBQUUsRUFBRTtnQkFDaEIsTUFBTTthQUNQO1NBQ0Y7UUFDRCxPQUFPLFdBQVcsQ0FBQztJQUNyQixDQUFDO0NBQ0YiLCJzb3VyY2VzQ29udGVudCI6WyJpbXBvcnQge09wdGlvbnMsIENvb3JkaW5hdGVzLCBWZWN0b3JzLCBEaXN0YW5jZU1ldHJpY30gZnJvbSAnLi90eXBlX2RlY2xhcmF0aW9ucyc7XHJcbmltcG9ydCB7Y2FsY3VsYXRlRXVjbGlkZWFuRGlzdGFuY2UsIGNhbGNEaXN0YW5jZU1hdHJpeCwgZmlsbFJhbmRvbU1hdHJpeCwgdmVjdG9yQWRkLCByYW5kb21JbnR9IGZyb20gJy4vb3BlcmF0aW9ucyc7XHJcblxyXG4vKipcclxuICogSW1wbGVtZW50cyBzdG9jaGFzdGljIHByb3hpbWl0eSBlbWJlZGRpbmcuXHJcbiAqXHJcbiAqIEBleHBvcnRcclxuICogQGNsYXNzIFNQRUJhc2VcclxuICogQGxpbmsgZG9pOjEwLjEwMTYvUzEwOTMtMzI2MygwMykwMDE1NS00XHJcbiAqL1xyXG5leHBvcnQgY2xhc3MgU1BFQmFzZSB7XHJcbiAgcHJvdGVjdGVkIHN0YXRpYyBkaW1lbnNpb24gPSAyO1xyXG4gIHByb3RlY3RlZCBzdGVwczogbnVtYmVyO1xyXG4gIHByb3RlY3RlZCBjeWNsZXM6IG51bWJlcjtcclxuICBwcm90ZWN0ZWQgY3V0b2ZmOiBudW1iZXI7XHJcbiAgcHJvdGVjdGVkIGxhbWJkYTogbnVtYmVyO1xyXG4gIHByb3RlY3RlZCBkbGFtYmRhOiBudW1iZXI7XHJcbiAgcHJvdGVjdGVkIGxhbWJkYTI6IG51bWJlcjtcclxuICBwcm90ZWN0ZWQgZGxhbWJkYTI6IG51bWJlcjtcclxuICBwcm90ZWN0ZWQgZXBzaWxvbjogbnVtYmVyO1xyXG4gIHByb3RlY3RlZCBkaXN0YW5jZUZ1bmN0aW9uOiBEaXN0YW5jZU1ldHJpYztcclxuICBwcm90ZWN0ZWQgZGlzdGFuY2U6IENvb3JkaW5hdGVzO1xyXG5cclxuICAvKipcclxuICAgKiBDcmVhdGVzIGFuIGluc3RhbmNlIG9mIFNQRUJhc2UuXHJcbiAgICogQHBhcmFtIHtPcHRpb25zfSBbb3B0aW9uc10gT3B0aW9ucyB0byBwYXNzIHRvIHRoZSBjb25zdHJ1Y3Rvci5cclxuICAgKiBAbWVtYmVyb2YgU1BFQmFzZVxyXG4gICAqL1xyXG4gIGNvbnN0cnVjdG9yKG9wdGlvbnM/OiBPcHRpb25zKSB7XHJcbiAgICB0aGlzLnN0ZXBzID0gb3B0aW9ucz8uc3RlcHMgPz8gMDtcclxuICAgIHRoaXMuY3ljbGVzID0gb3B0aW9ucz8uY3ljbGVzID8/IDFlNjtcclxuICAgIC8vIFNlbGVjdCBhIGN1dG9mZiBkaXN0YW5jZSB7Y3V0b2ZmfSBhbmQgLi4uXHJcbiAgICB0aGlzLmN1dG9mZiA9IG9wdGlvbnM/LmN1dG9mZiA/PyAwO1xyXG4gICAgLy8gLi4uIGFuIGluaXRpYWwgbGVhcm5pbmcgcmF0ZSB7bGFtYmRhfSA+IDBcclxuICAgIHRoaXMubGFtYmRhID0gb3B0aW9ucz8ubGFtYmRhID8/IDIuMDtcclxuICAgIHRoaXMuZGxhbWJkYSA9IG9wdGlvbnM/LmRsYW1iZGEgPz8gMC4wMTtcclxuICAgIHRoaXMubGFtYmRhMiA9IHRoaXMubGFtYmRhLzIuO1xyXG4gICAgdGhpcy5kbGFtYmRhMiA9IHRoaXMuZGxhbWJkYS8yLjtcclxuICAgIHRoaXMuZXBzaWxvbiA9IG9wdGlvbnM/LmVwc2lsb24gPz8gMWUtMTA7XHJcbiAgICAvLyBlc2xpbnQtZGlzYWJsZS1uZXh0LWxpbmUgYnJhY2Utc3R5bGVcclxuICAgIHRoaXMuZGlzdGFuY2VGdW5jdGlvbiA9IG9wdGlvbnM/LmRpc3RhbmNlID8/IGNhbGN1bGF0ZUV1Y2xpZGVhbkRpc3RhbmNlO1xyXG4gICAgdGhpcy5kaXN0YW5jZSA9IFtdO1xyXG4gIH1cclxuXHJcbiAgLyoqXHJcbiAgICogSW5pdGlhbGl6ZXMgZGlzdGFuY2UgbWF0cml4LlxyXG4gICAqXHJcbiAgICogQHByb3RlY3RlZFxyXG4gICAqIEBwYXJhbSB7VmVjdG9yc30gdmVjdG9ycyBJbnB1dCB2ZWN0b3JzIHRvIGNhbGN1bGF0ZSBkaXN0YW5jZSBiZXR3ZWVuLlxyXG4gICAqIEBtZW1iZXJvZiBTUEVCYXNlXHJcbiAgICovXHJcbiAgcHJvdGVjdGVkIGluaXREaXN0YW5jZSh2ZWN0b3JzOiBWZWN0b3JzKSB7XHJcbiAgICB0aGlzLmRpc3RhbmNlID0gY2FsY0Rpc3RhbmNlTWF0cml4KHZlY3RvcnMsIHRoaXMuZGlzdGFuY2VGdW5jdGlvbik7XHJcbiAgfVxyXG5cclxuICAvKipcclxuICAgKiBDYWxjdWxhdGVzIGRpc3RhbmNlIGJldHdlZW4gdGhlIHR3byB2ZWN0b3JzIGdpdmVuLlxyXG4gICAqIFxyXG4gICAqIEBwYXJhbSB7VmVjdG9yc30gdmVjdG9ycyBTZXQgb2YgdmVjdG9ycyB0byBjYWxjdWxhdGUgZGlzdGFuY2VzIGJldHdlZW4uXHJcbiAgICogQHBhcmFtIHtudW1iZXJ9IGluZGV4MSBJbmRleCBvZiB0aGUgZmlyc3QgdmVjdG9yIG9mIHRoZSBwYWlyLlxyXG4gICAqIEBwYXJhbSB7bnVtYmVyfSBpbmRleDIgSW5kZXggb2YgdGhlIHNlY29uZCB2ZWN0b3Igb2YgdGhlIHBhaXIuXHJcbiAgICogQHJldHVybiB7bnVtYmVyfSBEaXN0YW5jZSBiZXR3ZWVuIHRoZXNlIHR3byB2ZWN0b3JzLlxyXG4gICAqL1xyXG4gIHByb3RlY3RlZCBjYWxjRGlzdGFuY2UodmVjdG9yczogVmVjdG9ycywgaW5kZXgxOiBudW1iZXIsIGluZGV4MjogbnVtYmVyKTogbnVtYmVyIHtcclxuICAgIHJldHVybiB0aGlzLmRpc3RhbmNlW2luZGV4MV1baW5kZXgyXTtcclxuICB9XHJcblxyXG4gIC8qKlxyXG4gICAqIEVtYmVkcyB0aGUgdmVjdG9ycyBnaXZlbiBpbnRvIGEgdHdvLWRpbWVuc2lvbmFsIHNwYWNlLlxyXG4gICAqIFxyXG4gICAqIEBwYXJhbSB7VmVjdG9yc30gdmVjdG9ycyBELWRpbWVuc2lvbmFsIGNvb3JkaW5hdGVzLlxyXG4gICAqIEByZXR1cm4ge0Nvb3JkaW5hdGVzfSBTUEUgY29vcmRpbmF0ZXMgaW4gRCBzcGFjZS5cclxuICAgKi9cclxuICBwdWJsaWMgZW1iZWQodmVjdG9yczogVmVjdG9ycyk6IENvb3JkaW5hdGVzIHtcclxuICAgIGNvbnN0IG5JdGVtcyA9IHZlY3RvcnMubGVuZ3RoO1xyXG4gICAgY29uc3QgYXJlYVdpZHRoID0gNDA7XHJcbiAgICAvLyBJbml0aWFsaXplIHRoZSBELWRpbWVuc2lvbmFsIGNvb3JkaW5hdGVzIG9mIHRoZSBOIHBvaW50cy5cclxuICAgIGNvbnN0IGNvb3JkaW5hdGVzID0gZmlsbFJhbmRvbU1hdHJpeChuSXRlbXMsIFNQRUJhc2UuZGltZW5zaW9uLCBhcmVhV2lkdGgpO1xyXG5cclxuICAgIGxldCBsYW1iZGEyID0gdGhpcy5sYW1iZGEyO1xyXG5cclxuICAgIGlmICh0aGlzLnN0ZXBzID09IDApIHtcclxuICAgICAgdGhpcy5zdGVwcyA9IHZlY3RvcnMubGVuZ3RoLTE7XHJcbiAgICB9XHJcblxyXG4gICAgdGhpcy5pbml0RGlzdGFuY2UodmVjdG9ycyk7XHJcblxyXG4gICAgZm9yIChsZXQgY3ljbGUgPSAwOyBjeWNsZSA8IHRoaXMuY3ljbGVzOyArK2N5Y2xlKSB7XHJcbiAgICAgIGZvciAobGV0IHN0ZXAgPSAwOyBzdGVwIDwgdGhpcy5zdGVwczsgKytzdGVwKSB7XHJcbiAgICAgICAgLy8gU2VsZWN0IHR3byBwb2ludHMsIGkgYW5kIGosIGF0IHJhbmRvbSwgLi4uXHJcbiAgICAgICAgY29uc3QgaSA9IHJhbmRvbUludChuSXRlbXMpOyBsZXQgaiA9IHJhbmRvbUludChuSXRlbXMpO1xyXG4gICAgICAgIHdoaWxlIChpID09IGopIGogPSByYW5kb21JbnQobkl0ZW1zKTtcclxuXHJcbiAgICAgICAgY29uc3Qgcm93aSA9IGNvb3JkaW5hdGVzW2ldOyBjb25zdCByb3dqID0gY29vcmRpbmF0ZXNbal07XHJcblxyXG4gICAgICAgIC8vIC4uLiByZXRyaWV2ZSAob3IgZXZhbHVhdGUpIHRoZWlyIHByb3hpbWl0eSBpbiB0aGUgaW5wdXQgc3BhY2UsIHJpaiBhbmQgLi4uXHJcbiAgICAgICAgY29uc3QgciA9IHRoaXMuY2FsY0Rpc3RhbmNlKHZlY3RvcnMsIGksIGopO1xyXG4gICAgICAgIC8vIC4uLiBjb21wdXRlIHRoZWlyIEV1Y2xpZGVhbiBkaXN0YW5jZSBvbiB0aGUgRC1kaW1lbnNpb25hbCBtYXAsIGRpai5cclxuICAgICAgICBjb25zdCBkID0gY2FsY3VsYXRlRXVjbGlkZWFuRGlzdGFuY2Uocm93aSwgcm93aik7XHJcblxyXG4gICAgICAgIC8vIElmIHJpaiA8PSByYywgb3IgaWYgcmlqID4gcmMgYW5kIGRpaiA8IHJpaiAuLi5cclxuICAgICAgICBpZiAoKHRoaXMuY3V0b2ZmID09IDApIHx8IChyIDw9IHRoaXMuY3V0b2ZmKSB8fCAoZCA8IHIpKSB7XHJcbiAgICAgICAgICBjb25zdCBtdWx0aXBsaWVyID0gbGFtYmRhMiooci1kKS8oZCt0aGlzLmVwc2lsb24pO1xyXG4gICAgICAgICAgLy8gLi4uIHVwZGF0ZSB0aGUgY29vcmRpbmF0ZXMgeGkgYW5kIHhqLlxyXG4gICAgICAgICAgY29uc3QgZGlmZklKID0gdmVjdG9yQWRkKHJvd2ksIHJvd2osIC0xKTtcclxuICAgICAgICAgIGNvb3JkaW5hdGVzW2ldID0gdmVjdG9yQWRkKHJvd2ksIGRpZmZJSiwgbXVsdGlwbGllcik7XHJcbiAgICAgICAgICBjb29yZGluYXRlc1tqXSA9IHZlY3RvckFkZChyb3dqLCBkaWZmSUosIC1tdWx0aXBsaWVyKTtcclxuICAgICAgICB9XHJcbiAgICAgIH1cclxuICAgICAgLy8gRGVjcmVhc2UgdGhlIGxlYXJuaW5nIHJhdGUge2xhbWJkYX0gYnkgYSBwcmVzY3JpYmVkIHtkbGFtYmRhfS5cclxuICAgICAgbGFtYmRhMiAtPSB0aGlzLmRsYW1iZGEyO1xyXG4gICAgICBpZiAobGFtYmRhMiA8PSAwLikge1xyXG4gICAgICAgIGJyZWFrO1xyXG4gICAgICB9XHJcbiAgICB9XHJcbiAgICByZXR1cm4gY29vcmRpbmF0ZXM7XHJcbiAgfVxyXG59XHJcblxyXG4vKipcclxuICogSW1wbGVtZW50cyBtb2RpZmllZCBzdG9jaGFzdGljIHByb3hpbWl0eSBlbWJlZGRpbmcuXHJcbiAqXHJcbiAqIEBleHBvcnRcclxuICogQGNsYXNzIFBTUEVCYXNlXHJcbiAqIEBsaW5rIGRvaToxMC4xMDE2L1MxMDkzLTMyNjMoMDMpMDAxNTUtNFxyXG4gKi9cclxuIGV4cG9ydCBjbGFzcyBQU1BFQmFzZSBleHRlbmRzIFNQRUJhc2Uge1xyXG4gIC8qKlxyXG4gICAqIEVtYmVkcyB0aGUgdmVjdG9ycyBnaXZlbiBpbnRvIGEgdHdvLWRpbWVuc2lvbmFsIHNwYWNlIHVzaW5nIGEgbW9kaWZpZWQgdXBkYXRlIHJ1bGUuXHJcbiAgICogXHJcbiAgICogQHBhcmFtIHtWZWN0b3JzfSB2ZWN0b3JzIEQtZGltZW5zaW9uYWwgY29vcmRpbmF0ZXMuXHJcbiAgICogQHJldHVybiB7Q29vcmRpbmF0ZXN9IFNQRSBjb29yZGluYXRlcyBpbiBEIHNwYWNlLlxyXG4gICAqL1xyXG4gIHB1YmxpYyBlbWJlZCh2ZWN0b3JzOiBWZWN0b3JzKTogQ29vcmRpbmF0ZXMge1xyXG4gICAgY29uc3Qgbkl0ZW1zID0gdmVjdG9ycy5sZW5ndGg7XHJcbiAgICBjb25zdCBhcmVhV2lkdGggPSA0MDtcclxuICAgIC8vICBJbml0aWFsaXplIHRoZSBELWRpbWVuc2lvbmFsIGNvb3JkaW5hdGVzIG9mIHRoZSBOIHBvaW50cy5cclxuICAgIGNvbnN0IGNvb3JkaW5hdGVzID0gZmlsbFJhbmRvbU1hdHJpeChuSXRlbXMsIFBTUEVCYXNlLmRpbWVuc2lvbiwgYXJlYVdpZHRoKTtcclxuICAgIGxldCBsYW1iZGEgPSB0aGlzLmxhbWJkYTtcclxuXHJcbiAgICB0aGlzLmluaXREaXN0YW5jZSh2ZWN0b3JzKTtcclxuXHJcbiAgICBmb3IgKGxldCBjeWNsZSA9IDA7IGN5Y2xlIDwgdGhpcy5jeWNsZXM7ICsrY3ljbGUpIHtcclxuICAgICAgLy8gU2VsZWN0IGEgcG9pbnQsIGksIGF0IHJhbmRvbSAocGl2b3QpLlxyXG4gICAgICBjb25zdCBpOiBudW1iZXIgPSByYW5kb21JbnQobkl0ZW1zKTtcclxuICAgICAgY29uc3Qgcm93aSA9IGNvb3JkaW5hdGVzW2ldO1xyXG5cclxuICAgICAgLy8gRm9yIGV2ZXJ5IHBvaW50IGogIT0gaSAuLi5cclxuICAgICAgZm9yIChsZXQgaiA9IDA7IGogPCBuSXRlbXM7ICsraikge1xyXG4gICAgICAgIGlmIChpID09IGopIGNvbnRpbnVlO1xyXG4gICAgICAgIGNvbnN0IHJvd2ogPSBjb29yZGluYXRlc1tqXTtcclxuICAgICAgICAvLyAuLi4gcmV0cmlldmUgKG9yIGV2YWx1YXRlKSBpdHMgcHJveGltaXR5IHRvIGkgaW4gdGhlIGlucHV0IHNwYWNlLCByaWogLi4uXHJcbiAgICAgICAgY29uc3QgciA9IHRoaXMuY2FsY0Rpc3RhbmNlKHZlY3RvcnMsIGksIGopO1xyXG4gICAgICAgIC8vIC4uLiBhbmQgY29tcHV0ZSB0aGVpciBFdWNsaWRlYW4gZGlzdGFuY2Ugb24gdGhlIEQtZGltZW5zaW9uYWwgbWFwLCBkaWouXHJcbiAgICAgICAgY29uc3QgZCA9IGNhbGN1bGF0ZUV1Y2xpZGVhbkRpc3RhbmNlKHJvd2ksIHJvd2opO1xyXG4gICAgICAgIC8vIElmIHJpaiA8PSByYywgb3IgaWYgcmlqID4gcmMgYW5kIGRpaiA8IHJpaiAuLi5cclxuICAgICAgICBpZiAoKHRoaXMuY3V0b2ZmID09IDApIHx8IChyIDw9IHRoaXMuY3V0b2ZmKSB8fCAoZCA8IHIpKSB7XHJcbiAgICAgICAgICBjb25zdCBtdWx0aXBsaWVyID0gbGFtYmRhKihyLWQpLyhkK3RoaXMuZXBzaWxvbik7XHJcbiAgICAgICAgICBjb25zdCBkaWZmSUogPSB2ZWN0b3JBZGQocm93aSwgcm93aiwgLTEpO1xyXG4gICAgICAgICAgLy8gLi4uIHVwZGF0ZSB0aGUgY29vcmRpbmF0ZXMgeGouXHJcbiAgICAgICAgICBjb29yZGluYXRlc1tqXSA9IHZlY3RvckFkZChyb3dqLCBkaWZmSUosIC1tdWx0aXBsaWVyKTtcclxuICAgICAgICB9XHJcbiAgICAgIH1cclxuICAgICAgLy8gRGVjcmVhc2UgdGhlIGxlYXJuaW5nIHJhdGUge2xhbWJkYX0gYnkgYSBwcmVzY3JpYmVkIHtkbGFtYmRhfS5cclxuICAgICAgbGFtYmRhIC09IHRoaXMuZGxhbWJkYTtcclxuICAgICAgaWYgKGxhbWJkYSA8PSAwLikge1xyXG4gICAgICAgIGJyZWFrO1xyXG4gICAgICB9XHJcbiAgICB9XHJcbiAgICByZXR1cm4gY29vcmRpbmF0ZXM7XHJcbiAgfVxyXG59XHJcbiJdfQ==

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) {
+        this.receipt = {
+            'Levenshtein': fl.distance,
+            'Jaro-Winkler': jaroWinkler,
+        };
+        assert(this.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 this.receipt[this.method];
+    }
+    /**
+     * Returns available string distance metrics.
+     *
+     * @readonly
+     * @type {string[]}
+     * @memberof Measurer
+     */
+    get availableMeasures() {
+        let keys = [];
+        for (let key in this.receipt) {
+            keys.push(key);
+        }
+        return keys;
+    }
+}
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoic3RyaW5nLW1lYXN1cmUuanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyJzdHJpbmctbWVhc3VyZS50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiQUFBQSxPQUFPLEtBQUssRUFBRSxNQUFNLHFCQUFxQixDQUFDO0FBQzFDLE9BQU8sRUFBQyxXQUFXLEVBQUMsTUFBTSx5QkFBeUIsQ0FBQztBQUdwRCxPQUFPLEVBQUMsTUFBTSxFQUFDLE1BQU0sY0FBYyxDQUFDO0FBRXBDOzs7OztHQUtHO0FBQ0gsTUFBTSxPQUFPLFFBQVE7SUFPakI7Ozs7T0FJRztJQUNILFlBQVksTUFBYztRQVZoQixZQUFPLEdBQXFDO1lBQ3BELGFBQWEsRUFBRSxFQUFFLENBQUMsUUFBUTtZQUMxQixjQUFjLEVBQUUsV0FBVztTQUM1QixDQUFDO1FBUUEsTUFBTSxDQUFDLElBQUksQ0FBQyxpQkFBaUIsQ0FBQyxRQUFRLENBQUMsTUFBTSxDQUFDLEVBQUUsOEJBQThCLENBQUMsQ0FBQTtRQUMvRSxJQUFJLENBQUMsTUFBTSxHQUFHLE1BQU0sQ0FBQztJQUN2QixDQUFDO0lBRUQ7OztPQUdHO0lBQ0ksVUFBVTtRQUNmLE9BQU8sSUFBSSxDQUFDLE9BQU8sQ0FBQyxJQUFJLENBQUMsTUFBTSxDQUFDLENBQUM7SUFDbkMsQ0FBQztJQUVEOzs7Ozs7T0FNRztJQUNILElBQVcsaUJBQWlCO1FBQzFCLElBQUksSUFBSSxHQUFHLEVBQUUsQ0FBQztRQUVkLEtBQUssSUFBSSxHQUFHLElBQUksSUFBSSxDQUFDLE9BQU8sRUFBRTtZQUM1QixJQUFJLENBQUMsSUFBSSxDQUFDLEdBQUcsQ0FBQyxDQUFDO1NBQ2hCO1FBQ0QsT0FBTyxJQUFJLENBQUM7SUFDZCxDQUFDO0NBRUoiLCJzb3VyY2VzQ29udGVudCI6WyJpbXBvcnQgKiBhcyBmbCBmcm9tICdmYXN0ZXN0LWxldmVuc2h0ZWluJztcclxuaW1wb3J0IHtqYXJvV2lua2xlcn0gZnJvbSAnamFyby13aW5rbGVyLXR5cGVzY3JpcHQnO1xyXG5cclxuaW1wb3J0IHtEaXN0YW5jZU1ldHJpY30gZnJvbSAnLi90eXBlX2RlY2xhcmF0aW9ucyc7XHJcbmltcG9ydCB7YXNzZXJ0fSBmcm9tICcuL29wZXJhdGlvbnMnO1xyXG5cclxuLyoqXHJcbiAqIFVuaWZpZWQgY2xhc3MgaW1wbGVtZW50aW5nIHRoZSBkaWZmZXJlbnQgc3RyaW5nIG1lYXN1cmVzLlxyXG4gKlxyXG4gKiBAZXhwb3J0XHJcbiAqIEBjbGFzcyBNZWFzdXJlclxyXG4gKi9cclxuZXhwb3J0IGNsYXNzIE1lYXN1cmVyIHtcclxuICAgIHByb3RlY3RlZCBtZXRob2Q6IHN0cmluZztcclxuICAgIHByb3RlY3RlZCByZWNlaXB0OiB7W25hbWU6IHN0cmluZ106IERpc3RhbmNlTWV0cmljfSA9IHtcclxuICAgICAgJ0xldmVuc2h0ZWluJzogZmwuZGlzdGFuY2UsXHJcbiAgICAgICdKYXJvLVdpbmtsZXInOiBqYXJvV2lua2xlcixcclxuICAgIH07XHJcblxyXG4gICAgLyoqXHJcbiAgICAgKiBDcmVhdGVzIGFuIGluc3RhbmNlIG9mIE1lYXN1cmVyLlxyXG4gICAgICogQHBhcmFtIHtzdHJpbmd9IG1ldGhvZCBNZXRob2QgdG8gY2FsY3VsYXRlIGRpc3RhbmNlIGJldHdlZW4gc3RyaW5ncy5cclxuICAgICAqIEBtZW1iZXJvZiBNZWFzdXJlclxyXG4gICAgICovXHJcbiAgICBjb25zdHJ1Y3RvcihtZXRob2Q6IHN0cmluZykge1xyXG4gICAgICBhc3NlcnQodGhpcy5hdmFpbGFibGVNZWFzdXJlcy5pbmNsdWRlcyhtZXRob2QpLCAnVGhlICR7bWV0aG9kfSB3YXMgbm90IGZvdW5kLicpXHJcbiAgICAgIHRoaXMubWV0aG9kID0gbWV0aG9kO1xyXG4gICAgfVxyXG5cclxuICAgIC8qKlxyXG4gICAgICogUmV0dXJucyBjdXN0b20gc3RyaW5nIGRpc3RhbmNlIGZ1bmN0aW9uIHNwZWNpZmllZC5cclxuICAgICAqIEByZXR1cm4ge0Rpc3RhbmNlTWV0cmljfSBDYWxsYmFjayBvZiB0aGUgbWVhc3VyZSBjaG9zZW4uXHJcbiAgICAgKi9cclxuICAgIHB1YmxpYyBnZXRNZWFzdXJlKCk6IERpc3RhbmNlTWV0cmljIHtcclxuICAgICAgcmV0dXJuIHRoaXMucmVjZWlwdFt0aGlzLm1ldGhvZF07XHJcbiAgICB9XHJcblxyXG4gICAgLyoqXHJcbiAgICAgKiBSZXR1cm5zIGF2YWlsYWJsZSBzdHJpbmcgZGlzdGFuY2UgbWV0cmljcy5cclxuICAgICAqXHJcbiAgICAgKiBAcmVhZG9ubHlcclxuICAgICAqIEB0eXBlIHtzdHJpbmdbXX1cclxuICAgICAqIEBtZW1iZXJvZiBNZWFzdXJlclxyXG4gICAgICovXHJcbiAgICBwdWJsaWMgZ2V0IGF2YWlsYWJsZU1lYXN1cmVzKCkgOiBzdHJpbmdbXSB7XHJcbiAgICAgIGxldCBrZXlzID0gW107XHJcblxyXG4gICAgICBmb3IgKGxldCBrZXkgaW4gdGhpcy5yZWNlaXB0KSB7XHJcbiAgICAgICAga2V5cy5wdXNoKGtleSk7XHJcbiAgICAgIH1cclxuICAgICAgcmV0dXJuIGtleXM7XHJcbiAgICB9XHJcbiAgICBcclxufVxyXG4iXX0=

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;
+    protected 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(this.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 this.receipt[this.method];
+    }
+
+    /**
+     * Returns available string distance metrics.
+     *
+     * @readonly
+     * @type {string[]}
+     * @memberof Measurer
+     */
+    public get availableMeasures() : string[] {
+      let keys = [];
+
+      for (let key in this.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,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidGFibGUtdmFsaWRhdGlvbi5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbInRhYmxlLXZhbGlkYXRpb24udHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6IkFBR0EsT0FBTyxLQUFLLEVBQUUsTUFBTSxpQkFBaUIsQ0FBQztBQWN0QyxhQUFhO0FBQ2IsSUFBSSxPQUFPLEdBQVk7SUFDckIsSUFBSSxFQUFFLGVBQWU7SUFDckIsS0FBSyxFQUFFO1FBQ0w7WUFDRSxNQUFNLEVBQUUsS0FBSztZQUNiLElBQUksRUFBRSxLQUFLO1lBQ1gsT0FBTyxFQUFFLE9BQU87U0FDakI7UUFDRDtZQUNFLE1BQU0sRUFBRSxRQUFRO1lBQ2hCLElBQUksRUFBRSxXQUFXO1lBQ2pCLE9BQU8sRUFBRSxPQUFPO1NBQ2pCO1FBQ0Q7WUFDRSxNQUFNLEVBQUUsS0FBSztZQUNiLElBQUksRUFBRSxRQUFRO1lBQ2QsT0FBTyxFQUFFLFdBQVc7U0FDckI7UUFDRDtZQUNFLE1BQU0sRUFBRSxTQUFTO1lBQ2pCLElBQUksRUFBRSxVQUFVO1lBQ2hCLE9BQU8sRUFBRSxZQUFZO1NBQ3RCO0tBQ0Y7Q0FDRixDQUFDO0FBRUYsK0NBQStDO0FBQy9DLFNBQVMsWUFBWSxDQUFDLElBQVMsRUFBRSxPQUFnQjs7SUFDL0MsSUFBSSxJQUFJLEdBQUcsSUFBSSxDQUFDLE1BQU0sQ0FBQyxJQUFJLENBQUMsV0FBVyxFQUFFLENBQUM7SUFFMUMsS0FBSyxJQUFJLElBQUksSUFBSSxPQUFPLENBQUMsS0FBSztRQUM1QixJQUFJLElBQUksQ0FBQyxNQUFNLElBQUksSUFBSSxFQUFFO1lBQ3ZCLElBQUksQ0FBQyxPQUFPLFNBQUcsSUFBSSxDQUFDLE9BQU8sbUNBQUksRUFBRSxDQUFDLFlBQVksQ0FBQyxTQUFTLENBQUMsSUFBSSxDQUFDLE1BQU0sRUFBRSxJQUFJLENBQUMsT0FBTyxDQUFDLENBQUM7WUFDcEYsSUFBSSxLQUFLLEdBQUcsSUFBSSxDQUFDLE9BQU8sQ0FBQyxRQUFRLENBQUMsSUFBSSxDQUFDLEtBQUssQ0FBQyxDQUFDO1lBQzlDLElBQUksS0FBSyxLQUFLLElBQUk7Z0JBQ2hCLE9BQU8sS0FBSyxDQUFDO1NBQ2hCO0lBRUgsT0FBTyxJQUFJLENBQUM7QUFDZCxDQUFDO0FBRUQsU0FBUyxJQUFJO0lBQ1gsT0FBTyxDQUFDLEdBQUcsQ0FBQyxLQUFLLENBQUMsQ0FBQztBQUNyQixDQUFDO0FBRUQsSUFBSSxJQUFJLEdBQUcsRUFBQyxNQUFNLEVBQUUsRUFBRSxJQUFJLEVBQUUsS0FBSyxFQUFFLEVBQUUsS0FBSyxFQUFFLEdBQUcsRUFBRSxDQUFDO0FBQ2xELFlBQVksQ0FBQyxJQUFJLEVBQUUsT0FBTyxDQUFDLENBQUM7QUFFNUIsT0FBTyxDQUFDLEdBQUcsQ0FBQyxLQUFLLENBQUMsQ0FBQyIsInNvdXJjZXNDb250ZW50IjpbIi8qIERvIG5vdCBjaGFuZ2UgdGhlc2UgaW1wb3J0IGxpbmVzLiBEYXRhZ3JvayB3aWxsIGltcG9ydCBBUEkgbGlicmFyeSBpbiBleGFjdGx5IHRoZSBzYW1lIG1hbm5lciAqL1xyXG5pbXBvcnQgKiBhcyBncm9rIGZyb20gJ2RhdGFncm9rLWFwaS9ncm9rJztcclxuaW1wb3J0ICogYXMgdWkgZnJvbSAnZGF0YWdyb2stYXBpL3VpJztcclxuaW1wb3J0ICogYXMgREcgZnJvbSBcImRhdGFncm9rLWFwaS9kZ1wiO1xyXG5cclxuaW50ZXJmYWNlIFJ1bGUge1xyXG4gIGNvbHVtbjogc3RyaW5nO1xyXG4gIHR5cGU6IHN0cmluZztcclxuICBwYXR0ZXJuOiBzdHJpbmc7XHJcbiAgbWF0Y2hlcj86IGFueTtcclxufVxyXG5cclxuaW50ZXJmYWNlIFJ1bGVTZXQge1xyXG4gIG5hbWU6IHN0cmluZztcclxuICBydWxlczogUnVsZVtdO1xyXG59XHJcblxyXG4vLyBAdHMtaWdub3JlXHJcbmxldCBydWxlU2V0OiBSdWxlU2V0ID0ge1xyXG4gIG5hbWU6ICdTRFRNIERNIHJ1bGVzJyxcclxuICBydWxlczogW1xyXG4gICAge1xyXG4gICAgICBjb2x1bW46ICdhZ2UnLFxyXG4gICAgICB0eXBlOiAnaW50JyxcclxuICAgICAgcGF0dGVybjogJzE4LTkwJ1xyXG4gICAgfSxcclxuICAgIHtcclxuICAgICAgY29sdW1uOiAnaGVpZ2h0JyxcclxuICAgICAgdHlwZTogJ251bWVyaWNhbCcsXHJcbiAgICAgIHBhdHRlcm46ICcxOC05MCdcclxuICAgIH0sXHJcbiAgICB7XHJcbiAgICAgIGNvbHVtbjogJ3NleCcsXHJcbiAgICAgIHR5cGU6ICdzdHJpbmcnLFxyXG4gICAgICBwYXR0ZXJuOiAnaW4gKEYsIE0pJ1xyXG4gICAgfSxcclxuICAgIHtcclxuICAgICAgY29sdW1uOiAnc3RhcnRlZCcsXHJcbiAgICAgIHR5cGU6ICdkYXRldGltZScsXHJcbiAgICAgIHBhdHRlcm46ICc+IDEvMS8xOTAwJ1xyXG4gICAgfVxyXG4gIF1cclxufTtcclxuXHJcbi8qKiBSZXR1cm5zIGVycm9yIG1lc3NhZ2UsIG9yIG51bGwgaWYgdmFsaWQuICovXHJcbmZ1bmN0aW9uIHZhbGlkYXRlQ2VsbChjZWxsOiBhbnksIHJ1bGVzZXQ6IFJ1bGVTZXQpIDogc3RyaW5nIHwgbnVsbCB7XHJcbiAgbGV0IG5hbWUgPSBjZWxsLmNvbHVtbi5uYW1lLnRvTG93ZXJDYXNlKCk7XHJcblxyXG4gIGZvciAobGV0IHJ1bGUgb2YgcnVsZXNldC5ydWxlcylcclxuICAgIGlmIChydWxlLmNvbHVtbiA9PSBuYW1lKSB7XHJcbiAgICAgIHJ1bGUubWF0Y2hlciA9IHJ1bGUubWF0Y2hlciA/PyBERy5WYWx1ZU1hdGNoZXIuZm9yQ29sdW1uKGNlbGwuY29sdW1uLCBydWxlLnBhdHRlcm4pO1xyXG4gICAgICBsZXQgZXJyb3IgPSBydWxlLm1hdGNoZXIudmFsaWRhdGUoY2VsbC52YWx1ZSk7XHJcbiAgICAgIGlmIChlcnJvciAhPT0gbnVsbClcclxuICAgICAgICByZXR1cm4gZXJyb3I7XHJcbiAgICB9XHJcblxyXG4gIHJldHVybiBudWxsO1xyXG59XHJcblxyXG5mdW5jdGlvbiBtYWluKCkgOiB2b2lkIHtcclxuICBjb25zb2xlLmxvZygnZm9vJyk7XHJcbn1cclxuXHJcbmxldCBjZWxsID0ge2NvbHVtbjogeyBuYW1lOiAnYWdlJyB9LCB2YWx1ZTogOTk5IH07XHJcbnZhbGlkYXRlQ2VsbChjZWxsLCBydWxlU2V0KTtcclxuXHJcbmNvbnNvbGUubG9nKCdkZGQnKTsiXX0=

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');