backbone.ts

/**
 * Copyright (c) 2018 mol* contributors, licensed under MIT, See LICENSE file for more info.
 *
 * @author Alexander Rose <alexander.rose@weirdbyte.de>
 */

import { StructureRepresentation, StructureUnitsRepresentation } from '.';
import { PickingId } from '../../util/picking';
import { Structure } from 'mol-model/structure';
import { Task } from 'mol-task';
import { Loci } from 'mol-model/loci';
import { MarkerAction } from '../../util/marker-data';
import { PolymerBackboneVisual, DefaultPolymerBackboneProps } from './visual/polymer-backbone-cylinder';

export const DefaultBackboneProps = {
    ...DefaultPolymerBackboneProps
}
export type BackboneProps = Partial<typeof DefaultBackboneProps>

export function BackboneRepresentation(): StructureRepresentation<BackboneProps> {
    const traceRepr = StructureUnitsRepresentation(PolymerBackboneVisual)

    return {
        get renderObjects() {
            return [ ...traceRepr.renderObjects ]
        },
        get props() {
            return { ...traceRepr.props }
        },
        create: (structure: Structure, props: BackboneProps = {} as BackboneProps) => {
            const p = Object.assign({}, DefaultBackboneProps, props)
            return Task.create('BackboneRepresentation', async ctx => {
                await traceRepr.create(structure, p).runInContext(ctx)
            })
        },
        update: (props: BackboneProps) => {
            const p = Object.assign({}, props)
            return Task.create('Updating BackboneRepresentation', async ctx => {
                await traceRepr.update(p).runInContext(ctx)
            })
        },
        getLoci: (pickingId: PickingId) => {
            return traceRepr.getLoci(pickingId)
        },
        mark: (loci: Loci, action: MarkerAction) => {
            traceRepr.mark(loci, action)
        },
        destroy() {
            traceRepr.destroy()
        }
    }
}