MultitrackVideoOutput.cpp 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845
  1. #include "MultitrackVideoError.hpp"
  2. #include "MultitrackVideoOutput.hpp"
  3. #include "models/multitrack-video.hpp"
  4. #include "GoLiveAPI_Network.hpp"
  5. #include "GoLiveAPI_PostData.hpp"
  6. #include <OBSApp.hpp>
  7. #include <bpm.h>
  8. #include <util/dstr.hpp>
  9. #include <QPushButton>
  10. #include <QMessageBox>
  11. #include <QThread>
  12. #include <QUrl>
  13. #include <QUrlQuery>
  14. #include <cinttypes>
  15. Qt::ConnectionType BlockingConnectionTypeFor(QObject *object)
  16. {
  17. return object->thread() == QThread::currentThread() ? Qt::DirectConnection : Qt::BlockingQueuedConnection;
  18. }
  19. bool MultitrackVideoDeveloperModeEnabled()
  20. {
  21. static bool developer_mode = [] {
  22. auto args = qApp->arguments();
  23. for (const auto &arg : args) {
  24. if (arg == "--enable-multitrack-video-dev") {
  25. return true;
  26. }
  27. }
  28. return false;
  29. }();
  30. return developer_mode;
  31. }
  32. static OBSServiceAutoRelease create_service(const GoLiveApi::Config &go_live_config,
  33. const std::optional<std::string> &rtmp_url, const QString &in_stream_key,
  34. std::optional<bool> use_rtmps)
  35. {
  36. const char *url = nullptr;
  37. QString stream_key = in_stream_key;
  38. const auto &ingest_endpoints = go_live_config.ingest_endpoints;
  39. for (auto &endpoint : ingest_endpoints) {
  40. if (qstrnicmp("RTMP", endpoint.protocol.c_str(), 4))
  41. continue;
  42. if (use_rtmps.has_value() && *use_rtmps != (qstricmp("RTMPS", endpoint.protocol.c_str()) == 0))
  43. continue;
  44. url = endpoint.url_template.c_str();
  45. if (endpoint.authentication && !endpoint.authentication->empty()) {
  46. blog(LOG_INFO, "Using stream key supplied by autoconfig");
  47. stream_key = QString::fromStdString(*endpoint.authentication);
  48. }
  49. break;
  50. }
  51. if (rtmp_url.has_value()) {
  52. // Despite being set by user, it was set to a ""
  53. if (rtmp_url->empty()) {
  54. throw MultitrackVideoError::warning(QTStr("FailedToStartStream.NoCustomRTMPURLInSettings"));
  55. }
  56. url = rtmp_url->c_str();
  57. blog(LOG_INFO, "Using custom RTMP URL: '%s'", url);
  58. } else {
  59. if (!url) {
  60. blog(LOG_ERROR, "No RTMP URL in go live config");
  61. throw MultitrackVideoError::warning(QTStr("FailedToStartStream.NoRTMPURLInConfig"));
  62. }
  63. blog(LOG_INFO, "Using URL template: '%s'", url);
  64. }
  65. DStr str;
  66. dstr_cat(str, url);
  67. // dstr_find does not protect against null, and dstr_cat will
  68. // not initialize str if cat'ing with a null url
  69. if (!dstr_is_empty(str)) {
  70. auto found = dstr_find(str, "/{stream_key}");
  71. if (found)
  72. dstr_remove(str, found - str->array, str->len - (found - str->array));
  73. }
  74. /* The stream key itself may contain query parameters, such as
  75. * "bandwidthtest" that need to be carried over. */
  76. QUrl parsed_user_key{in_stream_key};
  77. QUrlQuery user_key_query{parsed_user_key};
  78. QUrl parsed_key{stream_key};
  79. QUrl parsed_url{url};
  80. QUrlQuery parsed_query{parsed_url};
  81. for (const auto &[key, value] : user_key_query.queryItems())
  82. parsed_query.addQueryItem(key, value);
  83. if (!go_live_config.meta.config_id.empty()) {
  84. parsed_query.addQueryItem("clientConfigId", QString::fromStdString(go_live_config.meta.config_id));
  85. }
  86. parsed_key.setQuery(parsed_query);
  87. OBSDataAutoRelease settings = obs_data_create();
  88. obs_data_set_string(settings, "server", str->array);
  89. obs_data_set_string(settings, "key", parsed_key.toString().toUtf8().constData());
  90. auto service = obs_service_create("rtmp_custom", "multitrack video service", settings, nullptr);
  91. if (!service) {
  92. blog(LOG_WARNING, "Failed to create multitrack video service");
  93. throw MultitrackVideoError::warning(QTStr("FailedToStartStream.FailedToCreateMultitrackVideoService"));
  94. }
  95. return service;
  96. }
  97. static OBSOutputAutoRelease create_output()
  98. {
  99. OBSOutputAutoRelease output = obs_output_create("rtmp_output", "rtmp multitrack video", nullptr, nullptr);
  100. if (!output) {
  101. blog(LOG_ERROR, "Failed to create multitrack video rtmp output");
  102. throw MultitrackVideoError::warning(QTStr("FailedToStartStream.FailedToCreateMultitrackVideoOutput"));
  103. }
  104. return output;
  105. }
  106. static OBSOutputAutoRelease create_recording_output(obs_data_t *settings)
  107. {
  108. OBSOutputAutoRelease output;
  109. bool useMP4 = obs_data_get_bool(settings, "use_mp4");
  110. if (useMP4) {
  111. output = obs_output_create("mp4_output", "mp4 multitrack video", settings, nullptr);
  112. } else {
  113. output = obs_output_create("flv_output", "flv multitrack video", settings, nullptr);
  114. }
  115. if (!output) {
  116. blog(LOG_ERROR, "Failed to create multitrack video %s output", useMP4 ? "mp4" : "flv");
  117. }
  118. return output;
  119. }
  120. static void adjust_video_encoder_scaling(const obs_video_info &ovi, obs_encoder_t *video_encoder,
  121. const GoLiveApi::VideoEncoderConfiguration &encoder_config,
  122. size_t encoder_index)
  123. {
  124. auto requested_width = encoder_config.width;
  125. auto requested_height = encoder_config.height;
  126. if (ovi.base_width < requested_width || ovi.base_height < requested_height) {
  127. blog(LOG_WARNING,
  128. "Requested resolution exceeds canvas/available resolution for encoder %zu: %" PRIu32 "x%" PRIu32
  129. " > %" PRIu32 "x%" PRIu32,
  130. encoder_index, requested_width, requested_height, ovi.base_width, ovi.base_height);
  131. }
  132. obs_encoder_set_scaled_size(video_encoder, requested_width, requested_height);
  133. obs_encoder_set_gpu_scale_type(video_encoder, encoder_config.gpu_scale_type.value_or(OBS_SCALE_BICUBIC));
  134. obs_encoder_set_preferred_video_format(video_encoder, encoder_config.format.value_or(VIDEO_FORMAT_NV12));
  135. obs_encoder_set_preferred_color_space(video_encoder, encoder_config.colorspace.value_or(VIDEO_CS_709));
  136. obs_encoder_set_preferred_range(video_encoder, encoder_config.range.value_or(VIDEO_RANGE_PARTIAL));
  137. }
  138. static uint32_t closest_divisor(const obs_video_info &ovi, const media_frames_per_second &target_fps)
  139. {
  140. auto target = (uint64_t)target_fps.numerator * ovi.fps_den;
  141. auto source = (uint64_t)ovi.fps_num * target_fps.denominator;
  142. return std::max(1u, static_cast<uint32_t>(source / target));
  143. }
  144. static void adjust_encoder_frame_rate_divisor(const obs_video_info &ovi, obs_encoder_t *video_encoder,
  145. const GoLiveApi::VideoEncoderConfiguration &encoder_config,
  146. const size_t encoder_index)
  147. {
  148. if (!encoder_config.framerate) {
  149. blog(LOG_WARNING, "`framerate` not specified for encoder %zu", encoder_index);
  150. return;
  151. }
  152. media_frames_per_second requested_fps = *encoder_config.framerate;
  153. if (ovi.fps_num == requested_fps.numerator && ovi.fps_den == requested_fps.denominator)
  154. return;
  155. auto divisor = closest_divisor(ovi, requested_fps);
  156. if (divisor <= 1)
  157. return;
  158. blog(LOG_INFO, "Setting frame rate divisor to %u for encoder %zu", divisor, encoder_index);
  159. obs_encoder_set_frame_rate_divisor(video_encoder, divisor);
  160. }
  161. static bool encoder_available(const char *type)
  162. {
  163. const char *id = nullptr;
  164. for (size_t idx = 0; obs_enum_encoder_types(idx, &id); idx++) {
  165. if (strcmp(id, type) == 0)
  166. return true;
  167. }
  168. return false;
  169. }
  170. static OBSEncoderAutoRelease create_video_encoder(DStr &name_buffer, size_t encoder_index,
  171. const GoLiveApi::VideoEncoderConfiguration &encoder_config)
  172. {
  173. auto encoder_type = encoder_config.type.c_str();
  174. if (!encoder_available(encoder_type)) {
  175. blog(LOG_ERROR, "Encoder type '%s' not available", encoder_type);
  176. throw MultitrackVideoError::warning(QTStr("FailedToStartStream.EncoderNotAvailable").arg(encoder_type));
  177. }
  178. dstr_printf(name_buffer, "multitrack video video encoder %zu", encoder_index);
  179. OBSDataAutoRelease encoder_settings = obs_data_create_from_json(encoder_config.settings.dump().c_str());
  180. obs_data_set_bool(encoder_settings, "disable_scenecut", true);
  181. OBSEncoderAutoRelease video_encoder =
  182. obs_video_encoder_create(encoder_type, name_buffer, encoder_settings, nullptr);
  183. if (!video_encoder) {
  184. blog(LOG_ERROR, "Failed to create video encoder '%s'", name_buffer->array);
  185. throw MultitrackVideoError::warning(
  186. QTStr("FailedToStartStream.FailedToCreateVideoEncoder").arg(name_buffer->array, encoder_type));
  187. }
  188. obs_encoder_set_video(video_encoder, obs_get_video());
  189. obs_video_info ovi;
  190. if (!obs_get_video_info(&ovi)) {
  191. blog(LOG_WARNING, "Failed to get obs_video_info while creating encoder %zu", encoder_index);
  192. throw MultitrackVideoError::warning(
  193. QTStr("FailedToStartStream.FailedToGetOBSVideoInfo").arg(name_buffer->array, encoder_type));
  194. }
  195. adjust_video_encoder_scaling(ovi, video_encoder, encoder_config, encoder_index);
  196. adjust_encoder_frame_rate_divisor(ovi, video_encoder, encoder_config, encoder_index);
  197. return video_encoder;
  198. }
  199. static OBSEncoderAutoRelease create_audio_encoder(const char *name, const char *audio_encoder_id, obs_data_t *settings,
  200. size_t mixer_idx)
  201. {
  202. OBSEncoderAutoRelease audio_encoder =
  203. obs_audio_encoder_create(audio_encoder_id, name, settings, mixer_idx, nullptr);
  204. if (!audio_encoder) {
  205. blog(LOG_ERROR, "Failed to create audio encoder");
  206. throw MultitrackVideoError::warning(QTStr("FailedToStartStream.FailedToCreateAudioEncoder"));
  207. }
  208. obs_encoder_set_audio(audio_encoder, obs_get_audio());
  209. return audio_encoder;
  210. }
  211. struct OBSOutputs {
  212. OBSOutputAutoRelease output, recording_output;
  213. };
  214. static OBSOutputs SetupOBSOutput(QWidget *parent, const QString &multitrack_video_name,
  215. obs_data_t *dump_stream_to_file_config, const GoLiveApi::Config &go_live_config,
  216. std::vector<OBSEncoderAutoRelease> &audio_encoders,
  217. std::shared_ptr<obs_encoder_group_t> &video_encoder_group,
  218. const char *audio_encoder_id, size_t main_audio_mixer,
  219. std::optional<size_t> vod_track_mixer);
  220. static void SetupSignalHandlers(bool recording, MultitrackVideoOutput *self, obs_output_t *output, OBSSignal &start,
  221. OBSSignal &stop);
  222. void MultitrackVideoOutput::PrepareStreaming(QWidget *parent, const char *service_name, obs_service_t *service,
  223. const std::optional<std::string> &rtmp_url, const QString &stream_key,
  224. const char *audio_encoder_id,
  225. std::optional<uint32_t> maximum_aggregate_bitrate,
  226. std::optional<uint32_t> maximum_video_tracks,
  227. std::optional<std::string> custom_config,
  228. obs_data_t *dump_stream_to_file_config, size_t main_audio_mixer,
  229. std::optional<size_t> vod_track_mixer, std::optional<bool> use_rtmps)
  230. {
  231. {
  232. const std::lock_guard<std::mutex> current_lock{current_mutex};
  233. const std::lock_guard<std::mutex> current_stream_dump_lock{current_stream_dump_mutex};
  234. if (current || current_stream_dump) {
  235. blog(LOG_WARNING, "Tried to prepare multitrack video output while it's already active");
  236. return;
  237. }
  238. }
  239. restart_on_error = false;
  240. std::optional<GoLiveApi::Config> go_live_config;
  241. std::optional<GoLiveApi::Config> custom;
  242. bool is_custom_config = custom_config.has_value();
  243. auto auto_config_url = MultitrackVideoAutoConfigURL(service);
  244. OBSDataAutoRelease service_settings = obs_service_get_settings(service);
  245. auto multitrack_video_name = QTStr("Basic.Settings.Stream.MultitrackVideoLabel");
  246. if (obs_data_has_user_value(service_settings, "multitrack_video_name")) {
  247. multitrack_video_name = obs_data_get_string(service_settings, "multitrack_video_name");
  248. }
  249. auto auto_config_url_data = auto_config_url.toUtf8();
  250. DStr vod_track_info_storage;
  251. if (vod_track_mixer.has_value())
  252. dstr_printf(vod_track_info_storage, "Yes (mixer: %zu)", vod_track_mixer.value());
  253. blog(LOG_INFO,
  254. "Preparing enhanced broadcasting stream for:\n"
  255. " custom config: %s\n"
  256. " config url: %s\n"
  257. " settings:\n"
  258. " service: %s\n"
  259. " max aggregate bitrate: %s (%" PRIu32 ")\n"
  260. " max video tracks: %s (%" PRIu32 ")\n"
  261. " custom rtmp url: %s ('%s')\n"
  262. " vod track: %s",
  263. is_custom_config ? "Yes" : "No", !auto_config_url.isEmpty() ? auto_config_url_data.constData() : "(null)",
  264. service_name, maximum_aggregate_bitrate.has_value() ? "Set" : "Auto",
  265. maximum_aggregate_bitrate.value_or(0), maximum_video_tracks.has_value() ? "Set" : "Auto",
  266. maximum_video_tracks.value_or(0), rtmp_url.has_value() ? "Yes" : "No",
  267. rtmp_url.has_value() ? rtmp_url->c_str() : "",
  268. vod_track_info_storage->array ? vod_track_info_storage->array : "No");
  269. const bool custom_config_only = auto_config_url.isEmpty() && MultitrackVideoDeveloperModeEnabled() &&
  270. custom_config.has_value() &&
  271. strcmp(obs_service_get_id(service), "rtmp_custom") == 0;
  272. if (!custom_config_only) {
  273. auto go_live_post = constructGoLivePost(stream_key, maximum_aggregate_bitrate, maximum_video_tracks,
  274. vod_track_mixer.has_value());
  275. go_live_config = DownloadGoLiveConfig(parent, auto_config_url, go_live_post, multitrack_video_name);
  276. }
  277. if (custom_config.has_value()) {
  278. GoLiveApi::Config parsed_custom;
  279. try {
  280. parsed_custom = nlohmann::json::parse(*custom_config);
  281. } catch (const nlohmann::json::exception &exception) {
  282. blog(LOG_WARNING, "Failed to parse custom config: %s", exception.what());
  283. throw MultitrackVideoError::critical(QTStr("FailedToStartStream.InvalidCustomConfig"));
  284. }
  285. // copy unique ID from go live request
  286. if (go_live_config.has_value()) {
  287. parsed_custom.meta.config_id = go_live_config->meta.config_id;
  288. blog(LOG_INFO, "Using config_id from go live config with custom config: %s",
  289. parsed_custom.meta.config_id.c_str());
  290. }
  291. nlohmann::json custom_data = parsed_custom;
  292. blog(LOG_INFO, "Using custom go live config: %s", custom_data.dump(4).c_str());
  293. custom.emplace(std::move(parsed_custom));
  294. }
  295. if (go_live_config.has_value()) {
  296. blog(LOG_INFO, "Enhanced broadcasting config_id: '%s'", go_live_config->meta.config_id.c_str());
  297. }
  298. if (!go_live_config && !custom) {
  299. blog(LOG_ERROR, "MultitrackVideoOutput: no config set, this should never happen");
  300. throw MultitrackVideoError::warning(QTStr("FailedToStartStream.NoConfig"));
  301. }
  302. const auto &output_config = custom ? *custom : *go_live_config;
  303. const auto &service_config = go_live_config ? *go_live_config : *custom;
  304. std::vector<OBSEncoderAutoRelease> audio_encoders;
  305. std::shared_ptr<obs_encoder_group_t> video_encoder_group;
  306. auto outputs = SetupOBSOutput(parent, multitrack_video_name, dump_stream_to_file_config, output_config,
  307. audio_encoders, video_encoder_group, audio_encoder_id, main_audio_mixer,
  308. vod_track_mixer);
  309. auto output = std::move(outputs.output);
  310. auto recording_output = std::move(outputs.recording_output);
  311. if (!output)
  312. throw MultitrackVideoError::warning(
  313. QTStr("FailedToStartStream.FallbackToDefault").arg(multitrack_video_name));
  314. auto multitrack_video_service = create_service(service_config, rtmp_url, stream_key, use_rtmps);
  315. if (!multitrack_video_service)
  316. throw MultitrackVideoError::warning(
  317. QTStr("FailedToStartStream.FallbackToDefault").arg(multitrack_video_name));
  318. obs_output_set_service(output, multitrack_video_service);
  319. // Register the BPM (Broadcast Performance Metrics) callback
  320. obs_output_add_packet_callback(output, bpm_inject, NULL);
  321. // Set callback to prevent reconnection attempts once the stream key has become invalid
  322. static auto reconnect_cb = [](void *, obs_output_t *, int code) -> bool {
  323. return code != OBS_OUTPUT_INVALID_STREAM;
  324. };
  325. obs_output_set_reconnect_callback(output, reconnect_cb, nullptr);
  326. OBSSignal start_streaming;
  327. OBSSignal stop_streaming;
  328. SetupSignalHandlers(false, this, output, start_streaming, stop_streaming);
  329. if (dump_stream_to_file_config && recording_output) {
  330. OBSSignal start_recording;
  331. OBSSignal stop_recording;
  332. SetupSignalHandlers(true, this, recording_output, start_recording, stop_recording);
  333. decltype(audio_encoders) recording_audio_encoders;
  334. recording_audio_encoders.reserve(audio_encoders.size());
  335. for (auto &encoder : audio_encoders) {
  336. recording_audio_encoders.emplace_back(obs_encoder_get_ref(encoder));
  337. }
  338. {
  339. const std::lock_guard current_stream_dump_lock{current_stream_dump_mutex};
  340. current_stream_dump.emplace(OBSOutputObjects{
  341. std::move(recording_output),
  342. video_encoder_group,
  343. std::move(recording_audio_encoders),
  344. nullptr,
  345. std::move(start_recording),
  346. std::move(stop_recording),
  347. });
  348. }
  349. }
  350. const std::lock_guard current_lock{current_mutex};
  351. current.emplace(OBSOutputObjects{
  352. std::move(output),
  353. video_encoder_group,
  354. std::move(audio_encoders),
  355. std::move(multitrack_video_service),
  356. std::move(start_streaming),
  357. std::move(stop_streaming),
  358. });
  359. }
  360. signal_handler_t *MultitrackVideoOutput::StreamingSignalHandler()
  361. {
  362. const std::lock_guard current_lock{current_mutex};
  363. return current.has_value() ? obs_output_get_signal_handler(current->output_) : nullptr;
  364. }
  365. void MultitrackVideoOutput::StartedStreaming()
  366. {
  367. OBSOutputAutoRelease dump_output;
  368. {
  369. const std::lock_guard current_stream_dump_lock{current_stream_dump_mutex};
  370. if (current_stream_dump && current_stream_dump->output_) {
  371. dump_output = obs_output_get_ref(current_stream_dump->output_);
  372. }
  373. }
  374. if (!dump_output)
  375. return;
  376. auto result = obs_output_start(dump_output);
  377. blog(LOG_INFO, "MultitrackVideoOutput: starting recording%s", result ? "" : " failed");
  378. }
  379. void MultitrackVideoOutput::StopStreaming()
  380. {
  381. restart_on_error = false;
  382. OBSOutputAutoRelease current_output;
  383. {
  384. const std::lock_guard current_lock{current_mutex};
  385. if (current && current->output_)
  386. current_output = obs_output_get_ref(current->output_);
  387. }
  388. if (current_output)
  389. obs_output_stop(current_output);
  390. OBSOutputAutoRelease dump_output;
  391. {
  392. const std::lock_guard current_stream_dump_lock{current_stream_dump_mutex};
  393. if (current_stream_dump && current_stream_dump->output_)
  394. dump_output = obs_output_get_ref(current_stream_dump->output_);
  395. }
  396. if (dump_output)
  397. obs_output_stop(dump_output);
  398. }
  399. bool MultitrackVideoOutput::HandleIncompatibleSettings(QWidget *parent, config_t *config, obs_service_t *service,
  400. bool &useDelay, bool &enableNewSocketLoop,
  401. bool &enableDynBitrate)
  402. {
  403. QString incompatible_settings;
  404. QString where_to_disable;
  405. QString incompatible_settings_list;
  406. size_t num = 1;
  407. auto check_setting = [&](bool setting, const char *name, const char *section) {
  408. if (!setting)
  409. return;
  410. incompatible_settings += QString(" %1. %2\n").arg(num).arg(QTStr(name));
  411. where_to_disable += QString(" %1. [%2 → %3 → %4]\n")
  412. .arg(num)
  413. .arg(QTStr("Settings"))
  414. .arg(QTStr("Basic.Settings.Advanced"))
  415. .arg(QTStr(section));
  416. incompatible_settings_list += QString("%1, ").arg(name);
  417. num += 1;
  418. };
  419. check_setting(useDelay, "Basic.Settings.Advanced.StreamDelay", "Basic.Settings.Advanced.StreamDelay");
  420. #ifdef _WIN32
  421. check_setting(enableNewSocketLoop, "Basic.Settings.Advanced.Network.EnableNewSocketLoop",
  422. "Basic.Settings.Advanced.Network");
  423. #endif
  424. check_setting(enableDynBitrate, "Basic.Settings.Output.DynamicBitrate.Beta", "Basic.Settings.Advanced.Network");
  425. if (incompatible_settings.isEmpty())
  426. return true;
  427. OBSDataAutoRelease service_settings = obs_service_get_settings(service);
  428. QMessageBox mb(parent);
  429. mb.setIcon(QMessageBox::Critical);
  430. mb.setWindowTitle(QTStr("MultitrackVideo.IncompatibleSettings.Title"));
  431. mb.setText(QString(QTStr("MultitrackVideo.IncompatibleSettings.Text"))
  432. .arg(obs_data_get_string(service_settings, "multitrack_video_name"))
  433. .arg(incompatible_settings)
  434. .arg(where_to_disable));
  435. auto this_stream = mb.addButton(QTStr("MultitrackVideo.IncompatibleSettings.DisableAndStartStreaming"),
  436. QMessageBox::AcceptRole);
  437. auto all_streams = mb.addButton(QString(QTStr("MultitrackVideo.IncompatibleSettings.UpdateAndStartStreaming")),
  438. QMessageBox::AcceptRole);
  439. mb.setStandardButtons(QMessageBox::StandardButton::Cancel);
  440. mb.exec();
  441. const char *action = "cancel";
  442. if (mb.clickedButton() == this_stream) {
  443. action = "DisableAndStartStreaming";
  444. } else if (mb.clickedButton() == all_streams) {
  445. action = "UpdateAndStartStreaming";
  446. }
  447. blog(LOG_INFO,
  448. "MultitrackVideoOutput: attempted to start stream with incompatible"
  449. "settings (%s); action taken: %s",
  450. incompatible_settings_list.toUtf8().constData(), action);
  451. if (mb.clickedButton() == this_stream || mb.clickedButton() == all_streams) {
  452. useDelay = false;
  453. enableNewSocketLoop = false;
  454. enableDynBitrate = false;
  455. if (mb.clickedButton() == all_streams) {
  456. config_set_bool(config, "Output", "DelayEnable", false);
  457. #ifdef _WIN32
  458. config_set_bool(config, "Output", "NewSocketLoopEnable", false);
  459. #endif
  460. config_set_bool(config, "Output", "DynamicBitrate", false);
  461. }
  462. return true;
  463. }
  464. MultitrackVideoOutput::ReleaseOnMainThread(take_current());
  465. MultitrackVideoOutput::ReleaseOnMainThread(take_current_stream_dump());
  466. return false;
  467. }
  468. static bool create_video_encoders(const GoLiveApi::Config &go_live_config,
  469. std::shared_ptr<obs_encoder_group_t> &video_encoder_group, obs_output_t *output,
  470. obs_output_t *recording_output)
  471. {
  472. DStr video_encoder_name_buffer;
  473. if (go_live_config.encoder_configurations.empty()) {
  474. blog(LOG_WARNING, "MultitrackVideoOutput: Missing video encoder configurations");
  475. throw MultitrackVideoError::warning(QTStr("FailedToStartStream.MissingEncoderConfigs"));
  476. }
  477. std::shared_ptr<obs_encoder_group_t> encoder_group(obs_encoder_group_create(), obs_encoder_group_destroy);
  478. if (!encoder_group)
  479. return false;
  480. for (size_t i = 0; i < go_live_config.encoder_configurations.size(); i++) {
  481. auto encoder =
  482. create_video_encoder(video_encoder_name_buffer, i, go_live_config.encoder_configurations[i]);
  483. if (!encoder)
  484. return false;
  485. if (!obs_encoder_set_group(encoder, encoder_group.get()))
  486. return false;
  487. obs_output_set_video_encoder2(output, encoder, i);
  488. if (recording_output)
  489. obs_output_set_video_encoder2(recording_output, encoder, i);
  490. }
  491. video_encoder_group = encoder_group;
  492. return true;
  493. }
  494. static void create_audio_encoders(const GoLiveApi::Config &go_live_config,
  495. std::vector<OBSEncoderAutoRelease> &audio_encoders, obs_output_t *output,
  496. obs_output_t *recording_output, const char *audio_encoder_id, size_t main_audio_mixer,
  497. std::optional<size_t> vod_track_mixer, std::vector<speaker_layout> &speaker_layouts,
  498. speaker_layout &current_layout)
  499. {
  500. speaker_layout speakers = SPEAKERS_UNKNOWN;
  501. obs_audio_info oai = {};
  502. if (obs_get_audio_info(&oai))
  503. speakers = oai.speakers;
  504. current_layout = speakers;
  505. auto sanitize_audio_channels = [&](obs_encoder_t *encoder, uint32_t channels) {
  506. speaker_layout target_speakers = SPEAKERS_UNKNOWN;
  507. for (size_t i = 0; i <= (size_t)SPEAKERS_7POINT1; i++) {
  508. if (get_audio_channels((speaker_layout)i) != channels)
  509. continue;
  510. target_speakers = (speaker_layout)i;
  511. break;
  512. }
  513. if (target_speakers == SPEAKERS_UNKNOWN) {
  514. blog(LOG_WARNING,
  515. "MultitrackVideoOutput: Could not find "
  516. "speaker layout for %" PRIu32 "channels "
  517. "while configuring encoder '%s'",
  518. channels, obs_encoder_get_name(encoder));
  519. return;
  520. }
  521. if (speakers != SPEAKERS_UNKNOWN &&
  522. (channels > get_audio_channels(speakers) || speakers == target_speakers))
  523. return;
  524. auto it = std::find(std::begin(speaker_layouts), std::end(speaker_layouts), target_speakers);
  525. if (it == std::end(speaker_layouts))
  526. speaker_layouts.push_back(target_speakers);
  527. };
  528. using encoder_configs_type = decltype(go_live_config.audio_configurations.live);
  529. DStr encoder_name_buffer;
  530. size_t output_encoder_index = 0;
  531. auto create_encoders = [&](const char *name_prefix, const encoder_configs_type &configs, size_t mixer_idx) {
  532. if (configs.empty()) {
  533. blog(LOG_WARNING, "MultitrackVideoOutput: Missing audio encoder configurations (for '%s')",
  534. name_prefix);
  535. throw MultitrackVideoError::warning(QTStr("FailedToStartStream.MissingEncoderConfigs"));
  536. }
  537. for (size_t i = 0; i < configs.size(); i++) {
  538. dstr_printf(encoder_name_buffer, "%s %zu", name_prefix, i);
  539. OBSDataAutoRelease settings = obs_data_create_from_json(configs[i].settings.dump().c_str());
  540. OBSEncoderAutoRelease audio_encoder =
  541. create_audio_encoder(encoder_name_buffer->array, audio_encoder_id, settings, mixer_idx);
  542. sanitize_audio_channels(audio_encoder, configs[i].channels);
  543. obs_output_set_audio_encoder(output, audio_encoder, output_encoder_index);
  544. if (recording_output)
  545. obs_output_set_audio_encoder(recording_output, audio_encoder, output_encoder_index);
  546. output_encoder_index += 1;
  547. audio_encoders.emplace_back(std::move(audio_encoder));
  548. }
  549. };
  550. create_encoders("multitrack video live audio", go_live_config.audio_configurations.live, main_audio_mixer);
  551. if (!vod_track_mixer.has_value())
  552. return;
  553. // we already check for empty inside of `create_encoders`
  554. encoder_configs_type empty = {};
  555. create_encoders("multitrack video vod audio", go_live_config.audio_configurations.vod.value_or(empty),
  556. *vod_track_mixer);
  557. return;
  558. }
  559. static const char *speaker_layout_to_string(speaker_layout layout)
  560. {
  561. switch (layout) {
  562. case SPEAKERS_MONO:
  563. return "Mono";
  564. case SPEAKERS_2POINT1:
  565. return "2.1";
  566. case SPEAKERS_4POINT0:
  567. return "4.0";
  568. case SPEAKERS_4POINT1:
  569. return "4.1";
  570. case SPEAKERS_5POINT1:
  571. return "5.1";
  572. case SPEAKERS_7POINT1:
  573. return "7.1";
  574. case SPEAKERS_UNKNOWN:
  575. case SPEAKERS_STEREO:
  576. return "Stereo";
  577. }
  578. return "Stereo";
  579. }
  580. static void handle_speaker_layout_issues(QWidget *parent, const QString &multitrack_video_name,
  581. const std::vector<speaker_layout> &requested_layouts, speaker_layout layout)
  582. {
  583. if (requested_layouts.empty())
  584. return;
  585. QString message;
  586. if (requested_layouts.size() == 1) {
  587. message = QTStr("MultitrackVideo.IncompatibleSettings.AudioChannelsSingle")
  588. .arg(QTStr(speaker_layout_to_string(requested_layouts.front())));
  589. } else {
  590. message =
  591. QTStr("MultitrackVideo.IncompatibleSettings.AudioChannelsMultiple").arg(multitrack_video_name);
  592. }
  593. QMetaObject::invokeMethod(
  594. parent,
  595. [&] {
  596. QMessageBox mb(parent);
  597. mb.setIcon(QMessageBox::Critical);
  598. mb.setWindowTitle(QTStr("MultitrackVideo.IncompatibleSettings.Title"));
  599. mb.setText(QTStr("MultitrackVideo.IncompatibleSettings.AudioChannels")
  600. .arg(multitrack_video_name)
  601. .arg(QTStr(speaker_layout_to_string(layout)))
  602. .arg(message));
  603. mb.setStandardButtons(QMessageBox::StandardButton::Cancel);
  604. mb.exec();
  605. },
  606. BlockingConnectionTypeFor(parent));
  607. blog(LOG_INFO, "MultitrackVideoOutput: Attempted to start stream with incompatible "
  608. "audio channel setting. Action taken: cancel");
  609. throw MultitrackVideoError::cancel();
  610. }
  611. static OBSOutputs SetupOBSOutput(QWidget *parent, const QString &multitrack_video_name,
  612. obs_data_t *dump_stream_to_file_config, const GoLiveApi::Config &go_live_config,
  613. std::vector<OBSEncoderAutoRelease> &audio_encoders,
  614. std::shared_ptr<obs_encoder_group_t> &video_encoder_group,
  615. const char *audio_encoder_id, size_t main_audio_mixer,
  616. std::optional<size_t> vod_track_mixer)
  617. {
  618. auto output = create_output();
  619. OBSOutputAutoRelease recording_output;
  620. if (dump_stream_to_file_config)
  621. recording_output = create_recording_output(dump_stream_to_file_config);
  622. if (!create_video_encoders(go_live_config, video_encoder_group, output, recording_output))
  623. return {nullptr, nullptr};
  624. std::vector<speaker_layout> requested_speaker_layouts;
  625. speaker_layout current_layout = SPEAKERS_UNKNOWN;
  626. create_audio_encoders(go_live_config, audio_encoders, output, recording_output, audio_encoder_id,
  627. main_audio_mixer, vod_track_mixer, requested_speaker_layouts, current_layout);
  628. handle_speaker_layout_issues(parent, multitrack_video_name, requested_speaker_layouts, current_layout);
  629. return {std::move(output), std::move(recording_output)};
  630. }
  631. void SetupSignalHandlers(bool recording, MultitrackVideoOutput *self, obs_output_t *output, OBSSignal &start,
  632. OBSSignal &stop)
  633. {
  634. auto handler = obs_output_get_signal_handler(output);
  635. start.Connect(handler, "start", !recording ? StreamStartHandler : RecordingStartHandler, self);
  636. stop.Connect(handler, "stop", !recording ? StreamStopHandler : RecordingStopHandler, self);
  637. }
  638. std::optional<MultitrackVideoOutput::OBSOutputObjects> MultitrackVideoOutput::take_current()
  639. {
  640. const std::lock_guard<std::mutex> current_lock{current_mutex};
  641. auto val = std::move(current);
  642. current.reset();
  643. return val;
  644. }
  645. std::optional<MultitrackVideoOutput::OBSOutputObjects> MultitrackVideoOutput::take_current_stream_dump()
  646. {
  647. const std::lock_guard<std::mutex> current_stream_dump_lock{current_stream_dump_mutex};
  648. auto val = std::move(current_stream_dump);
  649. current_stream_dump.reset();
  650. return val;
  651. }
  652. void MultitrackVideoOutput::ReleaseOnMainThread(std::optional<OBSOutputObjects> objects)
  653. {
  654. if (!objects.has_value())
  655. return;
  656. QMetaObject::invokeMethod(
  657. QApplication::instance()->thread(), [objects = std::move(objects)] {}, Qt::QueuedConnection);
  658. }
  659. void StreamStartHandler(void *arg, calldata_t *)
  660. {
  661. auto self = static_cast<MultitrackVideoOutput *>(arg);
  662. self->restart_on_error = true;
  663. }
  664. void StreamStopHandler(void *arg, calldata_t *data)
  665. {
  666. auto self = static_cast<MultitrackVideoOutput *>(arg);
  667. OBSOutputAutoRelease stream_dump_output;
  668. {
  669. const std::lock_guard<std::mutex> current_stream_dump_lock{self->current_stream_dump_mutex};
  670. if (self->current_stream_dump && self->current_stream_dump->output_)
  671. stream_dump_output = obs_output_get_ref(self->current_stream_dump->output_);
  672. }
  673. if (stream_dump_output)
  674. obs_output_stop(stream_dump_output);
  675. /* Unregister the BPM (Broadcast Performance Metrics) callback and destroy the allocated metrics data. */
  676. obs_output_remove_packet_callback(static_cast<obs_output_t *>(calldata_ptr(data, "output")), bpm_inject, NULL);
  677. bpm_destroy(static_cast<obs_output_t *>(calldata_ptr(data, "output")));
  678. MultitrackVideoOutput::ReleaseOnMainThread(self->take_current());
  679. }
  680. void RecordingStartHandler(void * /* arg */, calldata_t * /* data */)
  681. {
  682. blog(LOG_INFO, "MultitrackVideoOutput: recording started");
  683. }
  684. void RecordingStopHandler(void *arg, calldata_t *)
  685. {
  686. auto self = static_cast<MultitrackVideoOutput *>(arg);
  687. blog(LOG_INFO, "MultitrackVideoOutput: recording stopped");
  688. MultitrackVideoOutput::ReleaseOnMainThread(self->take_current_stream_dump());
  689. }