window-basic-main.cpp 46 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908
  1. /******************************************************************************
  2. Copyright (C) 2013-2014 by Hugh Bailey <[email protected]>
  3. 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 <QNetworkRequest>
  20. #include <QNetworkReply>
  21. #include <util/dstr.h>
  22. #include <util/util.hpp>
  23. #include <util/platform.h>
  24. #include <graphics/math-defs.h>
  25. #include "obs-app.hpp"
  26. #include "platform.hpp"
  27. #include "window-basic-settings.hpp"
  28. #include "window-namedialog.hpp"
  29. #include "window-basic-source-select.hpp"
  30. #include "window-basic-main.hpp"
  31. #include "window-basic-properties.hpp"
  32. #include "window-log-reply.hpp"
  33. #include "qt-wrappers.hpp"
  34. #include "display-helpers.hpp"
  35. #include "volume-control.hpp"
  36. #include "ui_OBSBasic.h"
  37. #include <fstream>
  38. #include <sstream>
  39. #include <QScreen>
  40. #include <QWindow>
  41. #define PREVIEW_EDGE_SIZE 10
  42. using namespace std;
  43. Q_DECLARE_METATYPE(OBSScene);
  44. Q_DECLARE_METATYPE(OBSSceneItem);
  45. Q_DECLARE_METATYPE(order_movement);
  46. OBSBasic::OBSBasic(QWidget *parent)
  47. : OBSMainWindow (parent),
  48. ui (new Ui::OBSBasic)
  49. {
  50. ui->setupUi(this);
  51. connect(windowHandle(), &QWindow::screenChanged, [this]() {
  52. struct obs_video_info ovi;
  53. if (obs_get_video_info(&ovi))
  54. ResizePreview(ovi.base_width, ovi.base_height);
  55. });
  56. stringstream name;
  57. name << "OBS " << App()->GetVersionString();
  58. blog(LOG_INFO, "%s", name.str().c_str());
  59. setWindowTitle(QT_UTF8(name.str().c_str()));
  60. }
  61. static void SaveAudioDevice(const char *name, int channel, obs_data_t parent)
  62. {
  63. obs_source_t source = obs_get_output_source(channel);
  64. if (!source)
  65. return;
  66. obs_data_t data = obs_save_source(source);
  67. obs_data_setobj(parent, name, data);
  68. obs_data_release(data);
  69. obs_source_release(source);
  70. }
  71. static obs_data_t GenerateSaveData()
  72. {
  73. obs_data_t saveData = obs_data_create();
  74. obs_data_array_t sourcesArray = obs_save_sources();
  75. obs_source_t currentScene = obs_get_output_source(0);
  76. const char *sceneName = obs_source_getname(currentScene);
  77. SaveAudioDevice(DESKTOP_AUDIO_1, 1, saveData);
  78. SaveAudioDevice(DESKTOP_AUDIO_2, 2, saveData);
  79. SaveAudioDevice(AUX_AUDIO_1, 3, saveData);
  80. SaveAudioDevice(AUX_AUDIO_2, 4, saveData);
  81. SaveAudioDevice(AUX_AUDIO_3, 5, saveData);
  82. obs_data_setstring(saveData, "current_scene", sceneName);
  83. obs_data_setarray(saveData, "sources", sourcesArray);
  84. obs_data_array_release(sourcesArray);
  85. obs_source_release(currentScene);
  86. return saveData;
  87. }
  88. void OBSBasic::ClearVolumeControls()
  89. {
  90. VolControl *control;
  91. for (size_t i = 0; i < volumes.size(); i++) {
  92. control = volumes[i];
  93. delete control;
  94. }
  95. volumes.clear();
  96. }
  97. void OBSBasic::Save(const char *file)
  98. {
  99. obs_data_t saveData = GenerateSaveData();
  100. const char *jsonData = obs_data_getjson(saveData);
  101. /* TODO maybe a message box here? */
  102. if (!os_quick_write_utf8_file(file, jsonData, strlen(jsonData), false))
  103. blog(LOG_ERROR, "Could not save scene data to %s", file);
  104. obs_data_release(saveData);
  105. }
  106. static void LoadAudioDevice(const char *name, int channel, obs_data_t parent)
  107. {
  108. obs_data_t data = obs_data_getobj(parent, name);
  109. if (!data)
  110. return;
  111. obs_source_t source = obs_load_source(data);
  112. if (source) {
  113. obs_set_output_source(channel, source);
  114. obs_source_release(source);
  115. }
  116. obs_data_release(data);
  117. }
  118. void OBSBasic::CreateDefaultScene()
  119. {
  120. obs_scene_t scene = obs_scene_create(Str("Basic.Scene"));
  121. obs_source_t source = obs_scene_getsource(scene);
  122. obs_add_source(source);
  123. #ifdef __APPLE__
  124. source = obs_source_create(OBS_SOURCE_TYPE_INPUT, "display_capture",
  125. Str("Basic.DisplayCapture"), NULL);
  126. if (source) {
  127. obs_scene_add(scene, source);
  128. obs_add_source(source);
  129. obs_source_release(source);
  130. }
  131. #endif
  132. obs_set_output_source(0, obs_scene_getsource(scene));
  133. obs_scene_release(scene);
  134. }
  135. void OBSBasic::Load(const char *file)
  136. {
  137. if (!file) {
  138. blog(LOG_ERROR, "Could not find file %s", file);
  139. return;
  140. }
  141. BPtr<char> jsonData = os_quick_read_utf8_file(file);
  142. if (!jsonData) {
  143. CreateDefaultScene();
  144. return;
  145. }
  146. obs_data_t data = obs_data_create_from_json(jsonData);
  147. obs_data_array_t sources = obs_data_getarray(data, "sources");
  148. const char *sceneName = obs_data_getstring(data, "current_scene");
  149. obs_source_t curScene;
  150. LoadAudioDevice(DESKTOP_AUDIO_1, 1, data);
  151. LoadAudioDevice(DESKTOP_AUDIO_2, 2, data);
  152. LoadAudioDevice(AUX_AUDIO_1, 3, data);
  153. LoadAudioDevice(AUX_AUDIO_2, 4, data);
  154. LoadAudioDevice(AUX_AUDIO_3, 5, data);
  155. obs_load_sources(sources);
  156. curScene = obs_get_source_by_name(sceneName);
  157. obs_set_output_source(0, curScene);
  158. obs_source_release(curScene);
  159. obs_data_array_release(sources);
  160. obs_data_release(data);
  161. }
  162. static inline bool HasAudioDevices(const char *source_id)
  163. {
  164. const char *output_id = source_id;
  165. obs_properties_t props = obs_get_source_properties(
  166. OBS_SOURCE_TYPE_INPUT, output_id);
  167. size_t count = 0;
  168. if (!props)
  169. return false;
  170. obs_property_t devices = obs_properties_get(props, "device_id");
  171. if (devices)
  172. count = obs_property_list_item_count(devices);
  173. obs_properties_destroy(props);
  174. return count != 0;
  175. }
  176. static void OBSStartStreaming(void *data, calldata_t params)
  177. {
  178. UNUSED_PARAMETER(params);
  179. QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
  180. "StreamingStart");
  181. }
  182. static void OBSStopStreaming(void *data, calldata_t params)
  183. {
  184. int code = (int)calldata_int(params, "code");
  185. QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
  186. "StreamingStop", Q_ARG(int, code));
  187. }
  188. static void OBSStopRecording(void *data, calldata_t params)
  189. {
  190. UNUSED_PARAMETER(params);
  191. QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
  192. "RecordingStop");
  193. }
  194. #define SERVICE_PATH "obs-studio/basic/service.json"
  195. void OBSBasic::SaveService()
  196. {
  197. if (!service)
  198. return;
  199. BPtr<char> serviceJsonPath(os_get_config_path(SERVICE_PATH));
  200. if (!serviceJsonPath)
  201. return;
  202. obs_data_t data = obs_data_create();
  203. obs_data_t settings = obs_service_get_settings(service);
  204. obs_data_setstring(data, "type", obs_service_gettype(service));
  205. obs_data_setobj(data, "settings", settings);
  206. const char *json = obs_data_getjson(data);
  207. os_quick_write_utf8_file(serviceJsonPath, json, strlen(json), false);
  208. obs_data_release(settings);
  209. obs_data_release(data);
  210. }
  211. bool OBSBasic::LoadService()
  212. {
  213. const char *type;
  214. BPtr<char> serviceJsonPath(os_get_config_path(SERVICE_PATH));
  215. if (!serviceJsonPath)
  216. return false;
  217. BPtr<char> jsonText = os_quick_read_utf8_file(serviceJsonPath);
  218. if (!jsonText)
  219. return false;
  220. obs_data_t data = obs_data_create_from_json(jsonText);
  221. obs_data_set_default_string(data, "type", "rtmp_common");
  222. type = obs_data_getstring(data, "type");
  223. obs_data_t settings = obs_data_getobj(data, "settings");
  224. service = obs_service_create(type, "default", settings);
  225. obs_data_release(settings);
  226. obs_data_release(data);
  227. return !!service;
  228. }
  229. bool OBSBasic::InitOutputs()
  230. {
  231. fileOutput = obs_output_create("flv_output", "default", nullptr);
  232. if (!fileOutput)
  233. return false;
  234. streamOutput = obs_output_create("rtmp_output", "default", nullptr);
  235. if (!streamOutput)
  236. return false;
  237. signal_handler_connect(obs_output_signalhandler(streamOutput),
  238. "start", OBSStartStreaming, this);
  239. signal_handler_connect(obs_output_signalhandler(streamOutput),
  240. "stop", OBSStopStreaming, this);
  241. signal_handler_connect(obs_output_signalhandler(fileOutput),
  242. "stop", OBSStopRecording, this);
  243. return true;
  244. }
  245. bool OBSBasic::InitEncoders()
  246. {
  247. x264 = obs_video_encoder_create("obs_x264", "h264", nullptr);
  248. if (!x264)
  249. return false;
  250. aac = obs_audio_encoder_create("libfdk_aac", "aac", nullptr);
  251. if(!aac)
  252. aac = obs_audio_encoder_create("ffmpeg_aac", "aac", nullptr);
  253. if (!aac)
  254. return false;
  255. return true;
  256. }
  257. bool OBSBasic::InitService()
  258. {
  259. if (LoadService())
  260. return true;
  261. service = obs_service_create("rtmp_common", nullptr, nullptr);
  262. if (!service)
  263. return false;
  264. return true;
  265. }
  266. bool OBSBasic::InitBasicConfigDefaults()
  267. {
  268. bool hasDesktopAudio = HasAudioDevices(App()->OutputAudioSource());
  269. bool hasInputAudio = HasAudioDevices(App()->InputAudioSource());
  270. config_set_default_int(basicConfig, "Window", "PosX", -1);
  271. config_set_default_int(basicConfig, "Window", "PosY", -1);
  272. config_set_default_int(basicConfig, "Window", "SizeX", -1);
  273. config_set_default_int(basicConfig, "Window", "SizeY", -1);
  274. vector<MonitorInfo> monitors;
  275. GetMonitors(monitors);
  276. if (!monitors.size()) {
  277. OBSErrorBox(NULL, "There appears to be no monitors. Er, this "
  278. "technically shouldn't be possible.");
  279. return false;
  280. }
  281. uint32_t cx = monitors[0].cx;
  282. uint32_t cy = monitors[0].cy;
  283. /* TODO: temporary */
  284. config_set_default_string(basicConfig, "SimpleOutput", "FilePath",
  285. GetDefaultVideoSavePath().c_str());
  286. config_set_default_uint (basicConfig, "SimpleOutput", "VBitrate",
  287. 2500);
  288. config_set_default_uint (basicConfig, "SimpleOutput", "ABitrate", 128);
  289. config_set_default_uint (basicConfig, "Video", "BaseCX", cx);
  290. config_set_default_uint (basicConfig, "Video", "BaseCY", cy);
  291. cx = cx * 10 / 15;
  292. cy = cy * 10 / 15;
  293. config_set_default_uint (basicConfig, "Video", "OutputCX", cx);
  294. config_set_default_uint (basicConfig, "Video", "OutputCY", cy);
  295. config_set_default_uint (basicConfig, "Video", "FPSType", 0);
  296. config_set_default_string(basicConfig, "Video", "FPSCommon", "30");
  297. config_set_default_uint (basicConfig, "Video", "FPSInt", 30);
  298. config_set_default_uint (basicConfig, "Video", "FPSNum", 30);
  299. config_set_default_uint (basicConfig, "Video", "FPSDen", 1);
  300. config_set_default_uint (basicConfig, "Audio", "SampleRate", 44100);
  301. config_set_default_string(basicConfig, "Audio", "ChannelSetup",
  302. "Stereo");
  303. config_set_default_uint (basicConfig, "Audio", "BufferingTime", 1000);
  304. config_set_default_string(basicConfig, "Audio", "DesktopDevice1",
  305. hasDesktopAudio ? "default" : "disabled");
  306. config_set_default_string(basicConfig, "Audio", "DesktopDevice2",
  307. "disabled");
  308. config_set_default_string(basicConfig, "Audio", "AuxDevice1",
  309. hasInputAudio ? "default" : "disabled");
  310. config_set_default_string(basicConfig, "Audio", "AuxDevice2",
  311. "disabled");
  312. config_set_default_string(basicConfig, "Audio", "AuxDevice3",
  313. "disabled");
  314. return true;
  315. }
  316. bool OBSBasic::InitBasicConfig()
  317. {
  318. BPtr<char> configPath(os_get_config_path("obs-studio/basic/basic.ini"));
  319. int code = basicConfig.Open(configPath, CONFIG_OPEN_ALWAYS);
  320. if (code != CONFIG_SUCCESS) {
  321. OBSErrorBox(NULL, "Failed to open basic.ini: %d", code);
  322. return false;
  323. }
  324. return InitBasicConfigDefaults();
  325. }
  326. void OBSBasic::InitOBSCallbacks()
  327. {
  328. signal_handler_connect(obs_signalhandler(), "source_add",
  329. OBSBasic::SourceAdded, this);
  330. signal_handler_connect(obs_signalhandler(), "source_remove",
  331. OBSBasic::SourceRemoved, this);
  332. signal_handler_connect(obs_signalhandler(), "channel_change",
  333. OBSBasic::ChannelChanged, this);
  334. signal_handler_connect(obs_signalhandler(), "source_activate",
  335. OBSBasic::SourceActivated, this);
  336. signal_handler_connect(obs_signalhandler(), "source_deactivate",
  337. OBSBasic::SourceDeactivated, this);
  338. }
  339. void OBSBasic::InitPrimitives()
  340. {
  341. gs_entercontext(obs_graphics());
  342. gs_renderstart(true);
  343. gs_vertex2f(0.0f, 0.0f);
  344. gs_vertex2f(0.0f, 1.0f);
  345. gs_vertex2f(1.0f, 1.0f);
  346. gs_vertex2f(1.0f, 0.0f);
  347. gs_vertex2f(0.0f, 0.0f);
  348. box = gs_rendersave();
  349. gs_renderstart(true);
  350. for (int i = 0; i <= 360; i += (360/20)) {
  351. float pos = RAD(float(i));
  352. gs_vertex2f(cosf(pos), sinf(pos));
  353. }
  354. circle = gs_rendersave();
  355. gs_leavecontext();
  356. }
  357. void OBSBasic::OBSInit()
  358. {
  359. BPtr<char> savePath(os_get_config_path("obs-studio/basic/scenes.json"));
  360. /* make sure it's fully displayed before doing any initialization */
  361. show();
  362. App()->processEvents();
  363. if (!obs_startup(App()->GetLocale()))
  364. throw "Failed to initialize libobs";
  365. if (!InitBasicConfig())
  366. throw "Failed to load basic.ini";
  367. if (!ResetVideo())
  368. throw "Failed to initialize video";
  369. if (!ResetAudio())
  370. throw "Failed to initialize audio";
  371. InitOBSCallbacks();
  372. /* TODO: this is a test, all modules will be searched for and loaded
  373. * automatically later */
  374. obs_load_module("test-input");
  375. obs_load_module("obs-ffmpeg");
  376. obs_load_module("obs-libfdk");
  377. obs_load_module("obs-x264");
  378. obs_load_module("obs-outputs");
  379. obs_load_module("rtmp-services");
  380. #ifdef __APPLE__
  381. obs_load_module("mac-avcapture");
  382. obs_load_module("mac-capture");
  383. #elif _WIN32
  384. obs_load_module("win-wasapi");
  385. obs_load_module("win-capture");
  386. obs_load_module("win-dshow");
  387. #else
  388. obs_load_module("linux-xshm");
  389. obs_load_module("linux-xcomposite");
  390. obs_load_module("linux-pulseaudio");
  391. obs_load_module("linux-v4l2");
  392. #endif
  393. if (!InitOutputs())
  394. throw "Failed to initialize outputs";
  395. if (!InitEncoders())
  396. throw "Failed to initialize encoders";
  397. if (!InitService())
  398. throw "Failed to initialize service";
  399. InitPrimitives();
  400. Load(savePath);
  401. ResetAudioDevices();
  402. loaded = true;
  403. }
  404. OBSBasic::~OBSBasic()
  405. {
  406. BPtr<char> savePath(os_get_config_path("obs-studio/basic/scenes.json"));
  407. SaveService();
  408. Save(savePath);
  409. /* XXX: any obs data must be released before calling obs_shutdown.
  410. * currently, we can't automate this with C++ RAII because of the
  411. * delicate nature of obs_shutdown needing to be freed before the UI
  412. * can be freed, and we have no control over the destruction order of
  413. * the Qt UI stuff, so we have to manually clear any references to
  414. * libobs. */
  415. if (properties)
  416. delete properties;
  417. if (transformWindow)
  418. delete transformWindow;
  419. ClearVolumeControls();
  420. ui->sources->clear();
  421. ui->scenes->clear();
  422. gs_entercontext(obs_graphics());
  423. vertexbuffer_destroy(box);
  424. vertexbuffer_destroy(circle);
  425. gs_leavecontext();
  426. obs_shutdown();
  427. }
  428. OBSScene OBSBasic::GetCurrentScene()
  429. {
  430. QListWidgetItem *item = ui->scenes->currentItem();
  431. return item ? item->data(Qt::UserRole).value<OBSScene>() : nullptr;
  432. }
  433. OBSSceneItem OBSBasic::GetCurrentSceneItem()
  434. {
  435. QListWidgetItem *item = ui->sources->currentItem();
  436. return item ? item->data(Qt::UserRole).value<OBSSceneItem>() : nullptr;
  437. }
  438. void OBSBasic::UpdateSources(OBSScene scene)
  439. {
  440. ui->sources->clear();
  441. obs_scene_enum_items(scene,
  442. [] (obs_scene_t scene, obs_sceneitem_t item, void *p)
  443. {
  444. OBSBasic *window = static_cast<OBSBasic*>(p);
  445. window->InsertSceneItem(item);
  446. UNUSED_PARAMETER(scene);
  447. return true;
  448. }, this);
  449. }
  450. void OBSBasic::InsertSceneItem(obs_sceneitem_t item)
  451. {
  452. obs_source_t source = obs_sceneitem_getsource(item);
  453. const char *name = obs_source_getname(source);
  454. QListWidgetItem *listItem = new QListWidgetItem(QT_UTF8(name));
  455. listItem->setData(Qt::UserRole,
  456. QVariant::fromValue(OBSSceneItem(item)));
  457. ui->sources->insertItem(0, listItem);
  458. ui->sources->setCurrentRow(0);
  459. /* if the source was just created, open properties dialog */
  460. if (sourceSceneRefs[source] == 0 && loaded) {
  461. delete properties;
  462. properties = new OBSBasicProperties(this, source);
  463. properties->Init();
  464. }
  465. }
  466. /* Qt callbacks for invokeMethod */
  467. void OBSBasic::AddScene(OBSSource source)
  468. {
  469. const char *name = obs_source_getname(source);
  470. obs_scene_t scene = obs_scene_fromsource(source);
  471. QListWidgetItem *item = new QListWidgetItem(QT_UTF8(name));
  472. item->setData(Qt::UserRole, QVariant::fromValue(OBSScene(scene)));
  473. ui->scenes->addItem(item);
  474. signal_handler_t handler = obs_source_signalhandler(source);
  475. signal_handler_connect(handler, "item_add",
  476. OBSBasic::SceneItemAdded, this);
  477. signal_handler_connect(handler, "item_remove",
  478. OBSBasic::SceneItemRemoved, this);
  479. signal_handler_connect(handler, "item_move_up",
  480. OBSBasic::SceneItemMoveUp, this);
  481. signal_handler_connect(handler, "item_move_down",
  482. OBSBasic::SceneItemMoveDown, this);
  483. signal_handler_connect(handler, "item_move_top",
  484. OBSBasic::SceneItemMoveTop, this);
  485. signal_handler_connect(handler, "item_move_bottom",
  486. OBSBasic::SceneItemMoveBottom, this);
  487. }
  488. void OBSBasic::RemoveScene(OBSSource source)
  489. {
  490. const char *name = obs_source_getname(source);
  491. QListWidgetItem *sel = ui->scenes->currentItem();
  492. QList<QListWidgetItem*> items = ui->scenes->findItems(QT_UTF8(name),
  493. Qt::MatchExactly);
  494. if (sel != nullptr) {
  495. if (items.contains(sel))
  496. ui->sources->clear();
  497. delete sel;
  498. }
  499. }
  500. void OBSBasic::AddSceneItem(OBSSceneItem item)
  501. {
  502. obs_scene_t scene = obs_sceneitem_getscene(item);
  503. obs_source_t source = obs_sceneitem_getsource(item);
  504. if (GetCurrentScene() == scene)
  505. InsertSceneItem(item);
  506. sourceSceneRefs[source] = sourceSceneRefs[source] + 1;
  507. }
  508. void OBSBasic::RemoveSceneItem(OBSSceneItem item)
  509. {
  510. obs_scene_t scene = obs_sceneitem_getscene(item);
  511. if (GetCurrentScene() == scene) {
  512. for (int i = 0; i < ui->sources->count(); i++) {
  513. QListWidgetItem *listItem = ui->sources->item(i);
  514. QVariant userData = listItem->data(Qt::UserRole);
  515. if (userData.value<OBSSceneItem>() == item) {
  516. delete listItem;
  517. break;
  518. }
  519. }
  520. }
  521. obs_source_t source = obs_sceneitem_getsource(item);
  522. int scenes = sourceSceneRefs[source] - 1;
  523. sourceSceneRefs[source] = scenes;
  524. if (scenes == 0) {
  525. obs_source_remove(source);
  526. sourceSceneRefs.erase(source);
  527. }
  528. }
  529. void OBSBasic::UpdateSceneSelection(OBSSource source)
  530. {
  531. if (source) {
  532. obs_source_type type;
  533. obs_source_gettype(source, &type, NULL);
  534. obs_scene_t scene = obs_scene_fromsource(source);
  535. const char *name = obs_source_getname(source);
  536. if (!scene)
  537. return;
  538. QList<QListWidgetItem*> items =
  539. ui->scenes->findItems(QT_UTF8(name), Qt::MatchExactly);
  540. if (items.count()) {
  541. sceneChanging = true;
  542. ui->scenes->setCurrentItem(items.first());
  543. sceneChanging = false;
  544. UpdateSources(scene);
  545. }
  546. }
  547. }
  548. void OBSBasic::MoveSceneItem(OBSSceneItem item, order_movement movement)
  549. {
  550. OBSScene scene = obs_sceneitem_getscene(item);
  551. if (scene != GetCurrentScene())
  552. return;
  553. int curRow = ui->sources->currentRow();
  554. if (curRow == -1)
  555. return;
  556. QListWidgetItem *listItem = ui->sources->takeItem(curRow);
  557. switch (movement) {
  558. case ORDER_MOVE_UP:
  559. if (curRow > 0)
  560. curRow--;
  561. break;
  562. case ORDER_MOVE_DOWN:
  563. if (curRow < ui->sources->count())
  564. curRow++;
  565. break;
  566. case ORDER_MOVE_TOP:
  567. curRow = 0;
  568. break;
  569. case ORDER_MOVE_BOTTOM:
  570. curRow = ui->sources->count();
  571. break;
  572. }
  573. ui->sources->insertItem(curRow, listItem);
  574. ui->sources->setCurrentRow(curRow);
  575. }
  576. void OBSBasic::ActivateAudioSource(OBSSource source)
  577. {
  578. VolControl *vol = new VolControl(source);
  579. volumes.push_back(vol);
  580. ui->volumeWidgets->layout()->addWidget(vol);
  581. }
  582. void OBSBasic::DeactivateAudioSource(OBSSource source)
  583. {
  584. for (size_t i = 0; i < volumes.size(); i++) {
  585. if (volumes[i]->GetSource() == source) {
  586. delete volumes[i];
  587. volumes.erase(volumes.begin() + i);
  588. break;
  589. }
  590. }
  591. }
  592. /* OBS Callbacks */
  593. void OBSBasic::SceneItemAdded(void *data, calldata_t params)
  594. {
  595. OBSBasic *window = static_cast<OBSBasic*>(data);
  596. obs_sceneitem_t item = (obs_sceneitem_t)calldata_ptr(params, "item");
  597. QMetaObject::invokeMethod(window, "AddSceneItem",
  598. Q_ARG(OBSSceneItem, OBSSceneItem(item)));
  599. }
  600. void OBSBasic::SceneItemRemoved(void *data, calldata_t params)
  601. {
  602. OBSBasic *window = static_cast<OBSBasic*>(data);
  603. obs_sceneitem_t item = (obs_sceneitem_t)calldata_ptr(params, "item");
  604. QMetaObject::invokeMethod(window, "RemoveSceneItem",
  605. Q_ARG(OBSSceneItem, OBSSceneItem(item)));
  606. }
  607. void OBSBasic::SourceAdded(void *data, calldata_t params)
  608. {
  609. OBSBasic *window = static_cast<OBSBasic*>(data);
  610. obs_source_t source = (obs_source_t)calldata_ptr(params, "source");
  611. if (obs_scene_fromsource(source) != NULL)
  612. QMetaObject::invokeMethod(window,
  613. "AddScene",
  614. Q_ARG(OBSSource, OBSSource(source)));
  615. else
  616. window->sourceSceneRefs[source] = 0;
  617. }
  618. void OBSBasic::SourceRemoved(void *data, calldata_t params)
  619. {
  620. obs_source_t source = (obs_source_t)calldata_ptr(params, "source");
  621. if (obs_scene_fromsource(source) != NULL)
  622. QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
  623. "RemoveScene",
  624. Q_ARG(OBSSource, OBSSource(source)));
  625. }
  626. void OBSBasic::SourceActivated(void *data, calldata_t params)
  627. {
  628. obs_source_t source = (obs_source_t)calldata_ptr(params, "source");
  629. uint32_t flags = obs_source_get_output_flags(source);
  630. if (flags & OBS_SOURCE_AUDIO)
  631. QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
  632. "ActivateAudioSource",
  633. Q_ARG(OBSSource, OBSSource(source)));
  634. }
  635. void OBSBasic::SourceDeactivated(void *data, calldata_t params)
  636. {
  637. obs_source_t source = (obs_source_t)calldata_ptr(params, "source");
  638. uint32_t flags = obs_source_get_output_flags(source);
  639. if (flags & OBS_SOURCE_AUDIO)
  640. QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
  641. "DeactivateAudioSource",
  642. Q_ARG(OBSSource, OBSSource(source)));
  643. }
  644. void OBSBasic::ChannelChanged(void *data, calldata_t params)
  645. {
  646. obs_source_t source = (obs_source_t)calldata_ptr(params, "source");
  647. uint32_t channel = (uint32_t)calldata_int(params, "channel");
  648. if (channel == 0)
  649. QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
  650. "UpdateSceneSelection",
  651. Q_ARG(OBSSource, OBSSource(source)));
  652. }
  653. void OBSBasic::DrawBackdrop(float cx, float cy)
  654. {
  655. if (!box)
  656. return;
  657. effect_t solid = obs_get_solid_effect();
  658. eparam_t color = effect_getparambyname(solid, "color");
  659. technique_t tech = effect_gettechnique(solid, "Solid");
  660. vec4 colorVal;
  661. vec4_set(&colorVal, 0.0f, 0.0f, 0.0f, 1.0f);
  662. effect_setvec4(solid, color, &colorVal);
  663. technique_begin(tech);
  664. technique_beginpass(tech, 0);
  665. gs_matrix_push();
  666. gs_matrix_identity();
  667. gs_matrix_scale3f(float(cx), float(cy), 1.0f);
  668. gs_load_vertexbuffer(box);
  669. gs_draw(GS_TRISTRIP, 0, 0);
  670. gs_matrix_pop();
  671. technique_endpass(tech);
  672. technique_end(tech);
  673. gs_load_vertexbuffer(nullptr);
  674. }
  675. void OBSBasic::RenderMain(void *data, uint32_t cx, uint32_t cy)
  676. {
  677. OBSBasic *window = static_cast<OBSBasic*>(data);
  678. obs_video_info ovi;
  679. obs_get_video_info(&ovi);
  680. window->previewCX = int(window->previewScale * float(ovi.base_width));
  681. window->previewCY = int(window->previewScale * float(ovi.base_height));
  682. gs_viewport_push();
  683. gs_projection_push();
  684. /* --------------------------------------- */
  685. gs_ortho(0.0f, float(ovi.base_width), 0.0f, float(ovi.base_height),
  686. -100.0f, 100.0f);
  687. gs_setviewport(window->previewX, window->previewY,
  688. window->previewCX, window->previewCY);
  689. window->DrawBackdrop(float(ovi.base_width), float(ovi.base_height));
  690. obs_render_main_view();
  691. gs_load_vertexbuffer(nullptr);
  692. /* --------------------------------------- */
  693. float right = float(window->ui->preview->width()) - window->previewX;
  694. float bottom = float(window->ui->preview->height()) - window->previewY;
  695. gs_ortho(-window->previewX, right,
  696. -window->previewY, bottom,
  697. -100.0f, 100.0f);
  698. gs_resetviewport();
  699. window->ui->preview->DrawSceneEditing();
  700. /* --------------------------------------- */
  701. gs_projection_pop();
  702. gs_viewport_pop();
  703. UNUSED_PARAMETER(cx);
  704. UNUSED_PARAMETER(cy);
  705. }
  706. void OBSBasic::SceneItemMoveUp(void *data, calldata_t params)
  707. {
  708. OBSSceneItem item = (obs_sceneitem_t)calldata_ptr(params, "item");
  709. QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
  710. "MoveSceneItem",
  711. Q_ARG(OBSSceneItem, OBSSceneItem(item)),
  712. Q_ARG(order_movement, ORDER_MOVE_UP));
  713. }
  714. void OBSBasic::SceneItemMoveDown(void *data, calldata_t params)
  715. {
  716. OBSSceneItem item = (obs_sceneitem_t)calldata_ptr(params, "item");
  717. QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
  718. "MoveSceneItem",
  719. Q_ARG(OBSSceneItem, OBSSceneItem(item)),
  720. Q_ARG(order_movement, ORDER_MOVE_DOWN));
  721. }
  722. void OBSBasic::SceneItemMoveTop(void *data, calldata_t params)
  723. {
  724. OBSSceneItem item = (obs_sceneitem_t)calldata_ptr(params, "item");
  725. QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
  726. "MoveSceneItem",
  727. Q_ARG(OBSSceneItem, OBSSceneItem(item)),
  728. Q_ARG(order_movement, ORDER_MOVE_TOP));
  729. }
  730. void OBSBasic::SceneItemMoveBottom(void *data, calldata_t params)
  731. {
  732. OBSSceneItem item = (obs_sceneitem_t)calldata_ptr(params, "item");
  733. QMetaObject::invokeMethod(static_cast<OBSBasic*>(data),
  734. "MoveSceneItem",
  735. Q_ARG(OBSSceneItem, OBSSceneItem(item)),
  736. Q_ARG(order_movement, ORDER_MOVE_BOTTOM));
  737. }
  738. /* Main class functions */
  739. obs_service_t OBSBasic::GetService()
  740. {
  741. if (!service)
  742. service = obs_service_create("rtmp_common", NULL, NULL);
  743. return service;
  744. }
  745. void OBSBasic::SetService(obs_service_t newService)
  746. {
  747. if (newService) {
  748. if (service)
  749. obs_service_destroy(service);
  750. service = newService;
  751. }
  752. }
  753. bool OBSBasic::ResetVideo()
  754. {
  755. struct obs_video_info ovi;
  756. GetConfigFPS(ovi.fps_num, ovi.fps_den);
  757. ovi.graphics_module = App()->GetRenderModule();
  758. ovi.base_width = (uint32_t)config_get_uint(basicConfig,
  759. "Video", "BaseCX");
  760. ovi.base_height = (uint32_t)config_get_uint(basicConfig,
  761. "Video", "BaseCY");
  762. ovi.output_width = (uint32_t)config_get_uint(basicConfig,
  763. "Video", "OutputCX");
  764. ovi.output_height = (uint32_t)config_get_uint(basicConfig,
  765. "Video", "OutputCY");
  766. ovi.output_format = VIDEO_FORMAT_NV12;
  767. ovi.adapter = 0;
  768. ovi.gpu_conversion = true;
  769. QTToGSWindow(ui->preview->winId(), ovi.window);
  770. //required to make opengl display stuff on osx(?)
  771. ResizePreview(ovi.base_width, ovi.base_height);
  772. QSize size = GetPixelSize(ui->preview);
  773. ovi.window_width = size.width();
  774. ovi.window_height = size.height();
  775. if (!obs_reset_video(&ovi))
  776. return false;
  777. obs_add_draw_callback(OBSBasic::RenderMain, this);
  778. return true;
  779. }
  780. bool OBSBasic::ResetAudio()
  781. {
  782. struct audio_output_info ai;
  783. ai.name = "Main Audio Track";
  784. ai.format = AUDIO_FORMAT_FLOAT;
  785. ai.samples_per_sec = config_get_uint(basicConfig, "Audio",
  786. "SampleRate");
  787. const char *channelSetupStr = config_get_string(basicConfig,
  788. "Audio", "ChannelSetup");
  789. if (strcmp(channelSetupStr, "Mono") == 0)
  790. ai.speakers = SPEAKERS_MONO;
  791. else
  792. ai.speakers = SPEAKERS_STEREO;
  793. ai.buffer_ms = config_get_uint(basicConfig, "Audio", "BufferingTime");
  794. return obs_reset_audio(&ai);
  795. }
  796. void OBSBasic::ResetAudioDevice(const char *sourceId, const char *deviceName,
  797. const char *deviceDesc, int channel)
  798. {
  799. const char *deviceId = config_get_string(basicConfig, "Audio",
  800. deviceName);
  801. obs_source_t source;
  802. obs_data_t settings;
  803. bool same = false;
  804. source = obs_get_output_source(channel);
  805. if (source) {
  806. settings = obs_source_getsettings(source);
  807. const char *curId = obs_data_getstring(settings, "device_id");
  808. same = (strcmp(curId, deviceId) == 0);
  809. obs_data_release(settings);
  810. obs_source_release(source);
  811. }
  812. if (!same)
  813. obs_set_output_source(channel, nullptr);
  814. if (!same && strcmp(deviceId, "disabled") != 0) {
  815. obs_data_t settings = obs_data_create();
  816. obs_data_setstring(settings, "device_id", deviceId);
  817. source = obs_source_create(OBS_SOURCE_TYPE_INPUT,
  818. sourceId, deviceDesc, settings);
  819. obs_data_release(settings);
  820. obs_set_output_source(channel, source);
  821. obs_source_release(source);
  822. }
  823. }
  824. void OBSBasic::ResetAudioDevices()
  825. {
  826. ResetAudioDevice(App()->OutputAudioSource(), "DesktopDevice1",
  827. Str("Basic.DesktopDevice1"), 1);
  828. ResetAudioDevice(App()->OutputAudioSource(), "DesktopDevice2",
  829. Str("Basic.DesktopDevice2"), 2);
  830. ResetAudioDevice(App()->InputAudioSource(), "AuxDevice1",
  831. Str("Basic.AuxDevice1"), 3);
  832. ResetAudioDevice(App()->InputAudioSource(), "AuxDevice2",
  833. Str("Basic.AuxDevice2"), 4);
  834. ResetAudioDevice(App()->InputAudioSource(), "AuxDevice3",
  835. Str("Basic.AuxDevice3"), 5);
  836. }
  837. void OBSBasic::ResizePreview(uint32_t cx, uint32_t cy)
  838. {
  839. QSize targetSize;
  840. /* resize preview panel to fix to the top section of the window */
  841. targetSize = GetPixelSize(ui->preview);
  842. GetScaleAndCenterPos(int(cx), int(cy),
  843. targetSize.width() - PREVIEW_EDGE_SIZE * 2,
  844. targetSize.height() - PREVIEW_EDGE_SIZE * 2,
  845. previewX, previewY, previewScale);
  846. previewX += float(PREVIEW_EDGE_SIZE);
  847. previewY += float(PREVIEW_EDGE_SIZE);
  848. if (isVisible()) {
  849. if (resizeTimer)
  850. killTimer(resizeTimer);
  851. resizeTimer = startTimer(100);
  852. }
  853. }
  854. void OBSBasic::closeEvent(QCloseEvent *event)
  855. {
  856. QWidget::closeEvent(event);
  857. if (!event->isAccepted())
  858. return;
  859. // remove draw callback in case our drawable surfaces go away before
  860. // the destructor gets called
  861. obs_remove_draw_callback(OBSBasic::RenderMain, this);
  862. }
  863. void OBSBasic::changeEvent(QEvent *event)
  864. {
  865. /* TODO */
  866. UNUSED_PARAMETER(event);
  867. }
  868. void OBSBasic::resizeEvent(QResizeEvent *event)
  869. {
  870. struct obs_video_info ovi;
  871. if (obs_get_video_info(&ovi))
  872. ResizePreview(ovi.base_width, ovi.base_height);
  873. UNUSED_PARAMETER(event);
  874. }
  875. void OBSBasic::timerEvent(QTimerEvent *event)
  876. {
  877. if (event->timerId() == resizeTimer) {
  878. killTimer(resizeTimer);
  879. resizeTimer = 0;
  880. QSize size = GetPixelSize(ui->preview);
  881. obs_resize(size.width(), size.height());
  882. }
  883. }
  884. void OBSBasic::on_action_New_triggered()
  885. {
  886. /* TODO */
  887. }
  888. void OBSBasic::on_action_Open_triggered()
  889. {
  890. /* TODO */
  891. }
  892. void OBSBasic::on_action_Save_triggered()
  893. {
  894. /* TODO */
  895. }
  896. void OBSBasic::on_action_Settings_triggered()
  897. {
  898. OBSBasicSettings settings(this);
  899. settings.exec();
  900. }
  901. void OBSBasic::on_scenes_currentItemChanged(QListWidgetItem *current,
  902. QListWidgetItem *prev)
  903. {
  904. obs_source_t source = NULL;
  905. if (sceneChanging)
  906. return;
  907. if (current) {
  908. obs_scene_t scene;
  909. scene = current->data(Qt::UserRole).value<OBSScene>();
  910. source = obs_scene_getsource(scene);
  911. }
  912. /* TODO: allow transitions */
  913. obs_set_output_source(0, source);
  914. UNUSED_PARAMETER(prev);
  915. }
  916. void OBSBasic::on_scenes_customContextMenuRequested(const QPoint &pos)
  917. {
  918. /* TODO */
  919. UNUSED_PARAMETER(pos);
  920. }
  921. void OBSBasic::on_actionAddScene_triggered()
  922. {
  923. string name;
  924. QString format{QTStr("Basic.Main.DefaultSceneName.Text")};
  925. int i = 1;
  926. QString placeHolderText = format.arg(i);
  927. obs_source_t source = nullptr;
  928. while ((source = obs_get_source_by_name(QT_TO_UTF8(placeHolderText)))) {
  929. obs_source_release(source);
  930. placeHolderText = format.arg(++i);
  931. }
  932. bool accepted = NameDialog::AskForName(this,
  933. QTStr("Basic.Main.AddSceneDlg.Title"),
  934. QTStr("Basic.Main.AddSceneDlg.Text"),
  935. name,
  936. placeHolderText);
  937. if (accepted) {
  938. if (name.empty()) {
  939. QMessageBox::information(this,
  940. QTStr("NoNameEntered"),
  941. QTStr("NoNameEntered"));
  942. on_actionAddScene_triggered();
  943. return;
  944. }
  945. obs_source_t source = obs_get_source_by_name(name.c_str());
  946. if (source) {
  947. QMessageBox::information(this,
  948. QTStr("NameExists.Title"),
  949. QTStr("NameExists.Text"));
  950. obs_source_release(source);
  951. on_actionAddScene_triggered();
  952. return;
  953. }
  954. obs_scene_t scene = obs_scene_create(name.c_str());
  955. source = obs_scene_getsource(scene);
  956. obs_add_source(source);
  957. obs_scene_release(scene);
  958. obs_set_output_source(0, source);
  959. }
  960. }
  961. void OBSBasic::on_actionRemoveScene_triggered()
  962. {
  963. QListWidgetItem *item = ui->scenes->currentItem();
  964. if (!item)
  965. return;
  966. QVariant userData = item->data(Qt::UserRole);
  967. obs_scene_t scene = userData.value<OBSScene>();
  968. obs_source_t source = obs_scene_getsource(scene);
  969. obs_source_remove(source);
  970. }
  971. void OBSBasic::on_actionSceneProperties_triggered()
  972. {
  973. /* TODO */
  974. }
  975. void OBSBasic::on_actionSceneUp_triggered()
  976. {
  977. /* TODO */
  978. }
  979. void OBSBasic::on_actionSceneDown_triggered()
  980. {
  981. /* TODO */
  982. }
  983. void OBSBasic::on_sources_currentItemChanged(QListWidgetItem *current,
  984. QListWidgetItem *prev)
  985. {
  986. auto select_one = [] (obs_scene_t scene, obs_sceneitem_t item,
  987. void *param)
  988. {
  989. obs_sceneitem_t selectedItem =
  990. *reinterpret_cast<OBSSceneItem*>(param);
  991. obs_sceneitem_select(item, (selectedItem == item));
  992. UNUSED_PARAMETER(scene);
  993. return true;
  994. };
  995. if (!current)
  996. return;
  997. OBSSceneItem item = current->data(Qt::UserRole).value<OBSSceneItem>();
  998. obs_scene_enum_items(GetCurrentScene(), select_one, &item);
  999. UNUSED_PARAMETER(prev);
  1000. }
  1001. void OBSBasic::on_sources_customContextMenuRequested(const QPoint &pos)
  1002. {
  1003. /* TODO */
  1004. UNUSED_PARAMETER(pos);
  1005. }
  1006. void OBSBasic::AddSource(const char *id)
  1007. {
  1008. OBSBasicSourceSelect sourceSelect(this, id);
  1009. sourceSelect.exec();
  1010. }
  1011. void OBSBasic::AddSourcePopupMenu(const QPoint &pos)
  1012. {
  1013. const char *type;
  1014. bool foundValues = false;
  1015. size_t idx = 0;
  1016. if (!GetCurrentScene()) {
  1017. // Tell the user he needs a scene first (help beginners).
  1018. QMessageBox::information(this,
  1019. QTStr("Basic.Main.AddSourceHelp.Title"),
  1020. QTStr("Basic.Main.AddSourceHelp.Text"));
  1021. return;
  1022. }
  1023. QMenu popup;
  1024. while (obs_enum_input_types(idx++, &type)) {
  1025. const char *name = obs_source_getdisplayname(
  1026. OBS_SOURCE_TYPE_INPUT, type);
  1027. if (strcmp(type, "scene") == 0)
  1028. continue;
  1029. QAction *popupItem = new QAction(QT_UTF8(name), this);
  1030. popupItem->setData(QT_UTF8(type));
  1031. popup.addAction(popupItem);
  1032. foundValues = true;
  1033. }
  1034. if (foundValues) {
  1035. QAction *ret = popup.exec(pos);
  1036. if (ret)
  1037. AddSource(ret->data().toString().toUtf8());
  1038. }
  1039. }
  1040. void OBSBasic::on_actionAddSource_triggered()
  1041. {
  1042. AddSourcePopupMenu(QCursor::pos());
  1043. }
  1044. void OBSBasic::on_actionRemoveSource_triggered()
  1045. {
  1046. OBSSceneItem item = GetCurrentSceneItem();
  1047. if (item)
  1048. obs_sceneitem_remove(item);
  1049. }
  1050. void OBSBasic::on_actionSourceProperties_triggered()
  1051. {
  1052. OBSSceneItem item = GetCurrentSceneItem();
  1053. OBSSource source = obs_sceneitem_getsource(item);
  1054. if (source) {
  1055. delete properties;
  1056. properties = new OBSBasicProperties(this, source);
  1057. properties->Init();
  1058. }
  1059. }
  1060. void OBSBasic::on_actionSourceUp_triggered()
  1061. {
  1062. OBSSceneItem item = GetCurrentSceneItem();
  1063. obs_sceneitem_setorder(item, ORDER_MOVE_UP);
  1064. }
  1065. void OBSBasic::on_actionSourceDown_triggered()
  1066. {
  1067. OBSSceneItem item = GetCurrentSceneItem();
  1068. obs_sceneitem_setorder(item, ORDER_MOVE_DOWN);
  1069. }
  1070. static char *ReadLogFile(const char *log)
  1071. {
  1072. BPtr<char> logDir(os_get_config_path("obs-studio/logs"));
  1073. string path = (char*)logDir;
  1074. path += "/";
  1075. path += log;
  1076. char *file = os_quick_read_utf8_file(path.c_str());
  1077. if (!file)
  1078. blog(LOG_WARNING, "Failed to read log file %s", path.c_str());
  1079. return file;
  1080. }
  1081. void OBSBasic::UploadLog(const char *file)
  1082. {
  1083. dstr fileString = {};
  1084. stringstream ss;
  1085. string jsonData;
  1086. dstr_move_array(&fileString, ReadLogFile(file));
  1087. if (!fileString.array)
  1088. return;
  1089. if (!*fileString.array) {
  1090. dstr_free(&fileString);
  1091. return;
  1092. }
  1093. ui->menuLogFiles->setEnabled(false);
  1094. dstr_replace(&fileString, "\\", "\\\\");
  1095. dstr_replace(&fileString, "\"", "\\\"");
  1096. dstr_replace(&fileString, "\n", "\\n");
  1097. dstr_replace(&fileString, "\r", "\\r");
  1098. dstr_replace(&fileString, "\t", "\\t");
  1099. dstr_replace(&fileString, "\t", "\\t");
  1100. dstr_replace(&fileString, "/", "\\/");
  1101. ss << "{ \"public\": false, \"description\": \"OBS " <<
  1102. App()->GetVersionString() << " log file uploaded at " <<
  1103. CurrentDateTimeString() << "\", \"files\": { \"" <<
  1104. file << "\": { \"content\": \"" <<
  1105. fileString.array << "\" } } }";
  1106. jsonData = std::move(ss.str());
  1107. logUploadPostData.setData(jsonData.c_str(), (int)jsonData.size());
  1108. QUrl url("https://api.github.com/gists");
  1109. logUploadReply = networkManager.post(QNetworkRequest(url),
  1110. &logUploadPostData);
  1111. connect(logUploadReply, SIGNAL(finished()),
  1112. this, SLOT(logUploadFinished()));
  1113. connect(logUploadReply, SIGNAL(readyRead()),
  1114. this, SLOT(logUploadRead()));
  1115. dstr_free(&fileString);
  1116. }
  1117. void OBSBasic::on_actionUploadCurrentLog_triggered()
  1118. {
  1119. UploadLog(App()->GetCurrentLog());
  1120. }
  1121. void OBSBasic::on_actionUploadLastLog_triggered()
  1122. {
  1123. UploadLog(App()->GetLastLog());
  1124. }
  1125. void OBSBasic::logUploadRead()
  1126. {
  1127. logUploadReturnData.push_back(logUploadReply->readAll());
  1128. }
  1129. void OBSBasic::logUploadFinished()
  1130. {
  1131. ui->menuLogFiles->setEnabled(true);
  1132. if (logUploadReply->error()) {
  1133. QMessageBox::information(this,
  1134. QTStr("LogReturnDialog.ErrorUploadingLog"),
  1135. logUploadReply->errorString());
  1136. return;
  1137. }
  1138. const char *jsonReply = logUploadReturnData.constData();
  1139. if (!jsonReply || !*jsonReply)
  1140. return;
  1141. obs_data_t returnData = obs_data_create_from_json(jsonReply);
  1142. QString logURL = obs_data_getstring(returnData, "html_url");
  1143. obs_data_release(returnData);
  1144. OBSLogReply logDialog(this, logURL);
  1145. logDialog.exec();
  1146. }
  1147. void OBSBasic::StreamingStart()
  1148. {
  1149. ui->streamButton->setText("Stop Streaming");
  1150. ui->streamButton->setEnabled(true);
  1151. }
  1152. void OBSBasic::StreamingStop(int code)
  1153. {
  1154. const char *errorMessage;
  1155. switch (code) {
  1156. case OBS_OUTPUT_BAD_PATH:
  1157. errorMessage = Str("Output.ConnectFail.BadPath");
  1158. break;
  1159. case OBS_OUTPUT_CONNECT_FAILED:
  1160. errorMessage = Str("Output.ConnectFail.ConnectFailed");
  1161. break;
  1162. case OBS_OUTPUT_INVALID_STREAM:
  1163. errorMessage = Str("Output.ConnectFail.InvalidStream");
  1164. break;
  1165. default:
  1166. case OBS_OUTPUT_ERROR:
  1167. errorMessage = Str("Output.ConnectFail.Error");
  1168. break;
  1169. case OBS_OUTPUT_DISCONNECTED:
  1170. /* doesn't happen if output is set to reconnect. note that
  1171. * reconnects are handled in the output, not in the UI */
  1172. errorMessage = Str("Output.ConnectFail.Disconnected");
  1173. }
  1174. activeRefs--;
  1175. ui->streamButton->setText(QTStr("Basic.Main.StartStreaming"));
  1176. ui->streamButton->setEnabled(true);
  1177. if (code != OBS_OUTPUT_SUCCESS)
  1178. QMessageBox::information(this,
  1179. QTStr("Output.ConnectFail.Title"),
  1180. QT_UTF8(errorMessage));
  1181. }
  1182. void OBSBasic::RecordingStop()
  1183. {
  1184. activeRefs--;
  1185. ui->recordButton->setText(QTStr("Basic.Main.StartRecording"));
  1186. }
  1187. void OBSBasic::SetupEncoders()
  1188. {
  1189. if (activeRefs == 0) {
  1190. obs_data_t x264Settings = obs_data_create();
  1191. obs_data_t aacSettings = obs_data_create();
  1192. int videoBitrate = config_get_uint(basicConfig, "SimpleOutput",
  1193. "VBitrate");
  1194. int audioBitrate = config_get_uint(basicConfig, "SimpleOutput",
  1195. "ABitrate");
  1196. obs_data_setint(x264Settings, "bitrate", videoBitrate);
  1197. obs_data_setbool(x264Settings, "cbr", true);
  1198. obs_data_setint(aacSettings, "bitrate", audioBitrate);
  1199. obs_encoder_update(x264, x264Settings);
  1200. obs_encoder_update(aac, aacSettings);
  1201. obs_data_release(x264Settings);
  1202. obs_data_release(aacSettings);
  1203. obs_encoder_set_video(x264, obs_video());
  1204. obs_encoder_set_audio(aac, obs_audio());
  1205. }
  1206. }
  1207. void OBSBasic::on_streamButton_clicked()
  1208. {
  1209. if (obs_output_active(streamOutput)) {
  1210. obs_output_stop(streamOutput);
  1211. } else {
  1212. SaveService();
  1213. SetupEncoders();
  1214. obs_output_set_video_encoder(streamOutput, x264);
  1215. obs_output_set_audio_encoder(streamOutput, aac);
  1216. obs_output_set_service(streamOutput, service);
  1217. if (obs_output_start(streamOutput)) {
  1218. activeRefs++;
  1219. ui->streamButton->setEnabled(false);
  1220. ui->streamButton->setText(
  1221. QTStr("Basic.Main.Connecting"));
  1222. }
  1223. }
  1224. }
  1225. void OBSBasic::on_recordButton_clicked()
  1226. {
  1227. if (obs_output_active(fileOutput)) {
  1228. obs_output_stop(fileOutput);
  1229. } else {
  1230. const char *path = config_get_string(basicConfig,
  1231. "SimpleOutput", "FilePath");
  1232. os_dir_t dir = path ? os_opendir(path) : nullptr;
  1233. if (!dir) {
  1234. QMessageBox::information(this,
  1235. QTStr("Output.BadPath.Title"),
  1236. QTStr("Output.BadPath.Text"));
  1237. return;
  1238. }
  1239. os_closedir(dir);
  1240. string strPath;
  1241. strPath += path;
  1242. char lastChar = strPath.back();
  1243. if (lastChar != '/' && lastChar != '\\')
  1244. strPath += "/";
  1245. strPath += GenerateTimeDateFilename("flv");
  1246. SetupEncoders();
  1247. obs_output_set_video_encoder(fileOutput, x264);
  1248. obs_output_set_audio_encoder(fileOutput, aac);
  1249. obs_data_t settings = obs_data_create();
  1250. obs_data_setstring(settings, "path", strPath.c_str());
  1251. obs_output_update(fileOutput, settings);
  1252. obs_data_release(settings);
  1253. if (obs_output_start(fileOutput)) {
  1254. activeRefs++;
  1255. ui->recordButton->setText(
  1256. QTStr("Basic.Main.StopRecording"));
  1257. }
  1258. }
  1259. }
  1260. void OBSBasic::on_settingsButton_clicked()
  1261. {
  1262. OBSBasicSettings settings(this);
  1263. settings.exec();
  1264. }
  1265. void OBSBasic::GetFPSCommon(uint32_t &num, uint32_t &den) const
  1266. {
  1267. const char *val = config_get_string(basicConfig, "Video", "FPSCommon");
  1268. if (strcmp(val, "10") == 0) {
  1269. num = 10;
  1270. den = 1;
  1271. } else if (strcmp(val, "20") == 0) {
  1272. num = 20;
  1273. den = 1;
  1274. } else if (strcmp(val, "25") == 0) {
  1275. num = 25;
  1276. den = 1;
  1277. } else if (strcmp(val, "29.97") == 0) {
  1278. num = 30000;
  1279. den = 1001;
  1280. } else if (strcmp(val, "48") == 0) {
  1281. num = 48;
  1282. den = 1;
  1283. } else if (strcmp(val, "59.94") == 0) {
  1284. num = 60000;
  1285. den = 1001;
  1286. } else if (strcmp(val, "60") == 0) {
  1287. num = 60;
  1288. den = 1;
  1289. } else {
  1290. num = 30;
  1291. den = 1;
  1292. }
  1293. }
  1294. void OBSBasic::GetFPSInteger(uint32_t &num, uint32_t &den) const
  1295. {
  1296. num = (uint32_t)config_get_uint(basicConfig, "Video", "FPSInt");
  1297. den = 1;
  1298. }
  1299. void OBSBasic::GetFPSFraction(uint32_t &num, uint32_t &den) const
  1300. {
  1301. num = (uint32_t)config_get_uint(basicConfig, "Video", "FPSNum");
  1302. den = (uint32_t)config_get_uint(basicConfig, "Video", "FPSDen");
  1303. }
  1304. void OBSBasic::GetFPSNanoseconds(uint32_t &num, uint32_t &den) const
  1305. {
  1306. num = 1000000000;
  1307. den = (uint32_t)config_get_uint(basicConfig, "Video", "FPSNS");
  1308. }
  1309. void OBSBasic::GetConfigFPS(uint32_t &num, uint32_t &den) const
  1310. {
  1311. uint32_t type = config_get_uint(basicConfig, "Video", "FPSType");
  1312. if (type == 1) //"Integer"
  1313. GetFPSInteger(num, den);
  1314. else if (type == 2) //"Fraction"
  1315. GetFPSFraction(num, den);
  1316. else if (false) //"Nanoseconds", currently not implemented
  1317. GetFPSNanoseconds(num, den);
  1318. else
  1319. GetFPSCommon(num, den);
  1320. }
  1321. config_t OBSBasic::Config() const
  1322. {
  1323. return basicConfig;
  1324. }
  1325. void OBSBasic::on_actionEditTransform_triggered()
  1326. {
  1327. delete transformWindow;
  1328. transformWindow = new OBSBasicTransform(this);
  1329. transformWindow->show();
  1330. }
  1331. void OBSBasic::on_actionResetTransform_triggered()
  1332. {
  1333. auto func = [] (obs_scene_t scene, obs_sceneitem_t item, void *param)
  1334. {
  1335. if (!obs_sceneitem_selected(item))
  1336. return true;
  1337. obs_sceneitem_info info;
  1338. vec2_set(&info.pos, 0.0f, 0.0f);
  1339. vec2_set(&info.scale, 1.0f, 1.0f);
  1340. info.rot = 0.0f;
  1341. info.alignment = OBS_ALIGN_TOP | OBS_ALIGN_LEFT;
  1342. info.bounds_type = OBS_BOUNDS_NONE;
  1343. info.bounds_alignment = OBS_ALIGN_CENTER;
  1344. vec2_set(&info.bounds, 0.0f, 0.0f);
  1345. obs_sceneitem_set_info(item, &info);
  1346. UNUSED_PARAMETER(scene);
  1347. UNUSED_PARAMETER(param);
  1348. return true;
  1349. };
  1350. obs_scene_enum_items(GetCurrentScene(), func, nullptr);
  1351. }
  1352. static void GetItemBox(obs_sceneitem_t item, vec3 &tl, vec3 &br)
  1353. {
  1354. matrix4 boxTransform;
  1355. obs_sceneitem_get_box_transform(item, &boxTransform);
  1356. vec3_set(&tl, M_INFINITE, M_INFINITE, 0.0f);
  1357. vec3_set(&br, -M_INFINITE, -M_INFINITE, 0.0f);
  1358. auto GetMinPos = [&] (float x, float y)
  1359. {
  1360. vec3 pos;
  1361. vec3_set(&pos, x, y, 0.0f);
  1362. vec3_transform(&pos, &pos, &boxTransform);
  1363. vec3_min(&tl, &tl, &pos);
  1364. vec3_max(&br, &br, &pos);
  1365. };
  1366. GetMinPos(0.0f, 0.0f);
  1367. GetMinPos(1.0f, 0.0f);
  1368. GetMinPos(0.0f, 1.0f);
  1369. GetMinPos(1.0f, 1.0f);
  1370. }
  1371. static vec3 GetItemTL(obs_sceneitem_t item)
  1372. {
  1373. vec3 tl, br;
  1374. GetItemBox(item, tl, br);
  1375. return tl;
  1376. }
  1377. static void SetItemTL(obs_sceneitem_t item, const vec3 &tl)
  1378. {
  1379. vec3 newTL;
  1380. vec2 pos;
  1381. obs_sceneitem_getpos(item, &pos);
  1382. newTL = GetItemTL(item);
  1383. pos.x += tl.x - newTL.x;
  1384. pos.y += tl.y - newTL.y;
  1385. obs_sceneitem_setpos(item, &pos);
  1386. }
  1387. static bool RotateSelectedSources(obs_scene_t scene, obs_sceneitem_t item,
  1388. void *param)
  1389. {
  1390. if (!obs_sceneitem_selected(item))
  1391. return true;
  1392. float rot = *reinterpret_cast<float*>(param);
  1393. vec3 tl = GetItemTL(item);
  1394. rot += obs_sceneitem_getrot(item);
  1395. if (rot >= 360.0f) rot -= 360.0f;
  1396. else if (rot <= -360.0f) rot += 360.0f;
  1397. obs_sceneitem_setrot(item, rot);
  1398. SetItemTL(item, tl);
  1399. UNUSED_PARAMETER(scene);
  1400. UNUSED_PARAMETER(param);
  1401. return true;
  1402. };
  1403. void OBSBasic::on_actionRotate90CW_triggered()
  1404. {
  1405. float f90CW = 90.0f;
  1406. obs_scene_enum_items(GetCurrentScene(), RotateSelectedSources, &f90CW);
  1407. }
  1408. void OBSBasic::on_actionRotate90CCW_triggered()
  1409. {
  1410. float f90CCW = -90.0f;
  1411. obs_scene_enum_items(GetCurrentScene(), RotateSelectedSources, &f90CCW);
  1412. }
  1413. void OBSBasic::on_actionRotate180_triggered()
  1414. {
  1415. float f180 = 180.0f;
  1416. obs_scene_enum_items(GetCurrentScene(), RotateSelectedSources, &f180);
  1417. }
  1418. static bool MultiplySelectedItemScale(obs_scene_t scene, obs_sceneitem_t item,
  1419. void *param)
  1420. {
  1421. vec2 &mul = *reinterpret_cast<vec2*>(param);
  1422. if (!obs_sceneitem_selected(item))
  1423. return true;
  1424. vec3 tl = GetItemTL(item);
  1425. vec2 scale;
  1426. obs_sceneitem_getscale(item, &scale);
  1427. vec2_mul(&scale, &scale, &mul);
  1428. obs_sceneitem_setscale(item, &scale);
  1429. SetItemTL(item, tl);
  1430. UNUSED_PARAMETER(scene);
  1431. return true;
  1432. }
  1433. void OBSBasic::on_actionFlipHorizontal_triggered()
  1434. {
  1435. vec2 scale;
  1436. vec2_set(&scale, -1.0f, 1.0f);
  1437. obs_scene_enum_items(GetCurrentScene(), MultiplySelectedItemScale,
  1438. &scale);
  1439. }
  1440. void OBSBasic::on_actionFlipVertical_triggered()
  1441. {
  1442. vec2 scale;
  1443. vec2_set(&scale, 1.0f, -1.0f);
  1444. obs_scene_enum_items(GetCurrentScene(), MultiplySelectedItemScale,
  1445. &scale);
  1446. }
  1447. static bool CenterAlignSelectedItems(obs_scene_t scene, obs_sceneitem_t item,
  1448. void *param)
  1449. {
  1450. obs_bounds_type boundsType = *reinterpret_cast<obs_bounds_type*>(param);
  1451. if (!obs_sceneitem_selected(item))
  1452. return true;
  1453. obs_video_info ovi;
  1454. obs_get_video_info(&ovi);
  1455. obs_sceneitem_info itemInfo;
  1456. vec2_set(&itemInfo.pos, 0.0f, 0.0f);
  1457. vec2_set(&itemInfo.scale, 1.0f, 1.0f);
  1458. itemInfo.alignment = OBS_ALIGN_LEFT | OBS_ALIGN_TOP;
  1459. itemInfo.rot = 0.0f;
  1460. vec2_set(&itemInfo.bounds,
  1461. float(ovi.base_width), float(ovi.base_height));
  1462. itemInfo.bounds_type = boundsType;
  1463. itemInfo.bounds_alignment = OBS_ALIGN_CENTER;
  1464. obs_sceneitem_set_info(item, &itemInfo);
  1465. UNUSED_PARAMETER(scene);
  1466. return true;
  1467. }
  1468. void OBSBasic::on_actionFitToScreen_triggered()
  1469. {
  1470. obs_bounds_type boundsType = OBS_BOUNDS_SCALE_INNER;
  1471. obs_scene_enum_items(GetCurrentScene(), CenterAlignSelectedItems,
  1472. &boundsType);
  1473. }
  1474. void OBSBasic::on_actionStretchToScreen_triggered()
  1475. {
  1476. obs_bounds_type boundsType = OBS_BOUNDS_STRETCH;
  1477. obs_scene_enum_items(GetCurrentScene(), CenterAlignSelectedItems,
  1478. &boundsType);
  1479. }
  1480. void OBSBasic::on_actionCenterToScreen_triggered()
  1481. {
  1482. auto func = [] (obs_scene_t scene, obs_sceneitem_t item, void *param)
  1483. {
  1484. vec3 tl, br, itemCenter, screenCenter, offset;
  1485. obs_video_info ovi;
  1486. if (!obs_sceneitem_selected(item))
  1487. return true;
  1488. obs_get_video_info(&ovi);
  1489. vec3_set(&screenCenter, float(ovi.base_width),
  1490. float(ovi.base_height), 0.0f);
  1491. vec3_mulf(&screenCenter, &screenCenter, 0.5f);
  1492. GetItemBox(item, tl, br);
  1493. vec3_sub(&itemCenter, &br, &tl);
  1494. vec3_mulf(&itemCenter, &itemCenter, 0.5f);
  1495. vec3_add(&itemCenter, &itemCenter, &tl);
  1496. vec3_sub(&offset, &screenCenter, &itemCenter);
  1497. vec3_add(&tl, &tl, &offset);
  1498. SetItemTL(item, tl);
  1499. UNUSED_PARAMETER(scene);
  1500. UNUSED_PARAMETER(param);
  1501. return true;
  1502. };
  1503. obs_scene_enum_items(GetCurrentScene(), func, nullptr);
  1504. }