SelectionTab.cpp 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948
  1. /*
  2. * SelectionTab.cpp, part of VCMI engine
  3. *
  4. * Authors: listed in file AUTHORS in main folder
  5. *
  6. * License: GNU General Public License v2.0 or later
  7. * Full text of license available in license.txt file, in main folder
  8. *
  9. */
  10. #include "StdInc.h"
  11. #include "SelectionTab.h"
  12. #include "CSelectionBase.h"
  13. #include "CLobbyScreen.h"
  14. #include "../CGameInfo.h"
  15. #include "../CPlayerInterface.h"
  16. #include "../CServerHandler.h"
  17. #include "../gui/CGuiHandler.h"
  18. #include "../gui/Shortcut.h"
  19. #include "../gui/WindowHandler.h"
  20. #include "../widgets/CComponent.h"
  21. #include "../widgets/Buttons.h"
  22. #include "../widgets/CTextInput.h"
  23. #include "../widgets/MiscWidgets.h"
  24. #include "../widgets/ObjectLists.h"
  25. #include "../widgets/Slider.h"
  26. #include "../widgets/TextControls.h"
  27. #include "../windows/GUIClasses.h"
  28. #include "../windows/InfoWindows.h"
  29. #include "../windows/CMapOverview.h"
  30. #include "../render/CAnimation.h"
  31. #include "../render/IImage.h"
  32. #include "../render/IRenderHandler.h"
  33. #include "../../CCallback.h"
  34. #include "../../lib/CGeneralTextHandler.h"
  35. #include "../../lib/CConfigHandler.h"
  36. #include "../../lib/GameSettings.h"
  37. #include "../../lib/filesystem/Filesystem.h"
  38. #include "../../lib/campaign/CampaignState.h"
  39. #include "../../lib/mapping/CMapInfo.h"
  40. #include "../../lib/mapping/CMapHeader.h"
  41. #include "../../lib/mapping/MapFormat.h"
  42. #include "../../lib/TerrainHandler.h"
  43. bool mapSorter::operator()(const std::shared_ptr<ElementInfo> aaa, const std::shared_ptr<ElementInfo> bbb)
  44. {
  45. if(aaa->isFolder || bbb->isFolder)
  46. {
  47. if(aaa->isFolder != bbb->isFolder)
  48. return (aaa->isFolder > bbb->isFolder);
  49. else
  50. {
  51. if(boost::algorithm::starts_with(aaa->folderName, "..") || boost::algorithm::starts_with(bbb->folderName, ".."))
  52. return boost::algorithm::starts_with(aaa->folderName, "..");
  53. return boost::ilexicographical_compare(aaa->folderName, bbb->folderName);
  54. }
  55. }
  56. auto a = aaa->mapHeader.get();
  57. auto b = bbb->mapHeader.get();
  58. if(a && b) //if we are sorting scenarios
  59. {
  60. switch(sortBy)
  61. {
  62. case _format: //by map format (RoE, WoG, etc)
  63. return (a->version < b->version);
  64. break;
  65. case _loscon: //by loss conditions
  66. return (a->defeatIconIndex < b->defeatIconIndex);
  67. break;
  68. case _playerAm: //by player amount
  69. int playerAmntB;
  70. int humenPlayersB;
  71. int playerAmntA;
  72. int humenPlayersA;
  73. playerAmntB = humenPlayersB = playerAmntA = humenPlayersA = 0;
  74. for(int i = 0; i < 8; i++)
  75. {
  76. if(a->players[i].canHumanPlay)
  77. {
  78. playerAmntA++;
  79. humenPlayersA++;
  80. }
  81. else if(a->players[i].canComputerPlay)
  82. {
  83. playerAmntA++;
  84. }
  85. if(b->players[i].canHumanPlay)
  86. {
  87. playerAmntB++;
  88. humenPlayersB++;
  89. }
  90. else if(b->players[i].canComputerPlay)
  91. {
  92. playerAmntB++;
  93. }
  94. }
  95. if(playerAmntB != playerAmntA)
  96. return (playerAmntA < playerAmntB);
  97. else
  98. return (humenPlayersA < humenPlayersB);
  99. break;
  100. case _size: //by size of map
  101. return (a->width < b->width);
  102. break;
  103. case _viccon: //by victory conditions
  104. return (a->victoryIconIndex < b->victoryIconIndex);
  105. break;
  106. case _name: //by name
  107. return boost::ilexicographical_compare(a->name.toString(), b->name.toString());
  108. case _fileName: //by filename
  109. return boost::ilexicographical_compare(aaa->fileURI, bbb->fileURI);
  110. case _changeDate: //by changedate
  111. return aaa->lastWrite < bbb->lastWrite;
  112. default:
  113. return boost::ilexicographical_compare(a->name.toString(), b->name.toString());
  114. }
  115. }
  116. else //if we are sorting campaigns
  117. {
  118. switch(sortBy)
  119. {
  120. case _numOfMaps: //by number of maps in campaign
  121. return aaa->campaign->scenariosCount() < bbb->campaign->scenariosCount();
  122. case _name: //by name
  123. return boost::ilexicographical_compare(aaa->campaign->getNameTranslated(), bbb->campaign->getNameTranslated());
  124. default:
  125. return boost::ilexicographical_compare(aaa->campaign->getNameTranslated(), bbb->campaign->getNameTranslated());
  126. }
  127. }
  128. }
  129. // pick sorting order based on selection
  130. static ESortBy getSortBySelectionScreen(ESelectionScreen Type)
  131. {
  132. switch(Type)
  133. {
  134. case ESelectionScreen::newGame:
  135. return ESortBy::_name;
  136. case ESelectionScreen::loadGame:
  137. case ESelectionScreen::saveGame:
  138. return ESortBy::_fileName;
  139. case ESelectionScreen::campaignList:
  140. return ESortBy::_name;
  141. }
  142. // Should not reach here. But let's not crash the game.
  143. return ESortBy::_name;
  144. }
  145. SelectionTab::SelectionTab(ESelectionScreen Type)
  146. : CIntObject(LCLICK | SHOW_POPUP | KEYBOARD | DOUBLECLICK), callOnSelect(nullptr), tabType(Type), selectionPos(0), sortModeAscending(true), inputNameRect{32, 539, 350, 20}, curFolder(""), currentMapSizeFilter(0), showRandom(false)
  147. {
  148. OBJ_CONSTRUCTION;
  149. generalSortingBy = getSortBySelectionScreen(tabType);
  150. sortingBy = _format;
  151. bool enableUiEnhancements = settings["general"]["enableUiEnhancements"].Bool();
  152. if(tabType != ESelectionScreen::campaignList)
  153. {
  154. background = std::make_shared<CPicture>(ImagePath::builtin("SCSELBCK.bmp"), 0, 6);
  155. pos = background->pos;
  156. inputName = std::make_shared<CTextInput>(inputNameRect, Point(-32, -25), ImagePath::builtin("GSSTRIP.bmp"));
  157. inputName->setFilterFilename();
  158. labelMapSizes = std::make_shared<CLabel>(87, 62, FONT_SMALL, ETextAlignment::CENTER, Colors::YELLOW, CGI->generaltexth->allTexts[510]);
  159. // TODO: Global constants?
  160. int sizes[] = {CMapHeader::MAP_SIZE_SMALL,
  161. CMapHeader::MAP_SIZE_MIDDLE,
  162. CMapHeader::MAP_SIZE_LARGE,
  163. CMapHeader::MAP_SIZE_XLARGE,
  164. 0};
  165. const char * filterIconNmes[] = {"SCSMBUT.DEF", "SCMDBUT.DEF", "SCLGBUT.DEF", "SCXLBUT.DEF", "SCALBUT.DEF"};
  166. for(int i = 0; i < 5; i++)
  167. buttonsSortBy.push_back(std::make_shared<CButton>(Point(158 + 47 * i, 46), AnimationPath::builtin(filterIconNmes[i]), CGI->generaltexth->zelp[54 + i], std::bind(&SelectionTab::filter, this, sizes[i], true)));
  168. int xpos[] = {23, 55, 88, 121, 306, 339};
  169. const char * sortIconNames[] = {"SCBUTT1.DEF", "SCBUTT2.DEF", "SCBUTCP.DEF", "SCBUTT3.DEF", "SCBUTT4.DEF", "SCBUTT5.DEF"};
  170. for(int i = 0; i < 6; i++)
  171. {
  172. ESortBy criteria = (ESortBy)i;
  173. if(criteria == _name)
  174. criteria = generalSortingBy;
  175. buttonsSortBy.push_back(std::make_shared<CButton>(Point(xpos[i], 86), AnimationPath::builtin(sortIconNames[i]), CGI->generaltexth->zelp[107 + i], std::bind(&SelectionTab::sortBy, this, criteria)));
  176. }
  177. }
  178. int positionsToShow = 18;
  179. std::string tabTitle;
  180. switch(tabType)
  181. {
  182. case ESelectionScreen::newGame:
  183. tabTitle = CGI->generaltexth->arraytxt[229];
  184. break;
  185. case ESelectionScreen::loadGame:
  186. tabTitle = CGI->generaltexth->arraytxt[230];
  187. break;
  188. case ESelectionScreen::saveGame:
  189. positionsToShow = 16;
  190. tabTitle = CGI->generaltexth->arraytxt[231];
  191. break;
  192. case ESelectionScreen::campaignList:
  193. tabTitle = CGI->generaltexth->allTexts[726];
  194. setRedrawParent(true); // we use parent background so we need to make sure it's will be redrawn too
  195. pos.w = parent->pos.w;
  196. pos.h = parent->pos.h;
  197. pos.x += 3;
  198. pos.y += 6;
  199. buttonsSortBy.push_back(std::make_shared<CButton>(Point(23, 86), AnimationPath::builtin("CamCusM.DEF"), CButton::tooltip(), std::bind(&SelectionTab::sortBy, this, _numOfMaps)));
  200. buttonsSortBy.push_back(std::make_shared<CButton>(Point(55, 86), AnimationPath::builtin("CamCusL.DEF"), CButton::tooltip(), std::bind(&SelectionTab::sortBy, this, _name)));
  201. break;
  202. default:
  203. assert(0);
  204. break;
  205. }
  206. if(enableUiEnhancements)
  207. {
  208. auto sortByDate = std::make_shared<CButton>(Point(371, 85), AnimationPath::builtin("selectionTabSortDate"), CButton::tooltip("", CGI->generaltexth->translate("vcmi.lobby.sortDate")), std::bind(&SelectionTab::sortBy, this, ESortBy::_changeDate));
  209. sortByDate->setOverlay(std::make_shared<CPicture>(ImagePath::builtin("lobby/selectionTabSortDate")));
  210. buttonsSortBy.push_back(sortByDate);
  211. }
  212. iconsMapFormats = GH.renderHandler().loadAnimation(AnimationPath::builtin("SCSELC.DEF"));
  213. iconsVictoryCondition = GH.renderHandler().loadAnimation(AnimationPath::builtin("SCNRVICT.DEF"));
  214. iconsLossCondition = GH.renderHandler().loadAnimation(AnimationPath::builtin("SCNRLOSS.DEF"));
  215. for(int i = 0; i < positionsToShow; i++)
  216. listItems.push_back(std::make_shared<ListItem>(Point(30, 129 + i * 25), iconsMapFormats, iconsVictoryCondition, iconsLossCondition));
  217. labelTabTitle = std::make_shared<CLabel>(205, 28, FONT_MEDIUM, ETextAlignment::CENTER, Colors::YELLOW, tabTitle);
  218. slider = std::make_shared<CSlider>(Point(372, 86 + (enableUiEnhancements ? 30 : 0)), (tabType != ESelectionScreen::saveGame ? 480 : 430) - (enableUiEnhancements ? 30 : 0), std::bind(&SelectionTab::sliderMove, this, _1), positionsToShow, (int)curItems.size(), 0, Orientation::VERTICAL, CSlider::BLUE);
  219. slider->setPanningStep(24);
  220. // create scroll bounds that encompass all area in this UI element to the left of slider (including area of slider itself)
  221. // entire screen can't be used in here since map description might also have slider
  222. slider->setScrollBounds(Rect(pos.x - slider->pos.x, 0, slider->pos.x + slider->pos.w - pos.x, slider->pos.h ));
  223. filter(0);
  224. }
  225. void SelectionTab::toggleMode()
  226. {
  227. if(CSH->isGuest())
  228. {
  229. allItems.clear();
  230. curItems.clear();
  231. if(slider)
  232. slider->block(true);
  233. }
  234. else
  235. {
  236. switch(tabType)
  237. {
  238. case ESelectionScreen::newGame:
  239. {
  240. inputName->disable();
  241. auto files = getFiles("Maps/", EResType::MAP);
  242. files.erase(ResourcePath("Maps/Tutorial.tut", EResType::MAP));
  243. parseMaps(files);
  244. break;
  245. }
  246. case ESelectionScreen::loadGame:
  247. inputName->disable();
  248. parseSaves(getFiles("Saves/", EResType::SAVEGAME));
  249. break;
  250. case ESelectionScreen::saveGame:
  251. parseSaves(getFiles("Saves/", EResType::SAVEGAME));
  252. inputName->enable();
  253. inputName->activate();
  254. restoreLastSelection();
  255. break;
  256. case ESelectionScreen::campaignList:
  257. parseCampaigns(getFiles("Maps/", EResType::CAMPAIGN));
  258. break;
  259. default:
  260. assert(0);
  261. break;
  262. }
  263. if(slider)
  264. {
  265. slider->block(false);
  266. filter(0);
  267. }
  268. if(CSH->campaignStateToSend)
  269. {
  270. CSH->setCampaignState(CSH->campaignStateToSend);
  271. CSH->campaignStateToSend.reset();
  272. }
  273. else
  274. {
  275. restoreLastSelection();
  276. }
  277. }
  278. slider->setAmount((int)curItems.size());
  279. updateListItems();
  280. redraw();
  281. }
  282. void SelectionTab::clickReleased(const Point & cursorPosition)
  283. {
  284. int line = getLine();
  285. if(line != -1)
  286. {
  287. select(line);
  288. }
  289. #ifdef VCMI_MOBILE
  290. // focus input field if clicked inside it
  291. else if(inputName && inputName->isActive() && inputNameRect.isInside(cursorPosition))
  292. inputName->giveFocus();
  293. #endif
  294. }
  295. void SelectionTab::keyPressed(EShortcut key)
  296. {
  297. int moveBy = 0;
  298. switch(key)
  299. {
  300. case EShortcut::MOVE_UP:
  301. moveBy = -1;
  302. break;
  303. case EShortcut::MOVE_DOWN:
  304. moveBy = +1;
  305. break;
  306. case EShortcut::MOVE_PAGE_UP:
  307. moveBy = -(int)listItems.size() + 1;
  308. break;
  309. case EShortcut::MOVE_PAGE_DOWN:
  310. moveBy = +(int)listItems.size() - 1;
  311. break;
  312. case EShortcut::MOVE_FIRST:
  313. select(-slider->getValue());
  314. return;
  315. case EShortcut::MOVE_LAST:
  316. select((int)curItems.size() - slider->getValue());
  317. return;
  318. default:
  319. return;
  320. }
  321. select((int)selectionPos - slider->getValue() + moveBy);
  322. }
  323. void SelectionTab::clickDouble(const Point & cursorPosition)
  324. {
  325. int position = getLine();
  326. int itemIndex = position + slider->getValue();
  327. if(itemIndex >= curItems.size())
  328. return;
  329. auto clickedItem = curItems[itemIndex];
  330. auto selectedItem = getSelectedMapInfo();
  331. if (clickedItem != selectedItem)
  332. {
  333. // double-click BUT player hit different item than he had selected
  334. // ignore - clickReleased would still trigger and update selection.
  335. // After which another (3rd) click if it happens would still register as double-click
  336. return;
  337. }
  338. if(itemIndex >= 0 && curItems[itemIndex]->isFolder)
  339. {
  340. select(position);
  341. return;
  342. }
  343. if(getLine() != -1) //double clicked scenarios list
  344. {
  345. (static_cast<CLobbyScreen *>(parent))->buttonStart->clickPressed(cursorPosition);
  346. (static_cast<CLobbyScreen *>(parent))->buttonStart->clickReleased(cursorPosition);
  347. }
  348. }
  349. void SelectionTab::showPopupWindow(const Point & cursorPosition)
  350. {
  351. int position = getLine();
  352. int py = position + slider->getValue();
  353. if(py >= curItems.size())
  354. return;
  355. if(!curItems[py]->isFolder)
  356. GH.windows().createAndPushWindow<CMapOverview>(curItems[py]->getNameTranslated(), curItems[py]->fullFileURI, curItems[py]->date, ResourcePath(curItems[py]->fileURI), tabType);
  357. else
  358. CRClickPopup::createAndPush(curItems[py]->folderName);
  359. }
  360. auto SelectionTab::checkSubfolder(std::string path)
  361. {
  362. struct Ret
  363. {
  364. std::string folderName;
  365. std::string baseFolder;
  366. bool parentExists;
  367. bool fileInFolder;
  368. } ret;
  369. ret.parentExists = (curFolder != "");
  370. ret.fileInFolder = false;
  371. std::vector<std::string> filetree;
  372. // delete first element (e.g. 'MAPS')
  373. boost::split(filetree, path, boost::is_any_of("/"));
  374. filetree.erase(filetree.begin());
  375. std::string pathWithoutPrefix = boost::algorithm::join(filetree, "/");
  376. if(!filetree.empty())
  377. {
  378. filetree.pop_back();
  379. ret.baseFolder = boost::algorithm::join(filetree, "/");
  380. }
  381. else
  382. ret.baseFolder = "";
  383. if(boost::algorithm::starts_with(ret.baseFolder, curFolder))
  384. {
  385. std::string folder = ret.baseFolder.substr(curFolder.size());
  386. if(folder != "")
  387. {
  388. boost::split(filetree, folder, boost::is_any_of("/"));
  389. ret.folderName = filetree[0];
  390. }
  391. }
  392. if(boost::algorithm::starts_with(pathWithoutPrefix, curFolder))
  393. if(boost::count(pathWithoutPrefix.substr(curFolder.size()), '/') == 0)
  394. ret.fileInFolder = true;
  395. return ret;
  396. }
  397. // A new size filter (Small, Medium, ...) has been selected. Populate
  398. // selMaps with the relevant data.
  399. void SelectionTab::filter(int size, bool selectFirst)
  400. {
  401. if(size == -1)
  402. size = currentMapSizeFilter;
  403. currentMapSizeFilter = size;
  404. curItems.clear();
  405. for(auto elem : allItems)
  406. {
  407. if((elem->mapHeader && (!size || elem->mapHeader->width == size)) || tabType == ESelectionScreen::campaignList)
  408. {
  409. if(showRandom)
  410. curFolder = "RANDOMMAPS/";
  411. auto [folderName, baseFolder, parentExists, fileInFolder] = checkSubfolder(elem->originalFileURI);
  412. if((showRandom && baseFolder != "RANDOMMAPS") || (!showRandom && baseFolder == "RANDOMMAPS"))
  413. continue;
  414. if(parentExists && !showRandom)
  415. {
  416. auto folder = std::make_shared<ElementInfo>();
  417. folder->isFolder = true;
  418. folder->folderName = ".. (" + curFolder + ")";
  419. auto itemIt = boost::range::find_if(curItems, [](std::shared_ptr<ElementInfo> e) { return boost::starts_with(e->folderName, ".."); });
  420. if (itemIt == curItems.end()) {
  421. curItems.push_back(folder);
  422. }
  423. }
  424. auto folder = std::make_shared<ElementInfo>();
  425. folder->isFolder = true;
  426. folder->folderName = folderName;
  427. auto itemIt = boost::range::find_if(curItems, [folder](std::shared_ptr<ElementInfo> e) { return e->folderName == folder->folderName; });
  428. if (itemIt == curItems.end() && folderName != "") {
  429. curItems.push_back(folder);
  430. }
  431. if(fileInFolder)
  432. curItems.push_back(elem);
  433. }
  434. }
  435. if(curItems.size())
  436. {
  437. slider->block(false);
  438. slider->setAmount((int)curItems.size());
  439. sort();
  440. if(selectFirst)
  441. {
  442. int firstPos = boost::range::find_if(curItems, [](std::shared_ptr<ElementInfo> e) { return !e->isFolder; }) - curItems.begin();
  443. if(firstPos < curItems.size())
  444. {
  445. slider->scrollTo(firstPos);
  446. callOnSelect(curItems[firstPos]);
  447. selectAbs(firstPos);
  448. }
  449. }
  450. }
  451. else
  452. {
  453. updateListItems();
  454. redraw();
  455. slider->block(true);
  456. if(callOnSelect)
  457. callOnSelect(nullptr);
  458. }
  459. }
  460. void SelectionTab::sortBy(int criteria)
  461. {
  462. if(criteria == sortingBy)
  463. {
  464. sortModeAscending = !sortModeAscending;
  465. }
  466. else
  467. {
  468. sortingBy = (ESortBy)criteria;
  469. sortModeAscending = true;
  470. }
  471. sort();
  472. selectAbs(-1);
  473. }
  474. void SelectionTab::sort()
  475. {
  476. if(sortingBy != generalSortingBy)
  477. std::stable_sort(curItems.begin(), curItems.end(), mapSorter(generalSortingBy));
  478. std::stable_sort(curItems.begin(), curItems.end(), mapSorter(sortingBy));
  479. int firstMapIndex = boost::range::find_if(curItems, [](std::shared_ptr<ElementInfo> e) { return !e->isFolder; }) - curItems.begin();
  480. if(!sortModeAscending)
  481. std::reverse(std::next(curItems.begin(), firstMapIndex), curItems.end());
  482. updateListItems();
  483. redraw();
  484. }
  485. void SelectionTab::select(int position)
  486. {
  487. if(!curItems.size())
  488. return;
  489. // New selection. py is the index in curItems.
  490. int py = position + slider->getValue();
  491. vstd::amax(py, 0);
  492. vstd::amin(py, curItems.size() - 1);
  493. selectionPos = py;
  494. if(position < 0)
  495. slider->scrollBy(position);
  496. else if(position >= listItems.size())
  497. slider->scrollBy(position - (int)listItems.size() + 1);
  498. if(curItems[py]->isFolder) {
  499. if(boost::starts_with(curItems[py]->folderName, ".."))
  500. {
  501. std::vector<std::string> filetree;
  502. boost::split(filetree, curFolder, boost::is_any_of("/"));
  503. filetree.pop_back();
  504. filetree.pop_back();
  505. curFolder = filetree.size() > 0 ? boost::algorithm::join(filetree, "/") + "/" : "";
  506. }
  507. else
  508. curFolder += curItems[py]->folderName + "/";
  509. filter(-1);
  510. slider->scrollTo(0);
  511. int firstPos = boost::range::find_if(curItems, [](std::shared_ptr<ElementInfo> e) { return !e->isFolder; }) - curItems.begin();
  512. if(firstPos < curItems.size())
  513. {
  514. selectAbs(firstPos);
  515. }
  516. return;
  517. }
  518. rememberCurrentSelection();
  519. if(inputName && inputName->isActive())
  520. {
  521. auto filename = *CResourceHandler::get()->getResourceName(ResourcePath(curItems[py]->fileURI, EResType::SAVEGAME));
  522. inputName->setText(filename.stem().string());
  523. }
  524. updateListItems();
  525. redraw();
  526. if(callOnSelect)
  527. callOnSelect(curItems[py]);
  528. }
  529. void SelectionTab::selectAbs(int position)
  530. {
  531. if(position == -1)
  532. position = boost::range::find_if(curItems, [](std::shared_ptr<ElementInfo> e) { return !e->isFolder; }) - curItems.begin();
  533. select(position - slider->getValue());
  534. }
  535. void SelectionTab::sliderMove(int slidPos)
  536. {
  537. if(!slider)
  538. return; // ignore spurious call when slider is being created
  539. updateListItems();
  540. redraw();
  541. }
  542. void SelectionTab::updateListItems()
  543. {
  544. // elemIdx is the index of the maps or saved game to display on line 0
  545. // slider->capacity contains the number of available screen lines
  546. // slider->positionsAmnt is the number of elements after filtering
  547. int elemIdx = slider->getValue();
  548. for(auto item : listItems)
  549. {
  550. if(elemIdx < curItems.size())
  551. {
  552. item->updateItem(curItems[elemIdx], elemIdx == selectionPos);
  553. elemIdx++;
  554. }
  555. else
  556. {
  557. item->updateItem();
  558. }
  559. }
  560. }
  561. bool SelectionTab::receiveEvent(const Point & position, int eventType) const
  562. {
  563. // FIXME: widget should instead have well-defined pos so events will be filtered using standard routine
  564. return getLine(position - pos.topLeft()) != -1;
  565. }
  566. int SelectionTab::getLine() const
  567. {
  568. Point clickPos = GH.getCursorPosition() - pos.topLeft();
  569. return getLine(clickPos);
  570. }
  571. int SelectionTab::getLine(const Point & clickPos) const
  572. {
  573. int line = -1;
  574. // Ignore clicks on save name area
  575. int maxPosY;
  576. if(tabType == ESelectionScreen::saveGame)
  577. maxPosY = 516;
  578. else
  579. maxPosY = 564;
  580. if(clickPos.y > 115 && clickPos.y < maxPosY && clickPos.x > 22 && clickPos.x < 371)
  581. {
  582. line = (clickPos.y - 115) / 25; //which line
  583. }
  584. return line;
  585. }
  586. void SelectionTab::selectFileName(std::string fname)
  587. {
  588. boost::to_upper(fname);
  589. for(int i = (int)allItems.size() - 1; i >= 0; i--)
  590. {
  591. if(allItems[i]->fileURI == fname)
  592. {
  593. auto [folderName, baseFolder, parentExists, fileInFolder] = checkSubfolder(allItems[i]->originalFileURI);
  594. curFolder = baseFolder != "" ? baseFolder + "/" : "";
  595. }
  596. }
  597. for(int i = (int)curItems.size() - 1; i >= 0; i--)
  598. {
  599. if(curItems[i]->fileURI == fname)
  600. {
  601. slider->scrollTo(i);
  602. selectAbs(i);
  603. return;
  604. }
  605. }
  606. filter(-1);
  607. selectAbs(-1);
  608. if(tabType == ESelectionScreen::saveGame && inputName->getText().empty())
  609. inputName->setText("NEWGAME");
  610. }
  611. std::shared_ptr<ElementInfo> SelectionTab::getSelectedMapInfo() const
  612. {
  613. return curItems.empty() || curItems[selectionPos]->isFolder ? nullptr : curItems[selectionPos];
  614. }
  615. void SelectionTab::rememberCurrentSelection()
  616. {
  617. if(getSelectedMapInfo()->isFolder)
  618. return;
  619. // TODO: this can be more elegant
  620. if(tabType == ESelectionScreen::newGame)
  621. {
  622. Settings lastMap = settings.write["general"]["lastMap"];
  623. lastMap->String() = getSelectedMapInfo()->fileURI;
  624. }
  625. else if(tabType == ESelectionScreen::loadGame)
  626. {
  627. Settings lastSave = settings.write["general"]["lastSave"];
  628. lastSave->String() = getSelectedMapInfo()->fileURI;
  629. }
  630. else if(tabType == ESelectionScreen::campaignList)
  631. {
  632. Settings lastCampaign = settings.write["general"]["lastCampaign"];
  633. lastCampaign->String() = getSelectedMapInfo()->fileURI;
  634. }
  635. }
  636. void SelectionTab::restoreLastSelection()
  637. {
  638. switch(tabType)
  639. {
  640. case ESelectionScreen::newGame:
  641. selectFileName(settings["general"]["lastMap"].String());
  642. break;
  643. case ESelectionScreen::campaignList:
  644. selectFileName(settings["general"]["lastCampaign"].String());
  645. break;
  646. case ESelectionScreen::loadGame:
  647. case ESelectionScreen::saveGame:
  648. selectFileName(settings["general"]["lastSave"].String());
  649. }
  650. }
  651. bool SelectionTab::isMapSupported(const CMapInfo & info)
  652. {
  653. switch (info.mapHeader->version)
  654. {
  655. case EMapFormat::ROE:
  656. return CGI->settings()->getValue(EGameSettings::MAP_FORMAT_RESTORATION_OF_ERATHIA)["supported"].Bool();
  657. case EMapFormat::AB:
  658. return CGI->settings()->getValue(EGameSettings::MAP_FORMAT_ARMAGEDDONS_BLADE)["supported"].Bool();
  659. case EMapFormat::SOD:
  660. return CGI->settings()->getValue(EGameSettings::MAP_FORMAT_SHADOW_OF_DEATH)["supported"].Bool();
  661. case EMapFormat::WOG:
  662. return CGI->settings()->getValue(EGameSettings::MAP_FORMAT_IN_THE_WAKE_OF_GODS)["supported"].Bool();
  663. case EMapFormat::HOTA:
  664. return CGI->settings()->getValue(EGameSettings::MAP_FORMAT_HORN_OF_THE_ABYSS)["supported"].Bool();
  665. case EMapFormat::VCMI:
  666. return CGI->settings()->getValue(EGameSettings::MAP_FORMAT_JSON_VCMI)["supported"].Bool();
  667. }
  668. return false;
  669. }
  670. void SelectionTab::parseMaps(const std::unordered_set<ResourcePath> & files)
  671. {
  672. logGlobal->debug("Parsing %d maps", files.size());
  673. allItems.clear();
  674. for(auto & file : files)
  675. {
  676. try
  677. {
  678. auto mapInfo = std::make_shared<ElementInfo>();
  679. mapInfo->mapInit(file.getName());
  680. if (isMapSupported(*mapInfo))
  681. allItems.push_back(mapInfo);
  682. }
  683. catch(std::exception & e)
  684. {
  685. logGlobal->error("Map %s is invalid. Message: %s", file.getName(), e.what());
  686. }
  687. }
  688. }
  689. void SelectionTab::parseSaves(const std::unordered_set<ResourcePath> & files)
  690. {
  691. for(auto & file : files)
  692. {
  693. try
  694. {
  695. auto mapInfo = std::make_shared<ElementInfo>();
  696. mapInfo->saveInit(file);
  697. // Filter out other game modes
  698. bool isCampaign = mapInfo->scenarioOptionsOfSave->mode == EStartMode::CAMPAIGN;
  699. bool isMultiplayer = mapInfo->amountOfHumanPlayersInSave > 1;
  700. bool isTutorial = boost::to_upper_copy(mapInfo->scenarioOptionsOfSave->mapname) == "MAPS/TUTORIAL";
  701. switch(CSH->getLoadMode())
  702. {
  703. case ELoadMode::SINGLE:
  704. if(isMultiplayer || isCampaign || isTutorial)
  705. mapInfo->mapHeader.reset();
  706. break;
  707. case ELoadMode::CAMPAIGN:
  708. if(!isCampaign)
  709. mapInfo->mapHeader.reset();
  710. break;
  711. case ELoadMode::TUTORIAL:
  712. if(!isTutorial)
  713. mapInfo->mapHeader.reset();
  714. break;
  715. default:
  716. if(!isMultiplayer)
  717. mapInfo->mapHeader.reset();
  718. break;
  719. }
  720. allItems.push_back(mapInfo);
  721. }
  722. catch(const std::exception & e)
  723. {
  724. logGlobal->error("Error: Failed to process %s: %s", file.getName(), e.what());
  725. }
  726. }
  727. }
  728. void SelectionTab::parseCampaigns(const std::unordered_set<ResourcePath> & files)
  729. {
  730. allItems.reserve(files.size());
  731. for(auto & file : files)
  732. {
  733. auto info = std::make_shared<ElementInfo>();
  734. //allItems[i].date = std::asctime(std::localtime(&files[i].date));
  735. info->fileURI = file.getName();
  736. info->campaignInit();
  737. if(info->campaign)
  738. allItems.push_back(info);
  739. }
  740. }
  741. std::unordered_set<ResourcePath> SelectionTab::getFiles(std::string dirURI, EResType resType)
  742. {
  743. boost::to_upper(dirURI);
  744. CResourceHandler::get()->updateFilteredFiles([&](const std::string & mount)
  745. {
  746. return boost::algorithm::starts_with(mount, dirURI);
  747. });
  748. std::unordered_set<ResourcePath> ret = CResourceHandler::get()->getFilteredFiles([&](const ResourcePath & ident)
  749. {
  750. return ident.getType() == resType && boost::algorithm::starts_with(ident.getName(), dirURI);
  751. });
  752. return ret;
  753. }
  754. SelectionTab::ListItem::ListItem(Point position, std::shared_ptr<CAnimation> iconsFormats, std::shared_ptr<CAnimation> iconsVictory, std::shared_ptr<CAnimation> iconsLoss)
  755. : CIntObject(LCLICK, position)
  756. {
  757. OBJ_CONSTRUCTION_CAPTURING_ALL_NO_DISPOSE;
  758. pictureEmptyLine = std::make_shared<CPicture>(GH.renderHandler().loadImage(ImagePath::builtin("camcust")), Rect(25, 121, 349, 26), -8, -14);
  759. labelName = std::make_shared<CLabel>(184, 0, FONT_SMALL, ETextAlignment::CENTER, Colors::WHITE, "", 185);
  760. labelName->setAutoRedraw(false);
  761. labelAmountOfPlayers = std::make_shared<CLabel>(8, 0, FONT_SMALL, ETextAlignment::CENTER, Colors::WHITE);
  762. labelAmountOfPlayers->setAutoRedraw(false);
  763. labelNumberOfCampaignMaps = std::make_shared<CLabel>(8, 0, FONT_SMALL, ETextAlignment::CENTER, Colors::WHITE);
  764. labelNumberOfCampaignMaps->setAutoRedraw(false);
  765. labelMapSizeLetter = std::make_shared<CLabel>(41, 0, FONT_SMALL, ETextAlignment::CENTER, Colors::WHITE);
  766. labelMapSizeLetter->setAutoRedraw(false);
  767. // FIXME: This -12 should not be needed, but for some reason CAnimImage displaced otherwise
  768. iconFolder = std::make_shared<CPicture>(ImagePath::builtin("lobby/iconFolder.png"), -8, -12);
  769. iconFormat = std::make_shared<CAnimImage>(iconsFormats, 0, 0, 59, -12);
  770. iconVictoryCondition = std::make_shared<CAnimImage>(iconsVictory, 0, 0, 277, -12);
  771. iconLossCondition = std::make_shared<CAnimImage>(iconsLoss, 0, 0, 310, -12);
  772. }
  773. void SelectionTab::ListItem::updateItem(std::shared_ptr<ElementInfo> info, bool selected)
  774. {
  775. if(!info)
  776. {
  777. labelAmountOfPlayers->disable();
  778. labelMapSizeLetter->disable();
  779. iconFolder->disable();
  780. pictureEmptyLine->disable();
  781. iconFormat->disable();
  782. iconVictoryCondition->disable();
  783. iconLossCondition->disable();
  784. labelNumberOfCampaignMaps->disable();
  785. labelName->disable();
  786. return;
  787. }
  788. auto color = selected ? Colors::YELLOW : Colors::WHITE;
  789. if(info->isFolder)
  790. {
  791. labelAmountOfPlayers->disable();
  792. labelMapSizeLetter->disable();
  793. iconFolder->enable();
  794. pictureEmptyLine->enable();
  795. iconFormat->disable();
  796. iconVictoryCondition->disable();
  797. iconLossCondition->disable();
  798. labelNumberOfCampaignMaps->disable();
  799. labelName->enable();
  800. labelName->setMaxWidth(316);
  801. labelName->setText(info->folderName);
  802. labelName->setColor(color);
  803. return;
  804. }
  805. labelName->enable();
  806. if(info->campaign)
  807. {
  808. labelAmountOfPlayers->disable();
  809. labelMapSizeLetter->disable();
  810. iconFolder->disable();
  811. pictureEmptyLine->disable();
  812. iconFormat->disable();
  813. iconVictoryCondition->disable();
  814. iconLossCondition->disable();
  815. labelNumberOfCampaignMaps->enable();
  816. std::ostringstream ostr(std::ostringstream::out);
  817. ostr << info->campaign->scenariosCount();
  818. labelNumberOfCampaignMaps->setText(ostr.str());
  819. labelNumberOfCampaignMaps->setColor(color);
  820. labelName->setMaxWidth(316);
  821. }
  822. else
  823. {
  824. labelNumberOfCampaignMaps->disable();
  825. std::ostringstream ostr(std::ostringstream::out);
  826. ostr << info->amountOfPlayersOnMap << "/" << info->amountOfHumanControllablePlayers;
  827. labelAmountOfPlayers->enable();
  828. labelAmountOfPlayers->setText(ostr.str());
  829. labelAmountOfPlayers->setColor(color);
  830. labelMapSizeLetter->enable();
  831. labelMapSizeLetter->setText(info->getMapSizeName());
  832. labelMapSizeLetter->setColor(color);
  833. iconFolder->disable();
  834. pictureEmptyLine->disable();
  835. iconFormat->enable();
  836. iconFormat->setFrame(info->getMapSizeFormatIconId());
  837. iconVictoryCondition->enable();
  838. iconVictoryCondition->setFrame(info->mapHeader->victoryIconIndex, 0);
  839. iconLossCondition->enable();
  840. iconLossCondition->setFrame(info->mapHeader->defeatIconIndex, 0);
  841. labelName->setMaxWidth(185);
  842. }
  843. labelName->setText(info->getNameForList());
  844. labelName->setColor(color);
  845. }