jspanchu / vtkdearimguiinjector Goto Github PK

class MyCallback : public vtkCommand
{
public:
    MyCallback() = default;
    ~MyCallback() = default;

    static MyCallback* New()
    {
        return new MyCallback;
    }

    void Execute(vtkObject* caller, unsigned long eventId,
        void* vtkNotUsed(callData)) override
    {
        if (vtkCommand::LeftButtonPressEvent == eventId)
        {
            std::cout << "LeftButtonPressEvent" << std::endl;
        }
        else if (vtkCommand::MouseMoveEvent == eventId)
        {
            std::cout << "MouseMoveEvent" << std::endl;

        }
        else if (vtkCommand::LeftButtonReleaseEvent == eventId)
        {
            std::cout << "LeftButtonReleaseEvent" << std::endl;
        }
    }
};

//------------------------------------------------------------------------------
// Main
//------------------------------------------------------------------------------

int main(int argc, char* argv[])
{
  // Create a renderer, render window, and interactor
  vtkNew<vtkRenderer> renderer;
  vtkNew<vtkRenderWindow> renderWindow;
  vtkNew<vtkRenderWindowInteractor> iren;
  renderWindow->SetMultiSamples(8);
  renderWindow->AddRenderer(renderer);
  iren->SetRenderWindow(renderWindow);

  // Create pipeline
  vtkNew<vtkConeSource> coneSource;
  coneSource->Update();

  vtkNew<vtkPolyDataMapper> mapper;
  mapper->SetInputConnection(coneSource->GetOutputPort());

  vtkNew<vtkActor> actor;
  actor->SetMapper(mapper);

  // Add the actors to the scene
  renderer->AddActor(actor);

  // Start rendering app
  renderer->SetBackground(0.2, 0.3, 0.4);
  renderWindow->Render();

  /// Change to your code begins here. ///
  // Initialize an overlay with DearImgui elements.
  vtkNew<vtkDearImGuiInjector> dearImGuiOverlay;
  // 💉 the overlay.
  dearImGuiOverlay->Inject(iren);
  // These functions add callbacks to ImGuiSetupEvent and ImGuiDrawEvents.
  SetupUI(dearImGuiOverlay);
  // You can draw custom user interface elements using ImGui:: namespace.
  DrawUI(dearImGuiOverlay);
  /// Change to your code ends here. ///

  vtkNew<vtkCameraOrientationWidget> camManipulator;
  camManipulator->SetParentRenderer(renderer);
  camManipulator->On();
  auto rep = vtkCameraOrientationRepresentation::SafeDownCast(camManipulator->GetRepresentation());
  rep->AnchorToLowerRight();

  // Start event loop
  renderWindow->SetSize(1920, 1000);
  vtkInteractorStyleSwitch::SafeDownCast(iren->GetInteractorStyle())->SetCurrentStyleToTrackballCamera();
  iren->EnableRenderOff();
  vtkNew<MyCallback> cb;
  iren->AddObserver(vtkCommand::LeftButtonPressEvent, cb);
  iren->AddObserver(vtkCommand::MouseMoveEvent, cb);
  iren->AddObserver(vtkCommand::LeftButtonReleaseEvent, cb);
  iren->Start();

  return 0;
}