From 7d4c3ea57ba3a04f7812983cdafa7f578bd047b3 Mon Sep 17 00:00:00 2001
From: David Sehnal <david.sehnal@gmail.com>
Date: Wed, 3 Apr 2019 13:43:58 +0200
Subject: [PATCH] wip mol-state
---
src/mol-plugin/behavior/dynamic/labels.ts | 2 +-
.../behavior/static/representation.ts | 2 +-
src/mol-plugin/behavior/static/state.ts | 2 +-
src/mol-plugin/ui/state/tree.tsx | 26 +++++++++----------
src/mol-state/object.ts | 2 +-
src/mol-state/state.ts | 11 +++-----
src/mol-state/transform.ts | 13 ++++++++++
src/mol-state/tree/transient.ts | 12 ++++++---
8 files changed, 43 insertions(+), 27 deletions(-)
diff --git a/src/mol-plugin/behavior/dynamic/labels.ts b/src/mol-plugin/behavior/dynamic/labels.ts
index b3597932c..313ce27bf 100644
--- a/src/mol-plugin/behavior/dynamic/labels.ts
+++ b/src/mol-plugin/behavior/dynamic/labels.ts
@@ -118,7 +118,7 @@ export const SceneLabels = PluginBehavior.create<SceneLabelsProps>({
for (const s of structures) {
const rootStructure = getRootStructure(s, state)
if (!rootStructure || !SO.Molecule.Structure.is(rootStructure.obj)) continue
- if (!s.state.isHidden) {
+ if (!s.transform.state.isHidden) {
rootStructures.add(rootStructure.obj)
}
}
diff --git a/src/mol-plugin/behavior/static/representation.ts b/src/mol-plugin/behavior/static/representation.ts
index df0310e77..d17291691 100644
--- a/src/mol-plugin/behavior/static/representation.ts
+++ b/src/mol-plugin/behavior/static/representation.ts
@@ -92,5 +92,5 @@ export function UpdateRepresentationVisibility(ctx: PluginContext) {
}
function updateVisibility(cell: StateObjectCell, r: Representation<any>) {
- r.setState({ visible: !cell.state.isHidden });
+ r.setState({ visible: !cell.transform.state.isHidden });
}
\ No newline at end of file
diff --git a/src/mol-plugin/behavior/static/state.ts b/src/mol-plugin/behavior/static/state.ts
index c4111b371..0237d3263 100644
--- a/src/mol-plugin/behavior/static/state.ts
+++ b/src/mol-plugin/behavior/static/state.ts
@@ -87,7 +87,7 @@ export function ToggleExpanded(ctx: PluginContext) {
}
export function ToggleVisibility(ctx: PluginContext) {
- PluginCommands.State.ToggleVisibility.subscribe(ctx, ({ state, ref }) => setVisibility(state, ref, !state.cells.get(ref)!.state.isHidden));
+ PluginCommands.State.ToggleVisibility.subscribe(ctx, ({ state, ref }) => setVisibility(state, ref, !state.cells.get(ref)!.transform.state.isHidden));
}
function setVisibility(state: State, root: StateTransform.Ref, value: boolean) {
diff --git a/src/mol-plugin/ui/state/tree.tsx b/src/mol-plugin/ui/state/tree.tsx
index 5e0815d2d..d7680c969 100644
--- a/src/mol-plugin/ui/state/tree.tsx
+++ b/src/mol-plugin/ui/state/tree.tsx
@@ -54,8 +54,8 @@ class StateTreeNode extends PluginUIComponent<{ cell: StateObjectCell, depth: nu
this.subscribe(this.plugin.events.state.cell.stateUpdated, e => {
if (this.props.cell === e.cell && this.is(e) && e.state.cells.has(this.ref)) {
this.forceUpdate();
- // if (!!this.props.cell.state.isCollapsed !== this.state.isCollapsed) {
- // this.setState({ isCollapsed: !!e.cell.state.isCollapsed });
+ // if (!!this.props.cell.transform.state.isCollapsed !== this.state.isCollapsed) {
+ // this.setState({ isCollapsed: !!e.cell.transform.state.isCollapsed });
// }
}
});
@@ -74,22 +74,22 @@ class StateTreeNode extends PluginUIComponent<{ cell: StateObjectCell, depth: nu
}
state = {
- isCollapsed: !!this.props.cell.state.isCollapsed
+ isCollapsed: !!this.props.cell.transform.state.isCollapsed
}
static getDerivedStateFromProps(props: _Props<StateTreeNode>, state: _State<StateTreeNode>): _State<StateTreeNode> | null {
- if (!!props.cell.state.isCollapsed === state.isCollapsed) return null;
- return { isCollapsed: !!props.cell.state.isCollapsed };
+ if (!!props.cell.transform.state.isCollapsed === state.isCollapsed) return null;
+ return { isCollapsed: !!props.cell.transform.state.isCollapsed };
}
render() {
const cell = this.props.cell;
- if (!cell || cell.obj === StateObject.Null) {
+ if (!cell || cell.obj === StateObject.Null || !cell.parent.tree.transforms.has(cell.transform.ref)) {
return null;
}
- const cellState = cell.state;
- const showLabel = cell.status !== 'ok' || !cell.state.isGhost;
+ const cellState = cell.transform.state;
+ const showLabel = cell.status !== 'ok' || !cell.transform.state.isGhost;
const children = cell.parent.tree.children.get(this.ref);
const newDepth = showLabel ? this.props.depth + 1 : this.props.depth;
@@ -140,7 +140,7 @@ class StateTreeNodeLabel extends PluginUIComponent<
if (e.state.transforms.has(this.ref)) {
this.setState({
isCurrent: this.props.cell.parent.current === this.ref,
- isCollapsed: !!this.props.cell.state.isCollapsed
+ isCollapsed: !!this.props.cell.transform.state.isCollapsed
});
}
});
@@ -148,12 +148,12 @@ class StateTreeNodeLabel extends PluginUIComponent<
state = {
isCurrent: this.props.cell.parent.current === this.ref,
- isCollapsed: !!this.props.cell.state.isCollapsed
+ isCollapsed: !!this.props.cell.transform.state.isCollapsed
}
static getDerivedStateFromProps(props: _Props<StateTreeNodeLabel>, state: _State<StateTreeNodeLabel>): _State<StateTreeNodeLabel> | null {
const isCurrent = props.cell.parent.current === props.cell.transform.ref;
- const isCollapsed = !!props.cell.state.isCollapsed;
+ const isCollapsed = !!props.cell.transform.state.isCollapsed;
if (state.isCollapsed === isCollapsed && state.isCurrent === isCurrent) return null;
return { isCurrent, isCollapsed };
@@ -227,7 +227,7 @@ class StateTreeNodeLabel extends PluginUIComponent<
}
const children = cell.parent.tree.children.get(this.ref);
- const cellState = cell.state;
+ const cellState = cell.transform.state;
const visibility = <button onClick={this.toggleVisible} className={`msp-btn msp-btn-link msp-tree-visibility${cellState.isHidden ? ' msp-tree-visibility-hidden' : ''}`}>
<span className='msp-icon msp-icon-visual-visibility' />
@@ -244,7 +244,7 @@ class StateTreeNodeLabel extends PluginUIComponent<
{children.size > 0 && <button onClick={this.toggleExpanded} className='msp-btn msp-btn-link msp-tree-toggle-exp-button'>
<span className={`msp-icon msp-icon-${cellState.isCollapsed ? 'expand' : 'collapse'}`} />
</button>}
- {!cell.state.isLocked && <button onClick={this.remove} className='msp-btn msp-btn-link msp-tree-remove-button'>
+ {!cell.transform.state.isLocked && <button onClick={this.remove} className='msp-btn msp-btn-link msp-tree-remove-button'>
<span className='msp-icon msp-icon-remove' />
</button>}{visibility}
</div>;
diff --git a/src/mol-state/object.ts b/src/mol-state/object.ts
index dd4c5c1d3..a144ca8f9 100644
--- a/src/mol-state/object.ts
+++ b/src/mol-state/object.ts
@@ -64,7 +64,7 @@ interface StateObjectCell<T extends StateObject = StateObject, F extends StateTr
sourceRef: StateTransform.Ref | undefined,
status: StateObjectCell.Status,
- state: StateTransform.State,
+ // state: StateTransform.State,
params: {
definition: ParamDefinition.Params,
diff --git a/src/mol-state/state.ts b/src/mol-state/state.ts
index 5d0d00850..213ddb465 100644
--- a/src/mol-state/state.ts
+++ b/src/mol-state/state.ts
@@ -63,7 +63,6 @@ class State {
private spine = new StateTreeSpine.Impl(this.cells);
getSnapshot(): State.Snapshot {
- this.cells.forEach(c => this._tree.updateState(c.transform.ref, c.state));
return { tree: StateTree.toJSON(this._tree) };
}
@@ -80,10 +79,10 @@ class State {
const cell = this.cells.get(ref);
if (!cell) return;
- const update = typeof stateOrProvider === 'function' ? stateOrProvider(cell.state) : stateOrProvider;
+ const update = typeof stateOrProvider === 'function' ? stateOrProvider(cell.transform.state) : stateOrProvider;
- if (StateTransform.assignState(cell.state, update)) {
- // this._tree.updateCellState(ref, update)) {
+ if (this._tree.updateState(cell.transform.ref, update)) {
+ cell.transform = this._tree.transforms.get(cell.transform.ref);
this.events.cell.stateUpdated.next({ state: this, ref, cell });
}
}
@@ -211,7 +210,6 @@ class State {
sourceRef: void 0,
obj: rootObject,
status: 'ok',
- state: { ...root.state },
errorText: void 0,
params: {
definition: {},
@@ -409,7 +407,7 @@ function findDeletes(ctx: UpdateContext): Ref[] {
function syncNewStatesVisitor(n: StateTransform, tree: StateTree, ctx: UpdateContext) {
const cell = ctx.cells.get(n.ref);
- if (!cell || !StateTransform.assignState(cell.state, n.state)) return;
+ if (!cell || !StateTransform.syncState(cell.transform.state, n.state)) return;
ctx.parent.events.cell.stateUpdated.next({ state: ctx.parent, ref: n.ref, cell });
}
@@ -447,7 +445,6 @@ function initCellsVisitor(transform: StateTransform, _: any, { ctx, added }: Ini
transform,
sourceRef: void 0,
status: 'pending',
- state: { ...transform.state },
errorText: void 0,
params: void 0,
cache: void 0
diff --git a/src/mol-state/transform.ts b/src/mol-state/transform.ts
index 0e11a042e..fe4b251bf 100644
--- a/src/mol-state/transform.ts
+++ b/src/mol-state/transform.ts
@@ -53,6 +53,19 @@ namespace Transform {
export function assignState(a: State, b?: Partial<State>): boolean {
if (!b) return false;
+ let changed = false;
+ for (const k of Object.keys(b)) {
+ const s = (b as any)[k], t = (a as any)[k];
+ if (!!s === !!t) continue;
+ changed = true;
+ (a as any)[k] = s;
+ }
+ return changed;
+ }
+
+ export function syncState(a: State, b?: Partial<State>): boolean {
+ if (!b) return false;
+
let changed = false;
for (const k of Object.keys(b)) {
const s = (b as any)[k], t = (a as any)[k];
diff --git a/src/mol-state/tree/transient.ts b/src/mol-state/tree/transient.ts
index 068a5264d..e23b81aba 100644
--- a/src/mol-state/tree/transient.ts
+++ b/src/mol-state/tree/transient.ts
@@ -19,6 +19,7 @@ class TransientTree implements StateTree {
private changedChildren = false;
private _childMutations: Map<StateTransform.Ref, OrderedSet<StateTransform.Ref>> | undefined = void 0;
+ private _stateUpdates: Set<StateTransform.Ref> | undefined = void 0;
private get childMutations() {
if (this._childMutations) return this._childMutations;
@@ -145,9 +146,14 @@ class TransientTree implements StateTree {
const old = this.transforms.get(ref);
if (!StateTransform.isStateChange(old.state, state)) return false;
- this.changeNodes();
- // TODO: cache these changes?
- this.transforms.set(ref, StateTransform.withState(old, state));
+ if (this._stateUpdates && this._stateUpdates.has(old.ref)) {
+ StateTransform.assignState(old.state, state);
+ } else {
+ if (!this._stateUpdates) this._stateUpdates = new Set();
+ this._stateUpdates.add(old.ref);
+ this.changeNodes();
+ this.transforms.set(ref, StateTransform.withState(old, state));
+ }
return true;
}
--
GitLab