Added selection of the video device by the index (web cam).

MasteringOpenCV · Apr 26, 2016 · 3b6ac06 · 3b6ac06
1 parent 55a9299
commit 3b6ac06
Showing 1 changed file with 27 additions and 5 deletions.
diff --git a/Chapter3_MarkerlessAR/src/main.cpp b/Chapter3_MarkerlessAR/src/main.cpp
@@ -85,11 +85,33 @@ int main(int argc, const char * argv[])
         else 
         {
             cv::VideoCapture cap;
-            if (cap.open(input))
-            {
-                processVideo(patternImage, calibration, cap);
-            }
-        }
+			if (cap.open(input))
+			{
+				processVideo(patternImage, calibration, cap);
+			}
+			if (cap.open(input))
+			{
+				processVideo(patternImage, calibration, cap);
+			}
+			else
+			{
+				int deviceIndex;
+				try
+				{
+					deviceIndex = std::stoi(input);
+				}
+				catch (std::invalid_argument e)
+				{
+					std::cerr << "Can't parse device index \"" << input << "\" as an integer value!" << std::endl;
+					return 1;
+				}
+
+				if (cap.open(deviceIndex))
+				{
+					processVideo(patternImage, calibration, cap);
+				}
+			}
+		}
     }
     else
     {