molstar
Version:
A comprehensive macromolecular library.
82 lines (81 loc) • 4.99 kB
JavaScript
"use strict";
/**
* Copyright (c) 2020 mol* contributors, licensed under MIT, See LICENSE file for more info.
*
* @author David Sehnal <david.sehnal@gmail.com>
*/
Object.defineProperty(exports, "__esModule", { value: true });
exports.createStructureRepresentationParams = createStructureRepresentationParams;
exports.getStructureThemeTypes = getStructureThemeTypes;
exports.createStructureColorThemeParams = createStructureColorThemeParams;
exports.createStructureSizeThemeParams = createStructureSizeThemeParams;
const structure_1 = require("../../mol-model/structure");
const param_definition_1 = require("../../mol-util/param-definition");
function createStructureRepresentationParams(ctx, structure, props = {}) {
const p = props;
if (typeof p.type === 'string' || typeof p.color === 'string' || typeof p.size === 'string')
return createParamsByName(ctx, structure || structure_1.Structure.Empty, props);
return createParamsProvider(ctx, structure || structure_1.Structure.Empty, props);
}
function getStructureThemeTypes(ctx, structure) {
const { themes: themeCtx } = ctx.representation.structure;
if (!structure)
return themeCtx.colorThemeRegistry.types;
return themeCtx.colorThemeRegistry.getApplicableTypes({ structure });
}
function createStructureColorThemeParams(ctx, structure, typeName, themeName, params) {
const { registry, themes } = ctx.representation.structure;
const repr = registry.get(typeName || registry.default.name);
const color = themes.colorThemeRegistry.get(themeName || repr.defaultColorTheme.name);
const colorDefaultParams = param_definition_1.ParamDefinition.getDefaultValues(color.getParams({ structure: structure || structure_1.Structure.Empty }));
if (color.name === repr.defaultColorTheme.name)
Object.assign(colorDefaultParams, repr.defaultColorTheme.props);
return { name: color.name, params: Object.assign(colorDefaultParams, params) };
}
function createStructureSizeThemeParams(ctx, structure, typeName, themeName, params) {
const { registry, themes } = ctx.representation.structure;
const repr = registry.get(typeName || registry.default.name);
const size = themes.sizeThemeRegistry.get(themeName || repr.defaultSizeTheme.name);
const sizeDefaultParams = param_definition_1.ParamDefinition.getDefaultValues(size.getParams({ structure: structure || structure_1.Structure.Empty }));
if (size.name === repr.defaultSizeTheme.name)
Object.assign(sizeDefaultParams, repr.defaultSizeTheme.props);
return { name: size.name, params: Object.assign(sizeDefaultParams, params) };
}
function createParamsByName(ctx, structure, props) {
const typeProvider = (props.type && ctx.representation.structure.registry.get(props.type))
|| ctx.representation.structure.registry.default.provider;
const colorProvider = (props.color && ctx.representation.structure.themes.colorThemeRegistry.get(props.color))
|| ctx.representation.structure.themes.colorThemeRegistry.get(typeProvider.defaultColorTheme.name);
const sizeProvider = (props.size && ctx.representation.structure.themes.sizeThemeRegistry.get(props.size))
|| ctx.representation.structure.themes.sizeThemeRegistry.get(typeProvider.defaultSizeTheme.name);
return createParamsProvider(ctx, structure, {
type: typeProvider,
typeParams: props.typeParams,
color: colorProvider,
colorParams: props.colorParams,
size: sizeProvider,
sizeParams: props.sizeParams
});
}
function createParamsProvider(ctx, structure, props = {}) {
const { themes: themeCtx } = ctx.representation.structure;
const themeDataCtx = { structure };
const repr = props.type || ctx.representation.structure.registry.default.provider;
const reprDefaultParams = param_definition_1.ParamDefinition.getDefaultValues(repr.getParams(themeCtx, structure));
const reprParams = Object.assign(reprDefaultParams, props.typeParams);
const color = props.color || themeCtx.colorThemeRegistry.get(repr.defaultColorTheme.name);
const colorDefaultParams = param_definition_1.ParamDefinition.getDefaultValues(color.getParams(themeDataCtx));
if (color.name === repr.defaultColorTheme.name)
Object.assign(colorDefaultParams, repr.defaultColorTheme.props);
const colorParams = Object.assign(colorDefaultParams, props.colorParams);
const size = props.size || themeCtx.sizeThemeRegistry.get(repr.defaultSizeTheme.name);
const sizeDefaultParams = param_definition_1.ParamDefinition.getDefaultValues(size.getParams(themeDataCtx));
if (size.name === repr.defaultSizeTheme.name)
Object.assign(sizeDefaultParams, repr.defaultSizeTheme.props);
const sizeParams = Object.assign(sizeDefaultParams, props.sizeParams);
return ({
type: { name: repr.name, params: reprParams },
colorTheme: { name: color.name, params: colorParams },
sizeTheme: { name: size.name, params: sizeParams }
});
}