@datagrok-libraries/utils 0.0.4 → 0.0.5

package/src/reduce-dimensionality.ts

@@ -0,0 +1,272 @@
+import * as umj from 'umap-js';
+import {TSNE} from '@keckelt/tsne';
+
+import {Options, DistanceMetric, Coordinates, Vector, Vectors, Matrix} from './type_declarations';
+import {calcDistanceMatrix, transposeMatrix} from './operations';
+import {SPEBase, PSPEBase} from './spe';
+import {Measurer} from './string-measure';
+//import {AlignedSequenceEncoder} from './sequence-encoder';
+
+/**
+ * Abstract dimensionality reducer.
+ *
+ * @abstract
+ * @class Reducer
+ */
+abstract class Reducer {
+  protected data: Vectors;
+
+  /**
+   * Creates an instance of Reducer.
+   * @param {Options} options Options to pass to the constructor.
+   * @memberof Reducer
+   */
+  constructor(options: Options) {
+    this.data = options.data;
+  }
+
+  /**
+   * Is to embed the data given into the two-dimensional space.
+   *
+   * @abstract
+   * @return {Coordinates} Cartesian coordinate of this embedding.
+   * @memberof Reducer
+   */
+  abstract transform(): Coordinates;
+}
+
+/**
+ * Implements t-SNE dimensionality reduction.
+ *
+ * @class TSNEReducer
+ * @extends {Reducer}
+ */
+class TSNEReducer extends Reducer {
+  protected reducer: TSNE;
+  protected iterations: number;
+  protected distance: DistanceMetric;
+
+  /**
+   * Creates an instance of TSNEReducer.
+   * @param {Options} options Options to pass to the constructor.
+   * @memberof TSNEReducer
+   */
+  constructor(options: Options) {
+    super(options);
+    this.reducer = new TSNE(options);
+    this.iterations = options?.iterations ?? 100;
+    this.distance = options.distance;
+  }
+
+  /**
+   * Embeds the data given into the two-dimensional space using t-SNE method.
+   * @return {Coordinates} Cartesian coordinate of this embedding.
+   */
+  public transform(): Coordinates {
+    this.reducer.initDataDist(calcDistanceMatrix(this.data, this.distance));
+
+    for (let i = 0; i < this.iterations; ++i) {
+      this.reducer.step(); // every time you call this, solution gets better
+    }
+    return this.reducer.getSolution();
+  }
+}
+
+/**
+ * Implements UMAP dimensionality reduction.
+ *
+ * @class UMAPReducer
+ * @extends {Reducer}
+ */
+class UMAPReducer extends Reducer {
+  protected reducer: umj.UMAP;
+/*  protected encoder: AlignedSequenceEncoder;
+  protected distanceMatrix: Matrix;*/
+
+  /**
+   * Creates an instance of UMAPReducer.
+   * @param {Options} options Options to pass to the constructor.
+   * @memberof UMAPReducer
+   */
+  constructor(options: Options) {
+    super(options);
+    this.reducer = new umj.UMAP(options);
+/*    this.encoder = new AlignedSequenceEncoder();
+    this.distanceMatrix = calcDistanceMatrix(this.data, options.distanceFn);
+    this.distanceFn = options.distanceFn;*/
+  }
+
+/*  protected _encodedDistance(a: number[], b: number[]): number {
+    return this.distanceMatrix[a[a.length-1]][b[b.length-1]];
+  }
+
+  protected _decodedDistance(a: number[], b: number[]): number {
+    return this.encoder.decode(a);
+  }
+
+  protected encode(): number[][] {
+    let vectors: number[][] = [];
+
+    for (let i = 0; i < this.data.length; ++i) {
+      let encoded = this.encoder.encode(this.data[i]);
+      vectors.push(encoded);
+    }
+    return vectors;
+  }
+
+  protected _calcDistance() {
+    const nItems = this.data.length;
+    const vectors = this.encode();
+    
+  }*/
+
+  /**
+   * Embeds the data given into the two-dimensional space using UMAP method.
+   * @return {Coordinates} Cartesian coordinate of this embedding.
+   */
+  public transform(): Coordinates {
+    const embedding = this.reducer.fit(this.data);
+
+    function arrayCast2Coordinates(data: number[][]): Coordinates {
+      return new Array(data.length).fill(0).map((_, i) => (Vector.from(data[i])));
+    }
+    return arrayCast2Coordinates(embedding);
+  }
+}
+
+/**
+ * Implements original SPE dimensionality reduction.
+ *
+ * @class SPEReducer
+ * @extends {Reducer}
+ */
+class SPEReducer extends Reducer {
+  protected reducer: SPEBase;
+
+  /**
+   * Creates an instance of SPEReducer.
+   * @param {Options} options Options to pass to the constructor.
+   * @memberof SPEReducer
+   */
+  constructor(options: Options) {
+    super(options);
+    this.reducer = new SPEBase(options);
+  }
+
+  /**
+   * Embeds the data given into the two-dimensional space using the original SPE method.
+   * @return {Coordinates} Cartesian coordinate of this embedding.
+   */
+  public transform(): Coordinates {
+    return this.reducer.embed(this.data);
+  }
+}
+
+/**
+ * Implements modified SPE dimensionality reduction.
+ *
+ * @class PSPEReducer
+ * @extends {Reducer}
+ */
+class PSPEReducer extends Reducer {
+  protected reducer: PSPEBase;
+
+  /**
+   * Creates an instance of PSPEReducer.
+   * @param {Options} options Options to pass to the constructor.
+   * @memberof PSPEReducer
+   */
+  constructor(options: Options) {
+    super(options);
+    this.reducer = new PSPEBase(options);
+  }
+
+  /**
+   * Embeds the data given into the two-dimensional space using the modified SPE method.
+   * @return {Coordinates} Cartesian coordinate of this embedding.
+   */
+  public transform(): Coordinates {
+    return this.reducer.embed(this.data);
+  }
+}
+
+/**
+ * Unified class implementing different dimensionality reduction methods.
+ *
+ * @export
+ * @class DimensionalityReducer
+ */
+export class DimensionalityReducer {
+  private reducer: Reducer | undefined;
+  private methods: string[];
+  private measurer: Measurer;
+
+  /**
+   * Creates an instance of DimensionalityReducer.
+   * @param {any[]} data Vectors to embed.
+   * @param {string} method Embedding method to be applied
+   * @param {string} metric Distance metric to be computed between each of the vectors.
+   * @param {Options} [options] Options to pass to the implementing embedders.
+   * @memberof DimensionalityReducer
+   */
+  constructor(data: any[], method: string, metric: string, options?: Options) {
+    this.methods = ['UMAP', 'TSNE', 'SPE', 'PSPE'];
+
+    if (!this.availableMethods.includes(method)) {
+      throw new Error('The method "'+method+'" is not supported');
+    }
+
+    this.measurer = new Measurer(metric);
+    const measure = this.measurer.getMeasure()
+
+    if (method == 'UMAP') {
+      this.reducer = new UMAPReducer({
+        ...{data: data}, 
+        ...{distanceFn: measure}, 
+        ...{nEpochs: options?.cycles},
+        ...options
+      });
+    } else if (method == 'TSNE') {
+      this.reducer = new TSNEReducer({
+        ...{data: data},
+        ...{distance: measure},
+        ...{iterations: options?.cycles ?? undefined},
+        ...options,
+      });
+    } else if (method == 'SPE') {
+      this.reducer = new SPEReducer({...{data: data}, ...{distance: measure}, ...options});
+    } else {
+      this.reducer = new PSPEReducer({...{data: data}, ...{distance: measure}, ...options});
+    }
+  }
+
+  /**
+   * Embeds the data given into the two-dimensional space using the chosen method.
+   *
+   * @param {boolean} transpose Whether to transform coordinates to have columns-first orientation.
+   * @throws {Error} If the embedding method was not found.
+   * @return {Coordinates} Cartesian coordinate of this embedding.
+   * @memberof DimensionalityReducer
+   */
+  public transform(transpose: boolean = false): Coordinates {
+    if (this.reducer == undefined) {
+      throw new Error('Reducer was not defined.');
+    }
+    let embedding = this.reducer.transform();
+
+    if (transpose) {
+      embedding = transposeMatrix(embedding);
+    }
+    return embedding;
+  }
+
+  /**
+   * Returns dimensionality reduction methods available.
+   *
+   * @readonly
+   * @memberof DimensionalityReducer
+   */
+  get availableMethods() {
+    return this.methods;
+  }
+}

