added option to change filemode (ros-visualization#1537)

Author: jsupratman13

rviz_common/include/rviz_common/properties/file_picker_property.hpp

@@ -31,7 +31,11 @@
 #ifndef RVIZ_COMMON__PROPERTIES__FILE_PICKER_PROPERTY_HPP_
 #define RVIZ_COMMON__PROPERTIES__FILE_PICKER_PROPERTY_HPP_
 
+#include <QFileDialog>
+#include <QObject>
 #include <QString>
+#include <QStyleOptionViewItem>
+#include <QWidget>
 
 #include "rviz_common/properties/string_property.hpp"
 #include "rviz_common/visibility_control.hpp"
@@ -49,9 +53,13 @@ class RVIZ_COMMON_PUBLIC FilePickerProperty : public StringProperty
     const QString & description = QString(),
     Property * parent = nullptr,
     const char * changed_slot = nullptr,
-    QObject * receiver = nullptr);
+    QObject * receiver = nullptr,
+    QFileDialog::FileMode mode = QFileDialog::AnyFile);
 
   QWidget * createEditor(QWidget * parent, const QStyleOptionViewItem & option) override;
+
+private:
+  QFileDialog::FileMode mode_;
 };
 
 }  // namespace properties

rviz_common/src/rviz_common/properties/file_picker.cpp

@@ -32,6 +32,7 @@
 
 #include <QFileDialog>
 #include <QString>
+#include <QWidget>
 
 
 namespace rviz_common
@@ -41,14 +42,17 @@ namespace properties
 
 FilePicker::FilePicker(
   StringProperty * file_name_property,
-  QWidget * parent)
+  QWidget * parent,
+  QFileDialog::FileMode mode)
 : LineEditWithButton(parent),
-  file_name_property_(file_name_property)
+  file_name_property_(file_name_property),
+  mode_(mode)
 {}
 
 void FilePicker::onButtonClick()
 {
   auto dialog = new QFileDialog(parentWidget());
+  dialog->setFileMode(mode_);
 
   connect(
     dialog, SIGNAL(fileSelected(const QString&)),

rviz_common/src/rviz_common/properties/file_picker.hpp

@@ -31,6 +31,9 @@
 #ifndef RVIZ_COMMON__PROPERTIES__FILE_PICKER_HPP_
 #define RVIZ_COMMON__PROPERTIES__FILE_PICKER_HPP_
 
+#include <QFileDialog>
+#include <QWidget>
+
 #include "rviz_common/properties/line_edit_with_button.hpp"
 #include "rviz_common/properties/string_property.hpp"
 #include "rviz_common/visibility_control.hpp"
@@ -45,13 +48,15 @@ class FilePicker : public LineEditWithButton
 public:
   explicit FilePicker(
     StringProperty * file_name_property = nullptr,
-    QWidget * parent = nullptr);
+    QWidget * parent = nullptr,
+    QFileDialog::FileMode mode = QFileDialog::AnyFile);
 
 protected:
   void onButtonClick() override;
 
 private:
   StringProperty * file_name_property_;
+  QFileDialog::FileMode mode_;
 };
 
 }  // namespace properties

rviz_common/src/rviz_common/properties/file_picker_property.cpp

@@ -30,7 +30,11 @@
 
 #include "rviz_common/properties/file_picker_property.hpp"
 
+#include <QFileDialog>
+#include <QObject>
 #include <QString>
+#include <QStyleOptionViewItem>
+#include <QWidget>
 
 #include "file_picker.hpp"
 
@@ -45,14 +49,15 @@ FilePickerProperty::FilePickerProperty(
   const QString & description,
   Property * parent,
   const char * changed_slot,
-  QObject * receiver)
-: StringProperty(name, default_value, description, parent, changed_slot, receiver)
+  QObject * receiver,
+  QFileDialog::FileMode mode)
+: StringProperty(name, default_value, description, parent, changed_slot, receiver), mode_(mode)
 {}
 
 QWidget * FilePickerProperty::createEditor(QWidget * parent, const QStyleOptionViewItem & option)
 {
   (void) option;
-  auto file_picker = new FilePicker(this, parent);
+  auto file_picker = new FilePicker(this, parent, mode_);
   file_picker->setFrame(false);
   return file_picker;
 }