window-basic-main-outputs.cpp 21 KB

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