window-basic-main-outputs.cpp 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793
  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. UNUSED_PARAMETER(params);
  12. }
  13. static void OBSStopStreaming(void *data, calldata_t *params)
  14. {
  15. BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
  16. int code = (int)calldata_int(params, "code");
  17. QMetaObject::invokeMethod(output->main,
  18. "StreamingStop", Q_ARG(int, code));
  19. output->activeRefs--;
  20. }
  21. static void OBSStartRecording(void *data, calldata_t *params)
  22. {
  23. BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
  24. QMetaObject::invokeMethod(output->main, "RecordingStart");
  25. UNUSED_PARAMETER(params);
  26. }
  27. static void OBSStopRecording(void *data, calldata_t *params)
  28. {
  29. BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
  30. int code = (int)calldata_int(params, "code");
  31. QMetaObject::invokeMethod(output->main,
  32. "RecordingStop", Q_ARG(int, code));
  33. output->activeRefs--;
  34. UNUSED_PARAMETER(params);
  35. }
  36. /* ------------------------------------------------------------------------ */
  37. static bool CreateAACEncoder(OBSEncoder &res, string &id, int bitrate,
  38. const char *name, size_t idx)
  39. {
  40. const char *id_ = GetAACEncoderForBitrate(bitrate);
  41. if (!id_) {
  42. id.clear();
  43. res = nullptr;
  44. return false;
  45. }
  46. if (id == id_)
  47. return true;
  48. id = id_;
  49. res = obs_audio_encoder_create(id_, name, nullptr, idx, nullptr);
  50. if (res) {
  51. obs_encoder_release(res);
  52. return true;
  53. }
  54. return false;
  55. }
  56. /* ------------------------------------------------------------------------ */
  57. struct SimpleOutput : BasicOutputHandler {
  58. OBSEncoder aac;
  59. OBSEncoder h264;
  60. string aacEncoderID;
  61. SimpleOutput(OBSBasic *main_);
  62. virtual void Update() override;
  63. void SetupOutputs();
  64. int GetAudioBitrate() const;
  65. virtual bool StartStreaming(obs_service_t *service) override;
  66. virtual bool StartRecording() override;
  67. virtual void StopStreaming() override;
  68. virtual void StopRecording() override;
  69. virtual bool StreamingActive() const override;
  70. virtual bool RecordingActive() const override;
  71. };
  72. SimpleOutput::SimpleOutput(OBSBasic *main_) : BasicOutputHandler(main_)
  73. {
  74. streamOutput = obs_output_create("rtmp_output", "simple_stream",
  75. nullptr, nullptr);
  76. if (!streamOutput)
  77. throw "Failed to create stream output (simple output)";
  78. obs_output_release(streamOutput);
  79. fileOutput = obs_output_create("ffmpeg_muxer", "simple_file_output",
  80. nullptr, nullptr);
  81. if (!fileOutput)
  82. throw "Failed to create recording output (simple output)";
  83. obs_output_release(fileOutput);
  84. h264 = obs_video_encoder_create("obs_x264", "simple_h264", nullptr,
  85. nullptr);
  86. if (!h264)
  87. throw "Failed to create h264 encoder (simple output)";
  88. obs_encoder_release(h264);
  89. if (!CreateAACEncoder(aac, aacEncoderID, GetAudioBitrate(),
  90. "simple_aac", 0))
  91. throw "Failed to create audio encoder (simple output)";
  92. startStreaming.Connect(obs_output_get_signal_handler(streamOutput),
  93. "start", OBSStartStreaming, this);
  94. stopStreaming.Connect(obs_output_get_signal_handler(streamOutput),
  95. "stop", OBSStopStreaming, this);
  96. startRecording.Connect(obs_output_get_signal_handler(fileOutput),
  97. "start", OBSStartRecording, this);
  98. stopRecording.Connect(obs_output_get_signal_handler(fileOutput),
  99. "stop", OBSStopRecording, this);
  100. }
  101. int SimpleOutput::GetAudioBitrate() const
  102. {
  103. return config_get_uint(main->Config(), "SimpleOutput", "ABitrate");
  104. }
  105. void SimpleOutput::Update()
  106. {
  107. obs_data_t *h264Settings = obs_data_create();
  108. obs_data_t *aacSettings = obs_data_create();
  109. int videoBitrate = config_get_uint(main->Config(), "SimpleOutput",
  110. "VBitrate");
  111. int videoBufsize = config_get_uint(main->Config(), "SimpleOutput",
  112. "VBufsize");
  113. int audioBitrate = GetAudioBitrate();
  114. bool advanced = config_get_bool(main->Config(), "SimpleOutput",
  115. "UseAdvanced");
  116. bool useCBR = config_get_bool(main->Config(), "SimpleOutput",
  117. "UseCBR");
  118. bool useBufsize = config_get_bool(main->Config(), "SimpleOutput",
  119. "UseBufsize");
  120. const char *preset = config_get_string(main->Config(),
  121. "SimpleOutput", "Preset");
  122. const char *custom = config_get_string(main->Config(),
  123. "SimpleOutput", "x264Settings");
  124. obs_data_set_int(h264Settings, "bitrate", videoBitrate);
  125. obs_data_set_bool(h264Settings, "use_bufsize", useBufsize);
  126. obs_data_set_int(h264Settings, "buffer_size", videoBufsize);
  127. if (advanced) {
  128. obs_data_set_string(h264Settings, "preset", preset);
  129. obs_data_set_string(h264Settings, "x264opts", custom);
  130. obs_data_set_bool(h264Settings, "cbr", useCBR);
  131. } else {
  132. obs_data_set_bool(h264Settings, "cbr", true);
  133. }
  134. obs_data_set_int(aacSettings, "bitrate", audioBitrate);
  135. obs_service_apply_encoder_settings(main->GetService(),
  136. h264Settings, aacSettings);
  137. video_t *video = obs_get_video();
  138. enum video_format format = video_output_get_format(video);
  139. if (format != VIDEO_FORMAT_NV12 && format != VIDEO_FORMAT_I420)
  140. obs_encoder_set_preferred_video_format(h264, VIDEO_FORMAT_NV12);
  141. obs_encoder_update(h264, h264Settings);
  142. obs_encoder_update(aac, aacSettings);
  143. obs_data_release(h264Settings);
  144. obs_data_release(aacSettings);
  145. }
  146. inline void SimpleOutput::SetupOutputs()
  147. {
  148. SimpleOutput::Update();
  149. obs_encoder_set_video(h264, obs_get_video());
  150. obs_encoder_set_audio(aac, obs_get_audio());
  151. }
  152. bool SimpleOutput::StartStreaming(obs_service_t *service)
  153. {
  154. if (!Active())
  155. SetupOutputs();
  156. obs_output_set_video_encoder(streamOutput, h264);
  157. obs_output_set_audio_encoder(streamOutput, aac, 0);
  158. obs_output_set_service(streamOutput, service);
  159. bool reconnect = config_get_bool(main->Config(), "SimpleOutput",
  160. "Reconnect");
  161. int retryDelay = config_get_uint(main->Config(), "SimpleOutput",
  162. "RetryDelay");
  163. int maxRetries = config_get_uint(main->Config(), "SimpleOutput",
  164. "MaxRetries");
  165. if (!reconnect)
  166. maxRetries = 0;
  167. obs_output_set_reconnect_settings(streamOutput, maxRetries,
  168. retryDelay);
  169. if (obs_output_start(streamOutput)) {
  170. activeRefs++;
  171. return true;
  172. }
  173. return false;
  174. }
  175. bool SimpleOutput::StartRecording()
  176. {
  177. if (!Active())
  178. SetupOutputs();
  179. const char *path = config_get_string(main->Config(),
  180. "SimpleOutput", "FilePath");
  181. const char *format = config_get_string(main->Config(),
  182. "SimpleOutput", "RecFormat");
  183. os_dir_t *dir = path ? os_opendir(path) : nullptr;
  184. if (!dir) {
  185. QMessageBox::information(main,
  186. QTStr("Output.BadPath.Title"),
  187. QTStr("Output.BadPath.Text"));
  188. return false;
  189. }
  190. os_closedir(dir);
  191. string strPath;
  192. strPath += path;
  193. char lastChar = strPath.back();
  194. if (lastChar != '/' && lastChar != '\\')
  195. strPath += "/";
  196. strPath += GenerateTimeDateFilename(format);
  197. SetupOutputs();
  198. obs_output_set_video_encoder(fileOutput, h264);
  199. obs_output_set_audio_encoder(fileOutput, aac, 0);
  200. obs_data_t *settings = obs_data_create();
  201. obs_data_set_string(settings, "path", strPath.c_str());
  202. obs_output_update(fileOutput, settings);
  203. obs_data_release(settings);
  204. if (obs_output_start(fileOutput)) {
  205. activeRefs++;
  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. activeRefs++;
  564. return true;
  565. }
  566. return false;
  567. }
  568. bool AdvancedOutput::StartRecording()
  569. {
  570. const char *path;
  571. const char *format;
  572. if (!useStreamEncoder) {
  573. if (!ffmpegOutput) {
  574. UpdateRecordingSettings();
  575. }
  576. } else if (!obs_output_active(streamOutput)) {
  577. UpdateStreamSettings();
  578. }
  579. UpdateAudioSettings();
  580. if (!Active())
  581. SetupOutputs();
  582. if (!ffmpegOutput || ffmpegRecording) {
  583. path = config_get_string(main->Config(), "AdvOut",
  584. ffmpegRecording ? "FFFilePath" : "RecFilePath");
  585. format = config_get_string(main->Config(), "AdvOut",
  586. ffmpegRecording ? "FFExtension" : "RecFormat");
  587. os_dir_t *dir = path ? os_opendir(path) : nullptr;
  588. if (!dir) {
  589. QMessageBox::information(main,
  590. QTStr("Output.BadPath.Title"),
  591. QTStr("Output.BadPath.Text"));
  592. return false;
  593. }
  594. os_closedir(dir);
  595. string strPath;
  596. strPath += path;
  597. char lastChar = strPath.back();
  598. if (lastChar != '/' && lastChar != '\\')
  599. strPath += "/";
  600. strPath += GenerateTimeDateFilename(format);
  601. obs_data_t *settings = obs_data_create();
  602. obs_data_set_string(settings,
  603. ffmpegRecording ? "url" : "path",
  604. strPath.c_str());
  605. obs_output_update(fileOutput, settings);
  606. obs_data_release(settings);
  607. }
  608. if (obs_output_start(fileOutput)) {
  609. activeRefs++;
  610. return true;
  611. }
  612. return false;
  613. }
  614. void AdvancedOutput::StopStreaming()
  615. {
  616. obs_output_stop(streamOutput);
  617. }
  618. void AdvancedOutput::StopRecording()
  619. {
  620. obs_output_stop(fileOutput);
  621. }
  622. bool AdvancedOutput::StreamingActive() const
  623. {
  624. return obs_output_active(streamOutput);
  625. }
  626. bool AdvancedOutput::RecordingActive() const
  627. {
  628. return obs_output_active(fileOutput);
  629. }
  630. /* ------------------------------------------------------------------------ */
  631. BasicOutputHandler *CreateSimpleOutputHandler(OBSBasic *main)
  632. {
  633. return new SimpleOutput(main);
  634. }
  635. BasicOutputHandler *CreateAdvancedOutputHandler(OBSBasic *main)
  636. {
  637. return new AdvancedOutput(main);
  638. }