exportformatcombobox: categorize formats where possible

2022-09-22 10:56:52 -07:00
parent 5824b4873b
commit f84cd8644d
2 changed files with 111 additions and 52 deletions
@@ -20,75 +20,120 @@

 #include "exportformatcombobox.h"

+#include <QHBoxLayout>
+#include <QLabel>
+
+#include "ui/icons/icons.h"
+
 namespace olive {

 ExportFormatComboBox::ExportFormatComboBox(Mode mode, QWidget *parent) :
  QComboBox(parent)
 {
-  // Populate combobox formats
-  for (int i=0; i<ExportFormat::kFormatCount; i++) {
-    ExportFormat::Format f = static_cast<ExportFormat::Format>(i);
+  custom_menu_ = new Menu(this);

+  // Populate combobox formats
  switch (mode) {
  case kShowAllFormats:
+    custom_menu_->addAction(CreateHeader(icon::Video, tr("Video")));
+    PopulateType(Track::kVideo);
+    custom_menu_->addSeparator();
+
+    custom_menu_->addAction(CreateHeader(icon::Audio, tr("Audio")));
+    PopulateType(Track::kAudio);
+    custom_menu_->addSeparator();
+
+    custom_menu_->addAction(CreateHeader(icon::Subtitles, tr("Subtitle")));
+    PopulateType(Track::kSubtitle);
    break;
  case kShowAudioOnly:
-      if (!ExportFormat::GetVideoCodecs(f).isEmpty()
-          || !ExportFormat::GetSubtitleCodecs(f).isEmpty()
-          || ExportFormat::GetAudioCodecs(f).isEmpty()) {
-        continue;
-      }
+    PopulateType(Track::kAudio);
    break;
  case kShowVideoOnly:
-      if (ExportFormat::GetVideoCodecs(f).isEmpty()
-          || !ExportFormat::GetSubtitleCodecs(f).isEmpty()
-          || !ExportFormat::GetAudioCodecs(f).isEmpty()) {
-        continue;
-      }
+    PopulateType(Track::kVideo);
    break;
  case kShowSubtitlesOnly:
-      if (!ExportFormat::GetVideoCodecs(f).isEmpty()
-          || ExportFormat::GetSubtitleCodecs(f).isEmpty()
-          || !ExportFormat::GetAudioCodecs(f).isEmpty()) {
-        continue;
-      }
+    PopulateType(Track::kSubtitle);
    break;
  }

-    QString format_name = ExportFormat::GetName(f);
-
-    bool inserted = false;
-
-    // Sort formats alphabetically
-    for (int j=0; j<count(); j++) {
-      if (itemText(j) > format_name) {
-        insertItem(j, format_name, i);
-        inserted = true;
-        break;
-      }
+  connect(custom_menu_, &Menu::triggered, this, &ExportFormatComboBox::HandleIndexChange);
 }

-    if (!inserted) {
-      addItem(format_name, i);
-    }
-  }
-
-  connect(this, static_cast<void(QComboBox::*)(int)>(&QComboBox::currentIndexChanged), this, &ExportFormatComboBox::HandleIndexChange);
+void ExportFormatComboBox::showPopup()
+{
+  custom_menu_->setMinimumWidth(this->width());
+  custom_menu_->exec(mapToGlobal(QPoint(0, 0)));
 }

 void ExportFormatComboBox::SetFormat(ExportFormat::Format fmt)
 {
-  for (int i=0; i<count(); i++) {
-    if (itemData(i).toInt() == fmt) {
-      setCurrentIndex(i);
-      break;
-    }
-  }
+  current_ = fmt;
+  clear();
+  addItem(ExportFormat::GetName(current_));
 }

-void ExportFormatComboBox::HandleIndexChange(int index)
+void ExportFormatComboBox::HandleIndexChange(QAction *a)
 {
-  emit FormatChanged(static_cast<ExportFormat::Format>(itemData(index).toInt()));
+  ExportFormat::Format f = static_cast<ExportFormat::Format>(a->data().toInt());
+  SetFormat(f);
+  emit FormatChanged(f);
+}
+
+void ExportFormatComboBox::PopulateType(Track::Type type)
+{
+  for (int i=0; i<ExportFormat::kFormatCount; i++) {
+    ExportFormat::Format f = static_cast<ExportFormat::Format>(i);
+
+    if (type == Track::kVideo
+        && !ExportFormat::GetVideoCodecs(f).isEmpty()) {
+      // Do nothing
+    } else if (type == Track::kAudio
+               && ExportFormat::GetVideoCodecs(f).isEmpty()
+               && !ExportFormat::GetAudioCodecs(f).isEmpty()) {
+      // Do nothing
+    } else if (type == Track::kSubtitle
+               && ExportFormat::GetVideoCodecs(f).isEmpty()
+               && ExportFormat::GetAudioCodecs(f).isEmpty()
+               && !ExportFormat::GetSubtitleCodecs(f).isEmpty()) {
+      // Do nothing
+    } else {
+      continue;
+    }
+
+    QString format_name = ExportFormat::GetName(f);
+
+    QAction *a = custom_menu_->addAction(format_name);
+    a->setData(i);
+    a->setIconVisibleInMenu(false);
+  }
+}
+
+QWidgetAction *ExportFormatComboBox::CreateHeader(const QIcon &icon, const QString &title)
+{
+  QWidgetAction *a = new QWidgetAction(this);
+
+  QWidget *w = new QWidget();
+  QHBoxLayout *layout = new QHBoxLayout(w);
+
+  QLabel *icon_lbl = new QLabel();
+
+  QLabel *text_lbl = new QLabel(title);
+  text_lbl->setAlignment(Qt::AlignCenter);
+  QFont f = text_lbl->font();
+  f.setWeight(QFont::Bold);
+  text_lbl->setFont(f);
+
+  icon_lbl->setPixmap(icon.pixmap(text_lbl->sizeHint()));
+
+  layout->addStretch();
+  layout->addWidget(icon_lbl);
+  layout->addWidget(text_lbl);
+  layout->addStretch();
+
+  a->setDefaultWidget(w);
+  a->setEnabled(false);
+  return a;
 }

 }
@@ -22,8 +22,11 @@
 #define EXPORTFORMATCOMBOBOX_H

 #include <QComboBox>
+#include <QWidgetAction>

 #include "codec/exportformat.h"
+#include "node/output/track/track.h"
+#include "widget/menu/menu.h"

 namespace olive {

@@ -45,9 +48,11 @@ public:

  ExportFormat::Format GetFormat() const
  {
-    return static_cast<ExportFormat::Format>(currentData().toInt());
+    return current_;
  }

+  void showPopup();
+
 signals:
  void FormatChanged(ExportFormat::Format fmt);

@@ -55,7 +60,16 @@ public slots:
  void SetFormat(ExportFormat::Format fmt);

 private slots:
-  void HandleIndexChange(int index);
+  void HandleIndexChange(QAction *a);
+
+private:
+  void PopulateType(Track::Type type);
+
+  QWidgetAction *CreateHeader(const QIcon &icon, const QString &title);
+
+  Menu *custom_menu_;
+
+  ExportFormat::Format current_;

 };