iteractor work
This commit is contained in:
Anatoly Baksheev
parent
03bd82796d
commit
fb7e74c511
@ -1,635 +0,0 @@
|
|||||||
/*M///////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
//
|
|
||||||
// IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
|
|
||||||
//
|
|
||||||
// By downloading, copying, installing or using the software you agree to this license.
|
|
||||||
// If you do not agree to this license, do not download, install,
|
|
||||||
// copy or use the software.
|
|
||||||
//
|
|
||||||
//
|
|
||||||
// License Agreement
|
|
||||||
// For Open Source Computer Vision Library
|
|
||||||
//
|
|
||||||
// Copyright (C) 2013, OpenCV Foundation, all rights reserved.
|
|
||||||
// Third party copyrights are property of their respective owners.
|
|
||||||
//
|
|
||||||
// Redistribution and use in source and binary forms, with or without modification,
|
|
||||||
// are permitted provided that the following conditions are met:
|
|
||||||
//
|
|
||||||
// * Redistribution's of source code must retain the above copyright notice,
|
|
||||||
// this list of conditions and the following disclaimer.
|
|
||||||
//
|
|
||||||
// * Redistribution's in binary form must reproduce the above copyright notice,
|
|
||||||
// this list of conditions and the following disclaimer in the documentation
|
|
||||||
// and/or other materials provided with the distribution.
|
|
||||||
//
|
|
||||||
// * The name of the copyright holders may not be used to endorse or promote products
|
|
||||||
// derived from this software without specific prior written permission.
|
|
||||||
//
|
|
||||||
// This software is provided by the copyright holders and contributors "as is" and
|
|
||||||
// any express or implied warranties, including, but not limited to, the implied
|
|
||||||
// warranties of merchantability and fitness for a particular purpose are disclaimed.
|
|
||||||
// In no event shall the Intel Corporation or contributors be liable for any direct,
|
|
||||||
// indirect, incidental, special, exemplary, or consequential damages
|
|
||||||
// (including, but not limited to, procurement of substitute goods or services;
|
|
||||||
// loss of use, data, or profits; or business interruption) however caused
|
|
||||||
// and on any theory of liability, whether in contract, strict liability,
|
|
||||||
// or tort (including negligence or otherwise) arising in any way out of
|
|
||||||
// the use of this software, even if advised of the possibility of such damage.
|
|
||||||
//
|
|
||||||
// Authors:
|
|
||||||
// * Ozan Tonkal, ozantonkal@gmail.com
|
|
||||||
// * Anatoly Baksheev, Itseez Inc. myname.mysurname <> mycompany.com
|
|
||||||
//
|
|
||||||
// OpenCV Viz module is complete rewrite of
|
|
||||||
// PCL visualization module (www.pointclouds.org)
|
|
||||||
//
|
|
||||||
//M*/
|
|
||||||
|
|
||||||
#include "precomp.hpp"
|
|
||||||
|
|
||||||
|
|
||||||
namespace cv { namespace viz
|
|
||||||
{
|
|
||||||
vtkStandardNewMacro(InteractorStyle)
|
|
||||||
}}
|
|
||||||
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
void cv::viz::InteractorStyle::Initialize()
|
|
||||||
{
|
|
||||||
// Set windows size (width, height) to unknown (-1)
|
|
||||||
win_size_ = Vec2i(-1, -1);
|
|
||||||
win_pos_ = Vec2i(0, 0);
|
|
||||||
max_win_size_ = Vec2i(-1, -1);
|
|
||||||
|
|
||||||
init_ = true;
|
|
||||||
stereo_anaglyph_mask_default_ = true;
|
|
||||||
|
|
||||||
// Initialize the keyboard event callback as none
|
|
||||||
keyboardCallback_ = 0;
|
|
||||||
keyboard_callback_cookie_ = 0;
|
|
||||||
|
|
||||||
// Initialize the mouse event callback as none
|
|
||||||
mouseCallback_ = 0;
|
|
||||||
mouse_callback_cookie_ = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
void cv::viz::InteractorStyle::saveScreenshot(const String &file)
|
|
||||||
{
|
|
||||||
FindPokedRenderer(Interactor->GetEventPosition()[0], Interactor->GetEventPosition()[1]);
|
|
||||||
|
|
||||||
vtkSmartPointer<vtkWindowToImageFilter> wif = vtkSmartPointer<vtkWindowToImageFilter>::New();
|
|
||||||
wif->SetInput(Interactor->GetRenderWindow());
|
|
||||||
|
|
||||||
vtkSmartPointer<vtkPNGWriter> snapshot_writer = vtkSmartPointer<vtkPNGWriter>::New();
|
|
||||||
snapshot_writer->SetInputConnection(wif->GetOutputPort());
|
|
||||||
snapshot_writer->SetFileName(file.c_str());
|
|
||||||
snapshot_writer->Write();
|
|
||||||
|
|
||||||
cout << "Screenshot successfully captured (" << file.c_str() << ")" << endl;
|
|
||||||
}
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
void cv::viz::InteractorStyle::exportScene(const String &file)
|
|
||||||
{
|
|
||||||
vtkSmartPointer<vtkExporter> exporter;
|
|
||||||
if (file.size() > 5 && file.substr(file.size() - 5) == ".vrml")
|
|
||||||
{
|
|
||||||
exporter = vtkSmartPointer<vtkVRMLExporter>::New();
|
|
||||||
vtkVRMLExporter::SafeDownCast(exporter)->SetFileName(file.c_str());
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
exporter = vtkSmartPointer<vtkOBJExporter>::New();
|
|
||||||
vtkOBJExporter::SafeDownCast(exporter)->SetFilePrefix(file.c_str());
|
|
||||||
}
|
|
||||||
|
|
||||||
exporter->SetInput(Interactor->GetRenderWindow());
|
|
||||||
exporter->Write();
|
|
||||||
|
|
||||||
cout << "Scene successfully exported (" << file.c_str() << ")" << endl;
|
|
||||||
}
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
void cv::viz::InteractorStyle::zoomIn()
|
|
||||||
{
|
|
||||||
FindPokedRenderer(Interactor->GetEventPosition()[0], Interactor->GetEventPosition()[1]);
|
|
||||||
// Zoom in
|
|
||||||
StartDolly();
|
|
||||||
double factor = 10.0 * 0.2 * .5;
|
|
||||||
Dolly(std::pow(1.1, factor));
|
|
||||||
EndDolly();
|
|
||||||
}
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
void cv::viz::InteractorStyle::zoomOut()
|
|
||||||
{
|
|
||||||
FindPokedRenderer(Interactor->GetEventPosition()[0], Interactor->GetEventPosition()[1]);
|
|
||||||
// Zoom out
|
|
||||||
StartDolly();
|
|
||||||
double factor = 10.0 * -0.2 * .5;
|
|
||||||
Dolly(std::pow(1.1, factor));
|
|
||||||
EndDolly();
|
|
||||||
}
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
void cv::viz::InteractorStyle::OnChar()
|
|
||||||
{
|
|
||||||
// Make sure we ignore the same events we handle in OnKeyDown to avoid calling things twice
|
|
||||||
FindPokedRenderer(Interactor->GetEventPosition()[0], Interactor->GetEventPosition()[1]);
|
|
||||||
if (Interactor->GetKeyCode() >= '0' && Interactor->GetKeyCode() <= '9')
|
|
||||||
return;
|
|
||||||
|
|
||||||
String key(Interactor->GetKeySym());
|
|
||||||
if (key.find("XF86ZoomIn") != String::npos)
|
|
||||||
zoomIn();
|
|
||||||
else if (key.find("XF86ZoomOut") != String::npos)
|
|
||||||
zoomOut();
|
|
||||||
|
|
||||||
int keymod = Interactor->GetAltKey();
|
|
||||||
|
|
||||||
switch (Interactor->GetKeyCode())
|
|
||||||
{
|
|
||||||
// All of the options below simply exit
|
|
||||||
case 'h': case 'H':
|
|
||||||
case 'l': case 'L':
|
|
||||||
case 'p': case 'P':
|
|
||||||
case 'j': case 'J':
|
|
||||||
case 'c': case 'C':
|
|
||||||
case 43: // KEY_PLUS
|
|
||||||
case 45: // KEY_MINUS
|
|
||||||
case 'f': case 'F':
|
|
||||||
case 'g': case 'G':
|
|
||||||
case 'o': case 'O':
|
|
||||||
case 'u': case 'U':
|
|
||||||
case 'q': case 'Q':
|
|
||||||
{
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
// S and R have a special !ALT case
|
|
||||||
case 'r': case 'R':
|
|
||||||
case 's': case 'S':
|
|
||||||
{
|
|
||||||
if (!keymod)
|
|
||||||
Superclass::OnChar();
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
default:
|
|
||||||
{
|
|
||||||
Superclass::OnChar();
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
void cv::viz::InteractorStyle::registerMouseCallback(void (*callback)(const MouseEvent&, void*), void* cookie)
|
|
||||||
{
|
|
||||||
// Register the callback function and store the user data
|
|
||||||
mouseCallback_ = callback;
|
|
||||||
mouse_callback_cookie_ = cookie;
|
|
||||||
}
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
void cv::viz::InteractorStyle::registerKeyboardCallback(void (*callback)(const KeyboardEvent&, void*), void *cookie)
|
|
||||||
{
|
|
||||||
// Register the callback function and store the user data
|
|
||||||
keyboardCallback_ = callback;
|
|
||||||
keyboard_callback_cookie_ = cookie;
|
|
||||||
}
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
int cv::viz::InteractorStyle::getModifiers()
|
|
||||||
{
|
|
||||||
int modifiers = KeyboardEvent::NONE;
|
|
||||||
|
|
||||||
if (Interactor->GetAltKey())
|
|
||||||
modifiers |= KeyboardEvent::ALT;
|
|
||||||
|
|
||||||
if (Interactor->GetControlKey())
|
|
||||||
modifiers |= KeyboardEvent::CTRL;
|
|
||||||
|
|
||||||
if (Interactor->GetShiftKey())
|
|
||||||
modifiers |= KeyboardEvent::SHIFT;
|
|
||||||
return modifiers;
|
|
||||||
}
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
void cv::viz::InteractorStyle::OnKeyDown()
|
|
||||||
{
|
|
||||||
CV_Assert("Interactor style not initialized. Please call Initialize() before continuing" && init_);
|
|
||||||
FindPokedRenderer(Interactor->GetEventPosition()[0], Interactor->GetEventPosition()[1]);
|
|
||||||
|
|
||||||
// Save the initial windows width/height
|
|
||||||
if (win_size_[0] == -1 || win_size_[1] == -1)
|
|
||||||
win_size_ = Vec2i(Interactor->GetRenderWindow()->GetSize());
|
|
||||||
|
|
||||||
bool alt = Interactor->GetAltKey() != 0;
|
|
||||||
|
|
||||||
std::string key(Interactor->GetKeySym());
|
|
||||||
if (key.find("XF86ZoomIn") != std::string::npos)
|
|
||||||
zoomIn();
|
|
||||||
else if (key.find("XF86ZoomOut") != std::string::npos)
|
|
||||||
zoomOut();
|
|
||||||
|
|
||||||
switch (Interactor->GetKeyCode())
|
|
||||||
{
|
|
||||||
case 'h': case 'H':
|
|
||||||
{
|
|
||||||
std::cout << "| Help:\n"
|
|
||||||
"-------\n"
|
|
||||||
" p, P : switch to a point-based representation\n"
|
|
||||||
" w, W : switch to a wireframe-based representation (where available)\n"
|
|
||||||
" s, S : switch to a surface-based representation (where available)\n"
|
|
||||||
"\n"
|
|
||||||
" j, J : take a .PNG snapshot of the current window view\n"
|
|
||||||
" k, K : export scene to Wavefront .obj format\n"
|
|
||||||
" ALT + k, K : export scene to VRML format\n"
|
|
||||||
" c, C : display current camera/window parameters\n"
|
|
||||||
" f, F : fly to point mode, hold the key and move mouse where to fly\n"
|
|
||||||
"\n"
|
|
||||||
" e, E : exit the interactor\n"
|
|
||||||
" q, Q : stop and call VTK's TerminateApp\n"
|
|
||||||
"\n"
|
|
||||||
" +/- : increment/decrement overall point size\n"
|
|
||||||
" +/- [+ ALT] : zoom in/out \n"
|
|
||||||
"\n"
|
|
||||||
" r, R [+ ALT] : reset camera [to viewpoint = {0, 0, 0} -> center_{x, y, z}]\n"
|
|
||||||
"\n"
|
|
||||||
" ALT + s, S : turn stereo mode on/off\n"
|
|
||||||
" ALT + f, F : switch between maximized window mode and original size\n"
|
|
||||||
"\n"
|
|
||||||
<< std::endl;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Switch representation to points
|
|
||||||
case 'p': case 'P':
|
|
||||||
{
|
|
||||||
vtkSmartPointer<vtkActorCollection> ac = CurrentRenderer->GetActors();
|
|
||||||
vtkCollectionSimpleIterator ait;
|
|
||||||
for (ac->InitTraversal(ait); vtkActor* actor = ac->GetNextActor(ait); )
|
|
||||||
for (actor->InitPathTraversal(); vtkAssemblyPath* path = actor->GetNextPath(); )
|
|
||||||
{
|
|
||||||
vtkActor* apart = vtkActor::SafeDownCast(path->GetLastNode()->GetViewProp());
|
|
||||||
apart->GetProperty()->SetRepresentationToPoints();
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Save a PNG snapshot
|
|
||||||
case 'j': case 'J':
|
|
||||||
saveScreenshot(cv::format("screenshot-%d.png", (unsigned int)time(0))); break;
|
|
||||||
|
|
||||||
// Export scene as in obj or vrml format
|
|
||||||
case 'k': case 'K':
|
|
||||||
{
|
|
||||||
String format = alt ? "scene-%d.vrml" : "scene-%d";
|
|
||||||
exportScene(cv::format(format.c_str(), (unsigned int)time(0)));
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
// display current camera settings/parameters
|
|
||||||
case 'c': case 'C':
|
|
||||||
{
|
|
||||||
vtkSmartPointer<vtkCamera> cam = Interactor->GetRenderWindow()->GetRenderers()->GetFirstRenderer()->GetActiveCamera();
|
|
||||||
|
|
||||||
Vec2d clip(cam->GetClippingRange());
|
|
||||||
Vec3d focal(cam->GetFocalPoint()), pos(cam->GetPosition()), view(cam->GetViewUp());
|
|
||||||
Vec2i win_pos(Interactor->GetRenderWindow()->GetPosition());
|
|
||||||
Vec2i win_size(Interactor->GetRenderWindow()->GetSize());
|
|
||||||
double angle = cam->GetViewAngle () / 180.0 * CV_PI;
|
|
||||||
|
|
||||||
String data = cv::format("clip(%f,%f) focal(%f,%f,%f) pos(%f,%f,%f) view(%f,%f,%f) angle(%f) winsz(%d,%d) winpos(%d,%d)",
|
|
||||||
clip[0], clip[1], focal[0], focal[1], focal[2], pos[0], pos[1], pos[2], view[0], view[1], view[2],
|
|
||||||
angle, win_size[0], win_size[1], win_pos[0], win_pos[1]);
|
|
||||||
|
|
||||||
std::cout << data.c_str() << std::endl;
|
|
||||||
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
case '=':
|
|
||||||
{
|
|
||||||
zoomIn();
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
case 43: // KEY_PLUS
|
|
||||||
{
|
|
||||||
if (alt)
|
|
||||||
zoomIn();
|
|
||||||
else
|
|
||||||
{
|
|
||||||
vtkSmartPointer<vtkActorCollection> ac = CurrentRenderer->GetActors();
|
|
||||||
vtkCollectionSimpleIterator ait;
|
|
||||||
for (ac->InitTraversal(ait); vtkActor* actor = ac->GetNextActor(ait); )
|
|
||||||
for (actor->InitPathTraversal(); vtkAssemblyPath* path = actor->GetNextPath(); )
|
|
||||||
{
|
|
||||||
vtkActor* apart = vtkActor::SafeDownCast(path->GetLastNode()->GetViewProp());
|
|
||||||
float psize = apart->GetProperty()->GetPointSize();
|
|
||||||
if (psize < 63.0f)
|
|
||||||
apart->GetProperty()->SetPointSize(psize + 1.0f);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
case 45: // KEY_MINUS
|
|
||||||
{
|
|
||||||
if (alt)
|
|
||||||
zoomOut();
|
|
||||||
else
|
|
||||||
{
|
|
||||||
vtkSmartPointer<vtkActorCollection> ac = CurrentRenderer->GetActors();
|
|
||||||
vtkCollectionSimpleIterator ait;
|
|
||||||
for (ac->InitTraversal(ait); vtkActor* actor = ac->GetNextActor(ait); )
|
|
||||||
for (actor->InitPathTraversal(); vtkAssemblyPath* path = actor->GetNextPath(); )
|
|
||||||
{
|
|
||||||
vtkActor* apart = vtkActor::SafeDownCast(path->GetLastNode()->GetViewProp());
|
|
||||||
float psize = apart->GetProperty()->GetPointSize();
|
|
||||||
if (psize > 1.0f)
|
|
||||||
apart->GetProperty()->SetPointSize(psize - 1.0f);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
// Switch between maximize and original window size
|
|
||||||
case 'f': case 'F':
|
|
||||||
{
|
|
||||||
if (alt)
|
|
||||||
{
|
|
||||||
Vec2i screen_size(Interactor->GetRenderWindow()->GetScreenSize());
|
|
||||||
Vec2i win_size(Interactor->GetRenderWindow()->GetSize());
|
|
||||||
|
|
||||||
// Is window size = max?
|
|
||||||
if (win_size == max_win_size_)
|
|
||||||
{
|
|
||||||
Interactor->GetRenderWindow()->SetSize(win_size_.val);
|
|
||||||
Interactor->GetRenderWindow()->SetPosition(win_pos_.val);
|
|
||||||
Interactor->Render();
|
|
||||||
}
|
|
||||||
// Set to max
|
|
||||||
else
|
|
||||||
{
|
|
||||||
win_pos_ = Vec2i(Interactor->GetRenderWindow()->GetPosition());
|
|
||||||
win_size_ = win_size;
|
|
||||||
|
|
||||||
Interactor->GetRenderWindow()->SetSize(screen_size.val);
|
|
||||||
Interactor->Render();
|
|
||||||
max_win_size_ = Vec2i(Interactor->GetRenderWindow()->GetSize());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
AnimState = VTKIS_ANIM_ON;
|
|
||||||
Interactor->GetPicker()->Pick(Interactor->GetEventPosition()[0], Interactor->GetEventPosition()[1], 0.0, CurrentRenderer);
|
|
||||||
vtkSmartPointer<vtkAbstractPropPicker> picker = vtkAbstractPropPicker::SafeDownCast(Interactor->GetPicker());
|
|
||||||
if (picker)
|
|
||||||
if (picker->GetPath())
|
|
||||||
Interactor->FlyTo(CurrentRenderer, picker->GetPickPosition());
|
|
||||||
AnimState = VTKIS_ANIM_OFF;
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
// 's'/'S' w/out ALT
|
|
||||||
case 's': case 'S':
|
|
||||||
{
|
|
||||||
if (alt)
|
|
||||||
{
|
|
||||||
vtkSmartPointer<vtkRenderWindow> window = Interactor->GetRenderWindow();
|
|
||||||
if (!window->GetStereoRender())
|
|
||||||
{
|
|
||||||
static Vec2i red_blue(4, 3), magenta_green(2, 5);
|
|
||||||
window->SetAnaglyphColorMask (stereo_anaglyph_mask_default_ ? red_blue.val : magenta_green.val);
|
|
||||||
stereo_anaglyph_mask_default_ = !stereo_anaglyph_mask_default_;
|
|
||||||
}
|
|
||||||
window->SetStereoRender(!window->GetStereoRender());
|
|
||||||
Interactor->Render();
|
|
||||||
}
|
|
||||||
else
|
|
||||||
Superclass::OnKeyDown();
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
case 'o': case 'O':
|
|
||||||
{
|
|
||||||
vtkSmartPointer<vtkCamera> cam = CurrentRenderer->GetActiveCamera();
|
|
||||||
cam->SetParallelProjection(!cam->GetParallelProjection());
|
|
||||||
Interactor->Render();
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Overwrite the camera reset
|
|
||||||
case 'r': case 'R':
|
|
||||||
{
|
|
||||||
if (!alt)
|
|
||||||
{
|
|
||||||
Superclass::OnKeyDown();
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
WidgetActorMap::iterator it = widget_actor_map_->begin();
|
|
||||||
// it might be that some actors don't have a valid transformation set -> we skip them to avoid a seg fault.
|
|
||||||
for (; it != widget_actor_map_->end(); ++it)
|
|
||||||
{
|
|
||||||
vtkProp3D * actor = vtkProp3D::SafeDownCast(it->second);
|
|
||||||
if (actor && actor->GetUserMatrix())
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
vtkSmartPointer<vtkCamera> cam = CurrentRenderer->GetActiveCamera();
|
|
||||||
|
|
||||||
// if a valid transformation was found, use it otherwise fall back to default view point.
|
|
||||||
if (it != widget_actor_map_->end())
|
|
||||||
{
|
|
||||||
vtkMatrix4x4* m = vtkProp3D::SafeDownCast(it->second)->GetUserMatrix();
|
|
||||||
|
|
||||||
cam->SetFocalPoint(m->GetElement(0, 3) - m->GetElement(0, 2),
|
|
||||||
m->GetElement(1, 3) - m->GetElement(1, 2),
|
|
||||||
m->GetElement(2, 3) - m->GetElement(2, 2));
|
|
||||||
|
|
||||||
cam->SetViewUp (m->GetElement(0, 1), m->GetElement(1, 1), m->GetElement(2, 1));
|
|
||||||
cam->SetPosition(m->GetElement(0, 3), m->GetElement(1, 3), m->GetElement(2, 3));
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
cam->SetPosition(0, 0, 0);
|
|
||||||
cam->SetFocalPoint(0, 0, 1);
|
|
||||||
cam->SetViewUp(0, -1, 0);
|
|
||||||
}
|
|
||||||
|
|
||||||
// go to the next actor for the next key-press event.
|
|
||||||
if (it != widget_actor_map_->end())
|
|
||||||
++it;
|
|
||||||
else
|
|
||||||
it = widget_actor_map_->begin();
|
|
||||||
|
|
||||||
CurrentRenderer->SetActiveCamera(cam);
|
|
||||||
CurrentRenderer->ResetCameraClippingRange();
|
|
||||||
Interactor->Render();
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
case 'q': case 'Q':
|
|
||||||
{
|
|
||||||
Interactor->ExitCallback();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
default:
|
|
||||||
{
|
|
||||||
Superclass::OnKeyDown();
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
KeyboardEvent event(KeyboardEvent::KEY_DOWN, Interactor->GetKeySym(), Interactor->GetKeyCode(), getModifiers());
|
|
||||||
if (keyboardCallback_)
|
|
||||||
keyboardCallback_(event, keyboard_callback_cookie_);
|
|
||||||
Interactor->Render();
|
|
||||||
}
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
void cv::viz::InteractorStyle::OnKeyUp()
|
|
||||||
{
|
|
||||||
KeyboardEvent event(KeyboardEvent::KEY_UP, Interactor->GetKeySym(), Interactor->GetKeyCode(), getModifiers());
|
|
||||||
if (keyboardCallback_)
|
|
||||||
keyboardCallback_(event, keyboard_callback_cookie_);
|
|
||||||
Superclass::OnKeyUp();
|
|
||||||
}
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
void cv::viz::InteractorStyle::OnMouseMove()
|
|
||||||
{
|
|
||||||
Vec2i p(Interactor->GetEventPosition());
|
|
||||||
MouseEvent event(MouseEvent::MouseMove, MouseEvent::NoButton, p, getModifiers());
|
|
||||||
if (mouseCallback_)
|
|
||||||
mouseCallback_(event, mouse_callback_cookie_);
|
|
||||||
Superclass::OnMouseMove();
|
|
||||||
}
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
void cv::viz::InteractorStyle::OnLeftButtonDown()
|
|
||||||
{
|
|
||||||
Vec2i p(Interactor->GetEventPosition());
|
|
||||||
MouseEvent::Type type = (Interactor->GetRepeatCount() == 0) ? MouseEvent::MouseButtonPress : MouseEvent::MouseDblClick;
|
|
||||||
MouseEvent event(type, MouseEvent::LeftButton, p, getModifiers());
|
|
||||||
if (mouseCallback_)
|
|
||||||
mouseCallback_(event, mouse_callback_cookie_);
|
|
||||||
Superclass::OnLeftButtonDown();
|
|
||||||
}
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
void cv::viz::InteractorStyle::OnLeftButtonUp()
|
|
||||||
{
|
|
||||||
Vec2i p(Interactor->GetEventPosition());
|
|
||||||
MouseEvent event(MouseEvent::MouseButtonRelease, MouseEvent::LeftButton, p, getModifiers());
|
|
||||||
if (mouseCallback_)
|
|
||||||
mouseCallback_(event, mouse_callback_cookie_);
|
|
||||||
Superclass::OnLeftButtonUp();
|
|
||||||
}
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
void cv::viz::InteractorStyle::OnMiddleButtonDown()
|
|
||||||
{
|
|
||||||
Vec2i p(Interactor->GetEventPosition());
|
|
||||||
MouseEvent::Type type = (Interactor->GetRepeatCount() == 0) ? MouseEvent::MouseButtonPress : MouseEvent::MouseDblClick;
|
|
||||||
MouseEvent event(type, MouseEvent::MiddleButton, p, getModifiers());
|
|
||||||
if (mouseCallback_)
|
|
||||||
mouseCallback_(event, mouse_callback_cookie_);
|
|
||||||
Superclass::OnMiddleButtonDown();
|
|
||||||
}
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
void cv::viz::InteractorStyle::OnMiddleButtonUp()
|
|
||||||
{
|
|
||||||
Vec2i p(Interactor->GetEventPosition());
|
|
||||||
MouseEvent event(MouseEvent::MouseButtonRelease, MouseEvent::MiddleButton, p, getModifiers());
|
|
||||||
if (mouseCallback_)
|
|
||||||
mouseCallback_(event, mouse_callback_cookie_);
|
|
||||||
Superclass::OnMiddleButtonUp();
|
|
||||||
}
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
void cv::viz::InteractorStyle::OnRightButtonDown()
|
|
||||||
{
|
|
||||||
Vec2i p(Interactor->GetEventPosition());
|
|
||||||
MouseEvent::Type type = (Interactor->GetRepeatCount() == 0) ? MouseEvent::MouseButtonPress : MouseEvent::MouseDblClick;
|
|
||||||
MouseEvent event(type, MouseEvent::RightButton, p, getModifiers());
|
|
||||||
if (mouseCallback_)
|
|
||||||
mouseCallback_(event, mouse_callback_cookie_);
|
|
||||||
Superclass::OnRightButtonDown();
|
|
||||||
}
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
void cv::viz::InteractorStyle::OnRightButtonUp()
|
|
||||||
{
|
|
||||||
Vec2i p(Interactor->GetEventPosition());
|
|
||||||
MouseEvent event(MouseEvent::MouseButtonRelease, MouseEvent::RightButton, p, getModifiers());
|
|
||||||
if (mouseCallback_)
|
|
||||||
mouseCallback_(event, mouse_callback_cookie_);
|
|
||||||
Superclass::OnRightButtonUp();
|
|
||||||
}
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
void cv::viz::InteractorStyle::OnMouseWheelForward()
|
|
||||||
{
|
|
||||||
Vec2i p(Interactor->GetEventPosition());
|
|
||||||
MouseEvent event(MouseEvent::MouseScrollUp, MouseEvent::VScroll, p, getModifiers());
|
|
||||||
if (mouseCallback_)
|
|
||||||
mouseCallback_(event, mouse_callback_cookie_);
|
|
||||||
if (Interactor->GetRepeatCount() && mouseCallback_)
|
|
||||||
mouseCallback_(event, mouse_callback_cookie_);
|
|
||||||
|
|
||||||
if (Interactor->GetAltKey())
|
|
||||||
{
|
|
||||||
// zoom
|
|
||||||
vtkSmartPointer<vtkCamera> cam = CurrentRenderer->GetActiveCamera();
|
|
||||||
double opening_angle = cam->GetViewAngle();
|
|
||||||
if (opening_angle > 15.0)
|
|
||||||
opening_angle -= 1.0;
|
|
||||||
|
|
||||||
cam->SetViewAngle(opening_angle);
|
|
||||||
cam->Modified();
|
|
||||||
CurrentRenderer->ResetCameraClippingRange();
|
|
||||||
CurrentRenderer->Modified();
|
|
||||||
Interactor->Render();
|
|
||||||
}
|
|
||||||
else
|
|
||||||
Superclass::OnMouseWheelForward();
|
|
||||||
}
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
void cv::viz::InteractorStyle::OnMouseWheelBackward()
|
|
||||||
{
|
|
||||||
Vec2i p(Interactor->GetEventPosition());
|
|
||||||
MouseEvent event(MouseEvent::MouseScrollDown, MouseEvent::VScroll, p, getModifiers());
|
|
||||||
if (mouseCallback_)
|
|
||||||
mouseCallback_(event, mouse_callback_cookie_);
|
|
||||||
|
|
||||||
if (Interactor->GetRepeatCount() && mouseCallback_)
|
|
||||||
mouseCallback_(event, mouse_callback_cookie_);
|
|
||||||
|
|
||||||
if (Interactor->GetAltKey())
|
|
||||||
{
|
|
||||||
// zoom
|
|
||||||
vtkSmartPointer<vtkCamera> cam = CurrentRenderer->GetActiveCamera();
|
|
||||||
double opening_angle = cam->GetViewAngle();
|
|
||||||
if (opening_angle < 170.0)
|
|
||||||
opening_angle += 1.0;
|
|
||||||
|
|
||||||
cam->SetViewAngle(opening_angle);
|
|
||||||
cam->Modified();
|
|
||||||
CurrentRenderer->ResetCameraClippingRange();
|
|
||||||
CurrentRenderer->Modified();
|
|
||||||
Interactor->Render();
|
|
||||||
}
|
|
||||||
else
|
|
||||||
Superclass::OnMouseWheelBackward();
|
|
||||||
}
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////////////////////
|
|
||||||
void cv::viz::InteractorStyle::OnTimer()
|
|
||||||
{
|
|
||||||
CV_Assert("Interactor style not initialized." && init_);
|
|
||||||
Interactor->Render();
|
|
||||||
}
|
|
||||||
@ -130,6 +130,7 @@
|
|||||||
#include <vtkElevationFilter.h>
|
#include <vtkElevationFilter.h>
|
||||||
#include <vtkColorTransferFunction.h>
|
#include <vtkColorTransferFunction.h>
|
||||||
#include <vtkStreamingDemandDrivenPipeline.h>
|
#include <vtkStreamingDemandDrivenPipeline.h>
|
||||||
|
#include "vtkCallbackCommand.h"
|
||||||
|
|
||||||
#if !defined(_WIN32) || defined(__CYGWIN__)
|
#if !defined(_WIN32) || defined(__CYGWIN__)
|
||||||
# include <unistd.h> /* unlink */
|
# include <unistd.h> /* unlink */
|
||||||
@ -145,6 +146,7 @@
|
|||||||
#include <vtk/vtkTrajectorySource.h>
|
#include <vtk/vtkTrajectorySource.h>
|
||||||
#include <vtk/vtkImageMatSource.h>
|
#include <vtk/vtkImageMatSource.h>
|
||||||
|
|
||||||
|
|
||||||
#include <opencv2/core/core.hpp>
|
#include <opencv2/core/core.hpp>
|
||||||
#include <opencv2/viz/vizcore.hpp>
|
#include <opencv2/viz/vizcore.hpp>
|
||||||
#include <opencv2/viz/widget_accessor.hpp>
|
#include <opencv2/viz/widget_accessor.hpp>
|
||||||
@ -332,8 +334,7 @@ namespace cv
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#include "interactor_style.hpp"
|
#include "vtk/vtkVizInteractorStyle.hpp"
|
||||||
#include "vizimpl.hpp"
|
#include "vizimpl.hpp"
|
||||||
|
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|||||||
@ -87,6 +87,7 @@ cv::Affine3d cv::viz::makeCameraPose(const Vec3d& position, const Vec3d& focal_p
|
|||||||
|
|
||||||
#else
|
#else
|
||||||
|
|
||||||
|
void register_console_handler();
|
||||||
void register_console_handler() {}
|
void register_console_handler() {}
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|||||||
@ -60,10 +60,9 @@ cv::viz::Viz3d::VizImpl::VizImpl(const String &name) : spin_once_state_(false),
|
|||||||
window_->AddRenderer(renderer_);
|
window_->AddRenderer(renderer_);
|
||||||
|
|
||||||
// Create the interactor style
|
// Create the interactor style
|
||||||
style_ = vtkSmartPointer<InteractorStyle>::New();
|
style_ = vtkSmartPointer<vtkVizInteractorStyle>::New();
|
||||||
style_->setWidgetActorMap(widget_actor_map_);
|
style_->setWidgetActorMap(widget_actor_map_);
|
||||||
style_->UseTimersOn();
|
style_->UseTimersOn();
|
||||||
style_->Initialize();
|
|
||||||
|
|
||||||
timer_callback_ = vtkSmartPointer<TimerCallback>::New();
|
timer_callback_ = vtkSmartPointer<TimerCallback>::New();
|
||||||
exit_callback_ = vtkSmartPointer<ExitCallback>::New();
|
exit_callback_ = vtkSmartPointer<ExitCallback>::New();
|
||||||
|
|||||||
@ -128,7 +128,7 @@ private:
|
|||||||
vtkSmartPointer<ExitCallback> exit_callback_;
|
vtkSmartPointer<ExitCallback> exit_callback_;
|
||||||
|
|
||||||
vtkSmartPointer<vtkRenderer> renderer_;
|
vtkSmartPointer<vtkRenderer> renderer_;
|
||||||
vtkSmartPointer<InteractorStyle> style_;
|
vtkSmartPointer<vtkVizInteractorStyle> style_;
|
||||||
Ptr<WidgetActorMap> widget_actor_map_;
|
Ptr<WidgetActorMap> widget_actor_map_;
|
||||||
|
|
||||||
bool removeActorFromRenderer(vtkSmartPointer<vtkProp> actor);
|
bool removeActorFromRenderer(vtkSmartPointer<vtkProp> actor);
|
||||||
|
|||||||
@ -40,8 +40,7 @@
|
|||||||
// Authors:
|
// Authors:
|
||||||
// * Anatoly Baksheev, Itseez Inc. myname.mysurname <> mycompany.com
|
// * Anatoly Baksheev, Itseez Inc. myname.mysurname <> mycompany.com
|
||||||
//
|
//
|
||||||
// OpenCV Viz module is complete rewrite of
|
// This workaround code was taken from PCL library(www.pointclouds.org)
|
||||||
// PCL visualization module (www.pointclouds.org)
|
|
||||||
//
|
//
|
||||||
//M*/
|
//M*/
|
||||||
|
|
||||||
|
|||||||
@ -84,7 +84,7 @@ void cv::viz::vtkOBJWriter::WriteData()
|
|||||||
std::ostream& outfile = *outfilep;
|
std::ostream& outfile = *outfilep;
|
||||||
|
|
||||||
//write header
|
//write header
|
||||||
outfile << "# wavefront obj file written by the visualization toolkit" << std::endl << std::endl;
|
outfile << "# wavefront obj file written by opencv viz module" << std::endl << std::endl;
|
||||||
outfile << "mtllib NONE" << std::endl << std::endl;
|
outfile << "mtllib NONE" << std::endl << std::endl;
|
||||||
|
|
||||||
// write out the points
|
// write out the points
|
||||||
|
|||||||
1077
modules/viz/src/vtk/vtkVizInteractorStyle.cpp
Normal file
1077
modules/viz/src/vtk/vtkVizInteractorStyle.cpp
Normal file
File diff suppressed because it is too large
Load Diff
@ -46,46 +46,23 @@
|
|||||||
#ifndef __OPENCV_VIZ_INTERACTOR_STYLE_H__
|
#ifndef __OPENCV_VIZ_INTERACTOR_STYLE_H__
|
||||||
#define __OPENCV_VIZ_INTERACTOR_STYLE_H__
|
#define __OPENCV_VIZ_INTERACTOR_STYLE_H__
|
||||||
|
|
||||||
|
#include <vtkInteractorStyle.h>
|
||||||
|
|
||||||
namespace cv
|
namespace cv
|
||||||
{
|
{
|
||||||
namespace viz
|
namespace viz
|
||||||
{
|
{
|
||||||
class InteractorStyle : public vtkInteractorStyleTrackballCamera
|
class vtkVizInteractorStyle : public vtkInteractorStyle
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
static InteractorStyle *New();
|
static vtkVizInteractorStyle *New();
|
||||||
virtual ~InteractorStyle() {}
|
vtkTypeMacro(vtkVizInteractorStyle, vtkInteractorStyle)
|
||||||
|
void PrintSelf(ostream& os, vtkIndent indent);
|
||||||
|
|
||||||
// this macro defines Superclass, the isA functionality and the safe downcast method
|
|
||||||
vtkTypeMacro(InteractorStyle, vtkInteractorStyleTrackballCamera)
|
|
||||||
|
|
||||||
/** \brief Initialization routine. Must be called before anything else. */
|
|
||||||
virtual void Initialize();
|
|
||||||
|
|
||||||
void setWidgetActorMap(const Ptr<WidgetActorMap>& actors) { widget_actor_map_ = actors; }
|
|
||||||
void registerMouseCallback(void (*callback)(const MouseEvent&, void*), void* cookie = 0);
|
|
||||||
void registerKeyboardCallback(void (*callback)(const KeyboardEvent&, void*), void * cookie = 0);
|
|
||||||
void saveScreenshot(const String &file);
|
|
||||||
void exportScene(const String &file);
|
|
||||||
|
|
||||||
private:
|
|
||||||
/** \brief Set to true after initialization is complete. */
|
|
||||||
bool init_;
|
|
||||||
|
|
||||||
Ptr<WidgetActorMap> widget_actor_map_;
|
|
||||||
|
|
||||||
Vec2i win_size_;
|
|
||||||
Vec2i win_pos_;
|
|
||||||
Vec2i max_win_size_;
|
|
||||||
|
|
||||||
/** \brief Interactor style internal method. Gets called whenever a key is pressed. */
|
|
||||||
virtual void OnChar();
|
virtual void OnChar();
|
||||||
|
|
||||||
// Keyboard events
|
|
||||||
virtual void OnKeyDown();
|
virtual void OnKeyDown();
|
||||||
virtual void OnKeyUp();
|
virtual void OnKeyUp();
|
||||||
|
|
||||||
// mouse button events
|
|
||||||
virtual void OnMouseMove();
|
virtual void OnMouseMove();
|
||||||
virtual void OnLeftButtonDown();
|
virtual void OnLeftButtonDown();
|
||||||
virtual void OnLeftButtonUp();
|
virtual void OnLeftButtonUp();
|
||||||
@ -95,15 +72,75 @@ namespace cv
|
|||||||
virtual void OnRightButtonUp();
|
virtual void OnRightButtonUp();
|
||||||
virtual void OnMouseWheelForward();
|
virtual void OnMouseWheelForward();
|
||||||
virtual void OnMouseWheelBackward();
|
virtual void OnMouseWheelBackward();
|
||||||
|
|
||||||
/** \brief Interactor style internal method. Gets called periodically if a timer is set. */
|
|
||||||
virtual void OnTimer();
|
virtual void OnTimer();
|
||||||
|
|
||||||
|
virtual void Rotate();
|
||||||
|
virtual void Spin();
|
||||||
|
virtual void Pan();
|
||||||
|
virtual void Dolly();
|
||||||
|
|
||||||
|
vtkSetMacro(FlyMode,bool)
|
||||||
|
vtkGetMacro(FlyMode,bool)
|
||||||
|
|
||||||
|
|
||||||
|
vtkSetMacro(MotionFactor, double)
|
||||||
|
vtkGetMacro(MotionFactor, double)
|
||||||
|
|
||||||
|
void registerMouseCallback(void (*callback)(const MouseEvent&, void*), void* cookie = 0);
|
||||||
|
void registerKeyboardCallback(void (*callback)(const KeyboardEvent&, void*), void * cookie = 0);
|
||||||
|
|
||||||
|
void setWidgetActorMap(const Ptr<WidgetActorMap>& actors) { widget_actor_map_ = actors; }
|
||||||
|
void saveScreenshot(const String &file);
|
||||||
|
void exportScene(const String &file);
|
||||||
|
void exportScene();
|
||||||
|
void changePointsSize(float delta);
|
||||||
|
void setRepresentationToPoints();
|
||||||
|
void printCameraParams();
|
||||||
|
void toggleFullScreen();
|
||||||
|
void resetViewerPose();
|
||||||
|
void toggleStereo();
|
||||||
|
void printHelp();
|
||||||
|
|
||||||
|
// Set the basic unit step size : by default 1/250 of bounding diagonal
|
||||||
|
vtkSetMacro(MotionStepSize,double)
|
||||||
|
vtkGetMacro(MotionStepSize,double)
|
||||||
|
|
||||||
|
// Set acceleration factor when shift key is applied : default 10
|
||||||
|
vtkSetMacro(MotionAccelerationFactor,double)
|
||||||
|
vtkGetMacro(MotionAccelerationFactor,double)
|
||||||
|
|
||||||
|
// Set the basic angular unit for turning : efault 1 degree
|
||||||
|
vtkSetMacro(AngleStepSize,double)
|
||||||
|
vtkGetMacro(AngleStepSize,double)
|
||||||
|
|
||||||
|
private:
|
||||||
|
Ptr<WidgetActorMap> widget_actor_map_;
|
||||||
|
|
||||||
|
Vec2i win_size_;
|
||||||
|
Vec2i win_pos_;
|
||||||
|
Vec2i max_win_size_;
|
||||||
|
|
||||||
void zoomIn();
|
void zoomIn();
|
||||||
void zoomOut();
|
void zoomOut();
|
||||||
|
|
||||||
/** \brief True if we're using red-blue colors for anaglyphic stereo, false if magenta-green. */
|
protected:
|
||||||
bool stereo_anaglyph_mask_default_;
|
vtkVizInteractorStyle();
|
||||||
|
~vtkVizInteractorStyle();
|
||||||
|
|
||||||
|
virtual void Dolly(double factor);
|
||||||
|
|
||||||
|
void Fly();
|
||||||
|
void FlyByMouse();
|
||||||
|
void FlyByKey();
|
||||||
|
void SetupMotionVars();
|
||||||
|
void MotionAlongVector(const Vec3d& vector, double amount, vtkCamera* cam);
|
||||||
|
|
||||||
|
private:
|
||||||
|
vtkVizInteractorStyle(const vtkVizInteractorStyle&);
|
||||||
|
vtkVizInteractorStyle& operator=(const vtkVizInteractorStyle&);
|
||||||
|
|
||||||
|
//! True for red-blue colors, false for magenta-green.
|
||||||
|
bool stereo_anaglyph_redblue_;
|
||||||
|
|
||||||
void (*keyboardCallback_)(const KeyboardEvent&, void*);
|
void (*keyboardCallback_)(const KeyboardEvent&, void*);
|
||||||
void *keyboard_callback_cookie_;
|
void *keyboard_callback_cookie_;
|
||||||
@ -111,7 +148,20 @@ namespace cv
|
|||||||
void (*mouseCallback_)(const MouseEvent&, void*);
|
void (*mouseCallback_)(const MouseEvent&, void*);
|
||||||
void *mouse_callback_cookie_;
|
void *mouse_callback_cookie_;
|
||||||
|
|
||||||
|
bool FlyMode;
|
||||||
|
double MotionFactor;
|
||||||
|
|
||||||
int getModifiers();
|
int getModifiers();
|
||||||
|
|
||||||
|
// from fly
|
||||||
|
unsigned char KeysDown;
|
||||||
|
double DiagonalLength;
|
||||||
|
double MotionStepSize;
|
||||||
|
double MotionUserScale;
|
||||||
|
double MotionAccelerationFactor;
|
||||||
|
double AngleStepSize;
|
||||||
|
double DeltaYaw;
|
||||||
|
double DeltaPitch;
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Loading…
Reference in New Issue
Block a user