From cd9af4fc489af0b31cb34a96ddb740e1baf19966 Mon Sep 17 00:00:00 2001
From: John Zhao <johnzhao@slightech.com>
Date: Tue, 3 Apr 2018 13:53:32 +0800
Subject: [PATCH] Update types

---
 include/mynteye/types.h   | 4 ++++
 src/public/types.cc       | 2 ++
 test/public/types_test.cc | 2 ++
 3 files changed, 8 insertions(+)

diff --git a/include/mynteye/types.h b/include/mynteye/types.h
index 7cbdfa1..f8b849d 100644
--- a/include/mynteye/types.h
+++ b/include/mynteye/types.h
@@ -52,6 +52,10 @@ enum class Capabilities : std::uint8_t {
   POINTS,
   /** Provices fisheye stream */
   FISHEYE,
+  /** Provices infrared stream */
+  INFRARED,
+  /** Provices second infrared stream */
+  INFRARED2,
   /** Provices IMU (accelerometer, gyroscope) data */
   IMU,
   /** Last guard */
diff --git a/src/public/types.cc b/src/public/types.cc
index 08964db..66bea9c 100644
--- a/src/public/types.cc
+++ b/src/public/types.cc
@@ -34,6 +34,8 @@ const char *to_string(const Capabilities &value) {
     CASE(DEPTH)
     CASE(POINTS)
     CASE(FISHEYE)
+    CASE(INFRARED)
+    CASE(INFRARED2)
     CASE(IMU)
     default:
       CHECK(is_valid(value));
diff --git a/test/public/types_test.cc b/test/public/types_test.cc
index e77fecb..b007c30 100644
--- a/test/public/types_test.cc
+++ b/test/public/types_test.cc
@@ -22,6 +22,8 @@ TEST(Capabilities, VerifyToString) {
   EXPECT_STREQ("Capabilities::DEPTH", to_string(Capabilities::DEPTH));
   EXPECT_STREQ("Capabilities::POINTS", to_string(Capabilities::POINTS));
   EXPECT_STREQ("Capabilities::FISHEYE", to_string(Capabilities::FISHEYE));
+  EXPECT_STREQ("Capabilities::INFRARED", to_string(Capabilities::INFRARED));
+  EXPECT_STREQ("Capabilities::INFRARED2", to_string(Capabilities::INFRARED2));
   EXPECT_STREQ("Capabilities::IMU", to_string(Capabilities::IMU));
 }