threaded song data?

Author: RayDeeUx

src/SongManager.cpp

@@ -306,4 +306,12 @@ void SongManager::setVibecodedVentilla(const bool value) {
 
 bool SongManager::getVibecodedVentilla() const {
 	return m_vibecodedVentilla;
+}
+
+void SongManager::setFinishedCalculatingSongLengths(const bool value) {
+	m_finishedCalculatingSongLengths = value;
+}
+
+bool SongManager::getFinishedCalculatingSongLengths() const {
+	return m_finishedCalculatingSongLengths;
 }

src/SongManager.hpp

@@ -92,6 +92,8 @@ class SongManager {
 	geode::Mod* getColonMenuLoopStartTime() const;
 	void setVibecodedVentilla(const bool);
 	bool getVibecodedVentilla() const;
+	void setFinishedCalculatingSongLengths(const bool);
+	bool getFinishedCalculatingSongLengths() const;
 
   private:
 	SongManager();
@@ -121,4 +123,5 @@ class SongManager {
 	int m_towerRepeatCount = 0;
 	geode::Mod* m_colonMenuLoopStartTime {};
 	SongToSongData m_songToSongDataMap;
+	bool m_finishedCalculatingSongLengths;
 };

src/Utils.cpp

@@ -470,6 +470,7 @@ void Utils::popualteSongToSongDataMap() {
 
 	const std::vector<std::string>& blacklist = songManager.getBlacklist();
 	const std::vector<std::string>& favorites = songManager.getFavorites();
+	std::vector<std::filesystem::path> tempKeys = {};
 
 	for (const std::string_view song : songManager.getSongs()) {
 		SongType songType = SongType::Regular;
@@ -488,7 +489,16 @@ void Utils::popualteSongToSongDataMap() {
 		};
 
 		songToSongData.emplace(theirPath, songData);
+		tempKeys.push_back(theirPath);
 	}
+
+	songManager.setFinishedCalculatingSongLengths(false);
+	std::thread([&songToSongData, &songManager]() {
+		for (auto& [path, songData] : songToSongData) {
+			songData.songLength = SongListLayer::getLength(geode::utils::string::pathToString(path), false);
+		}
+		songManager.setFinishedCalculatingSongLengths(true);
+	}).detach();
 }
 
 std::vector<std::string> Utils::parseBlacklistFile(const std::filesystem::path& blacklistFilePath) {

src/main.cpp

@@ -17,6 +17,7 @@ bool originalOverrideWasEmpty = false;
 	songManager.setConstantShuffleMode();
 	songManager.setLastMenuLoopPosition(0);
 	songManager.setShouldRestoreMenuLoopPoint(true);
+	songManager.setFinishedCalculatingSongLengths(false);
 	songManager.setAdvancedLogs(Mod::get()->getSettingValue<bool>("advancedLogs"));
 	songManager.setVibecodedVentilla(VIBECODED_RADIO && (VIBECODED_RADIO->isEnabled() || VIBECODED_RADIO->shouldLoad()));
 	if (!std::filesystem::exists(configDir / "playlistOne.txt")) Utils::writeToFile("# This file was generated automatically as it hadn't existed previously.", configDir / "playlistOne.txt");

src/ui/SongControlMenu.cpp

@@ -204,6 +204,7 @@ void SongControlMenu::onPreviousButton(CCObject*) {
 }
 
 void SongControlMenu::onPlaylistButton(CCObject*) {
+	if (!SongManager::get().getFinishedCalculatingSongLengths()) return FLAlertLayer::create("Hold on a sec!", "Menu Loop Randomizer is still calculating a few things. Hang tight, and try again later.", "I Understand")->show();
 	SongControlMenu::onClose(nullptr);
 	SongListLayer::create()->show();
 }