molstar
Version:
A comprehensive macromolecular library.
76 lines (75 loc) • 2.89 kB
JavaScript
/**
* Copyright (c) 2019-2024 mol* contributors, licensed under MIT, See LICENSE file for more info.
*
* @author Alexander Rose <alexander.rose@weirdbyte.de>
* @author Adam Midlik <midlik@gmail.com>
*/
import { Interval, OrderedSet } from '../../mol-data/int';
import { Structure, StructureElement } from '../../mol-model/structure';
import { ColorNames } from '../../mol-util/color/names';
import { SequenceWrapper } from './wrapper';
export class ElementSequenceWrapper extends SequenceWrapper {
residueLabel(seqIdx) {
return 'X';
}
residueColor(seqIdx) {
return ColorNames.black;
}
residueClass(seqIdx) {
return 'msp-sequence-present';
}
getSeqIndices(loci) {
const { structure, units } = this.data;
if (StructureElement.Loci.is(loci)) {
if (!Structure.areRootsEquivalent(loci.structure, structure))
return Interval.Empty;
loci = StructureElement.Loci.remap(loci, structure);
for (const e of loci.elements) {
const indices = this.unitIndices.get(e.unit.id);
if (indices) {
if (OrderedSet.isSubset(indices, e.indices)) {
// this assumes this SequnceWrapper has only a single unit, otherwise we'd need to collect indices from all units and apply offsets
return e.indices;
}
}
}
}
else if (Structure.isLoci(loci)) {
if (!Structure.areRootsEquivalent(loci.structure, structure))
return Interval.Empty;
for (let i = 0, il = units.length; i < il; ++i) {
const indices = this.unitIndices.get(units[i].id);
return indices;
}
}
return Interval.Empty;
}
getLoci(seqIdx) {
const { units } = this.data;
const lociElements = [];
let offset = 0;
for (let i = 0, il = units.length; i < il; ++i) {
const unit = units[i];
if (seqIdx < offset + unit.elements.length) {
lociElements.push({ unit, indices: Interval.ofSingleton(seqIdx - offset) });
break;
}
offset += unit.elements.length;
}
return StructureElement.Loci(this.data.structure, lociElements);
}
constructor(data) {
let length = 0;
const unitIndices = new Map();
const lociElements = [];
for (let i = 0, il = data.units.length; i < il; ++i) {
const unit = data.units[i];
length += unit.elements.length;
const indices = Interval.ofBounds(0, unit.elements.length);
unitIndices.set(unit.id, indices);
lociElements.push({ unit, indices });
}
super(data, length);
this.unitIndices = unitIndices;
}
}