window-basic-main-outputs.cpp 6.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251
  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. QMetaObject::invokeMethod(output->main, "RecordingStop");
  30. output->activeRefs--;
  31. UNUSED_PARAMETER(params);
  32. }
  33. /* ------------------------------------------------------------------------ */
  34. struct SimpleOutput : BasicOutputHandler {
  35. OBSEncoder aac;
  36. OBSEncoder h264;
  37. SimpleOutput(OBSBasic *main_);
  38. virtual void Update() override;
  39. void SetupOutputs();
  40. virtual bool StartStreaming(obs_service_t *service) override;
  41. virtual bool StartRecording() override;
  42. virtual void StopStreaming() override;
  43. virtual void StopRecording() override;
  44. virtual bool StreamingActive() const override;
  45. virtual bool RecordingActive() const override;
  46. };
  47. SimpleOutput::SimpleOutput(OBSBasic *main_) : BasicOutputHandler(main_)
  48. {
  49. streamOutput = obs_output_create("rtmp_output", "simple_stream",
  50. nullptr);
  51. if (!streamOutput)
  52. throw "Failed to create stream output (simple output)";
  53. fileOutput = obs_output_create("flv_output", "simple_file_output",
  54. nullptr);
  55. if (!fileOutput)
  56. throw "Failed to create recording output (simple output)";
  57. h264 = obs_video_encoder_create("obs_x264", "simple_h264", nullptr);
  58. if (!h264)
  59. throw "Failed to create h264 encoder (simple output)";
  60. aac = obs_audio_encoder_create("libfdk_aac", "simple_aac", nullptr, 0);
  61. if (!aac)
  62. aac = obs_audio_encoder_create("ffmpeg_aac", "simple_aac",
  63. nullptr, 0);
  64. if (!aac)
  65. throw "Failed to create audio encoder (simple output)";
  66. signal_handler_connect(obs_output_get_signal_handler(streamOutput),
  67. "start", OBSStartStreaming, this);
  68. signal_handler_connect(obs_output_get_signal_handler(streamOutput),
  69. "stop", OBSStopStreaming, this);
  70. signal_handler_connect(obs_output_get_signal_handler(fileOutput),
  71. "start", OBSStartRecording, this);
  72. signal_handler_connect(obs_output_get_signal_handler(fileOutput),
  73. "stop", OBSStopRecording, this);
  74. }
  75. void SimpleOutput::Update()
  76. {
  77. obs_data_t *h264Settings = obs_data_create();
  78. obs_data_t *aacSettings = obs_data_create();
  79. int videoBitrate = config_get_uint(main->Config(), "SimpleOutput",
  80. "VBitrate");
  81. int videoBufsize = config_get_uint(main->Config(), "SimpleOutput",
  82. "Bufsize");
  83. int audioBitrate = config_get_uint(main->Config(), "SimpleOutput",
  84. "ABitrate");
  85. bool advanced = config_get_bool(main->Config(), "SimpleOutput",
  86. "UseAdvanced");
  87. bool useCBR = config_get_bool(main->Config(), "SimpleOutput",
  88. "UseCBR");
  89. bool useBufsize = config_get_bool(main->Config(), "SimpleOutput",
  90. "UseBufsize");
  91. const char *preset = config_get_string(main->Config(),
  92. "SimpleOutput", "Preset");
  93. const char *custom = config_get_string(main->Config(),
  94. "SimpleOutput", "x264Settings");
  95. obs_data_set_int(h264Settings, "bitrate", videoBitrate);
  96. obs_data_set_bool(h264Settings, "use_bufsize", useBufsize);
  97. obs_data_set_int(h264Settings, "buffer_size", videoBufsize);
  98. if (advanced) {
  99. obs_data_set_string(h264Settings, "preset", preset);
  100. obs_data_set_string(h264Settings, "x264opts", custom);
  101. obs_data_set_bool(h264Settings, "cbr", useCBR);
  102. } else {
  103. obs_data_set_bool(h264Settings, "cbr", true);
  104. }
  105. obs_data_set_int(aacSettings, "bitrate", audioBitrate);
  106. obs_encoder_update(h264, h264Settings);
  107. obs_encoder_update(aac, aacSettings);
  108. obs_data_release(h264Settings);
  109. obs_data_release(aacSettings);
  110. }
  111. inline void SimpleOutput::SetupOutputs()
  112. {
  113. SimpleOutput::Update();
  114. obs_encoder_set_video(h264, obs_get_video());
  115. obs_encoder_set_audio(aac, obs_get_audio());
  116. }
  117. bool SimpleOutput::StartStreaming(obs_service_t *service)
  118. {
  119. if (!Active())
  120. SetupOutputs();
  121. obs_output_set_video_encoder(streamOutput, h264);
  122. obs_output_set_audio_encoder(streamOutput, aac, 0);
  123. obs_output_set_service(streamOutput, service);
  124. bool reconnect = config_get_bool(main->Config(), "SimpleOutput",
  125. "Reconnect");
  126. int retryDelay = config_get_uint(main->Config(), "SimpleOutput",
  127. "RetryDelay");
  128. int maxRetries = config_get_uint(main->Config(), "SimpleOutput",
  129. "MaxRetries");
  130. if (!reconnect)
  131. maxRetries = 0;
  132. obs_output_set_reconnect_settings(streamOutput, maxRetries,
  133. retryDelay);
  134. if (obs_output_start(streamOutput)) {
  135. activeRefs++;
  136. return true;
  137. }
  138. return false;
  139. }
  140. bool SimpleOutput::StartRecording()
  141. {
  142. if (!Active())
  143. SetupOutputs();
  144. const char *path = config_get_string(main->Config(),
  145. "SimpleOutput", "FilePath");
  146. os_dir_t *dir = path ? os_opendir(path) : nullptr;
  147. if (!dir) {
  148. QMessageBox::information(main,
  149. QTStr("Output.BadPath.Title"),
  150. QTStr("Output.BadPath.Text"));
  151. return false;
  152. }
  153. os_closedir(dir);
  154. string strPath;
  155. strPath += path;
  156. char lastChar = strPath.back();
  157. if (lastChar != '/' && lastChar != '\\')
  158. strPath += "/";
  159. strPath += GenerateTimeDateFilename("flv");
  160. SetupOutputs();
  161. obs_output_set_video_encoder(fileOutput, h264);
  162. obs_output_set_audio_encoder(fileOutput, aac, 0);
  163. obs_data_t *settings = obs_data_create();
  164. obs_data_set_string(settings, "path", strPath.c_str());
  165. obs_output_update(fileOutput, settings);
  166. obs_data_release(settings);
  167. if (obs_output_start(fileOutput)) {
  168. activeRefs++;
  169. return true;
  170. }
  171. return false;
  172. }
  173. void SimpleOutput::StopStreaming()
  174. {
  175. obs_output_stop(streamOutput);
  176. }
  177. void SimpleOutput::StopRecording()
  178. {
  179. obs_output_stop(fileOutput);
  180. }
  181. bool SimpleOutput::StreamingActive() const
  182. {
  183. return obs_output_active(streamOutput);
  184. }
  185. bool SimpleOutput::RecordingActive() const
  186. {
  187. return obs_output_active(fileOutput);
  188. }
  189. /* ------------------------------------------------------------------------ */
  190. BasicOutputHandler *CreateSimpleOutputHandler(OBSBasic *main)
  191. {
  192. return new SimpleOutput(main);
  193. }