import macro from 'vtk.js/Sources/macros'; import * as vtkMath from 'vtk.js/Sources/Common/Core/Math'; import vtkMatrixBuilder from 'vtk.js/Sources/Common/Core/MatrixBuilder'; import vtkBoundingBox from 'vtk.js/Sources/Common/DataModel/BoundingBox'; import vtkInteractorStyleManipulator from 'vtk.js/Sources/Interaction/Style/InteractorStyleManipulator'; import vtkMouseCameraTrackballRotateManipulator from 'vtk.js/Sources/Interaction/Manipulators/MouseCameraTrackballRotateManipulator'; import vtkMouseCameraTrackballPanManipulator from 'vtk.js/Sources/Interaction/Manipulators/MouseCameraTrackballPanManipulator'; import vtkMouseCameraTrackballZoomManipulator from 'vtk.js/Sources/Interaction/Manipulators/MouseCameraTrackballZoomManipulator'; import vtkMouseRangeManipulator from 'vtk.js/Sources/Interaction/Manipulators/MouseRangeManipulator';
import { mat4 } from 'gl-matrix';
function clamp(value, min, max) { if (value < min) { return min; } if (value > max) { return max; } return value; }
function vtkInteractorStyleMPRSlice(publicAPI, model) { model.classHierarchy.push('vtkInteractorStyleMPRSlice');
model.trackballManipulator = vtkMouseCameraTrackballRotateManipulator.newInstance({ button: 1, }); model.panManipulator = vtkMouseCameraTrackballPanManipulator.newInstance({ button: 1, shift: true, }); model.zoomManipulator = vtkMouseCameraTrackballZoomManipulator.newInstance({ button: 3, }); model.scrollManipulator = vtkMouseRangeManipulator.newInstance({ scrollEnabled: true, dragEnabled: false, });
const cache = { sliceNormal: [0, 0, 0], sliceRange: [0, 0], };
let cameraSub = null;
function updateScrollManipulator() { const range = publicAPI.getSliceRange(); model.scrollManipulator.removeScrollListener(); model.scrollManipulator.setScrollListener( range[0], range[1], 1, publicAPI.getSlice, publicAPI.setSlice ); }
function setManipulators() { publicAPI.removeAllMouseManipulators(); publicAPI.addMouseManipulator(model.trackballManipulator); publicAPI.addMouseManipulator(model.panManipulator); publicAPI.addMouseManipulator(model.zoomManipulator); publicAPI.addMouseManipulator(model.scrollManipulator); updateScrollManipulator(); }
const superSetInteractor = publicAPI.setInteractor; publicAPI.setInteractor = (interactor) => { superSetInteractor(interactor);
if (cameraSub) { cameraSub.unsubscribe(); cameraSub = null; }
if (interactor) { const renderer = interactor.getCurrentRenderer(); const camera = renderer.getActiveCamera();
cameraSub = camera.onModified(() => { updateScrollManipulator(); publicAPI.modified(); }); } };
publicAPI.handleMouseMove = macro.chain(publicAPI.handleMouseMove, () => { const renderer = model._interactor.getCurrentRenderer(); const camera = renderer.getActiveCamera(); const dist = camera.getDistance(); camera.setClippingRange(dist, dist + 0.1); });
const superSetVolumeMapper = publicAPI.setVolumeMapper; publicAPI.setVolumeMapper = (mapper) => { if (superSetVolumeMapper(mapper)) { const renderer = model._interactor.getCurrentRenderer(); const camera = renderer.getActiveCamera(); if (mapper) { camera.setFreezeFocalPoint(true); publicAPI.setSliceNormal(...publicAPI.getSliceNormal()); } else { camera.setFreezeFocalPoint(false); } } };
publicAPI.getSlice = () => { const renderer = model._interactor.getCurrentRenderer(); const camera = renderer.getActiveCamera(); const sliceNormal = publicAPI.getSliceNormal();
const transform = vtkMatrixBuilder .buildFromDegree() .identity() .rotateFromDirections(sliceNormal, [1, 0, 0]);
const fp = camera.getFocalPoint(); transform.apply(fp); return fp[0]; };
publicAPI.setSlice = (slice) => { const renderer = model._interactor.getCurrentRenderer(); const camera = renderer.getActiveCamera();
if (model.volumeMapper) { const range = publicAPI.getSliceRange();
const clampedSlice = clamp(slice, ...range); const center = model.volumeMapper.getCenter();
const distance = camera.getDistance(); const dop = camera.getDirectionOfProjection(); vtkMath.normalize(dop);
const midPoint = (range[1] + range[0]) / 2.0; const zeroPoint = [ center[0] - dop[0] * midPoint, center[1] - dop[1] * midPoint, center[2] - dop[2] * midPoint, ]; const slicePoint = [ zeroPoint[0] + dop[0] * clampedSlice, zeroPoint[1] + dop[1] * clampedSlice, zeroPoint[2] + dop[2] * clampedSlice, ];
const newPos = [ slicePoint[0] - dop[0] * distance, slicePoint[1] - dop[1] * distance, slicePoint[2] - dop[2] * distance, ];
camera.setPosition(...newPos); camera.setFocalPoint(...slicePoint); } };
publicAPI.getSliceRange = () => { if (model.volumeMapper) { const sliceNormal = publicAPI.getSliceNormal();
if (vtkMath.areEquals(sliceNormal, cache.sliceNormal)) { return cache.sliceRange; }
const sliceOrientation = vtkMatrixBuilder .buildFromDegree() .identity() .rotateFromDirections(sliceNormal, [1, 0, 0]); const imageAlongSliceNormal = mat4.create(); mat4.multiply( imageAlongSliceNormal, sliceOrientation.getMatrix(), model.volumeMapper.getInputData().getIndexToWorld() );
const transformedBounds = vtkBoundingBox.transformBounds( model.volumeMapper.getInputData().getSpatialExtent(), imageAlongSliceNormal );
const minX = transformedBounds[0]; const maxX = transformedBounds[1];
cache.sliceNormal = sliceNormal; cache.sliceRange = [minX, maxX]; return cache.sliceRange; } return [0, 0]; };
publicAPI.getSliceNormal = () => { if (model.volumeMapper) { const renderer = model._interactor.getCurrentRenderer(); const camera = renderer.getActiveCamera(); return camera.getDirectionOfProjection(); } return [0, 0, 0]; };
publicAPI.setSliceNormal = (...normal) => { const renderer = model._interactor.getCurrentRenderer(); const camera = renderer.getActiveCamera();
vtkMath.normalize(normal);
if (model.volumeMapper) { const bounds = model.volumeMapper.getBounds();
const diagonal = Math.sqrt( vtkMath.distance2BetweenPoints( [bounds[0], bounds[2], bounds[4]], [bounds[1], bounds[3], bounds[5]] ) );
const center = [ (bounds[0] + bounds[1]) / 2.0, (bounds[2] + bounds[3]) / 2.0, (bounds[4] + bounds[5]) / 2.0, ];
const angle = 90; const dist = diagonal / (2 * Math.tan((angle / 360) * Math.PI));
const cameraPos = [ center[0] - normal[0] * dist, center[1] - normal[1] * dist, center[2] - normal[2] * dist, ];
const oldDop = camera.getDirectionOfProjection(); const transform = vtkMatrixBuilder .buildFromDegree() .identity() .rotateFromDirections(oldDop, normal);
const viewUp = [0, 1, 0]; transform.apply(viewUp);
camera.setPosition(...cameraPos); camera.setDistance(dist); camera.setDirectionOfProjection(...normal); camera.setViewUp(...viewUp); camera.setViewAngle(angle); camera.setClippingRange(dist, dist + 0.1);
publicAPI.setCenterOfRotation(center); } };
setManipulators(); }
const DEFAULT_VALUES = {};
export function extend(publicAPI, model, initialValues = {}) { Object.assign(model, DEFAULT_VALUES, initialValues);
vtkInteractorStyleManipulator.extend(publicAPI, model, initialValues);
macro.setGet(publicAPI, model, ['volumeMapper']);
vtkInteractorStyleMPRSlice(publicAPI, model); }
export const newInstance = macro.newInstance( extend, 'vtkInteractorStyleMPRSlice' );
export default { newInstance, extend };
|