All files / packages/core/src/RenderingEngine/vtkClasses vtkStreamingOpenGLViewNodeFactory.js

95.83% Statements 46/48
72.72% Branches 8/11
100% Functions 5/5
95.83% Lines 46/48

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153                                        1x     2091x                       123x               123x 590x       590x 590x 590x 590x 590x 660x 590x   70x       590x       590x   590x 590x 590x     123x                       123x 590x   590x   590x 91x     590x               1x         123x     123x     123x     123x 123x 123x 123x 123x 123x 123x 123x       123x 123x 123x 123x 123x 123x 123x 123x                       1x                
import macro from '@kitware/vtk.js/macros';
// import vtkGenericWidgetRepresentation from '@kitware/vtk.js/Rendering/SceneGraph/GenericWidgetRepresentation'
import vtkOpenGLActor from '@kitware/vtk.js/Rendering/OpenGL/Actor';
import vtkOpenGLActor2D from '@kitware/vtk.js/Rendering/OpenGL/Actor2D';
import vtkOpenGLCamera from '@kitware/vtk.js/Rendering/OpenGL/Camera';
import vtkOpenGLGlyph3DMapper from '@kitware/vtk.js/Rendering/OpenGL/Glyph3DMapper';
import vtkOpenGLImageMapper from '@kitware/vtk.js/Rendering/OpenGL/ImageMapper';
import vtkOpenGLImageSlice from '@kitware/vtk.js/Rendering/OpenGL/ImageSlice';
import vtkOpenGLPixelSpaceCallbackMapper from '@kitware/vtk.js/Rendering/OpenGL/PixelSpaceCallbackMapper';
import vtkOpenGLPolyDataMapper from '@kitware/vtk.js/Rendering/OpenGL/PolyDataMapper';
import vtkOpenGLRenderer from '@kitware/vtk.js/Rendering/OpenGL/Renderer';
import vtkOpenGLSkybox from '@kitware/vtk.js/Rendering/OpenGL/Skybox';
import vtkOpenGLSphereMapper from '@kitware/vtk.js/Rendering/OpenGL/SphereMapper';
import vtkOpenGLStickMapper from '@kitware/vtk.js/Rendering/OpenGL/StickMapper';
import vtkOpenGLTexture from '@kitware/vtk.js/Rendering/OpenGL/Texture';
import vtkOpenGLVolume from '@kitware/vtk.js/Rendering/OpenGL/Volume';
import vtkOpenGLVolumeMapper from '@kitware/vtk.js/Rendering/OpenGL/VolumeMapper';
import vtkViewNodeFactory from '@kitware/vtk.js/Rendering/SceneGraph/ViewNodeFactory';
import vtkStreamingOpenGLVolumeMapper from './vtkStreamingOpenGLVolumeMapper';
 
const CLASS_MAPPING = Object.create(null);
 
export function registerOverride(className, fn) {
  CLASS_MAPPING[className] = fn;
}
 
/**
 * vtkStreamingOpenGLViewNodeFactory - A fork of the vtkOpenGLViewNodeFactory,
 * so that we can inject our custom derived "Streaming" classes.
 *
 * @param {*} publicAPI The public API to extend
 * @param {*} model The private model to extend.
 */
function vtkStreamingOpenGLViewNodeFactory(publicAPI, model) {
  // Set our className
  model.classHierarchy.push('vtkStreamingOpenGLViewNodeFactory');
 
  /**
   * createNode - fork of createNode from vtkOpenGLViewNodeFactory.
   * This fork is required to inject the properties from model.getModelInitialValues.
   *
   * @param {object} dataObject An instance of a vtk.js class.
   */
  publicAPI.createNode = (dataObject) => {
    Iif (dataObject.isDeleted()) {
      return null;
    }
 
    let cpt = 0;
    let className = dataObject.getClassName(cpt++);
    let isObject = false;
    const keys = Object.keys(model.overrides);
    while (className && !isObject) {
      if (keys.indexOf(className) !== -1) {
        isObject = true;
      } else {
        className = dataObject.getClassName(cpt++);
      }
    }
 
    Iif (!isObject) {
      return null;
    }
 
    const initialValues = model.getModelInitialValues(dataObject);
 
    const vn = model.overrides[className](initialValues);
    vn.setMyFactory(publicAPI);
    return vn;
  };
 
  model.overrides = CLASS_MAPPING;
 
  /**
   * getModelInitialValues - This function allows us to pass textures down from our
   * vtkSharedVolumeMapper to new instances of vtkStreamingOpenGLVolumeMapper.
   * The prevents us from sharing memory.
   *
   * TODO: It would be beneficial to push similar, but generalized, functionality
   * back to vtk.js in the future.
   *
   * @param {object} dataObject An instance of a vtk.js class.
   */
  model.getModelInitialValues = (dataObject) => {
    const initialValues = {};
 
    const className = dataObject.getClassName();
 
    if (className === 'vtkSharedVolumeMapper') {
      initialValues.scalarTexture = dataObject.getScalarTexture();
    }
 
    return initialValues;
  };
}
 
// ----------------------------------------------------------------------------
// Object factory
// ----------------------------------------------------------------------------
 
const DEFAULT_VALUES = {};
 
// ----------------------------------------------------------------------------
 
export function extend(publicAPI, model, initialValues = {}) {
  Object.assign(model, DEFAULT_VALUES, initialValues);
 
  // Inheritance
  vtkViewNodeFactory.extend(publicAPI, model, initialValues);
 
  // Object methods
  vtkStreamingOpenGLViewNodeFactory(publicAPI, model);
 
  // Initialization
  registerOverride('vtkActor', vtkOpenGLActor.newInstance);
  registerOverride('vtkActor2D', vtkOpenGLActor2D.newInstance);
  registerOverride('vtkCamera', vtkOpenGLCamera.newInstance);
  registerOverride('vtkGlyph3DMapper', vtkOpenGLGlyph3DMapper.newInstance);
  registerOverride('vtkImageMapper', vtkOpenGLImageMapper.newInstance);
  registerOverride('vtkImageSlice', vtkOpenGLImageSlice.newInstance);
  registerOverride('vtkMapper', vtkOpenGLPolyDataMapper.newInstance);
  registerOverride(
    'vtkPixelSpaceCallbackMapper',
    vtkOpenGLPixelSpaceCallbackMapper.newInstance
  );
  registerOverride('vtkRenderer', vtkOpenGLRenderer.newInstance);
  registerOverride('vtkSkybox', vtkOpenGLSkybox.newInstance);
  registerOverride('vtkSphereMapper', vtkOpenGLSphereMapper.newInstance);
  registerOverride('vtkStickMapper', vtkOpenGLStickMapper.newInstance);
  registerOverride('vtkTexture', vtkOpenGLTexture.newInstance);
  registerOverride('vtkVolume', vtkOpenGLVolume.newInstance);
  registerOverride('vtkVolumeMapper', vtkOpenGLVolumeMapper.newInstance);
  registerOverride(
    'vtkSharedVolumeMapper',
    vtkStreamingOpenGLVolumeMapper.newInstance
  );
  // registerOverride(
  //   'vtkWidgetRepresentation',
  //   vtkGenericWidgetRepresentation.newInstance
  // )
}
 
// ----------------------------------------------------------------------------
 
export const newInstance = macro.newInstance(
  extend,
  'vtkStreamingOpenGLViewNodeFactory'
);
 
// ----------------------------------------------------------------------------
 
export default { newInstance, extend };