X264实时对USB摄像头编码（H264）

#include <stdio.h> #include <iostream> #include "stdafx.h" #include "stdint.h" #include "opencv2/highgui/highgui.hpp" #include "opencv2/imgproc/imgproc.hpp" extern "C" { #include "x264.h" #include "x264_config.h" }; using namespace cv; using namespace std; int main(int argc, char* argv[]) { VideoCapture capture(0); if (!capture.isOpened()) { cout << "Cannot open the video cam" << endl; return -1; } int w = capture.get(CV_CAP_PROP_FRAME_WIDTH); int h = capture.get(CV_CAP_PROP_FRAME_HEIGHT); int yuv_bufLen = w * h * 3 / 2; size_t yuv_size = w * h * 3 / 2; x264_t *encoder; x264_picture_t pic_in, pic_out; uint8_t *yuv_buffer; x264_param_t param; x264_param_default_preset(&param, "veryfast", "animation"); param.i_width = w; param.i_height = h; x264_param_apply_profile(&param, "baseline"); encoder = x264_encoder_open(&param); x264_picture_alloc(&pic_in, X264_CSP_I420, w, h); yuv_buffer = (uint8_t*)malloc(yuv_size); pic_in.img.plane[0] = yuv_buffer; pic_in.img.plane[1] = pic_in.img.plane[0] + w * h; pic_in.img.plane[2] = pic_in.img.plane[1] + w * h / 4; int64_t i_pts = 0; x264_nal_t *nals; int nnal; FILE *fp_out = fopen("test.h264", "wb"); if (!fp_out) { printf("Could not open output 264 file\n"); return -1; } FILE* pFileOut = fopen("test.yuv", "w+"); if (!pFileOut) { printf("Could not open input yuv file\n"); return -1; } cout << "Frame size : " << w << " x " << h << endl; namedWindow("opencamera", CV_WINDOW_AUTOSIZE); Mat frame; while (1) { capture >> frame; imshow("opencamera", frame); waitKey(1); cv::Mat yuvImg; cv::cvtColor(frame, yuvImg, CV_BGR2YUV_I420); memcpy(yuv_buffer, yuvImg.data, yuv_bufLen*sizeof(unsigned char)); fwrite(yuv_buffer, yuv_bufLen*sizeof(unsigned char), 1, pFileOut); pic_in.i_pts = i_pts++; x264_encoder_encode(encoder, &nals, &nnal, &pic_in, &pic_out); x264_nal_t *nal; for (nal = nals; nal < nals + nnal; nal++) { fwrite(nal->p_payload, 1, nal->i_payload, fp_out); } } x264_encoder_close(encoder); free(yuv_buffer); return 0; }