Merge pull request #73 from lobocobra/patch-1

Fix error in start script

Author: poljvd

bin/hyperion.init.sh

@@ -7,22 +7,27 @@ DAEMON=hyperiond
 DAEMONOPTS="/etc/hyperion.config.json"
 DAEMON_PATH="/usr/bin"
 
-NAME=$DEAMON
+NAME=$DAEMON
 DESC="Hyperion ambilight server"
 PIDFILE=/var/run/$NAME.pid
 SCRIPTNAME=/etc/init.d/$NAME
-
 case "$1" in
 start)
-	printf "%-50s" "Starting $NAME..."
-	cd $DAEMON_PATH
-	PID=`$DAEMON $DAEMONOPTS > /dev/null 2>&1 & echo $!`
-	#echo "Saving PID" $PID " to " $PIDFILE
-        if [ -z $PID ]; then
-            printf "%s\n" "Fail"
+        if [ $(pgrep -l $NAME |wc -l) = 1 ]
+        then
+                printf  "%-50s\n" "Already running..."
+                exit 1
         else
-            echo $PID > $PIDFILE
-            printf "%s\n" "Ok"
+                printf "%-50s" "Starting $NAME..."
+                cd $DAEMON_PATH
+                PID=`$DAEMON $DAEMONOPTS > /dev/null 2>&1 & echo $!`
+                #echo "Saving PID" $PID " to " $PIDFILE
+                if [ -z $PID ]; then
+                    printf "%s\n" "Fail"
+                else
+                    echo $PID > $PIDFILE
+                    printf "%s\n" "Ok"
+                fi
         fi
 ;;
 status)
@@ -39,24 +44,31 @@ status)
         fi
 ;;
 stop)
-        printf "%-50s" "Stopping $NAME"
-            PID=`cat $PIDFILE`
-            cd $DAEMON_PATH
-        if [ -f $PIDFILE ]; then
-            kill -HUP $PID
-            printf "%s\n" "Ok"
-            rm -f $PIDFILE
+        if [ -f $PIDFILE ]
+        then
+                printf "%-50s" "Stopping $NAME"
+                    PID=`cat $PIDFILE`
+                    cd $DAEMON_PATH
+                if [ -f $PIDFILE ]; then
+                    kill -HUP $PID
+                    printf "%s\n" "Ok"
+                    rm -f $PIDFILE
+                else
+                    printf "%s\n" "pidfile not found"
+                fi
         else
-            printf "%s\n" "pidfile not found"
+                printf "%-50s\n" "No PID file $NAME not running?"
         fi
 ;;
 
 restart)
-  	$0 stop
-  	$0 start
+        $0 stop
+        $0 start
 ;;
 
 *)
         echo "Usage: $0 {status|start|stop|restart}"
         exit 1
 esac
+
+exit 0

include/grabber/PixelFormat.h

@@ -0,0 +1,36 @@
+#pragma once
+
+#include <string>
+#include <algorithm>
+
+/**
+ * Enumeration of the possible pixel formats the grabber can be set to
+ */
+enum PixelFormat {
+	PIXELFORMAT_YUYV,
+	PIXELFORMAT_UYVY,
+	PIXELFORMAT_RGB32,
+	PIXELFORMAT_NO_CHANGE
+};
+
+inline PixelFormat parsePixelFormat(std::string pixelFormat)
+{
+	// convert to lower case
+	std::transform(pixelFormat.begin(), pixelFormat.end(), pixelFormat.begin(), ::tolower);
+
+	if (pixelFormat == "yuyv")
+	{
+		return PIXELFORMAT_YUYV;
+	}
+	else if (pixelFormat == "uyvy")
+	{
+		return PIXELFORMAT_UYVY;
+	}
+	else if (pixelFormat == "rgb32")
+	{
+		return PIXELFORMAT_RGB32;
+	}
+
+	// return the default NO_CHANGE
+	return PIXELFORMAT_NO_CHANGE;
+}

include/grabber/V4L2Grabber.h

@@ -15,6 +15,7 @@
 
 // grabber includes
 #include <grabber/VideoStandard.h>
+#include <grabber/PixelFormat.h>
 
 /// Capture class for V4L2 devices
 ///
