1
0

window-basic-main.cpp 22 KB

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