@kitware/vtk.js/Common/DataModel/DataSet.js

Visualization Toolkit for the Web

import { m as macro } from '../../macros2.js';
import vtk from '../../vtk.js';
import vtkBoundingBox from './BoundingBox.js';
import vtkDataSetAttributes from './DataSetAttributes.js';
import { b as vtkMath } from '../Core/Math/index.js';
import Constants from './DataSet/Constants.js';

// ----------------------------------------------------------------------------
// Global methods
// ----------------------------------------------------------------------------

const DATASET_FIELDS = ['pointData', 'cellData', 'fieldData'];

// ----------------------------------------------------------------------------
// vtkDataSet methods
// ----------------------------------------------------------------------------

function vtkDataSet(publicAPI, model) {
  // Set our className
  model.classHierarchy.push('vtkDataSet');

  // Add dataset attributes
  DATASET_FIELDS.forEach(fieldName => {
    if (!model[fieldName]) {
      model[fieldName] = vtkDataSetAttributes.newInstance();
    } else {
      model[fieldName] = vtk(model[fieldName]);
    }
  });

  //------------------------------------------------------------------------------
  // Compute the data bounding box from data points.
  publicAPI.computeBounds = () => {
    if (model.modifiedTime && model.computeTime && model.modifiedTime > model.computeTime || !model.computeTime) {
      const points = publicAPI.getPoints();
      if (points?.getNumberOfPoints()) {
        // Compute bounds from points
        vtkBoundingBox.setBounds(model.bounds, points.getBoundsByReference());
      } else {
        model.bounds = vtkMath.createUninitializedBounds();
      }
      // Update computeTime
      model.computeTime = macro.getCurrentGlobalMTime();
    }
  };

  /**
   * Returns the squared length of the diagonal of the bounding box
   */
  publicAPI.getLength2 = () => {
    const bounds = publicAPI.getBoundsByReference();
    if (!bounds || bounds.length !== 6) return 0;
    return vtkBoundingBox.getDiagonalLength2(bounds);
  };

  /**
   * Returns the length of the diagonal of the bounding box
   */
  publicAPI.getLength = () => Math.sqrt(publicAPI.getLength2());

  /**
   * Returns the center of the bounding box as [x, y, z]
   */
  publicAPI.getCenter = () => {
    const bounds = publicAPI.getBoundsByReference();
    if (!bounds || bounds.length !== 6) return [0, 0, 0];
    return vtkBoundingBox.getCenter(bounds);
  };

  /**
   * Get the bounding box of a cell with the given cellId
   * @param {Number} cellId - The id of the cell
   * @returns {Number[]} - The bounds as [xmin, xmax, ymin, ymax, zmin, zmax]
   */
  publicAPI.getCellBounds = cellId => {
    const cell = publicAPI.getCell(cellId);
    if (cell) {
      return cell.getBounds();
    }
    return vtkMath.createUninitializedBounds();
  };
  publicAPI.getBounds = macro.chain(() => publicAPI.computeBounds, publicAPI.getBounds);
  publicAPI.getBoundsByReference = macro.chain(() => publicAPI.computeBounds, publicAPI.getBoundsByReference);
  const superShallowCopy = publicAPI.shallowCopy;
  publicAPI.shallowCopy = (other, debug = false) => {
    superShallowCopy(other, debug);
    DATASET_FIELDS.forEach(fieldName => {
      model[fieldName] = vtkDataSetAttributes.newInstance();
      model[fieldName].shallowCopy(other.getReferenceByName(fieldName));
    });
  };
  const superGetMTime = publicAPI.getMTime;
  publicAPI.getMTime = () => DATASET_FIELDS.reduce((mTime, fieldName) => Math.max(mTime, model[fieldName]?.getMTime() ?? mTime), superGetMTime());
  publicAPI.initialize = () => {
    DATASET_FIELDS.forEach(fieldName => model[fieldName]?.initialize());
    return publicAPI;
  };
}

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

const DEFAULT_VALUES = {
  // pointData: null,
  // cellData: null,
  // fieldData: null,
};

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

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

  // Object methods
  macro.obj(publicAPI, model);
  macro.setGet(publicAPI, model, DATASET_FIELDS);
  macro.getArray(publicAPI, model, ['bounds'], 6);
  // Object specific methods
  vtkDataSet(publicAPI, model);
}

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

const newInstance = macro.newInstance(extend, 'vtkDataSet');

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

var vtkDataSet$1 = {
  newInstance,
  extend,
  ...Constants
};

export { vtkDataSet$1 as default, extend, newInstance };