pdbe-molstar
Version:
Molstar implementation for PDBe
60 lines (59 loc) • 2.8 kB
JavaScript
"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.PDBeSIFTSMapping = void 0;
const int_1 = require("molstar/lib/mol-data/int");
const sifts_mapping_1 = require("molstar/lib/mol-model-props/sequence/themes/sifts-mapping");
const structure_1 = require("molstar/lib/mol-model/structure");
const behavior_1 = require("molstar/lib/mol-plugin/behavior");
const param_definition_1 = require("molstar/lib/mol-util/param-definition");
const sifts_mapping_2 = require("./sifts-mapping");
exports.PDBeSIFTSMapping = behavior_1.PluginBehavior.create({
name: 'pdbe-sifts-mapping-prop',
category: 'custom-props',
display: { name: 'PDBe SIFTS Mapping' },
ctor: class extends behavior_1.PluginBehavior.Handler {
constructor() {
super(...arguments);
this.provider = sifts_mapping_2.SIFTSMapping.Provider;
this.labelProvider = {
label: (loci) => {
if (!this.params.showTooltip)
return;
return PDBeBestDatabaseSequenceMappingLabel(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(sifts_mapping_1.SIFTSMappingColorThemeProvider);
this.ctx.managers.lociLabels.addProvider(this.labelProvider);
}
unregister() {
this.ctx.customModelProperties.unregister(this.provider.descriptor.name);
this.ctx.representation.structure.themes.colorThemeRegistry.remove(sifts_mapping_1.SIFTSMappingColorThemeProvider);
this.ctx.managers.lociLabels.removeProvider(this.labelProvider);
}
},
params: () => ({
autoAttach: param_definition_1.ParamDefinition.Boolean(true),
showTooltip: param_definition_1.ParamDefinition.Boolean(true),
}),
});
function PDBeBestDatabaseSequenceMappingLabel(loci) {
if (loci.kind === 'element-loci') {
if (loci.elements.length === 0)
return;
const e = loci.elements[0];
const u = e.unit;
const se = structure_1.StructureElement.Location.create(loci.structure, u, u.elements[int_1.OrderedSet.getAt(e.indices, 0)]);
return sifts_mapping_2.SIFTSMapping.getLabel(se);
}
}