window-basic-main-outputs.cpp 42 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485
  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. using namespace std;
  9. static void OBSStreamStarting(void *data, calldata_t *params)
  10. {
  11. BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
  12. obs_output_t *obj = (obs_output_t*)calldata_ptr(params, "output");
  13. int sec = (int)obs_output_get_active_delay(obj);
  14. if (sec == 0)
  15. return;
  16. output->delayActive = true;
  17. QMetaObject::invokeMethod(output->main,
  18. "StreamDelayStarting", Q_ARG(int, sec));
  19. }
  20. static void OBSStreamStopping(void *data, calldata_t *params)
  21. {
  22. BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
  23. obs_output_t *obj = (obs_output_t*)calldata_ptr(params, "output");
  24. int sec = (int)obs_output_get_active_delay(obj);
  25. if (sec == 0)
  26. QMetaObject::invokeMethod(output->main, "StreamStopping");
  27. else
  28. QMetaObject::invokeMethod(output->main,
  29. "StreamDelayStopping", Q_ARG(int, sec));
  30. }
  31. static void OBSStartStreaming(void *data, calldata_t *params)
  32. {
  33. BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
  34. output->streamingActive = true;
  35. QMetaObject::invokeMethod(output->main, "StreamingStart");
  36. UNUSED_PARAMETER(params);
  37. }
  38. static void OBSStopStreaming(void *data, calldata_t *params)
  39. {
  40. BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
  41. int code = (int)calldata_int(params, "code");
  42. const char *last_error = calldata_string(params, "last_error");
  43. QString arg_last_error = QString::fromUtf8(last_error);
  44. output->streamingActive = false;
  45. output->delayActive = false;
  46. QMetaObject::invokeMethod(output->main,
  47. "StreamingStop", Q_ARG(int, code), Q_ARG(QString, arg_last_error));
  48. }
  49. static void OBSStartRecording(void *data, calldata_t *params)
  50. {
  51. BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
  52. output->recordingActive = true;
  53. QMetaObject::invokeMethod(output->main, "RecordingStart");
  54. UNUSED_PARAMETER(params);
  55. }
  56. static void OBSStopRecording(void *data, calldata_t *params)
  57. {
  58. BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
  59. int code = (int)calldata_int(params, "code");
  60. output->recordingActive = false;
  61. QMetaObject::invokeMethod(output->main,
  62. "RecordingStop", Q_ARG(int, code));
  63. UNUSED_PARAMETER(params);
  64. }
  65. static void OBSRecordStopping(void *data, calldata_t *params)
  66. {
  67. BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
  68. QMetaObject::invokeMethod(output->main, "RecordStopping");
  69. UNUSED_PARAMETER(params);
  70. }
  71. static void OBSStartReplayBuffer(void *data, calldata_t *params)
  72. {
  73. BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
  74. output->replayBufferActive = true;
  75. QMetaObject::invokeMethod(output->main, "ReplayBufferStart");
  76. UNUSED_PARAMETER(params);
  77. }
  78. static void OBSStopReplayBuffer(void *data, calldata_t *params)
  79. {
  80. BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
  81. int code = (int)calldata_int(params, "code");
  82. output->replayBufferActive = false;
  83. QMetaObject::invokeMethod(output->main,
  84. "ReplayBufferStop", Q_ARG(int, code));
  85. UNUSED_PARAMETER(params);
  86. }
  87. static void OBSReplayBufferStopping(void *data, calldata_t *params)
  88. {
  89. BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
  90. QMetaObject::invokeMethod(output->main, "ReplayBufferStopping");
  91. UNUSED_PARAMETER(params);
  92. }
  93. static void FindBestFilename(string &strPath, bool noSpace)
  94. {
  95. int num = 2;
  96. if (!os_file_exists(strPath.c_str()))
  97. return;
  98. const char *ext = strrchr(strPath.c_str(), '.');
  99. if (!ext)
  100. return;
  101. int extStart = int(ext - strPath.c_str());
  102. for (;;) {
  103. string testPath = strPath;
  104. string numStr;
  105. numStr = noSpace ? "_" : " (";
  106. numStr += to_string(num++);
  107. if (!noSpace)
  108. numStr += ")";
  109. testPath.insert(extStart, numStr);
  110. if (!os_file_exists(testPath.c_str())) {
  111. strPath = testPath;
  112. break;
  113. }
  114. }
  115. }
  116. /* ------------------------------------------------------------------------ */
  117. static bool CreateAACEncoder(OBSEncoder &res, string &id, int bitrate,
  118. const char *name, size_t idx)
  119. {
  120. const char *id_ = GetAACEncoderForBitrate(bitrate);
  121. if (!id_) {
  122. id.clear();
  123. res = nullptr;
  124. return false;
  125. }
  126. if (id == id_)
  127. return true;
  128. id = id_;
  129. res = obs_audio_encoder_create(id_, name, nullptr, idx, nullptr);
  130. if (res) {
  131. obs_encoder_release(res);
  132. return true;
  133. }
  134. return false;
  135. }
  136. /* ------------------------------------------------------------------------ */
  137. struct SimpleOutput : BasicOutputHandler {
  138. OBSEncoder aacStreaming;
  139. OBSEncoder h264Streaming;
  140. OBSEncoder aacRecording;
  141. OBSEncoder h264Recording;
  142. string aacRecEncID;
  143. string aacStreamEncID;
  144. string videoEncoder;
  145. string videoQuality;
  146. bool usingRecordingPreset = false;
  147. bool recordingConfigured = false;
  148. bool ffmpegOutput = false;
  149. bool lowCPUx264 = false;
  150. SimpleOutput(OBSBasic *main_);
  151. int CalcCRF(int crf);
  152. void UpdateStreamingSettings_amd(obs_data_t *settings, int bitrate);
  153. void UpdateRecordingSettings_x264_crf(int crf);
  154. void UpdateRecordingSettings_qsv11(int crf);
  155. void UpdateRecordingSettings_nvenc(int cqp);
  156. void UpdateRecordingSettings_amd_cqp(int cqp);
  157. void UpdateRecordingSettings();
  158. void UpdateRecordingAudioSettings();
  159. virtual void Update() override;
  160. void SetupOutputs();
  161. int GetAudioBitrate() const;
  162. void LoadRecordingPreset_h264(const char *encoder);
  163. void LoadRecordingPreset_Lossless();
  164. void LoadRecordingPreset();
  165. void LoadStreamingPreset_h264(const char *encoder);
  166. void UpdateRecording();
  167. bool ConfigureRecording(bool useReplayBuffer);
  168. virtual bool StartStreaming(obs_service_t *service) override;
  169. virtual bool StartRecording() override;
  170. virtual bool StartReplayBuffer() override;
  171. virtual void StopStreaming(bool force) override;
  172. virtual void StopRecording(bool force) override;
  173. virtual void StopReplayBuffer(bool force) override;
  174. virtual bool StreamingActive() const override;
  175. virtual bool RecordingActive() const override;
  176. virtual bool ReplayBufferActive() const override;
  177. };
  178. void SimpleOutput::LoadRecordingPreset_Lossless()
  179. {
  180. fileOutput = obs_output_create("ffmpeg_output",
  181. "simple_ffmpeg_output", nullptr, nullptr);
  182. if (!fileOutput)
  183. throw "Failed to create recording FFmpeg output "
  184. "(simple output)";
  185. obs_output_release(fileOutput);
  186. obs_data_t *settings = obs_data_create();
  187. obs_data_set_string(settings, "format_name", "avi");
  188. obs_data_set_string(settings, "video_encoder", "utvideo");
  189. obs_data_set_string(settings, "audio_encoder", "pcm_s16le");
  190. obs_output_update(fileOutput, settings);
  191. obs_data_release(settings);
  192. }
  193. void SimpleOutput::LoadRecordingPreset_h264(const char *encoderId)
  194. {
  195. h264Recording = obs_video_encoder_create(encoderId,
  196. "simple_h264_recording", nullptr, nullptr);
  197. if (!h264Recording)
  198. throw "Failed to create h264 recording encoder (simple output)";
  199. obs_encoder_release(h264Recording);
  200. }
  201. void SimpleOutput::LoadStreamingPreset_h264(const char *encoderId)
  202. {
  203. h264Streaming = obs_video_encoder_create(encoderId,
  204. "simple_h264_stream", nullptr, nullptr);
  205. if (!h264Streaming)
  206. throw "Failed to create h264 streaming encoder (simple output)";
  207. obs_encoder_release(h264Streaming);
  208. }
  209. void SimpleOutput::LoadRecordingPreset()
  210. {
  211. const char *quality = config_get_string(main->Config(), "SimpleOutput",
  212. "RecQuality");
  213. const char *encoder = config_get_string(main->Config(), "SimpleOutput",
  214. "RecEncoder");
  215. videoEncoder = encoder;
  216. videoQuality = quality;
  217. ffmpegOutput = false;
  218. if (strcmp(quality, "Stream") == 0) {
  219. h264Recording = h264Streaming;
  220. aacRecording = aacStreaming;
  221. usingRecordingPreset = false;
  222. return;
  223. } else if (strcmp(quality, "Lossless") == 0) {
  224. LoadRecordingPreset_Lossless();
  225. usingRecordingPreset = true;
  226. ffmpegOutput = true;
  227. return;
  228. } else {
  229. lowCPUx264 = false;
  230. if (strcmp(encoder, SIMPLE_ENCODER_X264) == 0) {
  231. LoadRecordingPreset_h264("obs_x264");
  232. } else if (strcmp(encoder, SIMPLE_ENCODER_X264_LOWCPU) == 0) {
  233. LoadRecordingPreset_h264("obs_x264");
  234. lowCPUx264 = true;
  235. } else if (strcmp(encoder, SIMPLE_ENCODER_QSV) == 0) {
  236. LoadRecordingPreset_h264("obs_qsv11");
  237. } else if (strcmp(encoder, SIMPLE_ENCODER_AMD) == 0) {
  238. LoadRecordingPreset_h264("amd_amf_h264");
  239. } else if (strcmp(encoder, SIMPLE_ENCODER_NVENC) == 0) {
  240. LoadRecordingPreset_h264("ffmpeg_nvenc");
  241. }
  242. usingRecordingPreset = true;
  243. if (!CreateAACEncoder(aacRecording, aacRecEncID, 192,
  244. "simple_aac_recording", 0))
  245. throw "Failed to create aac recording encoder "
  246. "(simple output)";
  247. }
  248. }
  249. SimpleOutput::SimpleOutput(OBSBasic *main_) : BasicOutputHandler(main_)
  250. {
  251. streamOutput = obs_output_create("rtmp_output", "simple_stream",
  252. nullptr, nullptr);
  253. if (!streamOutput)
  254. throw "Failed to create stream output (simple output)";
  255. obs_output_release(streamOutput);
  256. const char *encoder = config_get_string(main->Config(), "SimpleOutput",
  257. "StreamEncoder");
  258. if (strcmp(encoder, SIMPLE_ENCODER_QSV) == 0)
  259. LoadStreamingPreset_h264("obs_qsv11");
  260. else if (strcmp(encoder, SIMPLE_ENCODER_AMD) == 0)
  261. LoadStreamingPreset_h264("amd_amf_h264");
  262. else if (strcmp(encoder, SIMPLE_ENCODER_NVENC) == 0)
  263. LoadStreamingPreset_h264("ffmpeg_nvenc");
  264. else
  265. LoadStreamingPreset_h264("obs_x264");
  266. if (!CreateAACEncoder(aacStreaming, aacStreamEncID, GetAudioBitrate(),
  267. "simple_aac", 0))
  268. throw "Failed to create aac streaming encoder (simple output)";
  269. streamDelayStarting.Connect(obs_output_get_signal_handler(streamOutput),
  270. "starting", OBSStreamStarting, this);
  271. streamStopping.Connect(obs_output_get_signal_handler(streamOutput),
  272. "stopping", OBSStreamStopping, this);
  273. startStreaming.Connect(obs_output_get_signal_handler(streamOutput),
  274. "start", OBSStartStreaming, this);
  275. stopStreaming.Connect(obs_output_get_signal_handler(streamOutput),
  276. "stop", OBSStopStreaming, this);
  277. LoadRecordingPreset();
  278. if (!ffmpegOutput) {
  279. bool useReplayBuffer = config_get_bool(main->Config(),
  280. "SimpleOutput", "RecRB");
  281. if (useReplayBuffer) {
  282. const char *str = config_get_string(main->Config(),
  283. "Hotkeys", "ReplayBuffer");
  284. obs_data_t *hotkey = obs_data_create_from_json(str);
  285. replayBuffer = obs_output_create("replay_buffer",
  286. Str("ReplayBuffer"), nullptr, hotkey);
  287. obs_data_release(hotkey);
  288. if (!replayBuffer)
  289. throw "Failed to create replay buffer output "
  290. "(simple output)";
  291. obs_output_release(replayBuffer);
  292. signal_handler_t *signal =
  293. obs_output_get_signal_handler(replayBuffer);
  294. startReplayBuffer.Connect(signal, "start",
  295. OBSStartReplayBuffer, this);
  296. stopReplayBuffer.Connect(signal, "stop",
  297. OBSStopReplayBuffer, this);
  298. replayBufferStopping.Connect(signal, "stopping",
  299. OBSReplayBufferStopping, this);
  300. }
  301. fileOutput = obs_output_create("ffmpeg_muxer",
  302. "simple_file_output", nullptr, nullptr);
  303. if (!fileOutput)
  304. throw "Failed to create recording output "
  305. "(simple output)";
  306. obs_output_release(fileOutput);
  307. }
  308. startRecording.Connect(obs_output_get_signal_handler(fileOutput),
  309. "start", OBSStartRecording, this);
  310. stopRecording.Connect(obs_output_get_signal_handler(fileOutput),
  311. "stop", OBSStopRecording, this);
  312. recordStopping.Connect(obs_output_get_signal_handler(fileOutput),
  313. "stopping", OBSRecordStopping, this);
  314. }
  315. int SimpleOutput::GetAudioBitrate() const
  316. {
  317. int bitrate = (int)config_get_uint(main->Config(), "SimpleOutput",
  318. "ABitrate");
  319. return FindClosestAvailableAACBitrate(bitrate);
  320. }
  321. void SimpleOutput::Update()
  322. {
  323. obs_data_t *h264Settings = obs_data_create();
  324. obs_data_t *aacSettings = obs_data_create();
  325. int videoBitrate = config_get_uint(main->Config(), "SimpleOutput",
  326. "VBitrate");
  327. int audioBitrate = GetAudioBitrate();
  328. bool advanced = config_get_bool(main->Config(), "SimpleOutput",
  329. "UseAdvanced");
  330. bool enforceBitrate = config_get_bool(main->Config(), "SimpleOutput",
  331. "EnforceBitrate");
  332. const char *custom = config_get_string(main->Config(),
  333. "SimpleOutput", "x264Settings");
  334. const char *encoder = config_get_string(main->Config(), "SimpleOutput",
  335. "StreamEncoder");
  336. const char *presetType;
  337. const char *preset;
  338. if (strcmp(encoder, SIMPLE_ENCODER_QSV) == 0) {
  339. presetType = "QSVPreset";
  340. } else if (strcmp(encoder, SIMPLE_ENCODER_AMD) == 0) {
  341. presetType = "AMDPreset";
  342. UpdateStreamingSettings_amd(h264Settings, videoBitrate);
  343. } else if (strcmp(encoder, SIMPLE_ENCODER_NVENC) == 0) {
  344. presetType = "NVENCPreset";
  345. } else {
  346. presetType = "Preset";
  347. }
  348. preset = config_get_string(main->Config(), "SimpleOutput", presetType);
  349. obs_data_set_string(h264Settings, "rate_control", "CBR");
  350. obs_data_set_int(h264Settings, "bitrate", videoBitrate);
  351. if (advanced) {
  352. obs_data_set_string(h264Settings, "preset", preset);
  353. obs_data_set_string(h264Settings, "x264opts", custom);
  354. }
  355. obs_data_set_string(aacSettings, "rate_control", "CBR");
  356. obs_data_set_int(aacSettings, "bitrate", audioBitrate);
  357. obs_service_apply_encoder_settings(main->GetService(),
  358. h264Settings, aacSettings);
  359. if (advanced && !enforceBitrate) {
  360. obs_data_set_int(h264Settings, "bitrate", videoBitrate);
  361. obs_data_set_int(aacSettings, "bitrate", audioBitrate);
  362. }
  363. video_t *video = obs_get_video();
  364. enum video_format format = video_output_get_format(video);
  365. if (format != VIDEO_FORMAT_NV12 && format != VIDEO_FORMAT_I420)
  366. obs_encoder_set_preferred_video_format(h264Streaming,
  367. VIDEO_FORMAT_NV12);
  368. obs_encoder_update(h264Streaming, h264Settings);
  369. obs_encoder_update(aacStreaming, aacSettings);
  370. obs_data_release(h264Settings);
  371. obs_data_release(aacSettings);
  372. }
  373. void SimpleOutput::UpdateRecordingAudioSettings()
  374. {
  375. obs_data_t *settings = obs_data_create();
  376. obs_data_set_int(settings, "bitrate", 192);
  377. obs_data_set_string(settings, "rate_control", "CBR");
  378. obs_encoder_update(aacRecording, settings);
  379. obs_data_release(settings);
  380. }
  381. #define CROSS_DIST_CUTOFF 2000.0
  382. int SimpleOutput::CalcCRF(int crf)
  383. {
  384. int cx = config_get_uint(main->Config(), "Video", "OutputCX");
  385. int cy = config_get_uint(main->Config(), "Video", "OutputCY");
  386. double fCX = double(cx);
  387. double fCY = double(cy);
  388. if (lowCPUx264)
  389. crf -= 2;
  390. double crossDist = sqrt(fCX * fCX + fCY * fCY);
  391. double crfResReduction =
  392. fmin(CROSS_DIST_CUTOFF, crossDist) / CROSS_DIST_CUTOFF;
  393. crfResReduction = (1.0 - crfResReduction) * 10.0;
  394. return crf - int(crfResReduction);
  395. }
  396. void SimpleOutput::UpdateRecordingSettings_x264_crf(int crf)
  397. {
  398. obs_data_t *settings = obs_data_create();
  399. obs_data_set_int(settings, "crf", crf);
  400. obs_data_set_bool(settings, "use_bufsize", true);
  401. obs_data_set_string(settings, "rate_control", "CRF");
  402. obs_data_set_string(settings, "profile", "high");
  403. obs_data_set_string(settings, "preset",
  404. lowCPUx264 ? "ultrafast" : "veryfast");
  405. obs_encoder_update(h264Recording, settings);
  406. obs_data_release(settings);
  407. }
  408. static bool icq_available(obs_encoder_t *encoder)
  409. {
  410. obs_properties_t *props = obs_encoder_properties(encoder);
  411. obs_property_t *p = obs_properties_get(props, "rate_control");
  412. bool icq_found = false;
  413. size_t num = obs_property_list_item_count(p);
  414. for (size_t i = 0; i < num; i++) {
  415. const char *val = obs_property_list_item_string(p, i);
  416. if (strcmp(val, "ICQ") == 0) {
  417. icq_found = true;
  418. break;
  419. }
  420. }
  421. obs_properties_destroy(props);
  422. return icq_found;
  423. }
  424. void SimpleOutput::UpdateRecordingSettings_qsv11(int crf)
  425. {
  426. bool icq = icq_available(h264Recording);
  427. obs_data_t *settings = obs_data_create();
  428. obs_data_set_string(settings, "profile", "high");
  429. if (icq) {
  430. obs_data_set_string(settings, "rate_control", "ICQ");
  431. obs_data_set_int(settings, "icq_quality", crf);
  432. } else {
  433. obs_data_set_string(settings, "rate_control", "CQP");
  434. obs_data_set_int(settings, "qpi", crf);
  435. obs_data_set_int(settings, "qpp", crf);
  436. obs_data_set_int(settings, "qpb", crf);
  437. }
  438. obs_encoder_update(h264Recording, settings);
  439. obs_data_release(settings);
  440. }
  441. void SimpleOutput::UpdateRecordingSettings_nvenc(int cqp)
  442. {
  443. obs_data_t *settings = obs_data_create();
  444. obs_data_set_string(settings, "rate_control", "CQP");
  445. obs_data_set_string(settings, "profile", "high");
  446. obs_data_set_string(settings, "preset", "hq");
  447. obs_data_set_int(settings, "cqp", cqp);
  448. obs_encoder_update(h264Recording, settings);
  449. obs_data_release(settings);
  450. }
  451. void SimpleOutput::UpdateStreamingSettings_amd(obs_data_t *settings,
  452. int bitrate)
  453. {
  454. // Static Properties
  455. obs_data_set_int(settings, "Usage", 0);
  456. obs_data_set_int(settings, "Profile", 100); // High
  457. obs_data_set_string(settings, "profile", "high"); // High
  458. // Rate Control Properties
  459. obs_data_set_int(settings, "RateControlMethod", 1);
  460. obs_data_set_string(settings, "rate_control", "CBR");
  461. obs_data_set_int(settings, "Bitrate.Target", bitrate);
  462. obs_data_set_int(settings, "bitrate", bitrate);
  463. obs_data_set_int(settings, "FillerData", 1);
  464. obs_data_set_int(settings, "VBVBuffer", 1);
  465. obs_data_set_int(settings, "VBVBuffer.Size", bitrate);
  466. // Picture Control Properties
  467. obs_data_set_double(settings, "KeyframeInterval", 2.0);
  468. obs_data_set_int(settings, "keyint_sec", 2);
  469. }
  470. void SimpleOutput::UpdateRecordingSettings_amd_cqp(int cqp)
  471. {
  472. obs_data_t *settings = obs_data_create();
  473. // Static Properties
  474. obs_data_set_int(settings, "Usage", 0);
  475. obs_data_set_int(settings, "Profile", 100); // High
  476. obs_data_set_string(settings, "profile", "high"); // High
  477. // Rate Control Properties
  478. obs_data_set_int(settings, "RateControlMethod", 0);
  479. obs_data_set_string(settings, "rate_control", "CQP");
  480. obs_data_set_int(settings, "QP.IFrame", cqp);
  481. obs_data_set_int(settings, "QP.PFrame", cqp);
  482. obs_data_set_int(settings, "QP.BFrame", cqp);
  483. obs_data_set_int(settings, "VBVBuffer", 1);
  484. obs_data_set_int(settings, "VBVBuffer.Size", 100000);
  485. // Picture Control Properties
  486. obs_data_set_double(settings, "KeyframeInterval", 2.0);
  487. obs_data_set_int(settings, "keyint_sec", 2);
  488. // Update and release
  489. obs_encoder_update(h264Recording, settings);
  490. obs_data_release(settings);
  491. }
  492. void SimpleOutput::UpdateRecordingSettings()
  493. {
  494. bool ultra_hq = (videoQuality == "HQ");
  495. int crf = CalcCRF(ultra_hq ? 16 : 23);
  496. if (astrcmp_n(videoEncoder.c_str(), "x264", 4) == 0) {
  497. UpdateRecordingSettings_x264_crf(crf);
  498. } else if (videoEncoder == SIMPLE_ENCODER_QSV) {
  499. UpdateRecordingSettings_qsv11(crf);
  500. } else if (videoEncoder == SIMPLE_ENCODER_AMD) {
  501. UpdateRecordingSettings_amd_cqp(crf);
  502. } else if (videoEncoder == SIMPLE_ENCODER_NVENC) {
  503. UpdateRecordingSettings_nvenc(crf);
  504. }
  505. }
  506. inline void SimpleOutput::SetupOutputs()
  507. {
  508. SimpleOutput::Update();
  509. obs_encoder_set_video(h264Streaming, obs_get_video());
  510. obs_encoder_set_audio(aacStreaming, obs_get_audio());
  511. if (usingRecordingPreset) {
  512. if (ffmpegOutput) {
  513. obs_output_set_media(fileOutput, obs_get_video(),
  514. obs_get_audio());
  515. } else {
  516. obs_encoder_set_video(h264Recording, obs_get_video());
  517. obs_encoder_set_audio(aacRecording, obs_get_audio());
  518. }
  519. }
  520. }
  521. bool SimpleOutput::StartStreaming(obs_service_t *service)
  522. {
  523. if (!Active())
  524. SetupOutputs();
  525. obs_output_set_video_encoder(streamOutput, h264Streaming);
  526. obs_output_set_audio_encoder(streamOutput, aacStreaming, 0);
  527. obs_output_set_service(streamOutput, service);
  528. bool reconnect = config_get_bool(main->Config(), "Output",
  529. "Reconnect");
  530. int retryDelay = config_get_uint(main->Config(), "Output",
  531. "RetryDelay");
  532. int maxRetries = config_get_uint(main->Config(), "Output",
  533. "MaxRetries");
  534. bool useDelay = config_get_bool(main->Config(), "Output",
  535. "DelayEnable");
  536. int delaySec = config_get_int(main->Config(), "Output",
  537. "DelaySec");
  538. bool preserveDelay = config_get_bool(main->Config(), "Output",
  539. "DelayPreserve");
  540. const char *bindIP = config_get_string(main->Config(), "Output",
  541. "BindIP");
  542. bool enableNewSocketLoop = config_get_bool(main->Config(), "Output",
  543. "NewSocketLoopEnable");
  544. bool enableLowLatencyMode = config_get_bool(main->Config(), "Output",
  545. "LowLatencyEnable");
  546. obs_data_t *settings = obs_data_create();
  547. obs_data_set_string(settings, "bind_ip", bindIP);
  548. obs_data_set_bool(settings, "new_socket_loop_enabled",
  549. enableNewSocketLoop);
  550. obs_data_set_bool(settings, "low_latency_mode_enabled",
  551. enableLowLatencyMode);
  552. obs_output_update(streamOutput, settings);
  553. obs_data_release(settings);
  554. if (!reconnect)
  555. maxRetries = 0;
  556. obs_output_set_delay(streamOutput, useDelay ? delaySec : 0,
  557. preserveDelay ? OBS_OUTPUT_DELAY_PRESERVE : 0);
  558. obs_output_set_reconnect_settings(streamOutput, maxRetries,
  559. retryDelay);
  560. if (obs_output_start(streamOutput)) {
  561. return true;
  562. }
  563. return false;
  564. }
  565. static void remove_reserved_file_characters(string &s)
  566. {
  567. replace(s.begin(), s.end(), '/', '_');
  568. replace(s.begin(), s.end(), '\\', '_');
  569. replace(s.begin(), s.end(), '*', '_');
  570. replace(s.begin(), s.end(), '?', '_');
  571. replace(s.begin(), s.end(), '"', '_');
  572. replace(s.begin(), s.end(), '|', '_');
  573. replace(s.begin(), s.end(), ':', '_');
  574. replace(s.begin(), s.end(), '>', '_');
  575. replace(s.begin(), s.end(), '<', '_');
  576. }
  577. static void ensure_directory_exists(string &path)
  578. {
  579. replace(path.begin(), path.end(), '\\', '/');
  580. size_t last = path.rfind('/');
  581. if (last == string::npos)
  582. return;
  583. string directory = path.substr(0, last);
  584. os_mkdirs(directory.c_str());
  585. }
  586. void SimpleOutput::UpdateRecording()
  587. {
  588. if (replayBufferActive || recordingActive)
  589. return;
  590. if (usingRecordingPreset) {
  591. if (!ffmpegOutput)
  592. UpdateRecordingSettings();
  593. } else if (!obs_output_active(streamOutput)) {
  594. Update();
  595. }
  596. if (!Active())
  597. SetupOutputs();
  598. if (!ffmpegOutput) {
  599. obs_output_set_video_encoder(fileOutput, h264Recording);
  600. obs_output_set_audio_encoder(fileOutput, aacRecording, 0);
  601. }
  602. if (replayBuffer) {
  603. obs_output_set_video_encoder(replayBuffer, h264Recording);
  604. obs_output_set_audio_encoder(replayBuffer, aacRecording, 0);
  605. }
  606. recordingConfigured = true;
  607. }
  608. bool SimpleOutput::ConfigureRecording(bool updateReplayBuffer)
  609. {
  610. const char *path = config_get_string(main->Config(),
  611. "SimpleOutput", "FilePath");
  612. const char *format = config_get_string(main->Config(),
  613. "SimpleOutput", "RecFormat");
  614. const char *mux = config_get_string(main->Config(), "SimpleOutput",
  615. "MuxerCustom");
  616. bool noSpace = config_get_bool(main->Config(), "SimpleOutput",
  617. "FileNameWithoutSpace");
  618. const char *filenameFormat = config_get_string(main->Config(), "Output",
  619. "FilenameFormatting");
  620. bool overwriteIfExists = config_get_bool(main->Config(), "Output",
  621. "OverwriteIfExists");
  622. const char *rbPrefix = config_get_string(main->Config(), "SimpleOutput",
  623. "RecRBPrefix");
  624. const char *rbSuffix = config_get_string(main->Config(), "SimpleOutput",
  625. "RecRBSuffix");
  626. int rbTime = config_get_int(main->Config(), "SimpleOutput",
  627. "RecRBTime");
  628. int rbSize = config_get_int(main->Config(), "SimpleOutput",
  629. "RecRBSize");
  630. os_dir_t *dir = path ? os_opendir(path) : nullptr;
  631. if (!dir) {
  632. if (main->isVisible())
  633. OBSMessageBox::information(main,
  634. QTStr("Output.BadPath.Title"),
  635. QTStr("Output.BadPath.Text"));
  636. else
  637. main->SysTrayNotify(QTStr("Output.BadPath.Text"),
  638. QSystemTrayIcon::Warning);
  639. return false;
  640. }
  641. os_closedir(dir);
  642. string strPath;
  643. strPath += path;
  644. char lastChar = strPath.back();
  645. if (lastChar != '/' && lastChar != '\\')
  646. strPath += "/";
  647. strPath += GenerateSpecifiedFilename(ffmpegOutput ? "avi" : format,
  648. noSpace, filenameFormat);
  649. ensure_directory_exists(strPath);
  650. if (!overwriteIfExists)
  651. FindBestFilename(strPath, noSpace);
  652. obs_data_t *settings = obs_data_create();
  653. if (updateReplayBuffer) {
  654. string f;
  655. if (rbPrefix && *rbPrefix) {
  656. f += rbPrefix;
  657. if (f.back() != ' ')
  658. f += " ";
  659. }
  660. f += filenameFormat;
  661. if (rbSuffix && *rbSuffix) {
  662. if (*rbSuffix != ' ')
  663. f += " ";
  664. f += rbSuffix;
  665. }
  666. remove_reserved_file_characters(f);
  667. obs_data_set_string(settings, "directory", path);
  668. obs_data_set_string(settings, "format", f.c_str());
  669. obs_data_set_string(settings, "extension", format);
  670. obs_data_set_int(settings, "max_time_sec", rbTime);
  671. obs_data_set_int(settings, "max_size_mb",
  672. usingRecordingPreset ? rbSize : 0);
  673. } else {
  674. obs_data_set_string(settings, ffmpegOutput ? "url" : "path",
  675. strPath.c_str());
  676. }
  677. obs_data_set_string(settings, "muxer_settings", mux);
  678. if (updateReplayBuffer)
  679. obs_output_update(replayBuffer, settings);
  680. else
  681. obs_output_update(fileOutput, settings);
  682. obs_data_release(settings);
  683. return true;
  684. }
  685. bool SimpleOutput::StartRecording()
  686. {
  687. UpdateRecording();
  688. if (!ConfigureRecording(false))
  689. return false;
  690. if (!obs_output_start(fileOutput)) {
  691. QMessageBox::critical(main,
  692. QTStr("Output.StartRecordingFailed"),
  693. QTStr("Output.StartFailedGeneric"));
  694. return false;
  695. }
  696. return true;
  697. }
  698. bool SimpleOutput::StartReplayBuffer()
  699. {
  700. UpdateRecording();
  701. if (!ConfigureRecording(true))
  702. return false;
  703. if (!obs_output_start(replayBuffer)) {
  704. QMessageBox::critical(main,
  705. QTStr("Output.StartReplayFailed"),
  706. QTStr("Output.StartFailedGeneric"));
  707. return false;
  708. }
  709. return true;
  710. }
  711. void SimpleOutput::StopStreaming(bool force)
  712. {
  713. if (force)
  714. obs_output_force_stop(streamOutput);
  715. else
  716. obs_output_stop(streamOutput);
  717. }
  718. void SimpleOutput::StopRecording(bool force)
  719. {
  720. if (force)
  721. obs_output_force_stop(fileOutput);
  722. else
  723. obs_output_stop(fileOutput);
  724. }
  725. void SimpleOutput::StopReplayBuffer(bool force)
  726. {
  727. if (force)
  728. obs_output_force_stop(replayBuffer);
  729. else
  730. obs_output_stop(replayBuffer);
  731. }
  732. bool SimpleOutput::StreamingActive() const
  733. {
  734. return obs_output_active(streamOutput);
  735. }
  736. bool SimpleOutput::RecordingActive() const
  737. {
  738. return obs_output_active(fileOutput);
  739. }
  740. bool SimpleOutput::ReplayBufferActive() const
  741. {
  742. return obs_output_active(replayBuffer);
  743. }
  744. /* ------------------------------------------------------------------------ */
  745. struct AdvancedOutput : BasicOutputHandler {
  746. OBSEncoder aacTrack[MAX_AUDIO_MIXES];
  747. OBSEncoder h264Streaming;
  748. OBSEncoder h264Recording;
  749. bool ffmpegOutput;
  750. bool ffmpegRecording;
  751. bool useStreamEncoder;
  752. string aacEncoderID[MAX_AUDIO_MIXES];
  753. AdvancedOutput(OBSBasic *main_);
  754. inline void UpdateStreamSettings();
  755. inline void UpdateRecordingSettings();
  756. inline void UpdateAudioSettings();
  757. virtual void Update() override;
  758. inline void SetupStreaming();
  759. inline void SetupRecording();
  760. inline void SetupFFmpeg();
  761. void SetupOutputs();
  762. int GetAudioBitrate(size_t i) const;
  763. virtual bool StartStreaming(obs_service_t *service) override;
  764. virtual bool StartRecording() override;
  765. virtual void StopStreaming(bool force) override;
  766. virtual void StopRecording(bool force) override;
  767. virtual bool StreamingActive() const override;
  768. virtual bool RecordingActive() const override;
  769. };
  770. static OBSData GetDataFromJsonFile(const char *jsonFile)
  771. {
  772. char fullPath[512];
  773. int ret = GetProfilePath(fullPath, sizeof(fullPath), jsonFile);
  774. if (ret > 0) {
  775. BPtr<char> jsonData = os_quick_read_utf8_file(fullPath);
  776. if (!!jsonData) {
  777. obs_data_t *data = obs_data_create_from_json(jsonData);
  778. OBSData dataRet(data);
  779. obs_data_release(data);
  780. return dataRet;
  781. }
  782. }
  783. return nullptr;
  784. }
  785. AdvancedOutput::AdvancedOutput(OBSBasic *main_) : BasicOutputHandler(main_)
  786. {
  787. const char *recType = config_get_string(main->Config(), "AdvOut",
  788. "RecType");
  789. const char *streamEncoder = config_get_string(main->Config(), "AdvOut",
  790. "Encoder");
  791. const char *recordEncoder = config_get_string(main->Config(), "AdvOut",
  792. "RecEncoder");
  793. ffmpegOutput = astrcmpi(recType, "FFmpeg") == 0;
  794. ffmpegRecording = ffmpegOutput &&
  795. config_get_bool(main->Config(), "AdvOut", "FFOutputToFile");
  796. useStreamEncoder = astrcmpi(recordEncoder, "none") == 0;
  797. OBSData streamEncSettings = GetDataFromJsonFile("streamEncoder.json");
  798. OBSData recordEncSettings = GetDataFromJsonFile("recordEncoder.json");
  799. streamOutput = obs_output_create("rtmp_output", "adv_stream",
  800. nullptr, nullptr);
  801. if (!streamOutput)
  802. throw "Failed to create stream output (advanced output)";
  803. obs_output_release(streamOutput);
  804. if (ffmpegOutput) {
  805. fileOutput = obs_output_create("ffmpeg_output",
  806. "adv_ffmpeg_output", nullptr, nullptr);
  807. if (!fileOutput)
  808. throw "Failed to create recording FFmpeg output "
  809. "(advanced output)";
  810. obs_output_release(fileOutput);
  811. } else {
  812. fileOutput = obs_output_create("ffmpeg_muxer",
  813. "adv_file_output", nullptr, nullptr);
  814. if (!fileOutput)
  815. throw "Failed to create recording output "
  816. "(advanced output)";
  817. obs_output_release(fileOutput);
  818. if (!useStreamEncoder) {
  819. h264Recording = obs_video_encoder_create(recordEncoder,
  820. "recording_h264", recordEncSettings,
  821. nullptr);
  822. if (!h264Recording)
  823. throw "Failed to create recording h264 "
  824. "encoder (advanced output)";
  825. obs_encoder_release(h264Recording);
  826. }
  827. }
  828. h264Streaming = obs_video_encoder_create(streamEncoder,
  829. "streaming_h264", streamEncSettings, nullptr);
  830. if (!h264Streaming)
  831. throw "Failed to create streaming h264 encoder "
  832. "(advanced output)";
  833. obs_encoder_release(h264Streaming);
  834. for (int i = 0; i < MAX_AUDIO_MIXES; i++) {
  835. char name[9];
  836. sprintf(name, "adv_aac%d", i);
  837. if (!CreateAACEncoder(aacTrack[i], aacEncoderID[i],
  838. GetAudioBitrate(i), name, i))
  839. throw "Failed to create audio encoder "
  840. "(advanced output)";
  841. }
  842. streamDelayStarting.Connect(obs_output_get_signal_handler(streamOutput),
  843. "starting", OBSStreamStarting, this);
  844. streamStopping.Connect(obs_output_get_signal_handler(streamOutput),
  845. "stopping", OBSStreamStopping, this);
  846. startStreaming.Connect(obs_output_get_signal_handler(streamOutput),
  847. "start", OBSStartStreaming, this);
  848. stopStreaming.Connect(obs_output_get_signal_handler(streamOutput),
  849. "stop", OBSStopStreaming, this);
  850. startRecording.Connect(obs_output_get_signal_handler(fileOutput),
  851. "start", OBSStartRecording, this);
  852. stopRecording.Connect(obs_output_get_signal_handler(fileOutput),
  853. "stop", OBSStopRecording, this);
  854. recordStopping.Connect(obs_output_get_signal_handler(fileOutput),
  855. "stopping", OBSRecordStopping, this);
  856. }
  857. void AdvancedOutput::UpdateStreamSettings()
  858. {
  859. bool applyServiceSettings = config_get_bool(main->Config(), "AdvOut",
  860. "ApplyServiceSettings");
  861. OBSData settings = GetDataFromJsonFile("streamEncoder.json");
  862. if (applyServiceSettings)
  863. obs_service_apply_encoder_settings(main->GetService(),
  864. settings, nullptr);
  865. video_t *video = obs_get_video();
  866. enum video_format format = video_output_get_format(video);
  867. if (format != VIDEO_FORMAT_NV12 && format != VIDEO_FORMAT_I420)
  868. obs_encoder_set_preferred_video_format(h264Streaming,
  869. VIDEO_FORMAT_NV12);
  870. obs_encoder_update(h264Streaming, settings);
  871. }
  872. inline void AdvancedOutput::UpdateRecordingSettings()
  873. {
  874. OBSData settings = GetDataFromJsonFile("recordEncoder.json");
  875. obs_encoder_update(h264Recording, settings);
  876. }
  877. void AdvancedOutput::Update()
  878. {
  879. UpdateStreamSettings();
  880. if (!useStreamEncoder && !ffmpegOutput)
  881. UpdateRecordingSettings();
  882. UpdateAudioSettings();
  883. }
  884. inline void AdvancedOutput::SetupStreaming()
  885. {
  886. bool rescale = config_get_bool(main->Config(), "AdvOut",
  887. "Rescale");
  888. const char *rescaleRes = config_get_string(main->Config(), "AdvOut",
  889. "RescaleRes");
  890. bool multitrack = config_get_bool(main->Config(), "AdvOut",
  891. "Multitrack");
  892. int trackIndex = config_get_int(main->Config(), "AdvOut",
  893. "TrackIndex");
  894. int trackCount = config_get_int(main->Config(), "AdvOut",
  895. "TrackCount");
  896. unsigned int cx = 0;
  897. unsigned int cy = 0;
  898. if (rescale && rescaleRes && *rescaleRes) {
  899. if (sscanf(rescaleRes, "%ux%u", &cx, &cy) != 2) {
  900. cx = 0;
  901. cy = 0;
  902. }
  903. }
  904. obs_encoder_set_scaled_size(h264Streaming, cx, cy);
  905. obs_encoder_set_video(h264Streaming, obs_get_video());
  906. obs_output_set_video_encoder(streamOutput, h264Streaming);
  907. if (multitrack) {
  908. int i = 0;
  909. for (; i < trackCount; i++)
  910. obs_output_set_audio_encoder(streamOutput, aacTrack[i],
  911. i);
  912. for (; i < MAX_AUDIO_MIXES; i++)
  913. obs_output_set_audio_encoder(streamOutput, nullptr, i);
  914. } else {
  915. obs_output_set_audio_encoder(streamOutput,
  916. aacTrack[trackIndex - 1], 0);
  917. }
  918. }
  919. inline void AdvancedOutput::SetupRecording()
  920. {
  921. const char *path = config_get_string(main->Config(), "AdvOut",
  922. "RecFilePath");
  923. const char *mux = config_get_string(main->Config(), "AdvOut",
  924. "RecMuxerCustom");
  925. bool rescale = config_get_bool(main->Config(), "AdvOut",
  926. "RecRescale");
  927. const char *rescaleRes = config_get_string(main->Config(), "AdvOut",
  928. "RecRescaleRes");
  929. int tracks = config_get_int(main->Config(), "AdvOut", "RecTracks");
  930. obs_data_t *settings = obs_data_create();
  931. unsigned int cx = 0;
  932. unsigned int cy = 0;
  933. int idx = 0;
  934. if (useStreamEncoder) {
  935. obs_output_set_video_encoder(fileOutput, h264Streaming);
  936. } else {
  937. if (rescale && rescaleRes && *rescaleRes) {
  938. if (sscanf(rescaleRes, "%ux%u", &cx, &cy) != 2) {
  939. cx = 0;
  940. cy = 0;
  941. }
  942. }
  943. obs_encoder_set_scaled_size(h264Recording, cx, cy);
  944. obs_encoder_set_video(h264Recording, obs_get_video());
  945. obs_output_set_video_encoder(fileOutput, h264Recording);
  946. }
  947. for (int i = 0; i < MAX_AUDIO_MIXES; i++) {
  948. if ((tracks & (1<<i)) != 0) {
  949. obs_output_set_audio_encoder(fileOutput, aacTrack[i],
  950. idx++);
  951. }
  952. }
  953. obs_data_set_string(settings, "path", path);
  954. obs_data_set_string(settings, "muxer_settings", mux);
  955. obs_output_update(fileOutput, settings);
  956. obs_data_release(settings);
  957. }
  958. inline void AdvancedOutput::SetupFFmpeg()
  959. {
  960. const char *url = config_get_string(main->Config(), "AdvOut", "FFURL");
  961. int vBitrate = config_get_int(main->Config(), "AdvOut",
  962. "FFVBitrate");
  963. int gopSize = config_get_int(main->Config(), "AdvOut",
  964. "FFVGOPSize");
  965. bool rescale = config_get_bool(main->Config(), "AdvOut",
  966. "FFRescale");
  967. const char *rescaleRes = config_get_string(main->Config(), "AdvOut",
  968. "FFRescaleRes");
  969. const char *formatName = config_get_string(main->Config(), "AdvOut",
  970. "FFFormat");
  971. const char *mimeType = config_get_string(main->Config(), "AdvOut",
  972. "FFFormatMimeType");
  973. const char *muxCustom = config_get_string(main->Config(), "AdvOut",
  974. "FFMCustom");
  975. const char *vEncoder = config_get_string(main->Config(), "AdvOut",
  976. "FFVEncoder");
  977. int vEncoderId = config_get_int(main->Config(), "AdvOut",
  978. "FFVEncoderId");
  979. const char *vEncCustom = config_get_string(main->Config(), "AdvOut",
  980. "FFVCustom");
  981. int aBitrate = config_get_int(main->Config(), "AdvOut",
  982. "FFABitrate");
  983. int aTrack = config_get_int(main->Config(), "AdvOut",
  984. "FFAudioTrack");
  985. const char *aEncoder = config_get_string(main->Config(), "AdvOut",
  986. "FFAEncoder");
  987. int aEncoderId = config_get_int(main->Config(), "AdvOut",
  988. "FFAEncoderId");
  989. const char *aEncCustom = config_get_string(main->Config(), "AdvOut",
  990. "FFACustom");
  991. obs_data_t *settings = obs_data_create();
  992. obs_data_set_string(settings, "url", url);
  993. obs_data_set_string(settings, "format_name", formatName);
  994. obs_data_set_string(settings, "format_mime_type", mimeType);
  995. obs_data_set_string(settings, "muxer_settings", muxCustom);
  996. obs_data_set_int(settings, "gop_size", gopSize);
  997. obs_data_set_int(settings, "video_bitrate", vBitrate);
  998. obs_data_set_string(settings, "video_encoder", vEncoder);
  999. obs_data_set_int(settings, "video_encoder_id", vEncoderId);
  1000. obs_data_set_string(settings, "video_settings", vEncCustom);
  1001. obs_data_set_int(settings, "audio_bitrate", aBitrate);
  1002. obs_data_set_string(settings, "audio_encoder", aEncoder);
  1003. obs_data_set_int(settings, "audio_encoder_id", aEncoderId);
  1004. obs_data_set_string(settings, "audio_settings", aEncCustom);
  1005. if (rescale && rescaleRes && *rescaleRes) {
  1006. int width;
  1007. int height;
  1008. int val = sscanf(rescaleRes, "%dx%d", &width, &height);
  1009. if (val == 2 && width && height) {
  1010. obs_data_set_int(settings, "scale_width", width);
  1011. obs_data_set_int(settings, "scale_height", height);
  1012. }
  1013. }
  1014. obs_output_set_mixer(fileOutput, aTrack - 1);
  1015. obs_output_set_media(fileOutput, obs_get_video(), obs_get_audio());
  1016. obs_output_update(fileOutput, settings);
  1017. obs_data_release(settings);
  1018. }
  1019. static inline void SetEncoderName(obs_encoder_t *encoder, const char *name,
  1020. const char *defaultName)
  1021. {
  1022. obs_encoder_set_name(encoder, (name && *name) ? name : defaultName);
  1023. }
  1024. inline void AdvancedOutput::UpdateAudioSettings()
  1025. {
  1026. bool applyServiceSettings = config_get_bool(main->Config(), "AdvOut",
  1027. "ApplyServiceSettings");
  1028. int streamTrackIndex = config_get_int(main->Config(), "AdvOut",
  1029. "TrackIndex");
  1030. obs_data_t *settings[MAX_AUDIO_MIXES];
  1031. for (size_t i = 0; i < MAX_AUDIO_MIXES; i++) {
  1032. settings[i] = obs_data_create();
  1033. obs_data_set_int(settings[i], "bitrate", GetAudioBitrate(i));
  1034. }
  1035. for (size_t i = 0; i < MAX_AUDIO_MIXES; i++) {
  1036. string cfg_name = "Track";
  1037. cfg_name += to_string((int)i + 1);
  1038. cfg_name += "Name";
  1039. const char *name = config_get_string(main->Config(), "AdvOut",
  1040. cfg_name.c_str());
  1041. string def_name = "Track";
  1042. def_name += to_string((int)i + 1);
  1043. SetEncoderName(aacTrack[i], name, def_name.c_str());
  1044. }
  1045. for (size_t i = 0; i < MAX_AUDIO_MIXES; i++) {
  1046. if (applyServiceSettings && (int)(i + 1) == streamTrackIndex)
  1047. obs_service_apply_encoder_settings(main->GetService(),
  1048. nullptr, settings[i]);
  1049. obs_encoder_update(aacTrack[i], settings[i]);
  1050. obs_data_release(settings[i]);
  1051. }
  1052. }
  1053. void AdvancedOutput::SetupOutputs()
  1054. {
  1055. obs_encoder_set_video(h264Streaming, obs_get_video());
  1056. if (h264Recording)
  1057. obs_encoder_set_video(h264Recording, obs_get_video());
  1058. for (size_t i = 0; i < MAX_AUDIO_MIXES; i++)
  1059. obs_encoder_set_audio(aacTrack[i], obs_get_audio());
  1060. SetupStreaming();
  1061. if (ffmpegOutput)
  1062. SetupFFmpeg();
  1063. else
  1064. SetupRecording();
  1065. }
  1066. int AdvancedOutput::GetAudioBitrate(size_t i) const
  1067. {
  1068. static const char *names[] = {
  1069. "Track1Bitrate", "Track2Bitrate",
  1070. "Track3Bitrate", "Track4Bitrate",
  1071. "Track5Bitrate", "Track6Bitrate",
  1072. };
  1073. int bitrate = (int)config_get_uint(main->Config(), "AdvOut", names[i]);
  1074. return FindClosestAvailableAACBitrate(bitrate);
  1075. }
  1076. bool AdvancedOutput::StartStreaming(obs_service_t *service)
  1077. {
  1078. if (!useStreamEncoder ||
  1079. (!ffmpegOutput && !obs_output_active(fileOutput))) {
  1080. UpdateStreamSettings();
  1081. }
  1082. UpdateAudioSettings();
  1083. if (!Active())
  1084. SetupOutputs();
  1085. obs_output_set_service(streamOutput, service);
  1086. bool reconnect = config_get_bool(main->Config(), "Output", "Reconnect");
  1087. int retryDelay = config_get_int(main->Config(), "Output", "RetryDelay");
  1088. int maxRetries = config_get_int(main->Config(), "Output", "MaxRetries");
  1089. bool useDelay = config_get_bool(main->Config(), "Output",
  1090. "DelayEnable");
  1091. int delaySec = config_get_int(main->Config(), "Output",
  1092. "DelaySec");
  1093. bool preserveDelay = config_get_bool(main->Config(), "Output",
  1094. "DelayPreserve");
  1095. const char *bindIP = config_get_string(main->Config(), "Output",
  1096. "BindIP");
  1097. bool enableNewSocketLoop = config_get_bool(main->Config(), "Output",
  1098. "NewSocketLoopEnable");
  1099. bool enableLowLatencyMode = config_get_bool(main->Config(), "Output",
  1100. "LowLatencyEnable");
  1101. obs_data_t *settings = obs_data_create();
  1102. obs_data_set_string(settings, "bind_ip", bindIP);
  1103. obs_data_set_bool(settings, "new_socket_loop_enabled",
  1104. enableNewSocketLoop);
  1105. obs_data_set_bool(settings, "low_latency_mode_enabled",
  1106. enableLowLatencyMode);
  1107. obs_output_update(streamOutput, settings);
  1108. obs_data_release(settings);
  1109. if (!reconnect)
  1110. maxRetries = 0;
  1111. obs_output_set_delay(streamOutput, useDelay ? delaySec : 0,
  1112. preserveDelay ? OBS_OUTPUT_DELAY_PRESERVE : 0);
  1113. obs_output_set_reconnect_settings(streamOutput, maxRetries,
  1114. retryDelay);
  1115. if (obs_output_start(streamOutput)) {
  1116. return true;
  1117. }
  1118. return false;
  1119. }
  1120. bool AdvancedOutput::StartRecording()
  1121. {
  1122. const char *path;
  1123. const char *recFormat;
  1124. const char *filenameFormat;
  1125. bool noSpace = false;
  1126. bool overwriteIfExists = false;
  1127. if (!useStreamEncoder) {
  1128. if (!ffmpegOutput) {
  1129. UpdateRecordingSettings();
  1130. }
  1131. } else if (!obs_output_active(streamOutput)) {
  1132. UpdateStreamSettings();
  1133. }
  1134. UpdateAudioSettings();
  1135. if (!Active())
  1136. SetupOutputs();
  1137. if (!ffmpegOutput || ffmpegRecording) {
  1138. path = config_get_string(main->Config(), "AdvOut",
  1139. ffmpegRecording ? "FFFilePath" : "RecFilePath");
  1140. recFormat = config_get_string(main->Config(), "AdvOut",
  1141. ffmpegRecording ? "FFExtension" : "RecFormat");
  1142. filenameFormat = config_get_string(main->Config(), "Output",
  1143. "FilenameFormatting");
  1144. overwriteIfExists = config_get_bool(main->Config(), "Output",
  1145. "OverwriteIfExists");
  1146. noSpace = config_get_bool(main->Config(), "AdvOut",
  1147. ffmpegRecording ?
  1148. "FFFileNameWithoutSpace" :
  1149. "RecFileNameWithoutSpace");
  1150. os_dir_t *dir = path ? os_opendir(path) : nullptr;
  1151. if (!dir) {
  1152. if (main->isVisible())
  1153. OBSMessageBox::information(main,
  1154. QTStr("Output.BadPath.Title"),
  1155. QTStr("Output.BadPath.Text"));
  1156. else
  1157. main->SysTrayNotify(QTStr("Output.BadPath.Text"),
  1158. QSystemTrayIcon::Warning);
  1159. return false;
  1160. }
  1161. os_closedir(dir);
  1162. string strPath;
  1163. strPath += path;
  1164. char lastChar = strPath.back();
  1165. if (lastChar != '/' && lastChar != '\\')
  1166. strPath += "/";
  1167. strPath += GenerateSpecifiedFilename(recFormat, noSpace,
  1168. filenameFormat);
  1169. ensure_directory_exists(strPath);
  1170. if (!overwriteIfExists)
  1171. FindBestFilename(strPath, noSpace);
  1172. obs_data_t *settings = obs_data_create();
  1173. obs_data_set_string(settings,
  1174. ffmpegRecording ? "url" : "path",
  1175. strPath.c_str());
  1176. obs_output_update(fileOutput, settings);
  1177. obs_data_release(settings);
  1178. }
  1179. if (!obs_output_start(fileOutput)) {
  1180. QMessageBox::critical(main,
  1181. QTStr("Output.StartRecordingFailed"),
  1182. QTStr("Output.StartFailedGeneric"));
  1183. return false;
  1184. }
  1185. return true;
  1186. }
  1187. void AdvancedOutput::StopStreaming(bool force)
  1188. {
  1189. if (force)
  1190. obs_output_force_stop(streamOutput);
  1191. else
  1192. obs_output_stop(streamOutput);
  1193. }
  1194. void AdvancedOutput::StopRecording(bool force)
  1195. {
  1196. if (force)
  1197. obs_output_force_stop(fileOutput);
  1198. else
  1199. obs_output_stop(fileOutput);
  1200. }
  1201. bool AdvancedOutput::StreamingActive() const
  1202. {
  1203. return obs_output_active(streamOutput);
  1204. }
  1205. bool AdvancedOutput::RecordingActive() const
  1206. {
  1207. return obs_output_active(fileOutput);
  1208. }
  1209. /* ------------------------------------------------------------------------ */
  1210. BasicOutputHandler *CreateSimpleOutputHandler(OBSBasic *main)
  1211. {
  1212. return new SimpleOutput(main);
  1213. }
  1214. BasicOutputHandler *CreateAdvancedOutputHandler(OBSBasic *main)
  1215. {
  1216. return new AdvancedOutput(main);
  1217. }