molstar
Version:
A comprehensive macromolecular library.
201 lines (200 loc) • 8.06 kB
JavaScript
;
/**
* Copyright (c) 2018-2026 mol* contributors, licensed under MIT, See LICENSE file for more info.
*
* @author Alexander Rose <alexander.rose@weirdbyte.de>
*/
Object.defineProperty(exports, "__esModule", { value: true });
exports.VolumeParams = void 0;
exports.VolumeRepresentationProvider = VolumeRepresentationProvider;
exports.VolumeRepresentation = VolumeRepresentation;
const param_definition_1 = require("../../mol-util/param-definition.js");
const theme_1 = require("../../mol-theme/theme.js");
const render_object_1 = require("../../mol-gl/render-object.js");
const loci_1 = require("../../mol-model/loci.js");
const util_1 = require("../util.js");
const linear_algebra_1 = require("../../mol-math/linear-algebra.js");
const representation_1 = require("../representation.js");
const base_1 = require("../../mol-geo/geometry/base.js");
const rxjs_1 = require("rxjs");
const mol_task_1 = require("../../mol-task/index.js");
function VolumeRepresentationProvider(p) { return p; }
//
exports.VolumeParams = {
...base_1.BaseGeometry.Params,
};
function VolumeRepresentation(label, ctx, getParams, visualCtor, getLoci, getKeys = () => [-1]) {
let version = 0;
const { webgl } = ctx;
const updated = new rxjs_1.Subject();
const geometryState = new representation_1.Representation.GeometryState();
const materialId = (0, render_object_1.getNextMaterialId)();
const renderObjects = [];
const _state = representation_1.Representation.createState();
const visuals = new Map();
let _volume;
let _keys;
let _params;
let _props;
let _theme = theme_1.Theme.createEmpty();
async function visual(runtime, key) {
var _a;
let visual = visuals.get(key);
if (!visual) {
visual = visualCtor(materialId, _volume, key, _props, webgl);
visuals.set(key, visual);
}
else if ((_a = visual.mustRecreate) === null || _a === void 0 ? void 0 : _a.call(visual, { volume: _volume, key }, _props, webgl)) {
visual.destroy();
visual = visualCtor(materialId, _volume, key, _props, webgl);
visuals.set(key, visual);
}
return visual.createOrUpdate({ webgl, runtime }, _theme, _props, { volume: _volume, key });
}
function createOrUpdate(props = {}, volume) {
if (volume && volume !== _volume) {
_params = getParams(ctx, volume);
_volume = volume;
if (!_props)
_props = param_definition_1.ParamDefinition.getDefaultValues(_params);
}
const qualityProps = (0, util_1.getQualityProps)(Object.assign({}, _props, props), _volume);
Object.assign(_props, props, qualityProps);
_keys = getKeys(_props, _volume);
return mol_task_1.Task.create('Creating or updating VolumeRepresentation', async (runtime) => {
const toDelete = new Set(visuals.keys());
for (let i = 0, il = _keys.length; i < il; ++i) {
const key = _keys[i];
toDelete.delete(key);
const promise = visual(runtime, key);
if (promise)
await promise;
}
toDelete.forEach(key => {
var _a;
(_a = visuals.get(key)) === null || _a === void 0 ? void 0 : _a.destroy();
visuals.delete(key);
});
// update list of renderObjects
renderObjects.length = 0;
visuals.forEach(visual => {
if (visual.renderObject) {
renderObjects.push(visual.renderObject);
geometryState.add(visual.renderObject.id, visual.geometryVersion);
}
});
geometryState.snapshot();
// increment version
updated.next(version++);
});
}
function mark(loci, action) {
let changed = false;
visuals.forEach(visual => {
changed = visual.mark(loci, action) || changed;
});
return changed;
}
function setVisualState(visual, state) {
if (state.visible !== undefined && visual)
visual.setVisibility(state.visible);
if (state.alphaFactor !== undefined && visual)
visual.setAlphaFactor(state.alphaFactor);
if (state.pickable !== undefined && visual)
visual.setPickable(state.pickable);
if (state.overpaint !== undefined && visual)
visual.setOverpaint(state.overpaint);
if (state.transparency !== undefined && visual)
visual.setTransparency(state.transparency);
if (state.emissive !== undefined && visual)
visual.setEmissive(state.emissive);
if (state.substance !== undefined && visual)
visual.setSubstance(state.substance);
if (state.clipping !== undefined && visual)
visual.setClipping(state.clipping);
if (state.transform !== undefined && visual)
visual.setTransform(state.transform);
if (state.themeStrength !== undefined && visual)
visual.setThemeStrength(state.themeStrength);
}
function setState(state) {
const { visible, alphaFactor, pickable, overpaint, transparency, emissive, substance, clipping, transform, themeStrength, syncManually, markerActions } = state;
const newState = {};
if (visible !== undefined)
newState.visible = visible;
if (alphaFactor !== undefined)
newState.alphaFactor = alphaFactor;
if (pickable !== undefined)
newState.pickable = pickable;
if (overpaint !== undefined)
newState.overpaint = overpaint;
if (transparency !== undefined)
newState.transparency = transparency;
if (emissive !== undefined)
newState.emissive = emissive;
if (substance !== undefined)
newState.substance = substance;
if (clipping !== undefined)
newState.clipping = clipping;
if (themeStrength !== undefined)
newState.themeStrength = themeStrength;
if (transform !== undefined && !linear_algebra_1.Mat4.areEqual(transform, _state.transform, linear_algebra_1.EPSILON)) {
newState.transform = transform;
}
if (syncManually !== undefined)
newState.syncManually = syncManually;
if (markerActions !== undefined)
newState.markerActions = markerActions;
visuals.forEach(visual => setVisualState(visual, newState));
representation_1.Representation.updateState(_state, state);
}
function setTheme(theme) {
_theme = theme;
}
function destroy() {
visuals.forEach(visual => visual.destroy());
visuals.clear();
}
return {
label,
get groupCount() {
let groupCount = 0;
visuals.forEach(visual => {
if (visual.renderObject)
groupCount += visual.groupCount;
});
return groupCount;
},
get props() { return _props; },
get params() { return _params; },
get state() { return _state; },
get theme() { return _theme; },
get geometryVersion() { return geometryState.version; },
renderObjects,
updated,
createOrUpdate,
setState,
setTheme,
getLoci: (pickingId) => {
let loci = loci_1.EmptyLoci;
for (const visual of visuals.values()) {
const _loci = visual.getLoci(pickingId);
if (!(0, loci_1.isEmptyLoci)(_loci)) {
loci = _loci;
break;
}
}
return loci;
},
getAllLoci: () => {
return [getLoci(_volume, _props)];
},
eachLocation: (cb) => {
visuals.forEach(visual => {
visual.eachLocation(cb);
});
},
mark,
destroy
};
}