git.sesse.net Git - nageru/blob - glwidget.cpp

   1 #include "glwidget.h"
   2
   3 #include <assert.h>
   4 #include <bmusb/bmusb.h>
   5 #include <movit/effect_chain.h>
   6 #include <movit/resource_pool.h>
   7 #include <stdbool.h>
   8 #include <stdint.h>
   9 #include <QAction>
  10 #include <QActionGroup>
  11 #include <QList>
  12 #include <QMenu>
  13 #include <QPoint>
  14 #include <QVariant>
  15 #include <QWidget>
  16 #include <functional>
  17 #include <map>
  18 #include <mutex>
  19 #include <utility>
  20
  21 #include "audio_mixer.h"
  22 #include "context.h"
  23 #include "context_menus.h"
  24 #include "flags.h"
  25 #include "mainwindow.h"
  26 #include "mixer.h"
  27 #include "ref_counted_gl_sync.h"
  28
  29 class QMouseEvent;
  30
  31 #undef Success
  32 #include <movit/util.h>
  33 #include <string>
  34
  35
  36 using namespace movit;
  37 using namespace std;
  38 using namespace std::placeholders;
  39
  40 GLWidget::GLWidget(QWidget *parent)
  41     : QGLWidget(parent, global_share_widget)
  42 {
  43 }
  44
  45 GLWidget::~GLWidget()
  46 {
  47 }
  48
  49 void GLWidget::shutdown()
  50 {
  51         if (resource_pool != nullptr) {
  52                 makeCurrent();
  53                 resource_pool->clean_context();
  54         }
  55         global_mixer->remove_frame_ready_callback(output, this);
  56 }
  57
  58 void GLWidget::initializeGL()
  59 {
  60         static once_flag flag;
  61         call_once(flag, [this]{
  62                 global_mixer = new Mixer(QGLFormat::toSurfaceFormat(format()), global_flags.num_cards);
  63                 global_audio_mixer = global_mixer->get_audio_mixer();
  64                 global_mainwindow->mixer_created(global_mixer);
  65                 global_mixer->start();
  66         });
  67         global_mixer->add_frame_ready_callback(output, this, [this]{
  68                 QMetaObject::invokeMethod(this, "update", Qt::AutoConnection);
  69         });
  70         if (output == Mixer::OUTPUT_LIVE) {
  71                 global_mixer->set_transition_names_updated_callback(output, [this](const vector<string> &names){
  72                         emit transition_names_updated(names);
  73                 });
  74         }
  75         if (output >= Mixer::OUTPUT_INPUT0) {
  76                 global_mixer->set_name_updated_callback(output, [this](const string &name){
  77                         emit name_updated(output, name);
  78                 });
  79                 global_mixer->set_color_updated_callback(output, [this](const string &color){
  80                         emit color_updated(output, color);
  81                 });
  82         }
  83         setContextMenuPolicy(Qt::CustomContextMenu);
  84         connect(this, &QWidget::customContextMenuRequested, bind(&GLWidget::show_context_menu, this, _1));
  85
  86         glDisable(GL_BLEND);
  87         glDisable(GL_DEPTH_TEST);
  88         glDepthMask(GL_FALSE);
  89 }
  90
  91 void GLWidget::resizeGL(int width, int height)
  92 {
  93         current_width = width;
  94         current_height = height;
  95         glViewport(0, 0, width, height);
  96 }
  97
  98 void GLWidget::paintGL()
  99 {
 100         Mixer::DisplayFrame frame;
 101         if (!global_mixer->get_display_frame(output, &frame)) {
 102                 glClearColor(0.0f, 1.0f, 0.0f, 1.0f);
 103                 check_error();
 104                 glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
 105                 check_error();
 106                 return;
 107         }
 108
 109         check_error();
 110         glWaitSync(frame.ready_fence.get(), /*flags=*/0, GL_TIMEOUT_IGNORED);
 111         check_error();
 112         frame.setup_chain();
 113         check_error();
 114         glDisable(GL_FRAMEBUFFER_SRGB);
 115         check_error();
 116         frame.chain->render_to_fbo(0, current_width, current_height);
 117         check_error();
 118
 119         if (resource_pool == nullptr) {
 120                 resource_pool = frame.chain->get_resource_pool();
 121         } else {
 122                 assert(resource_pool == frame.chain->get_resource_pool());
 123         }
 124 }
 125
 126 void GLWidget::mousePressEvent(QMouseEvent *event)
 127 {
 128         emit clicked();
 129 }
 130
 131 void GLWidget::show_context_menu(const QPoint &pos)
 132 {
 133         if (output == Mixer::OUTPUT_LIVE) {
 134                 show_live_context_menu(pos);
 135         }
 136         if (output >= Mixer::OUTPUT_INPUT0) {
 137                 int signal_num = global_mixer->get_channel_signal(output);
 138                 if (signal_num != -1) {
 139                         show_preview_context_menu(signal_num, pos);
 140                 }
 141         }
 142 }
 143
 144 void GLWidget::show_live_context_menu(const QPoint &pos)
 145 {
 146         QPoint global_pos = mapToGlobal(pos);
 147
 148         QMenu menu;
 149
 150         // Add a submenu for selecting output card, with an action for each card.
 151         QMenu card_submenu;
 152         fill_hdmi_sdi_output_device_menu(&card_submenu);
 153         card_submenu.setTitle("HDMI/SDI output device");
 154         menu.addMenu(&card_submenu);
 155
 156         // Add a submenu for choosing the output resolution. Since this is
 157         // card-dependent, it is disabled if we haven't chosen a card
 158         // (but it's still there so that the user will know it exists).
 159         QMenu resolution_submenu;
 160         fill_hdmi_sdi_output_resolution_menu(&resolution_submenu);
 161         resolution_submenu.setTitle("HDMI/SDI output resolution");
 162         menu.addMenu(&resolution_submenu);
 163
 164         // Show the menu; if there's an action selected, it will deal with it itself.
 165         menu.exec(global_pos);
 166 }
 167
 168 void GLWidget::show_preview_context_menu(unsigned signal_num, const QPoint &pos)
 169 {
 170         QPoint global_pos = mapToGlobal(pos);
 171
 172         QMenu menu;
 173
 174         // Add a submenu for selecting input card, with an action for each card.
 175         QMenu card_submenu;
 176         QActionGroup card_group(&card_submenu);
 177
 178         unsigned num_cards = global_mixer->get_num_cards();
 179         unsigned current_card = global_mixer->map_signal(signal_num);
 180         for (unsigned card_index = 0; card_index < num_cards; ++card_index) {
 181                 QString description(QString::fromStdString(global_mixer->get_card_description(card_index)));
 182                 QAction *action = new QAction(description, &card_group);
 183                 action->setCheckable(true);
 184                 if (current_card == card_index) {
 185                         action->setChecked(true);
 186                 }
 187                 action->setData(QList<QVariant>{"card", card_index});
 188                 card_submenu.addAction(action);
 189         }
 190
 191         card_submenu.setTitle("Input source");
 192         menu.addMenu(&card_submenu);
 193
 194         // Note that this setting depends on which card is active.
 195         // TODO: Consider hiding this for BGRA sources.
 196
 197         QMenu interpretation_submenu;
 198         QActionGroup interpretation_group(&interpretation_submenu);
 199
 200         YCbCrInterpretation current_interpretation = global_mixer->get_input_ycbcr_interpretation(current_card);
 201         {
 202                 QAction *action = new QAction("Auto", &interpretation_group);
 203                 action->setCheckable(true);
 204                 if (current_interpretation.ycbcr_coefficients_auto) {
 205                         action->setChecked(true);
 206                 }
 207                 action->setData(QList<QVariant>{"interpretation", true, YCBCR_REC_709, false});
 208                 interpretation_submenu.addAction(action);
 209         }
 210         for (YCbCrLumaCoefficients ycbcr_coefficients : { YCBCR_REC_709, YCBCR_REC_601 }) {
 211                 for (bool full_range : { false, true }) {
 212                         std::string description;
 213                         if (ycbcr_coefficients == YCBCR_REC_709) {
 214                                 description = "Rec. 709 (HD)";
 215                         } else {
 216                                 description = "Rec. 601 (SD)";
 217                         }
 218                         if (full_range) {
 219                                 description += ", full range (nonstandard)";
 220                         }
 221                         QAction *action = new QAction(QString::fromStdString(description), &interpretation_group);
 222                         action->setCheckable(true);
 223                         if (!current_interpretation.ycbcr_coefficients_auto &&
 224                             ycbcr_coefficients == current_interpretation.ycbcr_coefficients &&
 225                             full_range == current_interpretation.full_range) {
 226                                 action->setChecked(true);
 227                         }
 228                         action->setData(QList<QVariant>{"interpretation", false, ycbcr_coefficients, full_range});
 229                         interpretation_submenu.addAction(action);
 230                 }
 231         }
 232
 233         interpretation_submenu.setTitle("Input interpretation");
 234         menu.addMenu(&interpretation_submenu);
 235
 236         // --- The choices in the next few options depend a lot on which card is active ---
 237
 238         // Add a submenu for selecting video input, with an action for each input.
 239         QMenu video_input_submenu;
 240         QActionGroup video_input_group(&video_input_submenu);
 241         std::map<uint32_t, string> video_inputs = global_mixer->get_available_video_inputs(current_card);
 242         uint32_t current_video_input = global_mixer->get_current_video_input(current_card);
 243         for (const auto &mode : video_inputs) {
 244                 QString description(QString::fromStdString(mode.second));
 245                 QAction *action = new QAction(description, &video_input_group);
 246                 action->setCheckable(true);
 247                 if (mode.first == current_video_input) {
 248                         action->setChecked(true);
 249                 }
 250                 action->setData(QList<QVariant>{"video_input", mode.first});
 251                 video_input_submenu.addAction(action);
 252         }
 253
 254         video_input_submenu.setTitle("Video input");
 255         menu.addMenu(&video_input_submenu);
 256
 257         // The same for audio input.
 258         QMenu audio_input_submenu;
 259         QActionGroup audio_input_group(&audio_input_submenu);
 260         std::map<uint32_t, string> audio_inputs = global_mixer->get_available_audio_inputs(current_card);
 261         uint32_t current_audio_input = global_mixer->get_current_audio_input(current_card);
 262         for (const auto &mode : audio_inputs) {
 263                 QString description(QString::fromStdString(mode.second));
 264                 QAction *action = new QAction(description, &audio_input_group);
 265                 action->setCheckable(true);
 266                 if (mode.first == current_audio_input) {
 267                         action->setChecked(true);
 268                 }
 269                 action->setData(QList<QVariant>{"audio_input", mode.first});
 270                 audio_input_submenu.addAction(action);
 271         }
 272
 273         audio_input_submenu.setTitle("Audio input");
 274         menu.addMenu(&audio_input_submenu);
 275
 276         // The same for resolution.
 277         QMenu mode_submenu;
 278         QActionGroup mode_group(&mode_submenu);
 279         std::map<uint32_t, bmusb::VideoMode> video_modes = global_mixer->get_available_video_modes(current_card);
 280         uint32_t current_video_mode = global_mixer->get_current_video_mode(current_card);
 281         bool has_auto_mode = false;
 282         for (const auto &mode : video_modes) {
 283                 QString description(QString::fromStdString(mode.second.name));
 284                 QAction *action = new QAction(description, &mode_group);
 285                 action->setCheckable(true);
 286                 if (mode.first == current_video_mode) {
 287                         action->setChecked(true);
 288                 }
 289                 action->setData(QList<QVariant>{"video_mode", mode.first});
 290                 mode_submenu.addAction(action);
 291
 292                 // TODO: Relying on the 0 value here (from bmusb.h) is ugly, it should be a named constant.
 293                 if (mode.first == 0) {
 294                         has_auto_mode = true;
 295                 }
 296         }
 297
 298         // Add a “scan” menu if there's no “auto” mode.
 299         if (!has_auto_mode) {
 300                 QAction *action = new QAction("Scan", &mode_group);
 301                 action->setData(QList<QVariant>{"video_mode", 0});
 302                 mode_submenu.addSeparator();
 303                 mode_submenu.addAction(action);
 304         }
 305
 306         mode_submenu.setTitle("Input mode");
 307         menu.addMenu(&mode_submenu);
 308
 309         // --- End of card-dependent choices ---
 310
 311         // Add an audio source selector.
 312         QAction *audio_source_action = nullptr;
 313         if (global_audio_mixer->get_mapping_mode() == AudioMixer::MappingMode::SIMPLE) {
 314                 audio_source_action = new QAction("Use as audio source", &menu);
 315                 audio_source_action->setCheckable(true);
 316                 if (global_audio_mixer->get_simple_input() == signal_num) {
 317                         audio_source_action->setChecked(true);
 318                         audio_source_action->setEnabled(false);
 319                 }
 320                 menu.addAction(audio_source_action);
 321         }
 322
 323         // And a master clock selector.
 324         QAction *master_clock_action = new QAction("Use as master clock", &menu);
 325         master_clock_action->setCheckable(true);
 326         if (global_mixer->get_output_card_index() != -1) {
 327                 master_clock_action->setChecked(false);
 328                 master_clock_action->setEnabled(false);
 329         } else if (global_mixer->get_master_clock() == signal_num) {
 330                 master_clock_action->setChecked(true);
 331                 master_clock_action->setEnabled(false);
 332         }
 333         menu.addAction(master_clock_action);
 334
 335         // Show the menu and look at the result.
 336         QAction *selected_item = menu.exec(global_pos);
 337         if (audio_source_action != nullptr && selected_item == audio_source_action) {
 338                 global_audio_mixer->set_simple_input(signal_num);
 339         } else if (selected_item == master_clock_action) {
 340                 global_mixer->set_master_clock(signal_num);
 341         } else if (selected_item != nullptr) {
 342                 QList<QVariant> selected = selected_item->data().toList();
 343                 if (selected[0].toString() == "video_mode") {
 344                         uint32_t mode = selected[1].toUInt(nullptr);
 345                         if (mode == 0 && !has_auto_mode) {
 346                                 global_mixer->start_mode_scanning(current_card);
 347                         } else {
 348                                 global_mixer->set_video_mode(current_card, mode);
 349                         }
 350                 } else if (selected[0].toString() == "video_input") {
 351                         uint32_t input = selected[1].toUInt(nullptr);
 352                         global_mixer->set_video_input(current_card, input);
 353                 } else if (selected[0].toString() == "audio_input") {
 354                         uint32_t input = selected[1].toUInt(nullptr);
 355                         global_mixer->set_audio_input(current_card, input);
 356                 } else if (selected[0].toString() == "card") {
 357                         unsigned card_index = selected[1].toUInt(nullptr);
 358                         global_mixer->set_signal_mapping(signal_num, card_index);
 359                 } else if (selected[0].toString() == "interpretation") {
 360                         YCbCrInterpretation interpretation;
 361                         interpretation.ycbcr_coefficients_auto = selected[1].toBool();
 362                         interpretation.ycbcr_coefficients = YCbCrLumaCoefficients(selected[2].toUInt(nullptr));
 363                         interpretation.full_range = selected[3].toBool();
 364                         global_mixer->set_input_ycbcr_interpretation(current_card, interpretation);
 365                 } else {
 366                         assert(false);
 367                 }
 368         }
 369 }