TmMolStar/src/mol-plugin/state.ts

/**
 * Copyright (c) 2018 mol* contributors, licensed under MIT, See LICENSE file for more info.
 *
 * @author David Sehnal <david.sehnal@gmail.com>
 */

import { State, StateTree } from 'mol-state';
import { PluginStateObjects as SO } from './state/objects';
import { Camera } from 'mol-canvas3d/camera';
import { PluginBehavior } from './behavior';

export { PluginState }

class PluginState {
    readonly data: State;
    readonly behavior: State;

    getSnapshot(): PluginState.Snapshot {
        return {
            data: this.data.getSnapshot(),
            behaviour: this.behavior.getSnapshot(),
            canvas3d: {
                camera: this.plugin.canvas3d.camera.getSnapshot()
            }
        };
    }

    async setSnapshot(snapshot: PluginState.Snapshot) {
        await this.behavior.setSnapshot(snapshot.behaviour);
        await this.data.setSnapshot(snapshot.data);
        this.plugin.canvas3d.camera.setState(snapshot.canvas3d.camera);
        this.plugin.canvas3d.requestDraw(true);
    }

    updateData(tree: StateTree) {
        return this.plugin.runTask(this.data.update(tree));
    }

    updateBehaviour(tree: StateTree) {
        return this.plugin.runTask(this.behavior.update(tree));
    }

    dispose() {
        this.data.dispose();
    }

    constructor(private plugin: import('./context').PluginContext) {
        this.data = State.create(new SO.Root({ label: 'Root' }, { }), { globalContext: plugin });
        this.behavior = State.create(new PluginBehavior.Root({ label: 'Root' }, { }), { globalContext: plugin });
    }
}

namespace PluginState {
    export interface Snapshot {
        data: State.Snapshot,
        behaviour: State.Snapshot,
        canvas3d: {
            camera: Camera.Snapshot
        }
    }
}