feat: add projection toggle button to switch between perspective and orthographic views
This commit is contained in:
@@ -72,6 +72,7 @@ struct ViewerData {
|
||||
vtkRenderWindow *m_RenderWindow;
|
||||
vtkSmartPointer<vtkRenderWindowInteractor> m_Interactor;
|
||||
vtkSmartPointer<vtkButtonWidget> m_GridButton;
|
||||
vtkSmartPointer<vtkButtonWidget> m_ProjButton;
|
||||
|
||||
ViewerData() : m_RenderWindow(vtkRenderWindow::New()) {}
|
||||
~ViewerData() {
|
||||
@@ -97,6 +98,11 @@ Viewer::~Viewer() {
|
||||
dv->m_GridButton->SetInteractor(nullptr);
|
||||
dv->m_GridButton = nullptr;
|
||||
}
|
||||
if (dv->m_ProjButton) {
|
||||
dv->m_ProjButton->Off();
|
||||
dv->m_ProjButton->SetInteractor(nullptr);
|
||||
dv->m_ProjButton = nullptr;
|
||||
}
|
||||
if (this->GetRenderWindow()) {
|
||||
this->GetRenderWindow()->RemoveAllObservers();
|
||||
}
|
||||
@@ -123,6 +129,7 @@ void Viewer::InstallPipe() {
|
||||
// Setup native grid button
|
||||
if (!std::getenv("CTEST_PROJECT_NAME")) {
|
||||
SetupGridButton();
|
||||
SetupProjButton();
|
||||
}
|
||||
|
||||
// BUT we want to override the style with our custom NoSpin version
|
||||
@@ -238,9 +245,91 @@ void Viewer::UpdateGridButtonPosition() {
|
||||
rep->PlaceWidget(bds);
|
||||
}
|
||||
|
||||
void Viewer::Start() {
|
||||
void Viewer::SetupProjButton() {
|
||||
if (!dv->m_RenderWindow || !dv->m_RenderWindow->GetInteractor()) return;
|
||||
|
||||
vtkNew<vtkImageCanvasSource2D> canvas;
|
||||
canvas->SetScalarTypeToUnsignedChar();
|
||||
canvas->SetNumberOfScalarComponents(4);
|
||||
canvas->SetExtent(0, 63, 0, 63, 0, 0);
|
||||
|
||||
// State 0: Perspective (gray trapezoid-like lines)
|
||||
canvas->SetDrawColor(0, 0, 0, 0);
|
||||
canvas->FillBox(0, 63, 0, 63);
|
||||
canvas->SetDrawColor(120, 120, 120, 255);
|
||||
canvas->DrawSegment(16, 16, 48, 16);
|
||||
canvas->DrawSegment(48, 16, 56, 48);
|
||||
canvas->DrawSegment(56, 48, 8, 48);
|
||||
canvas->DrawSegment(8, 48, 16, 16);
|
||||
canvas->Update();
|
||||
|
||||
vtkNew<vtkImageData> imgPersp;
|
||||
imgPersp->DeepCopy(canvas->GetOutput());
|
||||
|
||||
// State 1: Orthographic (white rectangle)
|
||||
canvas->SetDrawColor(0, 0, 0, 0);
|
||||
canvas->FillBox(0, 63, 0, 63);
|
||||
canvas->SetDrawColor(255, 255, 255, 255);
|
||||
canvas->DrawSegment(12, 16, 52, 16);
|
||||
canvas->DrawSegment(52, 16, 52, 48);
|
||||
canvas->DrawSegment(52, 48, 12, 48);
|
||||
canvas->DrawSegment(12, 48, 12, 16);
|
||||
canvas->Update();
|
||||
|
||||
vtkNew<vtkImageData> imgOrtho;
|
||||
imgOrtho->DeepCopy(canvas->GetOutput());
|
||||
|
||||
vtkNew<vtkTexturedButtonRepresentation2D> rep;
|
||||
rep->SetNumberOfStates(2);
|
||||
rep->SetButtonTexture(0, imgPersp);
|
||||
rep->SetButtonTexture(1, imgOrtho);
|
||||
|
||||
dv->m_ProjButton = vtkSmartPointer<vtkButtonWidget>::New();
|
||||
dv->m_ProjButton->SetInteractor(dv->m_RenderWindow->GetInteractor());
|
||||
dv->m_ProjButton->SetRepresentation(rep);
|
||||
|
||||
UpdateProjButtonPosition();
|
||||
|
||||
vtkNew<vtkCallbackCommand> resizeCallback;
|
||||
resizeCallback->SetClientData(this);
|
||||
resizeCallback->SetCallback([](vtkObject*, unsigned long, void* clientdata, void*){
|
||||
static_cast<Viewer*>(clientdata)->UpdateProjButtonPosition();
|
||||
});
|
||||
dv->m_RenderWindow->AddObserver(vtkCommand::ModifiedEvent, resizeCallback);
|
||||
|
||||
vtkNew<vtkCallbackCommand> stateCallback;
|
||||
stateCallback->SetClientData(this);
|
||||
stateCallback->SetCallback([](vtkObject* caller, unsigned long, void* clientdata, void*){
|
||||
auto* btn = vtkButtonWidget::SafeDownCast(caller);
|
||||
auto* v = static_cast<Viewer*>(clientdata);
|
||||
auto* r = vtkTexturedButtonRepresentation2D::SafeDownCast(btn->GetRepresentation());
|
||||
v->SetParallelProjection(r->GetState() == 1);
|
||||
});
|
||||
dv->m_ProjButton->AddObserver(vtkCommand::StateChangedEvent, stateCallback);
|
||||
dv->m_ProjButton->On();
|
||||
|
||||
rep->SetState(GetParallelProjection() ? 1 : 0);
|
||||
}
|
||||
|
||||
void Viewer::UpdateProjButtonPosition() {
|
||||
if (!dv->m_ProjButton || !dv->m_RenderWindow) return;
|
||||
auto* rep = vtkTexturedButtonRepresentation2D::SafeDownCast(dv->m_ProjButton->GetRepresentation());
|
||||
if (!rep) return;
|
||||
|
||||
int *sz = dv->m_RenderWindow->GetSize();
|
||||
if (sz[0] == 0 || sz[1] == 0) return;
|
||||
|
||||
int margin_right = 23;
|
||||
int margin_top = 220; // below the grid button (170 + 50)
|
||||
int btnSz = 100;
|
||||
double bds[6] = { (double)sz[0] - btnSz - margin_right, (double)sz[0] - margin_right,
|
||||
(double)sz[1] - margin_top - btnSz/2.0, (double)sz[1] - margin_top + btnSz/2.0, 0, 0 };
|
||||
rep->PlaceWidget(bds);
|
||||
}
|
||||
|
||||
void Viewer::Start() {
|
||||
if (std::getenv("CTEST_PROJECT_NAME")) return;
|
||||
dv->m_RenderWindow->GetInteractor()->Start();
|
||||
dv->m_RenderWindow->GetInteractor()->Start();
|
||||
}
|
||||
|
||||
vtkRenderWindow *Viewer::GetRenderWindow() { return dv->m_RenderWindow; }
|
||||
|
||||
Reference in New Issue
Block a user