window-basic-auto-config.cpp 27 KB

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