window-basic-main.cpp 21 KB

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