feat: add function to generate matrix in cpp

Author: edmBernard

tests/cpp/test.cpp

@@ -27,6 +27,49 @@ cv::Mat read_image(std::string image_name) {
   return image;
 }
 
+constexpr int width = 10;
+constexpr int height = 10;
+constexpr int channel = 3;
+
+
+// Generate buffer for a matrix 10x10x3
+std::vector<uint8_t> generateBuffer() {
+  std::vector<uint8_t> matrice;
+  const int size = width * height * channel;
+  matrice.reserve(size);
+  for (int i = 0; i < size; ++i) {
+    matrice.push_back(i);
+  }
+  return matrice;
+}
+
+
+// Generate matrix 10x10x3
+// Example of function that return a matrix by value
+cv::Mat generateMatrix() {
+  const std::vector<uint8_t> buffer = generateBuffer();
+  return cv::Mat(buffer, true).reshape(3, {10, 10});
+}
+
+
+// Example of function that take a matrix as argument
+bool checkMatrixContent(const cv::Mat& mat) {
+  // I don't compare buffer from the matrix with expected buffer to correctly manage slicing.
+  const std::vector<uint8_t> expectedBuffer = generateBuffer();
+  bool match = true;
+  for (int j = 0; j < height; ++j) {
+    for (int i = 0; i < width; ++i) {
+      const cv::Vec3b values = mat.at<cv::Vec3b>(cv::Point(i,j));
+      match &= values(0) == expectedBuffer[j * width * 3 + i * 3 + 0];
+      match &= values(1) == expectedBuffer[j * width * 3 + i * 3 + 1];
+      match &= values(2) == expectedBuffer[j * width * 3 + i * 3 + 2];
+    }
+  }
+  return match;
+}
+
+
+
 cv::Mat passthru(cv::Mat image) {
   return image;
 }
@@ -66,6 +109,12 @@ PYBIND11_MODULE(test_module, m) {
   m.def("show_image", &show_image, "A function that show an image",
         py::arg("image"));
 
+
+  m.def("generate_matrix", &generateMatrix, "A function that generate a image");
+
+  m.def("check_matrix_content", &checkMatrixContent, "A function that check the content a an image",
+        py::arg("image"));
+
   m.def("passthru", &passthru, "Passthru function", py::arg("image"));
   m.def("clone", &cloneimg, "Clone function", py::arg("image"));