package/src/sequence-encoder.js

@@ -0,0 +1,116 @@
+import { assert } from './operations';
+/**
+ * Class to categorial encode/decode aligned amino acid residues sequence.
+ *
+ * @export
+ * @class AlignedSequenceEncoder
+ */
+export class AlignedSequenceEncoder {
+    constructor() {
+        this.aa2num = {
+            '-': 0,
+            'A': 1,
+            'C': 2,
+            'D': 3,
+            'E': 4,
+            'F': 5,
+            'G': 6,
+            'H': 7,
+            'I': 8,
+            'K': 9,
+            'L': 10,
+            'M': 11,
+            'N': 12,
+            'P': 13,
+            'Q': 14,
+            'R': 15,
+            'S': 16,
+            'T': 17,
+            'V': 18,
+            'W': 19,
+            'Y': 20,
+        };
+        this.num2aa = {};
+        Object.entries(this.aa2num).forEach(([k, v]) => (this.num2aa[v] = k));
+    }
+    /**
+     * Truncate NH2 and -COOH terminals of the given sequence.
+     *
+     * @protected
+     * @param {string} seq The sequence provided.
+     * @return {string} Truncated sequence.
+     * @memberof AlignedSequenceEncoder
+     */
+    _truncateSequence(seq) {
+        let start = 0;
+        let end = seq.length;
+        const termina = ['NH2', 'COOH'];
+        if (seq.startsWith(termina[0])) {
+            const l = termina[0].length; // Cut only 'NH2' without following '-'.
+            assert(seq[l] == '-', `Wrong sequence format: ${termina[0]} without following '-' in '${seq}'.`);
+            start = l;
+        }
+        if (seq.endsWith(termina[1])) {
+            const l = termina[1].length + 1; // Cut both 'COOH' and precending '-'.
+            assert(seq[end - l] == '-', `Wrong sequence format: ${termina[1]} without '-' precending in '${seq}'.`);
+            end -= l;
+        }
+        return seq.substring(start, end);
+    }
+    /**
+     * Cuts auxiliary defises before a residue.
+     *
+     * @protected
+     * @param {string} seq The sequence to process.
+     * @return {string} Processed sequence.
+     * @memberof AlignedSequenceEncoder
+     */
+    _dropDefises(seq) {
+        return seq.replace(/(-)([^-]+)/g, '$2');
+    }
+    /**
+     * Performs truncation and cutting auxiliary defises.
+     *
+     * @param {string} sequence The sequence work under process.
+     * @return {string} Result of cleaning.
+     * @memberof AlignedSequenceEncoder
+     */
+    clean(sequence) {
+        return this._dropDefises(this._truncateSequence(sequence));
+    }
+    /**
+     * Categorial encode of the sequence provided.
+     *
+     * @param {string} sequence The sequence.
+     * @return {number[]} Encoded vector.
+     * @memberof AlignedSequenceEncoder
+     */
+    encode(sequence) {
+        const seq = this.clean(sequence);
+        const nItems = seq.length;
+        let values = new Array(nItems).fill(0);
+        for (let i = 0; i < nItems; ++i) {
+            let char = seq[i];
+            assert(char in this.aa2num, `Unknown char '${char}' found in sequence '${seq}'`);
+            values[i] = this.aa2num[char];
+        }
+        return values;
+    }
+    /**
+     * Decode the encoded vector into the sequence back.
+     *
+     * @param {number[]} value The vector encoded.
+     * @return {string} Decoded sequence.
+     * @memberof AlignedSequenceEncoder
+     */
+    decode(value) {
+        let s = '';
+        for (let i = 0; i < value.length; ++i) {
+            let code = value[i];
+            assert(code in this.num2aa, `Unknown code '${code}' found in vector '${value}'`);
+            s += this.num2aa[code];
+        }
+        return s;
+    }
+}
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoic2VxdWVuY2UtZW5jb2Rlci5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbInNlcXVlbmNlLWVuY29kZXIudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6IkFBQUEsT0FBTyxFQUFDLE1BQU0sRUFBQyxNQUFNLGNBQWMsQ0FBQztBQUVwQzs7Ozs7R0FLRztBQUNILE1BQU0sT0FBTyxzQkFBc0I7SUFJL0I7UUFDSSxJQUFJLENBQUMsTUFBTSxHQUFHO1lBQ1YsR0FBRyxFQUFFLENBQUM7WUFDTixHQUFHLEVBQUUsQ0FBQztZQUNOLEdBQUcsRUFBRSxDQUFDO1lBQ04sR0FBRyxFQUFFLENBQUM7WUFDTixHQUFHLEVBQUUsQ0FBQztZQUNOLEdBQUcsRUFBRSxDQUFDO1lBQ04sR0FBRyxFQUFFLENBQUM7WUFDTixHQUFHLEVBQUUsQ0FBQztZQUNOLEdBQUcsRUFBRSxDQUFDO1lBQ04sR0FBRyxFQUFFLENBQUM7WUFDTixHQUFHLEVBQUUsRUFBRTtZQUNQLEdBQUcsRUFBRSxFQUFFO1lBQ1AsR0FBRyxFQUFFLEVBQUU7WUFDUCxHQUFHLEVBQUUsRUFBRTtZQUNQLEdBQUcsRUFBRSxFQUFFO1lBQ1AsR0FBRyxFQUFFLEVBQUU7WUFDUCxHQUFHLEVBQUUsRUFBRTtZQUNQLEdBQUcsRUFBRSxFQUFFO1lBQ1AsR0FBRyxFQUFFLEVBQUU7WUFDUCxHQUFHLEVBQUUsRUFBRTtZQUNQLEdBQUcsRUFBRSxFQUFFO1NBQ1YsQ0FBQztRQUNGLElBQUksQ0FBQyxNQUFNLEdBQUcsRUFBRSxDQUFDO1FBQ2pCLE1BQU0sQ0FBQyxPQUFPLENBQUMsSUFBSSxDQUFDLE1BQU0sQ0FBQyxDQUFDLE9BQU8sQ0FBQyxDQUFDLENBQUMsQ0FBQyxFQUFFLENBQUMsQ0FBQyxFQUFFLEVBQUUsQ0FBQyxDQUFDLElBQUksQ0FBQyxNQUFNLENBQUMsQ0FBQyxDQUFDLEdBQUcsQ0FBQyxDQUFDLENBQUMsQ0FBQztJQUMxRSxDQUFDO0lBRUQ7Ozs7Ozs7T0FPRztJQUNPLGlCQUFpQixDQUFDLEdBQVc7UUFDbkMsSUFBSSxLQUFLLEdBQUcsQ0FBQyxDQUFDO1FBQ2QsSUFBSSxHQUFHLEdBQUcsR0FBRyxDQUFDLE1BQU0sQ0FBQztRQUNyQixNQUFNLE9BQU8sR0FBRyxDQUFDLEtBQUssRUFBRSxNQUFNLENBQUMsQ0FBQztRQUVoQyxJQUFJLEdBQUcsQ0FBQyxVQUFVLENBQUMsT0FBTyxDQUFDLENBQUMsQ0FBQyxDQUFDLEVBQUU7WUFDNUIsTUFBTSxDQUFDLEdBQUcsT0FBTyxDQUFDLENBQUMsQ0FBQyxDQUFDLE1BQU0sQ0FBQyxDQUFDLHdDQUF3QztZQUNyRSxNQUFNLENBQUMsR0FBRyxDQUFDLENBQUMsQ0FBQyxJQUFJLEdBQUcsRUFBRSwwQkFBMEIsT0FBTyxDQUFDLENBQUMsQ0FBQyw4QkFBOEIsR0FBRyxJQUFJLENBQUMsQ0FBQTtZQUNoRyxLQUFLLEdBQUcsQ0FBQyxDQUFDO1NBQ2I7UUFDRCxJQUFJLEdBQUcsQ0FBQyxRQUFRLENBQUMsT0FBTyxDQUFDLENBQUMsQ0FBQyxDQUFDLEVBQUU7WUFDMUIsTUFBTSxDQUFDLEdBQUcsT0FBTyxDQUFDLENBQUMsQ0FBQyxDQUFDLE1BQU0sR0FBQyxDQUFDLENBQUMsQ0FBQyxzQ0FBc0M7WUFDckUsTUFBTSxDQUFDLEdBQUcsQ0FBQyxHQUFHLEdBQUMsQ0FBQyxDQUFDLElBQUksR0FBRyxFQUFFLDBCQUEwQixPQUFPLENBQUMsQ0FBQyxDQUFDLCtCQUErQixHQUFHLElBQUksQ0FBQyxDQUFBO1lBQ3JHLEdBQUcsSUFBSSxDQUFDLENBQUM7U0FDWjtRQUNELE9BQU8sR0FBRyxDQUFDLFNBQVMsQ0FBQyxLQUFLLEVBQUUsR0FBRyxDQUFDLENBQUM7SUFDckMsQ0FBQztJQUVEOzs7Ozs7O09BT0c7SUFDTyxZQUFZLENBQUMsR0FBVztRQUM5QixPQUFPLEdBQUcsQ0FBQyxPQUFPLENBQUMsYUFBYSxFQUFFLElBQUksQ0FBQyxDQUFBO0lBQzNDLENBQUM7SUFFRDs7Ozs7O09BTUc7SUFDSSxLQUFLLENBQUMsUUFBZ0I7UUFDekIsT0FBTyxJQUFJLENBQUMsWUFBWSxDQUFDLElBQUksQ0FBQyxpQkFBaUIsQ0FBQyxRQUFRLENBQUMsQ0FBQyxDQUFDO0lBQy9ELENBQUM7SUFFRDs7Ozs7O09BTUc7SUFDSSxNQUFNLENBQUMsUUFBZ0I7UUFDMUIsTUFBTSxHQUFHLEdBQUcsSUFBSSxDQUFDLEtBQUssQ0FBQyxRQUFRLENBQUMsQ0FBQztRQUNqQyxNQUFNLE1BQU0sR0FBRyxHQUFHLENBQUMsTUFBTSxDQUFDO1FBQzFCLElBQUksTUFBTSxHQUFHLElBQUksS0FBSyxDQUFDLE1BQU0sQ0FBQyxDQUFDLElBQUksQ0FBQyxDQUFDLENBQUMsQ0FBQztRQUV2QyxLQUFLLElBQUksQ0FBQyxHQUFHLENBQUMsRUFBRSxDQUFDLEdBQUcsTUFBTSxFQUFFLEVBQUUsQ0FBQyxFQUFFO1lBQzdCLElBQUksSUFBSSxHQUFHLEdBQUcsQ0FBQyxDQUFDLENBQUMsQ0FBQztZQUVsQixNQUFNLENBQUMsSUFBSSxJQUFJLElBQUksQ0FBQyxNQUFNLEVBQUUsaUJBQWlCLElBQUksd0JBQXdCLEdBQUcsR0FBRyxDQUFDLENBQUM7WUFFakYsTUFBTSxDQUFDLENBQUMsQ0FBQyxHQUFHLElBQUksQ0FBQyxNQUFNLENBQUMsSUFBSSxDQUFDLENBQUM7U0FDakM7UUFDRCxPQUFPLE1BQU0sQ0FBQztJQUNsQixDQUFDO0lBRUQ7Ozs7OztPQU1HO0lBQ0ksTUFBTSxDQUFDLEtBQWU7UUFDekIsSUFBSSxDQUFDLEdBQVcsRUFBRSxDQUFDO1FBRW5CLEtBQUssSUFBSSxDQUFDLEdBQUcsQ0FBQyxFQUFFLENBQUMsR0FBRyxLQUFLLENBQUMsTUFBTSxFQUFFLEVBQUUsQ0FBQyxFQUFFO1lBQ25DLElBQUksSUFBSSxHQUFHLEtBQUssQ0FBQyxDQUFDLENBQUMsQ0FBQztZQUVwQixNQUFNLENBQUMsSUFBSSxJQUFJLElBQUksQ0FBQyxNQUFNLEVBQUUsaUJBQWlCLElBQUksc0JBQXNCLEtBQUssR0FBRyxDQUFDLENBQUM7WUFFakYsQ0FBQyxJQUFJLElBQUksQ0FBQyxNQUFNLENBQUMsSUFBSSxDQUFDLENBQUM7U0FDMUI7UUFDRCxPQUFPLENBQUMsQ0FBQztJQUNiLENBQUM7Q0FDSiIsInNvdXJjZXNDb250ZW50IjpbImltcG9ydCB7YXNzZXJ0fSBmcm9tICcuL29wZXJhdGlvbnMnO1xyXG5cclxuLyoqXHJcbiAqIENsYXNzIHRvIGNhdGVnb3JpYWwgZW5jb2RlL2RlY29kZSBhbGlnbmVkIGFtaW5vIGFjaWQgcmVzaWR1ZXMgc2VxdWVuY2UuXHJcbiAqXHJcbiAqIEBleHBvcnRcclxuICogQGNsYXNzIEFsaWduZWRTZXF1ZW5jZUVuY29kZXJcclxuICovXHJcbmV4cG9ydCBjbGFzcyBBbGlnbmVkU2VxdWVuY2VFbmNvZGVyIHtcclxuICAgIHByb3RlY3RlZCBhYTJudW06IHtbbmFtZTogc3RyaW5nXTogbnVtYmVyfTtcclxuICAgIHByb3RlY3RlZCBudW0yYWE6IHtbY29kZTogbnVtYmVyXTogc3RyaW5nfTtcclxuXHJcbiAgICBjb25zdHJ1Y3RvciAoKSB7XHJcbiAgICAgICAgdGhpcy5hYTJudW0gPSB7XHJcbiAgICAgICAgICAgICctJzogMCxcclxuICAgICAgICAgICAgJ0EnOiAxLFxyXG4gICAgICAgICAgICAnQyc6IDIsXHJcbiAgICAgICAgICAgICdEJzogMyxcclxuICAgICAgICAgICAgJ0UnOiA0LFxyXG4gICAgICAgICAgICAnRic6IDUsXHJcbiAgICAgICAgICAgICdHJzogNixcclxuICAgICAgICAgICAgJ0gnOiA3LFxyXG4gICAgICAgICAgICAnSSc6IDgsXHJcbiAgICAgICAgICAgICdLJzogOSxcclxuICAgICAgICAgICAgJ0wnOiAxMCxcclxuICAgICAgICAgICAgJ00nOiAxMSxcclxuICAgICAgICAgICAgJ04nOiAxMixcclxuICAgICAgICAgICAgJ1AnOiAxMyxcclxuICAgICAgICAgICAgJ1EnOiAxNCxcclxuICAgICAgICAgICAgJ1InOiAxNSxcclxuICAgICAgICAgICAgJ1MnOiAxNixcclxuICAgICAgICAgICAgJ1QnOiAxNyxcclxuICAgICAgICAgICAgJ1YnOiAxOCxcclxuICAgICAgICAgICAgJ1cnOiAxOSxcclxuICAgICAgICAgICAgJ1knOiAyMCxcclxuICAgICAgICB9O1xyXG4gICAgICAgIHRoaXMubnVtMmFhID0ge307XHJcbiAgICAgICAgT2JqZWN0LmVudHJpZXModGhpcy5hYTJudW0pLmZvckVhY2goKFtrLCB2XSkgPT4gKHRoaXMubnVtMmFhW3ZdID0gaykpO1xyXG4gICAgfVxyXG5cclxuICAgIC8qKlxyXG4gICAgICogVHJ1bmNhdGUgTkgyIGFuZCAtQ09PSCB0ZXJtaW5hbHMgb2YgdGhlIGdpdmVuIHNlcXVlbmNlLlxyXG4gICAgICpcclxuICAgICAqIEBwcm90ZWN0ZWRcclxuICAgICAqIEBwYXJhbSB7c3RyaW5nfSBzZXEgVGhlIHNlcXVlbmNlIHByb3ZpZGVkLlxyXG4gICAgICogQHJldHVybiB7c3RyaW5nfSBUcnVuY2F0ZWQgc2VxdWVuY2UuXHJcbiAgICAgKiBAbWVtYmVyb2YgQWxpZ25lZFNlcXVlbmNlRW5jb2RlclxyXG4gICAgICovXHJcbiAgICBwcm90ZWN0ZWQgX3RydW5jYXRlU2VxdWVuY2Uoc2VxOiBzdHJpbmcpOiBzdHJpbmcge1xyXG4gICAgICAgIGxldCBzdGFydCA9IDA7XHJcbiAgICAgICAgbGV0IGVuZCA9IHNlcS5sZW5ndGg7XHJcbiAgICAgICAgY29uc3QgdGVybWluYSA9IFsnTkgyJywgJ0NPT0gnXTtcclxuICAgIFxyXG4gICAgICAgIGlmIChzZXEuc3RhcnRzV2l0aCh0ZXJtaW5hWzBdKSkge1xyXG4gICAgICAgICAgICBjb25zdCBsID0gdGVybWluYVswXS5sZW5ndGg7IC8vIEN1dCBvbmx5ICdOSDInIHdpdGhvdXQgZm9sbG93aW5nICctJy5cclxuICAgICAgICAgICAgYXNzZXJ0KHNlcVtsXSA9PSAnLScsIGBXcm9uZyBzZXF1ZW5jZSBmb3JtYXQ6ICR7dGVybWluYVswXX0gd2l0aG91dCBmb2xsb3dpbmcgJy0nIGluICcke3NlcX0nLmApXHJcbiAgICAgICAgICAgIHN0YXJ0ID0gbDtcclxuICAgICAgICB9XHJcbiAgICAgICAgaWYgKHNlcS5lbmRzV2l0aCh0ZXJtaW5hWzFdKSkge1xyXG4gICAgICAgICAgICBjb25zdCBsID0gdGVybWluYVsxXS5sZW5ndGgrMTsgLy8gQ3V0IGJvdGggJ0NPT0gnIGFuZCBwcmVjZW5kaW5nICctJy5cclxuICAgICAgICAgICAgYXNzZXJ0KHNlcVtlbmQtbF0gPT0gJy0nLCBgV3Jvbmcgc2VxdWVuY2UgZm9ybWF0OiAke3Rlcm1pbmFbMV19IHdpdGhvdXQgJy0nIHByZWNlbmRpbmcgaW4gJyR7c2VxfScuYClcclxuICAgICAgICAgICAgZW5kIC09IGw7XHJcbiAgICAgICAgfVxyXG4gICAgICAgIHJldHVybiBzZXEuc3Vic3RyaW5nKHN0YXJ0LCBlbmQpO1xyXG4gICAgfVxyXG5cclxuICAgIC8qKlxyXG4gICAgICogQ3V0cyBhdXhpbGlhcnkgZGVmaXNlcyBiZWZvcmUgYSByZXNpZHVlLlxyXG4gICAgICpcclxuICAgICAqIEBwcm90ZWN0ZWRcclxuICAgICAqIEBwYXJhbSB7c3RyaW5nfSBzZXEgVGhlIHNlcXVlbmNlIHRvIHByb2Nlc3MuXHJcbiAgICAgKiBAcmV0dXJuIHtzdHJpbmd9IFByb2Nlc3NlZCBzZXF1ZW5jZS5cclxuICAgICAqIEBtZW1iZXJvZiBBbGlnbmVkU2VxdWVuY2VFbmNvZGVyXHJcbiAgICAgKi9cclxuICAgIHByb3RlY3RlZCBfZHJvcERlZmlzZXMoc2VxOiBzdHJpbmcpOiBzdHJpbmcge1xyXG4gICAgICAgIHJldHVybiBzZXEucmVwbGFjZSgvKC0pKFteLV0rKS9nLCAnJDInKVxyXG4gICAgfVxyXG5cclxuICAgIC8qKlxyXG4gICAgICogUGVyZm9ybXMgdHJ1bmNhdGlvbiBhbmQgY3V0dGluZyBhdXhpbGlhcnkgZGVmaXNlcy5cclxuICAgICAqXHJcbiAgICAgKiBAcGFyYW0ge3N0cmluZ30gc2VxdWVuY2UgVGhlIHNlcXVlbmNlIHdvcmsgdW5kZXIgcHJvY2Vzcy5cclxuICAgICAqIEByZXR1cm4ge3N0cmluZ30gUmVzdWx0IG9mIGNsZWFuaW5nLlxyXG4gICAgICogQG1lbWJlcm9mIEFsaWduZWRTZXF1ZW5jZUVuY29kZXJcclxuICAgICAqL1xyXG4gICAgcHVibGljIGNsZWFuKHNlcXVlbmNlOiBzdHJpbmcpOiBzdHJpbmcge1xyXG4gICAgICAgIHJldHVybiB0aGlzLl9kcm9wRGVmaXNlcyh0aGlzLl90cnVuY2F0ZVNlcXVlbmNlKHNlcXVlbmNlKSk7XHJcbiAgICB9XHJcblxyXG4gICAgLyoqXHJcbiAgICAgKiBDYXRlZ29yaWFsIGVuY29kZSBvZiB0aGUgc2VxdWVuY2UgcHJvdmlkZWQuXHJcbiAgICAgKlxyXG4gICAgICogQHBhcmFtIHtzdHJpbmd9IHNlcXVlbmNlIFRoZSBzZXF1ZW5jZS5cclxuICAgICAqIEByZXR1cm4ge251bWJlcltdfSBFbmNvZGVkIHZlY3Rvci5cclxuICAgICAqIEBtZW1iZXJvZiBBbGlnbmVkU2VxdWVuY2VFbmNvZGVyXHJcbiAgICAgKi9cclxuICAgIHB1YmxpYyBlbmNvZGUoc2VxdWVuY2U6IHN0cmluZyk6IG51bWJlcltdIHtcclxuICAgICAgICBjb25zdCBzZXEgPSB0aGlzLmNsZWFuKHNlcXVlbmNlKTtcclxuICAgICAgICBjb25zdCBuSXRlbXMgPSBzZXEubGVuZ3RoO1xyXG4gICAgICAgIGxldCB2YWx1ZXMgPSBuZXcgQXJyYXkobkl0ZW1zKS5maWxsKDApO1xyXG4gICAgXHJcbiAgICAgICAgZm9yIChsZXQgaSA9IDA7IGkgPCBuSXRlbXM7ICsraSkge1xyXG4gICAgICAgICAgICBsZXQgY2hhciA9IHNlcVtpXTtcclxuICAgIFxyXG4gICAgICAgICAgICBhc3NlcnQoY2hhciBpbiB0aGlzLmFhMm51bSwgYFVua25vd24gY2hhciAnJHtjaGFyfScgZm91bmQgaW4gc2VxdWVuY2UgJyR7c2VxfSdgKTtcclxuICAgIFxyXG4gICAgICAgICAgICB2YWx1ZXNbaV0gPSB0aGlzLmFhMm51bVtjaGFyXTtcclxuICAgICAgICB9XHJcbiAgICAgICAgcmV0dXJuIHZhbHVlcztcclxuICAgIH1cclxuXHJcbiAgICAvKipcclxuICAgICAqIERlY29kZSB0aGUgZW5jb2RlZCB2ZWN0b3IgaW50byB0aGUgc2VxdWVuY2UgYmFjay5cclxuICAgICAqXHJcbiAgICAgKiBAcGFyYW0ge251bWJlcltdfSB2YWx1ZSBUaGUgdmVjdG9yIGVuY29kZWQuXHJcbiAgICAgKiBAcmV0dXJuIHtzdHJpbmd9IERlY29kZWQgc2VxdWVuY2UuXHJcbiAgICAgKiBAbWVtYmVyb2YgQWxpZ25lZFNlcXVlbmNlRW5jb2RlclxyXG4gICAgICovXHJcbiAgICBwdWJsaWMgZGVjb2RlKHZhbHVlOiBudW1iZXJbXSk6IHN0cmluZyB7XHJcbiAgICAgICAgbGV0IHM6IHN0cmluZyA9ICcnO1xyXG4gICAgXHJcbiAgICAgICAgZm9yIChsZXQgaSA9IDA7IGkgPCB2YWx1ZS5sZW5ndGg7ICsraSkge1xyXG4gICAgICAgICAgICBsZXQgY29kZSA9IHZhbHVlW2ldO1xyXG5cclxuICAgICAgICAgICAgYXNzZXJ0KGNvZGUgaW4gdGhpcy5udW0yYWEsIGBVbmtub3duIGNvZGUgJyR7Y29kZX0nIGZvdW5kIGluIHZlY3RvciAnJHt2YWx1ZX0nYCk7XHJcblxyXG4gICAgICAgICAgICBzICs9IHRoaXMubnVtMmFhW2NvZGVdO1xyXG4gICAgICAgIH1cclxuICAgICAgICByZXR1cm4gcztcclxuICAgIH1cclxufVxyXG4iXX0=

