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 127 128 129 130 131 132 133 134 | 1x 3x 3x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 3x 3x 3x 3x 3x 3x 1x | import vtk from 'vtk.js/Sources/vtk';
import macro from 'vtk.js/Sources/macros';
import vtkPoints from 'vtk.js/Sources/Common/Core/Points';
const { vtkDebugMacro, vtkErrorMacro } = macro;
// ----------------------------------------------------------------------------
// vtkWarpScalar methods
// ----------------------------------------------------------------------------
function vtkWarpScalar(publicAPI, model) {
// Set our className
model.classHierarchy.push('vtkWarpScalar');
publicAPI.requestData = (inData, outData) => {
// implement requestData
const input = inData[0];
Iif (!input) {
vtkErrorMacro('Invalid or missing input');
return 1;
}
// First, copy the input to the output as a starting point
// output->CopyStructure( input );
const inPts = input.getPoints();
const pd = input.getPointData();
const inNormals = pd.getNormals();
const inScalars = publicAPI.getInputArrayToProcess(0);
if (!inPts || !inScalars) {
vtkDebugMacro('No data to warp', !!inPts, !!inScalars);
outData[0] = inData[0];
return 1;
}
const numPts = inPts.getNumberOfPoints();
let pointNormal = null;
const normal = [0, 0, 1];
if (inNormals && !model.useNormal) {
pointNormal = (id, array) => [
array.getData()[id * 3],
array.getData()[id * 3 + 1],
array.getData()[id * 3 + 2],
];
vtkDebugMacro('Using data normals');
} else if (publicAPI.getXyPlane()) {
pointNormal = (id, array) => normal;
vtkDebugMacro('Using x-y plane normal');
} else {
pointNormal = (id, array) => model.normal;
vtkDebugMacro('Using Normal instance variable');
}
const newPtsData = new Float32Array(numPts * 3);
const inPoints = inPts.getData();
let ptOffset = 0;
let n = [0, 0, 1];
let s = 1;
// Loop over all points, adjusting locations
const scalarDataArray = inScalars.getData();
const nc = inScalars.getNumberOfComponents();
for (let ptId = 0; ptId < numPts; ++ptId) {
ptOffset = ptId * 3;
n = pointNormal(ptId, inNormals);
if (model.xyPlane) {
s = inPoints[ptOffset + 2];
} else {
// Use component 0 of array if there are multiple components
s = scalarDataArray[ptId * nc];
}
newPtsData[ptOffset] = inPoints[ptOffset] + model.scaleFactor * s * n[0];
newPtsData[ptOffset + 1] =
inPoints[ptOffset + 1] + model.scaleFactor * s * n[1];
newPtsData[ptOffset + 2] =
inPoints[ptOffset + 2] + model.scaleFactor * s * n[2];
}
const newDataSet = vtk({ vtkClass: input.getClassName() });
newDataSet.shallowCopy(input);
const points = vtkPoints.newInstance();
points.setData(newPtsData, 3);
newDataSet.setPoints(points);
outData[0] = newDataSet;
return 1;
};
}
// ----------------------------------------------------------------------------
// Object factory
// ----------------------------------------------------------------------------
const DEFAULT_VALUES = {
scaleFactor: 1,
useNormal: false,
normal: [0, 0, 1],
xyPlane: false,
};
// ----------------------------------------------------------------------------
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.setGet(publicAPI, model, ['scaleFactor', 'useNormal', 'xyPlane']);
macro.setGetArray(publicAPI, model, ['normal'], 3);
// Object specific methods
vtkWarpScalar(publicAPI, model);
}
// ----------------------------------------------------------------------------
export const newInstance = macro.newInstance(extend, 'vtkWarpScalar');
// ----------------------------------------------------------------------------
export default { newInstance, extend };
|