From 02328b82ffb90242f5afefac5a774e3a07b0943f Mon Sep 17 00:00:00 2001
From: f4exb <f4exb06@gmail.com>
Date: Fri, 10 Mar 2017 14:28:04 +0100
Subject: [PATCH] ATV modulator: added camera devices scan and selector in the
 GUI

---
 plugins/channeltx/modatv/atvmod.cpp    | 25 +++++++++++++++
 plugins/channeltx/modatv/atvmod.h      | 18 +++++++++++
 plugins/channeltx/modatv/atvmodgui.cpp |  7 +++++
 plugins/channeltx/modatv/atvmodgui.ui  | 42 +++++++++++++++++++++++++-
 4 files changed, 91 insertions(+), 1 deletion(-)

diff --git a/plugins/channeltx/modatv/atvmod.cpp b/plugins/channeltx/modatv/atvmod.cpp
index 2c20413ee..b062ce50f 100644
--- a/plugins/channeltx/modatv/atvmod.cpp
+++ b/plugins/channeltx/modatv/atvmod.cpp
@@ -49,6 +49,7 @@ ATVMod::ATVMod() :
 	m_videoOK(false)
 {
     setObjectName("ATVMod");
+    scanCameras();
 
     m_config.m_outputSampleRate = 1000000;
     m_config.m_inputFrequencyOffset = 0;
@@ -69,6 +70,7 @@ ATVMod::ATVMod() :
 ATVMod::~ATVMod()
 {
 	if (m_video.isOpened()) m_video.release();
+	releaseCameras();
 }
 
 void ATVMod::configure(MessageQueue* messageQueue,
@@ -567,4 +569,27 @@ void ATVMod::seekVideoFileStream(int seekPercentage)
     }
 }
 
+void ATVMod::scanCameras()
+{
+	for (int i = 0; i < 4; i++)
+	{
+		ATVCamera newCamera;
+		m_cameras.push_back(newCamera);
+		m_cameras.back().m_cameraNumber = i;
+		m_cameras.back().m_camera.open(i);
+
+		if (!m_cameras.back().m_camera.isOpened())
+		{
+			m_cameras.pop_back();
+		}
+	}
+}
+
+void ATVMod::releaseCameras()
+{
+	for (std::vector<ATVCamera>::iterator it = m_cameras.begin(); it != m_cameras.end(); ++it)
+	{
+		if (it->m_camera.isOpened()) it->m_camera.release();
+	}
+}
 
diff --git a/plugins/channeltx/modatv/atvmod.h b/plugins/channeltx/modatv/atvmod.h
index 733bc0c2f..0e98cb86e 100644
--- a/plugins/channeltx/modatv/atvmod.h
+++ b/plugins/channeltx/modatv/atvmod.h
@@ -20,6 +20,8 @@
 #include <QObject>
 #include <QMutex>
 
+#include <vector>
+
 #include <opencv2/core/core.hpp>
 #include <opencv2/highgui/highgui.hpp>
 
@@ -59,6 +61,12 @@ public:
 		ATVModulationFM
     } ATVModulation;
 
+    typedef struct
+    {
+    	cv::VideoCapture m_camera;
+    	int m_cameraNumber;
+    } ATVCamera;
+
     class MsgConfigureImageFileName : public Message
     {
         MESSAGE_CLASS_DECLARATION
@@ -206,6 +214,12 @@ public:
 
     Real getMagSq() const { return m_movingAverage.average(); }
 
+    void getCameraNumbers(std::vector<int>& numbers) {
+    	for (std::vector<ATVCamera>::iterator it = m_cameras.begin(); it != m_cameras.end(); ++it) {
+    		numbers.push_back(it->m_cameraNumber);
+    	}
+    }
+
     static int getSampleRateUnits(ATVStd std);
 
 signals:
@@ -361,6 +375,8 @@ private:
     bool m_videoEOF;             //!< current video has reached end of file
     bool m_videoOK;
 
+    std::vector<ATVCamera> m_cameras; //!< vector of available cameras
+
     static const float m_blackLevel;
     static const float m_spanLevel;
     static const int m_levelNbSamples;
@@ -378,6 +394,8 @@ private:
     void calculateVideoSizes();
     void resizeVideo();
     void seekVideoFileStream(int seekPercentage);
+    void scanCameras();
+    void releaseCameras();
 
     inline void pullImageLine(Real& sample)
     {
diff --git a/plugins/channeltx/modatv/atvmodgui.cpp b/plugins/channeltx/modatv/atvmodgui.cpp
index b88601d88..a3cff2652 100644
--- a/plugins/channeltx/modatv/atvmodgui.cpp
+++ b/plugins/channeltx/modatv/atvmodgui.cpp
@@ -356,6 +356,13 @@ ATVModGUI::ATVModGUI(PluginAPI* pluginAPI, DeviceSinkAPI *deviceAPI, QWidget* pa
 
 	connect(m_atvMod->getOutputMessageQueue(), SIGNAL(messageEnqueued()), this, SLOT(handleSourceMessages()));
     connect(m_atvMod, SIGNAL(levelChanged(qreal, qreal, int)), ui->volumeMeter, SLOT(levelChanged(qreal, qreal, int)));
+
+    std::vector<int> cameraNumbers;
+    m_atvMod->getCameraNumbers(cameraNumbers);
+
+    for (std::vector<int>::iterator it = cameraNumbers.begin(); it != cameraNumbers.end(); ++it) {
+        ui->camSelect->addItem(tr("%1").arg(*it));
+    }
 }
 
 ATVModGUI::~ATVModGUI()
diff --git a/plugins/channeltx/modatv/atvmodgui.ui b/plugins/channeltx/modatv/atvmodgui.ui
index 5f9bd5979..335b8ce54 100644
--- a/plugins/channeltx/modatv/atvmodgui.ui
+++ b/plugins/channeltx/modatv/atvmodgui.ui
@@ -7,7 +7,7 @@
     <x>0</x>
     <y>0</y>
     <width>342</width>
-    <height>366</height>
+    <height>364</height>
    </rect>
   </property>
   <property name="sizePolicy">
@@ -284,6 +284,12 @@
           <verstretch>0</verstretch>
          </sizepolicy>
         </property>
+        <property name="minimumSize">
+         <size>
+          <width>0</width>
+          <height>32</height>
+         </size>
+        </property>
         <property name="toolTip">
          <string>video signal level in % of 0:1 range</string>
         </property>
@@ -639,6 +645,40 @@
       </item>
      </layout>
     </item>
+    <item>
+     <widget class="Line" name="line_6">
+      <property name="orientation">
+       <enum>Qt::Horizontal</enum>
+      </property>
+     </widget>
+    </item>
+    <item>
+     <layout class="QHBoxLayout" name="camLayout">
+      <item>
+       <widget class="QLabel" name="camLabel">
+        <property name="text">
+         <string>Cam</string>
+        </property>
+       </widget>
+      </item>
+      <item>
+       <widget class="QComboBox" name="camSelect"/>
+      </item>
+      <item>
+       <spacer name="horizontalSpacer_4">
+        <property name="orientation">
+         <enum>Qt::Horizontal</enum>
+        </property>
+        <property name="sizeHint" stdset="0">
+         <size>
+          <width>40</width>
+          <height>20</height>
+         </size>
+        </property>
+       </spacer>
+      </item>
+     </layout>
+    </item>
    </layout>
   </widget>
  </widget>