multitrack-video-output.cpp 27 KB

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