UNPKG

molstar

Version:

A comprehensive macromolecular library.

78 lines 4.54 kB
/** * Copyright (c) 2020 mol* contributors, licensed under MIT, See LICENSE file for more info. * * @author David Sehnal <david.sehnal@gmail.com> */ import { Volume } from '../../mol-model/volume'; import { ParamDefinition as PD } from '../../mol-util/param-definition'; export function createVolumeRepresentationParams(ctx, volume, props) { if (props === void 0) { props = {}; } var p = props; if (typeof p.type === 'string' || typeof p.color === 'string' || typeof p.size === 'string') return createParamsByName(ctx, volume || Volume.One, props); return createParamsProvider(ctx, volume || Volume.One, props); } export function getVolumeThemeTypes(ctx, volume) { var themeCtx = ctx.representation.volume.themes; if (!volume) return themeCtx.colorThemeRegistry.types; return themeCtx.colorThemeRegistry.getApplicableTypes({ volume: volume }); } export function createVolumeColorThemeParams(ctx, volume, typeName, themeName, params) { var _a = ctx.representation.volume, registry = _a.registry, themes = _a.themes; var repr = registry.get(typeName || registry.default.name); var color = themes.colorThemeRegistry.get(themeName || repr.defaultColorTheme.name); var colorDefaultParams = PD.getDefaultValues(color.getParams({ volume: volume || Volume.One })); if (color.name === repr.defaultColorTheme.name) Object.assign(colorDefaultParams, repr.defaultColorTheme.props); return { name: color.name, params: Object.assign(colorDefaultParams, params) }; } export function createVolumeSizeThemeParams(ctx, volume, typeName, themeName, params) { var _a = ctx.representation.volume, registry = _a.registry, themes = _a.themes; var repr = registry.get(typeName || registry.default.name); var size = themes.sizeThemeRegistry.get(themeName || repr.defaultSizeTheme.name); var sizeDefaultParams = PD.getDefaultValues(size.getParams({ volume: volume || Volume.One })); if (size.name === repr.defaultSizeTheme.name) Object.assign(sizeDefaultParams, repr.defaultSizeTheme.props); return { name: size.name, params: Object.assign(sizeDefaultParams, params) }; } function createParamsByName(ctx, volume, props) { var typeProvider = (props.type && ctx.representation.volume.registry.get(props.type)) || ctx.representation.volume.registry.default.provider; var colorProvider = (props.color && ctx.representation.volume.themes.colorThemeRegistry.get(props.color)) || ctx.representation.volume.themes.colorThemeRegistry.get(typeProvider.defaultColorTheme.name); var sizeProvider = (props.size && ctx.representation.volume.themes.sizeThemeRegistry.get(props.size)) || ctx.representation.volume.themes.sizeThemeRegistry.get(typeProvider.defaultSizeTheme.name); return createParamsProvider(ctx, volume, { type: typeProvider, typeParams: props.typeParams, color: colorProvider, colorParams: props.colorParams, size: sizeProvider, sizeParams: props.sizeParams }); } function createParamsProvider(ctx, volume, props) { if (props === void 0) { props = {}; } var themeCtx = ctx.representation.volume.themes; var themeDataCtx = { volume: volume }; var repr = props.type || ctx.representation.volume.registry.default.provider; var reprDefaultParams = PD.getDefaultValues(repr.getParams(themeCtx, volume)); var reprParams = Object.assign(reprDefaultParams, props.typeParams); var color = props.color || themeCtx.colorThemeRegistry.get(repr.defaultColorTheme.name); var colorDefaultParams = PD.getDefaultValues(color.getParams(themeDataCtx)); if (color.name === repr.defaultColorTheme.name) Object.assign(colorDefaultParams, repr.defaultColorTheme.props); var colorParams = Object.assign(colorDefaultParams, props.colorParams); var size = props.size || themeCtx.sizeThemeRegistry.get(repr.defaultSizeTheme.name); var sizeDefaultParams = PD.getDefaultValues(size.getParams(themeDataCtx)); if (size.name === repr.defaultSizeTheme.name) Object.assign(sizeDefaultParams, repr.defaultSizeTheme.props); var 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 } }); } //# sourceMappingURL=volume-representation-params.js.map