package/src/sequence-encoder.ts

@@ -0,0 +1,131 @@
+import {assert} from './operations';
+
+/**
+ * Class to categorial encode/decode aligned amino acid residues sequence.
+ *
+ * @export
+ * @class AlignedSequenceEncoder
+ */
+export class AlignedSequenceEncoder {
+    protected aa2num: {[name: string]: number};
+    protected num2aa: {[code: number]: string};
+
+    constructor () {
+        this.aa2num = {
+            '-': 0,
+            'A': 1,
+            'C': 2,
+            'D': 3,
+            'E': 4,
+            'F': 5,
+            'G': 6,
+            'H': 7,
+            'I': 8,
+            'K': 9,
+            'L': 10,
+            'M': 11,
+            'N': 12,
+            'P': 13,
+            'Q': 14,
+            'R': 15,
+            'S': 16,
+            'T': 17,
+            'V': 18,
+            'W': 19,
+            'Y': 20,
+        };
+        this.num2aa = {};
+        Object.entries(this.aa2num).forEach(([k, v]) => (this.num2aa[v] = k));
+    }
+
+    /**
+     * Truncate NH2 and -COOH terminals of the given sequence.
+     *
+     * @protected
+     * @param {string} seq The sequence provided.
+     * @return {string} Truncated sequence.
+     * @memberof AlignedSequenceEncoder
+     */
+    protected _truncateSequence(seq: string): string {
+        let start = 0;
+        let end = seq.length;
+        const termina = ['NH2', 'COOH'];
+    
+        if (seq.startsWith(termina[0])) {
+            const l = termina[0].length; // Cut only 'NH2' without following '-'.
+            assert(seq[l] == '-', `Wrong sequence format: ${termina[0]} without following '-' in '${seq}'.`)
+            start = l;
+        }
+        if (seq.endsWith(termina[1])) {
+            const l = termina[1].length+1; // Cut both 'COOH' and precending '-'.
+            assert(seq[end-l] == '-', `Wrong sequence format: ${termina[1]} without '-' precending in '${seq}'.`)
+            end -= l;
+        }
+        return seq.substring(start, end);
+    }
+
+    /**
+     * Cuts auxiliary defises before a residue.
+     *
+     * @protected
+     * @param {string} seq The sequence to process.
+     * @return {string} Processed sequence.
+     * @memberof AlignedSequenceEncoder
+     */
+    protected _dropDefises(seq: string): string {
+        return seq.replace(/(-)([^-]+)/g, '$2')
+    }
+
+    /**
+     * Performs truncation and cutting auxiliary defises.
+     *
+     * @param {string} sequence The sequence work under process.
+     * @return {string} Result of cleaning.
+     * @memberof AlignedSequenceEncoder
+     */
+    public clean(sequence: string): string {
+        return this._dropDefises(this._truncateSequence(sequence));
+    }
+
+    /**
+     * Categorial encode of the sequence provided.
+     *
+     * @param {string} sequence The sequence.
+     * @return {number[]} Encoded vector.
+     * @memberof AlignedSequenceEncoder
+     */
+    public encode(sequence: string): number[] {
+        const seq = this.clean(sequence);
+        const nItems = seq.length;
+        let values = new Array(nItems).fill(0);
+    
+        for (let i = 0; i < nItems; ++i) {
+            let char = seq[i];
+    
+            assert(char in this.aa2num, `Unknown char '${char}' found in sequence '${seq}'`);
+    
+            values[i] = this.aa2num[char];
+        }
+        return values;
+    }
+
+    /**
+     * Decode the encoded vector into the sequence back.
+     *
+     * @param {number[]} value The vector encoded.
+     * @return {string} Decoded sequence.
+     * @memberof AlignedSequenceEncoder
+     */
+    public decode(value: number[]): string {
+        let s: string = '';
+    
+        for (let i = 0; i < value.length; ++i) {
+            let code = value[i];
+
+            assert(code in this.num2aa, `Unknown code '${code}' found in vector '${value}'`);
+
+            s += this.num2aa[code];
+        }
+        return s;
+    }
+}