From ea01adb9c978162873c91f1ee1a83f30e68faaa0 Mon Sep 17 00:00:00 2001
From: Alexey Spizhevoy <no@email>
Date: Fri, 14 Jan 2011 08:08:02 +0000
Subject: [PATCH] fixed writing video in HOG sample

---
 samples/gpu/hog.cpp | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/samples/gpu/hog.cpp b/samples/gpu/hog.cpp
index 81d61b0e0..8449004c5 100644
--- a/samples/gpu/hog.cpp
+++ b/samples/gpu/hog.cpp
@@ -288,7 +288,7 @@ void App::run()
             // Change format of the image
             if (make_gray) cvtColor(frame, img_aux, CV_BGR2GRAY);
             else if (use_gpu) cvtColor(frame, img_aux, CV_BGR2BGRA);
-            else img_aux = frame;
+            else frame.copyTo(img_aux);
 
             // Resize image
             if (args.resize_src) resize(img_aux, img, Size(args.width, args.height));
@@ -326,7 +326,6 @@ void App::run()
             putText(img_to_show, "FPS (HOG only): " + hogWorkFps(), Point(5, 65), FONT_HERSHEY_SIMPLEX, 1., Scalar(255, 100, 0), 2);
             putText(img_to_show, "FPS (total): " + workFps(), Point(5, 105), FONT_HERSHEY_SIMPLEX, 1., Scalar(255, 100, 0), 2);
             imshow("opencv_gpu_hog", img_to_show);
-            handleKey((char)waitKey(3));
 
             if (args.src_is_video || args.src_is_camera) vc >> frame;
 
@@ -341,9 +340,14 @@ void App::run()
                     if (!video_writer.isOpened())
                         throw std::runtime_error("can't create video writer");
                 }
-                cvtColor(img_to_show, img, CV_BGRA2BGR);
+
+                if (make_gray) cvtColor(img_to_show, img, CV_GRAY2BGR);
+                else cvtColor(img_to_show, img, CV_BGRA2BGR);
+
                 video_writer << img;
             }
+
+            handleKey((char)waitKey(3));
         }
     }
 }