/***

  Olive - Non-Linear Video Editor
  Copyright (C) 2022 Olive Team
  Modifications Copyright (C) 2025 mikesolar

  This program is free software: you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation, either version 3 of the License, or
  (at your option) any later version.

  This program is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with this program.  If not, see <http://www.gnu.org/licenses/>.

***/

#include "exportformatcombobox.h"

#include <QHBoxLayout>
#include <QLabel>

#include "oakengine/encoding.h"
#include "ui/icons/icons.h"

namespace olive
{

ExportFormatComboBox::ExportFormatComboBox(Mode mode, QWidget *parent)
	: QComboBox(parent)
{
	// The invalid placeholder format is the format count itself
	// (ExportFormat::k_format_count), not -1.
	current_ = oakengine_encoding_format_count();

	custom_menu_ = new Menu(this);

	// Populate combobox formats
	switch (mode) {
	case k_show_all_formats:
		custom_menu_->addAction(create_header(icon::video, tr("Video")));
		populate_type(TrackReference::k_video);
		custom_menu_->addSeparator();

		custom_menu_->addAction(create_header(icon::audio, tr("Audio")));
		populate_type(TrackReference::k_audio);
		custom_menu_->addSeparator();

		custom_menu_->addAction(create_header(icon::subtitles, tr("Subtitle")));
		populate_type(TrackReference::k_subtitle);
		break;
	case k_show_audio_only:
		populate_type(TrackReference::k_audio);
		break;
	case k_show_video_only:
		populate_type(TrackReference::k_video);
		break;
	case k_show_subtitles_only:
		populate_type(TrackReference::k_subtitle);
		break;
	}

	connect(custom_menu_, &Menu::triggered, this,
			&ExportFormatComboBox::handle_index_change);
}

void ExportFormatComboBox::showPopup()
{
	custom_menu_->setMinimumWidth(this->width());
	custom_menu_->exec(mapToGlobal(QPoint(0, 0)));
}

void ExportFormatComboBox::set_format(int fmt)
{
	current_ = fmt;
	clear();
	char buf[256];
	oakengine_encoding_format_name(fmt, buf, sizeof(buf));
	addItem(QString::fromUtf8(buf));
}

void ExportFormatComboBox::handle_index_change(QAction *a)
{
	int f = a->data().toInt();
	set_format(f);
	emit format_changed(f);
}

void ExportFormatComboBox::populate_type(TrackReference::Type type)
{
	const int fmt_count = oakengine_encoding_format_count();
	for (int i = 0; i < fmt_count; i++) {
		int f = i;
		char buf[256];

		bool has_video = oakengine_encoding_format_video_codec_count(f) > 0;
		bool has_audio = oakengine_encoding_format_audio_codec_count(f) > 0;
		bool has_sub = oakengine_encoding_format_subtitle_codec_count(f) > 0;

		if (type == TrackReference::k_video && has_video) {
			// Do nothing
		} else if (type == TrackReference::k_audio && !has_video && has_audio) {
			// Do nothing
		} else if (type == TrackReference::k_subtitle && !has_video && !has_audio && has_sub) {
			// Do nothing
		} else {
			continue;
		}

		oakengine_encoding_format_name(f, buf, sizeof(buf));
		QString format_name = QString::fromUtf8(buf);

		QAction *a = custom_menu_->addAction(format_name);
		a->setData(i);
		a->setIconVisibleInMenu(false);
	}
}

QWidgetAction *ExportFormatComboBox::create_header(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;
}

}