fix: Revert "refactor: Reduce bundle size (#1575)" (#1622)

This reverts commit 81dcbe294d.

platform/core/src/classes/StudyLoadingListener.js

@@ -1,10 +1,10 @@
 import cornerstone from 'cornerstone-core';
+import cornerstoneWADOImageLoader from 'cornerstone-wado-image-loader';
 import {
   clearStudyLoadingProgress,
   setStudyLoadingProgress,
 } from '../redux/actions';
 import StackManager from '../utils/StackManager';
-import { getCornerstoneWADOImageLoader } from '../utils/cornerstoneWADOImageLoader';
 
 class BaseLoadingListener {
   constructor(stack, options = {}) {
@@ -103,8 +103,7 @@ class DICOMFileLoadingListener extends BaseLoadingListener {
     this._checkCachedData();
   }
 
-  async _checkCachedData() {
-    const cornerstoneWADOImageLoader = await getCornerstoneWADOImageLoader();
+  _checkCachedData() {
     const dataSet = cornerstoneWADOImageLoader.wadouri.dataSetCacheManager.get(
       this._dataSetUrl
     );

platform/core/src/studies/services/wado/studyInstanceHelpers.js

@@ -1,8 +1,8 @@
 import DICOMWeb from '../../../DICOMWeb';
 import metadataProvider from '../../../classes/MetadataProvider';
 import getWADORSImageId from '../../../utils/getWADORSImageId';
+import cornerstoneWADOImageLoader from 'cornerstone-wado-image-loader';
 import getReferencedSeriesSequence from './getReferencedSeriesSequence';
-import { getCornerstoneWADOImageLoader } from '../../../utils/cornerstoneWADOImageLoader';
 
 /**
  * Create a plain JS object that describes a study (a study descriptor object)
@@ -160,7 +160,6 @@ async function makeSOPInstance(server, study, instance) {
 
     const wadoRSMetadata = Object.assign(instance);
 
-    const cornerstoneWADOImageLoader = await getCornerstoneWADOImageLoader();
     if (sopInstance.NumberOfFrames) {
       for (let i = 0; i < sopInstance.NumberOfFrames; i++) {
         const wadorsImageId = getWADORSImageId(sopInstance, i);

platform/core/src/utils/cornerstoneWADOImageLoader.js

@@ -1,5 +0,0 @@
-export function getCornerstoneWADOImageLoader() {
-  return import(
-    /* webpackChunkName: "CornerstoneWADOImageLoader" */ 'cornerstone-wado-image-loader'
-  );
-}

platform/core/src/utils/dicomLoaderService.js

@@ -1,7 +1,7 @@
 import cornerstone from 'cornerstone-core';
+import cornerstoneWADOImageLoader from 'cornerstone-wado-image-loader';
 import { api } from 'dicomweb-client';
 import DICOMWeb from '../DICOMWeb';
-import { getCornerstoneWADOImageLoader } from './cornerstoneWADOImageLoader';
 
 const getImageId = imageObj => {
   if (!imageObj) {
@@ -86,7 +86,7 @@ const getImageLoaderType = imageId => {
 };
 
 class DicomLoaderService {
-  async getLocalData(dataset, studies) {
+  getLocalData(dataset, studies) {
     if (dataset && dataset.localFile) {
       // Use referenced imageInstance
       const imageInstance = getImageInstance(dataset);
@@ -98,7 +98,6 @@ class DicomLoaderService {
       }
 
       if (!someInvalidStrings(imageId)) {
-        const cornerstoneWADOImageLoader = await getCornerstoneWADOImageLoader();
         return cornerstoneWADOImageLoader.wadouri.loadFileRequest(imageId);
       }
     }
@@ -177,16 +176,16 @@ class DicomLoaderService {
     }
   }
 
-  async *getLoaderIterator(dataset, studies) {
-    yield await this.getLocalData(dataset, studies);
+  *getLoaderIterator(dataset, studies) {
+    yield this.getLocalData(dataset, studies);
     yield this.getDataByImageType(dataset);
     yield this.getDataByDatasetType(dataset);
   }
 
-  async findDicomDataPromise(dataset, studies) {
+  findDicomDataPromise(dataset, studies) {
     const loaderIterator = this.getLoaderIterator(dataset, studies);
     // it returns first valid retriever method.
-    for await (const loader of loaderIterator) {
+    for (const loader of loaderIterator) {
       if (loader) {
         return loader;
       }

platform/viewer/src/App.js

@@ -34,7 +34,10 @@ import i18n from '@ohif/i18n';
 import './config';
 
 /** Utils */
-import { getUserManagerForOpenIdConnectClient } from './utils/index.js';
+import {
+  getUserManagerForOpenIdConnectClient,
+  initWebWorkers,
+} from './utils/index.js';
 
 /** Extensions */
 import { GenericViewerCommands, MeasurementsPanel } from './appExtensions';
@@ -137,6 +140,7 @@ class App extends Component {
      */
     _initHotkeys(appConfigHotkeys);
     _initServers(servers);
+    initWebWorkers();
   }
 
   render() {

platform/viewer/src/config.js

@@ -1,4 +1,7 @@
+import OHIF from '@ohif/core';
 import cornerstone from 'cornerstone-core';
+import cornerstoneWADOImageLoader from 'cornerstone-wado-image-loader';
+import dicomParser from 'dicom-parser';
 import version from './version.js';
 
 let homepage;
@@ -12,4 +15,31 @@ window.info = {
   homepage,
 };
 
+// For debugging
+//if (process.env.node_env === 'development') {
 window.cornerstone = cornerstone;
+window.cornerstoneWADOImageLoader = cornerstoneWADOImageLoader;
+//}
+
+cornerstoneWADOImageLoader.external.cornerstone = cornerstone;
+cornerstoneWADOImageLoader.external.dicomParser = dicomParser;
+
+OHIF.user.getAccessToken = () => {
+  // TODO: Get the Redux store from somewhere else
+  const state = window.store.getState();
+  if (!state.oidc || !state.oidc.user) {
+    return;
+  }
+
+  return state.oidc.user.access_token;
+};
+
+cornerstoneWADOImageLoader.configure({
+  beforeSend: function(xhr) {
+    const headers = OHIF.DICOMWeb.getAuthorizationHeader();
+
+    if (headers.Authorization) {
+      xhr.setRequestHeader('Authorization', headers.Authorization);
+    }
+  },
+});

platform/viewer/src/lib/filesToStudies.js

@@ -3,7 +3,7 @@ import FileLoaderService from './localFileLoaders/fileLoaderService';
 const processFile = async file => {
   try {
     const fileLoaderService = new FileLoaderService(file);
-    const imageId = await fileLoaderService.addFile(file);
+    const imageId = fileLoaderService.addFile(file);
     const image = await fileLoaderService.loadFile(file, imageId);
     const dataset = await fileLoaderService.getDataset(image, imageId);
     const studies = await fileLoaderService.getStudies(dataset, imageId);

platform/viewer/src/lib/localFileLoaders/dicomFileLoader.js

@@ -1,14 +1,13 @@
 import * as dcmjs from 'dcmjs';
+import cornerstoneWADOImageLoader from 'cornerstone-wado-image-loader';
 import FileLoader from './fileLoader';
 import OHIF from '@ohif/core';
-import { getCornerstoneWADOImageLoader } from '../../utils/cornerstoneWADOImageLoader';
 
 const metadataProvider = OHIF.cornerstone.metadataProvider;
 
 const DICOMFileLoader = new (class extends FileLoader {
   fileType = 'application/dicom';
-  async loadFile(file, imageId) {
-    const cornerstoneWADOImageLoader = await getCornerstoneWADOImageLoader();
+  loadFile(file, imageId) {
     return cornerstoneWADOImageLoader.wadouri.loadFileRequest(imageId);
   }
 

platform/viewer/src/lib/localFileLoaders/fileLoaderService.js

@@ -1,7 +1,7 @@
+import cornerstoneWADOImageLoader from 'cornerstone-wado-image-loader';
 import FileLoader from './fileLoader';
 import PDFFileLoader from './pdfFileLoader';
 import DICOMFileLoader from './dicomFileLoader';
-import { getCornerstoneWADOImageLoader } from '../../utils/cornerstoneWADOImageLoader';
 
 class FileLoaderService extends FileLoader {
   fileType;
@@ -53,8 +53,7 @@ class FileLoaderService extends FileLoader {
     return result;
   }
 
-  async addFile(file) {
-    const cornerstoneWADOImageLoader = await getCornerstoneWADOImageLoader();
+  addFile(file) {
     return cornerstoneWADOImageLoader.wadouri.fileManager.add(file);
   }
 

platform/viewer/src/lib/localFileLoaders/pdfFileLoader.js

@@ -1,10 +1,9 @@
+import cornerstoneWADOImageLoader from 'cornerstone-wado-image-loader';
 import FileLoader from './fileLoader';
-import { getCornerstoneWADOImageLoader } from '../../utils/cornerstoneWADOImageLoader';
 
 const PDFFileLoader = new (class extends FileLoader {
   fileType = 'application/pdf';
-  async loadFile(file, imageId) {
-    const cornerstoneWADOImageLoader = await getCornerstoneWADOImageLoader();
+  loadFile(file, imageId) {
     return cornerstoneWADOImageLoader.wadouri.loadFileRequest(imageId);
   }
 

platform/viewer/src/routes/routesUtil.js

@@ -1,5 +1,4 @@
 import asyncComponent from '../components/AsyncComponent.js';
-import { withCornerstone } from '../utils/cornerstoneWADOImageLoader';
 
 import OHIF from '@ohif/core';
 const { urlUtil: UrlUtil } = OHIF.utils;
@@ -10,10 +9,8 @@ const IHEInvokeImageDisplay = asyncComponent(() =>
     /* webpackChunkName: "IHEInvokeImageDisplay" */ './IHEInvokeImageDisplay.js'
   )
 );
-const ViewerRouting = withCornerstone(
-  asyncComponent(() =>
-    import(/* webpackChunkName: "ViewerRouting" */ './ViewerRouting.js')
-  )
+const ViewerRouting = asyncComponent(() =>
+  import(/* webpackChunkName: "ViewerRouting" */ './ViewerRouting.js')
 );
 
 const StudyListRouting = asyncComponent(() =>
@@ -21,18 +18,12 @@ const StudyListRouting = asyncComponent(() =>
     /* webpackChunkName: "StudyListRouting" */ '../studylist/StudyListRouting.js'
   )
 );
-const StandaloneRouting = withCornerstone(
-  asyncComponent(() =>
-    import(
-      /* webpackChunkName: "ConnectedStandaloneRouting" */ '../connectedComponents/ConnectedStandaloneRouting.js'
-    )
-  )
+const StandaloneRouting = asyncComponent(() =>
+  import(/* webpackChunkName: "ConnectedStandaloneRouting" */ '../connectedComponents/ConnectedStandaloneRouting.js')
 );
-const ViewerLocalFileData = withCornerstone(
-  asyncComponent(() =>
-    import(
-      /* webpackChunkName: "ViewerLocalFileData" */ '../connectedComponents/ViewerLocalFileData.js'
-    )
+const ViewerLocalFileData = asyncComponent(() =>
+  import(
+    /* webpackChunkName: "ViewerLocalFileData" */ '../connectedComponents/ViewerLocalFileData.js'
   )
 );
 

platform/viewer/src/utils/cornerstoneWADOImageLoader.js

@@ -1,79 +0,0 @@
-import OHIF from '@ohif/core';
-import cornerstone from 'cornerstone-core';
-import dicomParser from 'dicom-parser';
-import React from 'react';
-
-export function getCornerstoneWADOImageLoader() {
-  return import(
-    /* webpackChunkName: "CornerstoneWADOImageLoader" */ 'cornerstone-wado-image-loader'
-  );
-}
-
-let initialized = false;
-export async function initCornerstoneWADOImageLoader() {
-  if (initialized) return;
-  const cornerstoneWADOImageLoader = await getCornerstoneWADOImageLoader();
-  // For debugging
-  //if (process.env.node_env === 'development') {
-  window.cornerstoneWADOImageLoader = cornerstoneWADOImageLoader;
-  //}
-
-  cornerstoneWADOImageLoader.external.cornerstone = cornerstone;
-  cornerstoneWADOImageLoader.external.dicomParser = dicomParser;
-
-  OHIF.user.getAccessToken = () => {
-    // TODO: Get the Redux store from somewhere else
-    const state = window.store.getState();
-    if (!state.oidc || !state.oidc.user) {
-      return;
-    }
-
-    return state.oidc.user.accesstoken;
-  };
-
-  cornerstoneWADOImageLoader.configure({
-    beforeSend: function (xhr) {
-      const headers = OHIF.DICOMWeb.getAuthorizationHeader();
-
-      if (headers.Authorization) {
-        xhr.setRequestHeader('Authorization', headers.Authorization);
-      }
-    },
-  });
-
-  const config = {
-    maxWebWorkers: Math.max(navigator.hardwareConcurrency - 1, 1),
-    startWebWorkersOnDemand: true,
-    taskConfiguration: {
-      decodeTask: {
-        initializeCodecsOnStartup: false,
-        usePDFJS: false,
-        strict: false,
-      },
-    },
-  };
-  cornerstoneWADOImageLoader.webWorkerManager.initialize(config);
-
-  initialized = true;
-}
-
-export function withCornerstone(Component) {
-  // eslint-disable-next-line react/display-name
-  return props => {
-    const [
-      cornerstoneWADOImageLoaderInitialized,
-      setCornerstoneWADOImageLoaderInitialized,
-    ] = React.useState(false);
-
-    React.useEffect(() => {
-      initCornerstoneWADOImageLoader().then(() =>
-        setCornerstoneWADOImageLoaderInitialized(true)
-      );
-    }, []);
-
-    if (!cornerstoneWADOImageLoaderInitialized) {
-      return null;
-    }
-    return <Component {...props} />;
-  };
-}

platform/viewer/src/utils/index.js

@@ -1,3 +1,4 @@
 import getUserManagerForOpenIdConnectClient from './getUserManagerForOpenIdConnectClient.js';
+import initWebWorkers from './initWebWorkers.js';
 
-export { getUserManagerForOpenIdConnectClient };
+export { getUserManagerForOpenIdConnectClient, initWebWorkers };

platform/viewer/src/utils/index.test.js

@@ -5,7 +5,7 @@ describe('utils', () => {
     const utilExports = Object.keys(utils).sort();
 
     expect(utilExports).toEqual(
-      ['getUserManagerForOpenIdConnectClient'].sort()
+      ['getUserManagerForOpenIdConnectClient', 'initWebWorkers'].sort()
     );
   });
 });

platform/viewer/src/utils/initWebWorkers.js

@@ -0,0 +1,22 @@
+import cornerstoneWADOImageLoader from 'cornerstone-wado-image-loader';
+
+let initialized = false;
+
+export default function initWebWorkers() {
+  const config = {
+    maxWebWorkers: Math.max(navigator.hardwareConcurrency - 1, 1),
+    startWebWorkersOnDemand: true,
+    taskConfiguration: {
+      decodeTask: {
+        initializeCodecsOnStartup: false,
+        usePDFJS: false,
+        strict: false,
+      },
+    },
+  };
+
+  if (!initialized) {
+    cornerstoneWADOImageLoader.webWorkerManager.initialize(config);
+    initialized = true;
+  }
+}

platform/viewer/src/utils/initWebWorkers.test.js

@@ -0,0 +1,23 @@
+import initWebWorkers from './initWebWorkers.js';
+import cornerstoneWADOImageLoader from 'cornerstone-wado-image-loader';
+
+describe('initWebWorkers', () => {
+  it("initializes cornerstoneWADOImageLoader's web workers", () => {
+    initWebWorkers();
+
+    expect(
+      cornerstoneWADOImageLoader.webWorkerManager.initialize
+    ).toHaveBeenCalled();
+  });
+});
+
+describe('initWebWorkers', () => {
+  it("initializes cornerstoneWADOImageLoader's web workers only once", () => {
+    initWebWorkers();
+    initWebWorkers();
+
+    expect(
+      cornerstoneWADOImageLoader.webWorkerManager.initialize
+    ).toHaveBeenCalledTimes(1);
+  });
+});