Add templated AddArgument for enum types (like VendorID) (#10721)

project-chip · Oct 21, 2021 · 7c3e2a8 · 7c3e2a8
1 parent c78dea0
commit 7c3e2a8
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/examples/chip-tool/commands/common/Command.h b/examples/chip-tool/commands/common/Command.h
@@ -145,6 +145,12 @@ class Command
         return AddArgument(name, min, max, reinterpret_cast<void *>(out), Number_uint64);
     }
 
+    template <typename T, typename = std::enable_if_t<std::is_enum<T>::value>>
+    size_t AddArgument(const char * name, int64_t min, uint64_t max, T * out)
+    {
+        return AddArgument(name, min, max, reinterpret_cast<std::underlying_type_t<T> *>(out));
+    }
+
     virtual CHIP_ERROR Run() = 0;
 
 private: