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 + 0 + + 1 + 1 + 1 + 1 + + + + + + + + + 1 + 0 + 1 + + 1 + + 0 + 0 + + Dock + 0 + Left + 1 + + 1 + + + 0 + 0 + wxID_ANY + Edit credits + + 0 + + 0 + + + 0 + + 1 + _creditsEditButton + 1 + + + protected + 1 + + + + Resizable + 1 + + + ; ; forward_declare + 0 + + + wxFILTER_NONE + wxDefaultValidator + + + + + creditsEditEvent + + 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;