window-basic-auto-config.cpp 22 KB

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