udp_client.cpp

#include <stdio.h>
#include <vector>
#include <opencv2/opencv.hpp>
#include <opencv2/imgcodecs.hpp>
#ifdef RASP_PI
#include <raspicam/raspicam_cv.h>
using namespace raspicam;
#endif
#include "UDPMessenger.hpp"
using namespace cv;
using std::vector;

//udp_client demo
//Grabs an image from the camera and sends it to
//the server using UDPMessenger.
int main(int argc, char* argv[])
{
	if (argc < 3)
	{
		printf("Not enough arguments!\n");
		printf("Usage: ./client [dest] [port]\n");
		return 1;
	}

	UDPMessenger udpmsg((char*)"0", 1701); //ugh
	udpmsg.setDest(argv[1], atoi(argv[2]));
	
#ifdef RASP_PI
	RaspiCam_Cv cam;
	if (!cam.open())
#else	
	VideoCapture cam;
	if (!cam.open(0))
#endif
	{
		printf("Couldn't open camera!\n");
		return 1;
	}

	vector<int> param(2);
	param[0] = IMWRITE_JPEG_QUALITY;
	param[1] = 80;

	while (waitKey(1) != 27)
	{	
		Mat img;

		cam.grab();
		cam.retrieve(img);

		//Encode to jpeg, quality 80%
		vector<uchar> encodeBuffer;
		imencode(".jpg", img, encodeBuffer, param);

		udpmsg.sendPacket((const char*)&encodeBuffer[0], encodeBuffer.size());
		waitKey(16); //wait 16ms
	}
	cam.release();
	return 0;
}