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 | 1x 2x 2x 2x 20x 20x 2x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 50x 50x 2500x 2500x 125000x 125000x 125000x 1x 1x 2x 2x 2x 2x 2x 2x 1x | import macro from 'vtk.js/Sources/macros';
import vtkDataArray from 'vtk.js/Sources/Common/Core/DataArray';
import vtkImageData from 'vtk.js/Sources/Common/DataModel/ImageData';
const { vtkErrorMacro } = macro;
// ----------------------------------------------------------------------------
// vtkSampleFunction methods
// ----------------------------------------------------------------------------
function vtkSampleFunction(publicAPI, model) {
// Set our className
model.classHierarchy.push('vtkSampleFunction');
// Capture "parentClass" api for internal use
const superClass = { ...publicAPI };
publicAPI.getMTime = () => {
Iif (!(model.implicitFunction && model.implicitFunction.getMTime)) {
return superClass.getMTime();
}
return Math.max(superClass.getMTime(), model.implicitFunction.getMTime());
};
publicAPI.requestData = (inData, outData) => {
// implement requestData
const imp = model.implicitFunction;
Iif (!imp) {
vtkErrorMacro('An implicit function must be defined');
return;
}
// Create a volume
const dims = [
model.sampleDimensions[0],
model.sampleDimensions[1],
model.sampleDimensions[2],
];
const numScalars = dims[0] * dims[1] * dims[2];
Iif (numScalars < 1 || dims[1] < 2 || dims[1] < 2 || dims[2] < 2) {
vtkErrorMacro('Bad volume dimensions');
return;
}
const volume = vtkImageData.newInstance();
const origin = [
model.modelBounds[0],
model.modelBounds[2],
model.modelBounds[4],
];
const spacing = [
(model.modelBounds[1] - model.modelBounds[0]) / (dims[0] - 1),
(model.modelBounds[3] - model.modelBounds[2]) / (dims[1] - 1),
(model.modelBounds[5] - model.modelBounds[4]) / (dims[2] - 1),
];
const sliceSize = dims[0] * dims[1];
volume.setDimensions(dims);
volume.setOrigin(origin);
volume.setSpacing(spacing);
// Create scalars array
const s = macro.newTypedArray(model.pointType, numScalars);
const scalars = vtkDataArray.newInstance({
name: 'Scalars',
values: s,
numberOfComponents: 1,
});
volume.getPointData().setScalars(scalars);
// Now loop over volume dimensions and generate scalar values
let sValue = 0.0;
const xyz = [0.0, 0.0, 0.0];
for (let k = 0; k < dims[2]; k++) {
xyz[2] = origin[2] + k * spacing[2];
for (let j = 0; j < dims[1]; j++) {
xyz[1] = origin[1] + j * spacing[1];
for (let i = 0; i < dims[0]; i++) {
xyz[0] = origin[0] + i * spacing[0];
sValue = imp.evaluateFunction(xyz);
s[i + j * dims[0] + k * sliceSize] = sValue;
}
}
}
// Update output
outData[0] = volume;
};
}
// ----------------------------------------------------------------------------
// Object factory
// ----------------------------------------------------------------------------
const DEFAULT_VALUES = {
implicitFunction: undefined,
sampleDimensions: [50, 50, 50],
modelBounds: [-1.0, 1.0, -1.0, 1.0, -1.0, 1.0],
pointType: 'Float32Array',
};
// ----------------------------------------------------------------------------
export function extend(publicAPI, model, initialValues = {}) {
Object.assign(model, DEFAULT_VALUES, initialValues);
// Make this a VTK object
macro.obj(publicAPI, model);
// Also make it an algorithm with one input and one output
macro.algo(publicAPI, model, 1, 1);
// Generate macros for properties
macro.setGetArray(publicAPI, model, ['sampleDimensions'], 3);
macro.setGetArray(publicAPI, model, ['modelBounds'], 6);
// Object specific methods
vtkSampleFunction(publicAPI, model);
}
// ----------------------------------------------------------------------------
export const newInstance = macro.newInstance(extend, 'vtkSampleFunction');
// ----------------------------------------------------------------------------
export default { newInstance, extend };
|