ImageDisplay: toggle fullscreen button

ros-visualization · Sep 16, 2023 · 767911c · 767911c
1 parent d560273
commit 767911c
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/src/rviz/default_plugin/image_display.cpp b/src/rviz/default_plugin/image_display.cpp
@@ -42,8 +42,12 @@
 #include <OgreTechnique.h>
 #include <OgreCamera.h>
 
+#include <QToolButton>
+#include <QHBoxLayout>
+
 #include <rviz/display_context.h>
 #include <rviz/frame_manager.h>
+#include <rviz/panel_dock_widget.h>
 #include <rviz/ogre_helpers/compatibility.h>
 #include <rviz/render_panel.h>
 #include <rviz/validate_floats.h>
@@ -126,6 +130,14 @@ void ImageDisplay::onInitialize()
 
   setAssociatedWidget(render_panel_);
 
+  // fullscreen button
+  QToolButton* fullscreen_button = new QToolButton();
+  fullscreen_button->setText("FullScreen");
+  fullscreen_button->setIconSize(QSize(10, 10));
+  connect(fullscreen_button, &QToolButton::clicked, this, &ImageDisplay::toggleFullScreen);
+  dynamic_cast<QHBoxLayout*>(getAssociatedWidgetPanel()->titleBarWidget()->layout())
+      ->insertWidget(2, fullscreen_button);
+
   render_panel_->setAutoRender(false);
   render_panel_->setOverlaysEnabled(false);
   render_panel_->getCamera()->setNearClipDistance(0.01f);
@@ -156,6 +168,21 @@ void ImageDisplay::onDisable()
   reset();
 }
 
+void ImageDisplay::toggleFullScreen()
+{
+  auto* panel = getAssociatedWidgetPanel();
+  if (panel->windowState() & Qt::WindowFullScreen)
+  {
+    panel->setFloating(false);
+    panel->showNormal();
+  }
+  else
+  {
+    panel->setFloating(true);
+    panel->showFullScreen();
+  }
+}
+
 void ImageDisplay::updateNormalizeOptions()
 {
   if (got_float_image_)

diff --git a/src/rviz/default_plugin/image_display.h b/src/rviz/default_plugin/image_display.h
@@ -73,6 +73,7 @@ class ImageDisplay : public ImageDisplayBase
 
 public Q_SLOTS:
   virtual void updateNormalizeOptions();
+  virtual void toggleFullScreen();
 
 protected:
   // overrides from Display