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

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
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1x
 
 
 
 
 
 
 
 
 
1x
 
1x
 
 
1x
 
 
 
 
 
 
 
 
 
 
 
 
 
1x
 
 
 
 
 
 
 
1x
 
 
1x
1x
1x
 
1x
 
 
 
 
1x
 
 
 
 
 
import macro from 'vtk.js/Sources/macros';
import vtkImplicitFunction from 'vtk.js/Sources/Common/DataModel/ImplicitFunction';
 
// ----------------------------------------------------------------------------
// Global methods
// ----------------------------------------------------------------------------
 
function evaluate(radius, center, xyz) {
  if (!Array.isArray(radius)) {
    const retVal =
      (xyz[0] - center[0]) * (xyz[0] - center[0]) +
      (xyz[1] - center[1]) * (xyz[1] - center[1]) +
      (xyz[2] - center[2]) * (xyz[2] - center[2]) -
      radius * radius;
 
    return retVal;
  }
  const r = [
    (xyz[0] - center[0]) / radius[0],
    (xyz[1] - center[1]) / radius[1],
    (xyz[2] - center[2]) / radius[2],
  ];
 
  return r[0] * r[0] + r[1] * r[1] + r[2] * r[2] - 1;
}
 
// ----------------------------------------------------------------------------
// Static API
// ----------------------------------------------------------------------------
 
export const STATIC = {
  evaluate,
};
 
// ----------------------------------------------------------------------------
// vtkSphere methods
// ----------------------------------------------------------------------------
 
function vtkSphere(publicAPI, model) {
  // Set our className
  model.classHierarchy.push('vtkSphere');
 
  publicAPI.evaluateFunction = (xyz) =>
    evaluate(model.radius, model.center, xyz);
 
  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
  vtkImplicitFunction.extend(publicAPI, model, initialValues);
  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 };