Added param -http_port 8090 to show MJPEG-stream in the WebBrowser (Chrome/Firefox)

src/http_stream.cpp

@@ -0,0 +1,197 @@
+#ifdef OPENCV
+//
+// a single-threaded, multi client(using select), debug webserver - streaming out mjpg.
+//  on win, _WIN32 has to be defined, must link against ws2_32.lib (socks on linux are for free)
+//
+
+//
+// socket related abstractions:
+//
+#ifdef _WIN32  
+#pragma comment(lib, "ws2_32.lib")
+#include <winsock.h>
+#include <windows.h>
+#include <time.h>
+#define PORT        unsigned long
+#define ADDRPOINTER   int*
+struct _INIT_W32DATA
+{
+	WSADATA w;
+	_INIT_W32DATA() { WSAStartup(MAKEWORD(2, 1), &w); }
+} _init_once;
+#else       /* ! win32 */
+#include <unistd.h>
+#include <sys/time.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netdb.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
+#define PORT        unsigned short
+#define SOCKET    int
+#define HOSTENT  struct hostent
+#define SOCKADDR    struct sockaddr
+#define SOCKADDR_IN  struct sockaddr_in
+#define ADDRPOINTER  unsigned int*
+#define INVALID_SOCKET -1
+#define SOCKET_ERROR   -1
+#endif /* _WIN32 */
+
+#include <vector>
+#include <iostream>
+using std::cerr;
+using std::endl;
+
+#include "opencv2/opencv.hpp"
+using namespace cv;
+
+#include "http_stream.h"
+
+
+class MJPGWriter
+{
+	SOCKET sock;
+	SOCKET maxfd;
+	fd_set master;
+	int timeout; // master sock timeout, shutdown after timeout millis.
+	int quality; // jpeg compression [1..100]
+
+	int _write(int sock, char *s, int len)
+	{
+		if (len < 1) { len = strlen(s); }
+		return ::send(sock, s, len, 0);
+	}
+
+public:
+
+	MJPGWriter(int port = 0, int _timeout = 200000, int _quality = 30)
+		: sock(INVALID_SOCKET)
+		, timeout(_timeout)
+		, quality(_quality)
+	{
+		FD_ZERO(&master);
+		if (port)
+			open(port);
+	}
+
+	~MJPGWriter()
+	{
+		release();
+	}
+
+	bool release()
+	{
+		if (sock != INVALID_SOCKET)
+			::shutdown(sock, 2);
+		sock = (INVALID_SOCKET);
+		return false;
+	}
+
+	bool open(int port)
+	{
+		sock = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
+
+		SOCKADDR_IN address;
+		address.sin_addr.s_addr = INADDR_ANY;
+		address.sin_family = AF_INET;
+		address.sin_port = htons(port);	// ::htons(port);
+		if (::bind(sock, (SOCKADDR*)&address, sizeof(SOCKADDR_IN)) == SOCKET_ERROR)
+		{
+			cerr << "error : couldn't bind sock " << sock << " to port " << port << "!" << endl;
+			return release();
+		}
+		if (::listen(sock, 10) == SOCKET_ERROR)
+		{
+			cerr << "error : couldn't listen on sock " << sock << " on port " << port << " !" << endl;
+			return release();
+		}
+		FD_ZERO(&master);
+		FD_SET(sock, &master);
+		maxfd = sock;
+		return true;
+	}
+
+	bool isOpened()
+	{
+		return sock != INVALID_SOCKET;
+	}
+
+	bool write(const Mat & frame)
+	{
+		fd_set rread = master;
+		struct timeval to = { 0,timeout };
+		if (::select(maxfd+1, &rread, NULL, NULL, &to) <= 0)
+			return true; // nothing broken, there's just noone listening
+
+		std::vector<uchar> outbuf;
+		std::vector<int> params;
+		params.push_back(IMWRITE_JPEG_QUALITY);
+		params.push_back(quality);
+		cv::imencode(".jpg", frame, outbuf, params);
+		unsigned int outlen = outbuf.size();
+
+#ifdef _WIN32 
+		for (unsigned i = 0; i<rread.fd_count; i++)
+		{
+			int addrlen = sizeof(SOCKADDR);
+			SOCKET s = rread.fd_array[i];    // fd_set on win is an array, while ...
+#else         
+		for (int s = 0; s<=maxfd; s++)
+		{
+			socklen_t addrlen = sizeof(SOCKADDR);
+			if (!FD_ISSET(s, &rread))      // ... on linux it's a bitmask ;)
+				continue;
+#endif                   
+			if (s == sock) // request on master socket, accept and send main header.
+			{
+				SOCKADDR_IN address = { 0 };
+				SOCKET      client = ::accept(sock, (SOCKADDR*)&address, &addrlen);
+				if (client == SOCKET_ERROR)
+				{
+					cerr << "error : couldn't accept connection on sock " << sock << " !" << endl;
+					return false;
+				}
+				maxfd = (maxfd>client ? maxfd : client);
+				FD_SET(client, &master);
+				_write(client, "HTTP/1.0 200 OK\r\n", 0);
+				_write(client,
+					"Server: Mozarella/2.2\r\n"
+					"Accept-Range: bytes\r\n"
+					"Connection: close\r\n"
+					"Max-Age: 0\r\n"
+					"Expires: 0\r\n"
+					"Cache-Control: no-cache, private\r\n"
+					"Pragma: no-cache\r\n"
+					"Content-Type: multipart/x-mixed-replace; boundary=mjpegstream\r\n"
+					"\r\n", 0);
+				cerr << "new client " << client << endl;
+			}
+			else // existing client, just stream pix
+			{
+				char head[400];
+				sprintf(head, "--mjpegstream\r\nContent-Type: image/jpeg\r\nContent-Length: %lu\r\n\r\n", outlen);
+				_write(s, head, 0);
+				int n = _write(s, (char*)(&outbuf[0]), outlen);
+				//cerr << "known client " << s << " " << n << endl;
+				if (n < outlen)
+				{
+					cerr << "kill client " << s << endl;
+					::shutdown(s, 2);
+					FD_CLR(s, &master);
+				}
+			}
+		}
+		return true;
+	}
+};
+
+
+
+void send_mjpeg(IplImage* ipl, int port, int timeout, int quality) {
+	static MJPGWriter wri(port, timeout, quality);
+	cv::Mat mat = cv::cvarrToMat(ipl);
+	wri.write(mat);
+	std::cout << " MJPEG-stream sent. \n";
+}
+
+#endif	// OPENCV