TmMolStar/src/mol-plugin-ui/sequence/wrapper.ts

/**
 * Copyright (c) 2019 mol* contributors, licensed under MIT, See LICENSE file for more info.
 *
 * @author Alexander Rose <alexander.rose@weirdbyte.de>
 */

import { Interval } from '../../mol-data/int';
import { Loci, isEveryLoci } from '../../mol-model/loci';
import { MarkerAction, applyMarkerAction } from '../../mol-util/marker-action';
import { StructureElement, Structure, Unit } from '../../mol-model/structure';
import { Color } from '../../mol-util/color';

export type StructureUnit = { structure: Structure, units: Unit[] }

export { SequenceWrapper }

abstract class SequenceWrapper<D> {
    abstract residueLabel(seqIdx: number): string
    abstract residueColor(seqIdx: number): Color
    abstract residueClass(seqIdx: number): string

    abstract getLoci(seqIdx: number): StructureElement.Loci

    abstract mark(loci: Loci, action: MarkerAction): boolean;

    markResidue(loci: Loci, action: MarkerAction) {
        if (isEveryLoci(loci)) {
            return applyMarkerAction(this.markerArray, Interval.ofLength(this.length), action)
        } else {
            return this.mark(loci, action);
        }
    }

    constructor(readonly data: D, readonly markerArray: Uint8Array, readonly length: number) {

    }
}

namespace SequenceWrapper {
    export type Any = SequenceWrapper<any>
}