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)); }