index.js

import macro from 'vtk.js/Sources/macros';
import { getProfileForClass, getSuggestedProfile } from './profileHints';
import 'vtk.js/Sources/Rendering/OpenGL/Profiles/profileHintManifest';
import 'vtk.js/Sources/Rendering/WebGPU/Profiles/profileHintManifest';

const { vtkOnceErrorMacro } = macro;

function listClassHierarchy(dataObject) {
  const classNames = [];
  let depth = 0;
  let className = dataObject.getClassName(depth++);
  while (className) {
    classNames.push(className);
    className = dataObject.getClassName(depth++);
  }
  return classNames;
}

function buildMissingImplementationMessage(factoryName, classNames) {
  const classList = classNames.join(' → ');
  const suggestedClass = getSuggestedProfile(classNames);
  const suggestedProfile = suggestedClass && getProfileForClass(suggestedClass);

  if (suggestedProfile) {
    return [
      `No ${factoryName} implementation found for ${classNames[0]}.`,
      `Class hierarchy: ${classList}.`,
      'This usually means the rendering profile import is missing.',
      `Try importing '@kitware/vtk.js/Rendering/Profiles/${suggestedProfile}' or 'vtk.js/Sources/Rendering/Profiles/${suggestedProfile}'.`,
      "You can also use the full profile imports '@kitware/vtk.js/Rendering/Profiles/All' or 'vtk.js/Sources/Rendering/Profiles/All'.",
    ].join('\n');
  }

  return [
    `No ${factoryName} implementation found for ${classNames[0]}.`,
    `Class hierarchy: ${classList}.`,
    'If this is a built-in renderable, a rendering profile import may be missing.',
    "Try importing '@kitware/vtk.js/Rendering/Profiles/All' or 'vtk.js/Sources/Rendering/Profiles/All',",
    'or import the specific rendering profile needed by this renderable.',
    'If this is a custom renderable, register a matching override with the view node factory.',
  ].join('\n');
}

// ----------------------------------------------------------------------------
// vtkViewNodeFactory methods
// ----------------------------------------------------------------------------

function vtkViewNodeFactory(publicAPI, model) {
  // Make sure our overrides is just for our instance not shared with everyone...
  if (!model.overrides) {
    model.overrides = {};
  }

  // Set our className
  model.classHierarchy.push('vtkViewNodeFactory');

  publicAPI.createNode = (dataObject) => {
    if (dataObject.isDeleted()) {
      return null;
    }

    const classNames = listClassHierarchy(dataObject);
    let cpt = 0;
    let className = classNames[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++);
      }
    }

    if (!isObject) {
      vtkOnceErrorMacro(
        buildMissingImplementationMessage(publicAPI.getClassName(), classNames)
      );
      return null;
    }
    const vn = model.overrides[className]();
    vn.setMyFactory(publicAPI);
    return vn;
  };
}

// ----------------------------------------------------------------------------
// Object factory
// ----------------------------------------------------------------------------

const DEFAULT_VALUES = {
  // overrides: {},
};

// ----------------------------------------------------------------------------

export function extend(publicAPI, model, initialValues = {}) {
  Object.assign(model, DEFAULT_VALUES, initialValues);

  // Build VTK API
  macro.obj(publicAPI, model);

  // Object methods
  vtkViewNodeFactory(publicAPI, model);
}

// ----------------------------------------------------------------------------

export const newInstance = macro.newInstance(extend, 'vtkViewNodeFactory');

// ----------------------------------------------------------------------------

export default { newInstance, extend };