mainwindow.hpp

Go to the documentation of this file.

#ifndef RQT_BAG_EXPORTER_MAINWINDOW_HPP
#define RQT_BAG_EXPORTER_MAINWINDOW_HPP

#include <QApplication>
#include <QCheckBox>
#include <QDoubleSpinBox>
#include <QFileDialog>
#include <QLabel>
#include <QList>
#include <QMainWindow>
#include <QMenuBar>
#include <QMessageBox>
#include <QPushButton>
#include <QSettings>
#include <QSpinBox>
#include <QString>
#include <QTabWidget>
#include <QWidget>
#include <actionlib/client/simple_action_client.h>
#include <ros/ros.h>
#include <rqt_bag_exporter/CloseBag.h>
#include <rqt_bag_exporter/EstimateVideoFps.h>
#include <rqt_bag_exporter/ExportToCSVAction.h>
#include <rqt_bag_exporter/ExportToVideoAction.h>
#include <rqt_bag_exporter/GetDuration.h>
#include <rqt_bag_exporter/ListTopics.h>
#include <rqt_bag_exporter/OpenBag.h>
#include <rqt_bag_exporter/progress_dialog.hpp>

class MainWindow : public QMainWindow
{
Q_OBJECT

public:
  MainWindow(QWidget *parent = 0);
  ~MainWindow();

Q_SIGNALS:
  void displayMessageBox(const QString,
                         const QString,
                         const QString,
                         const QMessageBox::Icon);

private Q_SLOTS:
  void closeBag();
  void exportTopic();
  void exportToDirectory();
  void openBagFile();
  void updateTimeBegin(const double time_end);
  void updateTimeEnd(const double time_begin);
  void updateFps(const int fps);
  void displayMessageBoxHandler(const QString title,
                                const QString message,
                                const QString info_msg = "",
                                const QMessageBox::Icon icon = QMessageBox::Icon::Information);

private:
  bool isCsvWritableTopic(const std::string topic_type);
  bool isCompressedImageTopic(const std::string topic_type);
  void displayTopics();
  void fillStartEndTime();
  void fillVideoTab(const std::string topic_name);
  void listTopicsInBagFile();
  void fillTopicsToExport();
  void csvDoneCb(const actionlib::SimpleClientGoalState &state,
                 const rqt_bag_exporter::ExportToCSVResultConstPtr &result);
  void csvFeedbackCb(const rqt_bag_exporter::ExportToCSVFeedbackConstPtr &feedback);
  void videoDoneCb(const actionlib::SimpleClientGoalState &state,
                   const rqt_bag_exporter::ExportToVideoResultConstPtr &result);
  void videoFeedbackCb(const rqt_bag_exporter::ExportToVideoFeedbackConstPtr &feedback);
  void closeEvent(QCloseEvent *event);
  void load();

  ros::NodeHandle nh_;

  std::shared_ptr<ProgressDialog> progress_dialog_;

  typedef actionlib::SimpleActionClient<rqt_bag_exporter::ExportToCSVAction> ExportToCSVActionClient;
  typedef actionlib::SimpleActionClient<rqt_bag_exporter::ExportToVideoAction> ExportToVideoActionClient;

  std::unique_ptr<ExportToCSVActionClient> csv_ac_;
  std::unique_ptr<ExportToVideoActionClient> video_ac_;

  QSettings* settings_;

  QDoubleSpinBox* time_begin_;
  QDoubleSpinBox* time_end_;
  QTabWidget* video_tab_;
  QWidget* start_end_time_;
  QWidget* topics_container_;
  rqt_bag_exporter::ListTopicsResponse topics_list_;
  rqt_bag_exporter::ListTopicsResponse topics_to_export_;
  std::string directory_;
  unsigned fps_;
};

#endif