X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=server%2Fimage.c;h=dcc122d28f27ce323eb525217d3a83f6b36e0d53;hb=7547dad3aee722136894566bb6802fe6491cbfbe;hp=c7b626352a7ec32662c5e00a8faaafd7a81eba0f;hpb=8dfd2def9f3c84c4d85c073d07912a22e107713f;p=matches%2FMCTX3420.git

diff --git a/server/image.c b/server/image.c
index c7b62635..dcc122d 100644
--- a/server/image.c
+++ b/server/image.c
@@ -3,41 +3,104 @@
 #include "image.h"
 #include <string.h>
 #include <stdio.h>
+#include <pthread.h>
+
+static CvCapture * g_capture = NULL;
+static int g_captureID = -1;
 
 void Image_Handler(FCGIContext * context, char * params)
 {
-	static CvCapture * capture = NULL;
-	if (capture == NULL) {
-		capture = cvCreateCameraCapture(0);
-		//limit resolution to work on bbb
-		cvSetCaptureProperty(capture, CV_CAP_PROP_FRAME_WIDTH, 352);
-		cvSetCaptureProperty(capture, CV_CAP_PROP_FRAME_HEIGHT, 288);
-	}
-	static int p[] = {CV_IMWRITE_JPEG_QUALITY, 100, 0};
 
-	IplImage * frame = cvQueryFrame(capture);
-	assert(frame != NULL);
+	int num = 0, width = 1600, height = 1200;	// Set Default values
+	FCGIValue val[] = {
+		{"num", &num, FCGI_INT_T},
+		{"width", &width, FCGI_INT_T},
+		{"height", &height, FCGI_INT_T}
+	};
+	if (!FCGI_ParseRequest(context, params, val, 3))	// Populate val
+		return;
+	// Ensure the camera id is 0 or 1. Even though we plan to only have 1 camera attached at a time, this will allow 2. increase
+	else if (num < 0 || num > 1) {				
+		FCGI_RejectJSON(context, "Invalid capture number");
+		return;
+	// Ensure valid widths
+	} else if (width <= 0 || height <= 0) {
+		FCGI_RejectJSON(context, "Invalid width/height");
+		return;
+	}
+	
+	IplImage * src = NULL;   // Source Image
+	CvMat * encoded = NULL;   	// Encoded Image
 
-//        CvMat stub;
- //       CvMat * background = cvGetMat(frame, &stub, 0, 0);
+	Camera_GetImage( num, width, height ,&src); 
 
-//	CvMat *cv8u = cvCreateMat(frame->width, frame->height, CV_8U);
-//	double min, max;
-//	CvPoint a,b;	
-//	cvMinMaxLoc(background, &min, &max, &a, &b, 0);
-	
-//	double ccscale = 255.0/(max-min);
-//	double ccshift = -min;
-	//cvCvtScale(frame, cv8u, ccscale, ccshift);
-	CvMat * jpg = cvEncodeImage(".jpg", frame, p);
+	Log(LOGDEBUG, "About to encode");
+	encoded = cvEncodeImage(".jpg",src,0);
+	Log(LOGDEBUG, "Encoded");
 
-	// Will this work?
 	Log(LOGNOTE, "Sending image!");
 	FCGI_PrintRaw("Content-type: image/jpg\r\n");
 	FCGI_PrintRaw("Cache-Control: no-cache, no-store, must-revalidate\r\n\r\n");
-	//FCGI_PrintRaw("Content-Length: %d", jpg->rows*jpg->cols);
-	FCGI_WriteBinary(jpg->data.ptr,1,jpg->rows*jpg->cols);
+	//FCGI_PrintRaw("Content-Length: %d", g_encoded->rows*g_encoded->cols);
+	FCGI_WriteBinary(encoded->data.ptr,1,encoded->rows*encoded->cols);
 	
-	cvReleaseMat(&jpg);
-	cvReleaseImageHeader(&frame);
+	cvReleaseMat(&encoded);
+	cvReleaseImageHeader(&src);
 }
+	
+/**
+ * Attempts to get an image from a camera
+ * @param num - Camera id
+ * @param width - Width to force
+ * @param height - Height to force
+ * @param image - Pointer to CvMat* to set with result
+ * @returns true on success, false on error 
+ */
+ bool Camera_GetImage(int num, int width, int height,  IplImage ** frame)
+ {
+	Log(LOGDEBUG, "Called with arguments num=%d width=%d height=%d frame=%p", num,width,height, frame);
+	static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER; // Need to use a mutex to ensure 2 captures are not open at once
+	pthread_mutex_lock(&mutex);
+	bool result = false;
+
+	if( g_capture == NULL)
+	{
+		g_capture = cvCreateCameraCapture(num);
+		g_captureID = num;
+	}
+	else if( num != g_captureID)
+	{
+		cvReleaseCapture(&g_capture);
+		g_capture = cvCreateCameraCapture(num);	
+		g_captureID = num;
+	}
+
+	if (g_capture != NULL)
+	{
+
+		cvSetCaptureProperty(g_capture, CV_CAP_PROP_FRAME_WIDTH, width);
+		cvSetCaptureProperty(g_capture, CV_CAP_PROP_FRAME_HEIGHT, height);
+
+		*frame = cvQueryFrame(g_capture);
+		result = (*frame != NULL);
+
+	//cvShowImage("display", *image);
+	//cvWaitKey(0); 
+	//cvSaveImage("test.jpg",*image,0);
+		
+		Log(LOGDEBUG, "At end of mutex");
+	}
+
+	pthread_mutex_unlock(&mutex);	//Close the mutex
+
+	//NOTE: Never have a "return" statement before the mutex is unlocked; it causes deadlocks!
+	return result;
+}
+
+void Image_Cleanup()
+{
+	// Release the capture and IplImage pointers
+	//cvReleaseImageHeader(&g_frame);
+	cvReleaseCapture(&g_capture);
+}
+