increase coverage

Author: SSARCandy

include/postProcessing.h

@@ -6,7 +6,7 @@
 namespace postprocess
 {
 
-void visualizeFlowfield(const cv::Mat &flowfield, cv::Mat &dst);
+cv::Mat visualizeFlowfield(const cv::Mat &flowfield);
 cv::Mat visualizeETF(const cv::Mat &flowfield);
 cv::Mat antiAlias(const cv::Mat &src);
 

src/cmd.cpp

@@ -95,6 +95,12 @@ int main(int argc, char *argv[])
         cv::cvtColor(vis_etf, vis_etf, CV_GRAY2BGR);
         cv::imwrite("visualize-etf.jpg", vis_etf);
 
+        // Generate flow-field arrow direction visualization
+        cv::Mat vis_flow = postprocess::visualizeFlowfield(cld.etf.flowField);
+        vis_flow.convertTo(vis_flow, CV_8UC3, 255);
+        cv::cvtColor(vis_flow, vis_flow, CV_RGB2BGR);
+        cv::imwrite("arrow-etf.jpg", vis_flow);
+
         // Generate Anti-alias coherent line drawing
         cv::Mat anti_alias = postprocess::antiAlias(cld.result.clone());
         cv::imwrite("anti-alias.jpg", anti_alias);

src/gui.cpp

@@ -467,7 +467,9 @@ void BasicDrawPane::render(wxDC &dc, bool render_loop_on)
         dis.convertTo(dis, CV_8UC1, 255);
         cv::cvtColor(dis, dis, CV_GRAY2BGR);
     } else if (processingS == MODE_ETF_DEBUG) {
-        postprocess::visualizeFlowfield(cld.etf.flowField, dis);
+        dis = postprocess::visualizeFlowfield(cld.etf.flowField);
+        dis.convertTo(dis, CV_8UC3, 255);
+        cv::cvtColor(dis, dis, CV_RGB2BGR);
     } else if (processingS == MODE_CLD) {
         dis = cld.result.clone();
         cv::cvtColor(dis, dis, CV_GRAY2BGR);

src/postProcessing.cpp

@@ -57,19 +57,22 @@ cv::Mat visualizeETF(const cv::Mat &flowfield)
 }
 
 // visualize ETF by drawing red arrowline
-void visualizeFlowfield(const cv::Mat &flowfield, cv::Mat &dst)
+cv::Mat visualizeFlowfield(const cv::Mat &flowfield)
 {
     constexpr int resolution = 10;
+    cv::Mat dst{flowfield.size(), CV_32FC3, cv::Scalar(1, 1, 1)};
 
     for (int i = 0; i < dst.rows; i += resolution) {
         for (int j = 0; j < dst.cols; j += resolution) {
             const cv::Vec3f v = flowfield.at<cv::Vec3f>(i, j);
             const cv::Point2f p1(j, i);
             const cv::Point2f p2(j + v[1] * 5, i + v[0] * 5);
 
-            cv::arrowedLine(dst, p1, p2, cv::Scalar(255, 0, 0), 1.5, 8, 0, 0.3);
+            cv::arrowedLine(dst, p1, p2, cv::Scalar(1, 0, 0), 1.5, 8, 0, 0.3);
         }
     }
+
+    return dst;
 }
 
 cv::Mat antiAlias(const cv::Mat &src)

test.sh

@@ -6,6 +6,7 @@ time ./build/cld --src ./data/fingerprint.jpg -o output.jpg --ETF_iter 3 --CLD_i
 echo '10459bb86077235db3f85fe64d25f2d8 output.jpg' | md5sum -c - 
 echo '3ea95ae0f6f92478af7fbf97e96b2763 anti-alias.jpg' | md5sum -c - 
 echo '1a68133901751c29efe4febec29d014e visualize-etf.jpg' | md5sum -c - 
+echo '365b29b4d15520b5c92eb78bc6c236b8 arrow-etf.jpg' | md5sum -c - 
 
 if ! which lcov; then
     exit 0