window-basic-auto-config.cpp 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002
  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. auth = OAuthStreamKey::Login(this, service);
  336. if (!!auth)
  337. OnAuthConnected();
  338. #endif
  339. }
  340. #define DISCONNECT_COMFIRM_TITLE \
  341. "Basic.AutoConfig.StreamPage.DisconnectAccount.Confirm.Title"
  342. #define DISCONNECT_COMFIRM_TEXT \
  343. "Basic.AutoConfig.StreamPage.DisconnectAccount.Confirm.Text"
  344. void AutoConfigStreamPage::on_disconnectAccount_clicked()
  345. {
  346. QMessageBox::StandardButton button;
  347. button = OBSMessageBox::question(this, QTStr(DISCONNECT_COMFIRM_TITLE),
  348. QTStr(DISCONNECT_COMFIRM_TEXT));
  349. if (button == QMessageBox::No) {
  350. return;
  351. }
  352. OBSBasic *main = OBSBasic::Get();
  353. main->auth.reset();
  354. auth.reset();
  355. std::string service = QT_TO_UTF8(ui->service->currentText());
  356. #ifdef BROWSER_AVAILABLE
  357. OAuth::DeleteCookies(service);
  358. #endif
  359. ui->streamKeyWidget->setVisible(true);
  360. ui->streamKeyLabel->setVisible(true);
  361. ui->connectAccount2->setVisible(true);
  362. ui->disconnectAccount->setVisible(false);
  363. ui->key->setText("");
  364. }
  365. void AutoConfigStreamPage::on_useStreamKey_clicked()
  366. {
  367. ui->stackedWidget->setCurrentIndex((int)Section::StreamKey);
  368. UpdateCompleted();
  369. }
  370. static inline bool is_auth_service(const std::string &service)
  371. {
  372. return Auth::AuthType(service) != Auth::Type::None;
  373. }
  374. void AutoConfigStreamPage::ServiceChanged()
  375. {
  376. bool showMore = ui->service->currentData().toInt() ==
  377. (int)ListOpt::ShowAll;
  378. if (showMore)
  379. return;
  380. std::string service = QT_TO_UTF8(ui->service->currentText());
  381. bool regionBased = service == "Twitch" || service == "Smashcast";
  382. bool testBandwidth = ui->doBandwidthTest->isChecked();
  383. bool custom = IsCustom();
  384. ui->disconnectAccount->setVisible(false);
  385. #ifdef BROWSER_AVAILABLE
  386. if (cef) {
  387. if (lastService != service.c_str()) {
  388. bool can_auth = is_auth_service(service);
  389. int page = can_auth ? (int)Section::Connect
  390. : (int)Section::StreamKey;
  391. ui->stackedWidget->setCurrentIndex(page);
  392. ui->streamKeyWidget->setVisible(true);
  393. ui->streamKeyLabel->setVisible(true);
  394. ui->connectAccount2->setVisible(can_auth);
  395. auth.reset();
  396. if (lastService.isEmpty())
  397. lastService = service.c_str();
  398. }
  399. } else {
  400. ui->connectAccount2->setVisible(false);
  401. }
  402. #else
  403. ui->connectAccount2->setVisible(false);
  404. #endif
  405. /* Test three closest servers if "Auto" is available for Twitch */
  406. if (service == "Twitch" && wiz->twitchAuto)
  407. regionBased = false;
  408. ui->streamkeyPageLayout->removeWidget(ui->serverLabel);
  409. ui->streamkeyPageLayout->removeWidget(ui->serverStackedWidget);
  410. if (custom) {
  411. ui->streamkeyPageLayout->insertRow(1, ui->serverLabel,
  412. ui->serverStackedWidget);
  413. ui->region->setVisible(false);
  414. ui->serverStackedWidget->setCurrentIndex(1);
  415. ui->serverStackedWidget->setVisible(true);
  416. ui->serverLabel->setVisible(true);
  417. } else {
  418. if (!testBandwidth)
  419. ui->streamkeyPageLayout->insertRow(
  420. 2, ui->serverLabel, ui->serverStackedWidget);
  421. ui->region->setVisible(regionBased && testBandwidth);
  422. ui->serverStackedWidget->setCurrentIndex(0);
  423. ui->serverStackedWidget->setHidden(testBandwidth);
  424. ui->serverLabel->setHidden(testBandwidth);
  425. }
  426. wiz->testRegions = regionBased && testBandwidth;
  427. ui->bitrateLabel->setHidden(testBandwidth);
  428. ui->bitrate->setHidden(testBandwidth);
  429. #ifdef BROWSER_AVAILABLE
  430. OBSBasic *main = OBSBasic::Get();
  431. if (!!main->auth &&
  432. service.find(main->auth->service()) != std::string::npos) {
  433. auth = main->auth;
  434. OnAuthConnected();
  435. }
  436. #endif
  437. UpdateCompleted();
  438. }
  439. void AutoConfigStreamPage::UpdateKeyLink()
  440. {
  441. bool custom = IsCustom();
  442. QString serviceName = ui->service->currentText();
  443. bool isYoutube = false;
  444. if (custom)
  445. serviceName = "";
  446. QString text = QTStr("Basic.AutoConfig.StreamPage.StreamKey");
  447. if (serviceName == "Twitch") {
  448. text += " <a href=\"https://";
  449. text += "www.twitch.tv/broadcast/dashboard/streamkey";
  450. text += "\">";
  451. text += QTStr(
  452. "Basic.AutoConfig.StreamPage.StreamKey.LinkToSite");
  453. text += "</a>";
  454. } else if (serviceName == "YouTube / YouTube Gaming") {
  455. text += " <a href=\"https://";
  456. text += "www.youtube.com/live_dashboard";
  457. text += "\">";
  458. text += QTStr(
  459. "Basic.AutoConfig.StreamPage.StreamKey.LinkToSite");
  460. text += "</a>";
  461. isYoutube = true;
  462. } else if (serviceName.startsWith("Restream.io")) {
  463. text += " <a href=\"https://";
  464. text += "restream.io/settings/streaming-setup?from=OBS";
  465. text += "\">";
  466. text += QTStr(
  467. "Basic.AutoConfig.StreamPage.StreamKey.LinkToSite");
  468. text += "</a>";
  469. } else if (serviceName == "Facebook Live") {
  470. text += " <a href=\"https://";
  471. text += "www.facebook.com/live/create";
  472. text += "\">";
  473. text += QTStr(
  474. "Basic.AutoConfig.StreamPage.StreamKey.LinkToSite");
  475. text += "</a>";
  476. }
  477. if (isYoutube) {
  478. ui->doBandwidthTest->setChecked(false);
  479. ui->doBandwidthTest->setEnabled(false);
  480. } else {
  481. ui->doBandwidthTest->setEnabled(true);
  482. }
  483. ui->streamKeyLabel->setText(text);
  484. }
  485. void AutoConfigStreamPage::LoadServices(bool showAll)
  486. {
  487. obs_properties_t *props = obs_get_service_properties("rtmp_common");
  488. OBSData settings = obs_data_create();
  489. obs_data_release(settings);
  490. obs_data_set_bool(settings, "show_all", showAll);
  491. obs_property_t *prop = obs_properties_get(props, "show_all");
  492. obs_property_modified(prop, settings);
  493. ui->service->blockSignals(true);
  494. ui->service->clear();
  495. QStringList names;
  496. obs_property_t *services = obs_properties_get(props, "service");
  497. size_t services_count = obs_property_list_item_count(services);
  498. for (size_t i = 0; i < services_count; i++) {
  499. const char *name = obs_property_list_item_string(services, i);
  500. names.push_back(name);
  501. }
  502. if (showAll)
  503. names.sort();
  504. for (QString &name : names)
  505. ui->service->addItem(name);
  506. if (!showAll) {
  507. ui->service->addItem(
  508. QTStr("Basic.AutoConfig.StreamPage.Service.ShowAll"),
  509. QVariant((int)ListOpt::ShowAll));
  510. }
  511. ui->service->insertItem(
  512. 0, QTStr("Basic.AutoConfig.StreamPage.Service.Custom"),
  513. QVariant((int)ListOpt::Custom));
  514. if (!lastService.isEmpty()) {
  515. int idx = ui->service->findText(lastService);
  516. if (idx != -1)
  517. ui->service->setCurrentIndex(idx);
  518. }
  519. obs_properties_destroy(props);
  520. ui->service->blockSignals(false);
  521. }
  522. void AutoConfigStreamPage::UpdateServerList()
  523. {
  524. QString serviceName = ui->service->currentText();
  525. bool showMore = ui->service->currentData().toInt() ==
  526. (int)ListOpt::ShowAll;
  527. if (showMore) {
  528. LoadServices(true);
  529. ui->service->showPopup();
  530. return;
  531. } else {
  532. lastService = serviceName;
  533. }
  534. obs_properties_t *props = obs_get_service_properties("rtmp_common");
  535. obs_property_t *services = obs_properties_get(props, "service");
  536. OBSData settings = obs_data_create();
  537. obs_data_release(settings);
  538. obs_data_set_string(settings, "service", QT_TO_UTF8(serviceName));
  539. obs_property_modified(services, settings);
  540. obs_property_t *servers = obs_properties_get(props, "server");
  541. ui->server->clear();
  542. size_t servers_count = obs_property_list_item_count(servers);
  543. for (size_t i = 0; i < servers_count; i++) {
  544. const char *name = obs_property_list_item_name(servers, i);
  545. const char *server = obs_property_list_item_string(servers, i);
  546. ui->server->addItem(name, server);
  547. }
  548. obs_properties_destroy(props);
  549. }
  550. void AutoConfigStreamPage::UpdateCompleted()
  551. {
  552. if (ui->stackedWidget->currentIndex() == (int)Section::Connect ||
  553. (ui->key->text().isEmpty() && !auth)) {
  554. ready = false;
  555. } else {
  556. bool custom = IsCustom();
  557. if (custom) {
  558. ready = !ui->customServer->text().isEmpty();
  559. } else {
  560. ready = !wiz->testRegions ||
  561. ui->regionUS->isChecked() ||
  562. ui->regionEU->isChecked() ||
  563. ui->regionAsia->isChecked() ||
  564. ui->regionOther->isChecked();
  565. }
  566. }
  567. emit completeChanged();
  568. }
  569. /* ------------------------------------------------------------------------- */
  570. AutoConfig::AutoConfig(QWidget *parent) : QWizard(parent)
  571. {
  572. EnableThreadedMessageBoxes(true);
  573. calldata_t cd = {0};
  574. calldata_set_int(&cd, "seconds", 5);
  575. proc_handler_t *ph = obs_get_proc_handler();
  576. proc_handler_call(ph, "twitch_ingests_refresh", &cd);
  577. calldata_free(&cd);
  578. OBSBasic *main = reinterpret_cast<OBSBasic *>(parent);
  579. main->EnableOutputs(false);
  580. installEventFilter(CreateShortcutFilter());
  581. std::string serviceType;
  582. GetServiceInfo(serviceType, serviceName, server, key);
  583. #ifdef _WIN32
  584. setWizardStyle(QWizard::ModernStyle);
  585. #endif
  586. streamPage = new AutoConfigStreamPage();
  587. setPage(StartPage, new AutoConfigStartPage());
  588. setPage(VideoPage, new AutoConfigVideoPage());
  589. setPage(StreamPage, streamPage);
  590. setPage(TestPage, new AutoConfigTestPage());
  591. setWindowTitle(QTStr("Basic.AutoConfig"));
  592. setWindowFlags(windowFlags() & ~Qt::WindowContextHelpButtonHint);
  593. obs_video_info ovi;
  594. obs_get_video_info(&ovi);
  595. baseResolutionCX = ovi.base_width;
  596. baseResolutionCY = ovi.base_height;
  597. /* ----------------------------------------- */
  598. /* check to see if Twitch's "auto" available */
  599. OBSData twitchSettings = obs_data_create();
  600. obs_data_release(twitchSettings);
  601. obs_data_set_string(twitchSettings, "service", "Twitch");
  602. obs_properties_t *props = obs_get_service_properties("rtmp_common");
  603. obs_properties_apply_settings(props, twitchSettings);
  604. obs_property_t *p = obs_properties_get(props, "server");
  605. const char *first = obs_property_list_item_string(p, 0);
  606. twitchAuto = strcmp(first, "auto") == 0;
  607. obs_properties_destroy(props);
  608. /* ----------------------------------------- */
  609. /* load service/servers */
  610. customServer = serviceType == "rtmp_custom";
  611. QComboBox *serviceList = streamPage->ui->service;
  612. if (!serviceName.empty()) {
  613. serviceList->blockSignals(true);
  614. int count = serviceList->count();
  615. bool found = false;
  616. for (int i = 0; i < count; i++) {
  617. QString name = serviceList->itemText(i);
  618. if (name == serviceName.c_str()) {
  619. serviceList->setCurrentIndex(i);
  620. found = true;
  621. break;
  622. }
  623. }
  624. if (!found) {
  625. serviceList->insertItem(0, serviceName.c_str());
  626. serviceList->setCurrentIndex(0);
  627. }
  628. serviceList->blockSignals(false);
  629. }
  630. streamPage->UpdateServerList();
  631. streamPage->UpdateKeyLink();
  632. streamPage->lastService.clear();
  633. if (!customServer) {
  634. QComboBox *serverList = streamPage->ui->server;
  635. int idx = serverList->findData(QString(server.c_str()));
  636. if (idx == -1)
  637. idx = 0;
  638. serverList->setCurrentIndex(idx);
  639. } else {
  640. streamPage->ui->customServer->setText(server.c_str());
  641. int idx = streamPage->ui->service->findData(
  642. QVariant((int)ListOpt::Custom));
  643. streamPage->ui->service->setCurrentIndex(idx);
  644. }
  645. if (!key.empty())
  646. streamPage->ui->key->setText(key.c_str());
  647. int bitrate =
  648. config_get_int(main->Config(), "SimpleOutput", "VBitrate");
  649. streamPage->ui->bitrate->setValue(bitrate);
  650. streamPage->ServiceChanged();
  651. TestHardwareEncoding();
  652. if (!hardwareEncodingAvailable) {
  653. delete streamPage->ui->preferHardware;
  654. streamPage->ui->preferHardware = nullptr;
  655. } else {
  656. /* Newer generations of NVENC have a high enough quality to
  657. * bitrate ratio that if NVENC is available, it makes sense to
  658. * just always prefer hardware encoding by default */
  659. bool preferHardware = nvencAvailable ||
  660. os_get_physical_cores() <= 4;
  661. streamPage->ui->preferHardware->setChecked(preferHardware);
  662. }
  663. setOptions(0);
  664. setButtonText(QWizard::FinishButton,
  665. QTStr("Basic.AutoConfig.ApplySettings"));
  666. setButtonText(QWizard::BackButton, QTStr("Back"));
  667. setButtonText(QWizard::NextButton, QTStr("Next"));
  668. setButtonText(QWizard::CancelButton, QTStr("Cancel"));
  669. }
  670. AutoConfig::~AutoConfig()
  671. {
  672. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  673. main->EnableOutputs(true);
  674. EnableThreadedMessageBoxes(false);
  675. }
  676. void AutoConfig::TestHardwareEncoding()
  677. {
  678. size_t idx = 0;
  679. const char *id;
  680. while (obs_enum_encoder_types(idx++, &id)) {
  681. if (strcmp(id, "ffmpeg_nvenc") == 0)
  682. hardwareEncodingAvailable = nvencAvailable = true;
  683. else if (strcmp(id, "obs_qsv11") == 0)
  684. hardwareEncodingAvailable = qsvAvailable = true;
  685. else if (strcmp(id, "amd_amf_h264") == 0)
  686. hardwareEncodingAvailable = vceAvailable = true;
  687. }
  688. }
  689. bool AutoConfig::CanTestServer(const char *server)
  690. {
  691. if (!testRegions || (regionUS && regionEU && regionAsia && regionOther))
  692. return true;
  693. if (service == Service::Twitch) {
  694. if (astrcmp_n(server, "US West:", 8) == 0 ||
  695. astrcmp_n(server, "US East:", 8) == 0 ||
  696. astrcmp_n(server, "US Central:", 11) == 0) {
  697. return regionUS;
  698. } else if (astrcmp_n(server, "EU:", 3) == 0) {
  699. return regionEU;
  700. } else if (astrcmp_n(server, "Asia:", 5) == 0) {
  701. return regionAsia;
  702. } else if (regionOther) {
  703. return true;
  704. }
  705. } else if (service == Service::Smashcast) {
  706. if (strcmp(server, "Default") == 0) {
  707. return true;
  708. } else if (astrcmp_n(server, "US-West:", 8) == 0 ||
  709. astrcmp_n(server, "US-East:", 8) == 0) {
  710. return regionUS;
  711. } else if (astrcmp_n(server, "EU-", 3) == 0) {
  712. return regionEU;
  713. } else if (astrcmp_n(server, "South Korea:", 12) == 0 ||
  714. astrcmp_n(server, "Asia:", 5) == 0 ||
  715. astrcmp_n(server, "China:", 6) == 0) {
  716. return regionAsia;
  717. } else if (regionOther) {
  718. return true;
  719. }
  720. } else {
  721. return true;
  722. }
  723. return false;
  724. }
  725. void AutoConfig::done(int result)
  726. {
  727. QWizard::done(result);
  728. if (result == QDialog::Accepted) {
  729. if (type == Type::Streaming)
  730. SaveStreamSettings();
  731. SaveSettings();
  732. }
  733. }
  734. inline const char *AutoConfig::GetEncoderId(Encoder enc)
  735. {
  736. switch (enc) {
  737. case Encoder::NVENC:
  738. return SIMPLE_ENCODER_NVENC;
  739. case Encoder::QSV:
  740. return SIMPLE_ENCODER_QSV;
  741. case Encoder::AMD:
  742. return SIMPLE_ENCODER_AMD;
  743. default:
  744. return SIMPLE_ENCODER_X264;
  745. }
  746. };
  747. void AutoConfig::SaveStreamSettings()
  748. {
  749. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  750. /* ---------------------------------- */
  751. /* save service */
  752. const char *service_id = customServer ? "rtmp_custom" : "rtmp_common";
  753. obs_service_t *oldService = main->GetService();
  754. OBSData hotkeyData = obs_hotkeys_save_service(oldService);
  755. obs_data_release(hotkeyData);
  756. OBSData settings = obs_data_create();
  757. obs_data_release(settings);
  758. if (!customServer)
  759. obs_data_set_string(settings, "service", serviceName.c_str());
  760. obs_data_set_string(settings, "server", server.c_str());
  761. obs_data_set_string(settings, "key", key.c_str());
  762. OBSService newService = obs_service_create(
  763. service_id, "default_service", settings, hotkeyData);
  764. obs_service_release(newService);
  765. if (!newService)
  766. return;
  767. main->SetService(newService);
  768. main->SaveService();
  769. main->auth = streamPage->auth;
  770. if (!!main->auth)
  771. main->auth->LoadUI();
  772. /* ---------------------------------- */
  773. /* save stream settings */
  774. config_set_int(main->Config(), "SimpleOutput", "VBitrate",
  775. idealBitrate);
  776. config_set_string(main->Config(), "SimpleOutput", "StreamEncoder",
  777. GetEncoderId(streamingEncoder));
  778. config_remove_value(main->Config(), "SimpleOutput", "UseAdvanced");
  779. }
  780. void AutoConfig::SaveSettings()
  781. {
  782. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  783. if (recordingEncoder != Encoder::Stream)
  784. config_set_string(main->Config(), "SimpleOutput", "RecEncoder",
  785. GetEncoderId(recordingEncoder));
  786. const char *quality = recordingQuality == Quality::High ? "Small"
  787. : "Stream";
  788. config_set_string(main->Config(), "Output", "Mode", "Simple");
  789. config_set_string(main->Config(), "SimpleOutput", "RecQuality",
  790. quality);
  791. config_set_int(main->Config(), "Video", "BaseCX", baseResolutionCX);
  792. config_set_int(main->Config(), "Video", "BaseCY", baseResolutionCY);
  793. config_set_int(main->Config(), "Video", "OutputCX", idealResolutionCX);
  794. config_set_int(main->Config(), "Video", "OutputCY", idealResolutionCY);
  795. if (fpsType != FPSType::UseCurrent) {
  796. config_set_uint(main->Config(), "Video", "FPSType", 0);
  797. config_set_string(main->Config(), "Video", "FPSCommon",
  798. std::to_string(idealFPSNum).c_str());
  799. }
  800. main->ResetVideo();
  801. main->ResetOutputs();
  802. config_save_safe(main->Config(), "tmp", nullptr);
  803. }