window-basic-main-outputs.cpp 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767
  1. #include <string>
  2. #include <QMessageBox>
  3. #include "window-basic-main.hpp"
  4. #include "window-basic-main-outputs.hpp"
  5. using namespace std;
  6. static void OBSStartStreaming(void *data, calldata_t *params)
  7. {
  8. BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
  9. QMetaObject::invokeMethod(output->main, "StreamingStart");
  10. UNUSED_PARAMETER(params);
  11. }
  12. static void OBSStopStreaming(void *data, calldata_t *params)
  13. {
  14. BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
  15. int code = (int)calldata_int(params, "code");
  16. QMetaObject::invokeMethod(output->main,
  17. "StreamingStop", Q_ARG(int, code));
  18. output->activeRefs--;
  19. }
  20. static void OBSStartRecording(void *data, calldata_t *params)
  21. {
  22. BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
  23. QMetaObject::invokeMethod(output->main, "RecordingStart");
  24. UNUSED_PARAMETER(params);
  25. }
  26. static void OBSStopRecording(void *data, calldata_t *params)
  27. {
  28. BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
  29. QMetaObject::invokeMethod(output->main, "RecordingStop");
  30. output->activeRefs--;
  31. UNUSED_PARAMETER(params);
  32. }
  33. /* ------------------------------------------------------------------------ */
  34. struct SimpleOutput : BasicOutputHandler {
  35. OBSEncoder aac;
  36. OBSEncoder h264;
  37. SimpleOutput(OBSBasic *main_);
  38. virtual void Update() override;
  39. void SetupOutputs();
  40. virtual bool StartStreaming(obs_service_t *service) override;
  41. virtual bool StartRecording() override;
  42. virtual void StopStreaming() override;
  43. virtual void StopRecording() override;
  44. virtual bool StreamingActive() const override;
  45. virtual bool RecordingActive() const override;
  46. };
  47. SimpleOutput::SimpleOutput(OBSBasic *main_) : BasicOutputHandler(main_)
  48. {
  49. streamOutput = obs_output_create("rtmp_output", "simple_stream",
  50. nullptr, nullptr);
  51. if (!streamOutput)
  52. throw "Failed to create stream output (simple output)";
  53. obs_output_release(streamOutput);
  54. fileOutput = obs_output_create("flv_output", "simple_file_output",
  55. nullptr, nullptr);
  56. if (!fileOutput)
  57. throw "Failed to create recording output (simple output)";
  58. obs_output_release(fileOutput);
  59. h264 = obs_video_encoder_create("obs_x264", "simple_h264", nullptr,
  60. nullptr);
  61. if (!h264)
  62. throw "Failed to create h264 encoder (simple output)";
  63. obs_encoder_release(h264);
  64. aac = obs_audio_encoder_create("libfdk_aac", "simple_aac", nullptr, 0,
  65. nullptr);
  66. if (!aac)
  67. aac = obs_audio_encoder_create("ffmpeg_aac", "simple_aac",
  68. nullptr, 0, nullptr);
  69. if (!aac)
  70. throw "Failed to create audio encoder (simple output)";
  71. obs_encoder_release(aac);
  72. signal_handler_connect(obs_output_get_signal_handler(streamOutput),
  73. "start", OBSStartStreaming, this);
  74. signal_handler_connect(obs_output_get_signal_handler(streamOutput),
  75. "stop", OBSStopStreaming, this);
  76. signal_handler_connect(obs_output_get_signal_handler(fileOutput),
  77. "start", OBSStartRecording, this);
  78. signal_handler_connect(obs_output_get_signal_handler(fileOutput),
  79. "stop", OBSStopRecording, this);
  80. }
  81. void SimpleOutput::Update()
  82. {
  83. obs_data_t *h264Settings = obs_data_create();
  84. obs_data_t *aacSettings = obs_data_create();
  85. int videoBitrate = config_get_uint(main->Config(), "SimpleOutput",
  86. "VBitrate");
  87. int videoBufsize = config_get_uint(main->Config(), "SimpleOutput",
  88. "VBufsize");
  89. int audioBitrate = config_get_uint(main->Config(), "SimpleOutput",
  90. "ABitrate");
  91. bool advanced = config_get_bool(main->Config(), "SimpleOutput",
  92. "UseAdvanced");
  93. bool useCBR = config_get_bool(main->Config(), "SimpleOutput",
  94. "UseCBR");
  95. bool useBufsize = config_get_bool(main->Config(), "SimpleOutput",
  96. "UseBufsize");
  97. const char *preset = config_get_string(main->Config(),
  98. "SimpleOutput", "Preset");
  99. const char *custom = config_get_string(main->Config(),
  100. "SimpleOutput", "x264Settings");
  101. obs_data_set_int(h264Settings, "bitrate", videoBitrate);
  102. obs_data_set_bool(h264Settings, "use_bufsize", useBufsize);
  103. obs_data_set_int(h264Settings, "buffer_size", videoBufsize);
  104. if (advanced) {
  105. obs_data_set_string(h264Settings, "preset", preset);
  106. obs_data_set_string(h264Settings, "x264opts", custom);
  107. obs_data_set_bool(h264Settings, "cbr", useCBR);
  108. } else {
  109. obs_data_set_bool(h264Settings, "cbr", true);
  110. }
  111. obs_data_set_int(aacSettings, "bitrate", audioBitrate);
  112. obs_service_apply_encoder_settings(main->GetService(),
  113. h264Settings, aacSettings);
  114. video_t *video = obs_get_video();
  115. enum video_format format = video_output_get_format(video);
  116. if (format != VIDEO_FORMAT_NV12 && format != VIDEO_FORMAT_I420)
  117. obs_encoder_set_preferred_video_format(h264, VIDEO_FORMAT_NV12);
  118. obs_encoder_update(h264, h264Settings);
  119. obs_encoder_update(aac, aacSettings);
  120. obs_data_release(h264Settings);
  121. obs_data_release(aacSettings);
  122. }
  123. inline void SimpleOutput::SetupOutputs()
  124. {
  125. SimpleOutput::Update();
  126. obs_encoder_set_video(h264, obs_get_video());
  127. obs_encoder_set_audio(aac, obs_get_audio());
  128. }
  129. bool SimpleOutput::StartStreaming(obs_service_t *service)
  130. {
  131. if (!Active())
  132. SetupOutputs();
  133. obs_output_set_video_encoder(streamOutput, h264);
  134. obs_output_set_audio_encoder(streamOutput, aac, 0);
  135. obs_output_set_service(streamOutput, service);
  136. bool reconnect = config_get_bool(main->Config(), "SimpleOutput",
  137. "Reconnect");
  138. int retryDelay = config_get_uint(main->Config(), "SimpleOutput",
  139. "RetryDelay");
  140. int maxRetries = config_get_uint(main->Config(), "SimpleOutput",
  141. "MaxRetries");
  142. if (!reconnect)
  143. maxRetries = 0;
  144. obs_output_set_reconnect_settings(streamOutput, maxRetries,
  145. retryDelay);
  146. if (obs_output_start(streamOutput)) {
  147. activeRefs++;
  148. return true;
  149. }
  150. return false;
  151. }
  152. bool SimpleOutput::StartRecording()
  153. {
  154. if (!Active())
  155. SetupOutputs();
  156. const char *path = config_get_string(main->Config(),
  157. "SimpleOutput", "FilePath");
  158. os_dir_t *dir = path ? os_opendir(path) : nullptr;
  159. if (!dir) {
  160. QMessageBox::information(main,
  161. QTStr("Output.BadPath.Title"),
  162. QTStr("Output.BadPath.Text"));
  163. return false;
  164. }
  165. os_closedir(dir);
  166. string strPath;
  167. strPath += path;
  168. char lastChar = strPath.back();
  169. if (lastChar != '/' && lastChar != '\\')
  170. strPath += "/";
  171. strPath += GenerateTimeDateFilename("flv");
  172. SetupOutputs();
  173. obs_output_set_video_encoder(fileOutput, h264);
  174. obs_output_set_audio_encoder(fileOutput, aac, 0);
  175. obs_data_t *settings = obs_data_create();
  176. obs_data_set_string(settings, "path", strPath.c_str());
  177. obs_output_update(fileOutput, settings);
  178. obs_data_release(settings);
  179. if (obs_output_start(fileOutput)) {
  180. activeRefs++;
  181. return true;
  182. }
  183. return false;
  184. }
  185. void SimpleOutput::StopStreaming()
  186. {
  187. obs_output_stop(streamOutput);
  188. }
  189. void SimpleOutput::StopRecording()
  190. {
  191. obs_output_stop(fileOutput);
  192. }
  193. bool SimpleOutput::StreamingActive() const
  194. {
  195. return obs_output_active(streamOutput);
  196. }
  197. bool SimpleOutput::RecordingActive() const
  198. {
  199. return obs_output_active(fileOutput);
  200. }
  201. /* ------------------------------------------------------------------------ */
  202. struct AdvancedOutput : BasicOutputHandler {
  203. OBSEncoder aacTrack[4];
  204. OBSEncoder h264Streaming;
  205. OBSEncoder h264Recording;
  206. bool ffmpegRecording;
  207. bool useStreamEncoder;
  208. AdvancedOutput(OBSBasic *main_);
  209. inline void UpdateStreamSettings();
  210. inline void UpdateRecordingSettings();
  211. inline void UpdateAudioSettings();
  212. virtual void Update() override;
  213. inline void SetupStreaming();
  214. inline void SetupRecording();
  215. inline void SetupFFmpeg();
  216. void SetupOutputs();
  217. virtual bool StartStreaming(obs_service_t *service) override;
  218. virtual bool StartRecording() override;
  219. virtual void StopStreaming() override;
  220. virtual void StopRecording() override;
  221. virtual bool StreamingActive() const override;
  222. virtual bool RecordingActive() const override;
  223. };
  224. static OBSData GetDataFromJsonFile(const char *jsonFile)
  225. {
  226. char fullPath[512];
  227. int ret = os_get_config_path(fullPath, sizeof(fullPath), jsonFile);
  228. if (ret > 0) {
  229. BPtr<char> jsonData = os_quick_read_utf8_file(fullPath);
  230. if (!!jsonData) {
  231. obs_data_t *data = obs_data_create_from_json(jsonData);
  232. OBSData dataRet(data);
  233. obs_data_release(data);
  234. return dataRet;
  235. }
  236. }
  237. return nullptr;
  238. }
  239. AdvancedOutput::AdvancedOutput(OBSBasic *main_) : BasicOutputHandler(main_)
  240. {
  241. const char *recType = config_get_string(main->Config(), "AdvOut",
  242. "RecType");
  243. const char *streamEncoder = config_get_string(main->Config(), "AdvOut",
  244. "Encoder");
  245. const char *recordEncoder = config_get_string(main->Config(), "AdvOut",
  246. "RecEncoder");
  247. ffmpegRecording = astrcmpi(recType, "FFmpeg") == 0;
  248. useStreamEncoder = astrcmpi(recordEncoder, "none") == 0;
  249. OBSData streamEncSettings = GetDataFromJsonFile(
  250. "obs-studio/basic/streamEncoder.json");
  251. OBSData recordEncSettings = GetDataFromJsonFile(
  252. "obs-studio/basic/recordEncoder.json");
  253. streamOutput = obs_output_create("rtmp_output", "adv_stream",
  254. nullptr, nullptr);
  255. if (!streamOutput)
  256. throw "Failed to create stream output (advanced output)";
  257. obs_output_release(streamOutput);
  258. if (ffmpegRecording) {
  259. fileOutput = obs_output_create("ffmpeg_output",
  260. "adv_ffmpeg_output", nullptr, nullptr);
  261. if (!fileOutput)
  262. throw "Failed to create recording FFmpeg output "
  263. "(advanced output)";
  264. obs_output_release(fileOutput);
  265. } else {
  266. fileOutput = obs_output_create("flv_output", "adv_file_output",
  267. nullptr, nullptr);
  268. if (!fileOutput)
  269. throw "Failed to create recording output "
  270. "(advanced output)";
  271. obs_output_release(fileOutput);
  272. if (!useStreamEncoder) {
  273. h264Recording = obs_video_encoder_create(recordEncoder,
  274. "recording_h264", recordEncSettings,
  275. nullptr);
  276. if (!h264Recording)
  277. throw "Failed to create recording h264 "
  278. "encoder (advanced output)";
  279. obs_encoder_release(h264Recording);
  280. }
  281. }
  282. h264Streaming = obs_video_encoder_create(streamEncoder,
  283. "streaming_h264", streamEncSettings, nullptr);
  284. if (!h264Streaming)
  285. throw "Failed to create streaming h264 encoder "
  286. "(advanced output)";
  287. obs_encoder_release(h264Streaming);
  288. for (int i = 0; i < 4; i++) {
  289. char name[9];
  290. sprintf(name, "adv_aac%d", i);
  291. aacTrack[i] = obs_audio_encoder_create("libfdk_aac",
  292. name, nullptr, i, nullptr);
  293. if (!aacTrack[i])
  294. aacTrack[i] = obs_audio_encoder_create("ffmpeg_aac",
  295. name, nullptr, i, nullptr);
  296. if (!aacTrack[i])
  297. throw "Failed to create audio encoder "
  298. "(advanced output)";
  299. obs_encoder_release(aacTrack[i]);
  300. }
  301. signal_handler_connect(obs_output_get_signal_handler(streamOutput),
  302. "start", OBSStartStreaming, this);
  303. signal_handler_connect(obs_output_get_signal_handler(streamOutput),
  304. "stop", OBSStopStreaming, this);
  305. signal_handler_connect(obs_output_get_signal_handler(fileOutput),
  306. "start", OBSStartRecording, this);
  307. signal_handler_connect(obs_output_get_signal_handler(fileOutput),
  308. "stop", OBSStopRecording, this);
  309. }
  310. void AdvancedOutput::UpdateStreamSettings()
  311. {
  312. bool applyServiceSettings = config_get_bool(main->Config(), "AdvOut",
  313. "ApplyServiceSettings");
  314. OBSData settings = GetDataFromJsonFile(
  315. "obs-studio/basic/streamEncoder.json");
  316. if (applyServiceSettings)
  317. obs_service_apply_encoder_settings(main->GetService(),
  318. settings, nullptr);
  319. video_t *video = obs_get_video();
  320. enum video_format format = video_output_get_format(video);
  321. if (format != VIDEO_FORMAT_NV12 && format != VIDEO_FORMAT_I420)
  322. obs_encoder_set_preferred_video_format(h264Streaming,
  323. VIDEO_FORMAT_NV12);
  324. obs_encoder_update(h264Streaming, settings);
  325. }
  326. inline void AdvancedOutput::UpdateRecordingSettings()
  327. {
  328. OBSData settings = GetDataFromJsonFile(
  329. "obs-studio/basic/recordEncoder.json");
  330. obs_encoder_update(h264Recording, settings);
  331. }
  332. void AdvancedOutput::Update()
  333. {
  334. UpdateStreamSettings();
  335. if (!useStreamEncoder && !ffmpegRecording)
  336. UpdateRecordingSettings();
  337. UpdateAudioSettings();
  338. }
  339. inline void AdvancedOutput::SetupStreaming()
  340. {
  341. bool rescale = config_get_bool(main->Config(), "AdvOut",
  342. "Rescale");
  343. const char *rescaleRes = config_get_string(main->Config(), "AdvOut",
  344. "RescaleRes");
  345. bool multitrack = config_get_bool(main->Config(), "AdvOut",
  346. "Multitrack");
  347. int trackIndex = config_get_int(main->Config(), "AdvOut",
  348. "TrackIndex");
  349. int trackCount = config_get_int(main->Config(), "AdvOut",
  350. "TrackCount");
  351. unsigned int cx = 0;
  352. unsigned int cy = 0;
  353. if (rescale && rescaleRes && *rescaleRes) {
  354. if (sscanf(rescaleRes, "%ux%u", &cx, &cy) != 2) {
  355. cx = 0;
  356. cy = 0;
  357. }
  358. }
  359. obs_encoder_set_scaled_size(h264Streaming, cx, cy);
  360. obs_encoder_set_video(h264Streaming, obs_get_video());
  361. obs_output_set_video_encoder(streamOutput, h264Streaming);
  362. if (multitrack) {
  363. int i = 0;
  364. for (; i < trackCount; i++)
  365. obs_output_set_audio_encoder(streamOutput, aacTrack[i],
  366. i);
  367. for (; i < 4; i++)
  368. obs_output_set_audio_encoder(streamOutput, nullptr, i);
  369. } else {
  370. obs_output_set_audio_encoder(streamOutput,
  371. aacTrack[trackIndex - 1], 0);
  372. }
  373. }
  374. inline void AdvancedOutput::SetupRecording()
  375. {
  376. const char *path = config_get_string(main->Config(), "AdvOut",
  377. "RecFilePath");
  378. bool rescale = config_get_bool(main->Config(), "AdvOut",
  379. "RecRescale");
  380. const char *rescaleRes = config_get_string(main->Config(), "AdvOut",
  381. "RecRescaleRes");
  382. bool multitrack = config_get_bool(main->Config(), "AdvOut",
  383. "RecMultitrack");
  384. int trackIndex = config_get_int(main->Config(), "AdvOut",
  385. "RecTrackIndex");
  386. int trackCount = config_get_int(main->Config(), "AdvOut",
  387. "RecTrackCount");
  388. obs_data_t *settings = obs_data_create();
  389. unsigned int cx = 0;
  390. unsigned int cy = 0;
  391. if (useStreamEncoder) {
  392. obs_output_set_video_encoder(fileOutput, h264Streaming);
  393. } else {
  394. if (rescale && rescaleRes && *rescaleRes) {
  395. if (sscanf(rescaleRes, "%ux%u", &cx, &cy) != 2) {
  396. cx = 0;
  397. cy = 0;
  398. }
  399. }
  400. obs_encoder_set_scaled_size(h264Recording, cx, cy);
  401. obs_encoder_set_video(h264Recording, obs_get_video());
  402. obs_output_set_video_encoder(fileOutput, h264Recording);
  403. }
  404. if (multitrack) {
  405. int i = 0;
  406. for (; i < trackCount; i++)
  407. obs_output_set_audio_encoder(fileOutput, aacTrack[i],
  408. i);
  409. for (; i < 4; i++)
  410. obs_output_set_audio_encoder(fileOutput, nullptr, i);
  411. } else {
  412. obs_output_set_audio_encoder(fileOutput,
  413. aacTrack[trackIndex - 1], 0);
  414. }
  415. obs_data_set_string(settings, "path", path);
  416. obs_output_update(fileOutput, settings);
  417. obs_data_release(settings);
  418. }
  419. inline void AdvancedOutput::SetupFFmpeg()
  420. {
  421. const char *url = config_get_string(main->Config(), "AdvOut", "FFURL");
  422. int vBitrate = config_get_int(main->Config(), "AdvOut",
  423. "FFVBitrate");
  424. bool rescale = config_get_bool(main->Config(), "AdvOut",
  425. "FFRescale");
  426. const char *rescaleRes = config_get_string(main->Config(), "AdvOut",
  427. "FFRescaleRes");
  428. const char *formatName = config_get_string(main->Config(), "AdvOut",
  429. "FFFormat");
  430. const char *mimeType = config_get_string(main->Config(), "AdvOut",
  431. "FFFormatMimeType");
  432. const char *vEncoder = config_get_string(main->Config(), "AdvOut",
  433. "FFVEncoder");
  434. int vEncoderId = config_get_int(main->Config(), "AdvOut",
  435. "FFVEncoderId");
  436. const char *vEncCustom = config_get_string(main->Config(), "AdvOut",
  437. "FFVCustom");
  438. int aBitrate = config_get_int(main->Config(), "AdvOut",
  439. "FFABitrate");
  440. int aTrack = config_get_int(main->Config(), "AdvOut",
  441. "FFAudioTrack");
  442. const char *aEncoder = config_get_string(main->Config(), "AdvOut",
  443. "FFAEncoder");
  444. int aEncoderId = config_get_int(main->Config(), "AdvOut",
  445. "FFAEncoderId");
  446. const char *aEncCustom = config_get_string(main->Config(), "AdvOut",
  447. "FFACustom");
  448. obs_data_t *settings = obs_data_create();
  449. obs_data_set_string(settings, "url", url);
  450. obs_data_set_string(settings, "format_name", formatName);
  451. obs_data_set_string(settings, "format_mime_type", mimeType);
  452. obs_data_set_int(settings, "video_bitrate", vBitrate);
  453. obs_data_set_string(settings, "video_encoder", vEncoder);
  454. obs_data_set_int(settings, "video_encoder_id", vEncoderId);
  455. obs_data_set_string(settings, "video_settings", vEncCustom);
  456. obs_data_set_int(settings, "audio_bitrate", aBitrate);
  457. obs_data_set_string(settings, "audio_encoder", aEncoder);
  458. obs_data_set_int(settings, "audio_encoder_id", aEncoderId);
  459. obs_data_set_string(settings, "audio_settings", aEncCustom);
  460. if (rescale && rescaleRes && *rescaleRes) {
  461. int width;
  462. int height;
  463. int val = sscanf(rescaleRes, "%dx%d", &width, &height);
  464. if (val == 2 && width && height) {
  465. obs_data_set_int(settings, "scale_width", width);
  466. obs_data_set_int(settings, "scale_height", height);
  467. }
  468. }
  469. obs_output_set_mixer(fileOutput, aTrack - 1);
  470. obs_output_set_media(fileOutput, obs_get_video(), obs_get_audio());
  471. obs_output_update(fileOutput, settings);
  472. obs_data_release(settings);
  473. }
  474. static inline void SetEncoderName(obs_encoder_t *encoder, const char *name,
  475. const char *defaultName)
  476. {
  477. obs_encoder_set_name(encoder, (name && *name) ? name : defaultName);
  478. }
  479. inline void AdvancedOutput::UpdateAudioSettings()
  480. {
  481. int track1Bitrate = config_get_uint(main->Config(), "AdvOut",
  482. "Track1Bitrate");
  483. int track2Bitrate = config_get_uint(main->Config(), "AdvOut",
  484. "Track2Bitrate");
  485. int track3Bitrate = config_get_uint(main->Config(), "AdvOut",
  486. "Track3Bitrate");
  487. int track4Bitrate = config_get_uint(main->Config(), "AdvOut",
  488. "Track4Bitrate");
  489. const char *name1 = config_get_string(main->Config(), "AdvOut",
  490. "Track1Name");
  491. const char *name2 = config_get_string(main->Config(), "AdvOut",
  492. "Track2Name");
  493. const char *name3 = config_get_string(main->Config(), "AdvOut",
  494. "Track3Name");
  495. const char *name4 = config_get_string(main->Config(), "AdvOut",
  496. "Track4Name");
  497. bool applyServiceSettings = config_get_bool(main->Config(), "AdvOut",
  498. "ApplyServiceSettings");
  499. int streamTrackIndex = config_get_int(main->Config(), "AdvOut",
  500. "TrackIndex");
  501. obs_data_t *settings[4];
  502. for (size_t i = 0; i < 4; i++)
  503. settings[i] = obs_data_create();
  504. obs_data_set_int(settings[0], "bitrate", track1Bitrate);
  505. obs_data_set_int(settings[1], "bitrate", track2Bitrate);
  506. obs_data_set_int(settings[2], "bitrate", track3Bitrate);
  507. obs_data_set_int(settings[3], "bitrate", track4Bitrate);
  508. SetEncoderName(aacTrack[0], name1, "Track1");
  509. SetEncoderName(aacTrack[1], name2, "Track2");
  510. SetEncoderName(aacTrack[2], name3, "Track3");
  511. SetEncoderName(aacTrack[3], name4, "Track4");
  512. for (size_t i = 0; i < 4; i++) {
  513. if (applyServiceSettings && (int)(i + 1) == streamTrackIndex)
  514. obs_service_apply_encoder_settings(main->GetService(),
  515. nullptr, settings[i]);
  516. obs_encoder_update(aacTrack[i], settings[i]);
  517. obs_data_release(settings[i]);
  518. }
  519. }
  520. void AdvancedOutput::SetupOutputs()
  521. {
  522. obs_encoder_set_video(h264Streaming, obs_get_video());
  523. if (h264Recording)
  524. obs_encoder_set_video(h264Recording, obs_get_video());
  525. obs_encoder_set_audio(aacTrack[0], obs_get_audio());
  526. obs_encoder_set_audio(aacTrack[1], obs_get_audio());
  527. obs_encoder_set_audio(aacTrack[2], obs_get_audio());
  528. obs_encoder_set_audio(aacTrack[3], obs_get_audio());
  529. SetupStreaming();
  530. if (ffmpegRecording)
  531. SetupFFmpeg();
  532. else
  533. SetupRecording();
  534. }
  535. bool AdvancedOutput::StartStreaming(obs_service_t *service)
  536. {
  537. if (!useStreamEncoder ||
  538. (!ffmpegRecording && !obs_output_active(fileOutput))) {
  539. UpdateStreamSettings();
  540. }
  541. UpdateAudioSettings();
  542. if (!Active())
  543. SetupOutputs();
  544. obs_output_set_service(streamOutput, service);
  545. bool reconnect = config_get_bool(main->Config(), "AdvOut", "Reconnect");
  546. int retryDelay = config_get_int(main->Config(), "AdvOut", "RetryDelay");
  547. int maxRetries = config_get_int(main->Config(), "AdvOut", "MaxRetries");
  548. if (!reconnect)
  549. maxRetries = 0;
  550. obs_output_set_reconnect_settings(streamOutput, maxRetries,
  551. retryDelay);
  552. if (obs_output_start(streamOutput)) {
  553. activeRefs++;
  554. return true;
  555. }
  556. return false;
  557. }
  558. bool AdvancedOutput::StartRecording()
  559. {
  560. if (!useStreamEncoder) {
  561. if (!ffmpegRecording) {
  562. UpdateRecordingSettings();
  563. }
  564. } else if (!obs_output_active(streamOutput)) {
  565. UpdateStreamSettings();
  566. }
  567. UpdateAudioSettings();
  568. if (!Active())
  569. SetupOutputs();
  570. if (!ffmpegRecording) {
  571. const char *path = config_get_string(main->Config(),
  572. "AdvOut", "RecFilePath");
  573. os_dir_t *dir = path ? os_opendir(path) : nullptr;
  574. if (!dir) {
  575. QMessageBox::information(main,
  576. QTStr("Output.BadPath.Title"),
  577. QTStr("Output.BadPath.Text"));
  578. return false;
  579. }
  580. os_closedir(dir);
  581. string strPath;
  582. strPath += path;
  583. char lastChar = strPath.back();
  584. if (lastChar != '/' && lastChar != '\\')
  585. strPath += "/";
  586. strPath += GenerateTimeDateFilename("flv");
  587. obs_data_t *settings = obs_data_create();
  588. obs_data_set_string(settings, "path", strPath.c_str());
  589. obs_output_update(fileOutput, settings);
  590. obs_data_release(settings);
  591. }
  592. if (obs_output_start(fileOutput)) {
  593. activeRefs++;
  594. return true;
  595. }
  596. return false;
  597. }
  598. void AdvancedOutput::StopStreaming()
  599. {
  600. obs_output_stop(streamOutput);
  601. }
  602. void AdvancedOutput::StopRecording()
  603. {
  604. obs_output_stop(fileOutput);
  605. }
  606. bool AdvancedOutput::StreamingActive() const
  607. {
  608. return obs_output_active(streamOutput);
  609. }
  610. bool AdvancedOutput::RecordingActive() const
  611. {
  612. return obs_output_active(fileOutput);
  613. }
  614. /* ------------------------------------------------------------------------ */
  615. BasicOutputHandler *CreateSimpleOutputHandler(OBSBasic *main)
  616. {
  617. return new SimpleOutput(main);
  618. }
  619. BasicOutputHandler *CreateAdvancedOutputHandler(OBSBasic *main)
  620. {
  621. return new AdvancedOutput(main);
  622. }