Fix update issues when component props change (ohif-cornerstone-extension: 0.0.7, ohif-viewer 0.0.9)

Packages-react/extensions/ohif-cornerstone-extension/src/OHIFCornerstoneViewport.js

@@ -125,7 +125,7 @@ class OHIFCornerstoneViewport extends Component {
     return viewportData;
   };
 
-  componentDidMount() {
+  setStateFromProps() {
     const { studies, displaySet } = this.props.viewportData;
     const {
       studyInstanceUid,
@@ -153,6 +153,20 @@ class OHIFCornerstoneViewport extends Component {
     });
   }
 
+  componentDidMount() {
+    this.setStateFromProps();
+  }
+
+  componentDidUpdate(prevProps) {
+    const { studies, displaySet } = this.props.viewportData;
+    const { displaySetInstanceUid } = displaySet;
+    const prevDisplaySet = prevProps.viewportData.displaySet;
+
+    if (displaySet.displaySetInstanceUid !== prevDisplaySet.displaySetInstanceUid) {
+      this.setStateFromProps();
+    }
+  }
+
   render() {
     return (<>
       {this.state.viewportData && (