From 53032e556a4479d468b4cec986956abc7cd8620d Mon Sep 17 00:00:00 2001
From: Siarhei Fedartsou <siarhei.fedartsou@gmail.com>
Date: Tue, 9 Jul 2024 22:53:24 +0200
Subject: [PATCH] wip

---
 include/contractor/contractor_heap.hpp |  3 ++-
 include/customizer/cell_customizer.hpp |  2 +-
 include/util/query_heap.hpp            | 13 ++++++++++---
 3 files changed, 13 insertions(+), 5 deletions(-)

diff --git a/include/contractor/contractor_heap.hpp b/include/contractor/contractor_heap.hpp
index fc581f3b0..77a0a85af 100644
--- a/include/contractor/contractor_heap.hpp
+++ b/include/contractor/contractor_heap.hpp
@@ -20,7 +20,8 @@ using ContractorHeap = util::QueryHeap<NodeID,
                                        NodeID,
                                        EdgeWeight,
                                        ContractorHeapData,
-                                       util::XORFastHashStorage<NodeID, NodeID>>;
+                                       util::XORFastHashStorage<NodeID, NodeID>,
+                                       false>;
 
 } // namespace osrm::contractor
 
diff --git a/include/customizer/cell_customizer.hpp b/include/customizer/cell_customizer.hpp
index 00d03163a..1e2072d7d 100644
--- a/include/customizer/cell_customizer.hpp
+++ b/include/customizer/cell_customizer.hpp
@@ -25,7 +25,7 @@ class CellCustomizer
 
   public:
     using Heap =
-        util::QueryHeap<NodeID, NodeID, EdgeWeight, HeapData, util::ArrayStorage<NodeID, int>>;
+        util::QueryHeap<NodeID, NodeID, EdgeWeight, HeapData, util::ArrayStorage<NodeID, int>, false>;
     using HeapPtr = tbb::enumerable_thread_specific<Heap>;
 
     CellCustomizer(const partitioner::MultiLevelPartition &partition) : partition(partition) {}
diff --git a/include/util/query_heap.hpp b/include/util/query_heap.hpp
index 83814a470..77584c73a 100644
--- a/include/util/query_heap.hpp
+++ b/include/util/query_heap.hpp
@@ -124,7 +124,7 @@ template <typename NodeID, typename Key> class UnorderedMapStorage
   private:
     template <typename K, typename V>
     using UnorderedMap = std::
-        unordered_map<K, V, std::hash<K>, std::equal_to<K>, PoolAllocator<std::pair<const K, V>>>;
+        unordered_map<K, V/*, std::hash<K>, std::equal_to<K>, PoolAllocator<std::pair<const K, V>>*/>;
 
     UnorderedMap<NodeID, Key> nodes;
 };
@@ -241,7 +241,8 @@ template <typename NodeID,
           typename Key,
           typename Weight,
           typename Data,
-          typename IndexStorage = ArrayStorage<NodeID, NodeID>>
+          typename IndexStorage = ArrayStorage<NodeID, NodeID>,
+          bool ThreadLocal = true>
 class QueryHeap
 {
   private:
@@ -259,11 +260,17 @@ class QueryHeap
             return weight > other.weight;
         }
     };
+
+  using AllocatorType = typename std::conditional<ThreadLocal,
+                                                    PoolAllocator<HeapData>,
+                                                    std::allocator<HeapData>>::type;
+
+
     using HeapContainer = boost::heap::d_ary_heap<HeapData,
                                                   boost::heap::arity<4>,
                                                   boost::heap::mutable_<true>,
                                                   boost::heap::compare<std::greater<HeapData>>,
-                                                  boost::heap::allocator<PoolAllocator<HeapData>>>;
+                                                  boost::heap::allocator<AllocatorType>>;
     using HeapHandle = typename HeapContainer::handle_type;
 
   public: