1 files changed, 43 insertions, 11 deletions
diff --git a/test/test-software-roi.c b/test/test-software-roi.c
index 2b628c5..08ab742 100644
--- a/test/test-software-roi.c
+++ b/test/test-software-roi.c
@@ -2,18 +2,34 @@
 #include "software-roi.h"
 
 static const guchar test_frame[] = {
-    0, 1, 2, 3, 4, 5, 6, 7, 8,
-    0, 1, 2, 3, 4, 5, 6, 7, 8,
-    0, 1, 2, 3, 4, 5, 6, 7, 8,
-    0, 1, 2, 3, 4, 5, 6, 7, 8,
-    0, 1, 2, 3, 4, 5, 6, 7, 8,
+    0, 10, 20, 30, 40, 50, 60, 70, 80,
+    0, 11, 21, 31, 41, 51, 61, 71, 81,
+    0, 12, 22, 32, 42, 52, 62, 72, 82,
+    0, 13, 23, 33, 43, 53, 63, 73, 83,
+    0, 14, 24, 34, 44, 54, 64, 74, 84,
 };
 static const guint test_frame_width = 9;
-
 static guchar test_roi_2x1_5x3[] = {
-    2, 3, 4, 5, 6,
-    2, 3, 4, 5, 6,
-    2, 3, 4, 5, 6,
+    21, 31, 41, 51, 61,
+    22, 32, 42, 52, 62,
+    23, 33, 43, 53, 63,
+};
+
+static const guchar test_frame16bit[] = {
+    0, 0, 10, 10, 20, 20, 30, 30, 40, 40, 50, 50, 60, 60, 70, 70,
+    1, 1, 11, 11, 21, 21, 31, 31, 41, 41, 51, 51, 61, 61, 71, 71,
+    2, 2, 12, 12, 22, 22, 32, 32, 42, 42, 52, 52, 62, 62, 72, 72,
+    3, 3, 13, 13, 23, 23, 33, 33, 43, 43, 53, 53, 63, 63, 73, 73,
+    4, 4, 14, 14, 24, 24, 34, 34, 44, 44, 54, 54, 64, 64, 74, 74,
+    5, 5, 15, 15, 25, 25, 35, 35, 45, 45, 55, 55, 65, 65, 75, 75,
+    6, 6, 16, 16, 26, 26, 36, 36, 46, 46, 56, 56, 66, 66, 76, 76,
+};
+static const guint test_frame16bit_width = 8;
+static guchar test_roi16bit_3x3_5x3[] = {
+    33, 33, 43, 43, 53, 53, 63, 63, 73, 73,
+    34, 34, 44, 44, 54, 54, 64, 64, 74, 74,
+    35, 35, 45, 45, 55, 55, 65, 65, 75, 75,
+    36, 36, 46, 46, 56, 56, 66, 66, 76, 76,
 };
 
 void typical_roi_test(void)
@@ -24,7 +40,7 @@ void typical_roi_test(void)
     guint roiHeight = 3;
     guint roiSize = roiWidth * roiHeight;
     guchar roiFrame[roiSize];
-    apply_software_roi(test_frame, test_frame_width, roiFrame, roiX, roiY, roiWidth, roiHeight);
+    apply_software_roi(test_frame, test_frame_width, 1, roiFrame, roiX, roiY, roiWidth, roiHeight);
     for (guint i = 0; i < roiSize; i++) {
         g_assert_cmpint(test_roi_2x1_5x3[i], ==, roiFrame[i]);
     }
@@ -38,16 +54,32 @@ void nrows_only_roi_test(void)
     guint roiHeight = 3;
     guint roiSize = roiWidth * roiHeight;
     guchar roiFrame[roiSize];
-    apply_software_roi(test_frame, test_frame_width, roiFrame, roiX, roiY, roiWidth, roiHeight);
+    apply_software_roi(test_frame, test_frame_width, 1, roiFrame, roiX, roiY, roiWidth, roiHeight);
     for (guint i = 0; i < roiSize; i++) {
         g_assert_cmpint(test_frame[i], ==, roiFrame[i]);
     }
 }
 
+void multibyte_image_test(void)
+{
+    guint roiX = 3;
+    guint roiY = 3;
+    guint roiWidth = 5;
+    guint roiHeight = 3;
+    guint bytesPerPixel = 2;
+    guint roiSize = roiWidth * bytesPerPixel * roiHeight;
+    guchar roiFrame[roiSize];
+    apply_software_roi(test_frame16bit, test_frame16bit_width, bytesPerPixel, roiFrame, roiX, roiY, roiWidth, roiHeight);
+    for (guint i = 0; i < roiSize; i++) {
+        g_assert_cmpint(test_roi16bit_3x3_5x3[i], ==, roiFrame[i]);
+    }
+}
+
 int main(int argc, char** argv)
 {
     g_test_init(&argc, &argv, NULL);
     g_test_add_func("/software-roi/apply-typical-roi", typical_roi_test);
     g_test_add_func("/software-roi/apply-roi-nrows-only", nrows_only_roi_test);
+    g_test_add_func("/software-roi/apply-roi-multibyte-image", multibyte_image_test);
     return g_test_run();
 }