window-basic-main-outputs.cpp 67 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347
  1. #include <string>
  2. #include <algorithm>
  3. #include <QMessageBox>
  4. #include "qt-wrappers.hpp"
  5. #include "audio-encoders.hpp"
  6. #include "window-basic-main.hpp"
  7. #include "window-basic-main-outputs.hpp"
  8. #include "window-basic-vcam-config.hpp"
  9. using namespace std;
  10. extern bool EncoderAvailable(const char *encoder);
  11. volatile bool streaming_active = false;
  12. volatile bool recording_active = false;
  13. volatile bool recording_paused = false;
  14. volatile bool replaybuf_active = false;
  15. volatile bool virtualcam_active = false;
  16. #define FTL_PROTOCOL "ftl"
  17. #define RTMP_PROTOCOL "rtmp"
  18. static void OBSStreamStarting(void *data, calldata_t *params)
  19. {
  20. BasicOutputHandler *output = static_cast<BasicOutputHandler *>(data);
  21. obs_output_t *obj = (obs_output_t *)calldata_ptr(params, "output");
  22. int sec = (int)obs_output_get_active_delay(obj);
  23. if (sec == 0)
  24. return;
  25. output->delayActive = true;
  26. QMetaObject::invokeMethod(output->main, "StreamDelayStarting",
  27. Q_ARG(int, sec));
  28. }
  29. static void OBSStreamStopping(void *data, calldata_t *params)
  30. {
  31. BasicOutputHandler *output = static_cast<BasicOutputHandler *>(data);
  32. obs_output_t *obj = (obs_output_t *)calldata_ptr(params, "output");
  33. int sec = (int)obs_output_get_active_delay(obj);
  34. if (sec == 0)
  35. QMetaObject::invokeMethod(output->main, "StreamStopping");
  36. else
  37. QMetaObject::invokeMethod(output->main, "StreamDelayStopping",
  38. Q_ARG(int, sec));
  39. }
  40. static void OBSStartStreaming(void *data, calldata_t *params)
  41. {
  42. BasicOutputHandler *output = static_cast<BasicOutputHandler *>(data);
  43. output->streamingActive = true;
  44. os_atomic_set_bool(&streaming_active, true);
  45. QMetaObject::invokeMethod(output->main, "StreamingStart");
  46. UNUSED_PARAMETER(params);
  47. }
  48. static void OBSStopStreaming(void *data, calldata_t *params)
  49. {
  50. BasicOutputHandler *output = static_cast<BasicOutputHandler *>(data);
  51. int code = (int)calldata_int(params, "code");
  52. const char *last_error = calldata_string(params, "last_error");
  53. QString arg_last_error = QString::fromUtf8(last_error);
  54. output->streamingActive = false;
  55. output->delayActive = false;
  56. os_atomic_set_bool(&streaming_active, false);
  57. QMetaObject::invokeMethod(output->main, "StreamingStop",
  58. Q_ARG(int, code),
  59. Q_ARG(QString, arg_last_error));
  60. }
  61. static void OBSStartRecording(void *data, calldata_t *params)
  62. {
  63. BasicOutputHandler *output = static_cast<BasicOutputHandler *>(data);
  64. output->recordingActive = true;
  65. os_atomic_set_bool(&recording_active, true);
  66. QMetaObject::invokeMethod(output->main, "RecordingStart");
  67. UNUSED_PARAMETER(params);
  68. }
  69. static void OBSStopRecording(void *data, calldata_t *params)
  70. {
  71. BasicOutputHandler *output = static_cast<BasicOutputHandler *>(data);
  72. int code = (int)calldata_int(params, "code");
  73. const char *last_error = calldata_string(params, "last_error");
  74. QString arg_last_error = QString::fromUtf8(last_error);
  75. output->recordingActive = false;
  76. os_atomic_set_bool(&recording_active, false);
  77. os_atomic_set_bool(&recording_paused, false);
  78. QMetaObject::invokeMethod(output->main, "RecordingStop",
  79. Q_ARG(int, code),
  80. Q_ARG(QString, arg_last_error));
  81. }
  82. static void OBSRecordStopping(void *data, calldata_t *params)
  83. {
  84. BasicOutputHandler *output = static_cast<BasicOutputHandler *>(data);
  85. QMetaObject::invokeMethod(output->main, "RecordStopping");
  86. UNUSED_PARAMETER(params);
  87. }
  88. static void OBSRecordFileChanged(void *data, calldata_t *params)
  89. {
  90. BasicOutputHandler *output = static_cast<BasicOutputHandler *>(data);
  91. const char *next_file = calldata_string(params, "next_file");
  92. QString arg_last_file =
  93. QString::fromUtf8(output->lastRecordingPath.c_str());
  94. QMetaObject::invokeMethod(output->main, "RecordingFileChanged",
  95. Q_ARG(QString, arg_last_file));
  96. output->lastRecordingPath = next_file;
  97. }
  98. static void OBSStartReplayBuffer(void *data, calldata_t *params)
  99. {
  100. BasicOutputHandler *output = static_cast<BasicOutputHandler *>(data);
  101. output->replayBufferActive = true;
  102. os_atomic_set_bool(&replaybuf_active, true);
  103. QMetaObject::invokeMethod(output->main, "ReplayBufferStart");
  104. UNUSED_PARAMETER(params);
  105. }
  106. static void OBSStopReplayBuffer(void *data, calldata_t *params)
  107. {
  108. BasicOutputHandler *output = static_cast<BasicOutputHandler *>(data);
  109. int code = (int)calldata_int(params, "code");
  110. output->replayBufferActive = false;
  111. os_atomic_set_bool(&replaybuf_active, false);
  112. QMetaObject::invokeMethod(output->main, "ReplayBufferStop",
  113. Q_ARG(int, code));
  114. }
  115. static void OBSReplayBufferStopping(void *data, calldata_t *params)
  116. {
  117. BasicOutputHandler *output = static_cast<BasicOutputHandler *>(data);
  118. QMetaObject::invokeMethod(output->main, "ReplayBufferStopping");
  119. UNUSED_PARAMETER(params);
  120. }
  121. static void OBSReplayBufferSaved(void *data, calldata_t *)
  122. {
  123. BasicOutputHandler *output = static_cast<BasicOutputHandler *>(data);
  124. QMetaObject::invokeMethod(output->main, "ReplayBufferSaved",
  125. Qt::QueuedConnection);
  126. }
  127. static void OBSStartVirtualCam(void *data, calldata_t *params)
  128. {
  129. BasicOutputHandler *output = static_cast<BasicOutputHandler *>(data);
  130. output->virtualCamActive = true;
  131. os_atomic_set_bool(&virtualcam_active, true);
  132. QMetaObject::invokeMethod(output->main, "OnVirtualCamStart");
  133. UNUSED_PARAMETER(params);
  134. }
  135. static void OBSStopVirtualCam(void *data, calldata_t *params)
  136. {
  137. BasicOutputHandler *output = static_cast<BasicOutputHandler *>(data);
  138. int code = (int)calldata_int(params, "code");
  139. output->virtualCamActive = false;
  140. os_atomic_set_bool(&virtualcam_active, false);
  141. QMetaObject::invokeMethod(output->main, "OnVirtualCamStop",
  142. Q_ARG(int, code));
  143. obs_output_set_media(output->virtualCam, nullptr, nullptr);
  144. output->DestroyVirtualCamView();
  145. }
  146. /* ------------------------------------------------------------------------ */
  147. static bool CreateSimpleAACEncoder(OBSEncoder &res, int bitrate,
  148. const char *name, size_t idx)
  149. {
  150. const char *id_ = GetSimpleAACEncoderForBitrate(bitrate);
  151. if (!id_) {
  152. res = nullptr;
  153. return false;
  154. }
  155. res = obs_audio_encoder_create(id_, name, nullptr, idx, nullptr);
  156. if (res) {
  157. obs_encoder_release(res);
  158. return true;
  159. }
  160. return false;
  161. }
  162. static bool CreateSimpleOpusEncoder(OBSEncoder &res, int bitrate,
  163. const char *name, size_t idx)
  164. {
  165. const char *id_ = GetSimpleOpusEncoderForBitrate(bitrate);
  166. if (!id_) {
  167. res = nullptr;
  168. return false;
  169. }
  170. res = obs_audio_encoder_create(id_, name, nullptr, idx, nullptr);
  171. if (res) {
  172. obs_encoder_release(res);
  173. return true;
  174. }
  175. return false;
  176. }
  177. static inline bool can_use_output(const char *prot, const char *output,
  178. const char *prot_test1,
  179. const char *prot_test2 = nullptr)
  180. {
  181. return (strcmp(prot, prot_test1) == 0 ||
  182. (prot_test2 && strcmp(prot, prot_test2) == 0)) &&
  183. (obs_get_output_flags(output) & OBS_OUTPUT_SERVICE) != 0;
  184. }
  185. static bool return_first_id(void *data, const char *id)
  186. {
  187. const char **output = (const char **)data;
  188. *output = id;
  189. return false;
  190. }
  191. static const char *GetStreamOutputType(const obs_service_t *service)
  192. {
  193. const char *protocol = obs_service_get_protocol(service);
  194. const char *output = nullptr;
  195. if (!protocol) {
  196. blog(LOG_WARNING, "The service '%s' has no protocol set",
  197. obs_service_get_id(service));
  198. return nullptr;
  199. }
  200. if (!obs_is_output_protocol_registered(protocol)) {
  201. blog(LOG_WARNING, "The protocol '%s' is not registered",
  202. protocol);
  203. return nullptr;
  204. }
  205. /* Check if the service has a preferred output type */
  206. output = obs_service_get_preferred_output_type(service);
  207. if (output) {
  208. if ((obs_get_output_flags(output) & OBS_OUTPUT_SERVICE) != 0)
  209. return output;
  210. blog(LOG_WARNING,
  211. "The output '%s' is not registered, fallback to another one",
  212. output);
  213. }
  214. /* Otherwise, prefer first-party output types */
  215. if (can_use_output(protocol, "rtmp_output", "RTMP", "RTMPS")) {
  216. return "rtmp_output";
  217. } else if (can_use_output(protocol, "ffmpeg_hls_muxer", "HLS")) {
  218. return "ffmpeg_hls_muxer";
  219. } else if (can_use_output(protocol, "ffmpeg_mpegts_muxer", "SRT",
  220. "RIST")) {
  221. return "ffmpeg_mpegts_muxer";
  222. }
  223. /* If third-party protocol, use the first enumerated type */
  224. obs_enum_output_types_with_protocol(protocol, &output, return_first_id);
  225. if (output)
  226. return output;
  227. blog(LOG_WARNING,
  228. "No output compatible with the service '%s' is registered",
  229. obs_service_get_id(service));
  230. return nullptr;
  231. }
  232. /* ------------------------------------------------------------------------ */
  233. inline BasicOutputHandler::BasicOutputHandler(OBSBasic *main_) : main(main_)
  234. {
  235. if (main->vcamEnabled) {
  236. virtualCam = obs_output_create("virtualcam_output",
  237. "virtualcam_output", nullptr,
  238. nullptr);
  239. signal_handler_t *signal =
  240. obs_output_get_signal_handler(virtualCam);
  241. startVirtualCam.Connect(signal, "start", OBSStartVirtualCam,
  242. this);
  243. stopVirtualCam.Connect(signal, "stop", OBSStopVirtualCam, this);
  244. }
  245. }
  246. bool BasicOutputHandler::StartVirtualCam()
  247. {
  248. if (!main->vcamEnabled)
  249. return false;
  250. if (!virtualCamView)
  251. virtualCamView = obs_view_create();
  252. UpdateVirtualCamOutputSource();
  253. if (!virtualCamVideo) {
  254. virtualCamVideo = obs_view_add(virtualCamView);
  255. if (!virtualCamVideo)
  256. return false;
  257. }
  258. obs_output_set_media(virtualCam, virtualCamVideo, obs_get_audio());
  259. if (!Active())
  260. SetupOutputs();
  261. bool success = obs_output_start(virtualCam);
  262. if (!success)
  263. DestroyVirtualCamView();
  264. return success;
  265. }
  266. void BasicOutputHandler::StopVirtualCam()
  267. {
  268. if (main->vcamEnabled) {
  269. obs_output_stop(virtualCam);
  270. }
  271. }
  272. bool BasicOutputHandler::VirtualCamActive() const
  273. {
  274. if (main->vcamEnabled) {
  275. return obs_output_active(virtualCam);
  276. }
  277. return false;
  278. }
  279. void BasicOutputHandler::UpdateVirtualCamOutputSource()
  280. {
  281. if (!main->vcamEnabled || !virtualCamView)
  282. return;
  283. OBSSourceAutoRelease source;
  284. switch (main->vcamConfig.type) {
  285. case VCamOutputType::InternalOutput:
  286. switch (main->vcamConfig.internal) {
  287. case VCamInternalType::Default:
  288. source = obs_get_output_source(0);
  289. break;
  290. case VCamInternalType::Preview:
  291. OBSSource s = main->GetCurrentSceneSource();
  292. obs_source_get_ref(s);
  293. source = s.Get();
  294. break;
  295. }
  296. break;
  297. case VCamOutputType::SceneOutput:
  298. source = obs_get_source_by_name(main->vcamConfig.scene.c_str());
  299. break;
  300. case VCamOutputType::SourceOutput:
  301. OBSSource s =
  302. obs_get_source_by_name(main->vcamConfig.source.c_str());
  303. if (!vCamSourceScene)
  304. vCamSourceScene =
  305. obs_scene_create_private("vcam_source");
  306. source = obs_source_get_ref(
  307. obs_scene_get_source(vCamSourceScene));
  308. if (vCamSourceSceneItem &&
  309. (obs_sceneitem_get_source(vCamSourceSceneItem) != s)) {
  310. obs_sceneitem_remove(vCamSourceSceneItem);
  311. vCamSourceSceneItem = nullptr;
  312. }
  313. if (!vCamSourceSceneItem) {
  314. vCamSourceSceneItem = obs_scene_add(vCamSourceScene, s);
  315. obs_source_release(s);
  316. obs_sceneitem_set_bounds_type(vCamSourceSceneItem,
  317. OBS_BOUNDS_SCALE_INNER);
  318. obs_sceneitem_set_bounds_alignment(vCamSourceSceneItem,
  319. OBS_ALIGN_CENTER);
  320. const struct vec2 size = {
  321. (float)obs_source_get_width(source),
  322. (float)obs_source_get_height(source),
  323. };
  324. obs_sceneitem_set_bounds(vCamSourceSceneItem, &size);
  325. }
  326. break;
  327. }
  328. OBSSourceAutoRelease current = obs_view_get_source(virtualCamView, 0);
  329. if (source != current)
  330. obs_view_set_source(virtualCamView, 0, source);
  331. }
  332. void BasicOutputHandler::DestroyVirtualCamView()
  333. {
  334. obs_view_remove(virtualCamView);
  335. obs_view_set_source(virtualCamView, 0, nullptr);
  336. virtualCamVideo = nullptr;
  337. obs_view_destroy(virtualCamView);
  338. virtualCamView = nullptr;
  339. if (!vCamSourceScene)
  340. return;
  341. obs_scene_release(vCamSourceScene);
  342. vCamSourceScene = nullptr;
  343. vCamSourceSceneItem = nullptr;
  344. }
  345. /* ------------------------------------------------------------------------ */
  346. struct SimpleOutput : BasicOutputHandler {
  347. OBSEncoder audioStreaming;
  348. OBSEncoder videoStreaming;
  349. OBSEncoder audioRecording;
  350. OBSEncoder audioArchive;
  351. OBSEncoder videoRecording;
  352. string videoEncoder;
  353. string videoQuality;
  354. bool usingRecordingPreset = false;
  355. bool recordingConfigured = false;
  356. bool ffmpegOutput = false;
  357. bool lowCPUx264 = false;
  358. SimpleOutput(OBSBasic *main_);
  359. int CalcCRF(int crf);
  360. void UpdateRecordingSettings_x264_crf(int crf);
  361. void UpdateRecordingSettings_qsv11(int crf, bool av1);
  362. void UpdateRecordingSettings_nvenc(int cqp);
  363. void UpdateRecordingSettings_nvenc_hevc_av1(int cqp);
  364. void UpdateRecordingSettings_amd_cqp(int cqp);
  365. void UpdateRecordingSettings_apple(int quality);
  366. #ifdef ENABLE_HEVC
  367. void UpdateRecordingSettings_apple_hevc(int quality);
  368. #endif
  369. void UpdateRecordingSettings();
  370. void UpdateRecordingAudioSettings();
  371. virtual void Update() override;
  372. void SetupOutputs() override;
  373. int GetAudioBitrate() const;
  374. void LoadRecordingPreset_Lossy(const char *encoder);
  375. void LoadRecordingPreset_Lossless();
  376. void LoadRecordingPreset();
  377. void LoadStreamingPreset_Lossy(const char *encoder);
  378. void UpdateRecording();
  379. bool ConfigureRecording(bool useReplayBuffer);
  380. void SetupVodTrack(obs_service_t *service);
  381. virtual bool SetupStreaming(obs_service_t *service) override;
  382. virtual bool StartStreaming(obs_service_t *service) override;
  383. virtual bool StartRecording() override;
  384. virtual bool StartReplayBuffer() override;
  385. virtual void StopStreaming(bool force) override;
  386. virtual void StopRecording(bool force) override;
  387. virtual void StopReplayBuffer(bool force) override;
  388. virtual bool StreamingActive() const override;
  389. virtual bool RecordingActive() const override;
  390. virtual bool ReplayBufferActive() const override;
  391. };
  392. void SimpleOutput::LoadRecordingPreset_Lossless()
  393. {
  394. fileOutput = obs_output_create("ffmpeg_output", "simple_ffmpeg_output",
  395. nullptr, nullptr);
  396. if (!fileOutput)
  397. throw "Failed to create recording FFmpeg output "
  398. "(simple output)";
  399. OBSDataAutoRelease settings = obs_data_create();
  400. obs_data_set_string(settings, "format_name", "avi");
  401. obs_data_set_string(settings, "video_encoder", "utvideo");
  402. obs_data_set_string(settings, "audio_encoder", "pcm_s16le");
  403. int aMixes = 1;
  404. obs_output_set_mixers(fileOutput, aMixes);
  405. obs_output_update(fileOutput, settings);
  406. }
  407. void SimpleOutput::LoadRecordingPreset_Lossy(const char *encoderId)
  408. {
  409. videoRecording = obs_video_encoder_create(
  410. encoderId, "simple_video_recording", nullptr, nullptr);
  411. if (!videoRecording)
  412. throw "Failed to create video recording encoder (simple output)";
  413. obs_encoder_release(videoRecording);
  414. }
  415. void SimpleOutput::LoadStreamingPreset_Lossy(const char *encoderId)
  416. {
  417. videoStreaming = obs_video_encoder_create(
  418. encoderId, "simple_video_stream", nullptr, nullptr);
  419. if (!videoStreaming)
  420. throw "Failed to create video streaming encoder (simple output)";
  421. obs_encoder_release(videoStreaming);
  422. }
  423. /* mistakes have been made to lead us to this. */
  424. const char *get_simple_output_encoder(const char *encoder)
  425. {
  426. if (strcmp(encoder, SIMPLE_ENCODER_X264) == 0) {
  427. return "obs_x264";
  428. } else if (strcmp(encoder, SIMPLE_ENCODER_X264_LOWCPU) == 0) {
  429. return "obs_x264";
  430. } else if (strcmp(encoder, SIMPLE_ENCODER_QSV) == 0) {
  431. return "obs_qsv11_v2";
  432. } else if (strcmp(encoder, SIMPLE_ENCODER_QSV_AV1) == 0) {
  433. return "obs_qsv11_av1";
  434. } else if (strcmp(encoder, SIMPLE_ENCODER_AMD) == 0) {
  435. return "h264_texture_amf";
  436. #ifdef ENABLE_HEVC
  437. } else if (strcmp(encoder, SIMPLE_ENCODER_AMD_HEVC) == 0) {
  438. return "h265_texture_amf";
  439. #endif
  440. } else if (strcmp(encoder, SIMPLE_ENCODER_AMD_AV1) == 0) {
  441. return "av1_texture_amf";
  442. } else if (strcmp(encoder, SIMPLE_ENCODER_NVENC) == 0) {
  443. return EncoderAvailable("jim_nvenc") ? "jim_nvenc"
  444. : "ffmpeg_nvenc";
  445. #ifdef ENABLE_HEVC
  446. } else if (strcmp(encoder, SIMPLE_ENCODER_NVENC_HEVC) == 0) {
  447. return EncoderAvailable("jim_hevc_nvenc") ? "jim_hevc_nvenc"
  448. : "ffmpeg_hevc_nvenc";
  449. #endif
  450. } else if (strcmp(encoder, SIMPLE_ENCODER_NVENC_AV1) == 0) {
  451. return "jim_av1_nvenc";
  452. } else if (strcmp(encoder, SIMPLE_ENCODER_APPLE_H264) == 0) {
  453. return "com.apple.videotoolbox.videoencoder.ave.avc";
  454. #ifdef ENABLE_HEVC
  455. } else if (strcmp(encoder, SIMPLE_ENCODER_APPLE_HEVC) == 0) {
  456. return "com.apple.videotoolbox.videoencoder.ave.hevc";
  457. #endif
  458. }
  459. return "obs_x264";
  460. }
  461. void SimpleOutput::LoadRecordingPreset()
  462. {
  463. const char *quality =
  464. config_get_string(main->Config(), "SimpleOutput", "RecQuality");
  465. const char *encoder =
  466. config_get_string(main->Config(), "SimpleOutput", "RecEncoder");
  467. const char *audio_encoder = config_get_string(
  468. main->Config(), "SimpleOutput", "RecAudioEncoder");
  469. videoEncoder = encoder;
  470. videoQuality = quality;
  471. ffmpegOutput = false;
  472. if (strcmp(quality, "Stream") == 0) {
  473. videoRecording = videoStreaming;
  474. audioRecording = audioStreaming;
  475. usingRecordingPreset = false;
  476. return;
  477. } else if (strcmp(quality, "Lossless") == 0) {
  478. LoadRecordingPreset_Lossless();
  479. usingRecordingPreset = true;
  480. ffmpegOutput = true;
  481. return;
  482. } else {
  483. lowCPUx264 = false;
  484. if (strcmp(encoder, SIMPLE_ENCODER_X264_LOWCPU) == 0)
  485. lowCPUx264 = true;
  486. LoadRecordingPreset_Lossy(get_simple_output_encoder(encoder));
  487. usingRecordingPreset = true;
  488. bool success = false;
  489. if (strcmp(audio_encoder, "opus") == 0)
  490. success = CreateSimpleOpusEncoder(
  491. audioRecording, 192, "simple_opus_recording",
  492. 0);
  493. else
  494. success = CreateSimpleAACEncoder(
  495. audioRecording, 192, "simple_aac_recording", 0);
  496. if (!success)
  497. throw "Failed to create audio recording encoder "
  498. "(simple output)";
  499. }
  500. }
  501. #define SIMPLE_ARCHIVE_NAME "simple_archive_audio"
  502. SimpleOutput::SimpleOutput(OBSBasic *main_) : BasicOutputHandler(main_)
  503. {
  504. const char *encoder = config_get_string(main->Config(), "SimpleOutput",
  505. "StreamEncoder");
  506. const char *audio_encoder = config_get_string(
  507. main->Config(), "SimpleOutput", "StreamAudioEncoder");
  508. LoadStreamingPreset_Lossy(get_simple_output_encoder(encoder));
  509. bool success = false;
  510. if (strcmp(audio_encoder, "opus") == 0)
  511. success = CreateSimpleOpusEncoder(
  512. audioStreaming, GetAudioBitrate(), "simple_opus", 0);
  513. else
  514. success = CreateSimpleAACEncoder(
  515. audioStreaming, GetAudioBitrate(), "simple_aac", 0);
  516. if (!success)
  517. throw "Failed to create audio streaming encoder (simple output)";
  518. if (strcmp(audio_encoder, "opus") == 0)
  519. success = CreateSimpleOpusEncoder(audioArchive,
  520. GetAudioBitrate(),
  521. SIMPLE_ARCHIVE_NAME, 1);
  522. else
  523. success = CreateSimpleAACEncoder(audioArchive,
  524. GetAudioBitrate(),
  525. SIMPLE_ARCHIVE_NAME, 1);
  526. if (!success)
  527. throw "Failed to create audio archive encoder (simple output)";
  528. LoadRecordingPreset();
  529. if (!ffmpegOutput) {
  530. bool useReplayBuffer = config_get_bool(main->Config(),
  531. "SimpleOutput", "RecRB");
  532. if (useReplayBuffer) {
  533. OBSDataAutoRelease hotkey;
  534. const char *str = config_get_string(
  535. main->Config(), "Hotkeys", "ReplayBuffer");
  536. if (str)
  537. hotkey = obs_data_create_from_json(str);
  538. else
  539. hotkey = nullptr;
  540. replayBuffer = obs_output_create("replay_buffer",
  541. Str("ReplayBuffer"),
  542. nullptr, hotkey);
  543. if (!replayBuffer)
  544. throw "Failed to create replay buffer output "
  545. "(simple output)";
  546. signal_handler_t *signal =
  547. obs_output_get_signal_handler(replayBuffer);
  548. startReplayBuffer.Connect(signal, "start",
  549. OBSStartReplayBuffer, this);
  550. stopReplayBuffer.Connect(signal, "stop",
  551. OBSStopReplayBuffer, this);
  552. replayBufferStopping.Connect(signal, "stopping",
  553. OBSReplayBufferStopping,
  554. this);
  555. replayBufferSaved.Connect(signal, "saved",
  556. OBSReplayBufferSaved, this);
  557. }
  558. fileOutput = obs_output_create(
  559. "ffmpeg_muxer", "simple_file_output", nullptr, nullptr);
  560. if (!fileOutput)
  561. throw "Failed to create recording output "
  562. "(simple output)";
  563. }
  564. startRecording.Connect(obs_output_get_signal_handler(fileOutput),
  565. "start", OBSStartRecording, this);
  566. stopRecording.Connect(obs_output_get_signal_handler(fileOutput), "stop",
  567. OBSStopRecording, this);
  568. recordStopping.Connect(obs_output_get_signal_handler(fileOutput),
  569. "stopping", OBSRecordStopping, this);
  570. }
  571. int SimpleOutput::GetAudioBitrate() const
  572. {
  573. const char *audio_encoder = config_get_string(
  574. main->Config(), "SimpleOutput", "StreamAudioEncoder");
  575. int bitrate = (int)config_get_uint(main->Config(), "SimpleOutput",
  576. "ABitrate");
  577. if (strcmp(audio_encoder, "opus") == 0)
  578. return FindClosestAvailableSimpleOpusBitrate(bitrate);
  579. return FindClosestAvailableSimpleAACBitrate(bitrate);
  580. }
  581. void SimpleOutput::Update()
  582. {
  583. OBSDataAutoRelease videoSettings = obs_data_create();
  584. OBSDataAutoRelease audioSettings = obs_data_create();
  585. int videoBitrate =
  586. config_get_uint(main->Config(), "SimpleOutput", "VBitrate");
  587. int audioBitrate = GetAudioBitrate();
  588. bool advanced =
  589. config_get_bool(main->Config(), "SimpleOutput", "UseAdvanced");
  590. bool enforceBitrate = !config_get_bool(main->Config(), "Stream1",
  591. "IgnoreRecommended");
  592. const char *custom = config_get_string(main->Config(), "SimpleOutput",
  593. "x264Settings");
  594. const char *encoder = config_get_string(main->Config(), "SimpleOutput",
  595. "StreamEncoder");
  596. const char *presetType;
  597. const char *preset;
  598. if (strcmp(encoder, SIMPLE_ENCODER_QSV) == 0) {
  599. presetType = "QSVPreset";
  600. } else if (strcmp(encoder, SIMPLE_ENCODER_QSV_AV1) == 0) {
  601. presetType = "QSVPreset";
  602. } else if (strcmp(encoder, SIMPLE_ENCODER_AMD) == 0) {
  603. presetType = "AMDPreset";
  604. #ifdef ENABLE_HEVC
  605. } else if (strcmp(encoder, SIMPLE_ENCODER_AMD_HEVC) == 0) {
  606. presetType = "AMDPreset";
  607. #endif
  608. } else if (strcmp(encoder, SIMPLE_ENCODER_NVENC) == 0) {
  609. presetType = "NVENCPreset2";
  610. #ifdef ENABLE_HEVC
  611. } else if (strcmp(encoder, SIMPLE_ENCODER_NVENC_HEVC) == 0) {
  612. presetType = "NVENCPreset2";
  613. #endif
  614. } else if (strcmp(encoder, SIMPLE_ENCODER_AMD_AV1) == 0) {
  615. presetType = "AMDAV1Preset";
  616. } else if (strcmp(encoder, SIMPLE_ENCODER_NVENC_AV1) == 0) {
  617. presetType = "NVENCPreset2";
  618. } else {
  619. presetType = "Preset";
  620. }
  621. preset = config_get_string(main->Config(), "SimpleOutput", presetType);
  622. obs_data_set_string(videoSettings,
  623. (strcmp(presetType, "NVENCPreset2") == 0)
  624. ? "preset2"
  625. : "preset",
  626. preset);
  627. obs_data_set_string(videoSettings, "rate_control", "CBR");
  628. obs_data_set_int(videoSettings, "bitrate", videoBitrate);
  629. if (advanced)
  630. obs_data_set_string(videoSettings, "x264opts", custom);
  631. obs_data_set_string(audioSettings, "rate_control", "CBR");
  632. obs_data_set_int(audioSettings, "bitrate", audioBitrate);
  633. obs_service_apply_encoder_settings(main->GetService(), videoSettings,
  634. audioSettings);
  635. if (!enforceBitrate) {
  636. obs_data_set_int(videoSettings, "bitrate", videoBitrate);
  637. obs_data_set_int(audioSettings, "bitrate", audioBitrate);
  638. }
  639. video_t *video = obs_get_video();
  640. enum video_format format = video_output_get_format(video);
  641. switch (format) {
  642. case VIDEO_FORMAT_I420:
  643. case VIDEO_FORMAT_NV12:
  644. case VIDEO_FORMAT_I010:
  645. case VIDEO_FORMAT_P010:
  646. break;
  647. default:
  648. obs_encoder_set_preferred_video_format(videoStreaming,
  649. VIDEO_FORMAT_NV12);
  650. }
  651. obs_encoder_update(videoStreaming, videoSettings);
  652. obs_encoder_update(audioStreaming, audioSettings);
  653. obs_encoder_update(audioArchive, audioSettings);
  654. }
  655. void SimpleOutput::UpdateRecordingAudioSettings()
  656. {
  657. OBSDataAutoRelease settings = obs_data_create();
  658. obs_data_set_int(settings, "bitrate", 192);
  659. obs_data_set_string(settings, "rate_control", "CBR");
  660. obs_encoder_update(audioRecording, settings);
  661. }
  662. #define CROSS_DIST_CUTOFF 2000.0
  663. int SimpleOutput::CalcCRF(int crf)
  664. {
  665. int cx = config_get_uint(main->Config(), "Video", "OutputCX");
  666. int cy = config_get_uint(main->Config(), "Video", "OutputCY");
  667. double fCX = double(cx);
  668. double fCY = double(cy);
  669. if (lowCPUx264)
  670. crf -= 2;
  671. double crossDist = sqrt(fCX * fCX + fCY * fCY);
  672. double crfResReduction =
  673. fmin(CROSS_DIST_CUTOFF, crossDist) / CROSS_DIST_CUTOFF;
  674. crfResReduction = (1.0 - crfResReduction) * 10.0;
  675. return crf - int(crfResReduction);
  676. }
  677. void SimpleOutput::UpdateRecordingSettings_x264_crf(int crf)
  678. {
  679. OBSDataAutoRelease settings = obs_data_create();
  680. obs_data_set_int(settings, "crf", crf);
  681. obs_data_set_bool(settings, "use_bufsize", true);
  682. obs_data_set_string(settings, "rate_control", "CRF");
  683. obs_data_set_string(settings, "profile", "high");
  684. obs_data_set_string(settings, "preset",
  685. lowCPUx264 ? "ultrafast" : "veryfast");
  686. obs_encoder_update(videoRecording, settings);
  687. }
  688. static bool icq_available(obs_encoder_t *encoder)
  689. {
  690. obs_properties_t *props = obs_encoder_properties(encoder);
  691. obs_property_t *p = obs_properties_get(props, "rate_control");
  692. bool icq_found = false;
  693. size_t num = obs_property_list_item_count(p);
  694. for (size_t i = 0; i < num; i++) {
  695. const char *val = obs_property_list_item_string(p, i);
  696. if (strcmp(val, "ICQ") == 0) {
  697. icq_found = true;
  698. break;
  699. }
  700. }
  701. obs_properties_destroy(props);
  702. return icq_found;
  703. }
  704. void SimpleOutput::UpdateRecordingSettings_qsv11(int crf, bool av1)
  705. {
  706. bool icq = icq_available(videoRecording);
  707. OBSDataAutoRelease settings = obs_data_create();
  708. obs_data_set_string(settings, "profile", "high");
  709. if (icq && !av1) {
  710. obs_data_set_string(settings, "rate_control", "ICQ");
  711. obs_data_set_int(settings, "icq_quality", crf);
  712. } else {
  713. obs_data_set_string(settings, "rate_control", "CQP");
  714. obs_data_set_int(settings, "cqp", crf);
  715. }
  716. obs_encoder_update(videoRecording, settings);
  717. }
  718. void SimpleOutput::UpdateRecordingSettings_nvenc(int cqp)
  719. {
  720. OBSDataAutoRelease settings = obs_data_create();
  721. obs_data_set_string(settings, "rate_control", "CQP");
  722. obs_data_set_string(settings, "profile", "high");
  723. obs_data_set_int(settings, "cqp", cqp);
  724. obs_encoder_update(videoRecording, settings);
  725. }
  726. void SimpleOutput::UpdateRecordingSettings_nvenc_hevc_av1(int cqp)
  727. {
  728. OBSDataAutoRelease settings = obs_data_create();
  729. obs_data_set_string(settings, "rate_control", "CQP");
  730. obs_data_set_string(settings, "profile", "main");
  731. obs_data_set_int(settings, "cqp", cqp);
  732. obs_encoder_update(videoRecording, settings);
  733. }
  734. void SimpleOutput::UpdateRecordingSettings_apple(int quality)
  735. {
  736. OBSDataAutoRelease settings = obs_data_create();
  737. obs_data_set_string(settings, "rate_control", "CRF");
  738. obs_data_set_string(settings, "profile", "high");
  739. obs_data_set_int(settings, "quality", quality);
  740. obs_encoder_update(videoRecording, settings);
  741. }
  742. #ifdef ENABLE_HEVC
  743. void SimpleOutput::UpdateRecordingSettings_apple_hevc(int quality)
  744. {
  745. OBSDataAutoRelease settings = obs_data_create();
  746. obs_data_set_string(settings, "rate_control", "CRF");
  747. obs_data_set_string(settings, "profile", "main");
  748. obs_data_set_int(settings, "quality", quality);
  749. obs_encoder_update(videoRecording, settings);
  750. }
  751. #endif
  752. void SimpleOutput::UpdateRecordingSettings_amd_cqp(int cqp)
  753. {
  754. OBSDataAutoRelease settings = obs_data_create();
  755. obs_data_set_string(settings, "rate_control", "CQP");
  756. obs_data_set_string(settings, "profile", "high");
  757. obs_data_set_string(settings, "preset", "quality");
  758. obs_data_set_int(settings, "cqp", cqp);
  759. obs_encoder_update(videoRecording, settings);
  760. }
  761. void SimpleOutput::UpdateRecordingSettings()
  762. {
  763. bool ultra_hq = (videoQuality == "HQ");
  764. int crf = CalcCRF(ultra_hq ? 16 : 23);
  765. if (astrcmp_n(videoEncoder.c_str(), "x264", 4) == 0) {
  766. UpdateRecordingSettings_x264_crf(crf);
  767. } else if (videoEncoder == SIMPLE_ENCODER_QSV) {
  768. UpdateRecordingSettings_qsv11(crf, false);
  769. } else if (videoEncoder == SIMPLE_ENCODER_QSV_AV1) {
  770. UpdateRecordingSettings_qsv11(crf, true);
  771. } else if (videoEncoder == SIMPLE_ENCODER_AMD) {
  772. UpdateRecordingSettings_amd_cqp(crf);
  773. #ifdef ENABLE_HEVC
  774. } else if (videoEncoder == SIMPLE_ENCODER_AMD_HEVC) {
  775. UpdateRecordingSettings_amd_cqp(crf);
  776. #endif
  777. } else if (videoEncoder == SIMPLE_ENCODER_NVENC) {
  778. UpdateRecordingSettings_nvenc(crf);
  779. #ifdef ENABLE_HEVC
  780. } else if (videoEncoder == SIMPLE_ENCODER_NVENC_HEVC) {
  781. UpdateRecordingSettings_nvenc_hevc_av1(crf);
  782. #endif
  783. } else if (videoEncoder == SIMPLE_ENCODER_NVENC_AV1) {
  784. UpdateRecordingSettings_nvenc_hevc_av1(crf);
  785. } else if (videoEncoder == SIMPLE_ENCODER_APPLE_H264) {
  786. /* These are magic numbers. 0 - 100, more is better. */
  787. UpdateRecordingSettings_apple(ultra_hq ? 70 : 50);
  788. #ifdef ENABLE_HEVC
  789. } else if (videoEncoder == SIMPLE_ENCODER_APPLE_HEVC) {
  790. UpdateRecordingSettings_apple_hevc(ultra_hq ? 70 : 50);
  791. #endif
  792. }
  793. UpdateRecordingAudioSettings();
  794. }
  795. inline void SimpleOutput::SetupOutputs()
  796. {
  797. SimpleOutput::Update();
  798. obs_encoder_set_video(videoStreaming, obs_get_video());
  799. obs_encoder_set_audio(audioStreaming, obs_get_audio());
  800. obs_encoder_set_audio(audioArchive, obs_get_audio());
  801. if (usingRecordingPreset) {
  802. if (ffmpegOutput) {
  803. obs_output_set_media(fileOutput, obs_get_video(),
  804. obs_get_audio());
  805. } else {
  806. obs_encoder_set_video(videoRecording, obs_get_video());
  807. obs_encoder_set_audio(audioRecording, obs_get_audio());
  808. }
  809. }
  810. }
  811. const char *FindAudioEncoderFromCodec(const char *type)
  812. {
  813. const char *alt_enc_id = nullptr;
  814. size_t i = 0;
  815. while (obs_enum_encoder_types(i++, &alt_enc_id)) {
  816. const char *codec = obs_get_encoder_codec(alt_enc_id);
  817. if (strcmp(type, codec) == 0) {
  818. return alt_enc_id;
  819. }
  820. }
  821. return nullptr;
  822. }
  823. bool SimpleOutput::SetupStreaming(obs_service_t *service)
  824. {
  825. if (!Active())
  826. SetupOutputs();
  827. Auth *auth = main->GetAuth();
  828. if (auth)
  829. auth->OnStreamConfig();
  830. /* --------------------- */
  831. const char *type = GetStreamOutputType(service);
  832. if (!type)
  833. return false;
  834. /* XXX: this is messy and disgusting and should be refactored */
  835. if (outputType != type) {
  836. streamDelayStarting.Disconnect();
  837. streamStopping.Disconnect();
  838. startStreaming.Disconnect();
  839. stopStreaming.Disconnect();
  840. streamOutput = obs_output_create(type, "simple_stream", nullptr,
  841. nullptr);
  842. if (!streamOutput) {
  843. blog(LOG_WARNING,
  844. "Creation of stream output type '%s' "
  845. "failed!",
  846. type);
  847. return false;
  848. }
  849. streamDelayStarting.Connect(
  850. obs_output_get_signal_handler(streamOutput), "starting",
  851. OBSStreamStarting, this);
  852. streamStopping.Connect(
  853. obs_output_get_signal_handler(streamOutput), "stopping",
  854. OBSStreamStopping, this);
  855. startStreaming.Connect(
  856. obs_output_get_signal_handler(streamOutput), "start",
  857. OBSStartStreaming, this);
  858. stopStreaming.Connect(
  859. obs_output_get_signal_handler(streamOutput), "stop",
  860. OBSStopStreaming, this);
  861. outputType = type;
  862. }
  863. obs_output_set_video_encoder(streamOutput, videoStreaming);
  864. obs_output_set_audio_encoder(streamOutput, audioStreaming, 0);
  865. obs_output_set_service(streamOutput, service);
  866. return true;
  867. }
  868. static inline bool ServiceSupportsVodTrack(const char *service);
  869. static void clear_archive_encoder(obs_output_t *output,
  870. const char *expected_name)
  871. {
  872. obs_encoder_t *last = obs_output_get_audio_encoder(output, 1);
  873. bool clear = false;
  874. /* ensures that we don't remove twitch's soundtrack encoder */
  875. if (last) {
  876. const char *name = obs_encoder_get_name(last);
  877. clear = name && strcmp(name, expected_name) == 0;
  878. obs_encoder_release(last);
  879. }
  880. if (clear)
  881. obs_output_set_audio_encoder(output, nullptr, 1);
  882. }
  883. void SimpleOutput::SetupVodTrack(obs_service_t *service)
  884. {
  885. bool advanced =
  886. config_get_bool(main->Config(), "SimpleOutput", "UseAdvanced");
  887. bool enable = config_get_bool(main->Config(), "SimpleOutput",
  888. "VodTrackEnabled");
  889. bool enableForCustomServer = config_get_bool(
  890. GetGlobalConfig(), "General", "EnableCustomServerVodTrack");
  891. OBSDataAutoRelease settings = obs_service_get_settings(service);
  892. const char *name = obs_data_get_string(settings, "service");
  893. const char *id = obs_service_get_id(service);
  894. if (strcmp(id, "rtmp_custom") == 0)
  895. enable = enableForCustomServer ? enable : false;
  896. else
  897. enable = advanced && enable && ServiceSupportsVodTrack(name);
  898. if (enable)
  899. obs_output_set_audio_encoder(streamOutput, audioArchive, 1);
  900. else
  901. clear_archive_encoder(streamOutput, SIMPLE_ARCHIVE_NAME);
  902. }
  903. bool SimpleOutput::StartStreaming(obs_service_t *service)
  904. {
  905. bool reconnect = config_get_bool(main->Config(), "Output", "Reconnect");
  906. int retryDelay =
  907. config_get_uint(main->Config(), "Output", "RetryDelay");
  908. int maxRetries =
  909. config_get_uint(main->Config(), "Output", "MaxRetries");
  910. bool useDelay =
  911. config_get_bool(main->Config(), "Output", "DelayEnable");
  912. int delaySec = config_get_int(main->Config(), "Output", "DelaySec");
  913. bool preserveDelay =
  914. config_get_bool(main->Config(), "Output", "DelayPreserve");
  915. const char *bindIP =
  916. config_get_string(main->Config(), "Output", "BindIP");
  917. #ifdef _WIN32
  918. bool enableNewSocketLoop = config_get_bool(main->Config(), "Output",
  919. "NewSocketLoopEnable");
  920. bool enableLowLatencyMode =
  921. config_get_bool(main->Config(), "Output", "LowLatencyEnable");
  922. #endif
  923. bool enableDynBitrate =
  924. config_get_bool(main->Config(), "Output", "DynamicBitrate");
  925. OBSDataAutoRelease settings = obs_data_create();
  926. obs_data_set_string(settings, "bind_ip", bindIP);
  927. #ifdef _WIN32
  928. obs_data_set_bool(settings, "new_socket_loop_enabled",
  929. enableNewSocketLoop);
  930. obs_data_set_bool(settings, "low_latency_mode_enabled",
  931. enableLowLatencyMode);
  932. #endif
  933. obs_data_set_bool(settings, "dyn_bitrate", enableDynBitrate);
  934. obs_output_update(streamOutput, settings);
  935. if (!reconnect)
  936. maxRetries = 0;
  937. obs_output_set_delay(streamOutput, useDelay ? delaySec : 0,
  938. preserveDelay ? OBS_OUTPUT_DELAY_PRESERVE : 0);
  939. obs_output_set_reconnect_settings(streamOutput, maxRetries, retryDelay);
  940. SetupVodTrack(service);
  941. if (obs_output_start(streamOutput)) {
  942. return true;
  943. }
  944. const char *error = obs_output_get_last_error(streamOutput);
  945. bool hasLastError = error && *error;
  946. if (hasLastError)
  947. lastError = error;
  948. else
  949. lastError = string();
  950. const char *type = obs_output_get_id(streamOutput);
  951. blog(LOG_WARNING, "Stream output type '%s' failed to start!%s%s", type,
  952. hasLastError ? " Last Error: " : "", hasLastError ? error : "");
  953. return false;
  954. }
  955. void SimpleOutput::UpdateRecording()
  956. {
  957. if (replayBufferActive || recordingActive)
  958. return;
  959. if (usingRecordingPreset) {
  960. if (!ffmpegOutput)
  961. UpdateRecordingSettings();
  962. } else if (!obs_output_active(streamOutput)) {
  963. Update();
  964. }
  965. if (!Active())
  966. SetupOutputs();
  967. if (!ffmpegOutput) {
  968. obs_output_set_video_encoder(fileOutput, videoRecording);
  969. obs_output_set_audio_encoder(fileOutput, audioRecording, 0);
  970. }
  971. if (replayBuffer) {
  972. obs_output_set_video_encoder(replayBuffer, videoRecording);
  973. obs_output_set_audio_encoder(replayBuffer, audioRecording, 0);
  974. }
  975. recordingConfigured = true;
  976. }
  977. bool SimpleOutput::ConfigureRecording(bool updateReplayBuffer)
  978. {
  979. const char *path =
  980. config_get_string(main->Config(), "SimpleOutput", "FilePath");
  981. const char *format =
  982. config_get_string(main->Config(), "SimpleOutput", "RecFormat");
  983. const char *mux = config_get_string(main->Config(), "SimpleOutput",
  984. "MuxerCustom");
  985. bool noSpace = config_get_bool(main->Config(), "SimpleOutput",
  986. "FileNameWithoutSpace");
  987. const char *filenameFormat = config_get_string(main->Config(), "Output",
  988. "FilenameFormatting");
  989. bool overwriteIfExists =
  990. config_get_bool(main->Config(), "Output", "OverwriteIfExists");
  991. const char *rbPrefix = config_get_string(main->Config(), "SimpleOutput",
  992. "RecRBPrefix");
  993. const char *rbSuffix = config_get_string(main->Config(), "SimpleOutput",
  994. "RecRBSuffix");
  995. int rbTime =
  996. config_get_int(main->Config(), "SimpleOutput", "RecRBTime");
  997. int rbSize =
  998. config_get_int(main->Config(), "SimpleOutput", "RecRBSize");
  999. bool is_fragmented = strcmp(format, "fmp4") == 0 ||
  1000. strcmp(format, "fmov") == 0;
  1001. string f;
  1002. string strPath;
  1003. if (is_fragmented)
  1004. ++format;
  1005. OBSDataAutoRelease settings = obs_data_create();
  1006. if (updateReplayBuffer) {
  1007. f = GetFormatString(filenameFormat, rbPrefix, rbSuffix);
  1008. strPath = GetOutputFilename(path, ffmpegOutput ? "avi" : format,
  1009. noSpace, overwriteIfExists,
  1010. f.c_str());
  1011. obs_data_set_string(settings, "directory", path);
  1012. obs_data_set_string(settings, "format", f.c_str());
  1013. obs_data_set_string(settings, "extension", format);
  1014. obs_data_set_bool(settings, "allow_spaces", !noSpace);
  1015. obs_data_set_int(settings, "max_time_sec", rbTime);
  1016. obs_data_set_int(settings, "max_size_mb",
  1017. usingRecordingPreset ? rbSize : 0);
  1018. } else {
  1019. f = GetFormatString(filenameFormat, nullptr, nullptr);
  1020. strPath = GetRecordingFilename(path,
  1021. ffmpegOutput ? "avi" : format,
  1022. noSpace, overwriteIfExists,
  1023. f.c_str(), ffmpegOutput);
  1024. obs_data_set_string(settings, ffmpegOutput ? "url" : "path",
  1025. strPath.c_str());
  1026. }
  1027. // Use fragmented MOV/MP4 if user has not already specified custom movflags
  1028. if (is_fragmented && (!mux || strstr(mux, "movflags") == NULL)) {
  1029. string mux_frag =
  1030. "movflags=frag_keyframe+empty_moov+delay_moov";
  1031. if (mux) {
  1032. mux_frag += " ";
  1033. mux_frag += mux;
  1034. }
  1035. obs_data_set_string(settings, "muxer_settings",
  1036. mux_frag.c_str());
  1037. } else {
  1038. if (is_fragmented)
  1039. blog(LOG_WARNING,
  1040. "User enabled fragmented recording, "
  1041. "but custom muxer settings contained movflags.");
  1042. obs_data_set_string(settings, "muxer_settings", mux);
  1043. }
  1044. if (updateReplayBuffer)
  1045. obs_output_update(replayBuffer, settings);
  1046. else
  1047. obs_output_update(fileOutput, settings);
  1048. return true;
  1049. }
  1050. bool SimpleOutput::StartRecording()
  1051. {
  1052. UpdateRecording();
  1053. if (!ConfigureRecording(false))
  1054. return false;
  1055. if (!obs_output_start(fileOutput)) {
  1056. QString error_reason;
  1057. const char *error = obs_output_get_last_error(fileOutput);
  1058. if (error)
  1059. error_reason = QT_UTF8(error);
  1060. else
  1061. error_reason = QTStr("Output.StartFailedGeneric");
  1062. QMessageBox::critical(main,
  1063. QTStr("Output.StartRecordingFailed"),
  1064. error_reason);
  1065. return false;
  1066. }
  1067. return true;
  1068. }
  1069. bool SimpleOutput::StartReplayBuffer()
  1070. {
  1071. UpdateRecording();
  1072. if (!ConfigureRecording(true))
  1073. return false;
  1074. if (!obs_output_start(replayBuffer)) {
  1075. QMessageBox::critical(main, QTStr("Output.StartReplayFailed"),
  1076. QTStr("Output.StartFailedGeneric"));
  1077. return false;
  1078. }
  1079. return true;
  1080. }
  1081. void SimpleOutput::StopStreaming(bool force)
  1082. {
  1083. if (force)
  1084. obs_output_force_stop(streamOutput);
  1085. else
  1086. obs_output_stop(streamOutput);
  1087. }
  1088. void SimpleOutput::StopRecording(bool force)
  1089. {
  1090. if (force)
  1091. obs_output_force_stop(fileOutput);
  1092. else
  1093. obs_output_stop(fileOutput);
  1094. }
  1095. void SimpleOutput::StopReplayBuffer(bool force)
  1096. {
  1097. if (force)
  1098. obs_output_force_stop(replayBuffer);
  1099. else
  1100. obs_output_stop(replayBuffer);
  1101. }
  1102. bool SimpleOutput::StreamingActive() const
  1103. {
  1104. return obs_output_active(streamOutput);
  1105. }
  1106. bool SimpleOutput::RecordingActive() const
  1107. {
  1108. return obs_output_active(fileOutput);
  1109. }
  1110. bool SimpleOutput::ReplayBufferActive() const
  1111. {
  1112. return obs_output_active(replayBuffer);
  1113. }
  1114. /* ------------------------------------------------------------------------ */
  1115. struct AdvancedOutput : BasicOutputHandler {
  1116. OBSEncoder streamAudioEnc;
  1117. OBSEncoder streamArchiveEnc;
  1118. OBSEncoder recordTrack[MAX_AUDIO_MIXES];
  1119. OBSEncoder videoStreaming;
  1120. OBSEncoder videoRecording;
  1121. bool ffmpegOutput;
  1122. bool ffmpegRecording;
  1123. bool useStreamEncoder;
  1124. bool useStreamAudioEncoder;
  1125. bool usesBitrate = false;
  1126. AdvancedOutput(OBSBasic *main_);
  1127. inline void UpdateStreamSettings();
  1128. inline void UpdateRecordingSettings();
  1129. inline void UpdateAudioSettings();
  1130. virtual void Update() override;
  1131. inline void SetupVodTrack(obs_service_t *service);
  1132. inline void SetupStreaming();
  1133. inline void SetupRecording();
  1134. inline void SetupFFmpeg();
  1135. void SetupOutputs() override;
  1136. int GetAudioBitrate(size_t i, const char *id) const;
  1137. virtual bool SetupStreaming(obs_service_t *service) override;
  1138. virtual bool StartStreaming(obs_service_t *service) override;
  1139. virtual bool StartRecording() override;
  1140. virtual bool StartReplayBuffer() override;
  1141. virtual void StopStreaming(bool force) override;
  1142. virtual void StopRecording(bool force) override;
  1143. virtual void StopReplayBuffer(bool force) override;
  1144. virtual bool StreamingActive() const override;
  1145. virtual bool RecordingActive() const override;
  1146. virtual bool ReplayBufferActive() const override;
  1147. };
  1148. static OBSData GetDataFromJsonFile(const char *jsonFile)
  1149. {
  1150. char fullPath[512];
  1151. OBSDataAutoRelease data = nullptr;
  1152. int ret = GetProfilePath(fullPath, sizeof(fullPath), jsonFile);
  1153. if (ret > 0) {
  1154. BPtr<char> jsonData = os_quick_read_utf8_file(fullPath);
  1155. if (!!jsonData) {
  1156. data = obs_data_create_from_json(jsonData);
  1157. }
  1158. }
  1159. if (!data)
  1160. data = obs_data_create();
  1161. return data.Get();
  1162. }
  1163. static void ApplyEncoderDefaults(OBSData &settings,
  1164. const obs_encoder_t *encoder)
  1165. {
  1166. OBSData dataRet = obs_encoder_get_defaults(encoder);
  1167. obs_data_release(dataRet);
  1168. if (!!settings)
  1169. obs_data_apply(dataRet, settings);
  1170. settings = std::move(dataRet);
  1171. }
  1172. #define ADV_ARCHIVE_NAME "adv_archive_audio"
  1173. #ifdef __APPLE__
  1174. static void translate_macvth264_encoder(const char *&encoder)
  1175. {
  1176. if (strcmp(encoder, "vt_h264_hw") == 0) {
  1177. encoder = "com.apple.videotoolbox.videoencoder.h264.gva";
  1178. } else if (strcmp(encoder, "vt_h264_sw") == 0) {
  1179. encoder = "com.apple.videotoolbox.videoencoder.h264";
  1180. }
  1181. }
  1182. #endif
  1183. AdvancedOutput::AdvancedOutput(OBSBasic *main_) : BasicOutputHandler(main_)
  1184. {
  1185. const char *recType =
  1186. config_get_string(main->Config(), "AdvOut", "RecType");
  1187. const char *streamEncoder =
  1188. config_get_string(main->Config(), "AdvOut", "Encoder");
  1189. const char *streamAudioEncoder =
  1190. config_get_string(main->Config(), "AdvOut", "AudioEncoder");
  1191. const char *recordEncoder =
  1192. config_get_string(main->Config(), "AdvOut", "RecEncoder");
  1193. const char *recAudioEncoder =
  1194. config_get_string(main->Config(), "AdvOut", "RecAudioEncoder");
  1195. #ifdef __APPLE__
  1196. translate_macvth264_encoder(streamEncoder);
  1197. translate_macvth264_encoder(recordEncoder);
  1198. #endif
  1199. ffmpegOutput = astrcmpi(recType, "FFmpeg") == 0;
  1200. ffmpegRecording =
  1201. ffmpegOutput &&
  1202. config_get_bool(main->Config(), "AdvOut", "FFOutputToFile");
  1203. useStreamEncoder = astrcmpi(recordEncoder, "none") == 0;
  1204. useStreamAudioEncoder = astrcmpi(recAudioEncoder, "none") == 0;
  1205. OBSData streamEncSettings = GetDataFromJsonFile("streamEncoder.json");
  1206. OBSData recordEncSettings = GetDataFromJsonFile("recordEncoder.json");
  1207. if (ffmpegOutput) {
  1208. fileOutput = obs_output_create(
  1209. "ffmpeg_output", "adv_ffmpeg_output", nullptr, nullptr);
  1210. if (!fileOutput)
  1211. throw "Failed to create recording FFmpeg output "
  1212. "(advanced output)";
  1213. } else {
  1214. bool useReplayBuffer =
  1215. config_get_bool(main->Config(), "AdvOut", "RecRB");
  1216. if (useReplayBuffer) {
  1217. OBSDataAutoRelease hotkey;
  1218. const char *str = config_get_string(
  1219. main->Config(), "Hotkeys", "ReplayBuffer");
  1220. if (str)
  1221. hotkey = obs_data_create_from_json(str);
  1222. else
  1223. hotkey = nullptr;
  1224. replayBuffer = obs_output_create("replay_buffer",
  1225. Str("ReplayBuffer"),
  1226. nullptr, hotkey);
  1227. if (!replayBuffer)
  1228. throw "Failed to create replay buffer output "
  1229. "(simple output)";
  1230. signal_handler_t *signal =
  1231. obs_output_get_signal_handler(replayBuffer);
  1232. startReplayBuffer.Connect(signal, "start",
  1233. OBSStartReplayBuffer, this);
  1234. stopReplayBuffer.Connect(signal, "stop",
  1235. OBSStopReplayBuffer, this);
  1236. replayBufferStopping.Connect(signal, "stopping",
  1237. OBSReplayBufferStopping,
  1238. this);
  1239. replayBufferSaved.Connect(signal, "saved",
  1240. OBSReplayBufferSaved, this);
  1241. }
  1242. fileOutput = obs_output_create(
  1243. "ffmpeg_muxer", "adv_file_output", nullptr, nullptr);
  1244. if (!fileOutput)
  1245. throw "Failed to create recording output "
  1246. "(advanced output)";
  1247. if (!useStreamEncoder) {
  1248. videoRecording = obs_video_encoder_create(
  1249. recordEncoder, "advanced_video_recording",
  1250. recordEncSettings, nullptr);
  1251. if (!videoRecording)
  1252. throw "Failed to create recording video "
  1253. "encoder (advanced output)";
  1254. obs_encoder_release(videoRecording);
  1255. }
  1256. }
  1257. videoStreaming = obs_video_encoder_create(streamEncoder,
  1258. "advanced_video_stream",
  1259. streamEncSettings, nullptr);
  1260. if (!videoStreaming)
  1261. throw "Failed to create streaming video encoder "
  1262. "(advanced output)";
  1263. obs_encoder_release(videoStreaming);
  1264. const char *rate_control = obs_data_get_string(
  1265. useStreamEncoder ? streamEncSettings : recordEncSettings,
  1266. "rate_control");
  1267. if (!rate_control)
  1268. rate_control = "";
  1269. usesBitrate = astrcmpi(rate_control, "CBR") == 0 ||
  1270. astrcmpi(rate_control, "VBR") == 0 ||
  1271. astrcmpi(rate_control, "ABR") == 0;
  1272. for (int i = 0; i < MAX_AUDIO_MIXES; i++) {
  1273. char name[19];
  1274. snprintf(name, sizeof(name), "adv_record_audio_%d", i);
  1275. recordTrack[i] = obs_audio_encoder_create(
  1276. useStreamAudioEncoder ? streamAudioEncoder
  1277. : recAudioEncoder,
  1278. name, nullptr, i, nullptr);
  1279. if (!recordTrack[i]) {
  1280. throw "Failed to create audio encoder "
  1281. "(advanced output)";
  1282. }
  1283. obs_encoder_release(recordTrack[i]);
  1284. }
  1285. std::string id;
  1286. int streamTrack =
  1287. config_get_int(main->Config(), "AdvOut", "TrackIndex") - 1;
  1288. streamAudioEnc = obs_audio_encoder_create(streamAudioEncoder,
  1289. "adv_stream_audio", nullptr,
  1290. streamTrack, nullptr);
  1291. if (!streamAudioEnc)
  1292. throw "Failed to create streaming audio encoder "
  1293. "(advanced output)";
  1294. obs_encoder_release(streamAudioEnc);
  1295. id = "";
  1296. int vodTrack =
  1297. config_get_int(main->Config(), "AdvOut", "VodTrackIndex") - 1;
  1298. streamArchiveEnc = obs_audio_encoder_create(streamAudioEncoder,
  1299. ADV_ARCHIVE_NAME, nullptr,
  1300. vodTrack, nullptr);
  1301. if (!streamArchiveEnc)
  1302. throw "Failed to create archive audio encoder "
  1303. "(advanced output)";
  1304. obs_encoder_release(streamArchiveEnc);
  1305. startRecording.Connect(obs_output_get_signal_handler(fileOutput),
  1306. "start", OBSStartRecording, this);
  1307. stopRecording.Connect(obs_output_get_signal_handler(fileOutput), "stop",
  1308. OBSStopRecording, this);
  1309. recordStopping.Connect(obs_output_get_signal_handler(fileOutput),
  1310. "stopping", OBSRecordStopping, this);
  1311. recordFileChanged.Connect(obs_output_get_signal_handler(fileOutput),
  1312. "file_changed", OBSRecordFileChanged, this);
  1313. }
  1314. void AdvancedOutput::UpdateStreamSettings()
  1315. {
  1316. bool applyServiceSettings = config_get_bool(main->Config(), "AdvOut",
  1317. "ApplyServiceSettings");
  1318. bool enforceBitrate = !config_get_bool(main->Config(), "Stream1",
  1319. "IgnoreRecommended");
  1320. bool dynBitrate =
  1321. config_get_bool(main->Config(), "Output", "DynamicBitrate");
  1322. const char *streamEncoder =
  1323. config_get_string(main->Config(), "AdvOut", "Encoder");
  1324. OBSData settings = GetDataFromJsonFile("streamEncoder.json");
  1325. ApplyEncoderDefaults(settings, videoStreaming);
  1326. if (applyServiceSettings) {
  1327. int bitrate = (int)obs_data_get_int(settings, "bitrate");
  1328. int keyint_sec = (int)obs_data_get_int(settings, "keyint_sec");
  1329. obs_service_apply_encoder_settings(main->GetService(), settings,
  1330. nullptr);
  1331. if (!enforceBitrate)
  1332. obs_data_set_int(settings, "bitrate", bitrate);
  1333. int enforced_keyint_sec =
  1334. (int)obs_data_get_int(settings, "keyint_sec");
  1335. if (keyint_sec != 0 && keyint_sec < enforced_keyint_sec)
  1336. obs_data_set_int(settings, "keyint_sec", keyint_sec);
  1337. }
  1338. if (dynBitrate && astrcmpi(streamEncoder, "jim_nvenc") == 0)
  1339. obs_data_set_bool(settings, "lookahead", false);
  1340. video_t *video = obs_get_video();
  1341. enum video_format format = video_output_get_format(video);
  1342. switch (format) {
  1343. case VIDEO_FORMAT_I420:
  1344. case VIDEO_FORMAT_NV12:
  1345. case VIDEO_FORMAT_I010:
  1346. case VIDEO_FORMAT_P010:
  1347. break;
  1348. default:
  1349. obs_encoder_set_preferred_video_format(videoStreaming,
  1350. VIDEO_FORMAT_NV12);
  1351. }
  1352. obs_encoder_update(videoStreaming, settings);
  1353. }
  1354. inline void AdvancedOutput::UpdateRecordingSettings()
  1355. {
  1356. OBSData settings = GetDataFromJsonFile("recordEncoder.json");
  1357. obs_encoder_update(videoRecording, settings);
  1358. }
  1359. void AdvancedOutput::Update()
  1360. {
  1361. UpdateStreamSettings();
  1362. if (!useStreamEncoder && !ffmpegOutput)
  1363. UpdateRecordingSettings();
  1364. UpdateAudioSettings();
  1365. }
  1366. static inline bool ServiceSupportsVodTrack(const char *service)
  1367. {
  1368. static const char *vodTrackServices[] = {"Twitch"};
  1369. for (const char *vodTrackService : vodTrackServices) {
  1370. if (astrcmpi(vodTrackService, service) == 0)
  1371. return true;
  1372. }
  1373. return false;
  1374. }
  1375. inline void AdvancedOutput::SetupStreaming()
  1376. {
  1377. bool rescale = config_get_bool(main->Config(), "AdvOut", "Rescale");
  1378. const char *rescaleRes =
  1379. config_get_string(main->Config(), "AdvOut", "RescaleRes");
  1380. unsigned int cx = 0;
  1381. unsigned int cy = 0;
  1382. if (rescale && rescaleRes && *rescaleRes) {
  1383. if (sscanf(rescaleRes, "%ux%u", &cx, &cy) != 2) {
  1384. cx = 0;
  1385. cy = 0;
  1386. }
  1387. }
  1388. obs_output_set_audio_encoder(streamOutput, streamAudioEnc, 0);
  1389. obs_encoder_set_scaled_size(videoStreaming, cx, cy);
  1390. const char *id = obs_service_get_id(main->GetService());
  1391. if (strcmp(id, "rtmp_custom") == 0) {
  1392. OBSDataAutoRelease settings = obs_data_create();
  1393. obs_service_apply_encoder_settings(main->GetService(), settings,
  1394. nullptr);
  1395. obs_encoder_update(videoStreaming, settings);
  1396. }
  1397. }
  1398. inline void AdvancedOutput::SetupRecording()
  1399. {
  1400. const char *path =
  1401. config_get_string(main->Config(), "AdvOut", "RecFilePath");
  1402. const char *mux =
  1403. config_get_string(main->Config(), "AdvOut", "RecMuxerCustom");
  1404. bool rescale = config_get_bool(main->Config(), "AdvOut", "RecRescale");
  1405. const char *rescaleRes =
  1406. config_get_string(main->Config(), "AdvOut", "RecRescaleRes");
  1407. int tracks;
  1408. const char *recFormat =
  1409. config_get_string(main->Config(), "AdvOut", "RecFormat");
  1410. bool is_fragmented = strcmp(recFormat, "fmp4") == 0 ||
  1411. strcmp(recFormat, "fmov") == 0;
  1412. bool flv = strcmp(recFormat, "flv") == 0;
  1413. if (flv)
  1414. tracks = config_get_int(main->Config(), "AdvOut", "FLVTrack");
  1415. else
  1416. tracks = config_get_int(main->Config(), "AdvOut", "RecTracks");
  1417. OBSDataAutoRelease settings = obs_data_create();
  1418. unsigned int cx = 0;
  1419. unsigned int cy = 0;
  1420. int idx = 0;
  1421. if (tracks == 0)
  1422. tracks = config_get_int(main->Config(), "AdvOut", "TrackIndex");
  1423. if (useStreamEncoder) {
  1424. obs_output_set_video_encoder(fileOutput, videoStreaming);
  1425. if (replayBuffer)
  1426. obs_output_set_video_encoder(replayBuffer,
  1427. videoStreaming);
  1428. } else {
  1429. if (rescale && rescaleRes && *rescaleRes) {
  1430. if (sscanf(rescaleRes, "%ux%u", &cx, &cy) != 2) {
  1431. cx = 0;
  1432. cy = 0;
  1433. }
  1434. }
  1435. obs_encoder_set_scaled_size(videoRecording, cx, cy);
  1436. obs_output_set_video_encoder(fileOutput, videoRecording);
  1437. if (replayBuffer)
  1438. obs_output_set_video_encoder(replayBuffer,
  1439. videoRecording);
  1440. }
  1441. if (!flv) {
  1442. for (int i = 0; i < MAX_AUDIO_MIXES; i++) {
  1443. if ((tracks & (1 << i)) != 0) {
  1444. obs_output_set_audio_encoder(
  1445. fileOutput, recordTrack[i], idx);
  1446. if (replayBuffer)
  1447. obs_output_set_audio_encoder(
  1448. replayBuffer, recordTrack[i],
  1449. idx);
  1450. idx++;
  1451. }
  1452. }
  1453. } else if (flv && tracks != 0) {
  1454. obs_output_set_audio_encoder(fileOutput,
  1455. recordTrack[tracks - 1], idx);
  1456. if (replayBuffer)
  1457. obs_output_set_audio_encoder(
  1458. replayBuffer, recordTrack[tracks - 1], idx);
  1459. }
  1460. // Use fragmented MOV/MP4 if user has not already specified custom movflags
  1461. if (is_fragmented && (!mux || strstr(mux, "movflags") == NULL)) {
  1462. string mux_fmp4 =
  1463. "movflags=frag_keyframe+empty_moov+delay_moov";
  1464. if (mux) {
  1465. mux_fmp4 += " ";
  1466. mux_fmp4 += mux;
  1467. }
  1468. obs_data_set_string(settings, "muxer_settings",
  1469. mux_fmp4.c_str());
  1470. } else {
  1471. if (is_fragmented)
  1472. blog(LOG_WARNING,
  1473. "User enabled fragmented recording, "
  1474. "but custom muxer settings contained movflags.");
  1475. obs_data_set_string(settings, "muxer_settings", mux);
  1476. }
  1477. obs_data_set_string(settings, "path", path);
  1478. obs_output_update(fileOutput, settings);
  1479. if (replayBuffer)
  1480. obs_output_update(replayBuffer, settings);
  1481. }
  1482. inline void AdvancedOutput::SetupFFmpeg()
  1483. {
  1484. const char *url = config_get_string(main->Config(), "AdvOut", "FFURL");
  1485. int vBitrate = config_get_int(main->Config(), "AdvOut", "FFVBitrate");
  1486. int gopSize = config_get_int(main->Config(), "AdvOut", "FFVGOPSize");
  1487. bool rescale = config_get_bool(main->Config(), "AdvOut", "FFRescale");
  1488. const char *rescaleRes =
  1489. config_get_string(main->Config(), "AdvOut", "FFRescaleRes");
  1490. const char *formatName =
  1491. config_get_string(main->Config(), "AdvOut", "FFFormat");
  1492. const char *mimeType =
  1493. config_get_string(main->Config(), "AdvOut", "FFFormatMimeType");
  1494. const char *muxCustom =
  1495. config_get_string(main->Config(), "AdvOut", "FFMCustom");
  1496. const char *vEncoder =
  1497. config_get_string(main->Config(), "AdvOut", "FFVEncoder");
  1498. int vEncoderId =
  1499. config_get_int(main->Config(), "AdvOut", "FFVEncoderId");
  1500. const char *vEncCustom =
  1501. config_get_string(main->Config(), "AdvOut", "FFVCustom");
  1502. int aBitrate = config_get_int(main->Config(), "AdvOut", "FFABitrate");
  1503. int aMixes = config_get_int(main->Config(), "AdvOut", "FFAudioMixes");
  1504. const char *aEncoder =
  1505. config_get_string(main->Config(), "AdvOut", "FFAEncoder");
  1506. int aEncoderId =
  1507. config_get_int(main->Config(), "AdvOut", "FFAEncoderId");
  1508. const char *aEncCustom =
  1509. config_get_string(main->Config(), "AdvOut", "FFACustom");
  1510. OBSDataAutoRelease settings = obs_data_create();
  1511. obs_data_set_string(settings, "url", url);
  1512. obs_data_set_string(settings, "format_name", formatName);
  1513. obs_data_set_string(settings, "format_mime_type", mimeType);
  1514. obs_data_set_string(settings, "muxer_settings", muxCustom);
  1515. obs_data_set_int(settings, "gop_size", gopSize);
  1516. obs_data_set_int(settings, "video_bitrate", vBitrate);
  1517. obs_data_set_string(settings, "video_encoder", vEncoder);
  1518. obs_data_set_int(settings, "video_encoder_id", vEncoderId);
  1519. obs_data_set_string(settings, "video_settings", vEncCustom);
  1520. obs_data_set_int(settings, "audio_bitrate", aBitrate);
  1521. obs_data_set_string(settings, "audio_encoder", aEncoder);
  1522. obs_data_set_int(settings, "audio_encoder_id", aEncoderId);
  1523. obs_data_set_string(settings, "audio_settings", aEncCustom);
  1524. if (rescale && rescaleRes && *rescaleRes) {
  1525. int width;
  1526. int height;
  1527. int val = sscanf(rescaleRes, "%dx%d", &width, &height);
  1528. if (val == 2 && width && height) {
  1529. obs_data_set_int(settings, "scale_width", width);
  1530. obs_data_set_int(settings, "scale_height", height);
  1531. }
  1532. }
  1533. obs_output_set_mixers(fileOutput, aMixes);
  1534. obs_output_set_media(fileOutput, obs_get_video(), obs_get_audio());
  1535. obs_output_update(fileOutput, settings);
  1536. }
  1537. static inline void SetEncoderName(obs_encoder_t *encoder, const char *name,
  1538. const char *defaultName)
  1539. {
  1540. obs_encoder_set_name(encoder, (name && *name) ? name : defaultName);
  1541. }
  1542. inline void AdvancedOutput::UpdateAudioSettings()
  1543. {
  1544. bool applyServiceSettings = config_get_bool(main->Config(), "AdvOut",
  1545. "ApplyServiceSettings");
  1546. bool enforceBitrate = !config_get_bool(main->Config(), "Stream1",
  1547. "IgnoreRecommended");
  1548. int streamTrackIndex =
  1549. config_get_int(main->Config(), "AdvOut", "TrackIndex");
  1550. int vodTrackIndex =
  1551. config_get_int(main->Config(), "AdvOut", "VodTrackIndex");
  1552. const char *audioEncoder =
  1553. config_get_string(main->Config(), "AdvOut", "AudioEncoder");
  1554. const char *recAudioEncoder =
  1555. config_get_string(main->Config(), "AdvOut", "RecAudioEncoder");
  1556. OBSDataAutoRelease settings[MAX_AUDIO_MIXES];
  1557. for (size_t i = 0; i < MAX_AUDIO_MIXES; i++) {
  1558. string cfg_name = "Track";
  1559. cfg_name += to_string((int)i + 1);
  1560. cfg_name += "Name";
  1561. const char *name = config_get_string(main->Config(), "AdvOut",
  1562. cfg_name.c_str());
  1563. string def_name = "Track";
  1564. def_name += to_string((int)i + 1);
  1565. SetEncoderName(recordTrack[i], name, def_name.c_str());
  1566. }
  1567. for (size_t i = 0; i < MAX_AUDIO_MIXES; i++) {
  1568. int track = (int)(i + 1);
  1569. settings[i] = obs_data_create();
  1570. obs_data_set_int(settings[i], "bitrate",
  1571. GetAudioBitrate(i, recAudioEncoder));
  1572. obs_encoder_update(recordTrack[i], settings[i]);
  1573. obs_data_set_int(settings[i], "bitrate",
  1574. GetAudioBitrate(i, audioEncoder));
  1575. if (track == streamTrackIndex || track == vodTrackIndex) {
  1576. if (applyServiceSettings) {
  1577. int bitrate = (int)obs_data_get_int(settings[i],
  1578. "bitrate");
  1579. obs_service_apply_encoder_settings(
  1580. main->GetService(), nullptr,
  1581. settings[i]);
  1582. if (!enforceBitrate)
  1583. obs_data_set_int(settings[i], "bitrate",
  1584. bitrate);
  1585. }
  1586. }
  1587. if (track == streamTrackIndex)
  1588. obs_encoder_update(streamAudioEnc, settings[i]);
  1589. if (track == vodTrackIndex)
  1590. obs_encoder_update(streamArchiveEnc, settings[i]);
  1591. }
  1592. }
  1593. void AdvancedOutput::SetupOutputs()
  1594. {
  1595. obs_encoder_set_video(videoStreaming, obs_get_video());
  1596. if (videoRecording)
  1597. obs_encoder_set_video(videoRecording, obs_get_video());
  1598. for (size_t i = 0; i < MAX_AUDIO_MIXES; i++)
  1599. obs_encoder_set_audio(recordTrack[i], obs_get_audio());
  1600. obs_encoder_set_audio(streamAudioEnc, obs_get_audio());
  1601. obs_encoder_set_audio(streamArchiveEnc, obs_get_audio());
  1602. SetupStreaming();
  1603. if (ffmpegOutput)
  1604. SetupFFmpeg();
  1605. else
  1606. SetupRecording();
  1607. }
  1608. int AdvancedOutput::GetAudioBitrate(size_t i, const char *id) const
  1609. {
  1610. static const char *names[] = {
  1611. "Track1Bitrate", "Track2Bitrate", "Track3Bitrate",
  1612. "Track4Bitrate", "Track5Bitrate", "Track6Bitrate",
  1613. };
  1614. int bitrate = (int)config_get_uint(main->Config(), "AdvOut", names[i]);
  1615. return FindClosestAvailableAudioBitrate(id, bitrate);
  1616. }
  1617. inline void AdvancedOutput::SetupVodTrack(obs_service_t *service)
  1618. {
  1619. int streamTrack =
  1620. config_get_int(main->Config(), "AdvOut", "TrackIndex");
  1621. bool vodTrackEnabled =
  1622. config_get_bool(main->Config(), "AdvOut", "VodTrackEnabled");
  1623. int vodTrackIndex =
  1624. config_get_int(main->Config(), "AdvOut", "VodTrackIndex");
  1625. bool enableForCustomServer = config_get_bool(
  1626. GetGlobalConfig(), "General", "EnableCustomServerVodTrack");
  1627. const char *id = obs_service_get_id(service);
  1628. if (strcmp(id, "rtmp_custom") == 0) {
  1629. vodTrackEnabled = enableForCustomServer ? vodTrackEnabled
  1630. : false;
  1631. } else {
  1632. OBSDataAutoRelease settings = obs_service_get_settings(service);
  1633. const char *service = obs_data_get_string(settings, "service");
  1634. if (!ServiceSupportsVodTrack(service))
  1635. vodTrackEnabled = false;
  1636. }
  1637. if (vodTrackEnabled && streamTrack != vodTrackIndex)
  1638. obs_output_set_audio_encoder(streamOutput, streamArchiveEnc, 1);
  1639. else
  1640. clear_archive_encoder(streamOutput, ADV_ARCHIVE_NAME);
  1641. }
  1642. bool AdvancedOutput::SetupStreaming(obs_service_t *service)
  1643. {
  1644. if (!useStreamEncoder ||
  1645. (!ffmpegOutput && !obs_output_active(fileOutput))) {
  1646. UpdateStreamSettings();
  1647. }
  1648. UpdateAudioSettings();
  1649. if (!Active())
  1650. SetupOutputs();
  1651. Auth *auth = main->GetAuth();
  1652. if (auth)
  1653. auth->OnStreamConfig();
  1654. /* --------------------- */
  1655. const char *type = GetStreamOutputType(service);
  1656. if (!type)
  1657. return false;
  1658. /* XXX: this is messy and disgusting and should be refactored */
  1659. if (outputType != type) {
  1660. streamDelayStarting.Disconnect();
  1661. streamStopping.Disconnect();
  1662. startStreaming.Disconnect();
  1663. stopStreaming.Disconnect();
  1664. streamOutput =
  1665. obs_output_create(type, "adv_stream", nullptr, nullptr);
  1666. if (!streamOutput) {
  1667. blog(LOG_WARNING,
  1668. "Creation of stream output type '%s' "
  1669. "failed!",
  1670. type);
  1671. return false;
  1672. }
  1673. streamDelayStarting.Connect(
  1674. obs_output_get_signal_handler(streamOutput), "starting",
  1675. OBSStreamStarting, this);
  1676. streamStopping.Connect(
  1677. obs_output_get_signal_handler(streamOutput), "stopping",
  1678. OBSStreamStopping, this);
  1679. startStreaming.Connect(
  1680. obs_output_get_signal_handler(streamOutput), "start",
  1681. OBSStartStreaming, this);
  1682. stopStreaming.Connect(
  1683. obs_output_get_signal_handler(streamOutput), "stop",
  1684. OBSStopStreaming, this);
  1685. outputType = type;
  1686. }
  1687. obs_output_set_video_encoder(streamOutput, videoStreaming);
  1688. obs_output_set_audio_encoder(streamOutput, streamAudioEnc, 0);
  1689. return true;
  1690. }
  1691. bool AdvancedOutput::StartStreaming(obs_service_t *service)
  1692. {
  1693. obs_output_set_service(streamOutput, service);
  1694. bool reconnect = config_get_bool(main->Config(), "Output", "Reconnect");
  1695. int retryDelay = config_get_int(main->Config(), "Output", "RetryDelay");
  1696. int maxRetries = config_get_int(main->Config(), "Output", "MaxRetries");
  1697. bool useDelay =
  1698. config_get_bool(main->Config(), "Output", "DelayEnable");
  1699. int delaySec = config_get_int(main->Config(), "Output", "DelaySec");
  1700. bool preserveDelay =
  1701. config_get_bool(main->Config(), "Output", "DelayPreserve");
  1702. const char *bindIP =
  1703. config_get_string(main->Config(), "Output", "BindIP");
  1704. #ifdef _WIN32
  1705. bool enableNewSocketLoop = config_get_bool(main->Config(), "Output",
  1706. "NewSocketLoopEnable");
  1707. bool enableLowLatencyMode =
  1708. config_get_bool(main->Config(), "Output", "LowLatencyEnable");
  1709. #endif
  1710. bool enableDynBitrate =
  1711. config_get_bool(main->Config(), "Output", "DynamicBitrate");
  1712. OBSDataAutoRelease settings = obs_data_create();
  1713. obs_data_set_string(settings, "bind_ip", bindIP);
  1714. #ifdef _WIN32
  1715. obs_data_set_bool(settings, "new_socket_loop_enabled",
  1716. enableNewSocketLoop);
  1717. obs_data_set_bool(settings, "low_latency_mode_enabled",
  1718. enableLowLatencyMode);
  1719. #endif
  1720. obs_data_set_bool(settings, "dyn_bitrate", enableDynBitrate);
  1721. obs_output_update(streamOutput, settings);
  1722. if (!reconnect)
  1723. maxRetries = 0;
  1724. obs_output_set_delay(streamOutput, useDelay ? delaySec : 0,
  1725. preserveDelay ? OBS_OUTPUT_DELAY_PRESERVE : 0);
  1726. obs_output_set_reconnect_settings(streamOutput, maxRetries, retryDelay);
  1727. SetupVodTrack(service);
  1728. if (obs_output_start(streamOutput)) {
  1729. return true;
  1730. }
  1731. const char *error = obs_output_get_last_error(streamOutput);
  1732. bool hasLastError = error && *error;
  1733. if (hasLastError)
  1734. lastError = error;
  1735. else
  1736. lastError = string();
  1737. const char *type = obs_output_get_id(streamOutput);
  1738. blog(LOG_WARNING, "Stream output type '%s' failed to start!%s%s", type,
  1739. hasLastError ? " Last Error: " : "", hasLastError ? error : "");
  1740. return false;
  1741. }
  1742. bool AdvancedOutput::StartRecording()
  1743. {
  1744. const char *path;
  1745. const char *recFormat;
  1746. const char *filenameFormat;
  1747. bool noSpace = false;
  1748. bool overwriteIfExists = false;
  1749. bool splitFile;
  1750. const char *splitFileType;
  1751. int splitFileTime;
  1752. int splitFileSize;
  1753. if (!useStreamEncoder) {
  1754. if (!ffmpegOutput) {
  1755. UpdateRecordingSettings();
  1756. }
  1757. } else if (!obs_output_active(streamOutput)) {
  1758. UpdateStreamSettings();
  1759. }
  1760. UpdateAudioSettings();
  1761. if (!Active())
  1762. SetupOutputs();
  1763. if (!ffmpegOutput || ffmpegRecording) {
  1764. path = config_get_string(main->Config(), "AdvOut",
  1765. ffmpegRecording ? "FFFilePath"
  1766. : "RecFilePath");
  1767. recFormat = config_get_string(main->Config(), "AdvOut",
  1768. ffmpegRecording ? "FFExtension"
  1769. : "RecFormat");
  1770. filenameFormat = config_get_string(main->Config(), "Output",
  1771. "FilenameFormatting");
  1772. overwriteIfExists = config_get_bool(main->Config(), "Output",
  1773. "OverwriteIfExists");
  1774. noSpace = config_get_bool(main->Config(), "AdvOut",
  1775. ffmpegRecording
  1776. ? "FFFileNameWithoutSpace"
  1777. : "RecFileNameWithoutSpace");
  1778. splitFile = config_get_bool(main->Config(), "AdvOut",
  1779. "RecSplitFile");
  1780. // Strip leading "f" in case fragmented format was selected
  1781. if (strcmp(recFormat, "fmp4") == 0 ||
  1782. strcmp(recFormat, "fmov") == 0)
  1783. ++recFormat;
  1784. string strPath = GetRecordingFilename(path, recFormat, noSpace,
  1785. overwriteIfExists,
  1786. filenameFormat,
  1787. ffmpegRecording);
  1788. OBSDataAutoRelease settings = obs_data_create();
  1789. obs_data_set_string(settings, ffmpegRecording ? "url" : "path",
  1790. strPath.c_str());
  1791. if (splitFile) {
  1792. splitFileType = config_get_string(
  1793. main->Config(), "AdvOut", "RecSplitFileType");
  1794. splitFileTime =
  1795. (astrcmpi(splitFileType, "Time") == 0)
  1796. ? config_get_int(main->Config(),
  1797. "AdvOut",
  1798. "RecSplitFileTime")
  1799. : 0;
  1800. splitFileSize =
  1801. (astrcmpi(splitFileType, "Size") == 0)
  1802. ? config_get_int(main->Config(),
  1803. "AdvOut",
  1804. "RecSplitFileSize")
  1805. : 0;
  1806. obs_data_set_string(settings, "directory", path);
  1807. obs_data_set_string(settings, "format", filenameFormat);
  1808. obs_data_set_string(settings, "extension", recFormat);
  1809. obs_data_set_bool(settings, "allow_spaces", !noSpace);
  1810. obs_data_set_bool(settings, "allow_overwrite",
  1811. overwriteIfExists);
  1812. obs_data_set_bool(settings, "split_file", true);
  1813. obs_data_set_int(settings, "max_time_sec",
  1814. splitFileTime * 60);
  1815. obs_data_set_int(settings, "max_size_mb",
  1816. splitFileSize);
  1817. }
  1818. obs_output_update(fileOutput, settings);
  1819. }
  1820. if (!obs_output_start(fileOutput)) {
  1821. QString error_reason;
  1822. const char *error = obs_output_get_last_error(fileOutput);
  1823. if (error)
  1824. error_reason = QT_UTF8(error);
  1825. else
  1826. error_reason = QTStr("Output.StartFailedGeneric");
  1827. QMessageBox::critical(main,
  1828. QTStr("Output.StartRecordingFailed"),
  1829. error_reason);
  1830. return false;
  1831. }
  1832. return true;
  1833. }
  1834. bool AdvancedOutput::StartReplayBuffer()
  1835. {
  1836. const char *path;
  1837. const char *recFormat;
  1838. const char *filenameFormat;
  1839. bool noSpace = false;
  1840. bool overwriteIfExists = false;
  1841. const char *rbPrefix;
  1842. const char *rbSuffix;
  1843. int rbTime;
  1844. int rbSize;
  1845. if (!useStreamEncoder) {
  1846. if (!ffmpegOutput)
  1847. UpdateRecordingSettings();
  1848. } else if (!obs_output_active(streamOutput)) {
  1849. UpdateStreamSettings();
  1850. }
  1851. UpdateAudioSettings();
  1852. if (!Active())
  1853. SetupOutputs();
  1854. if (!ffmpegOutput || ffmpegRecording) {
  1855. path = config_get_string(main->Config(), "AdvOut",
  1856. ffmpegRecording ? "FFFilePath"
  1857. : "RecFilePath");
  1858. recFormat = config_get_string(main->Config(), "AdvOut",
  1859. ffmpegRecording ? "FFExtension"
  1860. : "RecFormat");
  1861. filenameFormat = config_get_string(main->Config(), "Output",
  1862. "FilenameFormatting");
  1863. overwriteIfExists = config_get_bool(main->Config(), "Output",
  1864. "OverwriteIfExists");
  1865. noSpace = config_get_bool(main->Config(), "AdvOut",
  1866. ffmpegRecording
  1867. ? "FFFileNameWithoutSpace"
  1868. : "RecFileNameWithoutSpace");
  1869. rbPrefix = config_get_string(main->Config(), "SimpleOutput",
  1870. "RecRBPrefix");
  1871. rbSuffix = config_get_string(main->Config(), "SimpleOutput",
  1872. "RecRBSuffix");
  1873. rbTime = config_get_int(main->Config(), "AdvOut", "RecRBTime");
  1874. rbSize = config_get_int(main->Config(), "AdvOut", "RecRBSize");
  1875. string f = GetFormatString(filenameFormat, rbPrefix, rbSuffix);
  1876. string strPath = GetOutputFilename(
  1877. path, recFormat, noSpace, overwriteIfExists, f.c_str());
  1878. OBSDataAutoRelease settings = obs_data_create();
  1879. obs_data_set_string(settings, "directory", path);
  1880. obs_data_set_string(settings, "format", f.c_str());
  1881. obs_data_set_string(settings, "extension", recFormat);
  1882. obs_data_set_bool(settings, "allow_spaces", !noSpace);
  1883. obs_data_set_int(settings, "max_time_sec", rbTime);
  1884. obs_data_set_int(settings, "max_size_mb",
  1885. usesBitrate ? 0 : rbSize);
  1886. obs_output_update(replayBuffer, settings);
  1887. }
  1888. if (!obs_output_start(replayBuffer)) {
  1889. QString error_reason;
  1890. const char *error = obs_output_get_last_error(replayBuffer);
  1891. if (error)
  1892. error_reason = QT_UTF8(error);
  1893. else
  1894. error_reason = QTStr("Output.StartFailedGeneric");
  1895. QMessageBox::critical(main, QTStr("Output.StartReplayFailed"),
  1896. error_reason);
  1897. return false;
  1898. }
  1899. return true;
  1900. }
  1901. void AdvancedOutput::StopStreaming(bool force)
  1902. {
  1903. if (force)
  1904. obs_output_force_stop(streamOutput);
  1905. else
  1906. obs_output_stop(streamOutput);
  1907. }
  1908. void AdvancedOutput::StopRecording(bool force)
  1909. {
  1910. if (force)
  1911. obs_output_force_stop(fileOutput);
  1912. else
  1913. obs_output_stop(fileOutput);
  1914. }
  1915. void AdvancedOutput::StopReplayBuffer(bool force)
  1916. {
  1917. if (force)
  1918. obs_output_force_stop(replayBuffer);
  1919. else
  1920. obs_output_stop(replayBuffer);
  1921. }
  1922. bool AdvancedOutput::StreamingActive() const
  1923. {
  1924. return obs_output_active(streamOutput);
  1925. }
  1926. bool AdvancedOutput::RecordingActive() const
  1927. {
  1928. return obs_output_active(fileOutput);
  1929. }
  1930. bool AdvancedOutput::ReplayBufferActive() const
  1931. {
  1932. return obs_output_active(replayBuffer);
  1933. }
  1934. /* ------------------------------------------------------------------------ */
  1935. void BasicOutputHandler::SetupAutoRemux(const char *&ext)
  1936. {
  1937. bool autoRemux = config_get_bool(main->Config(), "Video", "AutoRemux");
  1938. if (autoRemux && strcmp(ext, "mp4") == 0)
  1939. ext = "mkv";
  1940. }
  1941. std::string
  1942. BasicOutputHandler::GetRecordingFilename(const char *path, const char *ext,
  1943. bool noSpace, bool overwrite,
  1944. const char *format, bool ffmpeg)
  1945. {
  1946. if (!ffmpeg)
  1947. SetupAutoRemux(ext);
  1948. string dst = GetOutputFilename(path, ext, noSpace, overwrite, format);
  1949. lastRecordingPath = dst;
  1950. return dst;
  1951. }
  1952. BasicOutputHandler *CreateSimpleOutputHandler(OBSBasic *main)
  1953. {
  1954. return new SimpleOutput(main);
  1955. }
  1956. BasicOutputHandler *CreateAdvancedOutputHandler(OBSBasic *main)
  1957. {
  1958. return new AdvancedOutput(main);
  1959. }