molstar
Version:
A comprehensive macromolecular library.
63 lines (62 loc) • 2.79 kB
JavaScript
/**
* Copyright (c) 2021 mol* contributors, licensed under MIT, See LICENSE file for more info.
*
* @author David Sehnal <david.sehnal@gmail.com>
*/
import { OrderedSet } from '../../../../../mol-data/int';
import { SIFTSMapping as BestDatabaseSequenceMappingProp } from '../../../../../mol-model-props/sequence/sifts-mapping';
import { SIFTSMappingColorThemeProvider } from '../../../../../mol-model-props/sequence/themes/sifts-mapping';
import { StructureElement } from '../../../../../mol-model/structure';
import { ParamDefinition as PD } from '../../../../../mol-util/param-definition';
import { PluginBehavior } from '../../../behavior';
export const SIFTSMapping = PluginBehavior.create({
name: 'sifts-mapping-prop',
category: 'custom-props',
display: { name: 'SIFTS Mapping' },
ctor: class extends PluginBehavior.Handler {
constructor() {
super(...arguments);
this.provider = BestDatabaseSequenceMappingProp.Provider;
this.labelProvider = {
label: (loci) => {
if (!this.params.showTooltip)
return;
return bestDatabaseSequenceMappingLabel(loci);
}
};
}
update(p) {
const updated = (this.params.autoAttach !== p.autoAttach ||
this.params.showTooltip !== p.showTooltip);
this.params.autoAttach = p.autoAttach;
this.params.showTooltip = p.showTooltip;
this.ctx.customStructureProperties.setDefaultAutoAttach(this.provider.descriptor.name, this.params.autoAttach);
return updated;
}
register() {
this.ctx.customModelProperties.register(this.provider, this.params.autoAttach);
this.ctx.representation.structure.themes.colorThemeRegistry.add(SIFTSMappingColorThemeProvider);
this.ctx.managers.lociLabels.addProvider(this.labelProvider);
}
unregister() {
this.ctx.customModelProperties.unregister(this.provider.descriptor.name);
this.ctx.representation.structure.themes.colorThemeRegistry.remove(SIFTSMappingColorThemeProvider);
this.ctx.managers.lociLabels.removeProvider(this.labelProvider);
}
},
params: () => ({
autoAttach: PD.Boolean(true),
showTooltip: PD.Boolean(true)
})
});
//
function bestDatabaseSequenceMappingLabel(loci) {
if (loci.kind === 'element-loci') {
if (loci.elements.length === 0)
return;
const e = loci.elements[0];
const u = e.unit;
const se = StructureElement.Location.create(loci.structure, u, u.elements[OrderedSet.getAt(e.indices, 0)]);
return BestDatabaseSequenceMappingProp.getLabel(se);
}
}