1 files changed, 31 insertions, 0 deletions
diff --git a/python/gdbpp/gdbpp/enumset.py b/python/gdbpp/gdbpp/enumset.py
new file mode 100644
index 0000000000..c36e863627
--- /dev/null
+++ b/python/gdbpp/gdbpp/enumset.py
@@ -0,0 +1,31 @@
+# -*- Mode: python; indent-tabs-mode: nil; tab-width: 40 -*-
+# vim: set filetype=python:
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+import gdb
+
+from gdbpp import GeckoPrettyPrinter
+
+
+@GeckoPrettyPrinter("mozilla::EnumSet", "^mozilla::EnumSet<.*>$")
+class enumset_printer(object):
+    def __init__(self, value):
+        self.value = value
+        self.enum_type = value.type.template_argument(0)
+
+    def children(self):
+        bitfield = self.value["mBitField"]
+        max_bit = (self.enum_type.sizeof * 8) - 1
+        return (
+            ("flag", gdb.Value(i).cast(self.enum_type))
+            for i in range(0, max_bit)
+            if ((bitfield & (2 ** i)) != 0)
+        )
+
+    def to_string(self):
+        return str(self.value.type)
+
+    def display_hint(self):
+        return "array"