From c1ed17dd8623fe3921cc38bcfd5bd498789c548d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Du=C5=A1an=20Poizl?= <nou.spiro@gmail.com>
Date: Thu, 25 Mar 2021 09:33:03 +0100
Subject: [PATCH] Add RGB variants to RawImage

---
 rawimage.cpp | 12 ++++++++++++
 rawimage.h   |  3 +++
 2 files changed, 15 insertions(+)

diff --git a/rawimage.cpp b/rawimage.cpp
index e4e0898..93b63c1 100644
--- a/rawimage.cpp
+++ b/rawimage.cpp
@@ -12,6 +12,12 @@ RawImage::ImgType CV2Type(int cvtype)
         return RawImage::UINT16;
     case CV_32F:
         return RawImage::FLOAT32;
+    case CV_8UC3:
+        return RawImage::UINT8C3;
+    case CV_16UC3:
+        return RawImage::UINT16C3;
+    case CV_32FC3:
+        return RawImage::FLOAT32C3;
     default:
         return RawImage::UNKNOWN;
     }
@@ -27,6 +33,12 @@ int Type2CV(RawImage::ImgType type)
         return CV_16U;
     case RawImage::FLOAT32:
         return CV_32F;
+    case RawImage::UINT8C3:
+        return CV_8UC3;
+    case RawImage::UINT16C3:
+        return CV_16UC3;
+    case RawImage::FLOAT32C3:
+        return CV_32FC3;
     case RawImage::UNKNOWN:
         return CV_8S;
     default:
diff --git a/rawimage.h b/rawimage.h
index 1323a68..f68262e 100644
--- a/rawimage.h
+++ b/rawimage.h
@@ -40,6 +40,9 @@ public:
         UINT8,
         UINT16,
         FLOAT32,
+        UINT8C3,
+        UINT16C3,
+        FLOAT32C3,
         UNKNOWN,
     };
     RawImage();