Files
ThirdParty-6/ParaView-5.0.1/VTK/Examples/iOS/VolumeRender/Classes/MyGLKViewController.mm

482 lines
14 KiB
Plaintext

/*=========================================================================
Program: Visualization Toolkit
Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
All rights reserved.
See Copyright.txt or http://www.kitware.com/Copyright.htm for details.
This software is distributed WITHOUT ANY WARRANTY; without even
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE. See the above copyright notice for more information.
=========================================================================*/
#import "MyGLKViewController.h"
#include "vtk/vtkIOSRenderWindow.h"
#include "vtk/vtkIOSRenderWindowInteractor.h"
#include "vtk/vtkRenderingOpenGL2ObjectFactory.h"
#include "vtk/vtkNew.h"
#include "vtk/vtkNrrdReader.h"
#include "vtk/vtkImageCast.h"
#include "vtk/vtkRTAnalyticSource.h"
#include "vtk/vtkOpenGLGPUVolumeRayCastMapper.h"
#include "vtk/vtkVolumeProperty.h"
#include "vtk/vtkColorTransferFunction.h"
#include "vtk/vtkPiecewiseFunction.h"
#include "vtk/vtkVolume.h"
#include "vtk/vtkActor.h"
#include "vtk/vtkCamera.h"
#include "vtk/vtkConeSource.h"
#include "vtk/vtkDebugLeaks.h"
#include "vtk/vtkGlyph3D.h"
#include "vtk/vtkPolyData.h"
#include "vtk/vtkPolyDataMapper.h"
#include "vtk/vtkRenderWindow.h"
#include "vtk/vtkRenderer.h"
#include "vtk/vtkSphereSource.h"
#include "vtk/vtkTextActor.h"
#include "vtk/vtkTextProperty.h"
#include "vtk/vtkImageData.h"
#include "vtk/vtkPointData.h"
#include "vtk/vtkSmartPointer.h"
#include "vtk/vtkActor.h"
#include "vtk/vtkActor2D.h"
#include "vtk/vtkCamera.h"
#include "vtk/vtkCommand.h"
#include "vtk/vtkInteractorStyleMultiTouchCamera.h"
#include "vtk/vtkMath.h"
#include "vtk/vtkPoints.h"
#include "vtk/vtkPolyDataMapper.h"
#include "vtk/vtkRenderer.h"
#include "vtk/vtkTextMapper.h"
#include "vtk/vtkTextProperty.h"
#include <deque>
/* 2 or 3 -- needs to match VTK version */
#define GL_ES_VERSION 3
@interface MyGLKViewController () {
}
@property (strong, nonatomic) EAGLContext *context;
- (void)tearDownGL;
@end
@implementation MyGLKViewController
//----------------------------------------------------------------------------
- (vtkIOSRenderWindow *)getVTKRenderWindow
{
return _myVTKRenderWindow;
}
//----------------------------------------------------------------------------
- (void)setVTKRenderWindow:(vtkIOSRenderWindow *)theVTKRenderWindow
{
_myVTKRenderWindow = theVTKRenderWindow;
}
//----------------------------------------------------------------------------
- (vtkIOSRenderWindowInteractor *)getInteractor
{
if (_myVTKRenderWindow)
{
return (vtkIOSRenderWindowInteractor *)_myVTKRenderWindow->GetInteractor();
}
else
{
return NULL;
}
}
- (void)setupPipeline
{
// Register GL2 objects
vtkObjectFactory::RegisterFactory(vtkRenderingOpenGL2ObjectFactory::New());
vtkIOSRenderWindow *renWin = vtkIOSRenderWindow::New();
//renWin->DebugOn();
[self setVTKRenderWindow:renWin];
// this example uses VTK's built in interaction but you could choose
// to use your own instead.
vtkRenderWindowInteractor *iren = vtkRenderWindowInteractor::New();
vtkNew<vtkInteractorStyleMultiTouchCamera> ismt;
ismt->DebugOn();
iren->SetInteractorStyle(ismt.Get());
iren->SetRenderWindow(renWin);
vtkNew<vtkRenderer> renderer;
renWin->AddRenderer(renderer.Get());
vtkNew<vtkOpenGLGPUVolumeRayCastMapper> volumeMapper;
#if 0
vtkNew<vtkRTAnalyticSource> wavelet;
wavelet->SetWholeExtent(-127, 128,
-127, 128,
-127, 128);
wavelet->SetCenter(0.0, 0.0, 0.0);
vtkNew<vtkImageCast> ic;
ic->SetInputConnection(wavelet->GetOutputPort());
ic->SetOutputScalarTypeToUnsignedChar();
volumeMapper->SetInputConnection(ic->GetOutputPort());
#else
NSArray *paths = NSSearchPathForDirectoriesInDomains(NSDocumentDirectory, NSUserDomainMask, YES);
NSString *basePath = paths.firstObject;
std::string fname([basePath UTF8String]);
fname += "/CT-chest-quantized.nrrd";
vtkNew<vtkNrrdReader> mi;
mi->SetFileName(fname.c_str());
mi->Update();
double range[2];
mi->GetOutput()->GetPointData()->GetScalars()->GetRange(range);
volumeMapper->SetInputConnection(mi->GetOutputPort());
#endif
volumeMapper->SetAutoAdjustSampleDistances(1);
volumeMapper->SetSampleDistance(0.5);
vtkNew<vtkVolumeProperty> volumeProperty;
volumeProperty->SetShade(1);
volumeProperty->SetInterpolationTypeToLinear();
vtkNew<vtkColorTransferFunction> ctf;
// ctf->AddRGBPoint(90, 0.2, 0.29, 1);
// ctf->AddRGBPoint(157.091, 0.87, 0.87, 0.87);
// ctf->AddRGBPoint(250, 0.7, 0.015, 0.15);
ctf->AddRGBPoint(0, 0, 0, 0);
ctf->AddRGBPoint(255*67.0106/3150.0, 0.54902, 0.25098, 0.14902);
ctf->AddRGBPoint(255*251.105/3150.0, 0.882353, 0.603922, 0.290196);
ctf->AddRGBPoint(255*439.291/3150.0, 1, 0.937033, 0.954531);
ctf->AddRGBPoint(255*3071/3150.0, 0.827451, 0.658824, 1);
// vtkNew<vtkPiecewiseFunction> pwf;
// pwf->AddPoint(0, 0.0);
// pwf->AddPoint(7000, 1.0);
double tweak = 80.0;
vtkNew<vtkPiecewiseFunction> pwf;
pwf->AddPoint(0, 0);
pwf->AddPoint(255*(67.0106+tweak)/3150.0, 0);
pwf->AddPoint(255*(251.105+tweak)/3150.0, 0.3);
pwf->AddPoint(255*(439.291+tweak)/3150.0, 0.5);
pwf->AddPoint(255*3071/3150.0, 0.616071);
volumeProperty->SetColor(ctf.GetPointer());
volumeProperty->SetScalarOpacity(pwf.GetPointer());
vtkNew<vtkVolume> volume;
volume->SetMapper(volumeMapper.GetPointer());
volume->SetProperty(volumeProperty.GetPointer());
renderer->SetBackground2(0.2,0.3,0.4);
renderer->SetBackground(0.1,0.1,0.1);
renderer->GradientBackgroundOn();
renderer->AddVolume(volume.GetPointer());
renderer->ResetCamera();
renderer->GetActiveCamera()->Zoom(1.4);
}
- (void)viewDidLoad
{
[super viewDidLoad];
#if GL_ES_VERSION == 2
self.context = [[EAGLContext alloc] initWithAPI:kEAGLRenderingAPIOpenGLES2];
#else
self.context = [[EAGLContext alloc] initWithAPI:kEAGLRenderingAPIOpenGLES3];
#endif
if (!self.context) {
NSLog(@"Failed to create ES context");
}
GLKView *view = (GLKView *)self.view;
view.context = self.context;
view.drawableDepthFormat = GLKViewDrawableDepthFormat16;
//view.drawableMultisample = GLKViewDrawableMultisample4X;
// setup the vis pipeline
[self setupPipeline];
[EAGLContext setCurrentContext:self.context];
[self resizeView];
[self getVTKRenderWindow]->Render();
}
- (void)dealloc
{
[self tearDownGL];
if ([EAGLContext currentContext] == self.context) {
[EAGLContext setCurrentContext:nil];
}
}
- (void)didReceiveMemoryWarning
{
[super didReceiveMemoryWarning];
if ([self isViewLoaded] && ([[self view] window] == nil)) {
self.view = nil;
[self tearDownGL];
if ([EAGLContext currentContext] == self.context) {
[EAGLContext setCurrentContext:nil];
}
self.context = nil;
}
// Dispose of any resources that can be recreated.
}
- (void)tearDownGL
{
[EAGLContext setCurrentContext:self.context];
// free GL resources
// ...
}
-(void) resizeView
{
double scale = self.view.contentScaleFactor;
double newWidth = scale * self.view.bounds.size.width;
double newHeight = scale * self.view.bounds.size.height;
[self getVTKRenderWindow]->SetSize(newWidth, newHeight);
}
- (void)viewWillLayoutSubviews
{
[self resizeView];
}
- (void)glkView:(GLKView *)view drawInRect:(CGRect)rect
{
//std::cout << [self getVTKRenderWindow]->ReportCapabilities() << std::endl;
[self getVTKRenderWindow]->Render();
}
//=================================================================
// this example uses VTK's built in interaction but you could choose
// to use your own instead. The remaining methods forward touch events
// to VTKs interactor.
// Handles the start of a touch
- (void)touchesBegan:(NSSet *)touches withEvent:(UIEvent *)event
{
vtkIOSRenderWindowInteractor *interactor = [self getInteractor];
if (!interactor)
{
return;
}
vtkIOSRenderWindow *renWin = [self getVTKRenderWindow];
if (!renWin)
{
return;
}
CGRect bounds = [self.view bounds];
double scale = self.view.contentScaleFactor;
// set the position for all contacts
NSSet *myTouches = [event touchesForView:self.view];
for (UITouch *touch in myTouches)
{
// Convert touch point from UIView referential to OpenGL one (upside-down flip)
CGPoint location = [touch locationInView:self.view];
location.y = bounds.size.height - location.y;
// Account for the content scaling factor
location.x *= scale;
location.y *= scale;
int index = interactor->GetPointerIndexForContact((size_t)(__bridge void *)touch);
if (index < VTKI_MAX_POINTERS)
{
interactor->SetEventInformation((int)round(location.x),
(int)round(location.y),
0, 0,
0, 0, 0, index);
}
}
// handle begin events
for (UITouch *touch in touches)
{
int index = interactor->GetPointerIndexForContact((size_t)(__bridge void *)touch);
vtkGenericWarningMacro("down touch " << (size_t)(__bridge void *)touch << " index " << index);
interactor->SetPointerIndex(index);
interactor->InvokeEvent(vtkCommand::LeftButtonPressEvent,NULL);
//NSLog(@"Starting left mouse");
}
// Display the buffer
[(GLKView *)self.view display];
}
// Handles the continuation of a touch.
- (void)touchesMoved:(NSSet *)touches withEvent:(UIEvent *)event
{
vtkIOSRenderWindowInteractor *interactor = [self getInteractor];
if (!interactor)
{
return;
}
vtkIOSRenderWindow *renWin = [self getVTKRenderWindow];
if (!renWin)
{
return;
}
CGRect bounds = [self.view bounds];
double scale = self.view.contentScaleFactor;
// set the position for all contacts
int index;
NSSet *myTouches = [event touchesForView:self.view];
for (UITouch *touch in myTouches)
{
// Convert touch point from UIView referential to OpenGL one (upside-down flip)
CGPoint location = [touch locationInView:self.view];
location.y = bounds.size.height - location.y;
// Account for the content scaling factor
location.x *= scale;
location.y *= scale;
index = interactor->GetPointerIndexForContact((size_t)(__bridge void *)touch);
if (index < VTKI_MAX_POINTERS)
{
interactor->SetEventInformation((int)round(location.x),
(int)round(location.y),
0, 0,
0, 0, 0, index);
}
}
// fire move event on last index
interactor->SetPointerIndex(index);
interactor->InvokeEvent(vtkCommand::MouseMoveEvent,NULL);
NSLog(@"Moved left mouse");
// Display the buffer
[(GLKView *)self.view display];
}
// Handles the end of a touch event when the touch is a tap.
- (void)touchesEnded:(NSSet *)touches withEvent:(UIEvent *)event
{
vtkIOSRenderWindowInteractor *interactor = [self getInteractor];
if (!interactor)
{
return;
}
vtkIOSRenderWindow *renWin = [self getVTKRenderWindow];
if (!renWin)
{
return;
}
CGRect bounds = [self.view bounds];
double scale = self.view.contentScaleFactor;
// set the position for all contacts
NSSet *myTouches = [event touchesForView:self.view];
for (UITouch *touch in myTouches)
{
// Convert touch point from UIView referential to OpenGL one (upside-down flip)
CGPoint location = [touch locationInView:self.view];
location.y = bounds.size.height - location.y;
// Account for the content scaling factor
location.x *= scale;
location.y *= scale;
int index = interactor->GetPointerIndexForContact((size_t)(__bridge void *)touch);
if (index < VTKI_MAX_POINTERS)
{
interactor->SetEventInformation((int)round(location.x),
(int)round(location.y),
0, 0,
0, 0, 0, index);
}
}
// handle begin events
for (UITouch *touch in touches)
{
int index = interactor->GetPointerIndexForContact((size_t)(__bridge void *)touch);
vtkGenericWarningMacro("up touch " << (size_t)(__bridge void *)touch << " index " << index);
interactor->SetPointerIndex(index);
interactor->InvokeEvent(vtkCommand::LeftButtonReleaseEvent,NULL);
interactor->ClearContact((size_t)(__bridge void *)touch);
// NSLog(@"lifting left mouse");
}
// Display the buffer
[(GLKView *)self.view display];
}
// Handles the end of a touch event.
- (void)touchesCancelled:(NSSet *)touches withEvent:(UIEvent *)event
{
vtkIOSRenderWindowInteractor *interactor = [self getInteractor];
if (!interactor)
{
return;
}
vtkIOSRenderWindow *renWin = [self getVTKRenderWindow];
if (!renWin)
{
return;
}
CGRect bounds = [self.view bounds];
double scale = self.view.contentScaleFactor;
UITouch* touch = [[event touchesForView:self.view] anyObject];
// Convert touch point from UIView referential to OpenGL one (upside-down flip)
CGPoint location = [touch locationInView:self.view];
location.y = bounds.size.height - location.y;
// Account for the content scaling factor
location.x *= scale;
location.y *= scale;
interactor->SetEventInformation((int)round(location.x),
(int)round(location.y),
0, 0,
0, 0);
interactor->InvokeEvent(vtkCommand::LeftButtonReleaseEvent, NULL);
// NSLog(@"Ended left mouse");
// Display the buffer
[(GLKView *)self.view display];
}
@end