qt-wrappers.cpp 9.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401
  1. /******************************************************************************
  2. Copyright (C) 2013 by Hugh Bailey <[email protected]>
  3. This program is free software: you can redistribute it and/or modify
  4. it under the terms of the GNU General Public License as published by
  5. the Free Software Foundation, either version 2 of the License, or
  6. (at your option) any later version.
  7. This program is distributed in the hope that it will be useful,
  8. but WITHOUT ANY WARRANTY; without even the implied warranty of
  9. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  10. GNU General Public License for more details.
  11. You should have received a copy of the GNU General Public License
  12. along with this program. If not, see <http://www.gnu.org/licenses/>.
  13. ******************************************************************************/
  14. #include "qt-wrappers.hpp"
  15. #include "obs-app.hpp"
  16. #include <graphics/graphics.h>
  17. #include <util/threading.h>
  18. #include <QWidget>
  19. #include <QLayout>
  20. #include <QComboBox>
  21. #include <QMessageBox>
  22. #include <QDataStream>
  23. #include <QKeyEvent>
  24. #include <QFileDialog>
  25. #include <QStandardItemModel>
  26. #include <QLabel>
  27. #if !defined(_WIN32) && !defined(__APPLE__)
  28. #include <obs-nix-platform.h>
  29. #endif
  30. #ifdef ENABLE_WAYLAND
  31. #include <qpa/qplatformnativeinterface.h>
  32. #endif
  33. static inline void OBSErrorBoxva(QWidget *parent, const char *msg, va_list args)
  34. {
  35. char full_message[4096];
  36. vsnprintf(full_message, 4095, msg, args);
  37. QMessageBox::critical(parent, "Error", full_message);
  38. }
  39. void OBSErrorBox(QWidget *parent, const char *msg, ...)
  40. {
  41. va_list args;
  42. va_start(args, msg);
  43. OBSErrorBoxva(parent, msg, args);
  44. va_end(args);
  45. }
  46. QMessageBox::StandardButton
  47. OBSMessageBox::question(QWidget *parent, const QString &title,
  48. const QString &text,
  49. QMessageBox::StandardButtons buttons,
  50. QMessageBox::StandardButton defaultButton)
  51. {
  52. QMessageBox mb(QMessageBox::Question, title, text, buttons, parent);
  53. mb.setDefaultButton(defaultButton);
  54. if (buttons & QMessageBox::Ok)
  55. mb.setButtonText(QMessageBox::Ok, QTStr("OK"));
  56. #define translate_button(x) \
  57. if (buttons & QMessageBox::x) \
  58. mb.setButtonText(QMessageBox::x, QTStr(#x));
  59. translate_button(Open);
  60. translate_button(Save);
  61. translate_button(Cancel);
  62. translate_button(Close);
  63. translate_button(Discard);
  64. translate_button(Apply);
  65. translate_button(Reset);
  66. translate_button(Yes);
  67. translate_button(No);
  68. translate_button(Abort);
  69. translate_button(Retry);
  70. translate_button(Ignore);
  71. #undef translate_button
  72. return (QMessageBox::StandardButton)mb.exec();
  73. }
  74. void OBSMessageBox::information(QWidget *parent, const QString &title,
  75. const QString &text)
  76. {
  77. QMessageBox mb(QMessageBox::Information, title, text, QMessageBox::Ok,
  78. parent);
  79. mb.setButtonText(QMessageBox::Ok, QTStr("OK"));
  80. mb.exec();
  81. }
  82. void OBSMessageBox::warning(QWidget *parent, const QString &title,
  83. const QString &text, bool enableRichText)
  84. {
  85. QMessageBox mb(QMessageBox::Warning, title, text, QMessageBox::Ok,
  86. parent);
  87. if (enableRichText)
  88. mb.setTextFormat(Qt::RichText);
  89. mb.setButtonText(QMessageBox::Ok, QTStr("OK"));
  90. mb.exec();
  91. }
  92. void OBSMessageBox::critical(QWidget *parent, const QString &title,
  93. const QString &text)
  94. {
  95. QMessageBox mb(QMessageBox::Critical, title, text, QMessageBox::Ok,
  96. parent);
  97. mb.setButtonText(QMessageBox::Ok, QTStr("OK"));
  98. mb.exec();
  99. }
  100. bool QTToGSWindow(QWindow *window, gs_window &gswindow)
  101. {
  102. bool success = true;
  103. #ifdef _WIN32
  104. gswindow.hwnd = (HWND)window->winId();
  105. #elif __APPLE__
  106. gswindow.view = (id)window->winId();
  107. #else
  108. switch (obs_get_nix_platform()) {
  109. case OBS_NIX_PLATFORM_X11_EGL:
  110. gswindow.id = window->winId();
  111. gswindow.display = obs_get_nix_platform_display();
  112. break;
  113. #ifdef ENABLE_WAYLAND
  114. case OBS_NIX_PLATFORM_WAYLAND:
  115. QPlatformNativeInterface *native =
  116. QGuiApplication::platformNativeInterface();
  117. gswindow.display =
  118. native->nativeResourceForWindow("surface", window);
  119. success = gswindow.display != nullptr;
  120. break;
  121. #endif
  122. }
  123. #endif
  124. return success;
  125. }
  126. uint32_t TranslateQtKeyboardEventModifiers(Qt::KeyboardModifiers mods)
  127. {
  128. int obsModifiers = INTERACT_NONE;
  129. if (mods.testFlag(Qt::ShiftModifier))
  130. obsModifiers |= INTERACT_SHIFT_KEY;
  131. if (mods.testFlag(Qt::AltModifier))
  132. obsModifiers |= INTERACT_ALT_KEY;
  133. #ifdef __APPLE__
  134. // Mac: Meta = Control, Control = Command
  135. if (mods.testFlag(Qt::ControlModifier))
  136. obsModifiers |= INTERACT_COMMAND_KEY;
  137. if (mods.testFlag(Qt::MetaModifier))
  138. obsModifiers |= INTERACT_CONTROL_KEY;
  139. #else
  140. // Handle windows key? Can a browser even trap that key?
  141. if (mods.testFlag(Qt::ControlModifier))
  142. obsModifiers |= INTERACT_CONTROL_KEY;
  143. if (mods.testFlag(Qt::MetaModifier))
  144. obsModifiers |= INTERACT_COMMAND_KEY;
  145. #endif
  146. return obsModifiers;
  147. }
  148. QDataStream &operator<<(QDataStream &out,
  149. const std::vector<std::shared_ptr<OBSSignal>> &)
  150. {
  151. return out;
  152. }
  153. QDataStream &operator>>(QDataStream &in,
  154. std::vector<std::shared_ptr<OBSSignal>> &)
  155. {
  156. return in;
  157. }
  158. QDataStream &operator<<(QDataStream &out, const OBSScene &scene)
  159. {
  160. return out << QString(obs_source_get_name(obs_scene_get_source(scene)));
  161. }
  162. QDataStream &operator>>(QDataStream &in, OBSScene &scene)
  163. {
  164. QString sceneName;
  165. in >> sceneName;
  166. OBSSourceAutoRelease source =
  167. obs_get_source_by_name(QT_TO_UTF8(sceneName));
  168. scene = obs_scene_from_source(source);
  169. return in;
  170. }
  171. void DeleteLayout(QLayout *layout)
  172. {
  173. if (!layout)
  174. return;
  175. for (;;) {
  176. QLayoutItem *item = layout->takeAt(0);
  177. if (!item)
  178. break;
  179. QLayout *subLayout = item->layout();
  180. if (subLayout) {
  181. DeleteLayout(subLayout);
  182. } else {
  183. delete item->widget();
  184. delete item;
  185. }
  186. }
  187. delete layout;
  188. }
  189. class QuickThread : public QThread {
  190. public:
  191. explicit inline QuickThread(std::function<void()> func_) : func(func_)
  192. {
  193. }
  194. private:
  195. virtual void run() override { func(); }
  196. std::function<void()> func;
  197. };
  198. QThread *CreateQThread(std::function<void()> func)
  199. {
  200. return new QuickThread(func);
  201. }
  202. volatile long insideEventLoop = 0;
  203. void ExecuteFuncSafeBlock(std::function<void()> func)
  204. {
  205. QEventLoop eventLoop;
  206. auto wait = [&]() {
  207. func();
  208. QMetaObject::invokeMethod(&eventLoop, "quit",
  209. Qt::QueuedConnection);
  210. };
  211. os_atomic_inc_long(&insideEventLoop);
  212. QScopedPointer<QThread> thread(CreateQThread(wait));
  213. thread->start();
  214. eventLoop.exec();
  215. thread->wait();
  216. os_atomic_dec_long(&insideEventLoop);
  217. }
  218. void ExecuteFuncSafeBlockMsgBox(std::function<void()> func,
  219. const QString &title, const QString &text)
  220. {
  221. QMessageBox dlg;
  222. dlg.setWindowFlags(dlg.windowFlags() & ~Qt::WindowCloseButtonHint);
  223. dlg.setWindowTitle(title);
  224. dlg.setText(text);
  225. dlg.setStandardButtons(QMessageBox::StandardButtons());
  226. auto wait = [&]() {
  227. func();
  228. QMetaObject::invokeMethod(&dlg, "accept", Qt::QueuedConnection);
  229. };
  230. os_atomic_inc_long(&insideEventLoop);
  231. QScopedPointer<QThread> thread(CreateQThread(wait));
  232. thread->start();
  233. dlg.exec();
  234. thread->wait();
  235. os_atomic_dec_long(&insideEventLoop);
  236. }
  237. static bool enable_message_boxes = false;
  238. void EnableThreadedMessageBoxes(bool enable)
  239. {
  240. enable_message_boxes = enable;
  241. }
  242. void ExecThreadedWithoutBlocking(std::function<void()> func,
  243. const QString &title, const QString &text)
  244. {
  245. if (!enable_message_boxes)
  246. ExecuteFuncSafeBlock(func);
  247. else
  248. ExecuteFuncSafeBlockMsgBox(func, title, text);
  249. }
  250. bool LineEditCanceled(QEvent *event)
  251. {
  252. if (event->type() == QEvent::KeyPress) {
  253. QKeyEvent *keyEvent = reinterpret_cast<QKeyEvent *>(event);
  254. return keyEvent->key() == Qt::Key_Escape;
  255. }
  256. return false;
  257. }
  258. bool LineEditChanged(QEvent *event)
  259. {
  260. if (event->type() == QEvent::KeyPress) {
  261. QKeyEvent *keyEvent = reinterpret_cast<QKeyEvent *>(event);
  262. switch (keyEvent->key()) {
  263. case Qt::Key_Tab:
  264. case Qt::Key_Backtab:
  265. case Qt::Key_Enter:
  266. case Qt::Key_Return:
  267. return true;
  268. }
  269. } else if (event->type() == QEvent::FocusOut) {
  270. return true;
  271. }
  272. return false;
  273. }
  274. void SetComboItemEnabled(QComboBox *c, int idx, bool enabled)
  275. {
  276. QStandardItemModel *model =
  277. dynamic_cast<QStandardItemModel *>(c->model());
  278. QStandardItem *item = model->item(idx);
  279. item->setFlags(enabled ? Qt::ItemIsSelectable | Qt::ItemIsEnabled
  280. : Qt::NoItemFlags);
  281. }
  282. void setThemeID(QWidget *widget, const QString &themeID)
  283. {
  284. if (widget->property("themeID").toString() != themeID) {
  285. widget->setProperty("themeID", themeID);
  286. /* force style sheet recalculation */
  287. QString qss = widget->styleSheet();
  288. widget->setStyleSheet("/* */");
  289. widget->setStyleSheet(qss);
  290. }
  291. }
  292. QString SelectDirectory(QWidget *parent, QString title, QString path)
  293. {
  294. QString dir = QFileDialog::getExistingDirectory(
  295. parent, title, path,
  296. QFileDialog::ShowDirsOnly | QFileDialog::DontResolveSymlinks);
  297. return dir;
  298. }
  299. QString SaveFile(QWidget *parent, QString title, QString path,
  300. QString extensions)
  301. {
  302. QString file =
  303. QFileDialog::getSaveFileName(parent, title, path, extensions);
  304. return file;
  305. }
  306. QString OpenFile(QWidget *parent, QString title, QString path,
  307. QString extensions)
  308. {
  309. QString file =
  310. QFileDialog::getOpenFileName(parent, title, path, extensions);
  311. return file;
  312. }
  313. QStringList OpenFiles(QWidget *parent, QString title, QString path,
  314. QString extensions)
  315. {
  316. QStringList files =
  317. QFileDialog::getOpenFileNames(parent, title, path, extensions);
  318. return files;
  319. }
  320. static void SetLabelText(QLabel *label, const QString &newText)
  321. {
  322. if (label->text() != newText)
  323. label->setText(newText);
  324. }
  325. void TruncateLabel(QLabel *label, QString newText, int length)
  326. {
  327. if (newText.length() < length) {
  328. label->setToolTip(QString());
  329. SetLabelText(label, newText);
  330. return;
  331. }
  332. label->setToolTip(newText);
  333. newText.truncate(length);
  334. newText += "...";
  335. SetLabelText(label, newText);
  336. }