window-basic-main-outputs.cpp 21 KB

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