MultitrackVideoOutput.cpp 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864
  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.output_width == requested_width || ovi.output_height == requested_height)
  127. return;
  128. if (ovi.base_width < requested_width || ovi.base_height < requested_height) {
  129. blog(LOG_WARNING,
  130. "Requested resolution exceeds canvas/available resolution for encoder %zu: %" PRIu32 "x%" PRIu32
  131. " > %" PRIu32 "x%" PRIu32,
  132. encoder_index, requested_width, requested_height, ovi.base_width, ovi.base_height);
  133. }
  134. obs_encoder_set_scaled_size(video_encoder, requested_width, requested_height);
  135. obs_encoder_set_gpu_scale_type(video_encoder, encoder_config.gpu_scale_type.value_or(OBS_SCALE_BICUBIC));
  136. obs_encoder_set_preferred_video_format(video_encoder, VIDEO_FORMAT_NV12);
  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, OBSSignal &deactivate);
  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. std::optional<GoLiveApi::Config> go_live_config;
  240. std::optional<GoLiveApi::Config> custom;
  241. bool is_custom_config = custom_config.has_value();
  242. auto auto_config_url = MultitrackVideoAutoConfigURL(service);
  243. OBSDataAutoRelease service_settings = obs_service_get_settings(service);
  244. auto multitrack_video_name = QTStr("Basic.Settings.Stream.MultitrackVideoLabel");
  245. if (obs_data_has_user_value(service_settings, "multitrack_video_name")) {
  246. multitrack_video_name = obs_data_get_string(service_settings, "multitrack_video_name");
  247. }
  248. auto auto_config_url_data = auto_config_url.toUtf8();
  249. DStr vod_track_info_storage;
  250. if (vod_track_mixer.has_value())
  251. dstr_printf(vod_track_info_storage, "Yes (mixer: %zu)", vod_track_mixer.value());
  252. blog(LOG_INFO,
  253. "Preparing enhanced broadcasting stream for:\n"
  254. " custom config: %s\n"
  255. " config url: %s\n"
  256. " settings:\n"
  257. " service: %s\n"
  258. " max aggregate bitrate: %s (%" PRIu32 ")\n"
  259. " max video tracks: %s (%" PRIu32 ")\n"
  260. " custom rtmp url: %s ('%s')\n"
  261. " vod track: %s",
  262. is_custom_config ? "Yes" : "No", !auto_config_url.isEmpty() ? auto_config_url_data.constData() : "(null)",
  263. service_name, maximum_aggregate_bitrate.has_value() ? "Set" : "Auto",
  264. maximum_aggregate_bitrate.value_or(0), maximum_video_tracks.has_value() ? "Set" : "Auto",
  265. maximum_video_tracks.value_or(0), rtmp_url.has_value() ? "Yes" : "No",
  266. rtmp_url.has_value() ? rtmp_url->c_str() : "",
  267. vod_track_info_storage->array ? vod_track_info_storage->array : "No");
  268. const bool custom_config_only = auto_config_url.isEmpty() && MultitrackVideoDeveloperModeEnabled() &&
  269. custom_config.has_value() &&
  270. strcmp(obs_service_get_id(service), "rtmp_custom") == 0;
  271. if (!custom_config_only) {
  272. auto go_live_post = constructGoLivePost(stream_key, maximum_aggregate_bitrate, maximum_video_tracks,
  273. vod_track_mixer.has_value());
  274. go_live_config = DownloadGoLiveConfig(parent, auto_config_url, go_live_post, multitrack_video_name);
  275. }
  276. if (custom_config.has_value()) {
  277. GoLiveApi::Config parsed_custom;
  278. try {
  279. parsed_custom = nlohmann::json::parse(*custom_config);
  280. } catch (const nlohmann::json::exception &exception) {
  281. blog(LOG_WARNING, "Failed to parse custom config: %s", exception.what());
  282. throw MultitrackVideoError::critical(QTStr("FailedToStartStream.InvalidCustomConfig"));
  283. }
  284. // copy unique ID from go live request
  285. if (go_live_config.has_value()) {
  286. parsed_custom.meta.config_id = go_live_config->meta.config_id;
  287. blog(LOG_INFO, "Using config_id from go live config with custom config: %s",
  288. parsed_custom.meta.config_id.c_str());
  289. }
  290. nlohmann::json custom_data = parsed_custom;
  291. blog(LOG_INFO, "Using custom go live config: %s", custom_data.dump(4).c_str());
  292. custom.emplace(std::move(parsed_custom));
  293. }
  294. if (go_live_config.has_value()) {
  295. blog(LOG_INFO, "Enhanced broadcasting config_id: '%s'", go_live_config->meta.config_id.c_str());
  296. }
  297. if (!go_live_config && !custom) {
  298. blog(LOG_ERROR, "MultitrackVideoOutput: no config set, this should never happen");
  299. throw MultitrackVideoError::warning(QTStr("FailedToStartStream.NoConfig"));
  300. }
  301. const auto &output_config = custom ? *custom : *go_live_config;
  302. const auto &service_config = go_live_config ? *go_live_config : *custom;
  303. std::vector<OBSEncoderAutoRelease> audio_encoders;
  304. std::shared_ptr<obs_encoder_group_t> video_encoder_group;
  305. auto outputs = SetupOBSOutput(parent, multitrack_video_name, dump_stream_to_file_config, output_config,
  306. audio_encoders, video_encoder_group, audio_encoder_id, main_audio_mixer,
  307. vod_track_mixer);
  308. auto output = std::move(outputs.output);
  309. auto recording_output = std::move(outputs.recording_output);
  310. if (!output)
  311. throw MultitrackVideoError::warning(
  312. QTStr("FailedToStartStream.FallbackToDefault").arg(multitrack_video_name));
  313. auto multitrack_video_service = create_service(service_config, rtmp_url, stream_key, use_rtmps);
  314. if (!multitrack_video_service)
  315. throw MultitrackVideoError::warning(
  316. QTStr("FailedToStartStream.FallbackToDefault").arg(multitrack_video_name));
  317. obs_output_set_service(output, multitrack_video_service);
  318. // Register the BPM (Broadcast Performance Metrics) callback
  319. obs_output_add_packet_callback(output, bpm_inject, NULL);
  320. OBSSignal start_streaming;
  321. OBSSignal stop_streaming;
  322. OBSSignal deactivate_stream;
  323. SetupSignalHandlers(false, this, output, start_streaming, stop_streaming, deactivate_stream);
  324. if (dump_stream_to_file_config && recording_output) {
  325. OBSSignal start_recording;
  326. OBSSignal stop_recording;
  327. OBSSignal deactivate_recording;
  328. SetupSignalHandlers(true, this, recording_output, start_recording, stop_recording,
  329. deactivate_recording);
  330. decltype(audio_encoders) recording_audio_encoders;
  331. recording_audio_encoders.reserve(audio_encoders.size());
  332. for (auto &encoder : audio_encoders) {
  333. recording_audio_encoders.emplace_back(obs_encoder_get_ref(encoder));
  334. }
  335. {
  336. const std::lock_guard current_stream_dump_lock{current_stream_dump_mutex};
  337. current_stream_dump.emplace(OBSOutputObjects{
  338. std::move(recording_output),
  339. video_encoder_group,
  340. std::move(recording_audio_encoders),
  341. nullptr,
  342. std::move(start_recording),
  343. std::move(stop_recording),
  344. std::move(deactivate_recording),
  345. });
  346. }
  347. }
  348. const std::lock_guard current_lock{current_mutex};
  349. current.emplace(OBSOutputObjects{
  350. std::move(output),
  351. video_encoder_group,
  352. std::move(audio_encoders),
  353. std::move(multitrack_video_service),
  354. std::move(start_streaming),
  355. std::move(stop_streaming),
  356. std::move(deactivate_stream),
  357. });
  358. }
  359. signal_handler_t *MultitrackVideoOutput::StreamingSignalHandler()
  360. {
  361. const std::lock_guard current_lock{current_mutex};
  362. return current.has_value() ? obs_output_get_signal_handler(current->output_) : nullptr;
  363. }
  364. void MultitrackVideoOutput::StartedStreaming()
  365. {
  366. OBSOutputAutoRelease dump_output;
  367. {
  368. const std::lock_guard current_stream_dump_lock{current_stream_dump_mutex};
  369. if (current_stream_dump && current_stream_dump->output_) {
  370. dump_output = obs_output_get_ref(current_stream_dump->output_);
  371. }
  372. }
  373. if (!dump_output)
  374. return;
  375. auto result = obs_output_start(dump_output);
  376. blog(LOG_INFO, "MultitrackVideoOutput: starting recording%s", result ? "" : " failed");
  377. }
  378. void MultitrackVideoOutput::StopStreaming()
  379. {
  380. OBSOutputAutoRelease current_output;
  381. {
  382. const std::lock_guard current_lock{current_mutex};
  383. if (current && current->output_)
  384. current_output = obs_output_get_ref(current->output_);
  385. }
  386. if (current_output)
  387. obs_output_stop(current_output);
  388. OBSOutputAutoRelease dump_output;
  389. {
  390. const std::lock_guard current_stream_dump_lock{current_stream_dump_mutex};
  391. if (current_stream_dump && current_stream_dump->output_)
  392. dump_output = obs_output_get_ref(current_stream_dump->output_);
  393. }
  394. if (dump_output)
  395. obs_output_stop(dump_output);
  396. }
  397. bool MultitrackVideoOutput::HandleIncompatibleSettings(QWidget *parent, config_t *config, obs_service_t *service,
  398. bool &useDelay, bool &enableNewSocketLoop,
  399. bool &enableDynBitrate)
  400. {
  401. QString incompatible_settings;
  402. QString where_to_disable;
  403. QString incompatible_settings_list;
  404. size_t num = 1;
  405. auto check_setting = [&](bool setting, const char *name, const char *section) {
  406. if (!setting)
  407. return;
  408. incompatible_settings += QString(" %1. %2\n").arg(num).arg(QTStr(name));
  409. where_to_disable += QString(" %1. [%2 → %3 → %4]\n")
  410. .arg(num)
  411. .arg(QTStr("Settings"))
  412. .arg(QTStr("Basic.Settings.Advanced"))
  413. .arg(QTStr(section));
  414. incompatible_settings_list += QString("%1, ").arg(name);
  415. num += 1;
  416. };
  417. check_setting(useDelay, "Basic.Settings.Advanced.StreamDelay", "Basic.Settings.Advanced.StreamDelay");
  418. #ifdef _WIN32
  419. check_setting(enableNewSocketLoop, "Basic.Settings.Advanced.Network.EnableNewSocketLoop",
  420. "Basic.Settings.Advanced.Network");
  421. #endif
  422. check_setting(enableDynBitrate, "Basic.Settings.Output.DynamicBitrate.Beta", "Basic.Settings.Advanced.Network");
  423. if (incompatible_settings.isEmpty())
  424. return true;
  425. OBSDataAutoRelease service_settings = obs_service_get_settings(service);
  426. QMessageBox mb(parent);
  427. mb.setIcon(QMessageBox::Critical);
  428. mb.setWindowTitle(QTStr("MultitrackVideo.IncompatibleSettings.Title"));
  429. mb.setText(QString(QTStr("MultitrackVideo.IncompatibleSettings.Text"))
  430. .arg(obs_data_get_string(service_settings, "multitrack_video_name"))
  431. .arg(incompatible_settings)
  432. .arg(where_to_disable));
  433. auto this_stream = mb.addButton(QTStr("MultitrackVideo.IncompatibleSettings.DisableAndStartStreaming"),
  434. QMessageBox::AcceptRole);
  435. auto all_streams = mb.addButton(QString(QTStr("MultitrackVideo.IncompatibleSettings.UpdateAndStartStreaming")),
  436. QMessageBox::AcceptRole);
  437. mb.setStandardButtons(QMessageBox::StandardButton::Cancel);
  438. mb.exec();
  439. const char *action = "cancel";
  440. if (mb.clickedButton() == this_stream) {
  441. action = "DisableAndStartStreaming";
  442. } else if (mb.clickedButton() == all_streams) {
  443. action = "UpdateAndStartStreaming";
  444. }
  445. blog(LOG_INFO,
  446. "MultitrackVideoOutput: attempted to start stream with incompatible"
  447. "settings (%s); action taken: %s",
  448. incompatible_settings_list.toUtf8().constData(), action);
  449. if (mb.clickedButton() == this_stream || mb.clickedButton() == all_streams) {
  450. useDelay = false;
  451. enableNewSocketLoop = false;
  452. enableDynBitrate = false;
  453. if (mb.clickedButton() == all_streams) {
  454. config_set_bool(config, "Output", "DelayEnable", false);
  455. #ifdef _WIN32
  456. config_set_bool(config, "Output", "NewSocketLoopEnable", false);
  457. #endif
  458. config_set_bool(config, "Output", "DynamicBitrate", false);
  459. }
  460. return true;
  461. }
  462. MultitrackVideoOutput::ReleaseOnMainThread(take_current());
  463. MultitrackVideoOutput::ReleaseOnMainThread(take_current_stream_dump());
  464. return false;
  465. }
  466. static bool create_video_encoders(const GoLiveApi::Config &go_live_config,
  467. std::shared_ptr<obs_encoder_group_t> &video_encoder_group, obs_output_t *output,
  468. obs_output_t *recording_output)
  469. {
  470. DStr video_encoder_name_buffer;
  471. if (go_live_config.encoder_configurations.empty()) {
  472. blog(LOG_WARNING, "MultitrackVideoOutput: Missing video encoder configurations");
  473. throw MultitrackVideoError::warning(QTStr("FailedToStartStream.MissingEncoderConfigs"));
  474. }
  475. std::shared_ptr<obs_encoder_group_t> encoder_group(obs_encoder_group_create(), obs_encoder_group_destroy);
  476. if (!encoder_group)
  477. return false;
  478. for (size_t i = 0; i < go_live_config.encoder_configurations.size(); i++) {
  479. auto encoder =
  480. create_video_encoder(video_encoder_name_buffer, i, go_live_config.encoder_configurations[i]);
  481. if (!encoder)
  482. return false;
  483. if (!obs_encoder_set_group(encoder, encoder_group.get()))
  484. return false;
  485. obs_output_set_video_encoder2(output, encoder, i);
  486. if (recording_output)
  487. obs_output_set_video_encoder2(recording_output, encoder, i);
  488. }
  489. video_encoder_group = encoder_group;
  490. return true;
  491. }
  492. static void create_audio_encoders(const GoLiveApi::Config &go_live_config,
  493. std::vector<OBSEncoderAutoRelease> &audio_encoders, obs_output_t *output,
  494. obs_output_t *recording_output, const char *audio_encoder_id, size_t main_audio_mixer,
  495. std::optional<size_t> vod_track_mixer, std::vector<speaker_layout> &speaker_layouts,
  496. speaker_layout &current_layout)
  497. {
  498. speaker_layout speakers = SPEAKERS_UNKNOWN;
  499. obs_audio_info oai = {};
  500. if (obs_get_audio_info(&oai))
  501. speakers = oai.speakers;
  502. current_layout = speakers;
  503. auto sanitize_audio_channels = [&](obs_encoder_t *encoder, uint32_t channels) {
  504. speaker_layout target_speakers = SPEAKERS_UNKNOWN;
  505. for (size_t i = 0; i <= (size_t)SPEAKERS_7POINT1; i++) {
  506. if (get_audio_channels((speaker_layout)i) != channels)
  507. continue;
  508. target_speakers = (speaker_layout)i;
  509. break;
  510. }
  511. if (target_speakers == SPEAKERS_UNKNOWN) {
  512. blog(LOG_WARNING,
  513. "MultitrackVideoOutput: Could not find "
  514. "speaker layout for %" PRIu32 "channels "
  515. "while configuring encoder '%s'",
  516. channels, obs_encoder_get_name(encoder));
  517. return;
  518. }
  519. if (speakers != SPEAKERS_UNKNOWN &&
  520. (channels > get_audio_channels(speakers) || speakers == target_speakers))
  521. return;
  522. auto it = std::find(std::begin(speaker_layouts), std::end(speaker_layouts), target_speakers);
  523. if (it == std::end(speaker_layouts))
  524. speaker_layouts.push_back(target_speakers);
  525. };
  526. using encoder_configs_type = decltype(go_live_config.audio_configurations.live);
  527. DStr encoder_name_buffer;
  528. size_t output_encoder_index = 0;
  529. auto create_encoders = [&](const char *name_prefix, const encoder_configs_type &configs, size_t mixer_idx) {
  530. if (configs.empty()) {
  531. blog(LOG_WARNING, "MultitrackVideoOutput: Missing audio encoder configurations (for '%s')",
  532. name_prefix);
  533. throw MultitrackVideoError::warning(QTStr("FailedToStartStream.MissingEncoderConfigs"));
  534. }
  535. for (size_t i = 0; i < configs.size(); i++) {
  536. dstr_printf(encoder_name_buffer, "%s %zu", name_prefix, i);
  537. OBSDataAutoRelease settings = obs_data_create_from_json(configs[i].settings.dump().c_str());
  538. OBSEncoderAutoRelease audio_encoder =
  539. create_audio_encoder(encoder_name_buffer->array, audio_encoder_id, settings, mixer_idx);
  540. sanitize_audio_channels(audio_encoder, configs[i].channels);
  541. obs_output_set_audio_encoder(output, audio_encoder, output_encoder_index);
  542. if (recording_output)
  543. obs_output_set_audio_encoder(recording_output, audio_encoder, output_encoder_index);
  544. output_encoder_index += 1;
  545. audio_encoders.emplace_back(std::move(audio_encoder));
  546. }
  547. };
  548. create_encoders("multitrack video live audio", go_live_config.audio_configurations.live, main_audio_mixer);
  549. if (!vod_track_mixer.has_value())
  550. return;
  551. // we already check for empty inside of `create_encoders`
  552. encoder_configs_type empty = {};
  553. create_encoders("multitrack video vod audio", go_live_config.audio_configurations.vod.value_or(empty),
  554. *vod_track_mixer);
  555. return;
  556. }
  557. static const char *speaker_layout_to_string(speaker_layout layout)
  558. {
  559. switch (layout) {
  560. case SPEAKERS_MONO:
  561. return "Mono";
  562. case SPEAKERS_2POINT1:
  563. return "2.1";
  564. case SPEAKERS_4POINT0:
  565. return "4.0";
  566. case SPEAKERS_4POINT1:
  567. return "4.1";
  568. case SPEAKERS_5POINT1:
  569. return "5.1";
  570. case SPEAKERS_7POINT1:
  571. return "7.1";
  572. case SPEAKERS_UNKNOWN:
  573. case SPEAKERS_STEREO:
  574. return "Stereo";
  575. }
  576. return "Stereo";
  577. }
  578. static void handle_speaker_layout_issues(QWidget *parent, const QString &multitrack_video_name,
  579. const std::vector<speaker_layout> &requested_layouts, speaker_layout layout)
  580. {
  581. if (requested_layouts.empty())
  582. return;
  583. QString message;
  584. if (requested_layouts.size() == 1) {
  585. message = QTStr("MultitrackVideo.IncompatibleSettings.AudioChannelsSingle")
  586. .arg(QTStr(speaker_layout_to_string(requested_layouts.front())));
  587. } else {
  588. message =
  589. QTStr("MultitrackVideo.IncompatibleSettings.AudioChannelsMultiple").arg(multitrack_video_name);
  590. }
  591. QMetaObject::invokeMethod(
  592. parent,
  593. [&] {
  594. QMessageBox mb(parent);
  595. mb.setIcon(QMessageBox::Critical);
  596. mb.setWindowTitle(QTStr("MultitrackVideo.IncompatibleSettings.Title"));
  597. mb.setText(QTStr("MultitrackVideo.IncompatibleSettings.AudioChannels")
  598. .arg(multitrack_video_name)
  599. .arg(QTStr(speaker_layout_to_string(layout)))
  600. .arg(message));
  601. mb.setStandardButtons(QMessageBox::StandardButton::Cancel);
  602. mb.exec();
  603. },
  604. BlockingConnectionTypeFor(parent));
  605. blog(LOG_INFO, "MultitrackVideoOutput: Attempted to start stream with incompatible "
  606. "audio channel setting. Action taken: cancel");
  607. throw MultitrackVideoError::cancel();
  608. }
  609. static OBSOutputs SetupOBSOutput(QWidget *parent, const QString &multitrack_video_name,
  610. obs_data_t *dump_stream_to_file_config, const GoLiveApi::Config &go_live_config,
  611. std::vector<OBSEncoderAutoRelease> &audio_encoders,
  612. std::shared_ptr<obs_encoder_group_t> &video_encoder_group,
  613. const char *audio_encoder_id, size_t main_audio_mixer,
  614. std::optional<size_t> vod_track_mixer)
  615. {
  616. auto output = create_output();
  617. OBSOutputAutoRelease recording_output;
  618. if (dump_stream_to_file_config)
  619. recording_output = create_recording_output(dump_stream_to_file_config);
  620. if (!create_video_encoders(go_live_config, video_encoder_group, output, recording_output))
  621. return {nullptr, nullptr};
  622. std::vector<speaker_layout> requested_speaker_layouts;
  623. speaker_layout current_layout = SPEAKERS_UNKNOWN;
  624. create_audio_encoders(go_live_config, audio_encoders, output, recording_output, audio_encoder_id,
  625. main_audio_mixer, vod_track_mixer, requested_speaker_layouts, current_layout);
  626. handle_speaker_layout_issues(parent, multitrack_video_name, requested_speaker_layouts, current_layout);
  627. return {std::move(output), std::move(recording_output)};
  628. }
  629. void SetupSignalHandlers(bool recording, MultitrackVideoOutput *self, obs_output_t *output, OBSSignal &start,
  630. OBSSignal &stop, OBSSignal &deactivate)
  631. {
  632. auto handler = obs_output_get_signal_handler(output);
  633. if (recording)
  634. start.Connect(handler, "start", RecordingStartHandler, self);
  635. stop.Connect(handler, "stop", !recording ? StreamStopHandler : RecordingStopHandler, self);
  636. deactivate.Connect(handler, "deactivate", !recording ? StreamDeactivateHandler : RecordingDeactivateHandler,
  637. self);
  638. }
  639. std::optional<MultitrackVideoOutput::OBSOutputObjects> MultitrackVideoOutput::take_current()
  640. {
  641. const std::lock_guard<std::mutex> current_lock{current_mutex};
  642. auto val = std::move(current);
  643. current.reset();
  644. return val;
  645. }
  646. std::optional<MultitrackVideoOutput::OBSOutputObjects> MultitrackVideoOutput::take_current_stream_dump()
  647. {
  648. const std::lock_guard<std::mutex> current_stream_dump_lock{current_stream_dump_mutex};
  649. auto val = std::move(current_stream_dump);
  650. current_stream_dump.reset();
  651. return val;
  652. }
  653. void MultitrackVideoOutput::ReleaseOnMainThread(std::optional<OBSOutputObjects> objects)
  654. {
  655. if (!objects.has_value())
  656. return;
  657. QMetaObject::invokeMethod(
  658. QApplication::instance()->thread(), [objects = std::move(objects)] {}, Qt::QueuedConnection);
  659. }
  660. void StreamStopHandler(void *arg, calldata_t *params)
  661. {
  662. auto self = static_cast<MultitrackVideoOutput *>(arg);
  663. OBSOutputAutoRelease stream_dump_output;
  664. {
  665. const std::lock_guard<std::mutex> current_stream_dump_lock{self->current_stream_dump_mutex};
  666. if (self->current_stream_dump && self->current_stream_dump->output_)
  667. stream_dump_output = obs_output_get_ref(self->current_stream_dump->output_);
  668. }
  669. if (stream_dump_output)
  670. obs_output_stop(stream_dump_output);
  671. if (obs_output_active(static_cast<obs_output_t *>(calldata_ptr(params, "output"))))
  672. return;
  673. MultitrackVideoOutput::ReleaseOnMainThread(self->take_current());
  674. }
  675. void StreamDeactivateHandler(void *arg, calldata_t *params)
  676. {
  677. auto self = static_cast<MultitrackVideoOutput *>(arg);
  678. if (obs_output_reconnecting(static_cast<obs_output_t *>(calldata_ptr(params, "output"))))
  679. return;
  680. /* Unregister the BPM (Broadcast Performance Metrics) callback
  681. * and destroy the allocated metrics data.
  682. */
  683. obs_output_remove_packet_callback(static_cast<obs_output_t *>(calldata_ptr(params, "output")), bpm_inject,
  684. NULL);
  685. bpm_destroy(static_cast<obs_output_t *>(calldata_ptr(params, "output")));
  686. MultitrackVideoOutput::ReleaseOnMainThread(self->take_current());
  687. }
  688. void RecordingStartHandler(void * /* arg */, calldata_t * /* data */)
  689. {
  690. blog(LOG_INFO, "MultitrackVideoOutput: recording started");
  691. }
  692. void RecordingStopHandler(void *arg, calldata_t *params)
  693. {
  694. auto self = static_cast<MultitrackVideoOutput *>(arg);
  695. blog(LOG_INFO, "MultitrackVideoOutput: recording stopped");
  696. if (obs_output_active(static_cast<obs_output_t *>(calldata_ptr(params, "output"))))
  697. return;
  698. MultitrackVideoOutput::ReleaseOnMainThread(self->take_current_stream_dump());
  699. }
  700. void RecordingDeactivateHandler(void *arg, calldata_t * /*data*/)
  701. {
  702. auto self = static_cast<MultitrackVideoOutput *>(arg);
  703. MultitrackVideoOutput::ReleaseOnMainThread(self->take_current_stream_dump());
  704. }