@@ -26,7 +27,7 @@ class V4L2Grabber : public QObject
 public:
 	V4L2Grabber(const std::string & device,
 			int input,
-			VideoStandard videoStandard,
+			VideoStandard videoStandard, PixelFormat pixelFormat,
 			int width,
 			int height,
 			int frameDecimation,
@@ -104,7 +105,7 @@ private slots:
 	int _fileDescriptor;
 	std::vector<buffer> _buffers;
 
-	uint32_t _pixelFormat;
+	PixelFormat _pixelFormat;
 	int _width;
 	int _height;
 	int _cropLeft;

include/grabber/V4L2Wrapper.h

@@ -18,6 +18,7 @@ class V4L2Wrapper : public QObject
 	V4L2Wrapper(const std::string & device,
 			int input,
 			VideoStandard videoStandard,
+			PixelFormat pixelFormat,
 			int width,
 			int height,
 			int frameDecimation,

libsrc/grabber/v4l2/CMakeLists.txt

@@ -9,6 +9,7 @@ SET(V4L2_QT_HEADERS
 
 SET(V4L2_HEADERS
 		${CURRENT_HEADER_DIR}/VideoStandard.h
+		${CURRENT_HEADER_DIR}/PixelFormat.h
 )
 
 SET(V4L2_SOURCES

libsrc/grabber/v4l2/V4L2Grabber.cpp

@@ -39,6 +39,7 @@ static void yuv2rgb(uint8_t y, uint8_t u, uint8_t v, uint8_t & r, uint8_t & g, u
 V4L2Grabber::V4L2Grabber(const std::string & device,
 		int input,
 		VideoStandard videoStandard,
+		PixelFormat pixelFormat,
 		int width,
 		int height,
 		int frameDecimation,
@@ -48,7 +49,7 @@ V4L2Grabber::V4L2Grabber(const std::string & device,
 	_ioMethod(IO_METHOD_MMAP),
 	_fileDescriptor(-1),
 	_buffers(),
-	_pixelFormat(0),
+	_pixelFormat(pixelFormat),
 	_width(width),
 	_height(height),
 	_cropLeft(0),
@@ -380,17 +381,25 @@ void V4L2Grabber::init_device(VideoStandard videoStandard, int input)
 		throw_errno_exception("VIDIOC_G_FMT");
 	}
 
-	// check pixel format
-	switch (fmt.fmt.pix.pixelformat)
+	// set the requested pixel format
+	switch (_pixelFormat)
 	{
-	case V4L2_PIX_FMT_UYVY:
-	case V4L2_PIX_FMT_YUYV:
-		_pixelFormat = fmt.fmt.pix.pixelformat;
+	case PIXELFORMAT_UYVY:
+		fmt.fmt.pix.pixelformat = V4L2_PIX_FMT_UYVY;
+		break;
+	case PIXELFORMAT_YUYV:
+		fmt.fmt.pix.pixelformat = V4L2_PIX_FMT_YUYV;
 		break;
+	case PIXELFORMAT_RGB32:
+		fmt.fmt.pix.pixelformat = V4L2_PIX_FMT_RGB32;
+		break;
+	case PIXELFORMAT_NO_CHANGE:
 	default:
-		throw_exception("Only pixel formats UYVY and YUYV are supported");
+		// No change to device settings
+		break;
 	}
 
+	// set the requested withd and height
 	if (_width > 0 || _height > 0)
 	{
 		if (_width > 0)
@@ -402,19 +411,38 @@ void V4L2Grabber::init_device(VideoStandard videoStandard, int input)
 		{
 			fmt.fmt.pix.height = _height;
 		}
+	}
 
-		// set the settings
-		if (-1 == xioctl(VIDIOC_S_FMT, &fmt))
-		{
-			throw_errno_exception("VIDIOC_S_FMT");
-		}
+	// set the settings
+	if (-1 == xioctl(VIDIOC_S_FMT, &fmt))
+	{
+		throw_errno_exception("VIDIOC_S_FMT");
+	}
 
-		// get the format settings again
-		// (the size may not have been accepted without an error)
-		if (-1 == xioctl(VIDIOC_G_FMT, &fmt))
-		{
-			throw_errno_exception("VIDIOC_G_FMT");
-		}
+	// get the format settings again
+	// (the size may not have been accepted without an error)
+	if (-1 == xioctl(VIDIOC_G_FMT, &fmt))
+	{
+		throw_errno_exception("VIDIOC_G_FMT");
+	}
+
+	// check pixel format
+	switch (fmt.fmt.pix.pixelformat)
+	{
+	case V4L2_PIX_FMT_UYVY:
+		_pixelFormat = PIXELFORMAT_UYVY;
+		std::cout << "V4L2 pixel format=UYVY" << std::endl;
+		break;
+	case V4L2_PIX_FMT_YUYV:
+		_pixelFormat = PIXELFORMAT_YUYV;
+		std::cout << "V4L2 pixel format=YUYV" << std::endl;
+		break;
+	case V4L2_PIX_FMT_RGB32:
+		_pixelFormat = PIXELFORMAT_RGB32;
+		std::cout << "V4L2 pixel format=RGB32" << std::endl;
+		break;
+	default:
+		throw_exception("Only pixel formats UYVY, YUYV, and RGB32 are supported");
 	}
 
 	// store width & height
@@ -680,27 +708,40 @@ void V4L2Grabber::process_image(const uint8_t * data)
 	{
 		for (int xSource = _cropLeft + _horizontalPixelDecimation/2, xDest = 0; xSource < width - _cropRight; xSource += _horizontalPixelDecimation, ++xDest)
 		{
-			int index = (_width * ySource + xSource) * 2;
-			uint8_t y = 0;
-			uint8_t u = 0;
-			uint8_t v = 0;
+			ColorRgb & rgb = image(xDest, yDest);
 
 			switch (_pixelFormat)
 			{
-			case V4L2_PIX_FMT_UYVY:
-				y = data[index+1];
-				u = (xSource%2 == 0) ? data[index  ] : data[index-2];
-				v = (xSource%2 == 0) ? data[index+2] : data[index  ];
+			case PIXELFORMAT_UYVY:
+				{
+					int index = (_width * ySource + xSource) * 2;
+					uint8_t y = data[index+1];
+					uint8_t u = (xSource%2 == 0) ? data[index  ] : data[index-2];
+					uint8_t v = (xSource%2 == 0) ? data[index+2] : data[index  ];
+					yuv2rgb(y, u, v, rgb.red, rgb.green, rgb.blue);
+				}
+				break;
+			case PIXELFORMAT_YUYV:
+				{
+					int index = (_width * ySource + xSource) * 2;
+					uint8_t y = data[index];
+					uint8_t u = (xSource%2 == 0) ? data[index+1] : data[index-1];
+					uint8_t v = (xSource%2 == 0) ? data[index+3] : data[index+1];
+					yuv2rgb(y, u, v, rgb.red, rgb.green, rgb.blue);
+				}
 				break;
-			case V4L2_PIX_FMT_YUYV:
-				y = data[index];
-				u = (xSource%2 == 0) ? data[index+1] : data[index-1];
-				v = (xSource%2 == 0) ? data[index+3] : data[index+1];
+			case PIXELFORMAT_RGB32:
+				{
+					int index = (_width * ySource + xSource) * 4;
+					rgb.red   = data[index+1];
+					rgb.green = data[index+2];
+					rgb.blue  = data[index+3];
+				}
+				break;
+			default:
+				// this should not be possible
 				break;
 			}
-
-			ColorRgb & rgb = image(xDest, yDest);
-			yuv2rgb(y, u, v, rgb.red, rgb.green, rgb.blue);
 		}
 	}
 

libsrc/grabber/v4l2/V4L2Wrapper.cpp

@@ -7,6 +7,7 @@
 V4L2Wrapper::V4L2Wrapper(const std::string &device,
 		int input,
 		VideoStandard videoStandard,
+		PixelFormat pixelFormat,
 		int width,
 		int height,
 		int frameDecimation,
@@ -21,6 +22,7 @@ V4L2Wrapper::V4L2Wrapper(const std::string &device,
 	_grabber(device,
 			input,
 			videoStandard,
+			pixelFormat,
 			width,
 			height,
 			frameDecimation,

src/hyperion-v4l2/CMakeLists.txt

@@ -21,6 +21,7 @@ set(Hyperion_V4L2_QT_HEADERS
 
 set(Hyperion_V4L2_HEADERS
 	VideoStandardParameter.h
+	PixelFormatParameter.h
 	ProtoConnection.h
 )
 

src/hyperion-v4l2/PixelFormatParameter.h

@@ -0,0 +1,43 @@
+// getoptPlusPLus includes
+#include <getoptPlusPlus/getoptpp.h>
+
+// grabber includes
+#include <grabber/PixelFormat.h>
+
+using namespace vlofgren;
+
+/// Data parameter for the pixel format
+typedef vlofgren::PODParameter<PixelFormat> PixelFormatParameter;
+
+namespace vlofgren {
+	/// Translates a string (as passed on the commandline) to a pixel format
+	///
+	/// @param[in] s The string (as passed on the commandline)
+	/// @return The pixel format
+	/// @throws Parameter::ParameterRejected If the string did not result in a pixel format
+	template<>
+	PixelFormat PixelFormatParameter::validate(const std::string& s) throw (Parameter::ParameterRejected)
+	{
+		QString input = QString::fromStdString(s).toLower();
+
+		if (input == "yuyv")
+		{
+			return PIXELFORMAT_YUYV;
+		}
+		else if (input == "uyvy")
+		{
+			return PIXELFORMAT_UYVY;
+		}
+		else if (input == "rgb32")
+		{
+			return PIXELFORMAT_RGB32;
+		}
+		else if (input == "no-change")
+		{
+			return PIXELFORMAT_NO_CHANGE;
+		}
+
+		throw Parameter::ParameterRejected("Invalid value for pixel format. Valid values are: YUYV, UYVY, RGB32, and NO-CHANGE");
+		return PIXELFORMAT_NO_CHANGE;
+	}
+}