C++ (VtkSDIView.h):
// VtkSDIView.h : interface of the CVtkSDIView class
//
#pragma once
// Include the required header files for the vtk classes we are using
#include <vtkRenderer.h>
#include <vtkWin32OpenGLRenderWindow.h>
#include <vtkWin32RenderWindowInteractor.h>
#include <vtkSphereSource.h>
#include <vtkConeSource.h>
#include <vtkGlyph3D.h>
#include <vtkElevationFilter.h>
#include <vtkPolyDataMapper.h>
#include <vtkActor.h>
#include <vtkCubeAxesActor2D.h>
class CVtkSDIView : public CView
{
protected: // create from serialization only
CVtkSDIView();
DECLARE_DYNCREATE(CVtkSDIView)
// Attributes
public:
CVtkSDIDoc* GetDocument() const;
// Operations
public:
// Overrides
public:
virtual void OnDraw(CDC* pDC); // overridden to draw this view
virtual BOOL PreCreateWindow(CREATESTRUCT& cs);
protected:
virtual BOOL OnPreparePrinting(CPrintInfo* pInfo);
virtual void OnBeginPrinting(CDC* pDC, CPrintInfo* pInfo);
virtual void OnEndPrinting(CDC* pDC, CPrintInfo* pInfo);
// Implementation
public:
void Pipeline ( void );
virtual ~CVtkSDIView();
vtkWin32OpenGLRenderWindow *renWin;
vtkRenderer *ren;
vtkWin32RenderWindowInteractor *iren;
vtkSphereSource *sphere;
vtkPolyDataMapper *sphereMapper;
vtkElevationFilter *sphereElevation;
vtkActor *sphereActor;
vtkConeSource *cone;
vtkGlyph3D *glyph;
vtkPolyDataMapper *spikeMapper;
vtkActor *spikeActor;
vtkCubeAxesActor2D *sphereAxis;
#ifdef _DEBUG
virtual void AssertValid() const;
virtual void Dump(CDumpContext& dc) const;
#endif
protected:
// Generated message map functions
protected:
afx_msg void OnFilePrintPreview();
afx_msg void OnRButtonUp(UINT nFlags, CPoint point);
afx_msg void OnContextMenu(CWnd* pWnd, CPoint point);
DECLARE_MESSAGE_MAP()
public:
afx_msg void OnSize(UINT nType, int cx, int cy);
afx_msg BOOL OnEraseBkgnd(CDC* pDC);
afx_msg int OnCreate(LPCREATESTRUCT lpCreateStruct);
virtual LRESULT WindowProc(UINT message, WPARAM wParam, LPARAM lParam);
};
#ifndef _DEBUG // debug version in VtkSDIView.cpp
inline CVtkSDIDoc* CVtkSDIView::GetDocument() const
{ return reinterpret_cast<CVtkSDIDoc*>(m_pDocument); }
#endif
C++ (VtkSDIView.cpp):// VtkSDIView.cpp : implementation of the CVtkSDIView class
//
#include "stdafx.h"
// SHARED_HANDLERS can be defined in an ATL project implementing preview, thumbnail
// and search filter handlers and allows sharing of document code with that project.
#ifndef SHARED_HANDLERS
#include "VtkSDI.h"
#endif
#include "VtkSDIDoc.h"
#include "VtkSDIView.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#endif
// CVtkSDIView
IMPLEMENT_DYNCREATE(CVtkSDIView, CView)
BEGIN_MESSAGE_MAP(CVtkSDIView, CView)
// Standard printing commands
ON_COMMAND(ID_FILE_PRINT, &CView::OnFilePrint)
ON_COMMAND(ID_FILE_PRINT_DIRECT, &CView::OnFilePrint)
ON_COMMAND(ID_FILE_PRINT_PREVIEW, &CVtkSDIView::OnFilePrintPreview)
ON_WM_CONTEXTMENU()
ON_WM_RBUTTONUP()
ON_WM_SIZE()
ON_WM_ERASEBKGND()
ON_WM_CREATE()
END_MESSAGE_MAP()
// CVtkSDIView construction/destruction
CVtkSDIView::CVtkSDIView()
{
// Create the the renderer, window and interactor objects.
this->ren = vtkRenderer::New();
this->renWin = vtkWin32OpenGLRenderWindow::New();
this->iren = vtkWin32RenderWindowInteractor::New();
// Create the the objects used to form the visualisation.
this->sphere = vtkSphereSource::New();
this->sphereElevation = vtkElevationFilter::New();
this->sphereMapper = vtkPolyDataMapper::New();
this->sphereActor = vtkActor::New();
this->cone = vtkConeSource::New();
this->glyph = vtkGlyph3D::New();
this->spikeMapper = vtkPolyDataMapper::New();
this->spikeActor = vtkActor::New();
this->sphereAxis = vtkCubeAxesActor2D::New();
}
CVtkSDIView::~CVtkSDIView()
{
// Delete the the renderer, window and interactor objects.
this->ren->Delete();
this->iren->Delete();
this->renWin->Delete();
// Delete the the objects used to form the visualisation.
this->sphere->Delete();
this->sphereElevation->Delete();
this->sphereMapper->Delete();
this->sphereActor->Delete();
this->cone->Delete();
this->glyph->Delete();
this->spikeMapper->Delete();
this->spikeActor->Delete();
this->sphereAxis->Delete();
}
BOOL CVtkSDIView::PreCreateWindow(CREATESTRUCT& cs)
{
// TODO: Modify the Window class or styles here by modifying
// the CREATESTRUCT cs
return CView::PreCreateWindow(cs);
}
// CVtkSDIView drawing
void CVtkSDIView::OnDraw(CDC* pDC)
{
CVtkSDIDoc* pDoc = GetDocument();
ASSERT_VALID(pDoc);
if (!pDoc)
return;
if ( !this->iren->GetInitialized() )
{
CRect rect;
this->GetClientRect(&rect);
this->iren->Initialize();
this->renWin->SetSize(rect.right-rect.left,rect.bottom-rect.top);
this->ren->ResetCamera();
}
// Invoke the pipeline
Pipeline();
if ( pDC->IsPrinting() )
{
this->BeginWaitCursor();
// Obtain the size of the printer page in pixels.
int cxPage = pDC->GetDeviceCaps(HORZRES);
int cyPage = pDC->GetDeviceCaps(VERTRES);
// Get the size of the window in pixels.
int *size = this->renWin->GetSize();
int cxWindow = size[0];
int cyWindow = size[1];
float fx = float(cxPage) / float(cxWindow);
float fy = float(cyPage) / float(cyWindow);
float scale = min(fx,fy);
int x = int(scale * float(cxWindow));
int y = int(scale * float(cyWindow));
this->renWin->SetupMemoryRendering(cxWindow, cyWindow, pDC->GetSafeHdc());
this->renWin->Render();
HDC memDC = this->renWin->GetMemoryDC();
StretchBlt(pDC->GetSafeHdc(),0,0,x,y,memDC,0,0,cxWindow,cyWindow,SRCCOPY);
this->renWin->ResumeScreenRendering();
this->EndWaitCursor();
}
else
{
this->renWin->Render();
}
}
// CVtkSDIView printing
void CVtkSDIView::OnFilePrintPreview()
{
#ifndef SHARED_HANDLERS
AFXPrintPreview(this);
#endif
}
BOOL CVtkSDIView::OnPreparePrinting(CPrintInfo* pInfo)
{
// default preparation
return DoPreparePrinting(pInfo);
}
void CVtkSDIView::OnBeginPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
{
// TODO: add extra initialization before printing
}
void CVtkSDIView::OnEndPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
{
// TODO: add cleanup after printing
}
void CVtkSDIView::OnRButtonUp(UINT /* nFlags */, CPoint point)
{
ClientToScreen(&point);
OnContextMenu(this, point);
}
void CVtkSDIView::OnContextMenu(CWnd* /* pWnd */, CPoint point)
{
#ifndef SHARED_HANDLERS
theApp.GetContextMenuManager()->ShowPopupMenu(IDR_POPUP_EDIT, point.x, point.y, this, TRUE);
#endif
}
// CVtkSDIView diagnostics
#ifdef _DEBUG
void CVtkSDIView::AssertValid() const
{
CView::AssertValid();
}
void CVtkSDIView::Dump(CDumpContext& dc) const
{
CView::Dump(dc);
}
CVtkSDIDoc* CVtkSDIView::GetDocument() const // non-debug version is inline
{
ASSERT(m_pDocument->IsKindOf(RUNTIME_CLASS(CVtkSDIDoc)));
return (CVtkSDIDoc*)m_pDocument;
}
#endif //_DEBUG
// CVtkSDIView message handlers
void CVtkSDIView::OnSize(UINT nType, int cx, int cy)
{
CView::OnSize(nType, cx, cy);
CRect rect;
this->GetClientRect(&rect);
this->renWin->SetSize(rect.right-rect.left,rect.bottom-rect.top);
}
BOOL CVtkSDIView::OnEraseBkgnd(CDC* pDC)
{
return TRUE;
}
int CVtkSDIView::OnCreate(LPCREATESTRUCT lpCreateStruct)
{
if (CView::OnCreate(lpCreateStruct) == -1)
return -1;
this->renWin->AddRenderer(this->ren);
// setup the parent window
this->renWin->SetParentId(this->m_hWnd);
this->iren->SetRenderWindow(this->renWin);
return 0;
}
LRESULT CVtkSDIView::WindowProc(UINT message, WPARAM wParam, LPARAM lParam)
{
switch (message)
{
//case WM_PAINT:
case WM_LBUTTONDOWN:
case WM_LBUTTONUP:
case WM_MBUTTONDOWN:
case WM_MBUTTONUP:
case WM_RBUTTONDOWN:
case WM_RBUTTONUP:
case WM_MOUSEMOVE:
case WM_CHAR:
case WM_TIMER:
if (this->iren->GetInitialized())
{
return vtkHandleMessage2(this->m_hWnd, message, wParam, lParam, this->iren);
}
break;
}
return CView::WindowProc(message, wParam, lParam);
}
void CVtkSDIView::Pipeline()
{
// Construct the sphere
this->sphere->SetRadius(1);
this->sphere->SetThetaResolution(18);
this->sphere->SetPhiResolution(18);
this->sphere->LatLongTessellationOn();
// Generate elevations
this->sphereElevation->SetInput(this->sphere->GetOutput());
this->sphereElevation->SetLowPoint(0,0,-1);
this->sphereElevation->SetHighPoint(0,0,1);
this->sphereElevation->SetScalarRange(-1,1);
// Link the mapper
this->sphereMapper->SetInput(this->sphereElevation->GetPolyDataOutput());
this->sphereMapper->SetColorModeToMapScalars();
this->sphereMapper->SetScalarRange(-1,1);
// Link the actor
this->sphereActor->SetMapper(this->sphereMapper);
// Add it to the renderer
this->ren->AddActor(this->sphereActor);
// Construct the cone
this->cone->SetResolution(8);
// Construct the glyphs on the spherical surface
this->glyph->SetInput(this->sphere->GetOutput());
this->glyph->SetSource(this->cone->GetOutput());
this->glyph->SetVectorModeToUseNormal();
this->glyph->SetScaleModeToScaleByVector();
this->glyph->SetScaleFactor(0.1);
// Link the mapper to the glyph
this->spikeMapper->SetInput(this->glyph->GetOutput());
// Link the actor
this->spikeActor->SetMapper(this->spikeMapper);
// Add it to the renderer
this->ren->AddActor(this->spikeActor);
// Add in the cube axis actor
this->sphereAxis->SetInput(this->sphereElevation->GetOutput());
this->sphereAxis->SetCamera(this->ren->GetActiveCamera());
// Add it to the renderer
this->ren->AddActor(this->sphereAxis);
}
說明:
MFC使用VTK 5.10.0繪製3D模型範例。