Remove low performance containers with standard library ones

This commit is contained in:
Andreas Schmitt 2021-06-13 10:18:00 +02:00 committed by Jonathan G Rennison
parent 83ea6e9fd8
commit 35ebeff874
2 changed files with 50 additions and 15 deletions

View File

@ -26,6 +26,7 @@
#include "aystar.h" #include "aystar.h"
#include "../../safeguards.h" #include "../../safeguards.h"
#include "core/mem_func.hpp"
/** /**
* This looks in the hash whether a node exists in the closed list. * This looks in the hash whether a node exists in the closed list.
@ -34,7 +35,9 @@
*/ */
PathNode *AyStar::ClosedListIsInList(const AyStarNode *node) PathNode *AyStar::ClosedListIsInList(const AyStarNode *node)
{ {
return (PathNode*)this->closedlist_hash.Get(node->tile, node->direction); const auto result = this->closedlist_hash.find(std::make_pair(node->tile, node->direction));
return (result == this->closedlist_hash.end()) ? nullptr : result->second;
} }
/** /**
@ -45,9 +48,10 @@ PathNode *AyStar::ClosedListIsInList(const AyStarNode *node)
void AyStar::ClosedListAdd(const PathNode *node) void AyStar::ClosedListAdd(const PathNode *node)
{ {
/* Add a node to the ClosedList */ /* Add a node to the ClosedList */
PathNode *new_node = MallocT<PathNode>(1); const auto new_node = MallocT<PathNode>(1);
*new_node = *node; *new_node = *node;
this->closedlist_hash.Set(node->node.tile, node->node.direction, new_node);
this->closedlist_hash[std::make_pair(node->node.tile, node->node.direction)] = new_node;
} }
/** /**
@ -57,7 +61,9 @@ void AyStar::ClosedListAdd(const PathNode *node)
*/ */
OpenListNode *AyStar::OpenListIsInList(const AyStarNode *node) OpenListNode *AyStar::OpenListIsInList(const AyStarNode *node)
{ {
return (OpenListNode*)this->openlist_hash.Get(node->tile, node->direction); const auto result = this->openlist_hash.find(std::make_pair(node->tile, node->direction));
return (result == this->openlist_hash.end()) ? nullptr : result->second;
} }
/** /**
@ -70,7 +76,7 @@ OpenListNode *AyStar::OpenListPop()
/* Return the item the Queue returns.. the best next OpenList item. */ /* Return the item the Queue returns.. the best next OpenList item. */
OpenListNode *res = (OpenListNode*)this->openlist_queue.Pop(); OpenListNode *res = (OpenListNode*)this->openlist_queue.Pop();
if (res != nullptr) { if (res != nullptr) {
this->openlist_hash.DeleteValue(res->path.node.tile, res->path.node.direction); this->openlist_hash.erase(std::make_pair(res->path.node.tile, res->path.node.direction));
} }
return res; return res;
@ -87,7 +93,7 @@ void AyStar::OpenListAdd(PathNode *parent, const AyStarNode *node, int f, int g)
new_node->g = g; new_node->g = g;
new_node->path.parent = parent; new_node->path.parent = parent;
new_node->path.node = *node; new_node->path.node = *node;
this->openlist_hash.Set(node->tile, node->direction, new_node); this->openlist_hash[std::make_pair(node->tile, node->direction)] = new_node;
/* Add it to the queue */ /* Add it to the queue */
this->openlist_queue.Push(new_node, f); this->openlist_queue.Push(new_node, f);
@ -134,6 +140,7 @@ void AyStar::CheckTile(AyStarNode *current, OpenListNode *parent)
/* Yes, check if this g value is lower.. */ /* Yes, check if this g value is lower.. */
if (new_g > check->g) return; if (new_g > check->g) return;
this->openlist_queue.Delete(check, 0); this->openlist_queue.Delete(check, 0);
/* It is lower, so change it to this item */ /* It is lower, so change it to this item */
check->g = new_g; check->g = new_g;
check->path.parent = closedlist_parent; check->path.parent = closedlist_parent;
@ -191,7 +198,7 @@ int AyStar::Loop()
/* Free the node */ /* Free the node */
free(current); free(current);
if (this->max_search_nodes != 0 && this->closedlist_hash.GetSize() >= this->max_search_nodes) { if (this->max_search_nodes != 0 && this->closedlist_hash.size() >= this->max_search_nodes) {
/* We've expanded enough nodes */ /* We've expanded enough nodes */
return AYSTAR_LIMIT_REACHED; return AYSTAR_LIMIT_REACHED;
} else { } else {
@ -208,8 +215,14 @@ void AyStar::Free()
this->openlist_queue.Free(false); this->openlist_queue.Free(false);
/* 2nd argument above is false, below is true, to free the values only /* 2nd argument above is false, below is true, to free the values only
* once */ * once */
this->openlist_hash.Delete(true); for (const auto& pair : this->openlist_hash) {
this->closedlist_hash.Delete(true); free(pair.second);
}
this->openlist_hash.clear();
for (const auto& pair : this->closedlist_hash) {
free(pair.second);
}
this->closedlist_hash.clear();
#ifdef AYSTAR_DEBUG #ifdef AYSTAR_DEBUG
printf("[AyStar] Memory free'd\n"); printf("[AyStar] Memory free'd\n");
#endif #endif
@ -225,8 +238,15 @@ void AyStar::Clear()
* the hash. */ * the hash. */
this->openlist_queue.Clear(false); this->openlist_queue.Clear(false);
/* Clean the hashes */ /* Clean the hashes */
this->openlist_hash.Clear(true); for (const auto& pair : this->openlist_hash) {
this->closedlist_hash.Clear(true); free(pair.second);
}
this->openlist_hash.clear();
for (const auto& pair : this->closedlist_hash) {
free(pair.second);
}
this->closedlist_hash.clear();
#ifdef AYSTAR_DEBUG #ifdef AYSTAR_DEBUG
printf("[AyStar] Cleared AyStar\n"); printf("[AyStar] Cleared AyStar\n");
@ -292,9 +312,12 @@ void AyStar::AddStartNode(AyStarNode *start_node, uint g)
*/ */
void AyStar::Init(Hash_HashProc hash, uint num_buckets) void AyStar::Init(Hash_HashProc hash, uint num_buckets)
{ {
MemSetT(&neighbours, 0);
MemSetT(&openlist_queue, 0);
/* Allocated the Hash for the OpenList and ClosedList */ /* Allocated the Hash for the OpenList and ClosedList */
this->openlist_hash.Init(hash, num_buckets); this->openlist_hash.reserve(num_buckets);
this->closedlist_hash.Init(hash, num_buckets); this->closedlist_hash.reserve(num_buckets);
/* Set up our sorting queue /* Set up our sorting queue
* BinaryHeap allocates a block of 1024 nodes * BinaryHeap allocates a block of 1024 nodes

View File

@ -17,6 +17,9 @@
#define AYSTAR_H #define AYSTAR_H
#include "queue.h" #include "queue.h"
#include <unordered_map>
#include <memory>
#include "../../tile_type.h" #include "../../tile_type.h"
#include "../../track_type.h" #include "../../track_type.h"
@ -57,6 +60,15 @@ struct OpenListNode {
PathNode path; PathNode path;
}; };
struct PairHash {
public:
template <typename T, typename U>
std::size_t operator()(const std::pair<T, U> &x) const
{
return std::hash<T>()(x.first) ^ std::hash<U>()(x.second);
}
};
bool CheckIgnoreFirstTile(const PathNode *node); bool CheckIgnoreFirstTile(const PathNode *node);
struct AyStar; struct AyStar;
@ -156,9 +168,9 @@ struct AyStar {
void CheckTile(AyStarNode *current, OpenListNode *parent); void CheckTile(AyStarNode *current, OpenListNode *parent);
protected: protected:
Hash closedlist_hash; ///< The actual closed list. std::unordered_map<std::pair<TileIndex, Trackdir>, PathNode*, PairHash> closedlist_hash;
BinaryHeap openlist_queue; ///< The open queue. BinaryHeap openlist_queue; ///< The open queue.
Hash openlist_hash; ///< An extra hash to speed up the process of looking up an element in the open list. std::unordered_map<std::pair<TileIndex, Trackdir>, OpenListNode*, PairHash> openlist_hash;
void OpenListAdd(PathNode *parent, const AyStarNode *node, int f, int g); void OpenListAdd(PathNode *parent, const AyStarNode *node, int f, int g);
OpenListNode *OpenListIsInList(const AyStarNode *node); OpenListNode *OpenListIsInList(const AyStarNode *node);