From 6fba929a5037c5b23183437692687add7de68a74 Mon Sep 17 00:00:00 2001 From: David Sehnal <david.sehnal@gmail.com> Date: Mon, 3 Dec 2018 09:09:31 +0100 Subject: [PATCH] mol-plugin: pass ref to Behavior.register --- src/mol-plugin/behavior/behavior.ts | 4 ++-- src/mol-plugin/behavior/static/state.ts | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/mol-plugin/behavior/behavior.ts b/src/mol-plugin/behavior/behavior.ts index 6da2a7816..e42dabfdd 100644 --- a/src/mol-plugin/behavior/behavior.ts +++ b/src/mol-plugin/behavior/behavior.ts @@ -5,7 +5,7 @@ */ import { PluginStateTransform, PluginStateObject } from '../state/objects'; -import { Transformer } from 'mol-state'; +import { Transformer, Transform } from 'mol-state'; import { Task } from 'mol-task'; import { PluginContext } from 'mol-plugin/context'; import { PluginCommand } from '../command'; @@ -16,7 +16,7 @@ import { shallowEqual } from 'mol-util'; export { PluginBehavior } interface PluginBehavior<P = unknown> { - register(): void, + register(ref: Transform.Ref): void, unregister(): void, /** Update params in place. Optionally return a promise if it depends on an async action. */ diff --git a/src/mol-plugin/behavior/static/state.ts b/src/mol-plugin/behavior/static/state.ts index 5a6e6be01..af9541687 100644 --- a/src/mol-plugin/behavior/static/state.ts +++ b/src/mol-plugin/behavior/static/state.ts @@ -28,7 +28,7 @@ export function registerDefault(ctx: PluginContext) { export function SyncBehaviors(ctx: PluginContext) { ctx.events.state.object.created.subscribe(o => { if (!SO.isBehavior(o.obj)) return; - o.obj.data.register(); + o.obj.data.register(o.ref); }); ctx.events.state.object.removed.subscribe(o => { @@ -39,7 +39,7 @@ export function SyncBehaviors(ctx: PluginContext) { ctx.events.state.object.updated.subscribe(o => { if (o.action === 'recreate') { if (o.oldObj && SO.isBehavior(o.oldObj)) o.oldObj.data.unregister(); - if (o.obj && SO.isBehavior(o.obj)) o.obj.data.register(); + if (o.obj && SO.isBehavior(o.obj)) o.obj.data.register(o.ref); } }); } -- GitLab