[llvm] r284433 - [ADT] Add an initializer_list constructor to {Small, }DenseSet.

[Justin Lebar via llvm-commits]

Author: jlebar
Date: Mon Oct 17 17:24:32 2016
New Revision: 284433

URL: http://llvm.org/viewvc/llvm-project?rev=284433&view=rev
Log:
[ADT] Add an initializer_list constructor to {Small,}DenseSet.

Reviewers: timshen

Subscribers: llvm-commits

Differential Revision: https://reviews.llvm.org/D25629

Modified:
    llvm/trunk/include/llvm/ADT/DenseSet.h
    llvm/trunk/unittests/ADT/DenseSetTest.cpp

Modified: llvm/trunk/include/llvm/ADT/DenseSet.h
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/include/llvm/ADT/DenseSet.h?rev=284433&r1=284432&r2=284433&view=diff
==============================================================================
--- llvm/trunk/include/llvm/ADT/DenseSet.h (original)
+++ llvm/trunk/include/llvm/ADT/DenseSet.h Mon Oct 17 17:24:32 2016
@@ -15,6 +15,7 @@
 #define LLVM_ADT_DENSESET_H
 
 #include "llvm/ADT/DenseMap.h"
+#include <initializer_list>
 
 namespace llvm {
 
@@ -55,6 +56,11 @@ public:
 
   explicit DenseSetImpl(unsigned InitialReserve = 0) : TheMap(InitialReserve) {}
 
+  DenseSetImpl(std::initializer_list<ValueT> Elems)
+      : DenseSetImpl(Elems.size()) {
+    insert(Elems.begin(), Elems.end());
+  }
+
   bool empty() const { return TheMap.empty(); }
   size_type size() const { return TheMap.size(); }
   size_t getMemorySize() const { return TheMap.getMemorySize(); }

Modified: llvm/trunk/unittests/ADT/DenseSetTest.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/unittests/ADT/DenseSetTest.cpp?rev=284433&r1=284432&r2=284433&view=diff
==============================================================================
--- llvm/trunk/unittests/ADT/DenseSetTest.cpp (original)
+++ llvm/trunk/unittests/ADT/DenseSetTest.cpp Mon Oct 17 17:24:32 2016
@@ -64,6 +64,21 @@ typedef ::testing::Types<DenseSet<unsign
     DenseSetTestTypes;
 TYPED_TEST_CASE(DenseSetTest, DenseSetTestTypes);
 
+TYPED_TEST(DenseSetTest, InitializerList) {
+  TypeParam set({1, 2, 1, 4});
+  EXPECT_EQ(3u, set.size());
+  EXPECT_EQ(1u, set.count(1));
+  EXPECT_EQ(1u, set.count(2));
+  EXPECT_EQ(1u, set.count(4));
+  EXPECT_EQ(0u, set.count(3));
+}
+
+TYPED_TEST(DenseSetTest, EmptyInitializerList) {
+  TypeParam set({});
+  EXPECT_EQ(0u, set.size());
+  EXPECT_EQ(0u, set.count(0));
+}
+
 TYPED_TEST(DenseSetTest, FindAsTest) {
   auto &set = this->Set;
   // Size tests