基于嵌入式Linux的视频采集系统20-----源程序----stlmain.cpp

本文来自：
http://blog.chinaunix.net/uid-23093301-id-86409.html

#include "SDL.h"   /* All SDL App's need this */

#include <stdio.h>

#include <iostream>

#include <string>

using namespace std;

#include <ctime>

#include "decoder.h"

#include "encoder.h"

#include "log.h"

#include "rtp_receive.h"

#include "singleton.h"

int base_port = 5000;

const string dest_host = "192.168.1.6";

const int dest_port = 6000;

SDL_Surface *screen;

SDL_Overlay *overlay;

SDL_Event event;

SDL_Rect rect;

const int WIDTH = 640;

const int HEIGHT = 480;

static int XDIM=640,YDIM=480;

int SHXDIM =640;//display size

int SHYDIM =480;

int paused=0;

int resized=0;

AVFrame   pict;

//AVPicture pict;

CDecoder aDecoder;

CEncoder aEncoder("target.mp4",WIDTH,HEIGHT);

CEncoder aRawEncoder("target.avi",WIDTH,HEIGHT);

int ret;

AVFrame* mainFrame;

AVFrame* rawFrame;

FILE* ftrRawDta;

void show_rtp_pict()

{

string rtp_data;

SDL_LockSurface(screen);

    SDL_LockYUVOverlay(overlay);

        

    pict.data[0] = overlay->pixels[0];

    pict.data[1] = overlay->pixels[2];

    pict.data[2] = overlay->pixels[1];

    pict.linesize[0] = overlay->pitches[0];

    pict.linesize[1] = overlay->pitches[2];

    pict.linesize[2] = overlay->pitches[1];

    //cout << "begin to read...\n";

    rtp_data = singleton<CRtpReceive>::instance().pop_frame();

    fwrite(rtp_data.data(),rtp_data.size(),1,ftrRawDta);

    if(rtp_data.size()<6000)

{

     cout << "begin to read size="<<rtp_data.size()<<"\n";

goto end;

}

    aDecoder.push_frame((void*)rtp_data.data(),rtp_data.size());

//cin >> cmd;

ret = aDecoder.try_show(&pict,640,480,mainFrame,rawFrame);

if(ret<0) goto end;

//ret = aDecoder.try_show(mainFrame,640,480);

//if(ret<0) goto end;

//cin >> cmd;

//if(num!=0)continue;

aEncoder.try_encode(mainFrame);

aRawEncoder.try_encode(rawFrame);

end:

SDL_UnlockYUVOverlay(overlay);

    SDL_UnlockSurface(screen);

    

    SDL_DisplayYUVOverlay(overlay, &rect);

}

int main(int argc, char *argv[]) {

if(argc>1)

base_port = atoi(argv[1]);

cout <<"RTP info:\n"

<<"\tRTP base port:"<<base_port<<"\n"

<<"\tRTP dest host:"<<dest_host<<"\n"

<<"\tRTP dest port:"<<dest_port<<"\n"

<<"\n";

cout <<"SDL info:\n"

<<"\tSDL window width:640\n"

<<"\tSDL windor height:480\n"

<<"\n";

cout <<"FFMPEG info:\n"

<<"\toutput filename:target.mp4\n"

<<"\tencoder:H264\n"

<<"\n";

cout <<"main begin base port="<<base_port<<"...\n";

singleton<CRtpReceive>::instance().start(base_port,dest_host,dest_port);

mainFrame = aDecoder.do_alloc_picture(PIX_FMT_YUV420P,WIDTH,HEIGHT);

rawFrame = aDecoder.do_alloc_picture(PIX_FMT_YUV420P,WIDTH,HEIGHT);

const char* rawdata_filename = "rawdata.mpeg";

ftrRawDta = fopen(rawdata_filename,"wb");

if(ftrRawDta == NULL)

{

cout <<"main con't open="<<rawdata_filename<<"...\n";

exit(1);

}

sleep(1);

    printf("Initializing SDL.\n");

    

    /* Initialize defaults, Video and Audio */

    if((SDL_Init(SDL_INIT_VIDEO|SDL_INIT_AUDIO| SDL_INIT_TIMER)==-1)) { 

        printf("Could not initialize SDL: %s.\n", SDL_GetError());

        exit(-1);

    }

    atexit (SDL_Quit);

    printf("SDL initialized.\n");

    

    screen = SDL_SetVideoMode (WIDTH, HEIGHT, 0, SDL_HWSURFACE

                                              | SDL_DOUBLEBUF

                                              | SDL_ANYFORMAT

                                              | SDL_RESIZABLE);

    if (screen == NULL)

    {

        fprintf(stderr, "SDL_SetVideoMode ==>> %s\n", SDL_GetError());

        goto fatal;

    }

    if (0 == (screen->flags & SDL_HWSURFACE))

    {

       fprintf(stderr,"Can't get hardware surface\n");

    }

    SDL_WM_SetCaption ("USB Camera By Breeze", NULL);      

    overlay = SDL_CreateYUVOverlay(XDIM, YDIM, SDL_YV12_OVERLAY, screen);

    if (!overlay)

     {

            fprintf(stderr, "Couldn't create overlay: %s\n", SDL_GetError());

            exit(4);

     }   

    printf("SDL_CreateYUVOverlay info: %dx%dx%d %s %s overlay\n",overlay->w,overlay->h,overlay->planes,

               overlay->hw_overlay?"hardware":"software",

               overlay->format==SDL_YV12_OVERLAY?"YV12":

               overlay->format==SDL_IYUV_OVERLAY?"IYUV":

               overlay->format==SDL_YUY2_OVERLAY?"YUY2":

               overlay->format==SDL_UYVY_OVERLAY?"UYVY":

               overlay->format==SDL_YVYU_OVERLAY?"YVYU":

               "Unknown");

    rect.x=0;

    rect.y=0;

    rect.w=SHXDIM;

    rect.h=SHYDIM;

    

    do{

   while (SDL_PollEvent(&event))

   {

    switch (event.type)

    {

    case SDL_VIDEORESIZE:

    {

             screen=SDL_SetVideoMode(event.resize.w, event.resize.h, 0, SDL_RESIZABLE | SDL_SWSURFACE);

             rect.w=event.resize.w;

             rect.h=event.resize.h;

             if (paused)

              {

                 resized=1;

              }   

    }break;

    case SDL_QUIT:

    {

    aEncoder.finish_coder();

   

    fclose(ftrRawDta);

    goto release_all;

   

    }break;

    }//! end switch 

   

   }//! end while (SDL_PollEvent(&event))

   show_rtp_pict();

   SDL_Delay(10);//! delay 1 ms

   //!printf("Wvent Poll.\n");

    }while(1);

    //cin >>cmd;

    printf("Quiting SDL.\n");

release_all:

    printf("Quiting....\n");

    exit(0);

fatal:

printf("fatal!!,exit\n");

return -1;;

}