AdvancedOutput.cpp 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932
  1. #include "AdvancedOutput.hpp"
  2. #include <utility/audio-encoders.hpp>
  3. #include <utility/StartMultiTrackVideoStreamingGuard.hpp>
  4. #include <widgets/OBSBasic.hpp>
  5. #include <qt-wrappers.hpp>
  6. using namespace std;
  7. static OBSData GetDataFromJsonFile(const char *jsonFile)
  8. {
  9. const OBSBasic *basic = OBSBasic::Get();
  10. const OBSProfile &currentProfile = basic->GetCurrentProfile();
  11. const std::filesystem::path jsonFilePath = currentProfile.path / std::filesystem::u8path(jsonFile);
  12. OBSDataAutoRelease data = nullptr;
  13. if (!jsonFilePath.empty()) {
  14. BPtr<char> jsonData = os_quick_read_utf8_file(jsonFilePath.u8string().c_str());
  15. if (!!jsonData) {
  16. data = obs_data_create_from_json(jsonData);
  17. }
  18. }
  19. if (!data) {
  20. data = obs_data_create();
  21. }
  22. return data.Get();
  23. }
  24. static void ApplyEncoderDefaults(OBSData &settings, const obs_encoder_t *encoder)
  25. {
  26. OBSData dataRet = obs_encoder_get_defaults(encoder);
  27. obs_data_release(dataRet);
  28. if (!!settings)
  29. obs_data_apply(dataRet, settings);
  30. settings = std::move(dataRet);
  31. }
  32. #define ADV_ARCHIVE_NAME "adv_archive_audio"
  33. #ifdef __APPLE__
  34. static void translate_macvth264_encoder(const char *&encoder)
  35. {
  36. if (strcmp(encoder, "vt_h264_hw") == 0) {
  37. encoder = "com.apple.videotoolbox.videoencoder.h264.gva";
  38. } else if (strcmp(encoder, "vt_h264_sw") == 0) {
  39. encoder = "com.apple.videotoolbox.videoencoder.h264";
  40. }
  41. }
  42. #endif
  43. AdvancedOutput::AdvancedOutput(OBSBasic *main_) : BasicOutputHandler(main_)
  44. {
  45. const char *recType = config_get_string(main->Config(), "AdvOut", "RecType");
  46. const char *streamEncoder = config_get_string(main->Config(), "AdvOut", "Encoder");
  47. const char *streamAudioEncoder = config_get_string(main->Config(), "AdvOut", "AudioEncoder");
  48. const char *recordEncoder = config_get_string(main->Config(), "AdvOut", "RecEncoder");
  49. const char *recAudioEncoder = config_get_string(main->Config(), "AdvOut", "RecAudioEncoder");
  50. const char *recFormat = config_get_string(main->Config(), "AdvOut", "RecFormat2");
  51. #ifdef __APPLE__
  52. translate_macvth264_encoder(streamEncoder);
  53. translate_macvth264_encoder(recordEncoder);
  54. #endif
  55. ffmpegOutput = astrcmpi(recType, "FFmpeg") == 0;
  56. ffmpegRecording = ffmpegOutput && config_get_bool(main->Config(), "AdvOut", "FFOutputToFile");
  57. useStreamEncoder = astrcmpi(recordEncoder, "none") == 0;
  58. useStreamAudioEncoder = astrcmpi(recAudioEncoder, "none") == 0;
  59. OBSData streamEncSettings = GetDataFromJsonFile("streamEncoder.json");
  60. OBSData recordEncSettings = GetDataFromJsonFile("recordEncoder.json");
  61. if (ffmpegOutput) {
  62. fileOutput = obs_output_create("ffmpeg_output", "adv_ffmpeg_output", nullptr, nullptr);
  63. if (!fileOutput)
  64. throw "Failed to create recording FFmpeg output "
  65. "(advanced output)";
  66. } else {
  67. bool useReplayBuffer = config_get_bool(main->Config(), "AdvOut", "RecRB");
  68. if (useReplayBuffer) {
  69. OBSDataAutoRelease hotkey;
  70. const char *str = config_get_string(main->Config(), "Hotkeys", "ReplayBuffer");
  71. if (str)
  72. hotkey = obs_data_create_from_json(str);
  73. else
  74. hotkey = nullptr;
  75. replayBuffer = obs_output_create("replay_buffer", Str("ReplayBuffer"), nullptr, hotkey);
  76. if (!replayBuffer)
  77. throw "Failed to create replay buffer output "
  78. "(simple output)";
  79. signal_handler_t *signal = obs_output_get_signal_handler(replayBuffer);
  80. startReplayBuffer.Connect(signal, "start", OBSStartReplayBuffer, this);
  81. stopReplayBuffer.Connect(signal, "stop", OBSStopReplayBuffer, this);
  82. replayBufferStopping.Connect(signal, "stopping", OBSReplayBufferStopping, this);
  83. replayBufferSaved.Connect(signal, "saved", OBSReplayBufferSaved, this);
  84. }
  85. const char *mux = "ffmpeg_muxer";
  86. if (strcmp(recFormat, "hybrid_mp4") == 0)
  87. mux = "mp4_output";
  88. else if (strcmp(recFormat, "hybrid_mov") == 0)
  89. mux = "mov_output";
  90. fileOutput = obs_output_create(mux, "adv_file_output", nullptr, nullptr);
  91. if (!fileOutput)
  92. throw "Failed to create recording output "
  93. "(advanced output)";
  94. if (!useStreamEncoder) {
  95. videoRecording = obs_video_encoder_create(recordEncoder, "advanced_video_recording",
  96. recordEncSettings, nullptr);
  97. if (!videoRecording)
  98. throw "Failed to create recording video "
  99. "encoder (advanced output)";
  100. obs_encoder_release(videoRecording);
  101. }
  102. }
  103. videoStreaming = obs_video_encoder_create(streamEncoder, "advanced_video_stream", streamEncSettings, nullptr);
  104. if (!videoStreaming)
  105. throw "Failed to create streaming video encoder "
  106. "(advanced output)";
  107. obs_encoder_release(videoStreaming);
  108. const char *rate_control =
  109. obs_data_get_string(useStreamEncoder ? streamEncSettings : recordEncSettings, "rate_control");
  110. if (!rate_control)
  111. rate_control = "";
  112. usesBitrate = astrcmpi(rate_control, "CBR") == 0 || astrcmpi(rate_control, "VBR") == 0 ||
  113. astrcmpi(rate_control, "ABR") == 0;
  114. for (int i = 0; i < MAX_AUDIO_MIXES; i++) {
  115. char name[19];
  116. snprintf(name, sizeof(name), "adv_record_audio_%d", i);
  117. recordTrack[i] = obs_audio_encoder_create(useStreamAudioEncoder ? streamAudioEncoder : recAudioEncoder,
  118. name, nullptr, i, nullptr);
  119. if (!recordTrack[i]) {
  120. throw "Failed to create audio encoder "
  121. "(advanced output)";
  122. }
  123. obs_encoder_release(recordTrack[i]);
  124. snprintf(name, sizeof(name), "adv_stream_audio_%d", i);
  125. streamTrack[i] = obs_audio_encoder_create(streamAudioEncoder, name, nullptr, i, nullptr);
  126. if (!streamTrack[i]) {
  127. throw "Failed to create streaming audio encoders "
  128. "(advanced output)";
  129. }
  130. obs_encoder_release(streamTrack[i]);
  131. }
  132. std::string id;
  133. int streamTrackIndex = config_get_int(main->Config(), "AdvOut", "TrackIndex") - 1;
  134. streamAudioEnc =
  135. obs_audio_encoder_create(streamAudioEncoder, "adv_stream_audio", nullptr, streamTrackIndex, nullptr);
  136. if (!streamAudioEnc)
  137. throw "Failed to create streaming audio encoder "
  138. "(advanced output)";
  139. obs_encoder_release(streamAudioEnc);
  140. id = "";
  141. int vodTrack = config_get_int(main->Config(), "AdvOut", "VodTrackIndex") - 1;
  142. streamArchiveEnc = obs_audio_encoder_create(streamAudioEncoder, ADV_ARCHIVE_NAME, nullptr, vodTrack, nullptr);
  143. if (!streamArchiveEnc)
  144. throw "Failed to create archive audio encoder "
  145. "(advanced output)";
  146. obs_encoder_release(streamArchiveEnc);
  147. startRecording.Connect(obs_output_get_signal_handler(fileOutput), "start", OBSStartRecording, this);
  148. stopRecording.Connect(obs_output_get_signal_handler(fileOutput), "stop", OBSStopRecording, this);
  149. recordStopping.Connect(obs_output_get_signal_handler(fileOutput), "stopping", OBSRecordStopping, this);
  150. recordFileChanged.Connect(obs_output_get_signal_handler(fileOutput), "file_changed", OBSRecordFileChanged,
  151. this);
  152. }
  153. void AdvancedOutput::UpdateStreamSettings()
  154. {
  155. bool applyServiceSettings = config_get_bool(main->Config(), "AdvOut", "ApplyServiceSettings");
  156. bool enforceBitrate = !config_get_bool(main->Config(), "Stream1", "IgnoreRecommended");
  157. bool dynBitrate = config_get_bool(main->Config(), "Output", "DynamicBitrate");
  158. const char *streamEncoder = config_get_string(main->Config(), "AdvOut", "Encoder");
  159. OBSData settings = GetDataFromJsonFile("streamEncoder.json");
  160. ApplyEncoderDefaults(settings, videoStreaming);
  161. if (applyServiceSettings) {
  162. int bitrate = (int)obs_data_get_int(settings, "bitrate");
  163. int keyint_sec = (int)obs_data_get_int(settings, "keyint_sec");
  164. obs_service_apply_encoder_settings(main->GetService(), settings, nullptr);
  165. if (!enforceBitrate) {
  166. int maxVideoBitrate;
  167. int maxAudioBitrate;
  168. obs_service_get_max_bitrate(main->GetService(), &maxVideoBitrate, &maxAudioBitrate);
  169. std::string videoBitRateLogString = maxVideoBitrate > 0 ? std::to_string(maxVideoBitrate)
  170. : "None";
  171. std::string audioBitRateLogString = maxAudioBitrate > 0 ? std::to_string(maxAudioBitrate)
  172. : "None";
  173. blog(LOG_INFO,
  174. "User is ignoring service bitrate limits.\n"
  175. "Service Recommendations:\n"
  176. "\tvideo bitrate: %s\n"
  177. "\taudio bitrate: %s",
  178. videoBitRateLogString.c_str(), audioBitRateLogString.c_str());
  179. obs_data_set_int(settings, "bitrate", bitrate);
  180. }
  181. int enforced_keyint_sec = (int)obs_data_get_int(settings, "keyint_sec");
  182. if (keyint_sec != 0 && keyint_sec < enforced_keyint_sec)
  183. obs_data_set_int(settings, "keyint_sec", keyint_sec);
  184. } else {
  185. blog(LOG_WARNING, "User is ignoring service settings.");
  186. }
  187. if (dynBitrate && strstr(streamEncoder, "nvenc") != nullptr)
  188. obs_data_set_bool(settings, "lookahead", false);
  189. video_t *video = obs_get_video();
  190. enum video_format format = video_output_get_format(video);
  191. switch (format) {
  192. case VIDEO_FORMAT_I420:
  193. case VIDEO_FORMAT_NV12:
  194. case VIDEO_FORMAT_I010:
  195. case VIDEO_FORMAT_P010:
  196. break;
  197. default:
  198. obs_encoder_set_preferred_video_format(videoStreaming, VIDEO_FORMAT_NV12);
  199. }
  200. obs_encoder_update(videoStreaming, settings);
  201. }
  202. inline void AdvancedOutput::UpdateRecordingSettings()
  203. {
  204. OBSData settings = GetDataFromJsonFile("recordEncoder.json");
  205. obs_encoder_update(videoRecording, settings);
  206. }
  207. void AdvancedOutput::Update()
  208. {
  209. UpdateStreamSettings();
  210. if (!useStreamEncoder && !ffmpegOutput)
  211. UpdateRecordingSettings();
  212. UpdateAudioSettings();
  213. }
  214. inline bool AdvancedOutput::allowsMultiTrack()
  215. {
  216. const char *protocol = nullptr;
  217. obs_service_t *service_obj = main->GetService();
  218. protocol = obs_service_get_protocol(service_obj);
  219. if (!protocol)
  220. return false;
  221. return astrcmpi_n(protocol, SRT_PROTOCOL, strlen(SRT_PROTOCOL)) == 0 ||
  222. astrcmpi_n(protocol, RIST_PROTOCOL, strlen(RIST_PROTOCOL)) == 0;
  223. }
  224. inline void AdvancedOutput::SetupStreaming()
  225. {
  226. const char *rescaleRes = config_get_string(main->Config(), "AdvOut", "RescaleRes");
  227. int rescaleFilter = config_get_int(main->Config(), "AdvOut", "RescaleFilter");
  228. int multiTrackAudioMixes = config_get_int(main->Config(), "AdvOut", "StreamMultiTrackAudioMixes");
  229. unsigned int cx = 0;
  230. unsigned int cy = 0;
  231. int idx = 0;
  232. bool is_multitrack_output = allowsMultiTrack();
  233. if (rescaleFilter != OBS_SCALE_DISABLE && rescaleRes && *rescaleRes) {
  234. if (sscanf(rescaleRes, "%ux%u", &cx, &cy) != 2) {
  235. cx = 0;
  236. cy = 0;
  237. }
  238. }
  239. if (!is_multitrack_output) {
  240. obs_output_set_audio_encoder(streamOutput, streamAudioEnc, 0);
  241. } else {
  242. for (int i = 0; i < MAX_AUDIO_MIXES; i++) {
  243. if ((multiTrackAudioMixes & (1 << i)) != 0) {
  244. obs_output_set_audio_encoder(streamOutput, streamTrack[i], idx);
  245. idx++;
  246. }
  247. }
  248. }
  249. obs_encoder_set_scaled_size(videoStreaming, cx, cy);
  250. obs_encoder_set_gpu_scale_type(videoStreaming, (obs_scale_type)rescaleFilter);
  251. const char *id = obs_service_get_id(main->GetService());
  252. if (strcmp(id, "rtmp_custom") == 0) {
  253. OBSDataAutoRelease settings = obs_data_create();
  254. obs_service_apply_encoder_settings(main->GetService(), settings, nullptr);
  255. obs_encoder_update(videoStreaming, settings);
  256. }
  257. }
  258. inline void AdvancedOutput::SetupRecording()
  259. {
  260. const char *path = config_get_string(main->Config(), "AdvOut", "RecFilePath");
  261. const char *mux = config_get_string(main->Config(), "AdvOut", "RecMuxerCustom");
  262. const char *rescaleRes = config_get_string(main->Config(), "AdvOut", "RecRescaleRes");
  263. int rescaleFilter = config_get_int(main->Config(), "AdvOut", "RecRescaleFilter");
  264. int tracks;
  265. const char *recFormat = config_get_string(main->Config(), "AdvOut", "RecFormat2");
  266. bool is_fragmented = strncmp(recFormat, "fragmented", 10) == 0;
  267. bool flv = strcmp(recFormat, "flv") == 0;
  268. if (flv)
  269. tracks = config_get_int(main->Config(), "AdvOut", "FLVTrack");
  270. else
  271. tracks = config_get_int(main->Config(), "AdvOut", "RecTracks");
  272. OBSDataAutoRelease settings = obs_data_create();
  273. unsigned int cx = 0;
  274. unsigned int cy = 0;
  275. int idx = 0;
  276. /* Hack to allow recordings without any audio tracks selected. It is no
  277. * longer possible to select such a configuration in settings, but legacy
  278. * configurations might still have this configured and we don't want to
  279. * just break them. */
  280. if (tracks == 0)
  281. tracks = config_get_int(main->Config(), "AdvOut", "TrackIndex");
  282. if (useStreamEncoder) {
  283. obs_output_set_video_encoder(fileOutput, videoStreaming);
  284. if (replayBuffer)
  285. obs_output_set_video_encoder(replayBuffer, videoStreaming);
  286. } else {
  287. if (rescaleFilter != OBS_SCALE_DISABLE && rescaleRes && *rescaleRes) {
  288. if (sscanf(rescaleRes, "%ux%u", &cx, &cy) != 2) {
  289. cx = 0;
  290. cy = 0;
  291. }
  292. }
  293. obs_encoder_set_scaled_size(videoRecording, cx, cy);
  294. obs_encoder_set_gpu_scale_type(videoRecording, (obs_scale_type)rescaleFilter);
  295. obs_output_set_video_encoder(fileOutput, videoRecording);
  296. if (replayBuffer)
  297. obs_output_set_video_encoder(replayBuffer, videoRecording);
  298. }
  299. if (!flv) {
  300. for (int i = 0; i < MAX_AUDIO_MIXES; i++) {
  301. if ((tracks & (1 << i)) != 0) {
  302. obs_output_set_audio_encoder(fileOutput, recordTrack[i], idx);
  303. if (replayBuffer)
  304. obs_output_set_audio_encoder(replayBuffer, recordTrack[i], idx);
  305. idx++;
  306. }
  307. }
  308. } else if (flv && tracks != 0) {
  309. obs_output_set_audio_encoder(fileOutput, recordTrack[tracks - 1], idx);
  310. if (replayBuffer)
  311. obs_output_set_audio_encoder(replayBuffer, recordTrack[tracks - 1], idx);
  312. }
  313. // Use fragmented MOV/MP4 if user has not already specified custom movflags
  314. if (is_fragmented && (!mux || strstr(mux, "movflags") == NULL)) {
  315. string mux_frag = "movflags=frag_keyframe+empty_moov+delay_moov";
  316. if (mux) {
  317. mux_frag += " ";
  318. mux_frag += mux;
  319. }
  320. obs_data_set_string(settings, "muxer_settings", mux_frag.c_str());
  321. } else {
  322. if (is_fragmented)
  323. blog(LOG_WARNING, "User enabled fragmented recording, "
  324. "but custom muxer settings contained movflags.");
  325. obs_data_set_string(settings, "muxer_settings", mux);
  326. }
  327. obs_data_set_string(settings, "path", path);
  328. obs_output_update(fileOutput, settings);
  329. if (replayBuffer)
  330. obs_output_update(replayBuffer, settings);
  331. }
  332. inline void AdvancedOutput::SetupFFmpeg()
  333. {
  334. const char *url = config_get_string(main->Config(), "AdvOut", "FFURL");
  335. int vBitrate = config_get_int(main->Config(), "AdvOut", "FFVBitrate");
  336. int gopSize = config_get_int(main->Config(), "AdvOut", "FFVGOPSize");
  337. bool rescale = config_get_bool(main->Config(), "AdvOut", "FFRescale");
  338. const char *rescaleRes = config_get_string(main->Config(), "AdvOut", "FFRescaleRes");
  339. const char *formatName = config_get_string(main->Config(), "AdvOut", "FFFormat");
  340. const char *mimeType = config_get_string(main->Config(), "AdvOut", "FFFormatMimeType");
  341. const char *muxCustom = config_get_string(main->Config(), "AdvOut", "FFMCustom");
  342. const char *vEncoder = config_get_string(main->Config(), "AdvOut", "FFVEncoder");
  343. int vEncoderId = config_get_int(main->Config(), "AdvOut", "FFVEncoderId");
  344. const char *vEncCustom = config_get_string(main->Config(), "AdvOut", "FFVCustom");
  345. int aBitrate = config_get_int(main->Config(), "AdvOut", "FFABitrate");
  346. int aMixes = config_get_int(main->Config(), "AdvOut", "FFAudioMixes");
  347. const char *aEncoder = config_get_string(main->Config(), "AdvOut", "FFAEncoder");
  348. int aEncoderId = config_get_int(main->Config(), "AdvOut", "FFAEncoderId");
  349. const char *aEncCustom = config_get_string(main->Config(), "AdvOut", "FFACustom");
  350. OBSDataArrayAutoRelease audio_names = obs_data_array_create();
  351. for (size_t i = 0; i < MAX_AUDIO_MIXES; i++) {
  352. string cfg_name = "Track";
  353. cfg_name += to_string((int)i + 1);
  354. cfg_name += "Name";
  355. const char *audioName = config_get_string(main->Config(), "AdvOut", cfg_name.c_str());
  356. OBSDataAutoRelease item = obs_data_create();
  357. obs_data_set_string(item, "name", audioName);
  358. obs_data_array_push_back(audio_names, item);
  359. }
  360. OBSDataAutoRelease settings = obs_data_create();
  361. obs_data_set_array(settings, "audio_names", audio_names);
  362. obs_data_set_string(settings, "url", url);
  363. obs_data_set_string(settings, "format_name", formatName);
  364. obs_data_set_string(settings, "format_mime_type", mimeType);
  365. obs_data_set_string(settings, "muxer_settings", muxCustom);
  366. obs_data_set_int(settings, "gop_size", gopSize);
  367. obs_data_set_int(settings, "video_bitrate", vBitrate);
  368. obs_data_set_string(settings, "video_encoder", vEncoder);
  369. obs_data_set_int(settings, "video_encoder_id", vEncoderId);
  370. obs_data_set_string(settings, "video_settings", vEncCustom);
  371. obs_data_set_int(settings, "audio_bitrate", aBitrate);
  372. obs_data_set_string(settings, "audio_encoder", aEncoder);
  373. obs_data_set_int(settings, "audio_encoder_id", aEncoderId);
  374. obs_data_set_string(settings, "audio_settings", aEncCustom);
  375. if (rescale && rescaleRes && *rescaleRes) {
  376. int width;
  377. int height;
  378. int val = sscanf(rescaleRes, "%dx%d", &width, &height);
  379. if (val == 2 && width && height) {
  380. obs_data_set_int(settings, "scale_width", width);
  381. obs_data_set_int(settings, "scale_height", height);
  382. }
  383. }
  384. obs_output_set_mixers(fileOutput, aMixes);
  385. obs_output_set_media(fileOutput, obs_get_video(), obs_get_audio());
  386. obs_output_update(fileOutput, settings);
  387. }
  388. static inline void SetEncoderName(obs_encoder_t *encoder, const char *name, const char *defaultName)
  389. {
  390. obs_encoder_set_name(encoder, (name && *name) ? name : defaultName);
  391. }
  392. inline void AdvancedOutput::UpdateAudioSettings()
  393. {
  394. bool applyServiceSettings = config_get_bool(main->Config(), "AdvOut", "ApplyServiceSettings");
  395. bool enforceBitrate = !config_get_bool(main->Config(), "Stream1", "IgnoreRecommended");
  396. int streamTrackIndex = config_get_int(main->Config(), "AdvOut", "TrackIndex");
  397. int vodTrackIndex = config_get_int(main->Config(), "AdvOut", "VodTrackIndex");
  398. const char *audioEncoder = config_get_string(main->Config(), "AdvOut", "AudioEncoder");
  399. const char *recAudioEncoder = config_get_string(main->Config(), "AdvOut", "RecAudioEncoder");
  400. bool is_multitrack_output = allowsMultiTrack();
  401. OBSDataAutoRelease settings[MAX_AUDIO_MIXES];
  402. for (size_t i = 0; i < MAX_AUDIO_MIXES; i++) {
  403. string cfg_name = "Track";
  404. cfg_name += to_string((int)i + 1);
  405. cfg_name += "Name";
  406. const char *name = config_get_string(main->Config(), "AdvOut", cfg_name.c_str());
  407. string def_name = "Track";
  408. def_name += to_string((int)i + 1);
  409. SetEncoderName(recordTrack[i], name, def_name.c_str());
  410. SetEncoderName(streamTrack[i], name, def_name.c_str());
  411. }
  412. for (size_t i = 0; i < MAX_AUDIO_MIXES; i++) {
  413. int track = (int)(i + 1);
  414. settings[i] = obs_data_create();
  415. obs_data_set_int(settings[i], "bitrate", GetAudioBitrate(i, recAudioEncoder));
  416. obs_encoder_update(recordTrack[i], settings[i]);
  417. obs_data_set_int(settings[i], "bitrate", GetAudioBitrate(i, audioEncoder));
  418. if (!is_multitrack_output) {
  419. if (track == streamTrackIndex || track == vodTrackIndex) {
  420. if (applyServiceSettings) {
  421. int bitrate = (int)obs_data_get_int(settings[i], "bitrate");
  422. obs_service_apply_encoder_settings(main->GetService(), nullptr, settings[i]);
  423. if (!enforceBitrate)
  424. obs_data_set_int(settings[i], "bitrate", bitrate);
  425. }
  426. }
  427. if (track == streamTrackIndex)
  428. obs_encoder_update(streamAudioEnc, settings[i]);
  429. if (track == vodTrackIndex)
  430. obs_encoder_update(streamArchiveEnc, settings[i]);
  431. } else {
  432. obs_encoder_update(streamTrack[i], settings[i]);
  433. }
  434. }
  435. }
  436. void AdvancedOutput::SetupOutputs()
  437. {
  438. obs_encoder_set_video(videoStreaming, obs_get_video());
  439. if (videoRecording)
  440. obs_encoder_set_video(videoRecording, obs_get_video());
  441. for (size_t i = 0; i < MAX_AUDIO_MIXES; i++) {
  442. obs_encoder_set_audio(streamTrack[i], obs_get_audio());
  443. obs_encoder_set_audio(recordTrack[i], obs_get_audio());
  444. }
  445. obs_encoder_set_audio(streamAudioEnc, obs_get_audio());
  446. obs_encoder_set_audio(streamArchiveEnc, obs_get_audio());
  447. SetupStreaming();
  448. if (ffmpegOutput)
  449. SetupFFmpeg();
  450. else
  451. SetupRecording();
  452. }
  453. int AdvancedOutput::GetAudioBitrate(size_t i, const char *id) const
  454. {
  455. static const char *names[] = {
  456. "Track1Bitrate", "Track2Bitrate", "Track3Bitrate", "Track4Bitrate", "Track5Bitrate", "Track6Bitrate",
  457. };
  458. int bitrate = (int)config_get_uint(main->Config(), "AdvOut", names[i]);
  459. return FindClosestAvailableAudioBitrate(id, bitrate);
  460. }
  461. inline std::optional<size_t> AdvancedOutput::VodTrackMixerIdx(obs_service_t *service)
  462. {
  463. int streamTrackIndex = config_get_int(main->Config(), "AdvOut", "TrackIndex");
  464. bool vodTrackEnabled = config_get_bool(main->Config(), "AdvOut", "VodTrackEnabled");
  465. int vodTrackIndex = config_get_int(main->Config(), "AdvOut", "VodTrackIndex");
  466. bool enableForCustomServer = config_get_bool(App()->GetUserConfig(), "General", "EnableCustomServerVodTrack");
  467. const char *id = obs_service_get_id(service);
  468. if (strcmp(id, "rtmp_custom") == 0) {
  469. vodTrackEnabled = enableForCustomServer ? vodTrackEnabled : false;
  470. } else {
  471. OBSDataAutoRelease settings = obs_service_get_settings(service);
  472. const char *service = obs_data_get_string(settings, "service");
  473. if (!ServiceSupportsVodTrack(service))
  474. vodTrackEnabled = false;
  475. }
  476. if (vodTrackEnabled && streamTrackIndex != vodTrackIndex)
  477. return {vodTrackIndex - 1};
  478. return std::nullopt;
  479. }
  480. inline void AdvancedOutput::SetupVodTrack(obs_service_t *service)
  481. {
  482. if (VodTrackMixerIdx(service).has_value())
  483. obs_output_set_audio_encoder(streamOutput, streamArchiveEnc, 1);
  484. else
  485. clear_archive_encoder(streamOutput, ADV_ARCHIVE_NAME);
  486. }
  487. std::shared_future<void> AdvancedOutput::SetupStreaming(obs_service_t *service,
  488. SetupStreamingContinuation_t continuation)
  489. {
  490. int multiTrackAudioMixes = config_get_int(main->Config(), "AdvOut", "StreamMultiTrackAudioMixes");
  491. bool is_multitrack_output = allowsMultiTrack();
  492. if (!useStreamEncoder || (!ffmpegOutput && !obs_output_active(fileOutput))) {
  493. UpdateStreamSettings();
  494. }
  495. UpdateAudioSettings();
  496. if (!Active())
  497. SetupOutputs();
  498. Auth *auth = main->GetAuth();
  499. if (auth)
  500. auth->OnStreamConfig();
  501. /* --------------------- */
  502. const char *type = GetStreamOutputType(service);
  503. if (!type) {
  504. continuation(false);
  505. return StartMultitrackVideoStreamingGuard::MakeReadyFuture();
  506. }
  507. const char *audio_encoder_id = config_get_string(main->Config(), "AdvOut", "AudioEncoder");
  508. int streamTrackIndex = config_get_int(main->Config(), "AdvOut", "TrackIndex") - 1;
  509. auto handle_multitrack_video_result = [=](std::optional<bool> multitrackVideoResult) {
  510. if (multitrackVideoResult.has_value())
  511. return multitrackVideoResult.value();
  512. /* XXX: this is messy and disgusting and should be refactored */
  513. if (outputType != type) {
  514. streamDelayStarting.Disconnect();
  515. streamStopping.Disconnect();
  516. startStreaming.Disconnect();
  517. stopStreaming.Disconnect();
  518. streamOutput = obs_output_create(type, "adv_stream", nullptr, nullptr);
  519. if (!streamOutput) {
  520. blog(LOG_WARNING,
  521. "Creation of stream output type '%s' "
  522. "failed!",
  523. type);
  524. return false;
  525. }
  526. streamDelayStarting.Connect(obs_output_get_signal_handler(streamOutput), "starting",
  527. OBSStreamStarting, this);
  528. streamStopping.Connect(obs_output_get_signal_handler(streamOutput), "stopping",
  529. OBSStreamStopping, this);
  530. startStreaming.Connect(obs_output_get_signal_handler(streamOutput), "start", OBSStartStreaming,
  531. this);
  532. stopStreaming.Connect(obs_output_get_signal_handler(streamOutput), "stop", OBSStopStreaming,
  533. this);
  534. outputType = type;
  535. }
  536. obs_output_set_video_encoder(streamOutput, videoStreaming);
  537. obs_output_set_audio_encoder(streamOutput, streamAudioEnc, 0);
  538. if (!is_multitrack_output) {
  539. obs_output_set_audio_encoder(streamOutput, streamAudioEnc, 0);
  540. } else {
  541. int idx = 0;
  542. for (int i = 0; i < MAX_AUDIO_MIXES; i++) {
  543. if ((multiTrackAudioMixes & (1 << i)) != 0) {
  544. obs_output_set_audio_encoder(streamOutput, streamTrack[i], idx);
  545. idx++;
  546. }
  547. }
  548. }
  549. return true;
  550. };
  551. return SetupMultitrackVideo(service, audio_encoder_id, static_cast<size_t>(streamTrackIndex),
  552. VodTrackMixerIdx(service), [=](std::optional<bool> res) {
  553. continuation(handle_multitrack_video_result(res));
  554. });
  555. }
  556. bool AdvancedOutput::StartStreaming(obs_service_t *service)
  557. {
  558. obs_output_set_service(streamOutput, service);
  559. bool reconnect = config_get_bool(main->Config(), "Output", "Reconnect");
  560. int retryDelay = config_get_int(main->Config(), "Output", "RetryDelay");
  561. int maxRetries = config_get_int(main->Config(), "Output", "MaxRetries");
  562. bool useDelay = config_get_bool(main->Config(), "Output", "DelayEnable");
  563. int delaySec = config_get_int(main->Config(), "Output", "DelaySec");
  564. bool preserveDelay = config_get_bool(main->Config(), "Output", "DelayPreserve");
  565. const char *bindIP = config_get_string(main->Config(), "Output", "BindIP");
  566. const char *ipFamily = config_get_string(main->Config(), "Output", "IPFamily");
  567. #ifdef _WIN32
  568. bool enableNewSocketLoop = config_get_bool(main->Config(), "Output", "NewSocketLoopEnable");
  569. bool enableLowLatencyMode = config_get_bool(main->Config(), "Output", "LowLatencyEnable");
  570. #endif
  571. bool enableDynBitrate = config_get_bool(main->Config(), "Output", "DynamicBitrate");
  572. if (multitrackVideo && multitrackVideoActive &&
  573. !multitrackVideo->HandleIncompatibleSettings(main, main->Config(), service, enableDynBitrate)) {
  574. multitrackVideoActive = false;
  575. return false;
  576. }
  577. bool is_rtmp = false;
  578. obs_service_t *service_obj = main->GetService();
  579. const char *protocol = obs_service_get_protocol(service_obj);
  580. if (protocol) {
  581. if (astrcmpi_n(protocol, RTMP_PROTOCOL, strlen(RTMP_PROTOCOL)) == 0)
  582. is_rtmp = true;
  583. }
  584. OBSDataAutoRelease settings = obs_data_create();
  585. obs_data_set_string(settings, "bind_ip", bindIP);
  586. obs_data_set_string(settings, "ip_family", ipFamily);
  587. #ifdef _WIN32
  588. obs_data_set_bool(settings, "new_socket_loop_enabled", enableNewSocketLoop);
  589. obs_data_set_bool(settings, "low_latency_mode_enabled", enableLowLatencyMode);
  590. #endif
  591. obs_data_set_bool(settings, "dyn_bitrate", enableDynBitrate);
  592. auto streamOutput = StreamingOutput(); // shadowing is sort of bad, but also convenient
  593. obs_output_update(streamOutput, settings);
  594. if (!reconnect)
  595. maxRetries = 0;
  596. obs_output_set_delay(streamOutput, useDelay ? delaySec : 0, preserveDelay ? OBS_OUTPUT_DELAY_PRESERVE : 0);
  597. obs_output_set_reconnect_settings(streamOutput, maxRetries, retryDelay);
  598. if (is_rtmp) {
  599. SetupVodTrack(service);
  600. }
  601. if (obs_output_start(streamOutput)) {
  602. if (multitrackVideo && multitrackVideoActive)
  603. multitrackVideo->StartedStreaming();
  604. return true;
  605. }
  606. if (multitrackVideo && multitrackVideoActive)
  607. multitrackVideoActive = false;
  608. const char *error = obs_output_get_last_error(streamOutput);
  609. bool hasLastError = error && *error;
  610. if (hasLastError)
  611. lastError = error;
  612. else
  613. lastError = string();
  614. const char *type = obs_output_get_id(streamOutput);
  615. blog(LOG_WARNING, "Stream output type '%s' failed to start!%s%s", type, hasLastError ? " Last Error: " : "",
  616. hasLastError ? error : "");
  617. return false;
  618. }
  619. bool AdvancedOutput::StartRecording()
  620. {
  621. const char *path;
  622. const char *recFormat;
  623. const char *filenameFormat;
  624. bool noSpace = false;
  625. bool overwriteIfExists = false;
  626. bool splitFile;
  627. const char *splitFileType;
  628. int splitFileTime;
  629. int splitFileSize;
  630. if (!useStreamEncoder) {
  631. if (!ffmpegOutput) {
  632. UpdateRecordingSettings();
  633. }
  634. } else if (!obs_output_active(StreamingOutput())) {
  635. UpdateStreamSettings();
  636. }
  637. UpdateAudioSettings();
  638. if (!Active())
  639. SetupOutputs();
  640. if (!ffmpegOutput || ffmpegRecording) {
  641. path = config_get_string(main->Config(), "AdvOut", ffmpegRecording ? "FFFilePath" : "RecFilePath");
  642. recFormat = config_get_string(main->Config(), "AdvOut", ffmpegRecording ? "FFExtension" : "RecFormat2");
  643. filenameFormat = config_get_string(main->Config(), "Output", "FilenameFormatting");
  644. overwriteIfExists = config_get_bool(main->Config(), "Output", "OverwriteIfExists");
  645. noSpace = config_get_bool(main->Config(), "AdvOut",
  646. ffmpegRecording ? "FFFileNameWithoutSpace" : "RecFileNameWithoutSpace");
  647. splitFile = config_get_bool(main->Config(), "AdvOut", "RecSplitFile");
  648. string strPath = GetRecordingFilename(path, recFormat, noSpace, overwriteIfExists, filenameFormat,
  649. ffmpegRecording);
  650. OBSDataAutoRelease settings = obs_data_create();
  651. obs_data_set_string(settings, ffmpegRecording ? "url" : "path", strPath.c_str());
  652. if (splitFile) {
  653. splitFileType = config_get_string(main->Config(), "AdvOut", "RecSplitFileType");
  654. splitFileTime = (astrcmpi(splitFileType, "Time") == 0)
  655. ? config_get_int(main->Config(), "AdvOut", "RecSplitFileTime")
  656. : 0;
  657. splitFileSize = (astrcmpi(splitFileType, "Size") == 0)
  658. ? config_get_int(main->Config(), "AdvOut", "RecSplitFileSize")
  659. : 0;
  660. string ext = GetFormatExt(recFormat);
  661. obs_data_set_string(settings, "directory", path);
  662. obs_data_set_string(settings, "format", filenameFormat);
  663. obs_data_set_string(settings, "extension", ext.c_str());
  664. obs_data_set_bool(settings, "allow_spaces", !noSpace);
  665. obs_data_set_bool(settings, "allow_overwrite", overwriteIfExists);
  666. obs_data_set_bool(settings, "split_file", true);
  667. obs_data_set_int(settings, "max_time_sec", splitFileTime * 60);
  668. obs_data_set_int(settings, "max_size_mb", splitFileSize);
  669. }
  670. obs_output_update(fileOutput, settings);
  671. }
  672. if (!obs_output_start(fileOutput)) {
  673. QString error_reason;
  674. const char *error = obs_output_get_last_error(fileOutput);
  675. if (error)
  676. error_reason = QT_UTF8(error);
  677. else
  678. error_reason = QTStr("Output.StartFailedGeneric");
  679. QMessageBox::critical(main, QTStr("Output.StartRecordingFailed"), error_reason);
  680. return false;
  681. }
  682. return true;
  683. }
  684. bool AdvancedOutput::StartReplayBuffer()
  685. {
  686. const char *path;
  687. const char *recFormat;
  688. const char *filenameFormat;
  689. bool noSpace = false;
  690. bool overwriteIfExists = false;
  691. const char *rbPrefix;
  692. const char *rbSuffix;
  693. int rbTime;
  694. int rbSize;
  695. if (!useStreamEncoder) {
  696. if (!ffmpegOutput)
  697. UpdateRecordingSettings();
  698. } else if (!obs_output_active(StreamingOutput())) {
  699. UpdateStreamSettings();
  700. }
  701. UpdateAudioSettings();
  702. if (!Active())
  703. SetupOutputs();
  704. if (!ffmpegOutput || ffmpegRecording) {
  705. path = config_get_string(main->Config(), "AdvOut", ffmpegRecording ? "FFFilePath" : "RecFilePath");
  706. recFormat = config_get_string(main->Config(), "AdvOut", ffmpegRecording ? "FFExtension" : "RecFormat2");
  707. filenameFormat = config_get_string(main->Config(), "Output", "FilenameFormatting");
  708. overwriteIfExists = config_get_bool(main->Config(), "Output", "OverwriteIfExists");
  709. noSpace = config_get_bool(main->Config(), "AdvOut",
  710. ffmpegRecording ? "FFFileNameWithoutSpace" : "RecFileNameWithoutSpace");
  711. rbPrefix = config_get_string(main->Config(), "SimpleOutput", "RecRBPrefix");
  712. rbSuffix = config_get_string(main->Config(), "SimpleOutput", "RecRBSuffix");
  713. rbTime = config_get_int(main->Config(), "AdvOut", "RecRBTime");
  714. rbSize = config_get_int(main->Config(), "AdvOut", "RecRBSize");
  715. string f = GetFormatString(filenameFormat, rbPrefix, rbSuffix);
  716. string ext = GetFormatExt(recFormat);
  717. OBSDataAutoRelease settings = obs_data_create();
  718. obs_data_set_string(settings, "directory", path);
  719. obs_data_set_string(settings, "format", f.c_str());
  720. obs_data_set_string(settings, "extension", ext.c_str());
  721. obs_data_set_bool(settings, "allow_spaces", !noSpace);
  722. obs_data_set_int(settings, "max_time_sec", rbTime);
  723. obs_data_set_int(settings, "max_size_mb", usesBitrate ? 0 : rbSize);
  724. obs_output_update(replayBuffer, settings);
  725. }
  726. if (!obs_output_start(replayBuffer)) {
  727. QString error_reason;
  728. const char *error = obs_output_get_last_error(replayBuffer);
  729. if (error)
  730. error_reason = QT_UTF8(error);
  731. else
  732. error_reason = QTStr("Output.StartFailedGeneric");
  733. QMessageBox::critical(main, QTStr("Output.StartReplayFailed"), error_reason);
  734. return false;
  735. }
  736. return true;
  737. }
  738. void AdvancedOutput::StopStreaming(bool force)
  739. {
  740. auto output = StreamingOutput();
  741. if (force && output)
  742. obs_output_force_stop(output);
  743. else if (multitrackVideo && multitrackVideoActive)
  744. multitrackVideo->StopStreaming();
  745. else
  746. obs_output_stop(output);
  747. }
  748. void AdvancedOutput::StopRecording(bool force)
  749. {
  750. if (force)
  751. obs_output_force_stop(fileOutput);
  752. else
  753. obs_output_stop(fileOutput);
  754. }
  755. void AdvancedOutput::StopReplayBuffer(bool force)
  756. {
  757. if (force)
  758. obs_output_force_stop(replayBuffer);
  759. else
  760. obs_output_stop(replayBuffer);
  761. }
  762. bool AdvancedOutput::StreamingActive() const
  763. {
  764. return obs_output_active(StreamingOutput());
  765. }
  766. bool AdvancedOutput::RecordingActive() const
  767. {
  768. return obs_output_active(fileOutput);
  769. }
  770. bool AdvancedOutput::ReplayBufferActive() const
  771. {
  772. return obs_output_active(replayBuffer);
  773. }