All files / Sources/Common/DataModel/Sphere index.js

52.17% Statements 12/23
0% Branches 0/1
28.57% Functions 2/7
54.54% Lines 12/22

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      1x                                                                     1x                     1x   1x                   1x                           1x               1x     1x 1x 1x   1x         1x          
import macro from 'vtk.js/Sources/macros';
import vtkBoundingBox from 'vtk.js/Sources/Common/DataModel/BoundingBox';
 
const VTK_SMALL_NUMBER = 1e-12;
 
// ----------------------------------------------------------------------------
// Global methods
// ----------------------------------------------------------------------------
 
function evaluate(radius, center, x) {
  return (
    (x[0] - center[0]) * (x[0] - center[0]) +
    (x[1] - center[1]) * (x[1] - center[1]) +
    (x[2] - center[2]) * (x[2] - center[2]) -
    radius * radius
  );
}
 
function isPointIn3DEllipse(point, bounds) {
  const center = vtkBoundingBox.getCenter(bounds);
  let scale3 = vtkBoundingBox.computeScale3(bounds);
  scale3 = scale3.map((x) => Math.max(Math.abs(x), VTK_SMALL_NUMBER));
 
  const radius = [
    (point[0] - center[0]) / scale3[0],
    (point[1] - center[1]) / scale3[1],
    (point[2] - center[2]) / scale3[2],
  ];
 
  return (
    radius[0] * radius[0] + radius[1] * radius[1] + radius[2] * radius[2] <= 1
  );
}
 
// ----------------------------------------------------------------------------
// Static API
// ----------------------------------------------------------------------------
 
export const STATIC = {
  evaluate,
  isPointIn3DEllipse,
};
 
// ----------------------------------------------------------------------------
// vtkSphere methods
// ----------------------------------------------------------------------------
 
function vtkSphere(publicAPI, model) {
  // Set our className
  model.classHierarchy.push('vtkSphere');
 
  publicAPI.evaluateFunction = (xyz) => {
    const retVal =
      (xyz[0] - model.center[0]) * (xyz[0] - model.center[0]) +
      (xyz[1] - model.center[1]) * (xyz[1] - model.center[1]) +
      (xyz[2] - model.center[2]) * (xyz[2] - model.center[2]) -
      model.radius * model.radius;
 
    return retVal;
  };
 
  publicAPI.evaluateGradient = (xyz) => {
    const retVal = [
      2.0 - (xyz[0] - model.center[0]),
      2.0 - (xyz[1] - model.center[1]),
      2.0 - (xyz[2] - model.center[2]),
    ];
    return retVal;
  };
}
 
// ----------------------------------------------------------------------------
// Object factory
// ----------------------------------------------------------------------------
 
const DEFAULT_VALUES = {
  radius: 0.5,
  center: [0.0, 0.0, 0.0],
};
 
// ----------------------------------------------------------------------------
 
export function extend(publicAPI, model, initialValues = {}) {
  Object.assign(model, DEFAULT_VALUES, initialValues);
 
  // Object methods
  macro.obj(publicAPI, model);
  macro.setGet(publicAPI, model, ['radius']);
  macro.setGetArray(publicAPI, model, ['center'], 3);
 
  vtkSphere(publicAPI, model);
}
 
// ----------------------------------------------------------------------------
 
export const newInstance = macro.newInstance(extend, 'vtkSphere');
 
// ----------------------------------------------------------------------------
 
export default { newInstance, extend, ...STATIC };