window-basic-main-outputs.cpp 69 KB

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