drugflow-molstar 0.3.77 → 0.4.1

package/lib/ui/segment-tree.js

@@ -1,602 +1,475 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.SegmentTree = void 0;
-var tslib_1 = require("tslib");
-var jsx_runtime_1 = require("react/jsx-runtime");
-var operators_1 = require("rxjs/operators");
-var commands_1 = require("Molstar/mol-plugin/commands");
-var mol_state_1 = require("Molstar/mol-state");
-var base_1 = require("Molstar/mol-plugin-ui/base");
-var common_1 = require("Molstar/mol-plugin-ui/controls/common");
-var param_definition_1 = require("Molstar/mol-util/param-definition");
-var parameters_1 = require("Molstar/mol-plugin-ui/controls/parameters");
-var mol_state_2 = require("Molstar/mol-state");
-var superposition_1 = require("../superposition");
-var update_transform_1 = require("Molstar/mol-plugin-ui/state/update-transform");
-var icons_1 = require("Molstar/mol-plugin-ui/controls/icons");
-var lists_1 = require("Molstar/mol-util/color/lists");
-var color_1 = require("Molstar/mol-util/color");
-var builder_1 = require("Molstar/mol-script/language/builder");
-var rxjs_1 = require("rxjs");
-var transforms_1 = require("Molstar/mol-plugin-state/transforms");
-var superposition_2 = require("../superposition");
-var SuperpositionTag = 'SuperpositionTransform';
-var SegmentTree = /** @class */ (function (_super) {
-    tslib_1.__extends(SegmentTree, _super);
-    function SegmentTree() {
-        var _this = _super !== null && _super.apply(this, arguments) || this;
-        _this.getSegmentParams = function () {
-            var customState = _this.customState;
+const jsx_runtime_1 = require("react/jsx-runtime");
+const operators_1 = require("rxjs/operators");
+const commands_1 = require("Molstar/mol-plugin/commands");
+const mol_state_1 = require("Molstar/mol-state");
+const base_1 = require("Molstar/mol-plugin-ui/base");
+const common_1 = require("Molstar/mol-plugin-ui/controls/common");
+const param_definition_1 = require("Molstar/mol-util/param-definition");
+const parameters_1 = require("Molstar/mol-plugin-ui/controls/parameters");
+const mol_state_2 = require("Molstar/mol-state");
+const superposition_1 = require("../superposition");
+const update_transform_1 = require("Molstar/mol-plugin-ui/state/update-transform");
+const icons_1 = require("Molstar/mol-plugin-ui/controls/icons");
+const lists_1 = require("Molstar/mol-util/color/lists");
+const color_1 = require("Molstar/mol-util/color");
+const builder_1 = require("Molstar/mol-script/language/builder");
+const rxjs_1 = require("rxjs");
+const transforms_1 = require("Molstar/mol-plugin-state/transforms");
+const superposition_2 = require("../superposition");
+const SuperpositionTag = 'SuperpositionTransform';
+class SegmentTree extends base_1.PurePluginUIComponent {
+    constructor() {
+        super(...arguments);
+        this.getSegmentParams = () => {
+            const customState = this.customState;
             if (!customState.superpositionState || !customState.superpositionState.segmentData)
                 return;
-            var segmentData = customState.superpositionState.segmentData;
-            var segmentArr = segmentData.map(function (segment, i) {
-                var segmentLabel = "".concat(i + 1, " ( ").concat(segment.segment_start, " - ").concat(segment.segment_end, " )");
+            const segmentData = customState.superpositionState.segmentData;
+            const segmentArr = segmentData.map((segment, i) => {
+                const segmentLabel = `${i + 1} ( ${segment.segment_start} - ${segment.segment_end} )`;
                 return [segmentLabel, segmentLabel];
             });
-            var segmentOptions = {
+            const segmentOptions = {
                 segment: param_definition_1.ParamDefinition.Select('', segmentArr, { label: 'Select Segment', description: 'Select segment to view its clusters below' })
             };
-            var segmentIndex = customState.superpositionState.activeSegment - 1;
-            _this.setState({ segment: {
+            const segmentIndex = customState.superpositionState.activeSegment - 1;
+            this.setState({ segment: {
                     params: segmentOptions,
                     value: { segment: segmentArr[segmentIndex][0] }
                 } });
-            _this.setState({ isBusy: false });
+            this.setState({ isBusy: false });
         };
-        _this.updateSegment = function (val) { return tslib_1.__awaiter(_this, void 0, void 0, function () {
-            var customState, updatedParams;
-            var _this = this;
-            return tslib_1.__generator(this, function (_a) {
-                if (!this.state.segment)
-                    return [2 /*return*/];
-                customState = this.customState;
-                customState.events.isBusy.next(true);
-                // Hide pervious segement structures
-                this.hideStructures(customState.superpositionState.activeSegment - 1);
-                updatedParams = tslib_1.__assign({}, this.state.segment);
-                updatedParams.value = val;
-                this.setState({ segment: updatedParams });
-                setTimeout(function () { return tslib_1.__awaiter(_this, void 0, void 0, function () {
-                    var updatedSegmentIndex;
-                    return tslib_1.__generator(this, function (_a) {
-                        switch (_a.label) {
-                            case 0:
-                                updatedSegmentIndex = parseInt(val.segment.split(' ')[0]);
-                                customState.superpositionState.activeSegment = updatedSegmentIndex;
-                                // Display current segment visible structures
-                                return [4 /*yield*/, this.displayStructures(customState.superpositionState.activeSegment - 1)];
-                            case 1:
-                                // Display current segment visible structures
-                                _a.sent();
-                                customState.events.isBusy.next(false);
-                                this.plugin.customState.events.segmentUpdate.next(true);
-                                return [2 /*return*/];
-                        }
-                    });
-                }); }, 100);
-                return [2 /*return*/, false];
-            });
-        }); };
-        _this.hideStructures = function (segmentIndex) {
+        this.updateSegment = async (val) => {
+            if (!this.state.segment)
+                return;
+            const customState = this.customState;
+            customState.events.isBusy.next(true);
+            // Hide pervious segement structures
+            this.hideStructures(customState.superpositionState.activeSegment - 1);
+            // Set current segment params
+            const updatedParams = { ...this.state.segment };
+            updatedParams.value = val;
+            this.setState({ segment: updatedParams });
+            setTimeout(async () => {
+                const updatedSegmentIndex = parseInt(val.segment.split(' ')[0]);
+                customState.superpositionState.activeSegment = updatedSegmentIndex;
+                // Display current segment visible structures
+                await this.displayStructures(customState.superpositionState.activeSegment - 1);
+                customState.events.isBusy.next(false);
+                this.plugin.customState.events.segmentUpdate.next(true);
+            }, 100);
+            return false;
+        };
+        this.hideStructures = (segmentIndex) => {
             // clear selections
-            _this.plugin.managers.interactivity.lociSelects.deselectAll();
+            this.plugin.managers.interactivity.lociSelects.deselectAll();
             // clear Focus
-            _this.plugin.managers.structure.focus.clear();
+            this.plugin.managers.structure.focus.clear();
             // remove measurements
-            var measurements = _this.plugin.managers.structure.measurement.state;
-            var measureTypes = ['labels', 'distances', 'angles', 'dihedrals'];
-            var measurementCell = void 0;
-            measureTypes.forEach(function (type) {
+            const measurements = this.plugin.managers.structure.measurement.state;
+            const measureTypes = ['labels', 'distances', 'angles', 'dihedrals'];
+            let measurementCell = void 0;
+            measureTypes.forEach((type) => {
                 if (measurementCell)
                     return;
                 if (measurements[type][0]) {
-                    measurementCell = _this.plugin.state.data.cells.get(measurements[type][0].transform.parent);
+                    measurementCell = this.plugin.state.data.cells.get(measurements[type][0].transform.parent);
                 }
             });
             if (measurementCell) {
-                commands_1.PluginCommands.State.RemoveObject(_this.plugin, { state: measurementCell.parent, ref: measurementCell.transform.parent, removeParentGhosts: true });
+                commands_1.PluginCommands.State.RemoveObject(this.plugin, { state: measurementCell.parent, ref: measurementCell.transform.parent, removeParentGhosts: true });
             }
             // hide structures
-            var customState = _this.customState;
+            const customState = this.customState;
             customState.superpositionState.visibleRefs[segmentIndex] = [];
-            for (var _i = 0, _a = customState.superpositionState.loadedStructs[segmentIndex]; _i < _a.length; _i++) {
-                var struct = _a[_i];
-                var structRef = _this.customState.superpositionState.models[struct];
+            for (const struct of customState.superpositionState.loadedStructs[segmentIndex]) {
+                const structRef = this.customState.superpositionState.models[struct];
                 if (structRef) {
-                    var structHierarchy = _this.plugin.managers.structure.hierarchy.current.refs.get(structRef);
+                    const structHierarchy = this.plugin.managers.structure.hierarchy.current.refs.get(structRef);
                     if (structHierarchy && structHierarchy.components) {
-                        for (var _b = 0, _c = structHierarchy.components; _b < _c.length; _b++) {
-                            var c = _c[_b];
+                        for (const c of structHierarchy.components) {
                             if (c && c.cell && !c.cell.state.isHidden) {
                                 customState.superpositionState.visibleRefs[segmentIndex].push(c.cell.transform.ref);
-                                commands_1.PluginCommands.State.ToggleVisibility(_this.plugin, { state: c.cell.parent, ref: c.cell.transform.ref });
+                                commands_1.PluginCommands.State.ToggleVisibility(this.plugin, { state: c.cell.parent, ref: c.cell.transform.ref });
                             }
                         }
                     }
                 }
             }
             if (customState.superpositionState.alphafold.ref) {
-                var afStr = _this.plugin.managers.structure.hierarchy.current.refs.get(customState.superpositionState.alphafold.ref);
+                const afStr = this.plugin.managers.structure.hierarchy.current.refs.get(customState.superpositionState.alphafold.ref);
                 if (afStr && afStr.components) {
-                    for (var _d = 0, _e = afStr.components; _d < _e.length; _d++) {
-                        var c = _e[_d];
+                    for (const c of afStr.components) {
                         if (c && c.cell && !c.cell.state.isHidden) {
-                            commands_1.PluginCommands.State.ToggleVisibility(_this.plugin, { state: c.cell.parent, ref: c.cell.transform.ref });
+                            commands_1.PluginCommands.State.ToggleVisibility(this.plugin, { state: c.cell.parent, ref: c.cell.transform.ref });
                         }
                     }
                 }
             }
         };
-        _this.displayStructures = function (segmentIndex) { return tslib_1.__awaiter(_this, void 0, void 0, function () {
-            var customState, spState, loadStrs_1, _i, _a, ref, cell, afStr, _b, _c, c;
-            var _this = this;
-            return tslib_1.__generator(this, function (_d) {
-                switch (_d.label) {
-                    case 0:
-                        customState = this.customState;
-                        spState = customState.superpositionState;
-                        if (!(customState.superpositionState.visibleRefs[segmentIndex].length === 0)) return [3 /*break*/, 3];
-                        loadStrs_1 = [];
-                        customState.superpositionState.segmentData[segmentIndex].clusters.forEach(function (cluster) {
-                            var entryList = [cluster[0]];
-                            if (customState.initParams.superpositionParams && customState.initParams.superpositionParams.superposeAll) {
-                                entryList = cluster;
-                            }
-                            entryList.forEach(function (str) {
-                                var structStateId = "".concat(str.pdb_id, "_").concat(str.struct_asym_id);
-                                var structRef = customState.superpositionState.models[structStateId];
-                                if (structRef) {
-                                    var cell = _this.plugin.state.data.cells.get(structRef);
-                                    var isHidden = cell.state.isHidden ? true : false;
-                                    if (isHidden) {
-                                        commands_1.PluginCommands.State.ToggleVisibility(_this.plugin, { state: cell.parent, ref: structRef });
-                                        // PluginCommands.State.ToggleVisibility(this.plugin, { state: cell.parent!, ref: cell.transform.parent });
-                                    }
-                                }
-                                else {
-                                    loadStrs_1.push(str);
-                                }
-                            });
-                        });
-                        commands_1.PluginCommands.Camera.Reset(this.plugin);
-                        if (!(loadStrs_1.length > 0)) return [3 /*break*/, 2];
-                        return [4 /*yield*/, (0, superposition_1.renderSuperposition)(this.plugin, segmentIndex, loadStrs_1)];
-                    case 1:
-                        _d.sent();
-                        commands_1.PluginCommands.Camera.Reset(this.plugin);
-                        _d.label = 2;
-                    case 2: return [3 /*break*/, 4];
-                    case 3:
-                        for (_i = 0, _a = customState.superpositionState.visibleRefs[segmentIndex]; _i < _a.length; _i++) {
-                            ref = _a[_i];
-                            cell = this.plugin.state.data.cells.get(ref);
-                            if (cell && cell.state.isHidden) {
-                                commands_1.PluginCommands.State.ToggleVisibility(this.plugin, { state: cell.parent, ref: ref });
+        this.displayStructures = async (segmentIndex) => {
+            const customState = this.customState;
+            const spState = customState.superpositionState;
+            if (customState.superpositionState.visibleRefs[segmentIndex].length === 0) {
+                let loadStrs = [];
+                customState.superpositionState.segmentData[segmentIndex].clusters.forEach((cluster) => {
+                    let entryList = [cluster[0]];
+                    if (customState.initParams.superpositionParams && customState.initParams.superpositionParams.superposeAll) {
+                        entryList = cluster;
+                    }
+                    entryList.forEach((str) => {
+                        const structStateId = `${str.pdb_id}_${str.struct_asym_id}`;
+                        const structRef = customState.superpositionState.models[structStateId];
+                        if (structRef) {
+                            const cell = this.plugin.state.data.cells.get(structRef);
+                            const isHidden = cell.state.isHidden ? true : false;
+                            if (isHidden) {
+                                commands_1.PluginCommands.State.ToggleVisibility(this.plugin, { state: cell.parent, ref: structRef });
+                                // PluginCommands.State.ToggleVisibility(this.plugin, { state: cell.parent!, ref: cell.transform.parent });
                             }
                         }
-                        commands_1.PluginCommands.Camera.Reset(this.plugin);
-                        _d.label = 4;
-                    case 4:
-                        if (spState.alphafold.ref) {
-                            (0, superposition_2.superposeAf)(this.plugin, spState.alphafold.traceOnly);
-                            commands_1.PluginCommands.Camera.Reset(this.plugin);
+                        else {
+                            loadStrs.push(str);
                         }
-                        if (spState.alphafold.ref && spState.alphafold.visibility[segmentIndex]) {
-                            afStr = this.plugin.managers.structure.hierarchy.current.refs.get(customState.superpositionState.alphafold.ref);
-                            if (afStr && afStr.components) {
-                                for (_b = 0, _c = afStr.components; _b < _c.length; _b++) {
-                                    c = _c[_b];
-                                    if (c && c.cell && c.cell.state.isHidden) {
-                                        commands_1.PluginCommands.State.ToggleVisibility(this.plugin, { state: c.cell.parent, ref: c.cell.transform.ref });
-                                    }
-                                }
-                            }
+                    });
+                });
+                commands_1.PluginCommands.Camera.Reset(this.plugin);
+                if (loadStrs.length > 0) {
+                    await (0, superposition_1.renderSuperposition)(this.plugin, segmentIndex, loadStrs);
+                    commands_1.PluginCommands.Camera.Reset(this.plugin);
+                }
+            }
+            else {
+                for (const ref of customState.superpositionState.visibleRefs[segmentIndex]) {
+                    const cell = this.plugin.state.data.cells.get(ref);
+                    if (cell && cell.state.isHidden) {
+                        commands_1.PluginCommands.State.ToggleVisibility(this.plugin, { state: cell.parent, ref });
+                    }
+                }
+                commands_1.PluginCommands.Camera.Reset(this.plugin);
+            }
+            if (spState.alphafold.ref) {
+                (0, superposition_2.superposeAf)(this.plugin, spState.alphafold.traceOnly);
+                commands_1.PluginCommands.Camera.Reset(this.plugin);
+            }
+            if (spState.alphafold.ref && spState.alphafold.visibility[segmentIndex]) {
+                const afStr = this.plugin.managers.structure.hierarchy.current.refs.get(customState.superpositionState.alphafold.ref);
+                if (afStr && afStr.components) {
+                    for (const c of afStr.components) {
+                        if (c && c.cell && c.cell.state.isHidden) {
+                            commands_1.PluginCommands.State.ToggleVisibility(this.plugin, { state: c.cell.parent, ref: c.cell.transform.ref });
                         }
-                        return [2 /*return*/];
+                    }
                 }
-            });
-        }); };
-        return _this;
+            }
+        };
     }
-    SegmentTree.prototype.componentDidMount = function () {
-        var _this = this;
-        this.subscribe(this.plugin.customState.events.superpositionInit, function () {
-            var customState = _this.customState;
+    componentDidMount() {
+        this.subscribe(this.plugin.customState.events.superpositionInit, () => {
+            const customState = this.customState;
             if (customState && !customState.superpositionError) {
-                _this.getSegmentParams();
+                this.getSegmentParams();
             }
-            _this.forceUpdate();
+            this.forceUpdate();
         });
-        this.subscribe(this.plugin.customState.events.isBusy, function (e) {
-            _this.setState({ isBusy: e });
+        this.subscribe(this.plugin.customState.events.isBusy, (e) => {
+            this.setState({ isBusy: e });
             if (e) {
-                commands_1.PluginCommands.Toast.Show(_this.plugin, {
+                commands_1.PluginCommands.Toast.Show(this.plugin, {
                     title: 'Process',
                     message: 'Loading / computing large dataset!',
                     key: 'is-busy-toast'
                 });
             }
             else {
-                commands_1.PluginCommands.Toast.Hide(_this.plugin, { key: 'is-busy-toast' });
+                commands_1.PluginCommands.Toast.Hide(this.plugin, { key: 'is-busy-toast' });
             }
         });
-        this.subscribe(this.plugin.behaviors.layout.leftPanelTabName, function (e) {
+        this.subscribe(this.plugin.behaviors.layout.leftPanelTabName, (e) => {
             if (e !== 'segments')
                 return;
-            _this.getSegmentParams();
-            _this.forceUpdate();
-        });
-    };
-    Object.defineProperty(SegmentTree.prototype, "customState", {
-        get: function () {
-            return this.plugin.customState;
-        },
-        enumerable: false,
-        configurable: true
-    });
-    SegmentTree.prototype.transform = function (s, matrix) {
-        return tslib_1.__awaiter(this, void 0, void 0, function () {
-            var r, o, params, b;
-            return tslib_1.__generator(this, function (_a) {
-                switch (_a.label) {
-                    case 0:
-                        r = mol_state_2.StateObjectRef.resolveAndCheck(this.plugin.state.data, s);
-                        if (!r)
-                            return [2 /*return*/];
-                        o = mol_state_2.StateSelection.findTagInSubtree(this.plugin.state.data.tree, r.transform.ref, SuperpositionTag);
-                        params = {
-                            transform: {
-                                name: 'matrix',
-                                params: { data: matrix, transpose: false }
-                            }
-                        };
-                        b = o
-                            ? this.plugin.state.data.build().to(o).update(params)
-                            : this.plugin.state.data.build().to(s)
-                                .insert(transforms_1.StateTransforms.Model.TransformStructureConformation, params, { tags: SuperpositionTag });
-                        return [4 /*yield*/, this.plugin.runTask(this.plugin.state.data.updateTree(b))];
-                    case 1:
-                        _a.sent();
-                        return [2 /*return*/];
-                }
-            });
+            this.getSegmentParams();
+            this.forceUpdate();
         });
-    };
-    SegmentTree.prototype.render = function () {
-        var sectionHeader = (0, jsx_runtime_1.jsx)(common_1.SectionHeader, { title: "Structure clusters" });
-        var customState = this.customState;
+    }
+    get customState() {
+        return this.plugin.customState;
+    }
+    async transform(s, matrix) {
+        const r = mol_state_2.StateObjectRef.resolveAndCheck(this.plugin.state.data, s);
+        if (!r)
+            return;
+        // TODO should find any TransformStructureConformation decorator instance
+        const o = mol_state_2.StateSelection.findTagInSubtree(this.plugin.state.data.tree, r.transform.ref, SuperpositionTag);
+        const params = {
+            transform: {
+                name: 'matrix',
+                params: { data: matrix, transpose: false }
+            }
+        };
+        // TODO add .insertOrUpdate to StateBuilder?
+        const b = o
+            ? this.plugin.state.data.build().to(o).update(params)
+            : this.plugin.state.data.build().to(s)
+                .insert(transforms_1.StateTransforms.Model.TransformStructureConformation, params, { tags: SuperpositionTag });
+        await this.plugin.runTask(this.plugin.state.data.updateTree(b));
+    }
+    render() {
+        let sectionHeader = (0, jsx_runtime_1.jsx)(common_1.SectionHeader, { title: `Structure clusters` });
+        const customState = this.customState;
         if (customState && customState.initParams && !customState.initParams.superposition) {
             return (0, jsx_runtime_1.jsxs)(jsx_runtime_1.Fragment, { children: [sectionHeader, (0, jsx_runtime_1.jsx)("div", { children: "Functionality unavailable!" })] });
         }
         else {
             if (customState && customState.initParams && customState.initParams.superposition) {
-                sectionHeader = (0, jsx_runtime_1.jsx)(common_1.SectionHeader, { title: "Structure clusters - ".concat(customState.initParams.moleculeId) });
+                sectionHeader = (0, jsx_runtime_1.jsx)(common_1.SectionHeader, { title: `Structure clusters - ${customState.initParams.moleculeId}` });
                 if (customState.superpositionError) {
-                    return (0, jsx_runtime_1.jsxs)(jsx_runtime_1.Fragment, { children: [sectionHeader, (0, jsx_runtime_1.jsx)("div", tslib_1.__assign({ style: { textAlign: 'center' } }, { children: customState.superpositionError }))] });
+                    return (0, jsx_runtime_1.jsxs)(jsx_runtime_1.Fragment, { children: [sectionHeader, (0, jsx_runtime_1.jsx)("div", { style: { textAlign: 'center' }, children: customState.superpositionError })] });
                 }
                 else if (!customState.superpositionState || !customState.superpositionState.segmentData) {
-                    return (0, jsx_runtime_1.jsxs)(jsx_runtime_1.Fragment, { children: [sectionHeader, (0, jsx_runtime_1.jsx)("div", tslib_1.__assign({ style: { textAlign: 'center' } }, { children: "Loading Segment Data!" }))] });
+                    return (0, jsx_runtime_1.jsxs)(jsx_runtime_1.Fragment, { children: [sectionHeader, (0, jsx_runtime_1.jsx)("div", { style: { textAlign: 'center' }, children: "Loading Segment Data!" })] });
                 }
             }
         }
         if (this.state) {
-            var segmentIndex_1 = parseInt(this.state.segment.value.segment.split(' ')[0]) - 1;
-            var segmentData_1 = customState.superpositionState.segmentData;
-            var fullSegmentRange = "( ".concat(segmentData_1[0].segment_start, " - ").concat(segmentData_1[segmentData_1.length - 1].segment_end, " )");
-            sectionHeader = (0, jsx_runtime_1.jsx)(common_1.SectionHeader, { title: "Structure clusters ".concat(customState.initParams.moleculeId), desc: fullSegmentRange });
-            return (0, jsx_runtime_1.jsxs)(jsx_runtime_1.Fragment, { children: [sectionHeader, (0, jsx_runtime_1.jsx)(parameters_1.ParameterControls, { params: this.state.segment.params, values: this.state.segment.value, onChangeValues: this.updateSegment, isDisabled: this.state.isBusy }), segmentData_1[segmentIndex_1].clusters.map(function (c, i) { return (0, jsx_runtime_1.jsx)(ClusterNode, { cluster: c, totalClusters: segmentData_1[segmentIndex_1].clusters.length, segmentIndex: segmentIndex_1, clusterIndex: i }, "cluster-".concat(segmentIndex_1, "-").concat(i)); })] });
+            const segmentIndex = parseInt(this.state.segment.value.segment.split(' ')[0]) - 1;
+            const segmentData = customState.superpositionState.segmentData;
+            const fullSegmentRange = `( ${segmentData[0].segment_start} - ${segmentData[segmentData.length - 1].segment_end} )`;
+            sectionHeader = (0, jsx_runtime_1.jsx)(common_1.SectionHeader, { title: `Structure clusters ${customState.initParams.moleculeId}`, desc: fullSegmentRange });
+            return (0, jsx_runtime_1.jsxs)(jsx_runtime_1.Fragment, { children: [sectionHeader, (0, jsx_runtime_1.jsx)(parameters_1.ParameterControls, { params: this.state.segment.params, values: this.state.segment.value, onChangeValues: this.updateSegment, isDisabled: this.state.isBusy }), segmentData[segmentIndex].clusters.map((c, i) => (0, jsx_runtime_1.jsx)(ClusterNode, { cluster: c, totalClusters: segmentData[segmentIndex].clusters.length, segmentIndex: segmentIndex, clusterIndex: i }, `cluster-${segmentIndex}-${i}`))] });
         }
         return (0, jsx_runtime_1.jsx)(jsx_runtime_1.Fragment, {});
-    };
-    return SegmentTree;
-}(base_1.PurePluginUIComponent));
+    }
+}
 exports.SegmentTree = SegmentTree;
-var ClusterNode = /** @class */ (function (_super) {
-    tslib_1.__extends(ClusterNode, _super);
-    function ClusterNode() {
-        var _this = _super !== null && _super.apply(this, arguments) || this;
-        _this.state = {
+class ClusterNode extends base_1.PluginUIComponent {
+    constructor() {
+        super(...arguments);
+        this.state = {
             isCollapsed: false,
             showAll: false,
             showNone: false,
             showSearch: false,
             isBusy: false,
-            cluster: _this.props.cluster,
+            cluster: this.props.cluster,
             searchText: ''
         };
-        _this.inputStream = new rxjs_1.Subject();
-        _this.handleInputStream = function (inputStr) {
-            _this.setState({ searchText: inputStr });
-            var filteredRes = _this.props.cluster.filter(function (item) {
+        this.inputStream = new rxjs_1.Subject();
+        this.handleInputStream = (inputStr) => {
+            this.setState({ searchText: inputStr });
+            const filteredRes = this.props.cluster.filter((item) => {
                 return item.pdb_id.toLowerCase().indexOf(inputStr.toLowerCase()) >= 0;
             });
-            _this.setState({ cluster: filteredRes });
+            this.setState({ cluster: filteredRes });
         };
-        _this.toggleExpanded = function (e) {
+        this.toggleExpanded = (e) => {
             e.preventDefault();
-            _this.setState({ isCollapsed: !_this.state.isCollapsed });
+            this.setState({ isCollapsed: !this.state.isCollapsed });
             e.currentTarget.blur();
         };
-        _this.selectAll = function (e) {
+        this.selectAll = (e) => {
             e.preventDefault();
-            _this.setState({ showAll: !_this.state.showAll, showNone: false });
+            this.setState({ showAll: !this.state.showAll, showNone: false });
             e.currentTarget.blur();
         };
-        _this.selectNone = function (e) {
+        this.selectNone = (e) => {
             e.preventDefault();
-            _this.setState({ showAll: false, showNone: !_this.state.showNone });
+            this.setState({ showAll: false, showNone: !this.state.showNone });
             e.currentTarget.blur();
         };
-        _this.applyAction = function (e) { return tslib_1.__awaiter(_this, void 0, void 0, function () {
-            var customState, currentState;
-            var _this = this;
-            return tslib_1.__generator(this, function (_a) {
-                e.preventDefault();
-                e.currentTarget.blur();
-                customState = this.customState;
-                customState.events.isBusy.next(true);
-                currentState = tslib_1.__assign({}, this.state);
-                this.setState({ showAll: false, showNone: false });
-                setTimeout(function () { return tslib_1.__awaiter(_this, void 0, void 0, function () {
-                    var loadStrs, _a, _b, str, structStateId, structRef, cell, isHidden, e_1_1;
-                    var e_1, _c;
-                    return tslib_1.__generator(this, function (_d) {
-                        switch (_d.label) {
-                            case 0:
-                                loadStrs = [];
-                                _d.label = 1;
-                            case 1:
-                                _d.trys.push([1, 9, 10, 15]);
-                                _a = tslib_1.__asyncValues(this.state.cluster);
-                                _d.label = 2;
-                            case 2: return [4 /*yield*/, _a.next()];
-                            case 3:
-                                if (!(_b = _d.sent(), !_b.done)) return [3 /*break*/, 8];
-                                str = _b.value;
-                                structStateId = "".concat(str.pdb_id, "_").concat(str.struct_asym_id);
-                                structRef = undefined;
-                                if (customState && customState.superpositionState && customState.superpositionState.models[structStateId]) {
-                                    structRef = this.customState.superpositionState.models[structStateId];
-                                }
-                                if (!structRef) return [3 /*break*/, 6];
-                                cell = this.plugin.state.data.cells.get(structRef);
-                                if (!cell) return [3 /*break*/, 5];
-                                isHidden = cell.state.isHidden ? true : false;
-                                if (!((isHidden && currentState.showAll) || (!isHidden && currentState.showNone))) return [3 /*break*/, 5];
-                                return [4 /*yield*/, commands_1.PluginCommands.State.ToggleVisibility(this.plugin, { state: cell.parent, ref: structRef })];
-                            case 4:
-                                _d.sent();
-                                _d.label = 5;
-                            case 5: return [3 /*break*/, 7];
-                            case 6:
-                                if (currentState.showAll)
-                                    loadStrs.push(str);
-                                _d.label = 7;
-                            case 7: return [3 /*break*/, 2];
-                            case 8: return [3 /*break*/, 15];
-                            case 9:
-                                e_1_1 = _d.sent();
-                                e_1 = { error: e_1_1 };
-                                return [3 /*break*/, 15];
-                            case 10:
-                                _d.trys.push([10, , 13, 14]);
-                                if (!(_b && !_b.done && (_c = _a.return))) return [3 /*break*/, 12];
-                                return [4 /*yield*/, _c.call(_a)];
-                            case 11:
-                                _d.sent();
-                                _d.label = 12;
-                            case 12: return [3 /*break*/, 14];
-                            case 13:
-                                if (e_1) throw e_1.error;
-                                return [7 /*endfinally*/];
-                            case 14: return [7 /*endfinally*/];
-                            case 15:
-                                ;
-                                commands_1.PluginCommands.Camera.Reset(this.plugin);
-                                if (!(loadStrs.length > 0)) return [3 /*break*/, 17];
-                                return [4 /*yield*/, (0, superposition_1.renderSuperposition)(this.plugin, this.props.segmentIndex, loadStrs)];
-                            case 16:
-                                _d.sent();
-                                _d.label = 17;
-                            case 17:
-                                customState.events.isBusy.next(false);
-                                return [2 /*return*/];
+        this.applyAction = async (e) => {
+            e.preventDefault();
+            e.currentTarget.blur();
+            const customState = this.customState;
+            customState.events.isBusy.next(true);
+            const currentState = { ...this.state };
+            this.setState({ showAll: false, showNone: false });
+            setTimeout(async () => {
+                let loadStrs = [];
+                for await (const str of this.state.cluster) {
+                    const structStateId = `${str.pdb_id}_${str.struct_asym_id}`;
+                    let structRef = undefined;
+                    if (customState && customState.superpositionState && customState.superpositionState.models[structStateId]) {
+                        structRef = this.customState.superpositionState.models[structStateId];
+                    }
+                    if (structRef) {
+                        const cell = this.plugin.state.data.cells.get(structRef);
+                        if (cell) {
+                            const isHidden = cell.state.isHidden ? true : false;
+                            if ((isHidden && currentState.showAll) || (!isHidden && currentState.showNone)) {
+                                await commands_1.PluginCommands.State.ToggleVisibility(this.plugin, { state: cell.parent, ref: structRef });
+                                // await PluginCommands.State.ToggleVisibility(this.plugin, { state: cell.parent!, ref: cell.transform.parent });
+                            }
                         }
-                    });
-                }); });
-                return [2 /*return*/];
+                    }
+                    else {
+                        if (currentState.showAll)
+                            loadStrs.push(str);
+                    }
+                }
+                ;
+                commands_1.PluginCommands.Camera.Reset(this.plugin);
+                if (loadStrs.length > 0) {
+                    await (0, superposition_1.renderSuperposition)(this.plugin, this.props.segmentIndex, loadStrs);
+                }
+                customState.events.isBusy.next(false);
             });
-        }); };
-        _this.cancelAction = function (e) {
+        };
+        this.cancelAction = (e) => {
             e.preventDefault();
-            _this.setState({ showAll: false, showNone: false });
+            this.setState({ showAll: false, showNone: false });
             e.currentTarget.blur();
         };
-        _this.clearSearch = function (e) {
+        this.clearSearch = (e) => {
             e.preventDefault();
-            _this.setState({ searchText: '' });
-            _this.inputStream.next('');
+            this.setState({ searchText: '' });
+            this.inputStream.next('');
             e.currentTarget.blur();
         };
-        return _this;
     }
-    ClusterNode.prototype.componentDidMount = function () {
-        var _this = this;
-        this.subscribe(this.plugin.customState.events.isBusy, function (e) {
-            _this.setState({ isBusy: e, showAll: false, showNone: false });
+    componentDidMount() {
+        this.subscribe(this.plugin.customState.events.isBusy, (e) => {
+            this.setState({ isBusy: e, showAll: false, showNone: false });
         });
-        this.subscribe(this.inputStream.pipe((0, operators_1.debounceTime)(1000 / 24)), function (e) { return _this.handleInputStream(e); });
-    };
-    Object.defineProperty(ClusterNode.prototype, "customState", {
-        get: function () {
-            return this.plugin.customState;
-        },
-        enumerable: false,
-        configurable: true
-    });
-    ClusterNode.prototype.render = function () {
-        var _this = this;
-        var customState = this.customState;
+        this.subscribe(this.inputStream.pipe((0, operators_1.debounceTime)(1000 / 24)), (e) => this.handleInputStream(e));
+    }
+    get customState() {
+        return this.plugin.customState;
+    }
+    render() {
+        const customState = this.customState;
         if (!customState.superpositionState || !customState.superpositionState.segmentData)
             return (0, jsx_runtime_1.jsx)(jsx_runtime_1.Fragment, {});
-        var expand = (0, jsx_runtime_1.jsx)(common_1.IconButton, { svg: this.state.isCollapsed ? icons_1.ArrowRightSvg : icons_1.ArrowDropDownSvg, flex: '20px', onClick: this.toggleExpanded, transparent: true, disabled: this.state.isBusy, className: 'msp-no-hover-outline' });
-        var title = "Segment ".concat(customState.superpositionState.activeSegment, " Cluster ").concat(this.props.clusterIndex + 1);
-        var label = (0, jsx_runtime_1.jsxs)(common_1.Button, tslib_1.__assign({ className: "msp-btn-tree-label", noOverflow: true, title: title, disabled: this.state.isBusy }, { children: [(0, jsx_runtime_1.jsxs)("span", { children: ["Cluster ", this.props.clusterIndex + 1] }), " ", (0, jsx_runtime_1.jsxs)("small", { children: [this.state.cluster.length < this.props.cluster.length ? "".concat(this.state.cluster.length, " / ") : '', this.props.cluster.length, " chain", this.props.cluster.length > 1 ? 's' : ''] })] }));
-        var selectionControls = (0, jsx_runtime_1.jsxs)(jsx_runtime_1.Fragment, { children: [(0, jsx_runtime_1.jsx)(common_1.Button, tslib_1.__assign({ icon: icons_1.CheckSvg, flex: true, onClick: this.selectAll, style: { flex: '0 0 50px', textAlign: 'center', fontSize: '80%', color: '#9cacc3', padding: 0 }, disabled: this.state.isBusy, title: "Show all chains" }, { children: "All" })), (0, jsx_runtime_1.jsx)(common_1.Button, tslib_1.__assign({ icon: icons_1.CloseSvg, flex: true, onClick: this.selectNone, style: { flex: '0 0 50px', textAlign: 'center', fontSize: '80%', color: '#9cacc3', padding: 0 }, disabled: this.state.isBusy, title: "Hide all chains" }, { children: "None" }))] });
-        var mainRow = (0, jsx_runtime_1.jsxs)("div", tslib_1.__assign({ className: "msp-flex-row msp-tree-row", style: { marginTop: '10px' } }, { children: [expand, label, this.props.cluster.length > 1 && selectionControls] }));
-        var searchControls = (0, jsx_runtime_1.jsxs)("div", tslib_1.__assign({ className: 'msp-mapped-parameter-group', style: { fontSize: '90%' } }, { children: [(0, jsx_runtime_1.jsxs)("div", tslib_1.__assign({ className: 'msp-control-row msp-transform-header-brand-gray', style: { height: '33px', marginLeft: '30px' } }, { children: [(0, jsx_runtime_1.jsx)("span", tslib_1.__assign({ className: 'msp-control-row-label' }, { children: "Search PDB ID" })), (0, jsx_runtime_1.jsx)("div", tslib_1.__assign({ className: 'msp-control-row-ctrl' }, { children: (0, jsx_runtime_1.jsx)("input", { type: 'text', placeholder: 'Enter PDB ID..', disabled: this.state.isBusy, onChange: function (e) { return _this.inputStream.next(e.target.value); }, value: this.state.searchText, maxLength: 4 }) }))] })), (0, jsx_runtime_1.jsx)(common_1.IconButton, { svg: icons_1.CloseSvg, flex: true, onClick: this.clearSearch, style: { flex: '0 0 24px', padding: 0 }, disabled: this.state.isBusy || this.state.searchText === '', toggleState: this.state.searchText !== '', title: 'Clear search input' })] }));
-        return (0, jsx_runtime_1.jsxs)(jsx_runtime_1.Fragment, { children: [mainRow, (this.state.showAll || this.state.showNone) && (0, jsx_runtime_1.jsx)("div", { children: (0, jsx_runtime_1.jsxs)("div", tslib_1.__assign({ className: "msp-control-row msp-transform-header-brand-".concat(this.state.showAll ? 'green' : 'red'), style: { display: 'flex', marginLeft: '20px', height: '35px' } }, { children: [(0, jsx_runtime_1.jsxs)("span", tslib_1.__assign({ className: 'msp-control-row-label', style: { flex: '1 1 auto', textAlign: 'left', fontSize: '85%' } }, { children: [this.state.showAll ? 'Display' : 'Hide', " ", this.state.cluster.length < this.props.cluster.length ? "".concat(this.state.cluster.length, " / ") : 'all ', this.props.cluster.length, " chains"] })), (0, jsx_runtime_1.jsx)(common_1.Button, tslib_1.__assign({ icon: icons_1.CheckSvg, flex: true, onClick: this.applyAction, style: { flex: '0 0 60px', textAlign: 'center', fontSize: '78%', color: '#9cacc3', padding: 0, margin: '0 1px' }, title: "Apply action" }, { children: "Apply" })), (0, jsx_runtime_1.jsx)(common_1.Button, tslib_1.__assign({ icon: icons_1.CloseSvg, flex: true, onClick: this.cancelAction, style: { flex: '0 0 60px', textAlign: 'center', fontSize: '78%', color: '#9cacc3', padding: 0, margin: '0 1px' }, title: "Cancel action" }, { children: "Cancel" }))] })) }), (!this.state.isCollapsed && this.props.cluster.length > 5) && searchControls, !this.state.isCollapsed && (0, jsx_runtime_1.jsx)("div", tslib_1.__assign({ className: 'msp-tree-updates-wrapper', style: { maxHeight: (this.props.totalClusters > 1) ? '330px' : '87%', overflowY: 'auto' } }, { children: this.state.cluster.map(function (s, i) { return (0, jsx_runtime_1.jsx)(StructureNode, { segmentIndex: _this.props.segmentIndex, structure: s, isRep: i === 0 ? true : false }, "str-".concat(s.pdb_id).concat(s.struct_asym_id).concat(i)); }) }))] });
-    };
-    return ClusterNode;
-}(base_1.PluginUIComponent));
-var StructureNode = /** @class */ (function (_super) {
-    tslib_1.__extends(StructureNode, _super);
-    function StructureNode() {
-        var _this = _super !== null && _super.apply(this, arguments) || this;
-        _this.state = {
+        const expand = (0, jsx_runtime_1.jsx)(common_1.IconButton, { svg: this.state.isCollapsed ? icons_1.ArrowRightSvg : icons_1.ArrowDropDownSvg, flex: '20px', onClick: this.toggleExpanded, transparent: true, disabled: this.state.isBusy, className: 'msp-no-hover-outline' });
+        const title = `Segment ${customState.superpositionState.activeSegment} Cluster ${this.props.clusterIndex + 1}`;
+        const label = (0, jsx_runtime_1.jsxs)(common_1.Button, { className: `msp-btn-tree-label`, noOverflow: true, title: title, disabled: this.state.isBusy, children: [(0, jsx_runtime_1.jsxs)("span", { children: ["Cluster ", this.props.clusterIndex + 1] }), " ", (0, jsx_runtime_1.jsxs)("small", { children: [this.state.cluster.length < this.props.cluster.length ? `${this.state.cluster.length} / ` : '', this.props.cluster.length, " chain", this.props.cluster.length > 1 ? 's' : ''] })] });
+        const selectionControls = (0, jsx_runtime_1.jsxs)(jsx_runtime_1.Fragment, { children: [(0, jsx_runtime_1.jsx)(common_1.Button, { icon: icons_1.CheckSvg, flex: true, onClick: this.selectAll, style: { flex: '0 0 50px', textAlign: 'center', fontSize: '80%', color: '#9cacc3', padding: 0 }, disabled: this.state.isBusy, title: `Show all chains`, children: "All" }), (0, jsx_runtime_1.jsx)(common_1.Button, { icon: icons_1.CloseSvg, flex: true, onClick: this.selectNone, style: { flex: '0 0 50px', textAlign: 'center', fontSize: '80%', color: '#9cacc3', padding: 0 }, disabled: this.state.isBusy, title: `Hide all chains`, children: "None" })] });
+        const mainRow = (0, jsx_runtime_1.jsxs)("div", { className: `msp-flex-row msp-tree-row`, style: { marginTop: '10px' }, children: [expand, label, this.props.cluster.length > 1 && selectionControls] });
+        const searchControls = (0, jsx_runtime_1.jsxs)("div", { className: 'msp-mapped-parameter-group', style: { fontSize: '90%' }, children: [(0, jsx_runtime_1.jsxs)("div", { className: 'msp-control-row msp-transform-header-brand-gray', style: { height: '33px', marginLeft: '30px' }, children: [(0, jsx_runtime_1.jsx)("span", { className: 'msp-control-row-label', children: "Search PDB ID" }), (0, jsx_runtime_1.jsx)("div", { className: 'msp-control-row-ctrl', children: (0, jsx_runtime_1.jsx)("input", { type: 'text', placeholder: 'Enter PDB ID..', disabled: this.state.isBusy, onChange: e => this.inputStream.next(e.target.value), value: this.state.searchText, maxLength: 4 }) })] }), (0, jsx_runtime_1.jsx)(common_1.IconButton, { svg: icons_1.CloseSvg, flex: true, onClick: this.clearSearch, style: { flex: '0 0 24px', padding: 0 }, disabled: this.state.isBusy || this.state.searchText === '', toggleState: this.state.searchText !== '', title: 'Clear search input' })] });
+        return (0, jsx_runtime_1.jsxs)(jsx_runtime_1.Fragment, { children: [mainRow, (this.state.showAll || this.state.showNone) && (0, jsx_runtime_1.jsx)("div", { children: (0, jsx_runtime_1.jsxs)("div", { className: `msp-control-row msp-transform-header-brand-${this.state.showAll ? 'green' : 'red'}`, style: { display: 'flex', marginLeft: '20px', height: '35px' }, children: [(0, jsx_runtime_1.jsxs)("span", { className: 'msp-control-row-label', style: { flex: '1 1 auto', textAlign: 'left', fontSize: '85%' }, children: [this.state.showAll ? 'Display' : 'Hide', " ", this.state.cluster.length < this.props.cluster.length ? `${this.state.cluster.length} / ` : 'all ', this.props.cluster.length, " chains"] }), (0, jsx_runtime_1.jsx)(common_1.Button, { icon: icons_1.CheckSvg, flex: true, onClick: this.applyAction, style: { flex: '0 0 60px', textAlign: 'center', fontSize: '78%', color: '#9cacc3', padding: 0, margin: '0 1px' }, title: `Apply action`, children: "Apply" }), (0, jsx_runtime_1.jsx)(common_1.Button, { icon: icons_1.CloseSvg, flex: true, onClick: this.cancelAction, style: { flex: '0 0 60px', textAlign: 'center', fontSize: '78%', color: '#9cacc3', padding: 0, margin: '0 1px' }, title: `Cancel action`, children: "Cancel" })] }) }), (!this.state.isCollapsed && this.props.cluster.length > 5) && searchControls, !this.state.isCollapsed && (0, jsx_runtime_1.jsx)("div", { className: 'msp-tree-updates-wrapper', style: { maxHeight: (this.props.totalClusters > 1) ? '330px' : '87%', overflowY: 'auto' }, children: this.state.cluster.map((s, i) => (0, jsx_runtime_1.jsx)(StructureNode, { segmentIndex: this.props.segmentIndex, structure: s, isRep: i === 0 ? true : false }, `str-${s.pdb_id}${s.struct_asym_id}${i}`)) })] });
+    }
+}
+class StructureNode extends base_1.PluginUIComponent {
+    constructor() {
+        super(...arguments);
+        this.state = {
             showControls: false,
             isBusy: false,
             isProcessing: false,
             isHidden: true,
         };
-        _this.toggleVisible = function (e) { return tslib_1.__awaiter(_this, void 0, void 0, function () {
-            var structHierarchy, _i, _a, c, currentHiddenState;
-            return tslib_1.__generator(this, function (_b) {
-                switch (_b.label) {
-                    case 0:
-                        e.preventDefault();
-                        e.currentTarget.blur();
-                        this.setState({ isProcessing: true, showControls: false });
-                        if (!this.ref) return [3 /*break*/, 1];
-                        structHierarchy = this.plugin.managers.structure.hierarchy.current.refs.get(this.ref);
-                        if (structHierarchy && structHierarchy.components) {
-                            for (_i = 0, _a = structHierarchy.components; _i < _a.length; _i++) {
-                                c = _a[_i];
-                                currentHiddenState = c.cell.state.isHidden ? true : false;
-                                if (currentHiddenState === this.state.isHidden) {
-                                    commands_1.PluginCommands.State.ToggleVisibility(this.plugin, { state: c.cell.parent, ref: c.cell.transform.ref });
-                                }
-                            }
-                            this.setState({ isHidden: !this.state.isHidden });
+        this.toggleVisible = async (e) => {
+            e.preventDefault();
+            e.currentTarget.blur();
+            this.setState({ isProcessing: true, showControls: false });
+            if (this.ref) {
+                const structHierarchy = this.plugin.managers.structure.hierarchy.current.refs.get(this.ref);
+                if (structHierarchy && structHierarchy.components) {
+                    for (const c of structHierarchy.components) {
+                        const currentHiddenState = c.cell.state.isHidden ? true : false;
+                        if (currentHiddenState === this.state.isHidden) {
+                            commands_1.PluginCommands.State.ToggleVisibility(this.plugin, { state: c.cell.parent, ref: c.cell.transform.ref });
                         }
-                        return [3 /*break*/, 3];
-                    case 1: return [4 /*yield*/, (0, superposition_1.renderSuperposition)(this.plugin, this.props.segmentIndex, [this.props.structure])];
-                    case 2:
-                        _b.sent();
-                        _b.label = 3;
-                    case 3:
-                        this.setState({ isProcessing: false });
-                        commands_1.PluginCommands.Camera.Reset(this.plugin);
-                        return [2 /*return*/];
+                    }
+                    this.setState({ isHidden: !this.state.isHidden });
                 }
-            });
-        }); };
-        _this.selectAction = function (item) {
+            }
+            else {
+                await (0, superposition_1.renderSuperposition)(this.plugin, this.props.segmentIndex, [this.props.structure]);
+            }
+            this.setState({ isProcessing: false });
+            commands_1.PluginCommands.Camera.Reset(this.plugin);
+        };
+        this.selectAction = item => {
             if (!item)
                 return;
-            _this.setState({ showControls: false });
-            (item === null || item === void 0 ? void 0 : item.value)();
+            this.setState({ showControls: false });
+            (item?.value)();
         };
-        _this.highlight = function (e) {
+        this.highlight = (e) => {
             e.preventDefault();
-            if (_this.ref) {
-                var cell = _this.plugin.state.data.cells.get(_this.ref);
-                commands_1.PluginCommands.Interactivity.Object.Highlight(_this.plugin, { state: cell.parent, ref: _this.ref });
+            if (this.ref) {
+                const cell = this.plugin.state.data.cells.get(this.ref);
+                commands_1.PluginCommands.Interactivity.Object.Highlight(this.plugin, { state: cell.parent, ref: this.ref });
             }
             e.currentTarget.blur();
         };
-        _this.clearHighlight = function (e) {
+        this.clearHighlight = (e) => {
             e.preventDefault();
-            commands_1.PluginCommands.Interactivity.ClearHighlights(_this.plugin);
+            commands_1.PluginCommands.Interactivity.ClearHighlights(this.plugin);
             e.currentTarget.blur();
         };
-        _this.toggleControls = function (e) {
+        this.toggleControls = (e) => {
             e.preventDefault();
-            _this.setState({ showControls: !_this.state.showControls });
+            this.setState({ showControls: !this.state.showControls });
             e.currentTarget.blur();
         };
-        return _this;
     }
-    Object.defineProperty(StructureNode.prototype, "customState", {
-        get: function () {
-            return this.plugin.customState;
-        },
-        enumerable: false,
-        configurable: true
-    });
-    Object.defineProperty(StructureNode.prototype, "ref", {
-        get: function () {
-            if (this.customState && this.customState.superpositionState && this.customState.superpositionState.models["".concat(this.props.structure.pdb_id, "_").concat(this.props.structure.struct_asym_id)]) {
-                return this.customState.superpositionState.models["".concat(this.props.structure.pdb_id, "_").concat(this.props.structure.struct_asym_id)];
-            }
-            else {
-                return undefined;
-            }
-        },
-        enumerable: false,
-        configurable: true
-    });
-    Object.defineProperty(StructureNode.prototype, "modelCell", {
-        get: function () {
-            if (this.ref) {
-                return this.plugin.state.data.cells.get(this.ref);
-            }
-            else {
-                return undefined;
-            }
-        },
-        enumerable: false,
-        configurable: true
-    });
-    Object.defineProperty(StructureNode.prototype, "isAllHidden", {
-        get: function () {
-            var isHidden = true;
-            if (this.ref) {
-                var structHierarchy = this.plugin.managers.structure.hierarchy.current.refs.get(this.ref);
-                if (structHierarchy && structHierarchy.components) {
-                    for (var _i = 0, _a = structHierarchy.components; _i < _a.length; _i++) {
-                        var c = _a[_i];
-                        if (c && c.cell && !c.cell.state.isHidden) {
-                            isHidden = false;
-                            break;
-                        }
+    get customState() {
+        return this.plugin.customState;
+    }
+    get ref() {
+        if (this.customState && this.customState.superpositionState && this.customState.superpositionState.models[`${this.props.structure.pdb_id}_${this.props.structure.struct_asym_id}`]) {
+            return this.customState.superpositionState.models[`${this.props.structure.pdb_id}_${this.props.structure.struct_asym_id}`];
+        }
+        else {
+            return undefined;
+        }
+    }
+    get modelCell() {
+        if (this.ref) {
+            return this.plugin.state.data.cells.get(this.ref);
+        }
+        else {
+            return undefined;
+        }
+    }
+    get isAllHidden() {
+        let isHidden = true;
+        if (this.ref) {
+            const structHierarchy = this.plugin.managers.structure.hierarchy.current.refs.get(this.ref);
+            if (structHierarchy && structHierarchy.components) {
+                for (const c of structHierarchy.components) {
+                    if (c && c.cell && !c.cell.state.isHidden) {
+                        isHidden = false;
+                        break;
                     }
                 }
-                else {
-                    isHidden = false;
-                }
             }
-            return isHidden;
-        },
-        enumerable: false,
-        configurable: true
-    });
-    StructureNode.prototype.checkRelation = function (ref) {
-        var isRelated = false;
-        var cell = this.plugin.state.data.cells.get(ref);
+            else {
+                isHidden = false;
+            }
+        }
+        return isHidden;
+    }
+    checkRelation(ref) {
+        let isRelated = false;
+        const cell = this.plugin.state.data.cells.get(ref);
         if (cell && cell.transform.parent) {
             if (cell && cell.transform.parent === this.ref) {
                 isRelated = true;
             }
             else {
-                var pcell = this.plugin.state.data.cells.get(cell.transform.parent);
+                const pcell = this.plugin.state.data.cells.get(cell.transform.parent);
                 if (pcell && pcell.transform.parent === this.ref)
                     isRelated = true;
             }
         }
         else {
-            var currentNodeCell = this.plugin.state.data.cells.get(this.ref);
+            const currentNodeCell = this.plugin.state.data.cells.get(this.ref);
             if (currentNodeCell && currentNodeCell.transform.parent === cell.transform.parent) {
                 isRelated = true;
             }
         }
         return isRelated;
-    };
-    StructureNode.prototype.is = function (e) {
+    }
+    is(e) {
         if (!this.ref)
             return false;
-        var isRelated = false;
+        let isRelated = false;
         if (this.ref && e.ref !== this.ref) {
             isRelated = this.checkRelation(e.ref);
         }
@@ -604,31 +477,30 @@ var StructureNode = /** @class */ (function (_super) {
             return true;
         }
         else {
-            var invalidStruct = (this.customState.superpositionState.invalidStruct.indexOf("".concat(this.props.structure.pdb_id, "_").concat(this.props.structure.struct_asym_id)) > -1) ? true : false;
+            const invalidStruct = (this.customState.superpositionState.invalidStruct.indexOf(`${this.props.structure.pdb_id}_${this.props.structure.struct_asym_id}`) > -1) ? true : false;
             return invalidStruct ? true : false;
         }
-    };
-    StructureNode.prototype.componentDidMount = function () {
-        var _this = this;
+    }
+    componentDidMount() {
         this.setState({ isHidden: this.isAllHidden });
-        this.subscribe(this.plugin.customState.events.isBusy, function (e) {
-            _this.setState({ isBusy: e, showControls: false });
+        this.subscribe(this.plugin.customState.events.isBusy, (e) => {
+            this.setState({ isBusy: e, showControls: false });
         });
-        this.subscribe(this.plugin.state.events.cell.stateUpdated.pipe((0, operators_1.filter)(function (e) { return _this.is(e); }), (0, operators_1.debounceTime)(33)), function (e) {
-            _this.setState({ isHidden: _this.isAllHidden });
+        this.subscribe(this.plugin.state.events.cell.stateUpdated.pipe((0, operators_1.filter)(e => this.is(e)), (0, operators_1.debounceTime)(33)), e => {
+            this.setState({ isHidden: this.isAllHidden });
             // this.forceUpdate();
         });
-    };
-    StructureNode.prototype.getTagRefs = function (tags) {
-        var TagSet = new Set(tags);
-        var tree = this.plugin.state.data.tree;
+    }
+    getTagRefs(tags) {
+        const TagSet = new Set(tags);
+        const tree = this.plugin.state.data.tree;
         return mol_state_2.StateSelection.findUniqueTagsInSubtree(tree, this.modelCell.transform.ref, TagSet);
-    };
-    StructureNode.prototype.getRandomColor = function () {
-        var clList = lists_1.ColorLists;
-        var spState = this.plugin.customState.superpositionState;
-        var palleteIndex = spState.colorState[this.props.segmentIndex].palleteIndex;
-        var colorIndex = spState.colorState[this.props.segmentIndex].colorIndex;
+    }
+    getRandomColor() {
+        const clList = lists_1.ColorLists;
+        const spState = this.plugin.customState.superpositionState;
+        let palleteIndex = spState.colorState[this.props.segmentIndex].palleteIndex;
+        let colorIndex = spState.colorState[this.props.segmentIndex].colorIndex;
         if (clList[spState.colorPalette[palleteIndex]].list[colorIndex + 1]) {
             colorIndex += 1;
         }
@@ -636,169 +508,145 @@ var StructureNode = /** @class */ (function (_super) {
             colorIndex = 0;
             palleteIndex = spState.colorPalette[palleteIndex + 1] ? palleteIndex + 1 : 0;
         }
-        var palleteName = spState.colorPalette[palleteIndex];
+        const palleteName = spState.colorPalette[palleteIndex];
         this.plugin.customState.superpositionState.colorState[this.props.segmentIndex].palleteIndex = palleteIndex;
         this.plugin.customState.superpositionState.colorState[this.props.segmentIndex].colorIndex = colorIndex;
         return clList[palleteName].list[colorIndex];
-    };
-    StructureNode.prototype.addChainRepr = function () {
-        return tslib_1.__awaiter(this, void 0, void 0, function () {
-            var uniformColor1, strInstance, query, chainSel;
-            return tslib_1.__generator(this, function (_a) {
-                switch (_a.label) {
-                    case 0:
-                        uniformColor1 = this.getRandomColor();
-                        strInstance = this.plugin.state.data.select(this.ref)[0];
-                        query = builder_1.MolScriptBuilder.struct.generator.atomGroups({
-                            'chain-test': builder_1.MolScriptBuilder.core.rel.eq([builder_1.MolScriptBuilder.struct.atomProperty.macromolecular.label_asym_id(), this.props.structure.struct_asym_id])
-                        });
-                        return [4 /*yield*/, this.plugin.builders.structure.tryCreateComponentFromExpression(strInstance, query, "Chain-".concat(this.props.segmentIndex), { label: "Chain", tags: ["superposition-sel"] })];
-                    case 1:
-                        chainSel = _a.sent();
-                        if (!chainSel) return [3 /*break*/, 3];
-                        return [4 /*yield*/, this.plugin.builders.structure.representation.addRepresentation(chainSel, { type: 'cartoon', color: 'uniform', colorParams: { value: uniformColor1 } }, { tag: "superposition-visual" })];
-                    case 2:
-                        _a.sent();
-                        _a.label = 3;
-                    case 3: return [2 /*return*/];
-                }
-            });
+    }
+    async addChainRepr() {
+        const uniformColor1 = this.getRandomColor();
+        const strInstance = this.plugin.state.data.select(this.ref)[0];
+        const query = builder_1.MolScriptBuilder.struct.generator.atomGroups({
+            'chain-test': builder_1.MolScriptBuilder.core.rel.eq([builder_1.MolScriptBuilder.struct.atomProperty.macromolecular.label_asym_id(), this.props.structure.struct_asym_id])
         });
-    };
-    StructureNode.prototype.updates = function () {
-        var _this = this;
-        var structHierarchy = this.plugin.managers.structure.hierarchy.current.refs.get(this.ref);
+        const chainSel = await this.plugin.builders.structure.tryCreateComponentFromExpression(strInstance, query, `Chain-${this.props.segmentIndex}`, { label: `Chain`, tags: [`superposition-sel`] });
+        if (chainSel) {
+            await this.plugin.builders.structure.representation.addRepresentation(chainSel, { type: 'cartoon', color: 'uniform', colorParams: { value: uniformColor1 } }, { tag: `superposition-visual` });
+        }
+    }
+    updates() {
+        const structHierarchy = this.plugin.managers.structure.hierarchy.current.refs.get(this.ref);
         if (structHierarchy && structHierarchy.components) {
-            var representations_1 = [];
-            var showAddChainBtn_1 = true;
-            structHierarchy.components.forEach(function (comps) {
-                var gKeys = comps.key.split(',');
-                var cId1Arr = gKeys[0].split('-');
+            const representations = [];
+            let showAddChainBtn = true;
+            structHierarchy.components.forEach((comps) => {
+                const gKeys = comps.key.split(',');
+                const cId1Arr = gKeys[0].split('-');
                 if (cId1Arr[2] === 'Chain')
-                    showAddChainBtn_1 = false;
+                    showAddChainBtn = false;
                 if (comps.representations) {
-                    comps.representations.forEach(function (repr) {
-                        representations_1.push(repr);
+                    comps.representations.forEach((repr) => {
+                        representations.push(repr);
                     });
                 }
             });
-            var customState = this.plugin.customState;
+            const customState = this.plugin.customState;
             if (customState.initParams && customState.initParams.superpositionParams && !customState.initParams.superpositionParams.ligandView) {
-                showAddChainBtn_1 = false;
+                showAddChainBtn = false;
             }
-            if (representations_1.length > 0) {
-                return (0, jsx_runtime_1.jsxs)("div", tslib_1.__assign({ className: 'msp-accent-offset', style: { marginLeft: '40px' } }, { children: [representations_1.length > 0 && representations_1.map(function (r, i) { return (0, jsx_runtime_1.jsx)(StructureRepresentationEntry, { group: [structHierarchy], representation: r }, "".concat(r.cell.transform.ref, "-").concat(i)); }), showAddChainBtn_1 && (0, jsx_runtime_1.jsx)("div", tslib_1.__assign({ className: 'msp-control-group-header', style: { marginTop: '1px' } }, { children: (0, jsx_runtime_1.jsxs)(common_1.Button, tslib_1.__assign({ noOverflow: true, className: 'msp-control-button-label', title: "Click to add chain representaion", onClick: function () { return _this.addChainRepr(); } }, { children: ["\u00A0\u00A0Add Chain ", this.props.structure.struct_asym_id, " Representation"] })) }))] }));
+            if (representations.length > 0) {
+                return (0, jsx_runtime_1.jsxs)("div", { className: 'msp-accent-offset', style: { marginLeft: '40px' }, children: [representations.length > 0 && representations.map((r, i) => (0, jsx_runtime_1.jsx)(StructureRepresentationEntry, { group: [structHierarchy], representation: r }, `${r.cell.transform.ref}-${i}`)), showAddChainBtn && (0, jsx_runtime_1.jsx)("div", { className: 'msp-control-group-header', style: { marginTop: '1px' }, children: (0, jsx_runtime_1.jsxs)(common_1.Button, { noOverflow: true, className: 'msp-control-button-label', title: `Click to add chain representaion`, onClick: () => this.addChainRepr(), children: ["\u00A0\u00A0Add Chain ", this.props.structure.struct_asym_id, " Representation"] }) })] });
             }
         }
         return (0, jsx_runtime_1.jsx)(jsx_runtime_1.Fragment, {});
-    };
-    StructureNode.prototype.getSubtitle = function () {
-        var customState = this.plugin.customState;
-        var hetList = customState.superpositionState.hets["".concat(this.props.structure.pdb_id, "_").concat(this.props.structure.struct_asym_id)];
-        var subtitle;
+    }
+    getSubtitle() {
+        const customState = this.plugin.customState;
+        const hetList = customState.superpositionState.hets[`${this.props.structure.pdb_id}_${this.props.structure.struct_asym_id}`];
+        let subtitle;
         if (hetList) {
-            var hetLimit = this.props.structure.is_representative ? 1 : 4;
-            var totalHets = hetList.length;
-            var hetStr = hetList.join(', ');
+            const hetLimit = this.props.structure.is_representative ? 1 : 4;
+            const totalHets = hetList.length;
+            let hetStr = hetList.join(', ');
             if (totalHets > hetLimit) {
                 hetStr = hetList.slice(0, hetLimit).join(', ');
-                hetStr += " + ".concat(totalHets - hetLimit);
+                hetStr += ` + ${totalHets - hetLimit}`;
             }
-            subtitle = " ( ".concat(hetStr, " )");
+            subtitle = ` ( ${hetStr} )`;
             if (this.props.structure.is_representative)
-                subtitle = " ".concat(subtitle, " ( Representative )");
+                subtitle = ` ${subtitle} ( Representative )`;
         }
         else if (this.props.structure.is_representative) {
             subtitle = ' ( Representative )';
         }
         return subtitle;
-    };
-    Object.defineProperty(StructureNode.prototype, "panelColor", {
-        get: function () {
-            var panelColor = '#808080';
-            if (!this.state.isHidden) {
-                if (this.modelCell) {
-                    var refs = this.getTagRefs(["superposition-visual", "superposition-ligand-visual"]);
-                    var visualRef = refs["superposition-ligand-visual"] ? refs["superposition-ligand-visual"] : refs["superposition-visual"] ? refs["superposition-visual"] : undefined;
-                    if (visualRef) {
-                        var visualCell = this.plugin.state.data.cells.get(visualRef);
-                        if (visualCell.params && visualCell.params.values && visualCell.params.values.colorTheme) {
-                            var colorTheme = visualCell.params.values.colorTheme;
-                            if (colorTheme.params && colorTheme.params.value) {
-                                panelColor = "".concat(color_1.Color.toStyle(colorTheme.params.value));
-                            }
-                            else if (colorTheme.params && colorTheme.params.palette) {
-                                var colorList1 = colorTheme.params.palette.params.list.colors;
-                                panelColor = "".concat(color_1.Color.toStyle(colorList1[0]));
-                            }
-                            else if (colorTheme.params && colorTheme.params.list) {
-                                var colorList2 = colorTheme.params.list.colors;
-                                panelColor = "".concat(color_1.Color.toStyle(colorList2[0]));
-                            }
+    }
+    get panelColor() {
+        let panelColor = '#808080';
+        if (!this.state.isHidden) {
+            if (this.modelCell) {
+                const refs = this.getTagRefs([`superposition-visual`, `superposition-ligand-visual`]);
+                const visualRef = refs[`superposition-ligand-visual`] ? refs[`superposition-ligand-visual`] : refs[`superposition-visual`] ? refs[`superposition-visual`] : undefined;
+                if (visualRef) {
+                    const visualCell = this.plugin.state.data.cells.get(visualRef);
+                    if (visualCell.params && visualCell.params.values && visualCell.params.values.colorTheme) {
+                        const colorTheme = visualCell.params.values.colorTheme;
+                        if (colorTheme.params && colorTheme.params.value) {
+                            panelColor = `${color_1.Color.toStyle(colorTheme.params.value)}`;
+                        }
+                        else if (colorTheme.params && colorTheme.params.palette) {
+                            const colorList1 = colorTheme.params.palette.params.list.colors;
+                            panelColor = `${color_1.Color.toStyle(colorList1[0])}`;
+                        }
+                        else if (colorTheme.params && colorTheme.params.list) {
+                            const colorList2 = colorTheme.params.list.colors;
+                            panelColor = `${color_1.Color.toStyle(colorList2[0])}`;
                         }
                     }
                 }
             }
-            return panelColor;
-        },
-        enumerable: false,
-        configurable: true
-    });
-    StructureNode.prototype.render = function () {
-        var superpositionParams = this.customState.initParams.superpositionParams;
-        var strutStateId = "".concat(this.props.structure.pdb_id, "_").concat(this.props.structure.struct_asym_id);
-        var invalidStruct = (this.customState.superpositionState.invalidStruct.indexOf(strutStateId) > -1) ? true : false;
-        var noMatrixStruct = (this.customState.superpositionState.noMatrixStruct.indexOf(strutStateId) > -1) ? true : false;
-        var subTitle = invalidStruct ? noMatrixStruct ? " Matrix not available!" : " No Ligand found!" : this.getSubtitle();
-        var strTitle = "".concat(this.props.structure.pdb_id, " chain ").concat(this.props.structure.auth_asym_id);
+        }
+        return panelColor;
+    }
+    render() {
+        const superpositionParams = this.customState.initParams.superpositionParams;
+        const strutStateId = `${this.props.structure.pdb_id}_${this.props.structure.struct_asym_id}`;
+        const invalidStruct = (this.customState.superpositionState.invalidStruct.indexOf(strutStateId) > -1) ? true : false;
+        const noMatrixStruct = (this.customState.superpositionState.noMatrixStruct.indexOf(strutStateId) > -1) ? true : false;
+        const subTitle = invalidStruct ? noMatrixStruct ? ` Matrix not available!` : ` No Ligand found!` : this.getSubtitle();
+        let strTitle = `${this.props.structure.pdb_id} chain ${this.props.structure.auth_asym_id}`;
         if (superpositionParams && superpositionParams.ligandView) {
-            strTitle = "".concat(this.props.structure.pdb_id, " ").concat(this.props.structure.struct_asym_id);
+            strTitle = `${this.props.structure.pdb_id} ${this.props.structure.struct_asym_id}`;
         }
-        var label = (0, jsx_runtime_1.jsxs)(common_1.Button, tslib_1.__assign({ className: "msp-btn-tree-label", style: { borderLeftColor: this.panelColor }, noOverflow: true, title: strTitle, disabled: (invalidStruct || this.state.isBusy || this.state.isProcessing) ? true : false, onMouseEnter: this.highlight, onMouseLeave: this.clearHighlight }, { children: [(0, jsx_runtime_1.jsx)("span", { children: strTitle }), subTitle && (0, jsx_runtime_1.jsx)("small", { children: subTitle })] }));
-        var expand = (0, jsx_runtime_1.jsx)(common_1.IconButton, { svg: !this.state.showControls ? icons_1.ArrowRightSvg : icons_1.ArrowDropDownSvg, flex: '20px', onClick: this.toggleControls, transparent: true, className: 'msp-no-hover-outline', disabled: (invalidStruct || this.state.isBusy || this.state.isProcessing) ? true : false });
-        var visibility = (0, jsx_runtime_1.jsx)(common_1.IconButton, { svg: this.state.isHidden ? icons_1.VisibilityOffOutlinedSvg : icons_1.VisibilityOutlinedSvg, toggleState: false, small: true, onClick: this.toggleVisible, disabled: (invalidStruct || this.state.isBusy || this.state.isProcessing) ? true : false, title: this.state.isHidden ? "Show chain" : "Hide chain" });
-        var row = (0, jsx_runtime_1.jsxs)("div", tslib_1.__assign({ className: "msp-flex-row msp-tree-row", style: { marginLeft: !this.state.isHidden ? '10px' : '31px' } }, { children: [!this.state.isHidden && expand, label, visibility] }));
-        return (0, jsx_runtime_1.jsxs)("div", tslib_1.__assign({ style: { marginBottom: '1px' } }, { children: [row, this.state.showControls && this.updates()] }));
-    };
-    return StructureNode;
-}(base_1.PluginUIComponent));
-var StructureRepresentationEntry = /** @class */ (function (_super) {
-    tslib_1.__extends(StructureRepresentationEntry, _super);
-    function StructureRepresentationEntry() {
-        var _this = _super !== null && _super.apply(this, arguments) || this;
-        _this.toggleVisible = function (e) {
+        const label = (0, jsx_runtime_1.jsxs)(common_1.Button, { className: `msp-btn-tree-label`, style: { borderLeftColor: this.panelColor }, noOverflow: true, title: strTitle, disabled: (invalidStruct || this.state.isBusy || this.state.isProcessing) ? true : false, onMouseEnter: this.highlight, onMouseLeave: this.clearHighlight, children: [(0, jsx_runtime_1.jsx)("span", { children: strTitle }), subTitle && (0, jsx_runtime_1.jsx)("small", { children: subTitle })] });
+        const expand = (0, jsx_runtime_1.jsx)(common_1.IconButton, { svg: !this.state.showControls ? icons_1.ArrowRightSvg : icons_1.ArrowDropDownSvg, flex: '20px', onClick: this.toggleControls, transparent: true, className: 'msp-no-hover-outline', disabled: (invalidStruct || this.state.isBusy || this.state.isProcessing) ? true : false });
+        const visibility = (0, jsx_runtime_1.jsx)(common_1.IconButton, { svg: this.state.isHidden ? icons_1.VisibilityOffOutlinedSvg : icons_1.VisibilityOutlinedSvg, toggleState: false, small: true, onClick: this.toggleVisible, disabled: (invalidStruct || this.state.isBusy || this.state.isProcessing) ? true : false, title: this.state.isHidden ? `Show chain` : `Hide chain` });
+        const row = (0, jsx_runtime_1.jsxs)("div", { className: `msp-flex-row msp-tree-row`, style: { marginLeft: !this.state.isHidden ? '10px' : '31px' }, children: [!this.state.isHidden && expand, label, visibility] });
+        return (0, jsx_runtime_1.jsxs)("div", { style: { marginBottom: '1px' }, children: [row, this.state.showControls && this.updates()] });
+    }
+}
+class StructureRepresentationEntry extends base_1.PurePluginUIComponent {
+    constructor() {
+        super(...arguments);
+        this.toggleVisible = (e) => {
             e.preventDefault();
             e.currentTarget.blur();
-            var cell = _this.props.representation.cell;
-            commands_1.PluginCommands.State.ToggleVisibility(_this.plugin, { state: cell.parent, ref: cell.transform.parent });
+            const cell = this.props.representation.cell;
+            commands_1.PluginCommands.State.ToggleVisibility(this.plugin, { state: cell.parent, ref: cell.transform.parent });
         };
-        return _this;
     }
-    StructureRepresentationEntry.prototype.componentDidMount = function () {
-        var _this = this;
-        this.subscribe(this.plugin.state.events.cell.stateUpdated, function (e) {
-            if (mol_state_1.State.ObjectEvent.isCell(e, _this.props.representation.cell))
-                _this.forceUpdate();
+    componentDidMount() {
+        this.subscribe(this.plugin.state.events.cell.stateUpdated, e => {
+            if (mol_state_1.State.ObjectEvent.isCell(e, this.props.representation.cell))
+                this.forceUpdate();
         });
-    };
-    StructureRepresentationEntry.prototype.render = function () {
-        var _a, _b, _c, _d, _e, _f, _g;
-        var repr = this.props.representation.cell;
-        var label = (_a = repr.obj) === null || _a === void 0 ? void 0 : _a.label;
-        if (((_b = repr.obj) === null || _b === void 0 ? void 0 : _b.data.repr) && ((_c = repr.obj) === null || _c === void 0 ? void 0 : _c.data.repr.label)) {
-            var sourceLabel = (((_d = repr.obj) === null || _d === void 0 ? void 0 : _d.data.repr.label.indexOf('[Focus]')) >= 0) ? '[Focus]' : (_e = repr.obj) === null || _e === void 0 ? void 0 : _e.data.repr.label;
-            var isLargeLabel = sourceLabel.length > 10 ? true : false;
-            sourceLabel = "".concat(isLargeLabel ? "".concat(sourceLabel.substring(0, 28), "...") : sourceLabel);
+    }
+    render() {
+        const repr = this.props.representation.cell;
+        let label = repr.obj?.label;
+        if (repr.obj?.data.repr && repr.obj?.data.repr.label) {
+            let sourceLabel = (repr.obj?.data.repr.label.indexOf('[Focus]') >= 0) ? '[Focus]' : repr.obj?.data.repr.label;
+            const isLargeLabel = sourceLabel.length > 10 ? true : false;
+            sourceLabel = `${isLargeLabel ? `${sourceLabel.substring(0, 28)}...` : sourceLabel}`;
             if (isLargeLabel) {
                 label = sourceLabel;
             }
             else {
-                label = "".concat(sourceLabel, " ").concat((label && label.length < 21) ? ' - ' + label : '');
+                label = `${sourceLabel} ${(label && label.length < 21) ? ' - ' + label : ''}`;
             }
         }
-        if (((_f = repr.obj) === null || _f === void 0 ? void 0 : _f.data.repr) && ((_g = repr.obj) === null || _g === void 0 ? void 0 : _g.data.repr.label) === 'Custom Selection')
+        if (repr.obj?.data.repr && repr.obj?.data.repr.label === 'Custom Selection')
             label = 'Custom Selection';
-        return (0, jsx_runtime_1.jsxs)("div", tslib_1.__assign({ className: 'msp-representation-entry' }, { children: [repr.parent && (0, jsx_runtime_1.jsx)(common_1.ExpandGroup, tslib_1.__assign({ header: "".concat(label || 'Representation'), noOffset: true, headerStyle: { overflow: 'hidden' } }, { children: (0, jsx_runtime_1.jsx)(update_transform_1.UpdateTransformControl, { state: repr.parent, transform: repr.transform, customHeader: 'none', noMargin: true }) })), (0, jsx_runtime_1.jsx)(common_1.IconButton, { svg: this.props.representation.cell.state.isHidden ? icons_1.VisibilityOffOutlinedSvg : icons_1.VisibilityOutlinedSvg, toggleState: false, onClick: this.toggleVisible, title: this.props.representation.cell.state.isHidden ? "Show representation" : "Hide representation", small: true, className: 'msp-default-bg', style: { position: 'absolute', top: 0, right: 0, lineHeight: '24px', height: '24px', textAlign: 'right', width: '32px', paddingRight: '6px', background: 'none' } })] }));
-    };
-    return StructureRepresentationEntry;
-}(base_1.PurePluginUIComponent));
+        return (0, jsx_runtime_1.jsxs)("div", { className: 'msp-representation-entry', children: [repr.parent && (0, jsx_runtime_1.jsx)(common_1.ExpandGroup, { header: `${label || 'Representation'}`, noOffset: true, headerStyle: { overflow: 'hidden' }, children: (0, jsx_runtime_1.jsx)(update_transform_1.UpdateTransformControl, { state: repr.parent, transform: repr.transform, customHeader: 'none', noMargin: true }) }), (0, jsx_runtime_1.jsx)(common_1.IconButton, { svg: this.props.representation.cell.state.isHidden ? icons_1.VisibilityOffOutlinedSvg : icons_1.VisibilityOutlinedSvg, toggleState: false, onClick: this.toggleVisible, title: this.props.representation.cell.state.isHidden ? `Show representation` : `Hide representation`, small: true, className: 'msp-default-bg', style: { position: 'absolute', top: 0, right: 0, lineHeight: '24px', height: '24px', textAlign: 'right', width: '32px', paddingRight: '6px', background: 'none' } })] });
+    }
+}