window-basic-main.cpp 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924
  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.output_format = VIDEO_FORMAT_NV12;
  335. ovi.adapter = 0;
  336. ovi.gpu_conversion = true;
  337. QTToGSWindow(ui->preview->winId(), ovi.window);
  338. //required to make opengl display stuff on osx(?)
  339. ResizePreview(ovi.base_width, ovi.base_height);
  340. QSize size = ui->preview->size();
  341. ovi.window_width = size.width();
  342. ovi.window_height = size.height();
  343. if (!obs_reset_video(&ovi))
  344. return false;
  345. obs_add_draw_callback(OBSBasic::RenderMain, this);
  346. return true;
  347. }
  348. bool OBSBasic::ResetAudio()
  349. {
  350. struct audio_output_info ai;
  351. ai.name = "Main Audio Track";
  352. ai.format = AUDIO_FORMAT_FLOAT;
  353. ai.samples_per_sec = config_get_uint(basicConfig, "Audio",
  354. "SampleRate");
  355. const char *channelSetupStr = config_get_string(basicConfig,
  356. "Audio", "ChannelSetup");
  357. if (strcmp(channelSetupStr, "Mono") == 0)
  358. ai.speakers = SPEAKERS_MONO;
  359. else
  360. ai.speakers = SPEAKERS_STEREO;
  361. ai.buffer_ms = config_get_uint(basicConfig, "Audio", "BufferingTime");
  362. return obs_reset_audio(&ai);
  363. }
  364. void OBSBasic::ResetAudioDevice(const char *sourceId, const char *deviceName,
  365. int channel)
  366. {
  367. const char *deviceId = config_get_string(basicConfig, "Audio",
  368. deviceName);
  369. obs_source_t source;
  370. obs_data_t settings;
  371. bool same = false;
  372. source = obs_get_output_source(channel);
  373. if (source) {
  374. settings = obs_source_getsettings(source);
  375. const char *curId = obs_data_getstring(settings, "device_id");
  376. same = (strcmp(curId, deviceId) == 0);
  377. obs_data_release(settings);
  378. obs_source_release(source);
  379. }
  380. if (!same)
  381. obs_set_output_source(channel, nullptr);
  382. if (!same && strcmp(deviceId, "disabled") != 0) {
  383. obs_data_t settings = obs_data_create();
  384. obs_data_setstring(settings, "device_id", deviceId);
  385. source = obs_source_create(OBS_SOURCE_TYPE_INPUT,
  386. sourceId, deviceName, settings);
  387. obs_data_release(settings);
  388. obs_set_output_source(channel, source);
  389. obs_source_release(source);
  390. }
  391. }
  392. void OBSBasic::ResetAudioDevices()
  393. {
  394. ResetAudioDevice(App()->OutputAudioSource(), "DesktopDevice1", 1);
  395. ResetAudioDevice(App()->OutputAudioSource(), "DesktopDevice2", 2);
  396. ResetAudioDevice(App()->InputAudioSource(), "AuxDevice1", 3);
  397. ResetAudioDevice(App()->InputAudioSource(), "AuxDevice2", 4);
  398. ResetAudioDevice(App()->InputAudioSource(), "AuxDevice3", 5);
  399. }
  400. void OBSBasic::ResizePreview(uint32_t cx, uint32_t cy)
  401. {
  402. QSize targetSize;
  403. /* resize preview panel to fix to the top section of the window */
  404. targetSize = ui->preview->size();
  405. GetScaleAndCenterPos(int(cx), int(cy),
  406. targetSize.width(), targetSize.height(),
  407. previewX, previewY, previewScale);
  408. if (isVisible()) {
  409. if (resizeTimer)
  410. killTimer(resizeTimer);
  411. resizeTimer = startTimer(100);
  412. }
  413. }
  414. void OBSBasic::closeEvent(QCloseEvent *event)
  415. {
  416. /* TODO */
  417. UNUSED_PARAMETER(event);
  418. }
  419. void OBSBasic::changeEvent(QEvent *event)
  420. {
  421. /* TODO */
  422. UNUSED_PARAMETER(event);
  423. }
  424. void OBSBasic::resizeEvent(QResizeEvent *event)
  425. {
  426. struct obs_video_info ovi;
  427. if (obs_get_video_info(&ovi))
  428. ResizePreview(ovi.base_width, ovi.base_height);
  429. UNUSED_PARAMETER(event);
  430. }
  431. void OBSBasic::timerEvent(QTimerEvent *event)
  432. {
  433. if (event->timerId() == resizeTimer) {
  434. killTimer(resizeTimer);
  435. resizeTimer = 0;
  436. QSize size = ui->preview->size();
  437. obs_resize(size.width(), size.height());
  438. }
  439. }
  440. void OBSBasic::on_action_New_triggered()
  441. {
  442. /* TODO */
  443. }
  444. void OBSBasic::on_action_Open_triggered()
  445. {
  446. /* TODO */
  447. }
  448. void OBSBasic::on_action_Save_triggered()
  449. {
  450. /* TODO */
  451. }
  452. void OBSBasic::on_scenes_currentItemChanged(QListWidgetItem *current,
  453. QListWidgetItem *prev)
  454. {
  455. obs_source_t source = NULL;
  456. if (sceneChanging)
  457. return;
  458. if (current) {
  459. obs_scene_t scene;
  460. scene = current->data(Qt::UserRole).value<OBSScene>();
  461. source = obs_scene_getsource(scene);
  462. }
  463. /* TODO: allow transitions */
  464. obs_set_output_source(0, source);
  465. UNUSED_PARAMETER(prev);
  466. }
  467. void OBSBasic::on_scenes_customContextMenuRequested(const QPoint &pos)
  468. {
  469. /* TODO */
  470. UNUSED_PARAMETER(pos);
  471. }
  472. void OBSBasic::on_actionAddScene_triggered()
  473. {
  474. string name;
  475. bool accepted = NameDialog::AskForName(this,
  476. QTStr("MainWindow.AddSceneDlg.Title"),
  477. QTStr("MainWindow.AddSceneDlg.Text"),
  478. name);
  479. if (accepted) {
  480. if (name.empty()) {
  481. QMessageBox::information(this,
  482. QTStr("MainWindow.NoNameEntered"),
  483. QTStr("MainWindow.NoNameEntered"));
  484. on_actionAddScene_triggered();
  485. return;
  486. }
  487. obs_source_t source = obs_get_source_by_name(name.c_str());
  488. if (source) {
  489. QMessageBox::information(this,
  490. QTStr("MainWindow.NameExists.Title"),
  491. QTStr("MainWindow.NameExists.Text"));
  492. obs_source_release(source);
  493. on_actionAddScene_triggered();
  494. return;
  495. }
  496. obs_scene_t scene = obs_scene_create(name.c_str());
  497. source = obs_scene_getsource(scene);
  498. obs_add_source(source);
  499. obs_scene_release(scene);
  500. obs_set_output_source(0, source);
  501. }
  502. }
  503. void OBSBasic::on_actionRemoveScene_triggered()
  504. {
  505. QListWidgetItem *item = ui->scenes->currentItem();
  506. if (!item)
  507. return;
  508. QVariant userData = item->data(Qt::UserRole);
  509. obs_scene_t scene = userData.value<OBSScene>();
  510. obs_source_t source = obs_scene_getsource(scene);
  511. obs_source_remove(source);
  512. }
  513. void OBSBasic::on_actionSceneProperties_triggered()
  514. {
  515. /* TODO */
  516. }
  517. void OBSBasic::on_actionSceneUp_triggered()
  518. {
  519. /* TODO */
  520. }
  521. void OBSBasic::on_actionSceneDown_triggered()
  522. {
  523. /* TODO */
  524. }
  525. void OBSBasic::on_sources_currentItemChanged(QListWidgetItem *current,
  526. QListWidgetItem *prev)
  527. {
  528. /* TODO */
  529. UNUSED_PARAMETER(current);
  530. UNUSED_PARAMETER(prev);
  531. }
  532. void OBSBasic::on_sources_customContextMenuRequested(const QPoint &pos)
  533. {
  534. /* TODO */
  535. UNUSED_PARAMETER(pos);
  536. }
  537. void OBSBasic::AddSource(obs_scene_t scene, const char *id)
  538. {
  539. string name;
  540. bool success = false;
  541. while (!success) {
  542. bool accepted = NameDialog::AskForName(this,
  543. Str("MainWindow.AddSourceDlg.Title"),
  544. Str("MainWindow.AddSourceDlg.Text"),
  545. name);
  546. if (!accepted)
  547. break;
  548. if (name.empty()) {
  549. QMessageBox::information(this,
  550. QTStr("MainWindow.NoNameEntered"),
  551. QTStr("MainWindow.NoNameEntered"));
  552. continue;
  553. }
  554. obs_source_t source = obs_get_source_by_name(name.c_str());
  555. if (!source) {
  556. success = true;
  557. break;
  558. } else {
  559. QMessageBox::information(this,
  560. QTStr("MainWindow.NameExists.Title"),
  561. QTStr("MainWindow.NameExists.Text"));
  562. obs_source_release(source);
  563. }
  564. }
  565. if (success) {
  566. obs_source_t source = obs_source_create(OBS_SOURCE_TYPE_INPUT,
  567. id, name.c_str(), NULL);
  568. sourceSceneRefs[source] = 0;
  569. obs_add_source(source);
  570. obs_scene_add(scene, source);
  571. obs_source_release(source);
  572. }
  573. }
  574. void OBSBasic::AddSourcePopupMenu(const QPoint &pos)
  575. {
  576. OBSScene scene = GetCurrentScene();
  577. const char *type;
  578. bool foundValues = false;
  579. size_t idx = 0;
  580. if (!scene)
  581. return;
  582. QMenu popup;
  583. while (obs_enum_input_types(idx++, &type)) {
  584. const char *name = obs_source_getdisplayname(
  585. OBS_SOURCE_TYPE_INPUT,
  586. type, App()->GetLocale());
  587. QAction *popupItem = new QAction(QT_UTF8(name), this);
  588. popupItem->setData(QT_UTF8(type));
  589. popup.addAction(popupItem);
  590. foundValues = true;
  591. }
  592. if (foundValues) {
  593. QAction *ret = popup.exec(pos);
  594. if (ret)
  595. AddSource(scene, ret->data().toString().toUtf8());
  596. }
  597. }
  598. void OBSBasic::on_actionAddSource_triggered()
  599. {
  600. AddSourcePopupMenu(QCursor::pos());
  601. }
  602. void OBSBasic::on_actionRemoveSource_triggered()
  603. {
  604. OBSSceneItem item = GetCurrentSceneItem();
  605. if (item)
  606. obs_sceneitem_remove(item);
  607. }
  608. void OBSBasic::on_actionSourceProperties_triggered()
  609. {
  610. OBSSceneItem item = GetCurrentSceneItem();
  611. OBSSource source = obs_sceneitem_getsource(item);
  612. if (source) {
  613. delete properties;
  614. properties = new OBSBasicProperties(this, source);
  615. properties->Init();
  616. }
  617. }
  618. void OBSBasic::on_actionSourceUp_triggered()
  619. {
  620. }
  621. void OBSBasic::on_actionSourceDown_triggered()
  622. {
  623. }
  624. void OBSBasic::OutputStart(int errorcode)
  625. {
  626. if (errorcode != OBS_OUTPUT_SUCCESS) {
  627. obs_output_destroy(outputTest);
  628. outputTest = NULL;
  629. } else {
  630. ui->streamButton->setText("Stop Streaming");
  631. }
  632. ui->streamButton->setEnabled(true);
  633. }
  634. static void OBSOutputStart(void *data, calldata_t params)
  635. {
  636. int code = calldata_bool(params, "errorcode");
  637. QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
  638. "OutputStart", Q_ARG(int, code));
  639. }
  640. /* TODO: lots of temporary code */
  641. void OBSBasic::on_streamButton_clicked()
  642. {
  643. if (outputTest) {
  644. obs_output_destroy(outputTest);
  645. outputTest = NULL;
  646. ui->streamButton->setText("Start Streaming");
  647. } else {
  648. const char *url = config_get_string(basicConfig, "OutputTemp",
  649. "URL");
  650. const char *key = config_get_string(basicConfig, "OutputTemp",
  651. "Key");
  652. int vBitrate = config_get_uint(basicConfig, "OutputTemp",
  653. "VBitrate");
  654. int aBitrate = config_get_uint(basicConfig, "OutputTemp",
  655. "ABitrate");
  656. if (!url)
  657. return;
  658. string fullURL = url;
  659. if (key && *key)
  660. fullURL = fullURL + "/" + key;
  661. obs_data_t data = obs_data_create();
  662. obs_data_setstring(data, "filename", fullURL.c_str());
  663. obs_data_setint(data, "audio_bitrate", aBitrate);
  664. obs_data_setint(data, "video_bitrate", vBitrate);
  665. outputTest = obs_output_create("ffmpeg_output", "test", data);
  666. obs_data_release(data);
  667. if (!outputTest)
  668. return;
  669. signal_handler_connect(obs_output_signalhandler(outputTest),
  670. "start", OBSOutputStart, this);
  671. obs_output_start(outputTest);
  672. ui->streamButton->setEnabled(false);
  673. }
  674. }
  675. void OBSBasic::on_settingsButton_clicked()
  676. {
  677. OBSBasicSettings settings(this);
  678. settings.exec();
  679. }
  680. void OBSBasic::GetFPSCommon(uint32_t &num, uint32_t &den) const
  681. {
  682. const char *val = config_get_string(basicConfig, "Video", "FPSCommon");
  683. if (strcmp(val, "10") == 0) {
  684. num = 10;
  685. den = 1;
  686. } else if (strcmp(val, "20") == 0) {
  687. num = 20;
  688. den = 1;
  689. } else if (strcmp(val, "25") == 0) {
  690. num = 25;
  691. den = 1;
  692. } else if (strcmp(val, "29.97") == 0) {
  693. num = 30000;
  694. den = 1001;
  695. } else if (strcmp(val, "48") == 0) {
  696. num = 48;
  697. den = 1;
  698. } else if (strcmp(val, "59.94") == 0) {
  699. num = 60000;
  700. den = 1001;
  701. } else if (strcmp(val, "60") == 0) {
  702. num = 60;
  703. den = 1;
  704. } else {
  705. num = 30;
  706. den = 1;
  707. }
  708. }
  709. void OBSBasic::GetFPSInteger(uint32_t &num, uint32_t &den) const
  710. {
  711. num = (uint32_t)config_get_uint(basicConfig, "Video", "FPSInt");
  712. den = 1;
  713. }
  714. void OBSBasic::GetFPSFraction(uint32_t &num, uint32_t &den) const
  715. {
  716. num = (uint32_t)config_get_uint(basicConfig, "Video", "FPSNum");
  717. den = (uint32_t)config_get_uint(basicConfig, "Video", "FPSDen");
  718. }
  719. void OBSBasic::GetFPSNanoseconds(uint32_t &num, uint32_t &den) const
  720. {
  721. num = 1000000000;
  722. den = (uint32_t)config_get_uint(basicConfig, "Video", "FPSNS");
  723. }
  724. void OBSBasic::GetConfigFPS(uint32_t &num, uint32_t &den) const
  725. {
  726. uint32_t type = config_get_uint(basicConfig, "Video", "FPSType");
  727. if (type == 1) //"Integer"
  728. GetFPSInteger(num, den);
  729. else if (type == 2) //"Fraction"
  730. GetFPSFraction(num, den);
  731. else if (false) //"Nanoseconds", currently not implemented
  732. GetFPSNanoseconds(num, den);
  733. else
  734. GetFPSCommon(num, den);
  735. }
  736. config_t OBSBasic::Config() const
  737. {
  738. return basicConfig;
  739. }
  740. void OBSBasic::UnloadProperties()
  741. {
  742. properties = nullptr;
  743. }