1
0

window-basic-auto-config.cpp 27 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004
  1. #include "window-basic-auto-config.hpp"
  2. #include "window-basic-main.hpp"
  3. #include "qt-wrappers.hpp"
  4. #include "obs-app.hpp"
  5. #include <QMessageBox>
  6. #include <QScreen>
  7. #include <obs.hpp>
  8. #include "ui_AutoConfigStartPage.h"
  9. #include "ui_AutoConfigVideoPage.h"
  10. #include "ui_AutoConfigStreamPage.h"
  11. #ifdef BROWSER_AVAILABLE
  12. #include <browser-panel.hpp>
  13. #include "auth-oauth.hpp"
  14. #endif
  15. struct QCef;
  16. struct QCefCookieManager;
  17. extern QCef *cef;
  18. extern QCefCookieManager *panel_cookies;
  19. #define wiz reinterpret_cast<AutoConfig *>(wizard())
  20. /* ------------------------------------------------------------------------- */
  21. #define SERVICE_PATH "service.json"
  22. static OBSData OpenServiceSettings(std::string &type)
  23. {
  24. char serviceJsonPath[512];
  25. int ret = GetProfilePath(serviceJsonPath, sizeof(serviceJsonPath),
  26. SERVICE_PATH);
  27. if (ret <= 0)
  28. return OBSData();
  29. OBSData data =
  30. obs_data_create_from_json_file_safe(serviceJsonPath, "bak");
  31. obs_data_release(data);
  32. obs_data_set_default_string(data, "type", "rtmp_common");
  33. type = obs_data_get_string(data, "type");
  34. OBSData settings = obs_data_get_obj(data, "settings");
  35. obs_data_release(settings);
  36. return settings;
  37. }
  38. static void GetServiceInfo(std::string &type, std::string &service,
  39. std::string &server, std::string &key)
  40. {
  41. OBSData settings = OpenServiceSettings(type);
  42. service = obs_data_get_string(settings, "service");
  43. server = obs_data_get_string(settings, "server");
  44. key = obs_data_get_string(settings, "key");
  45. }
  46. /* ------------------------------------------------------------------------- */
  47. AutoConfigStartPage::AutoConfigStartPage(QWidget *parent)
  48. : QWizardPage(parent), ui(new Ui_AutoConfigStartPage)
  49. {
  50. ui->setupUi(this);
  51. setTitle(QTStr("Basic.AutoConfig.StartPage"));
  52. setSubTitle(QTStr("Basic.AutoConfig.StartPage.SubTitle"));
  53. }
  54. AutoConfigStartPage::~AutoConfigStartPage()
  55. {
  56. delete ui;
  57. }
  58. int AutoConfigStartPage::nextId() const
  59. {
  60. return AutoConfig::VideoPage;
  61. }
  62. void AutoConfigStartPage::on_prioritizeStreaming_clicked()
  63. {
  64. wiz->type = AutoConfig::Type::Streaming;
  65. }
  66. void AutoConfigStartPage::on_prioritizeRecording_clicked()
  67. {
  68. wiz->type = AutoConfig::Type::Recording;
  69. }
  70. /* ------------------------------------------------------------------------- */
  71. #define RES_TEXT(x) "Basic.AutoConfig.VideoPage." x
  72. #define RES_USE_CURRENT RES_TEXT("BaseResolution.UseCurrent")
  73. #define RES_USE_DISPLAY RES_TEXT("BaseResolution.Display")
  74. #define FPS_USE_CURRENT RES_TEXT("FPS.UseCurrent")
  75. #define FPS_PREFER_HIGH_FPS RES_TEXT("FPS.PreferHighFPS")
  76. #define FPS_PREFER_HIGH_RES RES_TEXT("FPS.PreferHighRes")
  77. AutoConfigVideoPage::AutoConfigVideoPage(QWidget *parent)
  78. : QWizardPage(parent), ui(new Ui_AutoConfigVideoPage)
  79. {
  80. ui->setupUi(this);
  81. setTitle(QTStr("Basic.AutoConfig.VideoPage"));
  82. setSubTitle(QTStr("Basic.AutoConfig.VideoPage.SubTitle"));
  83. obs_video_info ovi;
  84. obs_get_video_info(&ovi);
  85. long double fpsVal =
  86. (long double)ovi.fps_num / (long double)ovi.fps_den;
  87. QString fpsStr = (ovi.fps_den > 1) ? QString::number(fpsVal, 'f', 2)
  88. : QString::number(fpsVal, 'g', 2);
  89. ui->fps->addItem(QTStr(FPS_PREFER_HIGH_FPS),
  90. (int)AutoConfig::FPSType::PreferHighFPS);
  91. ui->fps->addItem(QTStr(FPS_PREFER_HIGH_RES),
  92. (int)AutoConfig::FPSType::PreferHighRes);
  93. ui->fps->addItem(QTStr(FPS_USE_CURRENT).arg(fpsStr),
  94. (int)AutoConfig::FPSType::UseCurrent);
  95. ui->fps->addItem(QStringLiteral("30"), (int)AutoConfig::FPSType::fps30);
  96. ui->fps->addItem(QStringLiteral("60"), (int)AutoConfig::FPSType::fps60);
  97. ui->fps->setCurrentIndex(0);
  98. QString cxStr = QString::number(ovi.base_width);
  99. QString cyStr = QString::number(ovi.base_height);
  100. int encRes = int(ovi.base_width << 16) | int(ovi.base_height);
  101. ui->canvasRes->addItem(QTStr(RES_USE_CURRENT).arg(cxStr, cyStr),
  102. (int)encRes);
  103. QList<QScreen *> screens = QGuiApplication::screens();
  104. for (int i = 0; i < screens.size(); i++) {
  105. QScreen *screen = screens[i];
  106. QSize as = screen->size();
  107. encRes = int(as.width() << 16) | int(as.height());
  108. QString str = QTStr(RES_USE_DISPLAY)
  109. .arg(QString::number(i + 1),
  110. QString::number(as.width()),
  111. QString::number(as.height()));
  112. ui->canvasRes->addItem(str, encRes);
  113. }
  114. auto addRes = [&](int cx, int cy) {
  115. encRes = (cx << 16) | cy;
  116. QString str = QString("%1x%2").arg(QString::number(cx),
  117. QString::number(cy));
  118. ui->canvasRes->addItem(str, encRes);
  119. };
  120. addRes(1920, 1080);
  121. addRes(1280, 720);
  122. ui->canvasRes->setCurrentIndex(0);
  123. }
  124. AutoConfigVideoPage::~AutoConfigVideoPage()
  125. {
  126. delete ui;
  127. }
  128. int AutoConfigVideoPage::nextId() const
  129. {
  130. return wiz->type == AutoConfig::Type::Recording
  131. ? AutoConfig::TestPage
  132. : AutoConfig::StreamPage;
  133. }
  134. bool AutoConfigVideoPage::validatePage()
  135. {
  136. int encRes = ui->canvasRes->currentData().toInt();
  137. wiz->baseResolutionCX = encRes >> 16;
  138. wiz->baseResolutionCY = encRes & 0xFFFF;
  139. wiz->fpsType = (AutoConfig::FPSType)ui->fps->currentData().toInt();
  140. obs_video_info ovi;
  141. obs_get_video_info(&ovi);
  142. switch (wiz->fpsType) {
  143. case AutoConfig::FPSType::PreferHighFPS:
  144. wiz->specificFPSNum = 0;
  145. wiz->specificFPSDen = 0;
  146. wiz->preferHighFPS = true;
  147. break;
  148. case AutoConfig::FPSType::PreferHighRes:
  149. wiz->specificFPSNum = 0;
  150. wiz->specificFPSDen = 0;
  151. wiz->preferHighFPS = false;
  152. break;
  153. case AutoConfig::FPSType::UseCurrent:
  154. wiz->specificFPSNum = ovi.fps_num;
  155. wiz->specificFPSDen = ovi.fps_den;
  156. wiz->preferHighFPS = false;
  157. break;
  158. case AutoConfig::FPSType::fps30:
  159. wiz->specificFPSNum = 30;
  160. wiz->specificFPSDen = 1;
  161. wiz->preferHighFPS = false;
  162. break;
  163. case AutoConfig::FPSType::fps60:
  164. wiz->specificFPSNum = 60;
  165. wiz->specificFPSDen = 1;
  166. wiz->preferHighFPS = false;
  167. break;
  168. }
  169. return true;
  170. }
  171. /* ------------------------------------------------------------------------- */
  172. enum class ListOpt : int {
  173. ShowAll = 1,
  174. Custom,
  175. };
  176. AutoConfigStreamPage::AutoConfigStreamPage(QWidget *parent)
  177. : QWizardPage(parent), ui(new Ui_AutoConfigStreamPage)
  178. {
  179. ui->setupUi(this);
  180. ui->bitrateLabel->setVisible(false);
  181. ui->bitrate->setVisible(false);
  182. ui->connectAccount2->setVisible(false);
  183. ui->disconnectAccount->setVisible(false);
  184. int vertSpacing = ui->topLayout->verticalSpacing();
  185. QMargins m = ui->topLayout->contentsMargins();
  186. m.setBottom(vertSpacing / 2);
  187. ui->topLayout->setContentsMargins(m);
  188. m = ui->loginPageLayout->contentsMargins();
  189. m.setTop(vertSpacing / 2);
  190. ui->loginPageLayout->setContentsMargins(m);
  191. m = ui->streamkeyPageLayout->contentsMargins();
  192. m.setTop(vertSpacing / 2);
  193. ui->streamkeyPageLayout->setContentsMargins(m);
  194. setTitle(QTStr("Basic.AutoConfig.StreamPage"));
  195. setSubTitle(QTStr("Basic.AutoConfig.StreamPage.SubTitle"));
  196. LoadServices(false);
  197. connect(ui->service, SIGNAL(currentIndexChanged(int)), this,
  198. SLOT(ServiceChanged()));
  199. connect(ui->customServer, SIGNAL(textChanged(const QString &)), this,
  200. SLOT(ServiceChanged()));
  201. connect(ui->doBandwidthTest, SIGNAL(toggled(bool)), this,
  202. SLOT(ServiceChanged()));
  203. connect(ui->service, SIGNAL(currentIndexChanged(int)), this,
  204. SLOT(UpdateServerList()));
  205. connect(ui->service, SIGNAL(currentIndexChanged(int)), this,
  206. SLOT(UpdateKeyLink()));
  207. connect(ui->key, SIGNAL(textChanged(const QString &)), this,
  208. SLOT(UpdateCompleted()));
  209. connect(ui->regionUS, SIGNAL(toggled(bool)), this,
  210. SLOT(UpdateCompleted()));
  211. connect(ui->regionEU, SIGNAL(toggled(bool)), this,
  212. SLOT(UpdateCompleted()));
  213. connect(ui->regionAsia, SIGNAL(toggled(bool)), this,
  214. SLOT(UpdateCompleted()));
  215. connect(ui->regionOther, SIGNAL(toggled(bool)), this,
  216. SLOT(UpdateCompleted()));
  217. }
  218. AutoConfigStreamPage::~AutoConfigStreamPage()
  219. {
  220. delete ui;
  221. }
  222. bool AutoConfigStreamPage::isComplete() const
  223. {
  224. return ready;
  225. }
  226. int AutoConfigStreamPage::nextId() const
  227. {
  228. return AutoConfig::TestPage;
  229. }
  230. inline bool AutoConfigStreamPage::IsCustom() const
  231. {
  232. return ui->service->currentData().toInt() == (int)ListOpt::Custom;
  233. }
  234. bool AutoConfigStreamPage::validatePage()
  235. {
  236. OBSData service_settings = obs_data_create();
  237. obs_data_release(service_settings);
  238. wiz->customServer = IsCustom();
  239. const char *serverType = wiz->customServer ? "rtmp_custom"
  240. : "rtmp_common";
  241. if (!wiz->customServer) {
  242. obs_data_set_string(service_settings, "service",
  243. QT_TO_UTF8(ui->service->currentText()));
  244. }
  245. OBSService service = obs_service_create(serverType, "temp_service",
  246. service_settings, nullptr);
  247. obs_service_release(service);
  248. int bitrate = 10000;
  249. if (!ui->doBandwidthTest->isChecked()) {
  250. bitrate = ui->bitrate->value();
  251. wiz->idealBitrate = bitrate;
  252. }
  253. OBSData settings = obs_data_create();
  254. obs_data_release(settings);
  255. obs_data_set_int(settings, "bitrate", bitrate);
  256. obs_service_apply_encoder_settings(service, settings, nullptr);
  257. if (wiz->customServer) {
  258. QString server = ui->customServer->text();
  259. wiz->server = wiz->serverName = QT_TO_UTF8(server);
  260. } else {
  261. wiz->serverName = QT_TO_UTF8(ui->server->currentText());
  262. wiz->server = QT_TO_UTF8(ui->server->currentData().toString());
  263. }
  264. wiz->bandwidthTest = ui->doBandwidthTest->isChecked();
  265. wiz->startingBitrate = (int)obs_data_get_int(settings, "bitrate");
  266. wiz->idealBitrate = wiz->startingBitrate;
  267. wiz->regionUS = ui->regionUS->isChecked();
  268. wiz->regionEU = ui->regionEU->isChecked();
  269. wiz->regionAsia = ui->regionAsia->isChecked();
  270. wiz->regionOther = ui->regionOther->isChecked();
  271. wiz->serviceName = QT_TO_UTF8(ui->service->currentText());
  272. if (ui->preferHardware)
  273. wiz->preferHardware = ui->preferHardware->isChecked();
  274. wiz->key = QT_TO_UTF8(ui->key->text());
  275. if (!wiz->customServer) {
  276. if (wiz->serviceName == "Twitch")
  277. wiz->service = AutoConfig::Service::Twitch;
  278. else if (wiz->serviceName == "Smashcast")
  279. wiz->service = AutoConfig::Service::Smashcast;
  280. else
  281. wiz->service = AutoConfig::Service::Other;
  282. } else {
  283. wiz->service = AutoConfig::Service::Other;
  284. }
  285. if (wiz->service != AutoConfig::Service::Twitch && wiz->bandwidthTest) {
  286. QMessageBox::StandardButton button;
  287. #define WARNING_TEXT(x) QTStr("Basic.AutoConfig.StreamPage.StreamWarning." x)
  288. button = OBSMessageBox::question(this, WARNING_TEXT("Title"),
  289. WARNING_TEXT("Text"));
  290. #undef WARNING_TEXT
  291. if (button == QMessageBox::No)
  292. return false;
  293. }
  294. return true;
  295. }
  296. void AutoConfigStreamPage::on_show_clicked()
  297. {
  298. if (ui->key->echoMode() == QLineEdit::Password) {
  299. ui->key->setEchoMode(QLineEdit::Normal);
  300. ui->show->setText(QTStr("Hide"));
  301. } else {
  302. ui->key->setEchoMode(QLineEdit::Password);
  303. ui->show->setText(QTStr("Show"));
  304. }
  305. }
  306. void AutoConfigStreamPage::OnOAuthStreamKeyConnected()
  307. {
  308. #ifdef BROWSER_AVAILABLE
  309. OAuthStreamKey *a = reinterpret_cast<OAuthStreamKey *>(auth.get());
  310. if (a) {
  311. bool validKey = !a->key().empty();
  312. if (validKey)
  313. ui->key->setText(QT_UTF8(a->key().c_str()));
  314. ui->streamKeyWidget->setVisible(!validKey);
  315. ui->streamKeyLabel->setVisible(!validKey);
  316. ui->connectAccount2->setVisible(!validKey);
  317. ui->disconnectAccount->setVisible(validKey);
  318. }
  319. ui->stackedWidget->setCurrentIndex((int)Section::StreamKey);
  320. UpdateCompleted();
  321. #endif
  322. }
  323. void AutoConfigStreamPage::OnAuthConnected()
  324. {
  325. std::string service = QT_TO_UTF8(ui->service->currentText());
  326. Auth::Type type = Auth::AuthType(service);
  327. if (type == Auth::Type::OAuth_StreamKey) {
  328. OnOAuthStreamKeyConnected();
  329. }
  330. }
  331. void AutoConfigStreamPage::on_connectAccount_clicked()
  332. {
  333. #ifdef BROWSER_AVAILABLE
  334. std::string service = QT_TO_UTF8(ui->service->currentText());
  335. OAuth::DeleteCookies(service);
  336. auth = OAuthStreamKey::Login(this, service);
  337. if (!!auth)
  338. OnAuthConnected();
  339. #endif
  340. }
  341. #define DISCONNECT_COMFIRM_TITLE \
  342. "Basic.AutoConfig.StreamPage.DisconnectAccount.Confirm.Title"
  343. #define DISCONNECT_COMFIRM_TEXT \
  344. "Basic.AutoConfig.StreamPage.DisconnectAccount.Confirm.Text"
  345. void AutoConfigStreamPage::on_disconnectAccount_clicked()
  346. {
  347. QMessageBox::StandardButton button;
  348. button = OBSMessageBox::question(this, QTStr(DISCONNECT_COMFIRM_TITLE),
  349. QTStr(DISCONNECT_COMFIRM_TEXT));
  350. if (button == QMessageBox::No) {
  351. return;
  352. }
  353. OBSBasic *main = OBSBasic::Get();
  354. main->auth.reset();
  355. auth.reset();
  356. std::string service = QT_TO_UTF8(ui->service->currentText());
  357. #ifdef BROWSER_AVAILABLE
  358. OAuth::DeleteCookies(service);
  359. #endif
  360. ui->streamKeyWidget->setVisible(true);
  361. ui->streamKeyLabel->setVisible(true);
  362. ui->connectAccount2->setVisible(true);
  363. ui->disconnectAccount->setVisible(false);
  364. ui->key->setText("");
  365. }
  366. void AutoConfigStreamPage::on_useStreamKey_clicked()
  367. {
  368. ui->stackedWidget->setCurrentIndex((int)Section::StreamKey);
  369. UpdateCompleted();
  370. }
  371. static inline bool is_auth_service(const std::string &service)
  372. {
  373. return Auth::AuthType(service) != Auth::Type::None;
  374. }
  375. void AutoConfigStreamPage::ServiceChanged()
  376. {
  377. bool showMore = ui->service->currentData().toInt() ==
  378. (int)ListOpt::ShowAll;
  379. if (showMore)
  380. return;
  381. std::string service = QT_TO_UTF8(ui->service->currentText());
  382. bool regionBased = service == "Twitch" || service == "Smashcast";
  383. bool testBandwidth = ui->doBandwidthTest->isChecked();
  384. bool custom = IsCustom();
  385. ui->disconnectAccount->setVisible(false);
  386. #ifdef BROWSER_AVAILABLE
  387. if (cef) {
  388. if (lastService != service.c_str()) {
  389. bool can_auth = is_auth_service(service);
  390. int page = can_auth ? (int)Section::Connect
  391. : (int)Section::StreamKey;
  392. ui->stackedWidget->setCurrentIndex(page);
  393. ui->streamKeyWidget->setVisible(true);
  394. ui->streamKeyLabel->setVisible(true);
  395. ui->connectAccount2->setVisible(can_auth);
  396. auth.reset();
  397. if (lastService.isEmpty())
  398. lastService = service.c_str();
  399. }
  400. } else {
  401. ui->connectAccount2->setVisible(false);
  402. }
  403. #else
  404. ui->connectAccount2->setVisible(false);
  405. #endif
  406. /* Test three closest servers if "Auto" is available for Twitch */
  407. if (service == "Twitch" && wiz->twitchAuto)
  408. regionBased = false;
  409. ui->streamkeyPageLayout->removeWidget(ui->serverLabel);
  410. ui->streamkeyPageLayout->removeWidget(ui->serverStackedWidget);
  411. if (custom) {
  412. ui->streamkeyPageLayout->insertRow(1, ui->serverLabel,
  413. ui->serverStackedWidget);
  414. ui->region->setVisible(false);
  415. ui->serverStackedWidget->setCurrentIndex(1);
  416. ui->serverStackedWidget->setVisible(true);
  417. ui->serverLabel->setVisible(true);
  418. } else {
  419. if (!testBandwidth)
  420. ui->streamkeyPageLayout->insertRow(
  421. 2, ui->serverLabel, ui->serverStackedWidget);
  422. ui->region->setVisible(regionBased && testBandwidth);
  423. ui->serverStackedWidget->setCurrentIndex(0);
  424. ui->serverStackedWidget->setHidden(testBandwidth);
  425. ui->serverLabel->setHidden(testBandwidth);
  426. }
  427. wiz->testRegions = regionBased && testBandwidth;
  428. ui->bitrateLabel->setHidden(testBandwidth);
  429. ui->bitrate->setHidden(testBandwidth);
  430. #ifdef BROWSER_AVAILABLE
  431. OBSBasic *main = OBSBasic::Get();
  432. if (!!main->auth &&
  433. service.find(main->auth->service()) != std::string::npos) {
  434. auth = main->auth;
  435. OnAuthConnected();
  436. }
  437. #endif
  438. UpdateCompleted();
  439. }
  440. void AutoConfigStreamPage::UpdateKeyLink()
  441. {
  442. bool custom = IsCustom();
  443. QString serviceName = ui->service->currentText();
  444. bool isYoutube = false;
  445. if (custom)
  446. serviceName = "";
  447. QString text = QTStr("Basic.AutoConfig.StreamPage.StreamKey");
  448. if (serviceName == "Twitch") {
  449. text += " <a href=\"https://";
  450. text += "www.twitch.tv/broadcast/dashboard/streamkey";
  451. text += "\">";
  452. text += QTStr(
  453. "Basic.AutoConfig.StreamPage.StreamKey.LinkToSite");
  454. text += "</a>";
  455. } else if (serviceName == "YouTube / YouTube Gaming") {
  456. text += " <a href=\"https://";
  457. text += "www.youtube.com/live_dashboard";
  458. text += "\">";
  459. text += QTStr(
  460. "Basic.AutoConfig.StreamPage.StreamKey.LinkToSite");
  461. text += "</a>";
  462. isYoutube = true;
  463. } else if (serviceName.startsWith("Restream.io")) {
  464. text += " <a href=\"https://";
  465. text += "restream.io/settings/streaming-setup?from=OBS";
  466. text += "\">";
  467. text += QTStr(
  468. "Basic.AutoConfig.StreamPage.StreamKey.LinkToSite");
  469. text += "</a>";
  470. } else if (serviceName == "Facebook Live") {
  471. text += " <a href=\"https://";
  472. text += "www.facebook.com/live/create";
  473. text += "\">";
  474. text += QTStr(
  475. "Basic.AutoConfig.StreamPage.StreamKey.LinkToSite");
  476. text += "</a>";
  477. }
  478. if (isYoutube) {
  479. ui->doBandwidthTest->setChecked(false);
  480. ui->doBandwidthTest->setEnabled(false);
  481. } else {
  482. ui->doBandwidthTest->setEnabled(true);
  483. }
  484. ui->streamKeyLabel->setText(text);
  485. }
  486. void AutoConfigStreamPage::LoadServices(bool showAll)
  487. {
  488. obs_properties_t *props = obs_get_service_properties("rtmp_common");
  489. OBSData settings = obs_data_create();
  490. obs_data_release(settings);
  491. obs_data_set_bool(settings, "show_all", showAll);
  492. obs_property_t *prop = obs_properties_get(props, "show_all");
  493. obs_property_modified(prop, settings);
  494. ui->service->blockSignals(true);
  495. ui->service->clear();
  496. QStringList names;
  497. obs_property_t *services = obs_properties_get(props, "service");
  498. size_t services_count = obs_property_list_item_count(services);
  499. for (size_t i = 0; i < services_count; i++) {
  500. const char *name = obs_property_list_item_string(services, i);
  501. names.push_back(name);
  502. }
  503. if (showAll)
  504. names.sort();
  505. for (QString &name : names)
  506. ui->service->addItem(name);
  507. if (!showAll) {
  508. ui->service->addItem(
  509. QTStr("Basic.AutoConfig.StreamPage.Service.ShowAll"),
  510. QVariant((int)ListOpt::ShowAll));
  511. }
  512. ui->service->insertItem(
  513. 0, QTStr("Basic.AutoConfig.StreamPage.Service.Custom"),
  514. QVariant((int)ListOpt::Custom));
  515. if (!lastService.isEmpty()) {
  516. int idx = ui->service->findText(lastService);
  517. if (idx != -1)
  518. ui->service->setCurrentIndex(idx);
  519. }
  520. obs_properties_destroy(props);
  521. ui->service->blockSignals(false);
  522. }
  523. void AutoConfigStreamPage::UpdateServerList()
  524. {
  525. QString serviceName = ui->service->currentText();
  526. bool showMore = ui->service->currentData().toInt() ==
  527. (int)ListOpt::ShowAll;
  528. if (showMore) {
  529. LoadServices(true);
  530. ui->service->showPopup();
  531. return;
  532. } else {
  533. lastService = serviceName;
  534. }
  535. obs_properties_t *props = obs_get_service_properties("rtmp_common");
  536. obs_property_t *services = obs_properties_get(props, "service");
  537. OBSData settings = obs_data_create();
  538. obs_data_release(settings);
  539. obs_data_set_string(settings, "service", QT_TO_UTF8(serviceName));
  540. obs_property_modified(services, settings);
  541. obs_property_t *servers = obs_properties_get(props, "server");
  542. ui->server->clear();
  543. size_t servers_count = obs_property_list_item_count(servers);
  544. for (size_t i = 0; i < servers_count; i++) {
  545. const char *name = obs_property_list_item_name(servers, i);
  546. const char *server = obs_property_list_item_string(servers, i);
  547. ui->server->addItem(name, server);
  548. }
  549. obs_properties_destroy(props);
  550. }
  551. void AutoConfigStreamPage::UpdateCompleted()
  552. {
  553. if (ui->stackedWidget->currentIndex() == (int)Section::Connect ||
  554. (ui->key->text().isEmpty() && !auth)) {
  555. ready = false;
  556. } else {
  557. bool custom = IsCustom();
  558. if (custom) {
  559. ready = !ui->customServer->text().isEmpty();
  560. } else {
  561. ready = !wiz->testRegions ||
  562. ui->regionUS->isChecked() ||
  563. ui->regionEU->isChecked() ||
  564. ui->regionAsia->isChecked() ||
  565. ui->regionOther->isChecked();
  566. }
  567. }
  568. emit completeChanged();
  569. }
  570. /* ------------------------------------------------------------------------- */
  571. AutoConfig::AutoConfig(QWidget *parent) : QWizard(parent)
  572. {
  573. EnableThreadedMessageBoxes(true);
  574. calldata_t cd = {0};
  575. calldata_set_int(&cd, "seconds", 5);
  576. proc_handler_t *ph = obs_get_proc_handler();
  577. proc_handler_call(ph, "twitch_ingests_refresh", &cd);
  578. calldata_free(&cd);
  579. OBSBasic *main = reinterpret_cast<OBSBasic *>(parent);
  580. main->EnableOutputs(false);
  581. installEventFilter(CreateShortcutFilter());
  582. std::string serviceType;
  583. GetServiceInfo(serviceType, serviceName, server, key);
  584. #ifdef _WIN32
  585. setWizardStyle(QWizard::ModernStyle);
  586. #endif
  587. streamPage = new AutoConfigStreamPage();
  588. setPage(StartPage, new AutoConfigStartPage());
  589. setPage(VideoPage, new AutoConfigVideoPage());
  590. setPage(StreamPage, streamPage);
  591. setPage(TestPage, new AutoConfigTestPage());
  592. setWindowTitle(QTStr("Basic.AutoConfig"));
  593. setWindowFlags(windowFlags() & ~Qt::WindowContextHelpButtonHint);
  594. obs_video_info ovi;
  595. obs_get_video_info(&ovi);
  596. baseResolutionCX = ovi.base_width;
  597. baseResolutionCY = ovi.base_height;
  598. /* ----------------------------------------- */
  599. /* check to see if Twitch's "auto" available */
  600. OBSData twitchSettings = obs_data_create();
  601. obs_data_release(twitchSettings);
  602. obs_data_set_string(twitchSettings, "service", "Twitch");
  603. obs_properties_t *props = obs_get_service_properties("rtmp_common");
  604. obs_properties_apply_settings(props, twitchSettings);
  605. obs_property_t *p = obs_properties_get(props, "server");
  606. const char *first = obs_property_list_item_string(p, 0);
  607. twitchAuto = strcmp(first, "auto") == 0;
  608. obs_properties_destroy(props);
  609. /* ----------------------------------------- */
  610. /* load service/servers */
  611. customServer = serviceType == "rtmp_custom";
  612. QComboBox *serviceList = streamPage->ui->service;
  613. if (!serviceName.empty()) {
  614. serviceList->blockSignals(true);
  615. int count = serviceList->count();
  616. bool found = false;
  617. for (int i = 0; i < count; i++) {
  618. QString name = serviceList->itemText(i);
  619. if (name == serviceName.c_str()) {
  620. serviceList->setCurrentIndex(i);
  621. found = true;
  622. break;
  623. }
  624. }
  625. if (!found) {
  626. serviceList->insertItem(0, serviceName.c_str());
  627. serviceList->setCurrentIndex(0);
  628. }
  629. serviceList->blockSignals(false);
  630. }
  631. streamPage->UpdateServerList();
  632. streamPage->UpdateKeyLink();
  633. streamPage->lastService.clear();
  634. if (!customServer) {
  635. QComboBox *serverList = streamPage->ui->server;
  636. int idx = serverList->findData(QString(server.c_str()));
  637. if (idx == -1)
  638. idx = 0;
  639. serverList->setCurrentIndex(idx);
  640. } else {
  641. streamPage->ui->customServer->setText(server.c_str());
  642. int idx = streamPage->ui->service->findData(
  643. QVariant((int)ListOpt::Custom));
  644. streamPage->ui->service->setCurrentIndex(idx);
  645. }
  646. if (!key.empty())
  647. streamPage->ui->key->setText(key.c_str());
  648. int bitrate =
  649. config_get_int(main->Config(), "SimpleOutput", "VBitrate");
  650. streamPage->ui->bitrate->setValue(bitrate);
  651. streamPage->ServiceChanged();
  652. TestHardwareEncoding();
  653. if (!hardwareEncodingAvailable) {
  654. delete streamPage->ui->preferHardware;
  655. streamPage->ui->preferHardware = nullptr;
  656. } else {
  657. /* Newer generations of NVENC have a high enough quality to
  658. * bitrate ratio that if NVENC is available, it makes sense to
  659. * just always prefer hardware encoding by default */
  660. bool preferHardware = nvencAvailable ||
  661. os_get_physical_cores() <= 4;
  662. streamPage->ui->preferHardware->setChecked(preferHardware);
  663. }
  664. setOptions(0);
  665. setButtonText(QWizard::FinishButton,
  666. QTStr("Basic.AutoConfig.ApplySettings"));
  667. setButtonText(QWizard::BackButton, QTStr("Back"));
  668. setButtonText(QWizard::NextButton, QTStr("Next"));
  669. setButtonText(QWizard::CancelButton, QTStr("Cancel"));
  670. }
  671. AutoConfig::~AutoConfig()
  672. {
  673. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  674. main->EnableOutputs(true);
  675. EnableThreadedMessageBoxes(false);
  676. }
  677. void AutoConfig::TestHardwareEncoding()
  678. {
  679. size_t idx = 0;
  680. const char *id;
  681. while (obs_enum_encoder_types(idx++, &id)) {
  682. if (strcmp(id, "ffmpeg_nvenc") == 0)
  683. hardwareEncodingAvailable = nvencAvailable = true;
  684. else if (strcmp(id, "obs_qsv11") == 0)
  685. hardwareEncodingAvailable = qsvAvailable = true;
  686. else if (strcmp(id, "amd_amf_h264") == 0)
  687. hardwareEncodingAvailable = vceAvailable = true;
  688. }
  689. }
  690. bool AutoConfig::CanTestServer(const char *server)
  691. {
  692. if (!testRegions || (regionUS && regionEU && regionAsia && regionOther))
  693. return true;
  694. if (service == Service::Twitch) {
  695. if (astrcmp_n(server, "US West:", 8) == 0 ||
  696. astrcmp_n(server, "US East:", 8) == 0 ||
  697. astrcmp_n(server, "US Central:", 11) == 0) {
  698. return regionUS;
  699. } else if (astrcmp_n(server, "EU:", 3) == 0) {
  700. return regionEU;
  701. } else if (astrcmp_n(server, "Asia:", 5) == 0) {
  702. return regionAsia;
  703. } else if (regionOther) {
  704. return true;
  705. }
  706. } else if (service == Service::Smashcast) {
  707. if (strcmp(server, "Default") == 0) {
  708. return true;
  709. } else if (astrcmp_n(server, "US-West:", 8) == 0 ||
  710. astrcmp_n(server, "US-East:", 8) == 0) {
  711. return regionUS;
  712. } else if (astrcmp_n(server, "EU-", 3) == 0) {
  713. return regionEU;
  714. } else if (astrcmp_n(server, "South Korea:", 12) == 0 ||
  715. astrcmp_n(server, "Asia:", 5) == 0 ||
  716. astrcmp_n(server, "China:", 6) == 0) {
  717. return regionAsia;
  718. } else if (regionOther) {
  719. return true;
  720. }
  721. } else {
  722. return true;
  723. }
  724. return false;
  725. }
  726. void AutoConfig::done(int result)
  727. {
  728. QWizard::done(result);
  729. if (result == QDialog::Accepted) {
  730. if (type == Type::Streaming)
  731. SaveStreamSettings();
  732. SaveSettings();
  733. }
  734. }
  735. inline const char *AutoConfig::GetEncoderId(Encoder enc)
  736. {
  737. switch (enc) {
  738. case Encoder::NVENC:
  739. return SIMPLE_ENCODER_NVENC;
  740. case Encoder::QSV:
  741. return SIMPLE_ENCODER_QSV;
  742. case Encoder::AMD:
  743. return SIMPLE_ENCODER_AMD;
  744. default:
  745. return SIMPLE_ENCODER_X264;
  746. }
  747. };
  748. void AutoConfig::SaveStreamSettings()
  749. {
  750. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  751. /* ---------------------------------- */
  752. /* save service */
  753. const char *service_id = customServer ? "rtmp_custom" : "rtmp_common";
  754. obs_service_t *oldService = main->GetService();
  755. OBSData hotkeyData = obs_hotkeys_save_service(oldService);
  756. obs_data_release(hotkeyData);
  757. OBSData settings = obs_data_create();
  758. obs_data_release(settings);
  759. if (!customServer)
  760. obs_data_set_string(settings, "service", serviceName.c_str());
  761. obs_data_set_string(settings, "server", server.c_str());
  762. obs_data_set_string(settings, "key", key.c_str());
  763. OBSService newService = obs_service_create(
  764. service_id, "default_service", settings, hotkeyData);
  765. obs_service_release(newService);
  766. if (!newService)
  767. return;
  768. main->SetService(newService);
  769. main->SaveService();
  770. main->auth = streamPage->auth;
  771. if (!!main->auth)
  772. main->auth->LoadUI();
  773. /* ---------------------------------- */
  774. /* save stream settings */
  775. config_set_int(main->Config(), "SimpleOutput", "VBitrate",
  776. idealBitrate);
  777. config_set_string(main->Config(), "SimpleOutput", "StreamEncoder",
  778. GetEncoderId(streamingEncoder));
  779. config_remove_value(main->Config(), "SimpleOutput", "UseAdvanced");
  780. }
  781. void AutoConfig::SaveSettings()
  782. {
  783. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  784. if (recordingEncoder != Encoder::Stream)
  785. config_set_string(main->Config(), "SimpleOutput", "RecEncoder",
  786. GetEncoderId(recordingEncoder));
  787. const char *quality = recordingQuality == Quality::High ? "Small"
  788. : "Stream";
  789. config_set_string(main->Config(), "Output", "Mode", "Simple");
  790. config_set_string(main->Config(), "SimpleOutput", "RecQuality",
  791. quality);
  792. config_set_int(main->Config(), "Video", "BaseCX", baseResolutionCX);
  793. config_set_int(main->Config(), "Video", "BaseCY", baseResolutionCY);
  794. config_set_int(main->Config(), "Video", "OutputCX", idealResolutionCX);
  795. config_set_int(main->Config(), "Video", "OutputCY", idealResolutionCY);
  796. if (fpsType != FPSType::UseCurrent) {
  797. config_set_uint(main->Config(), "Video", "FPSType", 0);
  798. config_set_string(main->Config(), "Video", "FPSCommon",
  799. std::to_string(idealFPSNum).c_str());
  800. }
  801. main->ResetVideo();
  802. main->ResetOutputs();
  803. config_save_safe(main->Config(), "tmp", nullptr);
  804. }