diff --git a/GUI/EvtMainFrame.cpp b/GUI/EvtMainFrame.cpp
index e6ac372..9338126 100644
--- a/GUI/EvtMainFrame.cpp
+++ b/GUI/EvtMainFrame.cpp
@@ -210,6 +210,30 @@ void EvtMainFrame::companyRenameEvent(wxCommandEvent&) {
}
}
+void EvtMainFrame::creditsEditEvent(wxCommandEvent&) {
+ const static std::string error_prefix = "Credits change failed:\n\n";
+
+ if(_unsafeMode == true || _mbManager.gameState() == GameState::NotRunning) {
+ long number = wxGetNumberFromUser("Please enter a number of credits between 0 and 2 000 000 000 included:", "Credits:", "Input credits",
+ _profileManager.currentProfile()->credits(), 0, 2000000000, this);
+ if(number == -1 || number == _profileManager.currentProfile()->credits()) {
+ return;
+ }
+
+ if(!_profileManager.currentProfile()->setCredits(number)) {
+ errorMessage(error_prefix + _profileManager.currentProfile()->lastError());
+ }
+
+ updateProfileStats();
+ }
+ else if(_mbManager.gameState() == GameState::Unknown) {
+ errorMessage(error_prefix + "For security reasons, changing credits is disabled if the game's status is unknown.");
+ }
+ else if(_mbManager.gameState() == GameState::Running) {
+ errorMessage(error_prefix + "For security reasons, changing credits is disabled if the game is running.");
+ }
+}
+
void EvtMainFrame::storyProgressSelectionEvent(wxCommandEvent& event) {
const static std::string error_prefix = "StoryProgress change failed:\n\n";
@@ -787,6 +811,7 @@ void EvtMainFrame::updateCommandsState() {
MassState mass_state = _massManager->massState(selection);
_companyRenameButton->Enable(_unsafeMode == true || game_state == GameState::NotRunning);
+ _creditsEditButton->Enable(_unsafeMode == true || game_state == GameState::NotRunning);
_storyProgressChangeButton->Enable(_unsafeMode == true || game_state == GameState::NotRunning);
wxPropertyGridConstIterator it = _researchInventoryPropGrid->GetIterator(wxPG_ITERATE_NORMAL);
diff --git a/GUI/EvtMainFrame.h b/GUI/EvtMainFrame.h
index 769c1af..bdbdbe5 100644
--- a/GUI/EvtMainFrame.h
+++ b/GUI/EvtMainFrame.h
@@ -42,33 +42,34 @@ class EvtMainFrame: public MainFrame {
protected:
// Profile-related events
- void profileSelectionEvent(wxCommandEvent&);
- void backupSelectedProfileEvent(wxCommandEvent&);
- void companyRenameEvent(wxCommandEvent&);
+ void profileSelectionEvent(wxCommandEvent&) override;
+ void backupSelectedProfileEvent(wxCommandEvent&) override;
+ void companyRenameEvent(wxCommandEvent&) override;
+ void creditsEditEvent(wxCommandEvent&) override;
void storyProgressSelectionEvent(wxCommandEvent& event);
- void openStoryProgressMenuEvent(wxCommandEvent&);
- void inventoryChangeEvent(wxPropertyGridEvent& event);
+ void openStoryProgressMenuEvent(wxCommandEvent&) override;
+ void inventoryChangeEvent(wxPropertyGridEvent& event) override;
// M.A.S.S.-related events
- void importMassEvent(wxCommandEvent&);
- void exportMassEvent(wxCommandEvent&);
- void moveMassEvent(wxCommandEvent&);
- void deleteMassEvent(wxCommandEvent&);
- void renameMassEvent(wxCommandEvent&);
- void openSaveDirEvent(wxCommandEvent&);
- void stagingSelectionEvent(wxCommandEvent&);
- void deleteStagedEvent(wxCommandEvent&);
- void openStagingDirEvent(wxCommandEvent&);
+ void importMassEvent(wxCommandEvent&) override;
+ void exportMassEvent(wxCommandEvent&) override;
+ void moveMassEvent(wxCommandEvent&) override;
+ void deleteMassEvent(wxCommandEvent&) override;
+ void renameMassEvent(wxCommandEvent&) override;
+ void openSaveDirEvent(wxCommandEvent&) override;
+ void stagingSelectionEvent(wxCommandEvent&) override;
+ void deleteStagedEvent(wxCommandEvent&) override;
+ void openStagingDirEvent(wxCommandEvent&) override;
void installedSelectionEvent(wxListEvent&);
void listColumnDragEvent(wxListEvent&);
// Screenshot-related events
- void openScreenshotDirEvent(wxCommandEvent&);
+ void openScreenshotDirEvent(wxCommandEvent&) override;
// General events
void fileUpdateEvent(wxFileSystemWatcherEvent& event);
- void gameCheckTimerEvent(wxTimerEvent&);
- void unsafeCheckboxEvent(wxCommandEvent& event);
+ void gameCheckTimerEvent(wxTimerEvent&) override;
+ void unsafeCheckboxEvent(wxCommandEvent& event) override;
private:
void saveFileEventHandler(int event_type, const wxString& event_file, const wxFileSystemWatcherEvent& event);
diff --git a/GUI/MainFrame.cpp b/GUI/MainFrame.cpp
index 7958b73..0250bc9 100644
--- a/GUI/MainFrame.cpp
+++ b/GUI/MainFrame.cpp
@@ -114,6 +114,9 @@ MainFrame::MainFrame( wxWindow* parent, wxWindowID id, const wxString& title, co
_companyRenameButton = new wxButton( sbSizerGeneralInfo->GetStaticBox(), wxID_ANY, wxT("Rename company"), wxDefaultPosition, wxDefaultSize, 0 );
bSizerProfileCommands->Add( _companyRenameButton, 0, wxALL, 5 );
+ _creditsEditButton = new wxButton( sbSizerGeneralInfo->GetStaticBox(), wxID_ANY, wxT("Edit credits"), wxDefaultPosition, wxDefaultSize, 0 );
+ bSizerProfileCommands->Add( _creditsEditButton, 0, wxALL, 5 );
+
_storyProgressChangeButton = new wxButton( sbSizerGeneralInfo->GetStaticBox(), wxID_ANY, wxT("Change story progress"), wxDefaultPosition, wxDefaultSize, 0 );
bSizerProfileCommands->Add( _storyProgressChangeButton, 0, wxALL, 5 );
@@ -298,6 +301,7 @@ MainFrame::MainFrame( wxWindow* parent, wxWindowID id, const wxString& title, co
_openScreenshotDirButton->Connect( wxEVT_COMMAND_BUTTON_CLICKED, wxCommandEventHandler( MainFrame::openScreenshotDirEvent ), NULL, this );
_unsafeCheckbox->Connect( wxEVT_COMMAND_CHECKBOX_CLICKED, wxCommandEventHandler( MainFrame::unsafeCheckboxEvent ), NULL, this );
_companyRenameButton->Connect( wxEVT_COMMAND_BUTTON_CLICKED, wxCommandEventHandler( MainFrame::companyRenameEvent ), NULL, this );
+ _creditsEditButton->Connect( wxEVT_COMMAND_BUTTON_CLICKED, wxCommandEventHandler( MainFrame::creditsEditEvent ), NULL, this );
_storyProgressChangeButton->Connect( wxEVT_COMMAND_BUTTON_CLICKED, wxCommandEventHandler( MainFrame::openStoryProgressMenuEvent ), NULL, this );
_researchInventoryPropGrid->Connect( wxEVT_PG_CHANGING, wxPropertyGridEventHandler( MainFrame::inventoryChangeEvent ), NULL, this );
_moveButton->Connect( wxEVT_COMMAND_BUTTON_CLICKED, wxCommandEventHandler( MainFrame::moveMassEvent ), NULL, this );
@@ -320,6 +324,7 @@ MainFrame::~MainFrame()
_openScreenshotDirButton->Disconnect( wxEVT_COMMAND_BUTTON_CLICKED, wxCommandEventHandler( MainFrame::openScreenshotDirEvent ), NULL, this );
_unsafeCheckbox->Disconnect( wxEVT_COMMAND_CHECKBOX_CLICKED, wxCommandEventHandler( MainFrame::unsafeCheckboxEvent ), NULL, this );
_companyRenameButton->Disconnect( wxEVT_COMMAND_BUTTON_CLICKED, wxCommandEventHandler( MainFrame::companyRenameEvent ), NULL, this );
+ _creditsEditButton->Disconnect( wxEVT_COMMAND_BUTTON_CLICKED, wxCommandEventHandler( MainFrame::creditsEditEvent ), NULL, this );
_storyProgressChangeButton->Disconnect( wxEVT_COMMAND_BUTTON_CLICKED, wxCommandEventHandler( MainFrame::openStoryProgressMenuEvent ), NULL, this );
_researchInventoryPropGrid->Disconnect( wxEVT_PG_CHANGING, wxPropertyGridEventHandler( MainFrame::inventoryChangeEvent ), NULL, this );
_moveButton->Disconnect( wxEVT_COMMAND_BUTTON_CLICKED, wxCommandEventHandler( MainFrame::moveMassEvent ), NULL, this );
diff --git a/GUI/MainFrame.fbp b/GUI/MainFrame.fbp
index 4a5b187..fc6d175 100644
--- a/GUI/MainFrame.fbp
+++ b/GUI/MainFrame.fbp
@@ -1194,6 +1194,79 @@
companyRenameEvent
+
5
wxALL
diff --git a/GUI/MainFrame.h b/GUI/MainFrame.h
index f3791f8..8f2aaf6 100644
--- a/GUI/MainFrame.h
+++ b/GUI/MainFrame.h
@@ -61,6 +61,7 @@ class MainFrame : public wxFrame
wxStaticText* _lastMissionIdLabel;
wxStaticText* _lastMissionId;
wxButton* _companyRenameButton;
+ wxButton* _creditsEditButton;
wxButton* _storyProgressChangeButton;
wxPropertyGrid* _researchInventoryPropGrid;
wxPGProperty* _materialsCategory;
@@ -109,6 +110,7 @@ class MainFrame : public wxFrame
virtual void openScreenshotDirEvent( wxCommandEvent& event ) { event.Skip(); }
virtual void unsafeCheckboxEvent( wxCommandEvent& event ) { event.Skip(); }
virtual void companyRenameEvent( wxCommandEvent& event ) { event.Skip(); }
+ virtual void creditsEditEvent( wxCommandEvent& event ) { event.Skip(); }
virtual void openStoryProgressMenuEvent( wxCommandEvent& event ) { event.Skip(); }
virtual void inventoryChangeEvent( wxPropertyGridEvent& event ) { event.Skip(); }
virtual void moveMassEvent( wxCommandEvent& event ) { event.Skip(); }
diff --git a/Profile/Profile.cpp b/Profile/Profile.cpp
index 4958cfb..be67d6d 100644
--- a/Profile/Profile.cpp
+++ b/Profile/Profile.cpp
@@ -187,6 +187,22 @@ auto Profile::getCredits() -> std::int32_t {
return _credits;
}
+auto Profile::setCredits(std::int32_t amount) -> bool {
+ auto mmap = Utility::Directory::map(Utility::Directory::join(_profileDirectory, _filename));
+
+ auto iter = std::search(mmap.begin(), mmap.end(), &credits_locator[0], &credits_locator[22]);
+
+ if(iter != mmap.end()) {
+ *reinterpret_cast(iter + 0x20) = amount;
+ _credits = amount;
+ return true;
+ }
+ else {
+ _lastError = "The profile save seems to be corrupted or the game didn't release the handle on the file.";
+ return false;
+ }
+}
+
auto Profile::storyProgress() const -> std::int32_t {
return _storyProgress;
}
diff --git a/Profile/Profile.h b/Profile/Profile.h
index 3781e73..01020b6 100644
--- a/Profile/Profile.h
+++ b/Profile/Profile.h
@@ -33,6 +33,7 @@ class Profile {
auto credits() const -> std::int32_t;
auto getCredits() -> std::int32_t;
+ auto setCredits(std::int32_t) -> bool;
auto storyProgress() const -> std::int32_t;
auto getStoryProgress() -> std::int32_t;