1 //===-- ThreadSafeDenseSet.h ------------------------------------------*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #ifndef liblldb_ThreadSafeDenseSet_h_
11 #define liblldb_ThreadSafeDenseSet_h_
12 
13 // C Includes
14 // C++ Includes
15 
16 // Other libraries and framework includes
17 #include "llvm/ADT/DenseSet.h"
18 
19 // Project includes
20 #include "lldb/Host/Mutex.h"
21 
22 namespace lldb_private {
23 
24     template <typename _ElementType>
25     class ThreadSafeDenseSet
26     {
27     public:
28         typedef llvm::DenseSet<_ElementType> LLVMSetType;
29 
30         ThreadSafeDenseSet(unsigned set_initial_capacity = 0,
31                            Mutex::Type mutex_type = Mutex::eMutexTypeNormal) :
32         m_set(set_initial_capacity),
33         m_mutex(mutex_type)
34         {
35         }
36 
37         void
38         Insert (_ElementType e)
39         {
40             Mutex::Locker locker(m_mutex);
41             m_set.insert(e);
42         }
43 
44         void
45         Erase (_ElementType e)
46         {
47             Mutex::Locker locker(m_mutex);
48             m_set.erase(e);
49         }
50 
51         bool
52         Lookup (_ElementType e)
53         {
54             Mutex::Locker locker(m_mutex);
55             return (m_set.count(e) > 0);
56         }
57 
58     protected:
59         LLVMSetType m_set;
60         Mutex m_mutex;
61     };
62 
63 } // namespace lldb_private
64 
65 #endif  // liblldb_ThreadSafeDenseSet_h_
66