1
0

window-basic-main-outputs.cpp 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778
  1. #include <string>
  2. #include <QMessageBox>
  3. #include "window-basic-main.hpp"
  4. #include "window-basic-main-outputs.hpp"
  5. using namespace std;
  6. static void OBSStartStreaming(void *data, calldata_t *params)
  7. {
  8. BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
  9. QMetaObject::invokeMethod(output->main, "StreamingStart");
  10. UNUSED_PARAMETER(params);
  11. }
  12. static void OBSStopStreaming(void *data, calldata_t *params)
  13. {
  14. BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
  15. int code = (int)calldata_int(params, "code");
  16. QMetaObject::invokeMethod(output->main,
  17. "StreamingStop", Q_ARG(int, code));
  18. output->activeRefs--;
  19. }
  20. static void OBSStartRecording(void *data, calldata_t *params)
  21. {
  22. BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
  23. QMetaObject::invokeMethod(output->main, "RecordingStart");
  24. UNUSED_PARAMETER(params);
  25. }
  26. static void OBSStopRecording(void *data, calldata_t *params)
  27. {
  28. BasicOutputHandler *output = static_cast<BasicOutputHandler*>(data);
  29. int code = (int)calldata_int(params, "code");
  30. QMetaObject::invokeMethod(output->main,
  31. "RecordingStop", Q_ARG(int, code));
  32. output->activeRefs--;
  33. UNUSED_PARAMETER(params);
  34. }
  35. /* ------------------------------------------------------------------------ */
  36. static OBSEncoder CreateAACEncoder(const char *name, size_t idx)
  37. {
  38. static const char *encoders[] = {
  39. "CoreAudio_AAC",
  40. "libfdk_aac",
  41. "ffmpeg_aac"
  42. };
  43. OBSEncoder result;
  44. for (const char *encoder : encoders) {
  45. result = obs_audio_encoder_create(encoder, name, nullptr, idx,
  46. nullptr);
  47. if (result) {
  48. obs_encoder_release(result);
  49. break;
  50. }
  51. }
  52. return result;
  53. }
  54. /* ------------------------------------------------------------------------ */
  55. struct SimpleOutput : BasicOutputHandler {
  56. OBSEncoder aac;
  57. OBSEncoder h264;
  58. SimpleOutput(OBSBasic *main_);
  59. virtual void Update() override;
  60. void SetupOutputs();
  61. virtual bool StartStreaming(obs_service_t *service) override;
  62. virtual bool StartRecording() override;
  63. virtual void StopStreaming() override;
  64. virtual void StopRecording() override;
  65. virtual bool StreamingActive() const override;
  66. virtual bool RecordingActive() const override;
  67. };
  68. SimpleOutput::SimpleOutput(OBSBasic *main_) : BasicOutputHandler(main_)
  69. {
  70. streamOutput = obs_output_create("rtmp_output", "simple_stream",
  71. nullptr, nullptr);
  72. if (!streamOutput)
  73. throw "Failed to create stream output (simple output)";
  74. obs_output_release(streamOutput);
  75. fileOutput = obs_output_create("ffmpeg_muxer", "simple_file_output",
  76. nullptr, nullptr);
  77. if (!fileOutput)
  78. throw "Failed to create recording output (simple output)";
  79. obs_output_release(fileOutput);
  80. h264 = obs_video_encoder_create("obs_x264", "simple_h264", nullptr,
  81. nullptr);
  82. if (!h264)
  83. throw "Failed to create h264 encoder (simple output)";
  84. obs_encoder_release(h264);
  85. aac = CreateAACEncoder("simple_aac", 0);
  86. if (!aac)
  87. throw "Failed to create audio encoder (simple output)";
  88. signal_handler_connect(obs_output_get_signal_handler(streamOutput),
  89. "start", OBSStartStreaming, this);
  90. signal_handler_connect(obs_output_get_signal_handler(streamOutput),
  91. "stop", OBSStopStreaming, this);
  92. signal_handler_connect(obs_output_get_signal_handler(fileOutput),
  93. "start", OBSStartRecording, this);
  94. signal_handler_connect(obs_output_get_signal_handler(fileOutput),
  95. "stop", OBSStopRecording, this);
  96. }
  97. void SimpleOutput::Update()
  98. {
  99. obs_data_t *h264Settings = obs_data_create();
  100. obs_data_t *aacSettings = obs_data_create();
  101. int videoBitrate = config_get_uint(main->Config(), "SimpleOutput",
  102. "VBitrate");
  103. int videoBufsize = config_get_uint(main->Config(), "SimpleOutput",
  104. "VBufsize");
  105. int audioBitrate = config_get_uint(main->Config(), "SimpleOutput",
  106. "ABitrate");
  107. bool advanced = config_get_bool(main->Config(), "SimpleOutput",
  108. "UseAdvanced");
  109. bool useCBR = config_get_bool(main->Config(), "SimpleOutput",
  110. "UseCBR");
  111. bool useBufsize = config_get_bool(main->Config(), "SimpleOutput",
  112. "UseBufsize");
  113. const char *preset = config_get_string(main->Config(),
  114. "SimpleOutput", "Preset");
  115. const char *custom = config_get_string(main->Config(),
  116. "SimpleOutput", "x264Settings");
  117. obs_data_set_int(h264Settings, "bitrate", videoBitrate);
  118. obs_data_set_bool(h264Settings, "use_bufsize", useBufsize);
  119. obs_data_set_int(h264Settings, "buffer_size", videoBufsize);
  120. if (advanced) {
  121. obs_data_set_string(h264Settings, "preset", preset);
  122. obs_data_set_string(h264Settings, "x264opts", custom);
  123. obs_data_set_bool(h264Settings, "cbr", useCBR);
  124. } else {
  125. obs_data_set_bool(h264Settings, "cbr", true);
  126. }
  127. obs_data_set_int(aacSettings, "bitrate", audioBitrate);
  128. obs_service_apply_encoder_settings(main->GetService(),
  129. h264Settings, aacSettings);
  130. video_t *video = obs_get_video();
  131. enum video_format format = video_output_get_format(video);
  132. if (format != VIDEO_FORMAT_NV12 && format != VIDEO_FORMAT_I420)
  133. obs_encoder_set_preferred_video_format(h264, VIDEO_FORMAT_NV12);
  134. obs_encoder_update(h264, h264Settings);
  135. obs_encoder_update(aac, aacSettings);
  136. obs_data_release(h264Settings);
  137. obs_data_release(aacSettings);
  138. }
  139. inline void SimpleOutput::SetupOutputs()
  140. {
  141. SimpleOutput::Update();
  142. obs_encoder_set_video(h264, obs_get_video());
  143. obs_encoder_set_audio(aac, obs_get_audio());
  144. }
  145. bool SimpleOutput::StartStreaming(obs_service_t *service)
  146. {
  147. if (!Active())
  148. SetupOutputs();
  149. obs_output_set_video_encoder(streamOutput, h264);
  150. obs_output_set_audio_encoder(streamOutput, aac, 0);
  151. obs_output_set_service(streamOutput, service);
  152. bool reconnect = config_get_bool(main->Config(), "SimpleOutput",
  153. "Reconnect");
  154. int retryDelay = config_get_uint(main->Config(), "SimpleOutput",
  155. "RetryDelay");
  156. int maxRetries = config_get_uint(main->Config(), "SimpleOutput",
  157. "MaxRetries");
  158. if (!reconnect)
  159. maxRetries = 0;
  160. obs_output_set_reconnect_settings(streamOutput, maxRetries,
  161. retryDelay);
  162. if (obs_output_start(streamOutput)) {
  163. activeRefs++;
  164. return true;
  165. }
  166. return false;
  167. }
  168. bool SimpleOutput::StartRecording()
  169. {
  170. if (!Active())
  171. SetupOutputs();
  172. const char *path = config_get_string(main->Config(),
  173. "SimpleOutput", "FilePath");
  174. const char *format = config_get_string(main->Config(),
  175. "SimpleOutput", "RecFormat");
  176. os_dir_t *dir = path ? os_opendir(path) : nullptr;
  177. if (!dir) {
  178. QMessageBox::information(main,
  179. QTStr("Output.BadPath.Title"),
  180. QTStr("Output.BadPath.Text"));
  181. return false;
  182. }
  183. os_closedir(dir);
  184. string strPath;
  185. strPath += path;
  186. char lastChar = strPath.back();
  187. if (lastChar != '/' && lastChar != '\\')
  188. strPath += "/";
  189. strPath += GenerateTimeDateFilename(format);
  190. SetupOutputs();
  191. obs_output_set_video_encoder(fileOutput, h264);
  192. obs_output_set_audio_encoder(fileOutput, aac, 0);
  193. obs_data_t *settings = obs_data_create();
  194. obs_data_set_string(settings, "path", strPath.c_str());
  195. obs_output_update(fileOutput, settings);
  196. obs_data_release(settings);
  197. if (obs_output_start(fileOutput)) {
  198. activeRefs++;
  199. return true;
  200. }
  201. return false;
  202. }
  203. void SimpleOutput::StopStreaming()
  204. {
  205. obs_output_stop(streamOutput);
  206. }
  207. void SimpleOutput::StopRecording()
  208. {
  209. obs_output_stop(fileOutput);
  210. }
  211. bool SimpleOutput::StreamingActive() const
  212. {
  213. return obs_output_active(streamOutput);
  214. }
  215. bool SimpleOutput::RecordingActive() const
  216. {
  217. return obs_output_active(fileOutput);
  218. }
  219. /* ------------------------------------------------------------------------ */
  220. struct AdvancedOutput : BasicOutputHandler {
  221. OBSEncoder aacTrack[4];
  222. OBSEncoder h264Streaming;
  223. OBSEncoder h264Recording;
  224. bool ffmpegRecording;
  225. bool useStreamEncoder;
  226. AdvancedOutput(OBSBasic *main_);
  227. inline void UpdateStreamSettings();
  228. inline void UpdateRecordingSettings();
  229. inline void UpdateAudioSettings();
  230. virtual void Update() override;
  231. inline void SetupStreaming();
  232. inline void SetupRecording();
  233. inline void SetupFFmpeg();
  234. void SetupOutputs();
  235. virtual bool StartStreaming(obs_service_t *service) override;
  236. virtual bool StartRecording() override;
  237. virtual void StopStreaming() override;
  238. virtual void StopRecording() override;
  239. virtual bool StreamingActive() const override;
  240. virtual bool RecordingActive() const override;
  241. };
  242. static OBSData GetDataFromJsonFile(const char *jsonFile)
  243. {
  244. char fullPath[512];
  245. int ret = GetConfigPath(fullPath, sizeof(fullPath), jsonFile);
  246. if (ret > 0) {
  247. BPtr<char> jsonData = os_quick_read_utf8_file(fullPath);
  248. if (!!jsonData) {
  249. obs_data_t *data = obs_data_create_from_json(jsonData);
  250. OBSData dataRet(data);
  251. obs_data_release(data);
  252. return dataRet;
  253. }
  254. }
  255. return nullptr;
  256. }
  257. AdvancedOutput::AdvancedOutput(OBSBasic *main_) : BasicOutputHandler(main_)
  258. {
  259. const char *recType = config_get_string(main->Config(), "AdvOut",
  260. "RecType");
  261. const char *streamEncoder = config_get_string(main->Config(), "AdvOut",
  262. "Encoder");
  263. const char *recordEncoder = config_get_string(main->Config(), "AdvOut",
  264. "RecEncoder");
  265. ffmpegRecording = astrcmpi(recType, "FFmpeg") == 0;
  266. useStreamEncoder = astrcmpi(recordEncoder, "none") == 0;
  267. OBSData streamEncSettings = GetDataFromJsonFile(
  268. "obs-studio/basic/streamEncoder.json");
  269. OBSData recordEncSettings = GetDataFromJsonFile(
  270. "obs-studio/basic/recordEncoder.json");
  271. streamOutput = obs_output_create("rtmp_output", "adv_stream",
  272. nullptr, nullptr);
  273. if (!streamOutput)
  274. throw "Failed to create stream output (advanced output)";
  275. obs_output_release(streamOutput);
  276. if (ffmpegRecording) {
  277. fileOutput = obs_output_create("ffmpeg_output",
  278. "adv_ffmpeg_output", nullptr, nullptr);
  279. if (!fileOutput)
  280. throw "Failed to create recording FFmpeg output "
  281. "(advanced output)";
  282. obs_output_release(fileOutput);
  283. } else {
  284. fileOutput = obs_output_create("ffmpeg_muxer",
  285. "adv_file_output", nullptr, nullptr);
  286. if (!fileOutput)
  287. throw "Failed to create recording output "
  288. "(advanced output)";
  289. obs_output_release(fileOutput);
  290. if (!useStreamEncoder) {
  291. h264Recording = obs_video_encoder_create(recordEncoder,
  292. "recording_h264", recordEncSettings,
  293. nullptr);
  294. if (!h264Recording)
  295. throw "Failed to create recording h264 "
  296. "encoder (advanced output)";
  297. obs_encoder_release(h264Recording);
  298. }
  299. }
  300. h264Streaming = obs_video_encoder_create(streamEncoder,
  301. "streaming_h264", streamEncSettings, nullptr);
  302. if (!h264Streaming)
  303. throw "Failed to create streaming h264 encoder "
  304. "(advanced output)";
  305. obs_encoder_release(h264Streaming);
  306. for (int i = 0; i < 4; i++) {
  307. char name[9];
  308. sprintf(name, "adv_aac%d", i);
  309. aacTrack[i] = CreateAACEncoder(name, i);
  310. if (!aacTrack[i])
  311. throw "Failed to create audio encoder "
  312. "(advanced output)";
  313. }
  314. signal_handler_connect(obs_output_get_signal_handler(streamOutput),
  315. "start", OBSStartStreaming, this);
  316. signal_handler_connect(obs_output_get_signal_handler(streamOutput),
  317. "stop", OBSStopStreaming, this);
  318. signal_handler_connect(obs_output_get_signal_handler(fileOutput),
  319. "start", OBSStartRecording, this);
  320. signal_handler_connect(obs_output_get_signal_handler(fileOutput),
  321. "stop", OBSStopRecording, this);
  322. }
  323. void AdvancedOutput::UpdateStreamSettings()
  324. {
  325. bool applyServiceSettings = config_get_bool(main->Config(), "AdvOut",
  326. "ApplyServiceSettings");
  327. OBSData settings = GetDataFromJsonFile(
  328. "obs-studio/basic/streamEncoder.json");
  329. if (applyServiceSettings)
  330. obs_service_apply_encoder_settings(main->GetService(),
  331. settings, nullptr);
  332. video_t *video = obs_get_video();
  333. enum video_format format = video_output_get_format(video);
  334. if (format != VIDEO_FORMAT_NV12 && format != VIDEO_FORMAT_I420)
  335. obs_encoder_set_preferred_video_format(h264Streaming,
  336. VIDEO_FORMAT_NV12);
  337. obs_encoder_update(h264Streaming, settings);
  338. }
  339. inline void AdvancedOutput::UpdateRecordingSettings()
  340. {
  341. OBSData settings = GetDataFromJsonFile(
  342. "obs-studio/basic/recordEncoder.json");
  343. obs_encoder_update(h264Recording, settings);
  344. }
  345. void AdvancedOutput::Update()
  346. {
  347. UpdateStreamSettings();
  348. if (!useStreamEncoder && !ffmpegRecording)
  349. UpdateRecordingSettings();
  350. UpdateAudioSettings();
  351. }
  352. inline void AdvancedOutput::SetupStreaming()
  353. {
  354. bool rescale = config_get_bool(main->Config(), "AdvOut",
  355. "Rescale");
  356. const char *rescaleRes = config_get_string(main->Config(), "AdvOut",
  357. "RescaleRes");
  358. bool multitrack = config_get_bool(main->Config(), "AdvOut",
  359. "Multitrack");
  360. int trackIndex = config_get_int(main->Config(), "AdvOut",
  361. "TrackIndex");
  362. int trackCount = config_get_int(main->Config(), "AdvOut",
  363. "TrackCount");
  364. unsigned int cx = 0;
  365. unsigned int cy = 0;
  366. if (rescale && rescaleRes && *rescaleRes) {
  367. if (sscanf(rescaleRes, "%ux%u", &cx, &cy) != 2) {
  368. cx = 0;
  369. cy = 0;
  370. }
  371. }
  372. obs_encoder_set_scaled_size(h264Streaming, cx, cy);
  373. obs_encoder_set_video(h264Streaming, obs_get_video());
  374. obs_output_set_video_encoder(streamOutput, h264Streaming);
  375. if (multitrack) {
  376. int i = 0;
  377. for (; i < trackCount; i++)
  378. obs_output_set_audio_encoder(streamOutput, aacTrack[i],
  379. i);
  380. for (; i < 4; i++)
  381. obs_output_set_audio_encoder(streamOutput, nullptr, i);
  382. } else {
  383. obs_output_set_audio_encoder(streamOutput,
  384. aacTrack[trackIndex - 1], 0);
  385. }
  386. }
  387. inline void AdvancedOutput::SetupRecording()
  388. {
  389. const char *path = config_get_string(main->Config(), "AdvOut",
  390. "RecFilePath");
  391. bool rescale = config_get_bool(main->Config(), "AdvOut",
  392. "RecRescale");
  393. const char *rescaleRes = config_get_string(main->Config(), "AdvOut",
  394. "RecRescaleRes");
  395. int tracks = config_get_int(main->Config(), "AdvOut", "RecTracks");
  396. obs_data_t *settings = obs_data_create();
  397. unsigned int cx = 0;
  398. unsigned int cy = 0;
  399. int idx = 0;
  400. if (useStreamEncoder) {
  401. obs_output_set_video_encoder(fileOutput, h264Streaming);
  402. } else {
  403. if (rescale && rescaleRes && *rescaleRes) {
  404. if (sscanf(rescaleRes, "%ux%u", &cx, &cy) != 2) {
  405. cx = 0;
  406. cy = 0;
  407. }
  408. }
  409. obs_encoder_set_scaled_size(h264Recording, cx, cy);
  410. obs_encoder_set_video(h264Recording, obs_get_video());
  411. obs_output_set_video_encoder(fileOutput, h264Recording);
  412. }
  413. for (int i = 0; i < MAX_AUDIO_MIXES; i++) {
  414. if ((tracks & (1<<i)) != 0) {
  415. obs_output_set_audio_encoder(fileOutput, aacTrack[i],
  416. idx++);
  417. }
  418. }
  419. obs_data_set_string(settings, "path", path);
  420. obs_output_update(fileOutput, settings);
  421. obs_data_release(settings);
  422. }
  423. inline void AdvancedOutput::SetupFFmpeg()
  424. {
  425. const char *url = config_get_string(main->Config(), "AdvOut", "FFURL");
  426. int vBitrate = config_get_int(main->Config(), "AdvOut",
  427. "FFVBitrate");
  428. bool rescale = config_get_bool(main->Config(), "AdvOut",
  429. "FFRescale");
  430. const char *rescaleRes = config_get_string(main->Config(), "AdvOut",
  431. "FFRescaleRes");
  432. const char *formatName = config_get_string(main->Config(), "AdvOut",
  433. "FFFormat");
  434. const char *mimeType = config_get_string(main->Config(), "AdvOut",
  435. "FFFormatMimeType");
  436. const char *vEncoder = config_get_string(main->Config(), "AdvOut",
  437. "FFVEncoder");
  438. int vEncoderId = config_get_int(main->Config(), "AdvOut",
  439. "FFVEncoderId");
  440. const char *vEncCustom = config_get_string(main->Config(), "AdvOut",
  441. "FFVCustom");
  442. int aBitrate = config_get_int(main->Config(), "AdvOut",
  443. "FFABitrate");
  444. int aTrack = config_get_int(main->Config(), "AdvOut",
  445. "FFAudioTrack");
  446. const char *aEncoder = config_get_string(main->Config(), "AdvOut",
  447. "FFAEncoder");
  448. int aEncoderId = config_get_int(main->Config(), "AdvOut",
  449. "FFAEncoderId");
  450. const char *aEncCustom = config_get_string(main->Config(), "AdvOut",
  451. "FFACustom");
  452. obs_data_t *settings = obs_data_create();
  453. obs_data_set_string(settings, "url", url);
  454. obs_data_set_string(settings, "format_name", formatName);
  455. obs_data_set_string(settings, "format_mime_type", mimeType);
  456. obs_data_set_int(settings, "video_bitrate", vBitrate);
  457. obs_data_set_string(settings, "video_encoder", vEncoder);
  458. obs_data_set_int(settings, "video_encoder_id", vEncoderId);
  459. obs_data_set_string(settings, "video_settings", vEncCustom);
  460. obs_data_set_int(settings, "audio_bitrate", aBitrate);
  461. obs_data_set_string(settings, "audio_encoder", aEncoder);
  462. obs_data_set_int(settings, "audio_encoder_id", aEncoderId);
  463. obs_data_set_string(settings, "audio_settings", aEncCustom);
  464. if (rescale && rescaleRes && *rescaleRes) {
  465. int width;
  466. int height;
  467. int val = sscanf(rescaleRes, "%dx%d", &width, &height);
  468. if (val == 2 && width && height) {
  469. obs_data_set_int(settings, "scale_width", width);
  470. obs_data_set_int(settings, "scale_height", height);
  471. }
  472. }
  473. obs_output_set_mixer(fileOutput, aTrack - 1);
  474. obs_output_set_media(fileOutput, obs_get_video(), obs_get_audio());
  475. obs_output_update(fileOutput, settings);
  476. obs_data_release(settings);
  477. }
  478. static inline void SetEncoderName(obs_encoder_t *encoder, const char *name,
  479. const char *defaultName)
  480. {
  481. obs_encoder_set_name(encoder, (name && *name) ? name : defaultName);
  482. }
  483. inline void AdvancedOutput::UpdateAudioSettings()
  484. {
  485. int track1Bitrate = config_get_uint(main->Config(), "AdvOut",
  486. "Track1Bitrate");
  487. int track2Bitrate = config_get_uint(main->Config(), "AdvOut",
  488. "Track2Bitrate");
  489. int track3Bitrate = config_get_uint(main->Config(), "AdvOut",
  490. "Track3Bitrate");
  491. int track4Bitrate = config_get_uint(main->Config(), "AdvOut",
  492. "Track4Bitrate");
  493. const char *name1 = config_get_string(main->Config(), "AdvOut",
  494. "Track1Name");
  495. const char *name2 = config_get_string(main->Config(), "AdvOut",
  496. "Track2Name");
  497. const char *name3 = config_get_string(main->Config(), "AdvOut",
  498. "Track3Name");
  499. const char *name4 = config_get_string(main->Config(), "AdvOut",
  500. "Track4Name");
  501. bool applyServiceSettings = config_get_bool(main->Config(), "AdvOut",
  502. "ApplyServiceSettings");
  503. int streamTrackIndex = config_get_int(main->Config(), "AdvOut",
  504. "TrackIndex");
  505. obs_data_t *settings[4];
  506. for (size_t i = 0; i < 4; i++)
  507. settings[i] = obs_data_create();
  508. obs_data_set_int(settings[0], "bitrate", track1Bitrate);
  509. obs_data_set_int(settings[1], "bitrate", track2Bitrate);
  510. obs_data_set_int(settings[2], "bitrate", track3Bitrate);
  511. obs_data_set_int(settings[3], "bitrate", track4Bitrate);
  512. SetEncoderName(aacTrack[0], name1, "Track1");
  513. SetEncoderName(aacTrack[1], name2, "Track2");
  514. SetEncoderName(aacTrack[2], name3, "Track3");
  515. SetEncoderName(aacTrack[3], name4, "Track4");
  516. for (size_t i = 0; i < 4; i++) {
  517. if (applyServiceSettings && (int)(i + 1) == streamTrackIndex)
  518. obs_service_apply_encoder_settings(main->GetService(),
  519. nullptr, settings[i]);
  520. obs_encoder_update(aacTrack[i], settings[i]);
  521. obs_data_release(settings[i]);
  522. }
  523. }
  524. void AdvancedOutput::SetupOutputs()
  525. {
  526. obs_encoder_set_video(h264Streaming, obs_get_video());
  527. if (h264Recording)
  528. obs_encoder_set_video(h264Recording, obs_get_video());
  529. obs_encoder_set_audio(aacTrack[0], obs_get_audio());
  530. obs_encoder_set_audio(aacTrack[1], obs_get_audio());
  531. obs_encoder_set_audio(aacTrack[2], obs_get_audio());
  532. obs_encoder_set_audio(aacTrack[3], obs_get_audio());
  533. SetupStreaming();
  534. if (ffmpegRecording)
  535. SetupFFmpeg();
  536. else
  537. SetupRecording();
  538. }
  539. bool AdvancedOutput::StartStreaming(obs_service_t *service)
  540. {
  541. if (!useStreamEncoder ||
  542. (!ffmpegRecording && !obs_output_active(fileOutput))) {
  543. UpdateStreamSettings();
  544. }
  545. UpdateAudioSettings();
  546. if (!Active())
  547. SetupOutputs();
  548. obs_output_set_service(streamOutput, service);
  549. bool reconnect = config_get_bool(main->Config(), "AdvOut", "Reconnect");
  550. int retryDelay = config_get_int(main->Config(), "AdvOut", "RetryDelay");
  551. int maxRetries = config_get_int(main->Config(), "AdvOut", "MaxRetries");
  552. if (!reconnect)
  553. maxRetries = 0;
  554. obs_output_set_reconnect_settings(streamOutput, maxRetries,
  555. retryDelay);
  556. if (obs_output_start(streamOutput)) {
  557. activeRefs++;
  558. return true;
  559. }
  560. return false;
  561. }
  562. bool AdvancedOutput::StartRecording()
  563. {
  564. if (!useStreamEncoder) {
  565. if (!ffmpegRecording) {
  566. UpdateRecordingSettings();
  567. }
  568. } else if (!obs_output_active(streamOutput)) {
  569. UpdateStreamSettings();
  570. }
  571. UpdateAudioSettings();
  572. if (!Active())
  573. SetupOutputs();
  574. if (!ffmpegRecording) {
  575. const char *path = config_get_string(main->Config(),
  576. "AdvOut", "RecFilePath");
  577. const char *format = config_get_string(main->Config(),
  578. "AdvOut", "RecFormat");
  579. os_dir_t *dir = path ? os_opendir(path) : nullptr;
  580. if (!dir) {
  581. QMessageBox::information(main,
  582. QTStr("Output.BadPath.Title"),
  583. QTStr("Output.BadPath.Text"));
  584. return false;
  585. }
  586. os_closedir(dir);
  587. string strPath;
  588. strPath += path;
  589. char lastChar = strPath.back();
  590. if (lastChar != '/' && lastChar != '\\')
  591. strPath += "/";
  592. strPath += GenerateTimeDateFilename(format);
  593. obs_data_t *settings = obs_data_create();
  594. obs_data_set_string(settings, "path", strPath.c_str());
  595. obs_output_update(fileOutput, settings);
  596. obs_data_release(settings);
  597. }
  598. if (obs_output_start(fileOutput)) {
  599. activeRefs++;
  600. return true;
  601. }
  602. return false;
  603. }
  604. void AdvancedOutput::StopStreaming()
  605. {
  606. obs_output_stop(streamOutput);
  607. }
  608. void AdvancedOutput::StopRecording()
  609. {
  610. obs_output_stop(fileOutput);
  611. }
  612. bool AdvancedOutput::StreamingActive() const
  613. {
  614. return obs_output_active(streamOutput);
  615. }
  616. bool AdvancedOutput::RecordingActive() const
  617. {
  618. return obs_output_active(fileOutput);
  619. }
  620. /* ------------------------------------------------------------------------ */
  621. BasicOutputHandler *CreateSimpleOutputHandler(OBSBasic *main)
  622. {
  623. return new SimpleOutput(main);
  624. }
  625. BasicOutputHandler *CreateAdvancedOutputHandler(OBSBasic *main)
  626. {
  627. return new AdvancedOutput(main);
  628. }