window-basic-main.cpp 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923
  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 <util/util.hpp>
  20. #include <util/platform.h>
  21. #include "obs-app.hpp"
  22. #include "platform.hpp"
  23. #include "window-basic-settings.hpp"
  24. #include "window-namedialog.hpp"
  25. #include "window-basic-main.hpp"
  26. #include "window-basic-properties.hpp"
  27. #include "qt-wrappers.hpp"
  28. #include "display-helpers.hpp"
  29. #include "ui_OBSBasic.h"
  30. using namespace std;
  31. Q_DECLARE_METATYPE(OBSScene);
  32. Q_DECLARE_METATYPE(OBSSceneItem);
  33. OBSBasic::OBSBasic(QWidget *parent)
  34. : OBSMainWindow (parent),
  35. outputTest (NULL),
  36. sceneChanging (false),
  37. resizeTimer (0),
  38. properties (nullptr),
  39. ui (new Ui::OBSBasic)
  40. {
  41. ui->setupUi(this);
  42. }
  43. static inline bool HasAudioDevices(const char *source_id)
  44. {
  45. const char *output_id = source_id;
  46. obs_properties_t props = obs_get_source_properties(
  47. OBS_SOURCE_TYPE_INPUT, output_id, App()->GetLocale());
  48. size_t count = 0;
  49. if (!props)
  50. return false;
  51. obs_property_t devices = obs_properties_get(props, "device_id");
  52. if (devices)
  53. count = obs_property_list_item_count(devices);
  54. obs_properties_destroy(props);
  55. return count != 0;
  56. }
  57. bool OBSBasic::InitBasicConfigDefaults()
  58. {
  59. bool hasDesktopAudio = HasAudioDevices(App()->OutputAudioSource());
  60. bool hasInputAudio = HasAudioDevices(App()->InputAudioSource());
  61. config_set_default_int(basicConfig, "Window", "PosX", -1);
  62. config_set_default_int(basicConfig, "Window", "PosY", -1);
  63. config_set_default_int(basicConfig, "Window", "SizeX", -1);
  64. config_set_default_int(basicConfig, "Window", "SizeY", -1);
  65. vector<MonitorInfo> monitors;
  66. GetMonitors(monitors);
  67. if (!monitors.size()) {
  68. OBSErrorBox(NULL, "There appears to be no monitors. Er, this "
  69. "technically shouldn't be possible.");
  70. return false;
  71. }
  72. uint32_t cx = monitors[0].cx;
  73. uint32_t cy = monitors[0].cy;
  74. /* TODO: temporary */
  75. config_set_default_string(basicConfig, "OutputTemp", "URL", "");
  76. config_set_default_string(basicConfig, "OutputTemp", "Key", "");
  77. config_set_default_uint (basicConfig, "OutputTemp", "VBitrate", 2500);
  78. config_set_default_uint (basicConfig, "OutputTemp", "ABitrate", 128);
  79. config_set_default_uint (basicConfig, "Video", "BaseCX", cx);
  80. config_set_default_uint (basicConfig, "Video", "BaseCY", cy);
  81. cx = cx * 10 / 15;
  82. cy = cy * 10 / 15;
  83. config_set_default_uint (basicConfig, "Video", "OutputCX", cx);
  84. config_set_default_uint (basicConfig, "Video", "OutputCY", cy);
  85. config_set_default_uint (basicConfig, "Video", "FPSType", 0);
  86. config_set_default_string(basicConfig, "Video", "FPSCommon", "30");
  87. config_set_default_uint (basicConfig, "Video", "FPSInt", 30);
  88. config_set_default_uint (basicConfig, "Video", "FPSNum", 30);
  89. config_set_default_uint (basicConfig, "Video", "FPSDen", 1);
  90. config_set_default_uint (basicConfig, "Audio", "SampleRate", 44100);
  91. config_set_default_string(basicConfig, "Audio", "ChannelSetup",
  92. "Stereo");
  93. config_set_default_uint (basicConfig, "Audio", "BufferingTime", 1000);
  94. config_set_default_string(basicConfig, "Audio", "DesktopDevice1",
  95. hasDesktopAudio ? "default" : "disabled");
  96. config_set_default_string(basicConfig, "Audio", "DesktopDevice2",
  97. "disabled");
  98. config_set_default_string(basicConfig, "Audio", "AuxDevice1",
  99. hasInputAudio ? "default" : "disabled");
  100. config_set_default_string(basicConfig, "Audio", "AuxDevice2",
  101. "disabled");
  102. config_set_default_string(basicConfig, "Audio", "AuxDevice3",
  103. "disabled");
  104. return true;
  105. }
  106. bool OBSBasic::InitBasicConfig()
  107. {
  108. BPtr<char> configPath(os_get_config_path("obs-studio/basic/basic.ini"));
  109. int errorcode = basicConfig.Open(configPath, CONFIG_OPEN_ALWAYS);
  110. if (errorcode != CONFIG_SUCCESS) {
  111. OBSErrorBox(NULL, "Failed to open basic.ini: %d", errorcode);
  112. return false;
  113. }
  114. return InitBasicConfigDefaults();
  115. }
  116. void OBSBasic::OBSInit()
  117. {
  118. /* make sure it's fully displayed before doing any initialization */
  119. show();
  120. App()->processEvents();
  121. if (!obs_startup())
  122. throw "Failed to initialize libobs";
  123. if (!InitBasicConfig())
  124. throw "Failed to load basic.ini";
  125. if (!ResetVideo())
  126. throw "Failed to initialize video";
  127. if (!ResetAudio())
  128. throw "Failed to initialize audio";
  129. signal_handler_connect(obs_signalhandler(), "source_add",
  130. OBSBasic::SourceAdded, this);
  131. signal_handler_connect(obs_signalhandler(), "source_remove",
  132. OBSBasic::SourceRemoved, this);
  133. signal_handler_connect(obs_signalhandler(), "channel_change",
  134. OBSBasic::ChannelChanged, this);
  135. /* TODO: this is a test, all modules will be searched for and loaded
  136. * automatically later */
  137. obs_load_module("test-input");
  138. obs_load_module("obs-ffmpeg");
  139. #ifdef __APPLE__
  140. obs_load_module("mac-capture");
  141. #elif _WIN32
  142. obs_load_module("win-wasapi");
  143. obs_load_module("win-capture");
  144. #else
  145. obs_load_module("linux-xshm");
  146. obs_load_module("linux-pulseaudio");
  147. #endif
  148. ResetAudioDevices();
  149. }
  150. OBSBasic::~OBSBasic()
  151. {
  152. if (properties)
  153. delete properties;
  154. /* free the lists before shutting down to remove the scene/item
  155. * references */
  156. ui->sources->clear();
  157. ui->scenes->clear();
  158. obs_shutdown();
  159. }
  160. OBSScene OBSBasic::GetCurrentScene()
  161. {
  162. QListWidgetItem *item = ui->scenes->currentItem();
  163. return item ? item->data(Qt::UserRole).value<OBSScene>() : nullptr;
  164. }
  165. OBSSceneItem OBSBasic::GetCurrentSceneItem()
  166. {
  167. QListWidgetItem *item = ui->sources->currentItem();
  168. return item ? item->data(Qt::UserRole).value<OBSSceneItem>() : nullptr;
  169. }
  170. void OBSBasic::UpdateSources(OBSScene scene)
  171. {
  172. ui->sources->clear();
  173. obs_scene_enum_items(scene,
  174. [] (obs_scene_t scene, obs_sceneitem_t item, void *p)
  175. {
  176. OBSBasic *window = static_cast<OBSBasic*>(p);
  177. window->InsertSceneItem(item);
  178. UNUSED_PARAMETER(scene);
  179. return true;
  180. }, this);
  181. }
  182. void OBSBasic::InsertSceneItem(obs_sceneitem_t item)
  183. {
  184. obs_source_t source = obs_sceneitem_getsource(item);
  185. const char *name = obs_source_getname(source);
  186. QListWidgetItem *listItem = new QListWidgetItem(QT_UTF8(name));
  187. listItem->setData(Qt::UserRole,
  188. QVariant::fromValue(OBSSceneItem(item)));
  189. ui->sources->insertItem(0, listItem);
  190. }
  191. /* Qt callbacks for invokeMethod */
  192. void OBSBasic::AddScene(OBSSource source)
  193. {
  194. const char *name = obs_source_getname(source);
  195. obs_scene_t scene = obs_scene_fromsource(source);
  196. QListWidgetItem *item = new QListWidgetItem(QT_UTF8(name));
  197. item->setData(Qt::UserRole, QVariant::fromValue(OBSScene(scene)));
  198. ui->scenes->addItem(item);
  199. signal_handler_t handler = obs_source_signalhandler(source);
  200. signal_handler_connect(handler, "item_add",
  201. OBSBasic::SceneItemAdded, this);
  202. signal_handler_connect(handler, "item_remove",
  203. OBSBasic::SceneItemRemoved, this);
  204. }
  205. void OBSBasic::RemoveScene(OBSSource source)
  206. {
  207. const char *name = obs_source_getname(source);
  208. QListWidgetItem *sel = ui->scenes->currentItem();
  209. QList<QListWidgetItem*> items = ui->scenes->findItems(QT_UTF8(name),
  210. Qt::MatchExactly);
  211. if (sel != nullptr) {
  212. if (items.contains(sel))
  213. ui->sources->clear();
  214. delete sel;
  215. }
  216. }
  217. void OBSBasic::AddSceneItem(OBSSceneItem item)
  218. {
  219. obs_scene_t scene = obs_sceneitem_getscene(item);
  220. obs_source_t source = obs_sceneitem_getsource(item);
  221. if (GetCurrentScene() == scene)
  222. InsertSceneItem(item);
  223. sourceSceneRefs[source] = sourceSceneRefs[source] + 1;
  224. }
  225. void OBSBasic::RemoveSceneItem(OBSSceneItem item)
  226. {
  227. obs_scene_t scene = obs_sceneitem_getscene(item);
  228. if (GetCurrentScene() == scene) {
  229. for (int i = 0; i < ui->sources->count(); i++) {
  230. QListWidgetItem *listItem = ui->sources->item(i);
  231. QVariant userData = listItem->data(Qt::UserRole);
  232. if (userData.value<OBSSceneItem>() == item) {
  233. delete listItem;
  234. break;
  235. }
  236. }
  237. }
  238. obs_source_t source = obs_sceneitem_getsource(item);
  239. int scenes = sourceSceneRefs[source] - 1;
  240. if (scenes == 0) {
  241. obs_source_remove(source);
  242. sourceSceneRefs.erase(source);
  243. }
  244. }
  245. void OBSBasic::UpdateSceneSelection(OBSSource source)
  246. {
  247. if (source) {
  248. obs_source_type type;
  249. obs_source_gettype(source, &type, NULL);
  250. if (type != OBS_SOURCE_TYPE_SCENE)
  251. return;
  252. obs_scene_t scene = obs_scene_fromsource(source);
  253. const char *name = obs_source_getname(source);
  254. QList<QListWidgetItem*> items =
  255. ui->scenes->findItems(QT_UTF8(name), Qt::MatchExactly);
  256. if (items.count()) {
  257. sceneChanging = true;
  258. ui->scenes->setCurrentItem(items.first());
  259. sceneChanging = false;
  260. UpdateSources(scene);
  261. }
  262. }
  263. }
  264. /* OBS Callbacks */
  265. void OBSBasic::SceneItemAdded(void *data, calldata_t params)
  266. {
  267. OBSBasic *window = static_cast<OBSBasic*>(data);
  268. obs_sceneitem_t item = (obs_sceneitem_t)calldata_ptr(params, "item");
  269. QMetaObject::invokeMethod(window, "AddSceneItem",
  270. Q_ARG(OBSSceneItem, OBSSceneItem(item)));
  271. }
  272. void OBSBasic::SceneItemRemoved(void *data, calldata_t params)
  273. {
  274. OBSBasic *window = static_cast<OBSBasic*>(data);
  275. obs_sceneitem_t item = (obs_sceneitem_t)calldata_ptr(params, "item");
  276. QMetaObject::invokeMethod(window, "RemoveSceneItem",
  277. Q_ARG(OBSSceneItem, OBSSceneItem(item)));
  278. }
  279. void OBSBasic::SourceAdded(void *data, calldata_t params)
  280. {
  281. obs_source_t source = (obs_source_t)calldata_ptr(params, "source");
  282. obs_source_type type;
  283. obs_source_gettype(source, &type, NULL);
  284. if (type == OBS_SOURCE_TYPE_SCENE)
  285. QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
  286. "AddScene",
  287. Q_ARG(OBSSource, OBSSource(source)));
  288. }
  289. void OBSBasic::SourceRemoved(void *data, calldata_t params)
  290. {
  291. obs_source_t source = (obs_source_t)calldata_ptr(params, "source");
  292. obs_source_type type;
  293. obs_source_gettype(source, &type, NULL);
  294. if (type == OBS_SOURCE_TYPE_SCENE)
  295. QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
  296. "RemoveScene",
  297. Q_ARG(OBSSource, OBSSource(source)));
  298. }
  299. void OBSBasic::ChannelChanged(void *data, calldata_t params)
  300. {
  301. obs_source_t source = (obs_source_t)calldata_ptr(params, "source");
  302. uint32_t channel = (uint32_t)calldata_int(params, "channel");
  303. if (channel == 0)
  304. QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
  305. "UpdateSceneSelection",
  306. Q_ARG(OBSSource, OBSSource(source)));
  307. }
  308. void OBSBasic::RenderMain(void *data, uint32_t cx, uint32_t cy)
  309. {
  310. OBSBasic *window = static_cast<OBSBasic*>(data);
  311. gs_matrix_push();
  312. gs_matrix_scale3f(window->previewScale, window->previewScale, 1.0f);
  313. gs_matrix_translate3f(-window->previewX, -window->previewY, 0.0f);
  314. obs_render_main_view();
  315. gs_matrix_pop();
  316. UNUSED_PARAMETER(cx);
  317. UNUSED_PARAMETER(cy);
  318. }
  319. /* Main class functions */
  320. bool OBSBasic::ResetVideo()
  321. {
  322. struct obs_video_info ovi;
  323. GetConfigFPS(ovi.fps_num, ovi.fps_den);
  324. ovi.graphics_module = App()->GetRenderModule();
  325. ovi.base_width = (uint32_t)config_get_uint(basicConfig,
  326. "Video", "BaseCX");
  327. ovi.base_height = (uint32_t)config_get_uint(basicConfig,
  328. "Video", "BaseCY");
  329. ovi.output_width = (uint32_t)config_get_uint(basicConfig,
  330. "Video", "OutputCX");
  331. ovi.output_height = (uint32_t)config_get_uint(basicConfig,
  332. "Video", "OutputCY");
  333. ovi.output_format = VIDEO_FORMAT_I420;
  334. ovi.adapter = 0;
  335. ovi.gpu_conversion = true;
  336. QTToGSWindow(ui->preview->winId(), ovi.window);
  337. //required to make opengl display stuff on osx(?)
  338. ResizePreview(ovi.base_width, ovi.base_height);
  339. QSize size = ui->preview->size();
  340. ovi.window_width = size.width();
  341. ovi.window_height = size.height();
  342. if (!obs_reset_video(&ovi))
  343. return false;
  344. obs_add_draw_callback(OBSBasic::RenderMain, this);
  345. return true;
  346. }
  347. bool OBSBasic::ResetAudio()
  348. {
  349. struct audio_output_info ai;
  350. ai.name = "Main Audio Track";
  351. ai.format = AUDIO_FORMAT_FLOAT;
  352. ai.samples_per_sec = config_get_uint(basicConfig, "Audio",
  353. "SampleRate");
  354. const char *channelSetupStr = config_get_string(basicConfig,
  355. "Audio", "ChannelSetup");
  356. if (strcmp(channelSetupStr, "Mono") == 0)
  357. ai.speakers = SPEAKERS_MONO;
  358. else
  359. ai.speakers = SPEAKERS_STEREO;
  360. ai.buffer_ms = config_get_uint(basicConfig, "Audio", "BufferingTime");
  361. return obs_reset_audio(&ai);
  362. }
  363. void OBSBasic::ResetAudioDevice(const char *sourceId, const char *deviceName,
  364. int channel)
  365. {
  366. const char *deviceId = config_get_string(basicConfig, "Audio",
  367. deviceName);
  368. obs_source_t source;
  369. obs_data_t settings;
  370. bool same = false;
  371. source = obs_get_output_source(channel);
  372. if (source) {
  373. settings = obs_source_getsettings(source);
  374. const char *curId = obs_data_getstring(settings, "device_id");
  375. same = (strcmp(curId, deviceId) == 0);
  376. obs_data_release(settings);
  377. obs_source_release(source);
  378. }
  379. if (!same)
  380. obs_set_output_source(channel, nullptr);
  381. if (!same && strcmp(deviceId, "disabled") != 0) {
  382. obs_data_t settings = obs_data_create();
  383. obs_data_setstring(settings, "device_id", deviceId);
  384. source = obs_source_create(OBS_SOURCE_TYPE_INPUT,
  385. sourceId, deviceName, settings);
  386. obs_data_release(settings);
  387. obs_set_output_source(channel, source);
  388. obs_source_release(source);
  389. }
  390. }
  391. void OBSBasic::ResetAudioDevices()
  392. {
  393. ResetAudioDevice(App()->OutputAudioSource(), "DesktopDevice1", 1);
  394. ResetAudioDevice(App()->OutputAudioSource(), "DesktopDevice2", 2);
  395. ResetAudioDevice(App()->InputAudioSource(), "AuxDevice1", 3);
  396. ResetAudioDevice(App()->InputAudioSource(), "AuxDevice2", 4);
  397. ResetAudioDevice(App()->InputAudioSource(), "AuxDevice3", 5);
  398. }
  399. void OBSBasic::ResizePreview(uint32_t cx, uint32_t cy)
  400. {
  401. QSize targetSize;
  402. /* resize preview panel to fix to the top section of the window */
  403. targetSize = ui->preview->size();
  404. GetScaleAndCenterPos(int(cx), int(cy),
  405. targetSize.width(), targetSize.height(),
  406. previewX, previewY, previewScale);
  407. if (isVisible()) {
  408. if (resizeTimer)
  409. killTimer(resizeTimer);
  410. resizeTimer = startTimer(100);
  411. }
  412. }
  413. void OBSBasic::closeEvent(QCloseEvent *event)
  414. {
  415. /* TODO */
  416. UNUSED_PARAMETER(event);
  417. }
  418. void OBSBasic::changeEvent(QEvent *event)
  419. {
  420. /* TODO */
  421. UNUSED_PARAMETER(event);
  422. }
  423. void OBSBasic::resizeEvent(QResizeEvent *event)
  424. {
  425. struct obs_video_info ovi;
  426. if (obs_get_video_info(&ovi))
  427. ResizePreview(ovi.base_width, ovi.base_height);
  428. UNUSED_PARAMETER(event);
  429. }
  430. void OBSBasic::timerEvent(QTimerEvent *event)
  431. {
  432. if (event->timerId() == resizeTimer) {
  433. killTimer(resizeTimer);
  434. resizeTimer = 0;
  435. QSize size = ui->preview->size();
  436. obs_resize(size.width(), size.height());
  437. }
  438. }
  439. void OBSBasic::on_action_New_triggered()
  440. {
  441. /* TODO */
  442. }
  443. void OBSBasic::on_action_Open_triggered()
  444. {
  445. /* TODO */
  446. }
  447. void OBSBasic::on_action_Save_triggered()
  448. {
  449. /* TODO */
  450. }
  451. void OBSBasic::on_scenes_currentItemChanged(QListWidgetItem *current,
  452. QListWidgetItem *prev)
  453. {
  454. obs_source_t source = NULL;
  455. if (sceneChanging)
  456. return;
  457. if (current) {
  458. obs_scene_t scene;
  459. scene = current->data(Qt::UserRole).value<OBSScene>();
  460. source = obs_scene_getsource(scene);
  461. }
  462. /* TODO: allow transitions */
  463. obs_set_output_source(0, source);
  464. UNUSED_PARAMETER(prev);
  465. }
  466. void OBSBasic::on_scenes_customContextMenuRequested(const QPoint &pos)
  467. {
  468. /* TODO */
  469. UNUSED_PARAMETER(pos);
  470. }
  471. void OBSBasic::on_actionAddScene_triggered()
  472. {
  473. string name;
  474. bool accepted = NameDialog::AskForName(this,
  475. QTStr("MainWindow.AddSceneDlg.Title"),
  476. QTStr("MainWindow.AddSceneDlg.Text"),
  477. name);
  478. if (accepted) {
  479. if (name.empty()) {
  480. QMessageBox::information(this,
  481. QTStr("MainWindow.NoNameEntered"),
  482. QTStr("MainWindow.NoNameEntered"));
  483. on_actionAddScene_triggered();
  484. return;
  485. }
  486. obs_source_t source = obs_get_source_by_name(name.c_str());
  487. if (source) {
  488. QMessageBox::information(this,
  489. QTStr("MainWindow.NameExists.Title"),
  490. QTStr("MainWindow.NameExists.Text"));
  491. obs_source_release(source);
  492. on_actionAddScene_triggered();
  493. return;
  494. }
  495. obs_scene_t scene = obs_scene_create(name.c_str());
  496. source = obs_scene_getsource(scene);
  497. obs_add_source(source);
  498. obs_scene_release(scene);
  499. obs_set_output_source(0, source);
  500. }
  501. }
  502. void OBSBasic::on_actionRemoveScene_triggered()
  503. {
  504. QListWidgetItem *item = ui->scenes->currentItem();
  505. if (!item)
  506. return;
  507. QVariant userData = item->data(Qt::UserRole);
  508. obs_scene_t scene = userData.value<OBSScene>();
  509. obs_source_t source = obs_scene_getsource(scene);
  510. obs_source_remove(source);
  511. }
  512. void OBSBasic::on_actionSceneProperties_triggered()
  513. {
  514. /* TODO */
  515. }
  516. void OBSBasic::on_actionSceneUp_triggered()
  517. {
  518. /* TODO */
  519. }
  520. void OBSBasic::on_actionSceneDown_triggered()
  521. {
  522. /* TODO */
  523. }
  524. void OBSBasic::on_sources_currentItemChanged(QListWidgetItem *current,
  525. QListWidgetItem *prev)
  526. {
  527. /* TODO */
  528. UNUSED_PARAMETER(current);
  529. UNUSED_PARAMETER(prev);
  530. }
  531. void OBSBasic::on_sources_customContextMenuRequested(const QPoint &pos)
  532. {
  533. /* TODO */
  534. UNUSED_PARAMETER(pos);
  535. }
  536. void OBSBasic::AddSource(obs_scene_t scene, const char *id)
  537. {
  538. string name;
  539. bool success = false;
  540. while (!success) {
  541. bool accepted = NameDialog::AskForName(this,
  542. Str("MainWindow.AddSourceDlg.Title"),
  543. Str("MainWindow.AddSourceDlg.Text"),
  544. name);
  545. if (!accepted)
  546. break;
  547. if (name.empty()) {
  548. QMessageBox::information(this,
  549. QTStr("MainWindow.NoNameEntered"),
  550. QTStr("MainWindow.NoNameEntered"));
  551. continue;
  552. }
  553. obs_source_t source = obs_get_source_by_name(name.c_str());
  554. if (!source) {
  555. success = true;
  556. break;
  557. } else {
  558. QMessageBox::information(this,
  559. QTStr("MainWindow.NameExists.Title"),
  560. QTStr("MainWindow.NameExists.Text"));
  561. obs_source_release(source);
  562. }
  563. }
  564. if (success) {
  565. obs_source_t source = obs_source_create(OBS_SOURCE_TYPE_INPUT,
  566. id, name.c_str(), NULL);
  567. sourceSceneRefs[source] = 0;
  568. obs_add_source(source);
  569. obs_scene_add(scene, source);
  570. obs_source_release(source);
  571. }
  572. }
  573. void OBSBasic::AddSourcePopupMenu(const QPoint &pos)
  574. {
  575. OBSScene scene = GetCurrentScene();
  576. const char *type;
  577. bool foundValues = false;
  578. size_t idx = 0;
  579. if (!scene)
  580. return;
  581. QMenu popup;
  582. while (obs_enum_input_types(idx++, &type)) {
  583. const char *name = obs_source_getdisplayname(
  584. OBS_SOURCE_TYPE_INPUT,
  585. type, App()->GetLocale());
  586. QAction *popupItem = new QAction(QT_UTF8(name), this);
  587. popupItem->setData(QT_UTF8(type));
  588. popup.addAction(popupItem);
  589. foundValues = true;
  590. }
  591. if (foundValues) {
  592. QAction *ret = popup.exec(pos);
  593. if (ret)
  594. AddSource(scene, ret->data().toString().toUtf8());
  595. }
  596. }
  597. void OBSBasic::on_actionAddSource_triggered()
  598. {
  599. AddSourcePopupMenu(QCursor::pos());
  600. }
  601. void OBSBasic::on_actionRemoveSource_triggered()
  602. {
  603. OBSSceneItem item = GetCurrentSceneItem();
  604. if (item)
  605. obs_sceneitem_remove(item);
  606. }
  607. void OBSBasic::on_actionSourceProperties_triggered()
  608. {
  609. OBSSceneItem item = GetCurrentSceneItem();
  610. OBSSource source = obs_sceneitem_getsource(item);
  611. if (source) {
  612. delete properties;
  613. properties = new OBSBasicProperties(this, source);
  614. properties->Init();
  615. }
  616. }
  617. void OBSBasic::on_actionSourceUp_triggered()
  618. {
  619. }
  620. void OBSBasic::on_actionSourceDown_triggered()
  621. {
  622. }
  623. void OBSBasic::OutputConnect(bool success)
  624. {
  625. if (!success) {
  626. obs_output_destroy(outputTest);
  627. outputTest = NULL;
  628. } else {
  629. ui->streamButton->setText("Stop Streaming");
  630. }
  631. ui->streamButton->setEnabled(true);
  632. }
  633. static void OBSOutputConnect(void *data, calldata_t params)
  634. {
  635. bool success = calldata_bool(params, "success");
  636. QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
  637. "OutputConnect", Q_ARG(bool, success));
  638. }
  639. /* TODO: lots of temporary code */
  640. void OBSBasic::on_streamButton_clicked()
  641. {
  642. if (outputTest) {
  643. obs_output_destroy(outputTest);
  644. outputTest = NULL;
  645. ui->streamButton->setText("Start Streaming");
  646. } else {
  647. const char *url = config_get_string(basicConfig, "OutputTemp",
  648. "URL");
  649. const char *key = config_get_string(basicConfig, "OutputTemp",
  650. "Key");
  651. int vBitrate = config_get_uint(basicConfig, "OutputTemp",
  652. "VBitrate");
  653. int aBitrate = config_get_uint(basicConfig, "OutputTemp",
  654. "ABitrate");
  655. if (!url)
  656. return;
  657. string fullURL = url;
  658. if (key && *key)
  659. fullURL = fullURL + "/" + key;
  660. obs_data_t data = obs_data_create();
  661. obs_data_setstring(data, "filename", fullURL.c_str());
  662. obs_data_setint(data, "audio_bitrate", aBitrate);
  663. obs_data_setint(data, "video_bitrate", vBitrate);
  664. outputTest = obs_output_create("ffmpeg_output", "test", data);
  665. obs_data_release(data);
  666. if (!outputTest)
  667. return;
  668. signal_handler_connect(obs_output_signalhandler(outputTest),
  669. "start", OBSOutputConnect, this);
  670. obs_output_start(outputTest);
  671. ui->streamButton->setEnabled(false);
  672. }
  673. }
  674. void OBSBasic::on_settingsButton_clicked()
  675. {
  676. OBSBasicSettings settings(this);
  677. settings.exec();
  678. }
  679. void OBSBasic::GetFPSCommon(uint32_t &num, uint32_t &den) const
  680. {
  681. const char *val = config_get_string(basicConfig, "Video", "FPSCommon");
  682. if (strcmp(val, "10") == 0) {
  683. num = 10;
  684. den = 1;
  685. } else if (strcmp(val, "20") == 0) {
  686. num = 20;
  687. den = 1;
  688. } else if (strcmp(val, "25") == 0) {
  689. num = 25;
  690. den = 1;
  691. } else if (strcmp(val, "29.97") == 0) {
  692. num = 30000;
  693. den = 1001;
  694. } else if (strcmp(val, "48") == 0) {
  695. num = 48;
  696. den = 1;
  697. } else if (strcmp(val, "59.94") == 0) {
  698. num = 60000;
  699. den = 1001;
  700. } else if (strcmp(val, "60") == 0) {
  701. num = 60;
  702. den = 1;
  703. } else {
  704. num = 30;
  705. den = 1;
  706. }
  707. }
  708. void OBSBasic::GetFPSInteger(uint32_t &num, uint32_t &den) const
  709. {
  710. num = (uint32_t)config_get_uint(basicConfig, "Video", "FPSInt");
  711. den = 1;
  712. }
  713. void OBSBasic::GetFPSFraction(uint32_t &num, uint32_t &den) const
  714. {
  715. num = (uint32_t)config_get_uint(basicConfig, "Video", "FPSNum");
  716. den = (uint32_t)config_get_uint(basicConfig, "Video", "FPSDen");
  717. }
  718. void OBSBasic::GetFPSNanoseconds(uint32_t &num, uint32_t &den) const
  719. {
  720. num = 1000000000;
  721. den = (uint32_t)config_get_uint(basicConfig, "Video", "FPSNS");
  722. }
  723. void OBSBasic::GetConfigFPS(uint32_t &num, uint32_t &den) const
  724. {
  725. uint32_t type = config_get_uint(basicConfig, "Video", "FPSType");
  726. if (type == 1) //"Integer"
  727. GetFPSInteger(num, den);
  728. else if (type == 2) //"Fraction"
  729. GetFPSFraction(num, den);
  730. else if (false) //"Nanoseconds", currently not implemented
  731. GetFPSNanoseconds(num, den);
  732. else
  733. GetFPSCommon(num, den);
  734. }
  735. config_t OBSBasic::Config() const
  736. {
  737. return basicConfig;
  738. }
  739. void OBSBasic::UnloadProperties()
  740. {
  741. properties = nullptr;
  742. }