95 lines
2.7 KiB
Text
95 lines
2.7 KiB
Text
/*
|
|
* grabFramesCam implementation
|
|
*
|
|
* Copyright (C) Stanislav Serebryakov
|
|
*
|
|
* Released under the GPL version 3.
|
|
*
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
#include <cv.h>
|
|
#include <highgui.h>
|
|
//TODO: flip image flag
|
|
|
|
void usage(char *progName) {
|
|
printf("%s <board-size-x> <board-size-y> <camera-id>\n", progName);
|
|
printf("i.e.: %s 6 4 0\n", progName);
|
|
printf("press space to detect chessboard and (again) to proceed.\n");
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
/* TODO:
|
|
* flags:
|
|
* subpixel
|
|
* camera id
|
|
* flip x and y
|
|
* pmd mode?
|
|
*/
|
|
|
|
if(argc < 4) {
|
|
usage(argv[0]);
|
|
exit(1);
|
|
}
|
|
CvCapture *capture = cvCaptureFromCAM(atoi(argv[3]));
|
|
|
|
IplImage *imgColor = cvQueryFrame(capture);
|
|
IplImage *img = cvCreateImage(cvGetSize(imgColor), 8, 1);
|
|
|
|
int patx = atoi(argv[1]);
|
|
int paty = atoi(argv[2]);
|
|
CvSize patternSize = cvSize(patx, paty);
|
|
|
|
int cornersTotal = patternSize.width * patternSize.height;
|
|
|
|
CvPoint2D32f *corners = (CvPoint2D32f*) malloc(cornersTotal * sizeof(CvPoint2D32f));
|
|
|
|
cvNamedWindow("Camera", 0);
|
|
|
|
int imageCnt = 0;
|
|
bool grabFrame = false;
|
|
|
|
while(1) {
|
|
imgColor = cvQueryFrame(capture);
|
|
|
|
cvFlip(imgColor, 0, -1); // flips image around the x and y axes
|
|
|
|
cvCvtColor(imgColor, img, CV_BGR2GRAY);
|
|
|
|
if(grabFrame) {
|
|
/* ----- Chessboard detection -----
|
|
-------------------------------- */
|
|
int cornersCount; // should be the same for wcam and pmd and equal totalCorners
|
|
|
|
int found = cvFindChessboardCorners(img, patternSize, corners,
|
|
&cornersCount, CV_CALIB_CB_ADAPTIVE_THRESH | CV_CALIB_CB_FILTER_QUADS);
|
|
cvFindCornerSubPix(img, corners, cornersCount, cvSize(11,11), cvSize(-1,-1),
|
|
cvTermCriteria(CV_TERMCRIT_EPS | CV_TERMCRIT_ITER, 30, 0.1));
|
|
|
|
if(found && (cornersCount == cornersTotal)) {
|
|
cvDrawChessboardCorners(imgColor, patternSize, corners, cornersCount, found);
|
|
cvShowImage("Camera", imgColor);
|
|
printf("grab?\n");
|
|
int k = cvWaitKey(0);
|
|
if(k == (int)' ') {
|
|
imageCnt++;
|
|
char *filename = (char*) malloc(11 * sizeof(char));
|
|
sprintf(filename, "image%02i.jpg", imageCnt);
|
|
printf("saved %s.\n", filename);
|
|
cvSaveImage(filename, img);
|
|
grabFrame = false;
|
|
continue;
|
|
}
|
|
}
|
|
}
|
|
|
|
cvShowImage("Camera", imgColor);
|
|
if((int)' ' == cvWaitKey(5)) grabFrame = true;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|