[feature][UI] Show recorded bag file and remaining disk space size in record widget

Author: Fleury

src/ui/interactive/ControlBagWidget.cpp

@@ -42,19 +42,19 @@ ControlBagWidget::ControlBagWidget(Parameters::SelectableBagTopicParameters& par
     buttonLayout->addWidget(stopButton);
     buttonLayout->setAlignment(stopButton, Qt::AlignLeft);
 
-    auto* const upperLayout = new QVBoxLayout;
-    upperLayout->addStretch();
-    upperLayout->addWidget(headerPixmapLabel);
-    upperLayout->addWidget(headerLabel);
-    upperLayout->addSpacing(40);
-    upperLayout->addLayout(m_controlsLayout);
-    upperLayout->addSpacing(10);
-    upperLayout->addWidget(m_loggerListWidget);
-    upperLayout->setContentsMargins(20, 20, 20, 20);
-    upperLayout->addStretch();
+    m_upperLayout = new QVBoxLayout;
+    m_upperLayout->addStretch();
+    m_upperLayout->addWidget(headerPixmapLabel);
+    m_upperLayout->addWidget(headerLabel);
+    m_upperLayout->addSpacing(40);
+    m_upperLayout->addLayout(m_controlsLayout);
+    m_upperLayout->addSpacing(10);
+    m_upperLayout->addWidget(m_loggerListWidget);
+    m_upperLayout->setContentsMargins(20, 20, 20, 20);
+    m_upperLayout->addStretch();
 
     auto* const mainLayout = new QVBoxLayout;
-    mainLayout->addLayout(upperLayout);
+    mainLayout->addLayout(m_upperLayout);
     mainLayout->addLayout(buttonLayout);
     setLayout(mainLayout);
 

src/ui/interactive/ControlBagWidget.hpp

@@ -8,6 +8,7 @@
 
 class QHBoxLayout;
 class QListWidget;
+class QVBoxLayout;
 
 // Widget used to control playing a bag file.
 class ControlBagWidget : public QWidget
@@ -46,6 +47,7 @@ protected slots:
 
 protected:
     QPointer<QHBoxLayout> m_controlsLayout;
+    QPointer<QVBoxLayout> m_upperLayout;
     QPointer<QToolButton> m_playPauseButton;
     QPointer<QListWidget> m_loggerListWidget;
 

src/ui/interactive/ControlRecordBagWidget.cpp

@@ -1,15 +1,35 @@
 #include "ControlRecordBagWidget.hpp"
 
+#include "UtilsGeneral.hpp"
+
 #include <QHBoxLayout>
+#include <QLabel>
 #include <QListWidget>
+#include <QTimer>
+
+#include <filesystem>
 
 ControlRecordBagWidget::ControlRecordBagWidget(Parameters::RecordBagParameters& parameters, QWidget* parent)
-    : ControlBagWidget(parameters, "Started recording Bag File\n", ":/icons/tools/record_bag_", true, parent)
+    : ControlBagWidget(parameters, "Started recording Bag File\n", ":/icons/tools/record_bag_", true, parent), m_parameters(parameters)
 {
     m_controlsLayout->addWidget(m_playPauseButton);
     m_controlsLayout->addStretch();
 
-    m_bagRecorder = std::make_unique<BagRecorder>(parameters);
+    m_bagSizeLabel = new QLabel("");
+    m_availableDiskSpaceLabel = new QLabel("");
+
+    auto font = m_bagSizeLabel->font();
+    font.setBold(true);
+    m_bagSizeLabel->setFont(font);
+    m_availableDiskSpaceLabel->setFont(font);
+
+    m_upperLayout->insertSpacing(7, 20);
+    m_upperLayout->insertWidget(8, m_bagSizeLabel);
+    m_upperLayout->insertWidget(9, m_availableDiskSpaceLabel);
+    m_upperLayout->setAlignment(m_bagSizeLabel, Qt::AlignHCenter);
+    m_upperLayout->setAlignment(m_availableDiskSpaceLabel, Qt::AlignHCenter);
+
+    m_bagRecorder = std::make_unique<BagRecorder>(m_parameters);
     // Simulate the first few terminal info messages displayed when calling ros2 bag play
     addRecordBagLoggerEntry("Press SPACE for pausing/resuming");
     addRecordBagLoggerEntry("Listening for topics...");
@@ -26,4 +46,34 @@ ControlRecordBagWidget::ControlRecordBagWidget(Parameters::RecordBagParameters&
     connect(m_bagRecorder.get(), &BagRecorder::allTopicsSubscribed, this, [this] {
         addRecordBagLoggerEntry("All requested topics are subscribed. Stopping discovery...");
     });
+
+    m_loggerListWidget->setFocus();
+
+    auto* const timer = new QTimer(this);
+    connect(timer, &QTimer::timeout, this, &ControlRecordBagWidget::updateSpaceInfoLabel);
+    timer->start(1000);
+}
+
+
+void
+ControlRecordBagWidget::updateSpaceInfoLabel()
+{
+    float bagFileSpace = 0;
+
+    std::error_code errorCode;
+    for (const auto& entry : std::filesystem::recursive_directory_iterator(m_parameters.sourceDirectory.toStdString(),
+                                                                           std::filesystem::directory_options::skip_permission_denied, errorCode)) {
+        if (!std::filesystem::is_regular_file(entry, errorCode)) {
+            continue;
+        }
+        bagFileSpace += static_cast<float>(std::filesystem::file_size(entry, errorCode));
+    }
+
+    const auto sizeIndicator = bagFileSpace < Utils::General::GIGABYTE_IN_BYTES? " MB" : " GB";
+    bagFileSpace /= bagFileSpace < Utils::General::GIGABYTE_IN_BYTES ? static_cast<float>(MEGABYTE_IN_BYTES)
+                                                                     : static_cast<float>(Utils::General::GIGABYTE_IN_BYTES);
+    m_bagSizeLabel->setText("Recorded Bag File Size: " + QString::number(bagFileSpace) + sizeIndicator);
+
+    const auto remainingSpace = Utils::General::getAvailableDriveSpace(m_parameters.sourceDirectory);
+    m_availableDiskSpaceLabel->setText("Available Disk Space: " + QString::number(remainingSpace) + "GB");
 }

src/ui/interactive/ControlRecordBagWidget.hpp

@@ -5,8 +5,11 @@
 #include "Parameters.hpp"
 #include "UtilsROS.hpp"
 
+#include <QPointer>
 #include <QWidget>
 
+class QLabel;
+
 // Widget used to control playing a bag file.
 class ControlRecordBagWidget : public ControlBagWidget
 {
@@ -17,6 +20,9 @@ class ControlRecordBagWidget : public ControlBagWidget
                            QWidget*                         parent = 0);
 
 private:
+    void
+    updateSpaceInfoLabel();
+
     void
     handleBagControlInstance() override
     {
@@ -31,4 +37,11 @@ class ControlRecordBagWidget : public ControlBagWidget
 
 private:
     std::unique_ptr<BagRecorder> m_bagRecorder;
+
+    QPointer<QLabel> m_bagSizeLabel;
+    QPointer<QLabel> m_availableDiskSpaceLabel;
+
+    const Parameters::RecordBagParameters& m_parameters;
+
+    static constexpr long MEGABYTE_IN_BYTES = 1048576;
 };