mainwindow.cpp 42 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381
  1. /*
  2. * mainwindow.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 "mainwindow.h"
  12. #include "ui_mainwindow.h"
  13. #include <QFileDialog>
  14. #include <QFile>
  15. #include <QMessageBox>
  16. #include <QFileInfo>
  17. #include "../lib/VCMIDirs.h"
  18. #include "../lib/VCMI_Lib.h"
  19. #include "../lib/logging/CBasicLogConfigurator.h"
  20. #include "../lib/CConfigHandler.h"
  21. #include "../lib/filesystem/Filesystem.h"
  22. #include "../lib/GameConstants.h"
  23. #include "../lib/mapObjectConstructors/AObjectTypeHandler.h"
  24. #include "../lib/mapObjectConstructors/CObjectClassesHandler.h"
  25. #include "../lib/mapObjects/ObjectTemplate.h"
  26. #include "../lib/mapping/CMapService.h"
  27. #include "../lib/mapping/CMap.h"
  28. #include "../lib/mapping/CMapEditManager.h"
  29. #include "../lib/mapping/MapFormat.h"
  30. #include "../lib/modding/ModIncompatibility.h"
  31. #include "../lib/RoadHandler.h"
  32. #include "../lib/RiverHandler.h"
  33. #include "../lib/TerrainHandler.h"
  34. #include "../lib/filesystem/CFilesystemLoader.h"
  35. #include "maphandler.h"
  36. #include "graphics.h"
  37. #include "windownewmap.h"
  38. #include "objectbrowser.h"
  39. #include "inspector/inspector.h"
  40. #include "mapsettings/mapsettings.h"
  41. #include "mapsettings/translations.h"
  42. #include "playersettings.h"
  43. #include "validator.h"
  44. static CBasicLogConfigurator * logConfig;
  45. QJsonValue jsonFromPixmap(const QPixmap &p)
  46. {
  47. QBuffer buffer;
  48. buffer.open(QIODevice::WriteOnly);
  49. p.save(&buffer, "PNG");
  50. auto const encoded = buffer.data().toBase64();
  51. return {QLatin1String(encoded)};
  52. }
  53. QPixmap pixmapFromJson(const QJsonValue &val)
  54. {
  55. auto const encoded = val.toString().toLatin1();
  56. QPixmap p;
  57. p.loadFromData(QByteArray::fromBase64(encoded), "PNG");
  58. return p;
  59. }
  60. void init()
  61. {
  62. loadDLLClasses();
  63. Settings config = settings.write["session"]["editor"];
  64. config->Bool() = true;
  65. logGlobal->info("Initializing VCMI_Lib");
  66. }
  67. void MainWindow::loadUserSettings()
  68. {
  69. //load window settings
  70. QSettings s(Ui::teamName, Ui::appName);
  71. auto size = s.value(mainWindowSizeSetting).toSize();
  72. if (size.isValid())
  73. {
  74. resize(size);
  75. }
  76. auto position = s.value(mainWindowPositionSetting).toPoint();
  77. if (!position.isNull())
  78. {
  79. move(position);
  80. }
  81. lastSavingDir = s.value(lastDirectorySetting).toString();
  82. if(lastSavingDir.isEmpty())
  83. lastSavingDir = QString::fromStdString(VCMIDirs::get().userDataPath().make_preferred().string());
  84. }
  85. void MainWindow::saveUserSettings()
  86. {
  87. QSettings s(Ui::teamName, Ui::appName);
  88. s.setValue(mainWindowSizeSetting, size());
  89. s.setValue(mainWindowPositionSetting, pos());
  90. s.setValue(lastDirectorySetting, lastSavingDir);
  91. }
  92. void MainWindow::parseCommandLine(ExtractionOptions & extractionOptions)
  93. {
  94. QCommandLineParser parser;
  95. parser.addHelpOption();
  96. parser.addPositionalArgument("map", QCoreApplication::translate("main", "Filepath of the map to open."));
  97. parser.addOptions({
  98. {"e", QCoreApplication::translate("main", "Extract original H3 archives into a separate folder.")},
  99. {"s", QCoreApplication::translate("main", "From an extracted archive, it Splits TwCrPort, CPRSMALL, FlagPort, ITPA, ITPt, Un32 and Un44 into individual PNG's.")},
  100. {"c", QCoreApplication::translate("main", "From an extracted archive, Converts single Images (found in Images folder) from .pcx to png.")},
  101. {"d", QCoreApplication::translate("main", "Delete original files, for the ones split / converted.")},
  102. });
  103. parser.process(qApp->arguments());
  104. const QStringList positionalArgs = parser.positionalArguments();
  105. if(!positionalArgs.isEmpty())
  106. mapFilePath = positionalArgs.at(0);
  107. extractionOptions = {
  108. parser.isSet("e"), {
  109. parser.isSet("s"),
  110. parser.isSet("c"),
  111. parser.isSet("d")}};
  112. }
  113. void MainWindow::loadTranslation()
  114. {
  115. #ifdef ENABLE_QT_TRANSLATIONS
  116. const std::string translationFile = settings["general"]["language"].String()+ ".qm";
  117. QString translationFileResourcePath = QString{":/translation/%1"}.arg(translationFile.c_str());
  118. logGlobal->info("Loading translation %s", translationFile);
  119. if(!QFile::exists(translationFileResourcePath))
  120. {
  121. logGlobal->warn("Translation file %s does not exist", translationFileResourcePath.toStdString());
  122. return;
  123. }
  124. if (!translator.load(translationFileResourcePath))
  125. {
  126. logGlobal->error("Failed to load translation file %s", translationFileResourcePath.toStdString());
  127. return;
  128. }
  129. if(translationFile == "english.qm")
  130. {
  131. // translator doesn't need to be installed for English
  132. return;
  133. }
  134. if (!qApp->installTranslator(&translator))
  135. {
  136. logGlobal->error("Failed to install translator for translation file %s", translationFileResourcePath.toStdString());
  137. }
  138. #endif
  139. }
  140. MainWindow::MainWindow(QWidget* parent) :
  141. QMainWindow(parent),
  142. ui(new Ui::MainWindow),
  143. controller(this)
  144. {
  145. // Set current working dir to executable folder.
  146. // This is important on Mac for relative paths to work inside DMG.
  147. QDir::setCurrent(QApplication::applicationDirPath());
  148. new QShortcut(QKeySequence("Backspace"), this, SLOT(on_actionErase_triggered()));
  149. ExtractionOptions extractionOptions;
  150. parseCommandLine(extractionOptions);
  151. //configure logging
  152. const boost::filesystem::path logPath = VCMIDirs::get().userLogsPath() / "VCMI_Editor_log.txt";
  153. console = new CConsoleHandler();
  154. logConfig = new CBasicLogConfigurator(logPath, console);
  155. logConfig->configureDefault();
  156. logGlobal->info("The log file will be saved to %s", logPath);
  157. //init
  158. preinitDLL(::console, extractionOptions.extractArchives);
  159. // Initialize logging based on settings
  160. logConfig->configure();
  161. logGlobal->debug("settings = %s", settings.toJsonNode().toString());
  162. // Some basic data validation to produce better error messages in cases of incorrect install
  163. auto testFile = [](std::string filename, std::string message) -> bool
  164. {
  165. if (CResourceHandler::get()->existsResource(ResourcePath(filename)))
  166. return true;
  167. logGlobal->error("Error: %s was not found!", message);
  168. return false;
  169. };
  170. if (!testFile("DATA/HELP.TXT", "Heroes III data") ||
  171. !testFile("MODS/VCMI/MOD.JSON", "VCMI data"))
  172. {
  173. QApplication::quit();
  174. }
  175. loadTranslation();
  176. ui->setupUi(this);
  177. setWindowIcon(QIcon{":/icons/menu-game.png"});
  178. ui->toolBrush->setIcon(QIcon{":/icons/brush-1.png"});
  179. ui->toolBrush2->setIcon(QIcon{":/icons/brush-2.png"});
  180. ui->toolBrush4->setIcon(QIcon{":/icons/brush-4.png"});
  181. ui->toolLasso->setIcon(QIcon{":/icons/tool-lasso.png"});
  182. ui->toolLine->setIcon(QIcon{":/icons/tool-line.png"});
  183. ui->toolArea->setIcon(QIcon{":/icons/tool-area.png"});
  184. ui->toolFill->setIcon(QIcon{":/icons/tool-fill.png"});
  185. ui->toolSelect->setIcon(QIcon{":/icons/tool-select.png"});
  186. ui->actionOpen->setIcon(QIcon{":/icons/document-open.png"});
  187. ui->actionSave->setIcon(QIcon{":/icons/document-save.png"});
  188. ui->actionNew->setIcon(QIcon{":/icons/document-new.png"});
  189. ui->actionLevel->setIcon(QIcon{":/icons/toggle-underground.png"});
  190. ui->actionPass->setIcon(QIcon{":/icons/toggle-pass.png"});
  191. ui->actionCut->setIcon(QIcon{":/icons/edit-cut.png"});
  192. ui->actionCopy->setIcon(QIcon{":/icons/edit-copy.png"});
  193. ui->actionPaste->setIcon(QIcon{":/icons/edit-paste.png"});
  194. ui->actionFill->setIcon(QIcon{":/icons/fill-obstacles.png"});
  195. ui->actionGrid->setIcon(QIcon{":/icons/toggle-grid.png"});
  196. ui->actionUndo->setIcon(QIcon{":/icons/edit-undo.png"});
  197. ui->actionRedo->setIcon(QIcon{":/icons/edit-redo.png"});
  198. ui->actionErase->setIcon(QIcon{":/icons/edit-clear.png"});
  199. ui->actionTranslations->setIcon(QIcon{":/icons/translations.png"});
  200. ui->actionLock->setIcon(QIcon{":/icons/lock-closed.png"});
  201. ui->actionUnlock->setIcon(QIcon{":/icons/lock-open.png"});
  202. ui->actionZoom_in->setIcon(QIcon{":/icons/zoom_plus.png"});
  203. ui->actionZoom_out->setIcon(QIcon{":/icons/zoom_minus.png"});
  204. ui->actionZoom_reset->setIcon(QIcon{":/icons/zoom_zero.png"});
  205. loadUserSettings(); //For example window size
  206. setTitle();
  207. init();
  208. graphics = new Graphics(); // should be before curh->init()
  209. graphics->load();//must be after Content loading but should be in main thread
  210. if (extractionOptions.extractArchives)
  211. ResourceConverter::convertExtractedResourceFiles(extractionOptions.conversionOptions);
  212. ui->mapView->setScene(controller.scene(0));
  213. ui->mapView->setController(&controller);
  214. ui->mapView->setOptimizationFlags(QGraphicsView::DontSavePainterState | QGraphicsView::DontAdjustForAntialiasing);
  215. connect(ui->mapView, &MapView::openObjectProperties, this, &MainWindow::loadInspector);
  216. connect(ui->mapView, &MapView::currentCoordinates, this, &MainWindow::currentCoordinatesChanged);
  217. ui->minimapView->setScene(controller.miniScene(0));
  218. ui->minimapView->setController(&controller);
  219. connect(ui->minimapView, &MinimapView::cameraPositionChanged, ui->mapView, &MapView::cameraChanged);
  220. scenePreview = new QGraphicsScene(this);
  221. ui->objectPreview->setScene(scenePreview);
  222. //loading objects
  223. loadObjectsTree();
  224. ui->tabWidget->setCurrentIndex(0);
  225. for(int i = 0; i < PlayerColor::PLAYER_LIMIT.getNum(); ++i)
  226. {
  227. connect(getActionPlayer(PlayerColor(i)), &QAction::toggled, this, [&, i](){switchDefaultPlayer(PlayerColor(i));});
  228. }
  229. connect(getActionPlayer(PlayerColor::NEUTRAL), &QAction::toggled, this, [&](){switchDefaultPlayer(PlayerColor::NEUTRAL);});
  230. onPlayersChanged();
  231. show();
  232. //Load map from command line
  233. if(!mapFilePath.isEmpty())
  234. openMap(mapFilePath);
  235. }
  236. MainWindow::~MainWindow()
  237. {
  238. saveUserSettings(); //save window size etc.
  239. delete ui;
  240. }
  241. bool MainWindow::getAnswerAboutUnsavedChanges()
  242. {
  243. if(unsaved)
  244. {
  245. auto sure = QMessageBox::question(this, tr("Confirmation"), tr("Unsaved changes will be lost, are you sure?"));
  246. if(sure == QMessageBox::No)
  247. {
  248. return false;
  249. }
  250. }
  251. return true;
  252. }
  253. void MainWindow::closeEvent(QCloseEvent *event)
  254. {
  255. if(getAnswerAboutUnsavedChanges())
  256. QMainWindow::closeEvent(event);
  257. else
  258. event->ignore();
  259. }
  260. void MainWindow::setStatusMessage(const QString & status)
  261. {
  262. statusBar()->showMessage(status);
  263. }
  264. void MainWindow::setTitle()
  265. {
  266. QString title = QString("%1%2 - %3 (v%4)").arg(filename, unsaved ? "*" : "", VCMI_EDITOR_NAME, VCMI_EDITOR_VERSION);
  267. setWindowTitle(title);
  268. }
  269. void MainWindow::mapChanged()
  270. {
  271. unsaved = true;
  272. setTitle();
  273. }
  274. void MainWindow::initializeMap(bool isNew)
  275. {
  276. unsaved = isNew;
  277. if(isNew)
  278. filename.clear();
  279. setTitle();
  280. mapLevel = 0;
  281. ui->mapView->setScene(controller.scene(mapLevel));
  282. ui->minimapView->setScene(controller.miniScene(mapLevel));
  283. ui->minimapView->dimensions();
  284. if(initialScale.isValid())
  285. on_actionZoom_reset_triggered();
  286. initialScale = ui->mapView->mapToScene(ui->mapView->viewport()->geometry()).boundingRect();
  287. //enable settings
  288. ui->actionMapSettings->setEnabled(true);
  289. ui->actionPlayers_settings->setEnabled(true);
  290. ui->actionTranslations->setEnabled(true);
  291. ui->actionLevel->setEnabled(controller.map()->twoLevel);
  292. //set minimal players count
  293. if(isNew)
  294. {
  295. controller.map()->players[0].canComputerPlay = true;
  296. controller.map()->players[0].canHumanPlay = true;
  297. }
  298. onPlayersChanged();
  299. }
  300. std::unique_ptr<CMap> MainWindow::openMapInternal(const QString & filenameSelect)
  301. {
  302. QFileInfo fi(filenameSelect);
  303. std::string fname = fi.fileName().toStdString();
  304. std::string fdir = fi.dir().path().toStdString();
  305. ResourcePath resId("MAPEDITOR/" + fname, EResType::MAP);
  306. //addFilesystem takes care about memory deallocation if case of failure, no memory leak here
  307. auto * mapEditorFilesystem = new CFilesystemLoader("MAPEDITOR/", fdir, 0);
  308. CResourceHandler::removeFilesystem("local", "mapEditor");
  309. CResourceHandler::addFilesystem("local", "mapEditor", mapEditorFilesystem);
  310. if(!CResourceHandler::get("mapEditor")->existsResource(resId))
  311. throw std::runtime_error("Cannot open map from this folder");
  312. CMapService mapService;
  313. if(auto header = mapService.loadMapHeader(resId))
  314. {
  315. auto missingMods = CMapService::verifyMapHeaderMods(*header);
  316. ModIncompatibility::ModList modList;
  317. for(const auto & m : missingMods)
  318. modList.push_back(m.second.name);
  319. if(!modList.empty())
  320. throw ModIncompatibility(modList);
  321. return mapService.loadMap(resId, nullptr);
  322. }
  323. else
  324. throw std::runtime_error("Corrupted map");
  325. }
  326. bool MainWindow::openMap(const QString & filenameSelect)
  327. {
  328. try
  329. {
  330. controller.setMap(openMapInternal(filenameSelect));
  331. }
  332. catch(const ModIncompatibility & e)
  333. {
  334. assert(e.whatExcessive().empty());
  335. QMessageBox::warning(this, "Mods are required", QString::fromStdString(e.whatMissing()));
  336. return false;
  337. }
  338. catch(const std::exception & e)
  339. {
  340. QMessageBox::critical(this, "Failed to open map", tr(e.what()));
  341. return false;
  342. }
  343. filename = filenameSelect;
  344. initializeMap(controller.map()->version != EMapFormat::VCMI);
  345. return true;
  346. }
  347. void MainWindow::on_actionOpen_triggered()
  348. {
  349. if(!getAnswerAboutUnsavedChanges())
  350. return;
  351. auto filenameSelect = QFileDialog::getOpenFileName(this, tr("Open map"),
  352. QString::fromStdString(VCMIDirs::get().userDataPath().make_preferred().string()),
  353. tr("All supported maps (*.vmap *.h3m);;VCMI maps(*.vmap);;HoMM3 maps(*.h3m)"));
  354. if(filenameSelect.isEmpty())
  355. return;
  356. openMap(filenameSelect);
  357. }
  358. void MainWindow::saveMap()
  359. {
  360. if(!controller.map())
  361. return;
  362. if(!unsaved)
  363. return;
  364. //validate map
  365. auto issues = Validator::validate(controller.map());
  366. bool critical = false;
  367. for(auto & issue : issues)
  368. critical |= issue.critical;
  369. if(!issues.empty())
  370. {
  371. if(critical)
  372. QMessageBox::warning(this, "Map validation", "Map has critical problems and most probably will not be playable. Open Validator from the Map menu to see issues found");
  373. else
  374. QMessageBox::information(this, "Map validation", "Map has some errors. Open Validator from the Map menu to see issues found");
  375. }
  376. Translations::cleanupRemovedItems(*controller.map());
  377. CMapService mapService;
  378. try
  379. {
  380. mapService.saveMap(controller.getMapUniquePtr(), filename.toStdString());
  381. }
  382. catch(const std::exception & e)
  383. {
  384. QMessageBox::critical(this, "Failed to save map", e.what());
  385. return;
  386. }
  387. unsaved = false;
  388. setTitle();
  389. }
  390. void MainWindow::on_actionSave_as_triggered()
  391. {
  392. if(!controller.map())
  393. return;
  394. auto filenameSelect = QFileDialog::getSaveFileName(this, tr("Save map"), lastSavingDir, tr("VCMI maps (*.vmap)"));
  395. if(filenameSelect.isNull())
  396. return;
  397. QFileInfo fileInfo(filenameSelect);
  398. lastSavingDir = fileInfo.dir().path();
  399. if(fileInfo.suffix().toLower() != "vmap")
  400. filenameSelect += ".vmap";
  401. filename = filenameSelect;
  402. saveMap();
  403. }
  404. void MainWindow::on_actionNew_triggered()
  405. {
  406. if(getAnswerAboutUnsavedChanges())
  407. new WindowNewMap(this);
  408. }
  409. void MainWindow::on_actionSave_triggered()
  410. {
  411. if(!controller.map())
  412. return;
  413. if(filename.isNull())
  414. on_actionSave_as_triggered();
  415. else
  416. saveMap();
  417. }
  418. void MainWindow::currentCoordinatesChanged(int x, int y)
  419. {
  420. setStatusMessage(QString("x: %1 y: %2").arg(x).arg(y));
  421. }
  422. void MainWindow::terrainButtonClicked(TerrainId terrain)
  423. {
  424. controller.commitTerrainChange(mapLevel, terrain);
  425. }
  426. void MainWindow::roadOrRiverButtonClicked(ui8 type, bool isRoad)
  427. {
  428. controller.commitRoadOrRiverChange(mapLevel, type, isRoad);
  429. }
  430. void MainWindow::addGroupIntoCatalog(const std::string & groupName, bool staticOnly)
  431. {
  432. auto knownObjects = VLC->objtypeh->knownObjects();
  433. for(auto ID : knownObjects)
  434. {
  435. if(catalog.count(ID))
  436. continue;
  437. addGroupIntoCatalog(groupName, true, staticOnly, ID);
  438. }
  439. }
  440. void MainWindow::addGroupIntoCatalog(const std::string & groupName, bool useCustomName, bool staticOnly, int ID)
  441. {
  442. QStandardItem * itemGroup = nullptr;
  443. auto itms = objectsModel.findItems(QString::fromStdString(groupName));
  444. if(itms.empty())
  445. {
  446. itemGroup = new QStandardItem(QString::fromStdString(groupName));
  447. objectsModel.appendRow(itemGroup);
  448. }
  449. else
  450. {
  451. itemGroup = itms.front();
  452. }
  453. if (VLC->objtypeh->knownObjects().count(ID) == 0)
  454. return;
  455. auto knownSubObjects = VLC->objtypeh->knownSubObjects(ID);
  456. for(auto secondaryID : knownSubObjects)
  457. {
  458. auto factory = VLC->objtypeh->getHandlerFor(ID, secondaryID);
  459. auto templates = factory->getTemplates();
  460. bool singleTemplate = templates.size() == 1;
  461. if(staticOnly && !factory->isStaticObject())
  462. continue;
  463. auto subGroupName = QString::fromStdString(VLC->objtypeh->getObjectName(ID, secondaryID));
  464. auto * itemType = new QStandardItem(subGroupName);
  465. for(int templateId = 0; templateId < templates.size(); ++templateId)
  466. {
  467. auto templ = templates[templateId];
  468. //selecting file
  469. const AnimationPath & afile = templ->editorAnimationFile.empty() ? templ->animationFile : templ->editorAnimationFile;
  470. //creating picture
  471. QPixmap preview(128, 128);
  472. preview.fill(QColor(255, 255, 255));
  473. QPainter painter(&preview);
  474. Animation animation(afile.getOriginalName());
  475. animation.preload();
  476. auto picture = animation.getImage(0);
  477. if(picture && picture->width() && picture->height())
  478. {
  479. qreal xscale = static_cast<qreal>(128) / static_cast<qreal>(picture->width());
  480. qreal yscale = static_cast<qreal>(128) / static_cast<qreal>(picture->height());
  481. qreal scale = std::min(xscale, yscale);
  482. painter.scale(scale, scale);
  483. painter.drawImage(QPoint(0, 0), *picture);
  484. }
  485. //create object to extract name
  486. std::unique_ptr<CGObjectInstance> temporaryObj(factory->create(nullptr, templ));
  487. QString translated = useCustomName ? QString::fromStdString(temporaryObj->getObjectName().c_str()) : subGroupName;
  488. itemType->setText(translated);
  489. //add parameters
  490. QJsonObject data{{"id", QJsonValue(ID)},
  491. {"subid", QJsonValue(secondaryID)},
  492. {"template", QJsonValue(templateId)},
  493. {"animationEditor", QString::fromStdString(templ->editorAnimationFile.getOriginalName())},
  494. {"animation", QString::fromStdString(templ->animationFile.getOriginalName())},
  495. {"preview", jsonFromPixmap(preview)},
  496. {"typeName", QString::fromStdString(factory->getJsonKey())}
  497. };
  498. //do not have extra level
  499. if(singleTemplate)
  500. {
  501. itemType->setIcon(QIcon(preview));
  502. itemType->setData(data);
  503. }
  504. else
  505. {
  506. auto * item = new QStandardItem(QIcon(preview), QString::fromStdString(templ->stringID));
  507. item->setData(data);
  508. itemType->appendRow(item);
  509. }
  510. }
  511. itemGroup->appendRow(itemType);
  512. catalog.insert(ID);
  513. }
  514. }
  515. void MainWindow::loadObjectsTree()
  516. {
  517. try
  518. {
  519. ui->terrainFilterCombo->addItem("");
  520. //adding terrains
  521. for(auto & terrain : VLC->terrainTypeHandler->objects)
  522. {
  523. auto *b = new QPushButton(QString::fromStdString(terrain->getNameTranslated()));
  524. ui->terrainLayout->addWidget(b);
  525. connect(b, &QPushButton::clicked, this, [this, terrainID=terrain->getId()]{ terrainButtonClicked(terrainID); });
  526. //filter
  527. QString displayName = QString::fromStdString(terrain->getNameTranslated());
  528. QString uniqueName = QString::fromStdString(terrain->getJsonKey());
  529. ui->terrainFilterCombo->addItem(displayName, QVariant(uniqueName));
  530. }
  531. //add spacer to keep terrain button on the top
  532. ui->terrainLayout->addItem(new QSpacerItem(20, 20, QSizePolicy::Minimum, QSizePolicy::Expanding));
  533. //adding roads
  534. for(auto & road : VLC->roadTypeHandler->objects)
  535. {
  536. auto *b = new QPushButton(QString::fromStdString(road->getNameTranslated()));
  537. ui->roadLayout->addWidget(b);
  538. connect(b, &QPushButton::clicked, this, [this, roadID=road->getIndex()]{ roadOrRiverButtonClicked(roadID, true); });
  539. }
  540. //add spacer to keep terrain button on the top
  541. ui->roadLayout->addItem(new QSpacerItem(20, 20, QSizePolicy::Minimum, QSizePolicy::Expanding));
  542. //adding rivers
  543. for(auto & river : VLC->riverTypeHandler->objects)
  544. {
  545. auto *b = new QPushButton(QString::fromStdString(river->getNameTranslated()));
  546. ui->riverLayout->addWidget(b);
  547. connect(b, &QPushButton::clicked, this, [this, riverID=river->getIndex()]{ roadOrRiverButtonClicked(riverID, false); });
  548. }
  549. //add spacer to keep terrain button on the top
  550. ui->riverLayout->addItem(new QSpacerItem(20, 20, QSizePolicy::Minimum, QSizePolicy::Expanding));
  551. if(objectBrowser)
  552. throw std::runtime_error("object browser exists");
  553. //model
  554. objectsModel.setHorizontalHeaderLabels(QStringList() << tr("Type"));
  555. objectBrowser = new ObjectBrowserProxyModel(this);
  556. objectBrowser->setSourceModel(&objectsModel);
  557. objectBrowser->setDynamicSortFilter(false);
  558. objectBrowser->setRecursiveFilteringEnabled(true);
  559. ui->treeView->setModel(objectBrowser);
  560. ui->treeView->setSelectionBehavior(QAbstractItemView::SelectItems);
  561. ui->treeView->setSelectionMode(QAbstractItemView::SingleSelection);
  562. connect(ui->treeView->selectionModel(), SIGNAL(currentChanged(const QModelIndex &, const QModelIndex &)), this, SLOT(treeViewSelected(const QModelIndex &, const QModelIndex &)));
  563. //adding objects
  564. addGroupIntoCatalog("TOWNS", false, false, Obj::TOWN);
  565. addGroupIntoCatalog("TOWNS", false, false, Obj::RANDOM_TOWN);
  566. addGroupIntoCatalog("TOWNS", true, false, Obj::SHIPYARD);
  567. addGroupIntoCatalog("TOWNS", true, false, Obj::GARRISON);
  568. addGroupIntoCatalog("TOWNS", true, false, Obj::GARRISON2);
  569. addGroupIntoCatalog("OBJECTS", true, false, Obj::ARENA);
  570. addGroupIntoCatalog("OBJECTS", true, false, Obj::BUOY);
  571. addGroupIntoCatalog("OBJECTS", true, false, Obj::CARTOGRAPHER);
  572. addGroupIntoCatalog("OBJECTS", true, false, Obj::SWAN_POND);
  573. addGroupIntoCatalog("OBJECTS", true, false, Obj::COVER_OF_DARKNESS);
  574. addGroupIntoCatalog("OBJECTS", true, false, Obj::CORPSE);
  575. addGroupIntoCatalog("OBJECTS", true, false, Obj::FAERIE_RING);
  576. addGroupIntoCatalog("OBJECTS", true, false, Obj::FOUNTAIN_OF_FORTUNE);
  577. addGroupIntoCatalog("OBJECTS", true, false, Obj::FOUNTAIN_OF_YOUTH);
  578. addGroupIntoCatalog("OBJECTS", true, false, Obj::GARDEN_OF_REVELATION);
  579. addGroupIntoCatalog("OBJECTS", true, false, Obj::HILL_FORT);
  580. addGroupIntoCatalog("OBJECTS", true, false, Obj::IDOL_OF_FORTUNE);
  581. addGroupIntoCatalog("OBJECTS", true, false, Obj::LIBRARY_OF_ENLIGHTENMENT);
  582. addGroupIntoCatalog("OBJECTS", true, false, Obj::LIGHTHOUSE);
  583. addGroupIntoCatalog("OBJECTS", true, false, Obj::SCHOOL_OF_MAGIC);
  584. addGroupIntoCatalog("OBJECTS", true, false, Obj::MAGIC_SPRING);
  585. addGroupIntoCatalog("OBJECTS", true, false, Obj::MAGIC_WELL);
  586. addGroupIntoCatalog("OBJECTS", true, false, Obj::MERCENARY_CAMP);
  587. addGroupIntoCatalog("OBJECTS", true, false, Obj::MERMAID);
  588. addGroupIntoCatalog("OBJECTS", true, false, Obj::MYSTICAL_GARDEN);
  589. addGroupIntoCatalog("OBJECTS", true, false, Obj::OASIS);
  590. addGroupIntoCatalog("OBJECTS", true, false, Obj::LEAN_TO);
  591. addGroupIntoCatalog("OBJECTS", true, false, Obj::OBELISK);
  592. addGroupIntoCatalog("OBJECTS", true, false, Obj::REDWOOD_OBSERVATORY);
  593. addGroupIntoCatalog("OBJECTS", true, false, Obj::PILLAR_OF_FIRE);
  594. addGroupIntoCatalog("OBJECTS", true, false, Obj::STAR_AXIS);
  595. addGroupIntoCatalog("OBJECTS", true, false, Obj::RALLY_FLAG);
  596. addGroupIntoCatalog("OBJECTS", true, false, Obj::WATERING_HOLE);
  597. addGroupIntoCatalog("OBJECTS", true, false, Obj::SCHOLAR);
  598. addGroupIntoCatalog("OBJECTS", true, false, Obj::SHRINE_OF_MAGIC_INCANTATION);
  599. addGroupIntoCatalog("OBJECTS", true, false, Obj::SHRINE_OF_MAGIC_GESTURE);
  600. addGroupIntoCatalog("OBJECTS", true, false, Obj::SHRINE_OF_MAGIC_THOUGHT);
  601. addGroupIntoCatalog("OBJECTS", true, false, Obj::SIRENS);
  602. addGroupIntoCatalog("OBJECTS", true, false, Obj::STABLES);
  603. addGroupIntoCatalog("OBJECTS", true, false, Obj::TAVERN);
  604. addGroupIntoCatalog("OBJECTS", true, false, Obj::TEMPLE);
  605. addGroupIntoCatalog("OBJECTS", true, false, Obj::DEN_OF_THIEVES);
  606. addGroupIntoCatalog("OBJECTS", true, false, Obj::LEARNING_STONE);
  607. addGroupIntoCatalog("OBJECTS", true, false, Obj::TREE_OF_KNOWLEDGE);
  608. addGroupIntoCatalog("OBJECTS", true, false, Obj::WAGON);
  609. addGroupIntoCatalog("OBJECTS", true, false, Obj::SCHOOL_OF_WAR);
  610. addGroupIntoCatalog("OBJECTS", true, false, Obj::WAR_MACHINE_FACTORY);
  611. addGroupIntoCatalog("OBJECTS", true, false, Obj::WARRIORS_TOMB);
  612. addGroupIntoCatalog("OBJECTS", true, false, Obj::WITCH_HUT);
  613. addGroupIntoCatalog("OBJECTS", true, false, Obj::SANCTUARY);
  614. addGroupIntoCatalog("OBJECTS", true, false, Obj::MARLETTO_TOWER);
  615. addGroupIntoCatalog("HEROES", true, false, Obj::PRISON);
  616. addGroupIntoCatalog("HEROES", false, false, Obj::HERO);
  617. addGroupIntoCatalog("HEROES", false, false, Obj::RANDOM_HERO);
  618. addGroupIntoCatalog("HEROES", false, false, Obj::HERO_PLACEHOLDER);
  619. addGroupIntoCatalog("HEROES", false, false, Obj::BOAT);
  620. addGroupIntoCatalog("ARTIFACTS", true, false, Obj::ARTIFACT);
  621. addGroupIntoCatalog("ARTIFACTS", false, false, Obj::RANDOM_ART);
  622. addGroupIntoCatalog("ARTIFACTS", false, false, Obj::RANDOM_TREASURE_ART);
  623. addGroupIntoCatalog("ARTIFACTS", false, false, Obj::RANDOM_MINOR_ART);
  624. addGroupIntoCatalog("ARTIFACTS", false, false, Obj::RANDOM_MAJOR_ART);
  625. addGroupIntoCatalog("ARTIFACTS", false, false, Obj::RANDOM_RELIC_ART);
  626. addGroupIntoCatalog("ARTIFACTS", true, false, Obj::SPELL_SCROLL);
  627. addGroupIntoCatalog("ARTIFACTS", true, false, Obj::PANDORAS_BOX);
  628. addGroupIntoCatalog("RESOURCES", true, false, Obj::RANDOM_RESOURCE);
  629. addGroupIntoCatalog("RESOURCES", false, false, Obj::RESOURCE);
  630. addGroupIntoCatalog("RESOURCES", true, false, Obj::SEA_CHEST);
  631. addGroupIntoCatalog("RESOURCES", true, false, Obj::TREASURE_CHEST);
  632. addGroupIntoCatalog("RESOURCES", true, false, Obj::CAMPFIRE);
  633. addGroupIntoCatalog("RESOURCES", true, false, Obj::SHIPWRECK_SURVIVOR);
  634. addGroupIntoCatalog("RESOURCES", true, false, Obj::FLOTSAM);
  635. addGroupIntoCatalog("BANKS", true, false, Obj::CREATURE_BANK);
  636. addGroupIntoCatalog("BANKS", true, false, Obj::DRAGON_UTOPIA);
  637. addGroupIntoCatalog("BANKS", true, false, Obj::CRYPT);
  638. addGroupIntoCatalog("BANKS", true, false, Obj::DERELICT_SHIP);
  639. addGroupIntoCatalog("BANKS", true, false, Obj::PYRAMID);
  640. addGroupIntoCatalog("BANKS", true, false, Obj::SHIPWRECK);
  641. addGroupIntoCatalog("DWELLINGS", true, false, Obj::CREATURE_GENERATOR1);
  642. addGroupIntoCatalog("DWELLINGS", true, false, Obj::CREATURE_GENERATOR2);
  643. addGroupIntoCatalog("DWELLINGS", true, false, Obj::CREATURE_GENERATOR3);
  644. addGroupIntoCatalog("DWELLINGS", true, false, Obj::CREATURE_GENERATOR4);
  645. addGroupIntoCatalog("DWELLINGS", true, false, Obj::REFUGEE_CAMP);
  646. addGroupIntoCatalog("DWELLINGS", false, false, Obj::RANDOM_DWELLING);
  647. addGroupIntoCatalog("DWELLINGS", false, false, Obj::RANDOM_DWELLING_LVL);
  648. addGroupIntoCatalog("DWELLINGS", false, false, Obj::RANDOM_DWELLING_FACTION);
  649. addGroupIntoCatalog("GROUNDS", true, false, Obj::CURSED_GROUND1);
  650. addGroupIntoCatalog("GROUNDS", true, false, Obj::MAGIC_PLAINS1);
  651. addGroupIntoCatalog("GROUNDS", true, false, Obj::CLOVER_FIELD);
  652. addGroupIntoCatalog("GROUNDS", true, false, Obj::CURSED_GROUND2);
  653. addGroupIntoCatalog("GROUNDS", true, false, Obj::EVIL_FOG);
  654. addGroupIntoCatalog("GROUNDS", true, false, Obj::FAVORABLE_WINDS);
  655. addGroupIntoCatalog("GROUNDS", true, false, Obj::FIERY_FIELDS);
  656. addGroupIntoCatalog("GROUNDS", true, false, Obj::HOLY_GROUNDS);
  657. addGroupIntoCatalog("GROUNDS", true, false, Obj::LUCID_POOLS);
  658. addGroupIntoCatalog("GROUNDS", true, false, Obj::MAGIC_CLOUDS);
  659. addGroupIntoCatalog("GROUNDS", true, false, Obj::MAGIC_PLAINS2);
  660. addGroupIntoCatalog("GROUNDS", true, false, Obj::ROCKLANDS);
  661. addGroupIntoCatalog("GROUNDS", true, false, Obj::HOLE);
  662. addGroupIntoCatalog("TELEPORTS", true, false, Obj::MONOLITH_ONE_WAY_ENTRANCE);
  663. addGroupIntoCatalog("TELEPORTS", true, false, Obj::MONOLITH_ONE_WAY_EXIT);
  664. addGroupIntoCatalog("TELEPORTS", true, false, Obj::MONOLITH_TWO_WAY);
  665. addGroupIntoCatalog("TELEPORTS", true, false, Obj::SUBTERRANEAN_GATE);
  666. addGroupIntoCatalog("TELEPORTS", true, false, Obj::WHIRLPOOL);
  667. addGroupIntoCatalog("MINES", true, false, Obj::MINE);
  668. addGroupIntoCatalog("MINES", false, false, Obj::ABANDONED_MINE);
  669. addGroupIntoCatalog("MINES", true, false, Obj::WINDMILL);
  670. addGroupIntoCatalog("MINES", true, false, Obj::WATER_WHEEL);
  671. addGroupIntoCatalog("TRIGGERS", true, false, Obj::EVENT);
  672. addGroupIntoCatalog("TRIGGERS", true, false, Obj::GRAIL);
  673. addGroupIntoCatalog("TRIGGERS", true, false, Obj::SIGN);
  674. addGroupIntoCatalog("TRIGGERS", true, false, Obj::OCEAN_BOTTLE);
  675. addGroupIntoCatalog("MONSTERS", false, false, Obj::MONSTER);
  676. addGroupIntoCatalog("MONSTERS", true, false, Obj::RANDOM_MONSTER);
  677. addGroupIntoCatalog("MONSTERS", true, false, Obj::RANDOM_MONSTER_L1);
  678. addGroupIntoCatalog("MONSTERS", true, false, Obj::RANDOM_MONSTER_L2);
  679. addGroupIntoCatalog("MONSTERS", true, false, Obj::RANDOM_MONSTER_L3);
  680. addGroupIntoCatalog("MONSTERS", true, false, Obj::RANDOM_MONSTER_L4);
  681. addGroupIntoCatalog("MONSTERS", true, false, Obj::RANDOM_MONSTER_L5);
  682. addGroupIntoCatalog("MONSTERS", true, false, Obj::RANDOM_MONSTER_L6);
  683. addGroupIntoCatalog("MONSTERS", true, false, Obj::RANDOM_MONSTER_L7);
  684. addGroupIntoCatalog("QUESTS", true, false, Obj::SEER_HUT);
  685. addGroupIntoCatalog("QUESTS", true, false, Obj::BORDER_GATE);
  686. addGroupIntoCatalog("QUESTS", true, false, Obj::QUEST_GUARD);
  687. addGroupIntoCatalog("QUESTS", true, false, Obj::HUT_OF_MAGI);
  688. addGroupIntoCatalog("QUESTS", true, false, Obj::EYE_OF_MAGI);
  689. addGroupIntoCatalog("QUESTS", true, false, Obj::BORDERGUARD);
  690. addGroupIntoCatalog("QUESTS", true, false, Obj::KEYMASTER);
  691. addGroupIntoCatalog("wog object", true, false, Obj::WOG_OBJECT);
  692. addGroupIntoCatalog("OBSTACLES", true);
  693. addGroupIntoCatalog("OTHER", false);
  694. }
  695. catch(const std::exception &)
  696. {
  697. QMessageBox::critical(this, "Mods loading problem", "Critical error during Mods loading. Disable invalid mods and restart.");
  698. }
  699. }
  700. void MainWindow::on_actionLevel_triggered()
  701. {
  702. if(controller.map() && controller.map()->twoLevel)
  703. {
  704. mapLevel = mapLevel ? 0 : 1;
  705. ui->mapView->setScene(controller.scene(mapLevel));
  706. ui->minimapView->setScene(controller.miniScene(mapLevel));
  707. if (mapLevel == 0)
  708. {
  709. ui->actionLevel->setToolTip(tr("View underground"));
  710. }
  711. else
  712. {
  713. ui->actionLevel->setToolTip(tr("View surface"));
  714. }
  715. }
  716. }
  717. void MainWindow::on_actionUndo_triggered()
  718. {
  719. QString str("Undo clicked");
  720. statusBar()->showMessage(str, 1000);
  721. if (controller.map())
  722. {
  723. controller.undo();
  724. }
  725. }
  726. void MainWindow::on_actionRedo_triggered()
  727. {
  728. QString str("Redo clicked");
  729. displayStatus(str);
  730. if (controller.map())
  731. {
  732. controller.redo();
  733. }
  734. }
  735. void MainWindow::on_actionPass_triggered(bool checked)
  736. {
  737. QString str("Passability clicked");
  738. displayStatus(str);
  739. if(controller.map())
  740. {
  741. controller.scene(0)->passabilityView.show(checked);
  742. controller.scene(1)->passabilityView.show(checked);
  743. }
  744. }
  745. void MainWindow::on_actionGrid_triggered(bool checked)
  746. {
  747. QString str("Grid clicked");
  748. displayStatus(str);
  749. if(controller.map())
  750. {
  751. controller.scene(0)->gridView.show(checked);
  752. controller.scene(1)->gridView.show(checked);
  753. }
  754. }
  755. void MainWindow::changeBrushState(int idx)
  756. {
  757. }
  758. void MainWindow::on_actionErase_triggered()
  759. {
  760. if(controller.map())
  761. {
  762. controller.commitObjectErase(mapLevel);
  763. }
  764. }
  765. void MainWindow::preparePreview(const QModelIndex &index)
  766. {
  767. scenePreview->clear();
  768. auto data = objectsModel.itemFromIndex(objectBrowser->mapToSource(index))->data().toJsonObject();
  769. if(!data.empty())
  770. {
  771. auto preview = data["preview"];
  772. if(preview != QJsonValue::Undefined)
  773. {
  774. QPixmap objPreview = pixmapFromJson(preview);
  775. scenePreview->addPixmap(objPreview);
  776. }
  777. }
  778. ui->objectPreview->fitInView(scenePreview->itemsBoundingRect(), Qt::KeepAspectRatio);
  779. }
  780. void MainWindow::treeViewSelected(const QModelIndex & index, const QModelIndex & deselected)
  781. {
  782. ui->toolSelect->setChecked(true);
  783. ui->mapView->selectionTool = MapView::SelectionTool::None;
  784. preparePreview(index);
  785. }
  786. void MainWindow::on_terrainFilterCombo_currentIndexChanged(int index)
  787. {
  788. if(!objectBrowser)
  789. return;
  790. QString uniqueName = ui->terrainFilterCombo->itemData(index).toString();
  791. objectBrowser->terrain = TerrainId(ETerrainId::ANY_TERRAIN);
  792. if (!uniqueName.isEmpty())
  793. {
  794. for (auto const & terrain : VLC->terrainTypeHandler->objects)
  795. if (terrain->getJsonKey() == uniqueName.toStdString())
  796. objectBrowser->terrain = terrain->getId();
  797. }
  798. objectBrowser->invalidate();
  799. objectBrowser->sort(0);
  800. }
  801. void MainWindow::on_filter_textChanged(const QString &arg1)
  802. {
  803. if(!objectBrowser)
  804. return;
  805. objectBrowser->filter = arg1;
  806. objectBrowser->invalidate();
  807. objectBrowser->sort(0);
  808. }
  809. void MainWindow::on_actionFill_triggered()
  810. {
  811. QString str("Fill clicked");
  812. displayStatus(str);
  813. if(!controller.map())
  814. return;
  815. controller.commitObstacleFill(mapLevel);
  816. }
  817. void MainWindow::loadInspector(CGObjectInstance * obj, bool switchTab)
  818. {
  819. if(switchTab)
  820. ui->tabWidget->setCurrentIndex(1);
  821. Inspector inspector(controller, obj, ui->inspectorWidget);
  822. inspector.updateProperties();
  823. }
  824. void MainWindow::on_inspectorWidget_itemChanged(QTableWidgetItem *item)
  825. {
  826. if(!item->isSelected() && !(item->flags() & Qt::ItemIsUserCheckable))
  827. return;
  828. int r = item->row();
  829. int c = item->column();
  830. if(c < 1)
  831. return;
  832. auto * tableWidget = item->tableWidget();
  833. //get identifier
  834. auto identifier = tableWidget->item(0, 1)->text();
  835. CGObjectInstance * obj = data_cast<CGObjectInstance>(identifier.toLongLong());
  836. //get parameter name
  837. auto param = tableWidget->item(r, c - 1)->text();
  838. //set parameter
  839. Inspector inspector(controller, obj, tableWidget);
  840. inspector.setProperty(param, item);
  841. controller.commitObjectChange(mapLevel);
  842. }
  843. void MainWindow::on_actionMapSettings_triggered()
  844. {
  845. auto settingsDialog = new MapSettings(controller, this);
  846. settingsDialog->setWindowModality(Qt::WindowModal);
  847. settingsDialog->setModal(true);
  848. }
  849. void MainWindow::on_actionPlayers_settings_triggered()
  850. {
  851. auto settingsDialog = new PlayerSettings(controller, this);
  852. settingsDialog->setWindowModality(Qt::WindowModal);
  853. settingsDialog->setModal(true);
  854. connect(settingsDialog, &QDialog::finished, this, &MainWindow::onPlayersChanged);
  855. }
  856. QAction * MainWindow::getActionPlayer(const PlayerColor & player)
  857. {
  858. if(player.getNum() == 0) return ui->actionPlayer_1;
  859. if(player.getNum() == 1) return ui->actionPlayer_2;
  860. if(player.getNum() == 2) return ui->actionPlayer_3;
  861. if(player.getNum() == 3) return ui->actionPlayer_4;
  862. if(player.getNum() == 4) return ui->actionPlayer_5;
  863. if(player.getNum() == 5) return ui->actionPlayer_6;
  864. if(player.getNum() == 6) return ui->actionPlayer_7;
  865. if(player.getNum() == 7) return ui->actionPlayer_8;
  866. return ui->actionNeutral;
  867. }
  868. void MainWindow::switchDefaultPlayer(const PlayerColor & player)
  869. {
  870. if(controller.defaultPlayer == player)
  871. return;
  872. ui->actionNeutral->blockSignals(true);
  873. ui->actionNeutral->setChecked(PlayerColor::NEUTRAL == player);
  874. ui->actionNeutral->blockSignals(false);
  875. for(int i = 0; i < PlayerColor::PLAYER_LIMIT.getNum(); ++i)
  876. {
  877. getActionPlayer(PlayerColor(i))->blockSignals(true);
  878. getActionPlayer(PlayerColor(i))->setChecked(PlayerColor(i) == player);
  879. getActionPlayer(PlayerColor(i))->blockSignals(false);
  880. }
  881. controller.defaultPlayer = player;
  882. }
  883. void MainWindow::onPlayersChanged()
  884. {
  885. if(controller.map())
  886. {
  887. getActionPlayer(PlayerColor::NEUTRAL)->setEnabled(true);
  888. for(int i = 0; i < controller.map()->players.size(); ++i)
  889. getActionPlayer(PlayerColor(i))->setEnabled(controller.map()->players.at(i).canAnyonePlay());
  890. if(!getActionPlayer(controller.defaultPlayer)->isEnabled() || controller.defaultPlayer == PlayerColor::NEUTRAL)
  891. switchDefaultPlayer(PlayerColor::NEUTRAL);
  892. }
  893. else
  894. {
  895. for(int i = 0; i < PlayerColor::PLAYER_LIMIT.getNum(); ++i)
  896. getActionPlayer(PlayerColor(i))->setEnabled(false);
  897. getActionPlayer(PlayerColor::NEUTRAL)->setEnabled(false);
  898. }
  899. }
  900. void MainWindow::enableUndo(bool enable)
  901. {
  902. ui->actionUndo->setEnabled(enable);
  903. }
  904. void MainWindow::enableRedo(bool enable)
  905. {
  906. ui->actionRedo->setEnabled(enable);
  907. }
  908. void MainWindow::onSelectionMade(int level, bool anythingSelected)
  909. {
  910. if (level == mapLevel)
  911. {
  912. ui->actionErase->setEnabled(anythingSelected);
  913. }
  914. }
  915. void MainWindow::displayStatus(const QString& message, int timeout /* = 2000 */)
  916. {
  917. statusBar()->showMessage(message, timeout);
  918. }
  919. void MainWindow::on_actionValidate_triggered()
  920. {
  921. new Validator(controller.map(), this);
  922. }
  923. void MainWindow::on_actionUpdate_appearance_triggered()
  924. {
  925. if(!controller.map())
  926. return;
  927. if(controller.scene(mapLevel)->selectionObjectsView.getSelection().empty())
  928. {
  929. QMessageBox::information(this, tr("Update appearance"), tr("No objects selected"));
  930. return;
  931. }
  932. if(QMessageBox::Yes != QMessageBox::question(this, tr("Update appearance"), tr("This operation is irreversible. Do you want to continue?")))
  933. return;
  934. controller.scene(mapLevel)->selectionTerrainView.clear();
  935. int errors = 0;
  936. std::set<CGObjectInstance*> staticObjects;
  937. for(auto * obj : controller.scene(mapLevel)->selectionObjectsView.getSelection())
  938. {
  939. auto handler = VLC->objtypeh->getHandlerFor(obj->ID, obj->subID);
  940. if(!controller.map()->isInTheMap(obj->visitablePos()))
  941. {
  942. ++errors;
  943. continue;
  944. }
  945. auto * terrain = controller.map()->getTile(obj->visitablePos()).terType;
  946. if(handler->isStaticObject())
  947. {
  948. staticObjects.insert(obj);
  949. if(obj->appearance->canBePlacedAt(terrain->getId()))
  950. {
  951. controller.scene(mapLevel)->selectionObjectsView.deselectObject(obj);
  952. continue;
  953. }
  954. for(auto & offset : obj->appearance->getBlockedOffsets())
  955. controller.scene(mapLevel)->selectionTerrainView.select(obj->pos + offset);
  956. }
  957. else
  958. {
  959. auto app = handler->getOverride(terrain->getId(), obj);
  960. if(!app)
  961. {
  962. if(obj->appearance->canBePlacedAt(terrain->getId()))
  963. continue;
  964. auto templates = handler->getTemplates(terrain->getId());
  965. if(templates.empty())
  966. {
  967. ++errors;
  968. continue;
  969. }
  970. app = templates.front();
  971. }
  972. obj->appearance = app;
  973. controller.mapHandler()->invalidate(obj);
  974. controller.scene(mapLevel)->selectionObjectsView.deselectObject(obj);
  975. }
  976. }
  977. controller.commitObjectChange(mapLevel);
  978. controller.commitObjectErase(mapLevel);
  979. controller.commitObstacleFill(mapLevel);
  980. if(errors)
  981. QMessageBox::warning(this, tr("Update appearance"), QString(tr("Errors occurred. %1 objects were not updated")).arg(errors));
  982. }
  983. void MainWindow::on_actionRecreate_obstacles_triggered()
  984. {
  985. }
  986. void MainWindow::on_actionCut_triggered()
  987. {
  988. if(controller.map())
  989. {
  990. controller.copyToClipboard(mapLevel);
  991. controller.commitObjectErase(mapLevel);
  992. }
  993. }
  994. void MainWindow::on_actionCopy_triggered()
  995. {
  996. if(controller.map())
  997. {
  998. controller.copyToClipboard(mapLevel);
  999. }
  1000. }
  1001. void MainWindow::on_actionPaste_triggered()
  1002. {
  1003. if(controller.map())
  1004. {
  1005. controller.pasteFromClipboard(mapLevel);
  1006. }
  1007. }
  1008. void MainWindow::on_actionExport_triggered()
  1009. {
  1010. QString fileName = QFileDialog::getSaveFileName(this, tr("Save to image"), lastSavingDir, "BMP (*.bmp);;JPEG (*.jpeg);;PNG (*.png)");
  1011. if(!fileName.isNull())
  1012. {
  1013. QImage image(ui->mapView->scene()->sceneRect().size().toSize(), QImage::Format_RGB888);
  1014. QPainter painter(&image);
  1015. ui->mapView->scene()->render(&painter);
  1016. image.save(fileName);
  1017. }
  1018. }
  1019. void MainWindow::on_actionTranslations_triggered()
  1020. {
  1021. auto translationsDialog = new Translations(*controller.map(), this);
  1022. translationsDialog->show();
  1023. }
  1024. void MainWindow::on_actionh3m_converter_triggered()
  1025. {
  1026. auto mapFiles = QFileDialog::getOpenFileNames(this, tr("Select maps to convert"),
  1027. QString::fromStdString(VCMIDirs::get().userDataPath().make_preferred().string()),
  1028. tr("HoMM3 maps(*.h3m)"));
  1029. if(mapFiles.empty())
  1030. return;
  1031. auto saveDirectory = QFileDialog::getExistingDirectory(this, tr("Choose directory to save converted maps"), QCoreApplication::applicationDirPath());
  1032. if(saveDirectory.isEmpty())
  1033. return;
  1034. try
  1035. {
  1036. for(auto & m : mapFiles)
  1037. {
  1038. CMapService mapService;
  1039. auto map = openMapInternal(m);
  1040. controller.repairMap(map.get());
  1041. mapService.saveMap(map, (saveDirectory + '/' + QFileInfo(m).completeBaseName() + ".vmap").toStdString());
  1042. }
  1043. QMessageBox::information(this, tr("Operation completed"), tr("Successfully converted %1 maps").arg(mapFiles.size()));
  1044. }
  1045. catch(const std::exception & e)
  1046. {
  1047. QMessageBox::critical(this, tr("Failed to convert the map. Abort operation"), tr(e.what()));
  1048. }
  1049. }
  1050. void MainWindow::on_actionLock_triggered()
  1051. {
  1052. if(controller.map())
  1053. {
  1054. if(controller.scene(mapLevel)->selectionObjectsView.getSelection().empty())
  1055. {
  1056. for(auto obj : controller.map()->objects)
  1057. {
  1058. controller.scene(mapLevel)->selectionObjectsView.setLockObject(obj, true);
  1059. controller.scene(mapLevel)->objectsView.setLockObject(obj, true);
  1060. }
  1061. }
  1062. else
  1063. {
  1064. for(auto * obj : controller.scene(mapLevel)->selectionObjectsView.getSelection())
  1065. {
  1066. controller.scene(mapLevel)->selectionObjectsView.setLockObject(obj, true);
  1067. controller.scene(mapLevel)->objectsView.setLockObject(obj, true);
  1068. }
  1069. controller.scene(mapLevel)->selectionObjectsView.clear();
  1070. }
  1071. controller.scene(mapLevel)->objectsView.update();
  1072. controller.scene(mapLevel)->selectionObjectsView.update();
  1073. }
  1074. }
  1075. void MainWindow::on_actionUnlock_triggered()
  1076. {
  1077. if(controller.map())
  1078. {
  1079. controller.scene(mapLevel)->selectionObjectsView.unlockAll();
  1080. controller.scene(mapLevel)->objectsView.unlockAll();
  1081. }
  1082. controller.scene(mapLevel)->objectsView.update();
  1083. }
  1084. void MainWindow::on_actionZoom_in_triggered()
  1085. {
  1086. auto rect = ui->mapView->mapToScene(ui->mapView->viewport()->geometry()).boundingRect();
  1087. rect -= QMargins{32 + 1, 32 + 1, 32 + 2, 32 + 2}; //compensate bounding box
  1088. ui->mapView->fitInView(rect, Qt::KeepAspectRatioByExpanding);
  1089. }
  1090. void MainWindow::on_actionZoom_out_triggered()
  1091. {
  1092. auto rect = ui->mapView->mapToScene(ui->mapView->viewport()->geometry()).boundingRect();
  1093. rect += QMargins{32 - 1, 32 - 1, 32 - 2, 32 - 2}; //compensate bounding box
  1094. ui->mapView->fitInView(rect, Qt::KeepAspectRatioByExpanding);
  1095. }
  1096. void MainWindow::on_actionZoom_reset_triggered()
  1097. {
  1098. auto center = ui->mapView->mapToScene(ui->mapView->viewport()->geometry().center());
  1099. ui->mapView->fitInView(initialScale, Qt::KeepAspectRatioByExpanding);
  1100. ui->mapView->centerOn(center);
  1101. }
  1102. void MainWindow::on_toolLine_toggled(bool checked)
  1103. {
  1104. if(checked)
  1105. {
  1106. ui->mapView->selectionTool = MapView::SelectionTool::Line;
  1107. ui->tabWidget->setCurrentIndex(0);
  1108. }
  1109. }
  1110. void MainWindow::on_toolBrush2_toggled(bool checked)
  1111. {
  1112. if(checked)
  1113. {
  1114. ui->mapView->selectionTool = MapView::SelectionTool::Brush2;
  1115. ui->tabWidget->setCurrentIndex(0);
  1116. }
  1117. }
  1118. void MainWindow::on_toolBrush_toggled(bool checked)
  1119. {
  1120. if(checked)
  1121. {
  1122. ui->mapView->selectionTool = MapView::SelectionTool::Brush;
  1123. ui->tabWidget->setCurrentIndex(0);
  1124. }
  1125. }
  1126. void MainWindow::on_toolBrush4_toggled(bool checked)
  1127. {
  1128. if(checked)
  1129. {
  1130. ui->mapView->selectionTool = MapView::SelectionTool::Brush4;
  1131. ui->tabWidget->setCurrentIndex(0);
  1132. }
  1133. }
  1134. void MainWindow::on_toolLasso_toggled(bool checked)
  1135. {
  1136. if(checked)
  1137. {
  1138. ui->mapView->selectionTool = MapView::SelectionTool::Lasso;
  1139. ui->tabWidget->setCurrentIndex(0);
  1140. }
  1141. }
  1142. void MainWindow::on_toolArea_toggled(bool checked)
  1143. {
  1144. if(checked)
  1145. {
  1146. ui->mapView->selectionTool = MapView::SelectionTool::Area;
  1147. ui->tabWidget->setCurrentIndex(0);
  1148. }
  1149. }
  1150. void MainWindow::on_toolFill_toggled(bool checked)
  1151. {
  1152. if(checked)
  1153. {
  1154. ui->mapView->selectionTool = MapView::SelectionTool::Fill;
  1155. ui->tabWidget->setCurrentIndex(0);
  1156. }
  1157. }
  1158. void MainWindow::on_toolSelect_toggled(bool checked)
  1159. {
  1160. if(checked)
  1161. {
  1162. ui->mapView->selectionTool = MapView::SelectionTool::None;
  1163. ui->tabWidget->setCurrentIndex(0);
  1164. }
  1165. }