From 4a692b9a88def79c7468fa11d0dc63855a30f6a5 Mon Sep 17 00:00:00 2001
From: Alexander Rose <alexander.rose@weirdbyte.de>
Date: Tue, 22 Oct 2019 10:20:55 -0700
Subject: [PATCH] ensure applicable repr types are up-to-date
---
src/mol-plugin/ui/structure/representation.tsx | 16 ++++++++++++++--
1 file changed, 14 insertions(+), 2 deletions(-)
diff --git a/src/mol-plugin/ui/structure/representation.tsx b/src/mol-plugin/ui/structure/representation.tsx
index 0eff7066a..758ec7e02 100644
--- a/src/mol-plugin/ui/structure/representation.tsx
+++ b/src/mol-plugin/ui/structure/representation.tsx
@@ -16,7 +16,7 @@ import { VisualQuality, VisualQualityOptions } from '../../../mol-geo/geometry/b
import { StructureRepresentationPresets as P } from '../../util/structure-representation-helper';
import { camelCaseToWords } from '../../../mol-util/string';
import { CollapsableControls } from '../base';
-import { StateSelection } from '../../../mol-state';
+import { StateSelection, StateObject } from '../../../mol-state';
import { PluginStateObject } from '../../state/objects';
abstract class BaseStructureRepresentationControls extends PluginUIComponent {
@@ -28,7 +28,7 @@ abstract class BaseStructureRepresentationControls extends PluginUIComponent {
return this.plugin.state.dataState.select(StateSelection.Generators.rootsOfType(PluginStateObject.Molecule.Structure)).map(s => s.obj!.data)
}
- /** applicapble types */
+ /** applicable types */
private get types() {
const types: [string, string][] = []
const structures = this.structures
@@ -40,6 +40,18 @@ abstract class BaseStructureRepresentationControls extends PluginUIComponent {
return types
}
+ private forceUpdateIfStructure = (obj?: StateObject) => {
+ if (obj && obj.type === PluginStateObject.Molecule.Structure.type) {
+ this.forceUpdate()
+ }
+ }
+
+ componentDidMount() {
+ this.subscribe(this.plugin.events.state.object.created, ({ obj }) => this.forceUpdateIfStructure(obj));
+
+ this.subscribe(this.plugin.events.state.object.removed, ({ obj }) => this.forceUpdateIfStructure(obj));
+ }
+
show = (value: string) => {
this.plugin.helpers.structureRepresentation.set('add', value, this.lociGetter)
}
--
GitLab