window-basic-main.cpp 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602
  1. /******************************************************************************
  2. Copyright (C) 2013-2014 by Hugh Bailey <[email protected]>
  3. Copyright (C) 2014 by Zachary Lund <[email protected]>
  4. This program is free software: you can redistribute it and/or modify
  5. it under the terms of the GNU General Public License as published by
  6. the Free Software Foundation, either version 2 of the License, or
  7. (at your option) any later version.
  8. This program is distributed in the hope that it will be useful,
  9. but WITHOUT ANY WARRANTY; without even the implied warranty of
  10. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  11. GNU General Public License for more details.
  12. You should have received a copy of the GNU General Public License
  13. along with this program. If not, see <http://www.gnu.org/licenses/>.
  14. ******************************************************************************/
  15. #include <obs.hpp>
  16. #include <QMessageBox>
  17. #include <QShowEvent>
  18. #include <QFileDialog>
  19. #include "obs-app.hpp"
  20. #include "window-basic-settings.hpp"
  21. #include "window-namedialog.hpp"
  22. #include "window-basic-main.hpp"
  23. #include "qt-wrappers.hpp"
  24. #include "ui_OBSBasic.h"
  25. using namespace std;
  26. Q_DECLARE_METATYPE(OBSScene);
  27. Q_DECLARE_METATYPE(OBSSceneItem);
  28. OBSBasic::OBSBasic(QWidget *parent)
  29. : OBSMainWindow (parent),
  30. outputTest (NULL),
  31. ui (new Ui::OBSBasic)
  32. {
  33. ui->setupUi(this);
  34. }
  35. void OBSBasic::OBSInit()
  36. {
  37. /* make sure it's fully displayed before doing any initialization */
  38. show();
  39. App()->processEvents();
  40. if (!obs_startup())
  41. throw "Failed to initialize libobs";
  42. if (!InitGraphics())
  43. throw "Failed to initialize graphics";
  44. if (!InitAudio())
  45. throw "Failed to initialize audio";
  46. obs_add_draw_callback(OBSBasic::RenderMain, this);
  47. signal_handler_connect(obs_signalhandler(), "source-add",
  48. OBSBasic::SourceAdded, this);
  49. signal_handler_connect(obs_signalhandler(), "source-remove",
  50. OBSBasic::SourceRemoved, this);
  51. signal_handler_connect(obs_signalhandler(), "channel-change",
  52. OBSBasic::ChannelChanged, this);
  53. /* TODO: this is a test */
  54. obs_load_module("test-input");
  55. obs_load_module("obs-ffmpeg");
  56. /* HACK: fixes a qt bug with native widgets with native repaint */
  57. ui->previewContainer->repaint();
  58. }
  59. OBSBasic::~OBSBasic()
  60. {
  61. /* free the lists before shutting down to remove the scene/item
  62. * references */
  63. ui->sources->clear();
  64. ui->scenes->clear();
  65. obs_shutdown();
  66. }
  67. OBSScene OBSBasic::GetCurrentScene()
  68. {
  69. QListWidgetItem *item = ui->scenes->currentItem();
  70. return item ? item->data(Qt::UserRole).value<OBSScene>() : nullptr;
  71. }
  72. OBSSceneItem OBSBasic::GetCurrentSceneItem()
  73. {
  74. QListWidgetItem *item = ui->sources->currentItem();
  75. return item ? item->data(Qt::UserRole).value<OBSSceneItem>() : nullptr;
  76. }
  77. void OBSBasic::UpdateSources(OBSScene scene)
  78. {
  79. ui->sources->clear();
  80. obs_scene_enum_items(scene,
  81. [] (obs_scene_t scene, obs_sceneitem_t item, void *p)
  82. {
  83. OBSBasic *window = static_cast<OBSBasic*>(p);
  84. window->AddSceneItem(item);
  85. UNUSED_PARAMETER(scene);
  86. return true;
  87. }, this);
  88. }
  89. /* Qt callbacks for invokeMethod */
  90. void OBSBasic::AddScene(OBSSource source)
  91. {
  92. const char *name = obs_source_getname(source);
  93. obs_scene_t scene = obs_scene_fromsource(source);
  94. QListWidgetItem *item = new QListWidgetItem(QT_UTF8(name));
  95. item->setData(Qt::UserRole, QVariant::fromValue(OBSScene(scene)));
  96. ui->scenes->addItem(item);
  97. signal_handler_t handler = obs_source_signalhandler(source);
  98. signal_handler_connect(handler, "add", OBSBasic::SceneItemAdded, this);
  99. signal_handler_connect(handler, "remove", OBSBasic::SceneItemRemoved,
  100. this);
  101. }
  102. void OBSBasic::RemoveScene(OBSSource source)
  103. {
  104. const char *name = obs_source_getname(source);
  105. QListWidgetItem *sel = ui->scenes->currentItem();
  106. QList<QListWidgetItem*> items = ui->scenes->findItems(QT_UTF8(name),
  107. Qt::MatchExactly);
  108. if (sel != nullptr) {
  109. if (items.contains(sel))
  110. ui->sources->clear();
  111. delete sel;
  112. }
  113. }
  114. void OBSBasic::AddSceneItem(OBSSceneItem item)
  115. {
  116. obs_scene_t scene = obs_sceneitem_getscene(item);
  117. obs_source_t source = obs_sceneitem_getsource(item);
  118. const char *name = obs_source_getname(source);
  119. if (GetCurrentScene() == scene) {
  120. QListWidgetItem *listItem = new QListWidgetItem(QT_UTF8(name));
  121. listItem->setData(Qt::UserRole,
  122. QVariant::fromValue(OBSSceneItem(item)));
  123. ui->sources->insertItem(0, listItem);
  124. }
  125. sourceSceneRefs[source] = sourceSceneRefs[source] + 1;
  126. }
  127. void OBSBasic::RemoveSceneItem(OBSSceneItem item)
  128. {
  129. obs_scene_t scene = obs_sceneitem_getscene(item);
  130. if (GetCurrentScene() == scene) {
  131. for (int i = 0; i < ui->sources->count(); i++) {
  132. QListWidgetItem *listItem = ui->sources->item(i);
  133. QVariant userData = listItem->data(Qt::UserRole);
  134. if (userData.value<OBSSceneItem>() == item) {
  135. delete listItem;
  136. break;
  137. }
  138. }
  139. }
  140. obs_source_t source = obs_sceneitem_getsource(item);
  141. int scenes = sourceSceneRefs[source] - 1;
  142. if (scenes == 0) {
  143. obs_source_remove(source);
  144. sourceSceneRefs.erase(source);
  145. }
  146. }
  147. void OBSBasic::UpdateSceneSelection(OBSSource source)
  148. {
  149. if (source) {
  150. obs_source_type type;
  151. obs_source_gettype(source, &type, NULL);
  152. if (type != OBS_SOURCE_TYPE_SCENE)
  153. return;
  154. obs_scene_t scene = obs_scene_fromsource(source);
  155. const char *name = obs_source_getname(source);
  156. QListWidgetItem *sel = ui->scenes->currentItem();
  157. QList<QListWidgetItem*> items =
  158. ui->scenes->findItems(QT_UTF8(name), Qt::MatchExactly);
  159. if (items.contains(sel)) {
  160. ui->scenes->setCurrentItem(sel);
  161. UpdateSources(scene);
  162. }
  163. }
  164. }
  165. /* OBS Callbacks */
  166. void OBSBasic::SceneItemAdded(void *data, calldata_t params)
  167. {
  168. OBSBasic *window = static_cast<OBSBasic*>(data);
  169. obs_sceneitem_t item = (obs_sceneitem_t)calldata_ptr(params, "item");
  170. QMetaObject::invokeMethod(window, "AddSceneItem",
  171. Q_ARG(OBSSceneItem, OBSSceneItem(item)));
  172. }
  173. void OBSBasic::SceneItemRemoved(void *data, calldata_t params)
  174. {
  175. OBSBasic *window = static_cast<OBSBasic*>(data);
  176. obs_sceneitem_t item = (obs_sceneitem_t)calldata_ptr(params, "item");
  177. QMetaObject::invokeMethod(window, "RemoveSceneItem",
  178. Q_ARG(OBSSceneItem, OBSSceneItem(item)));
  179. }
  180. void OBSBasic::SourceAdded(void *data, calldata_t params)
  181. {
  182. obs_source_t source = (obs_source_t)calldata_ptr(params, "source");
  183. obs_source_type type;
  184. obs_source_gettype(source, &type, NULL);
  185. if (type == OBS_SOURCE_TYPE_SCENE)
  186. QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
  187. "AddScene",
  188. Q_ARG(OBSSource, OBSSource(source)));
  189. }
  190. void OBSBasic::SourceRemoved(void *data, calldata_t params)
  191. {
  192. obs_source_t source = (obs_source_t)calldata_ptr(params, "source");
  193. obs_source_type type;
  194. obs_source_gettype(source, &type, NULL);
  195. if (type == OBS_SOURCE_TYPE_SCENE)
  196. QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
  197. "RemoveScene",
  198. Q_ARG(OBSSource, OBSSource(source)));
  199. }
  200. void OBSBasic::ChannelChanged(void *data, calldata_t params)
  201. {
  202. obs_source_t source = (obs_source_t)calldata_ptr(params, "source");
  203. uint32_t channel = calldata_uint32(params, "channel");
  204. if (channel == 0)
  205. QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
  206. "UpdateSceneSelection",
  207. Q_ARG(OBSSource, OBSSource(source)));
  208. }
  209. void OBSBasic::RenderMain(void *data, uint32_t cx, uint32_t cy)
  210. {
  211. obs_render_main_view();
  212. UNUSED_PARAMETER(data);
  213. UNUSED_PARAMETER(cx);
  214. UNUSED_PARAMETER(cy);
  215. }
  216. /* Main class functions */
  217. bool OBSBasic::InitGraphics()
  218. {
  219. struct obs_video_info ovi;
  220. App()->GetConfigFPS(ovi.fps_num, ovi.fps_den);
  221. ovi.graphics_module = App()->GetRenderModule();
  222. ovi.base_width = (uint32_t)config_get_uint(GetGlobalConfig(),
  223. "Video", "BaseCX");
  224. ovi.base_height = (uint32_t)config_get_uint(GetGlobalConfig(),
  225. "Video", "BaseCY");
  226. ovi.output_width = (uint32_t)config_get_uint(GetGlobalConfig(),
  227. "Video", "OutputCX");
  228. ovi.output_height = (uint32_t)config_get_uint(GetGlobalConfig(),
  229. "Video", "OutputCY");
  230. ovi.output_format = VIDEO_FORMAT_I420;
  231. ovi.adapter = 0;
  232. ovi.gpu_conversion = true;
  233. QTToGSWindow(ui->preview, ovi.window);
  234. //required to make opengl display stuff on osx(?)
  235. ResizePreview(ovi.base_width, ovi.base_height);
  236. QSize size = ui->preview->size();
  237. ovi.window_width = size.width();
  238. ovi.window_height = size.height();
  239. return obs_reset_video(&ovi);
  240. }
  241. bool OBSBasic::InitAudio()
  242. {
  243. /* TODO: load audio settings from config */
  244. struct audio_output_info ai;
  245. ai.name = "test";
  246. ai.samples_per_sec = 44100;
  247. ai.format = AUDIO_FORMAT_FLOAT_PLANAR;
  248. ai.speakers = SPEAKERS_STEREO;
  249. ai.buffer_ms = 700;
  250. return obs_reset_audio(&ai);
  251. }
  252. void OBSBasic::ResizePreview(uint32_t cx, uint32_t cy)
  253. {
  254. double targetAspect, baseAspect;
  255. QSize targetSize;
  256. int x, y;
  257. /* resize preview panel to fix to the top section of the window */
  258. targetSize = ui->previewContainer->size();
  259. targetAspect = double(targetSize.width()) / double(targetSize.height());
  260. baseAspect = double(cx) / double(cy);
  261. if (targetAspect > baseAspect) {
  262. cx = targetSize.height() * baseAspect;
  263. cy = targetSize.height();
  264. } else {
  265. cx = targetSize.width();
  266. cy = targetSize.width() / baseAspect;
  267. }
  268. x = targetSize.width() /2 - cx/2;
  269. y = targetSize.height()/2 - cy/2;
  270. ui->preview->setGeometry(x, y, cx, cy);
  271. if (isVisible())
  272. obs_resize(cx, cy);
  273. }
  274. void OBSBasic::closeEvent(QCloseEvent *event)
  275. {
  276. /* TODO */
  277. UNUSED_PARAMETER(event);
  278. }
  279. void OBSBasic::changeEvent(QEvent *event)
  280. {
  281. /* TODO */
  282. UNUSED_PARAMETER(event);
  283. }
  284. void OBSBasic::resizeEvent(QResizeEvent *event)
  285. {
  286. struct obs_video_info ovi;
  287. if (obs_get_video_info(&ovi))
  288. ResizePreview(ovi.base_width, ovi.base_height);
  289. UNUSED_PARAMETER(event);
  290. }
  291. void OBSBasic::on_action_New_triggered()
  292. {
  293. /* TODO */
  294. }
  295. void OBSBasic::on_action_Open_triggered()
  296. {
  297. /* TODO */
  298. }
  299. void OBSBasic::on_action_Save_triggered()
  300. {
  301. /* TODO */
  302. }
  303. void OBSBasic::on_scenes_itemChanged(QListWidgetItem *item)
  304. {
  305. obs_source_t source = NULL;
  306. if (item) {
  307. obs_scene_t scene;
  308. scene = item->data(Qt::UserRole).value<OBSScene>();
  309. source = obs_scene_getsource(scene);
  310. UpdateSources(scene);
  311. }
  312. /* TODO: allow transitions */
  313. obs_set_output_source(0, source);
  314. }
  315. void OBSBasic::on_scenes_customContextMenuRequested(const QPoint &pos)
  316. {
  317. /* TODO */
  318. UNUSED_PARAMETER(pos);
  319. }
  320. void OBSBasic::on_actionAddScene_triggered()
  321. {
  322. string name;
  323. bool accepted = NameDialog::AskForName(this,
  324. QTStr("MainWindow.AddSceneDlg.Title"),
  325. QTStr("MainWindow.AddSceneDlg.Text"),
  326. name);
  327. if (accepted) {
  328. obs_source_t source = obs_get_source_by_name(name.c_str());
  329. if (source) {
  330. QMessageBox::information(this,
  331. QTStr("MainWindow.NameExists.Title"),
  332. QTStr("MainWindow.NameExists.Text"));
  333. obs_source_release(source);
  334. on_actionAddScene_triggered();
  335. return;
  336. }
  337. obs_scene_t scene = obs_scene_create(name.c_str());
  338. source = obs_scene_getsource(scene);
  339. obs_add_source(source);
  340. obs_scene_release(scene);
  341. obs_set_output_source(0, source);
  342. }
  343. }
  344. void OBSBasic::on_actionRemoveScene_triggered()
  345. {
  346. QListWidgetItem *item = ui->scenes->currentItem();
  347. if (!item)
  348. return;
  349. QVariant userData = item->data(Qt::UserRole);
  350. obs_scene_t scene = userData.value<OBSScene>();
  351. obs_source_t source = obs_scene_getsource(scene);
  352. obs_source_remove(source);
  353. }
  354. void OBSBasic::on_actionSceneProperties_triggered()
  355. {
  356. /* TODO */
  357. }
  358. void OBSBasic::on_actionSceneUp_triggered()
  359. {
  360. /* TODO */
  361. }
  362. void OBSBasic::on_actionSceneDown_triggered()
  363. {
  364. /* TODO */
  365. }
  366. void OBSBasic::on_sources_itemChanged(QListWidgetItem *item)
  367. {
  368. /* TODO */
  369. UNUSED_PARAMETER(item);
  370. }
  371. void OBSBasic::on_sources_customContextMenuRequested(const QPoint &pos)
  372. {
  373. /* TODO */
  374. UNUSED_PARAMETER(pos);
  375. }
  376. void OBSBasic::AddSource(obs_scene_t scene, const char *id)
  377. {
  378. string name;
  379. bool success = false;
  380. while (!success) {
  381. bool accepted = NameDialog::AskForName(this,
  382. Str("MainWindow.AddSourceDlg.Title"),
  383. Str("MainWindow.AddSourceDlg.Text"),
  384. name);
  385. if (!accepted)
  386. break;
  387. obs_source_t source = obs_get_source_by_name(name.c_str());
  388. if (!source) {
  389. success = true;
  390. } else {
  391. QMessageBox::information(this,
  392. QTStr("MainWindow.NameExists.Title"),
  393. QTStr("MainWindow.NameExists.Text"));
  394. obs_source_release(source);
  395. }
  396. }
  397. if (success) {
  398. obs_source_t source = obs_source_create(OBS_SOURCE_TYPE_INPUT,
  399. id, name.c_str(), NULL);
  400. sourceSceneRefs[source] = 0;
  401. obs_add_source(source);
  402. obs_scene_add(scene, source);
  403. obs_source_release(source);
  404. }
  405. }
  406. void OBSBasic::AddSourcePopupMenu(const QPoint &pos)
  407. {
  408. OBSScene scene = GetCurrentScene();
  409. const char *type;
  410. bool foundValues = false;
  411. size_t idx = 0;
  412. if (!scene)
  413. return;
  414. QMenu popup;
  415. while (obs_enum_input_types(idx++, &type)) {
  416. const char *name = obs_source_getdisplayname(
  417. OBS_SOURCE_TYPE_INPUT,
  418. type, App()->GetLocale());
  419. QAction *popupItem = new QAction(QT_UTF8(name), this);
  420. popupItem->setData(QT_UTF8(type));
  421. popup.addAction(popupItem);
  422. foundValues = true;
  423. }
  424. if (foundValues) {
  425. QAction *ret = popup.exec(pos);
  426. if (ret)
  427. AddSource(scene, ret->data().toString().toUtf8());
  428. }
  429. }
  430. void OBSBasic::on_actionAddSource_triggered()
  431. {
  432. AddSourcePopupMenu(QCursor::pos());
  433. }
  434. void OBSBasic::on_actionRemoveSource_triggered()
  435. {
  436. OBSSceneItem item = GetCurrentSceneItem();
  437. if (item)
  438. obs_sceneitem_remove(item);
  439. }
  440. void OBSBasic::on_actionSourceProperties_triggered()
  441. {
  442. }
  443. void OBSBasic::on_actionSourceUp_triggered()
  444. {
  445. }
  446. void OBSBasic::on_actionSourceDown_triggered()
  447. {
  448. }
  449. void OBSBasic::on_recordButton_clicked()
  450. {
  451. if (outputTest) {
  452. obs_output_destroy(outputTest);
  453. outputTest = NULL;
  454. ui->recordButton->setText("Start Recording");
  455. } else {
  456. QString path = QFileDialog::getSaveFileName(this,
  457. "Please enter a file name", QString(),
  458. "Video Files (*.avi)");
  459. if (path.isNull() || path.isEmpty())
  460. return;
  461. obs_data_t data = obs_data_create();
  462. obs_data_setstring(data, "filename", QT_TO_UTF8(path));
  463. outputTest = obs_output_create("ffmpeg_output", "test", data);
  464. obs_data_release(data);
  465. if (!obs_output_start(outputTest)) {
  466. obs_output_destroy(outputTest);
  467. outputTest = NULL;
  468. return;
  469. }
  470. ui->recordButton->setText("Stop Recording");
  471. }
  472. }
  473. void OBSBasic::on_settingsButton_clicked()
  474. {
  475. OBSBasicSettings settings(this);
  476. settings.exec();
  477. }