88 lines
2.3 KiB
C++
88 lines
2.3 KiB
C++
#pragma once
|
|
|
|
#include <map>
|
|
#include <optional>
|
|
#include <set>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include <QDialog>
|
|
#include <QPoint>
|
|
|
|
#include "GameConfig.h"
|
|
#include "Rotation.h"
|
|
#include "ShipLayout.h"
|
|
#include "ShipLayoutBlueprint.h"
|
|
|
|
class QPushButton;
|
|
class ShipStatsPanel;
|
|
|
|
class ShipLayoutDialog : public QDialog
|
|
{
|
|
Q_OBJECT
|
|
|
|
public:
|
|
ShipLayoutDialog(const GameConfig* config,
|
|
const std::string& shipId,
|
|
const ShipLayoutConfig& currentLayout,
|
|
std::vector<ShipLayoutBlueprint>& allBlueprints,
|
|
std::set<std::string> unlockedModuleIds,
|
|
bool debugDraw,
|
|
QWidget* parent = nullptr);
|
|
|
|
std::optional<ShipLayoutConfig> getResult() const;
|
|
|
|
protected:
|
|
void keyPressEvent(QKeyEvent* event) override;
|
|
|
|
signals:
|
|
void gridCellClicked(QPoint cell);
|
|
|
|
private slots:
|
|
void onModuleButtonClicked(int index);
|
|
void onConfirm();
|
|
void onCancel();
|
|
|
|
public:
|
|
struct CellInfo
|
|
{
|
|
bool buildable;
|
|
std::optional<int> moduleIndex; // nullopt if empty
|
|
};
|
|
|
|
private:
|
|
|
|
void rebuildOccupancy();
|
|
void updateGridWidget();
|
|
void updateStats();
|
|
bool canPlaceModule(const ModuleDef& def, QPoint position, Rotation rotation) const;
|
|
std::vector<std::string> rotatedMask(const ModuleDef& def, Rotation rotation) const;
|
|
void loadLayoutBlueprint(const std::vector<PlacedModule>& modules);
|
|
|
|
const GameConfig* m_config;
|
|
std::string m_shipId;
|
|
std::set<std::string> m_unlockedModuleIds;
|
|
std::vector<std::string> m_shipLayout;
|
|
int m_rows;
|
|
int m_cols;
|
|
|
|
std::vector<PlacedModule> m_placedModules;
|
|
std::vector<std::vector<CellInfo>> m_grid;
|
|
|
|
// The module to place, as an index into config modules; nullopt when no
|
|
// module is selected for placement. m_removeMode is a separate mode in which
|
|
// clicking a cell removes the module there (mutually exclusive with a
|
|
// selected module).
|
|
std::optional<int> m_activeModuleIndex;
|
|
bool m_removeMode = false;
|
|
Rotation m_currentRotation;
|
|
|
|
std::vector<QPushButton*> m_moduleButtons;
|
|
QPushButton* m_removeButton;
|
|
QWidget* m_gridWidget;
|
|
ShipStatsPanel* m_statsPanel;
|
|
bool m_debugDraw;
|
|
|
|
std::optional<ShipLayoutConfig> m_result;
|
|
};
|