diff --git a/src/ai/ai_scanner.cpp b/src/ai/ai_scanner.cpp index bdf089a60b..8e802cafc8 100644 --- a/src/ai/ai_scanner.cpp +++ b/src/ai/ai_scanner.cpp @@ -166,8 +166,18 @@ AIScanner::~AIScanner() { AIInfoList::iterator it = this->info_list.begin(); for (; it != this->info_list.end(); it++) { + free((void *)(*it).first); delete (*it).second; } + it = this->info_single_list.begin(); + for (; it != this->info_single_list.end(); it++) { + free((void *)(*it).first); + } + AILibraryList::iterator lit = this->library_list.begin(); + for (; lit != this->library_list.end(); lit++) { + free((void *)(*lit).first); + delete (*lit).second; + } delete this->engine; } @@ -308,30 +318,32 @@ void AIScanner::RegisterAI(AIInfo *info) } this->info_list[strdup(ai_name)] = info; -} - -void AIScanner::UnregisterAI(AIInfo *info) -{ - char ai_name[1024]; - snprintf(ai_name, sizeof(ai_name), "%s.%d", info->GetInstanceName(), info->GetVersion()); - this->info_list.erase(ai_name); + /* Add the AI to the 'unique' AI list, where only the highest version of the + * AI is registered. */ + snprintf(ai_name, sizeof(ai_name), "%s", info->GetInstanceName()); + strtolower(ai_name); + if (this->info_single_list.find(ai_name) == this->info_single_list.end()) { + this->info_single_list[strdup(ai_name)] = info; + } else if (this->info_single_list[strdup(ai_name)]->GetVersion() < info->GetVersion()) { + this->info_single_list[ai_name] = info; + } } AIInfo *AIScanner::SelectRandomAI() { - if (this->info_list.size() == 0) { + if (this->info_single_list.size() == 0) { DEBUG(ai, 0, "No suitable AI found, loading 'dummy' AI."); return this->info_dummy; } /* Find a random AI */ uint pos; - if (_networking) pos = InteractiveRandomRange((uint16)this->info_list.size()); - else pos = RandomRange((uint16)this->info_list.size()); + if (_networking) pos = InteractiveRandomRange((uint16)this->info_single_list.size()); + else pos = RandomRange((uint16)this->info_single_list.size()); /* Find the Nth item from the array */ - AIInfoList::iterator it = this->info_list.begin(); + AIInfoList::iterator it = this->info_single_list.begin(); for (; pos > 0; pos--) it++; AIInfoList::iterator first_it = it; return (*it).second; @@ -354,19 +366,7 @@ AIInfo *AIScanner::FindInfo(const char *nameParam, int versionParam) strtolower(ai_name); /* We want to load the latest version of this AI; so find it */ - AIInfoList::iterator it = this->info_list.begin(); - for (; it != this->info_list.end(); it++) { - char ai_name_compare[1024]; - snprintf(ai_name_compare, sizeof(ai_name_compare), "%s", (*it).second->GetInstanceName()); - strtolower(ai_name_compare); - - if (strcasecmp(ai_name, ai_name_compare) == 0 && (*it).second->GetVersion() > version) { - version = (*it).second->GetVersion(); - info = (*it).second; - } - } - - if (info != NULL) return info; + if (this->info_single_list.find(ai_name) != this->info_single_list.end()) return this->info_single_list[ai_name]; /* If we didn't find a match AI, maybe the user included a version */ char *e = strrchr(name, '.'); diff --git a/src/ai/ai_scanner.hpp b/src/ai/ai_scanner.hpp index c974e14f71..3e82e46822 100644 --- a/src/ai/ai_scanner.hpp +++ b/src/ai/ai_scanner.hpp @@ -29,11 +29,6 @@ public: void SetDummyAI(class AIInfo *info) { this->info_dummy = info; } - /** - * Remove an AI from the available list. - */ - void UnregisterAI(class AIInfo *info); - /** * Select a Random AI. */ @@ -88,6 +83,7 @@ private: AIInfo *info_dummy; AIInfoList info_list; + AIInfoList info_single_list; AILibraryList library_list; class Squirrel *engine; char main_script[1024];