Separated graycode generation and displaying.

Author: Tsing

src/app/generateGraycode.cpp

@@ -4,9 +4,16 @@ int main()
 {
     // Generate gray code based on the resolution of the projector.
     SLS::GrayCode gc(1024, 768);
-    // Show the image sequence.
-    // This function can be called when projecting patterns to the 
-    // reconstruction object.
-    gc.generateGrayCode();
+    std::vector<cv::Mat> grayCodeImages = gc.generateGrayCode();
+
+    // Display graycode
+    std::cout<<"Press 'q' to exit\n";
+    std::cout<<"Press any key to show next image\n";
+    for (const auto & image: grayCodeImages)
+    {
+        cv::imshow("GrayCode", image);
+        if (cv::waitKey(0) == 'q')
+            break;
+    }
     return 0;
 }

src/lib/GrayCode/GrayCode.cpp

@@ -8,7 +8,7 @@ GrayCode::GrayCode(size_t projW, size_t projH):
     numRowBits = (size_t)std::ceil(std::log2((float) height_));
     grayCodes_.resize(numColBits*2+numRowBits*2+2);
 }
-void GrayCode::generateGrayCode()
+const std::vector<cv::Mat>& GrayCode::generateGrayCode()
 {
     // Init rest of the mat
     for (auto &mat: grayCodes_)
@@ -76,10 +76,6 @@ void GrayCode::generateGrayCode()
 			prevRem=rem;
 		} 
 	}
-    for (size_t i=0; i<grayCodes_.size(); i++)
-    {
-        cv::imshow("test",grayCodes_[i]);
-        cv::waitKey(0);
-    }
+    return grayCodes_;
 }
 } // namespace SLS

src/lib/GrayCode/GrayCode.hpp

@@ -22,7 +22,7 @@ class GrayCode
      * The binary are encoded separately for columns and rows. i.e. for a projector pixel (x, y)
      * we have (BitSeqX, BitSeqY).
      */
-    void generateGrayCode();
+    const std::vector<cv::Mat>& generateGrayCode();
 private:
     //! Projector width and height
     size_t width_, height_;