开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 多媒体 > 视频捕捉/采集 > 查看源码
main.cpp
资源名称：handvu-beta3.tar.gz　[点击查看]
上传用户：lijia5631
上传日期：2008-11-10
资源大小：1214k
文件大小：11k
源码类别：

视频捕捉/采集

开发平台：

MultiPlatform

main.cpp：源码内容
  1. #include "main.h"
  3. #include <stdio.h>
  4. #include <stdlib.h>
  5. #include <AR/param.h>
  6. #include <AR/ar.h>
  7. #include <AR/arMulti.h>
  8. #include <AR/video.h>
  9. #include "gsub.h"
  10. #include "sdlut.h"
  11. //#include "util.h"
  12. #include "callbacks.h"
  13. //#include "video.h"
  14. //#include "image.h"
  15. #include "hv_ARtk_demo.h"
  16. #include <sys/stat.h>
  17. #include <errno.h>
  19. using namespace std;
  21. void EXIT_WITH_ERROR ( char *msg );
  23. //-------------------------------------------------------------------------
  24. // global variables
  25. //-------------------------------------------------------------------------
  28. char *programName = "arwin";
  30. static char *single_filename  = "data/single.dat";
  31. static char *multi_filename   = "data/ARmulti.dat";
  32. //static char *multi_filename   = "data/multi.dat";
  33. static char *cparam_filename  = "data/camera_para.dat";
  34. //static char *startup_filename = "data/startup.dat";
  36. #ifdef USE_CG
  37. CGcontext cgContext;
  38. #endif
  40. #ifdef V4L
  41. static char *vconf = "";
  42. #else
  43. static char *vconf = "-mode=640x480_YUV411 -rate=30";
  44. #endif
  46. vector< Marker * > floater_markers;
  47. ARMultiMarkerInfoT *multi_markers;
  48. Marker *current_marker = NULL;
  50. ARParam cparam;
  51. int threshold = 150;
  53. int xsize, ysize;
  54. int windowx, windowy;
  56. double aspectRatio;
  57. GLint viewport[ 4 ];
  58. GLdouble projectionMatrix[ 16 ];
  60. int videopassthrough = 1;
  61. int fullscreen = 0;
  62. int grabinput = 0;
  63. char *displaystr = ":0";
  64. bool fsaa = true;
  66. bool take_snapshots = false;
  67. bool fullsize_snapshots = false;
  68. bool fixed_world = false;
  69. bool reset_world_once = true;
  70. char* conductor_fname = "../config/ARtk.conductor";
  71. char* save_fname_root = NULL;
  72. extern int hv_pointer_dist;
  75. //-------------------------------------------------------------------------
  76. // function prototypes
  77. //-------------------------------------------------------------------------
  80. int main( int, char ** );
  81. static void parseArgs( int, char ** );
  82. static void usage();
  83. static void init();
  84. static void contextInit();
  85. static void displayInit();
  86. static void readFloaters( const char * );
  87. //static void readStartupConfig( const char * );
  89. // dragging
  90. void initObjs();
  91. void ResetWorld();
  93. #ifdef USE_CG
  94. static void cgErrorCallback();
  95. #endif
  98. //-------------------------------------------------------------------------
  99. // startup and init functions
  100. //-------------------------------------------------------------------------
  103. int main( int argc, char **argv )
  104. {
  105.   parseArgs( argc, argv );
  107.   init();
  109.   sdlutDisplayFunc( displayInit );
  110.   sdlutMainLoop();
  112.   cleanup();
  113.   return 0;
  114. }
  117. void parseArgs( int argc, char **argv )
  118. {
  119.   int i = 1;
  120.   while( i < argc )
  121.   {
  122.     if( strncmp( argv[i], "--fullscreen", 13 ) == 0 )
  123.     {
  124.       fullscreen = 1;
  125.       ++i;
  126.     }
  127.     else if( strncmp( argv[i], "--nofullscreen", 15 ) == 0 )
  128.     {
  129.       fullscreen = 0;
  130.       ++i;
  131.     }
  132.     else if( strncmp( argv[i], "--grabinput", 12 ) == 0 )
  133.     {
  134.       grabinput = 1;
  135.       ++i;
  136.     }
  137.     else if( strncmp( argv[i], "--nograbinput", 14 ) == 0 )
  138.     {
  139.       grabinput = 0;
  140.       ++i;
  141.     }
  142.     else if( strncmp( argv[i], "--videopassthrough", 19 ) == 0 )
  143.     {
  144.       videopassthrough = 1;
  145.       ++i;
  146.     }
  147.     else if( strncmp( argv[i], "--novideopassthrough", 19 ) == 0 )
  148.     {
  149.       videopassthrough = 0;
  150.       ++i;
  151.     }
  152.     else if( strncmp( argv[i], "--display", 10 ) == 0 )
  153.     {
  154.       ++i;
  155.       displaystr = argv[i];
  156.       ++i;
  157.     }
  158.     else if( strncmp( argv[i], "--conductor", 12 ) == 0 )
  159.     {
  160.       ++i;
  161.       conductor_fname = argv[i];
  162.       ++i;
  163.     }
  164.     else if( strncmp( argv[i], "--pointerdist", 14 ) == 0 )
  165.     {
  166.       ++i;
  167.       hv_pointer_dist = atoi( argv[i] );
  168.       ++i;
  169.     }
  170.     else if( strncmp( argv[i], "--snapshots", 14 ) == 0 )
  171.     {
  172.       ++i;
  173.       take_snapshots = true;
  174.     }
  175.     else if( strncmp( argv[i], "--fullsnaps", 14 ) == 0 )
  176.     {
  177.       ++i;
  178.       fullsize_snapshots = true;
  179.     }
  180.     else if( strncmp( argv[i], "--fixedworld", 13 ) == 0 )
  181.     {
  182.       ++i;
  183.       fixed_world = true;
  184.       reset_world_once = true;
  185.     }
  186.     else if( strncmp( argv[i], "--threshold", 12 ) == 0 )
  187.     {
  188.       ++i;
  189.       threshold = atoi( argv[i] );
  190.       ++i;
  191.     }
  192.     else if( strncmp( argv[i], "--nofsaa", 9 ) == 0 )
  193.     {
  194.       ++i;
  195.       fsaa = false;
  196.     }
  197.     else
  198.     {
  199.       fprintf( stderr, "ERROR: unrecognized option: %sn", argv[i] );
  200.       fflush( stderr );
  202.       usage();
  203.       exit( 1 );
  204.     }
  205.   }
  206. }
  209. void usage()
  210. {
  211.   fprintf( stderr, "usage: ./hv_ARtk [ --fullscreen ] [ --conductor filename ]" );
  212.   fprintf( stderr, " [ --display :<id> ] [ --nofsaa ] [ --fixedword ] [ --pointerdist n ] [ --threshod n ]n" );
  213.   fflush( stderr );
  214. }
  217. void init()
  218. {
  219.   ARParam wparam;
  221.   if( arVideoOpen( vconf ) < 0 )
  222.     EXIT_WITH_ERROR( "arVdeoOpen" );
  223.   
  224.   if( arVideoInqSize( &xsize, &ysize ) < 0 )
  225.     EXIT_WITH_ERROR( "arVideoInqSize" );
  227.   if( arParamLoad( cparam_filename, 1, &wparam ) < 0 )
  228.     EXIT_WITH_ERROR( "arParamLoad" );
  230.   arParamChangeSize( &wparam, xsize, ysize, &cparam );
  231.   arInitCparam( &cparam );
  233.   // load floaters
  234.   readFloaters( single_filename );
  235.   if( floater_markers.size() == 0 )
  236.     EXIT_WITH_ERROR( "readSingles" );
  238.   // load space registers
  239.   multi_markers = arMultiReadConfigFile( multi_filename );
  240.   if( multi_markers == NULL )
  241.     EXIT_WITH_ERROR( "arMultiReadConfigFile" );
  243.   // for take_snapshots, which can be turned on dynamically
  244.   framePixels = new unsigned char[ xsize * ysize * 3 ];
  246.   contextInit();
  248. #ifdef USE_CG
  249.   // initialize Cg context
  250.   cgSetErrorCallback( cgErrorCallback );
  251.   cgContext = cgCreateContext();
  252. #endif
  254. //  readStartupConfig( startup_filename );
  255.   
  256.   //
  257.   // directory to save snapshots in
  258.   //
  259.   if (save_fname_root==NULL) {
  260.     const char *home = getenv("HOME");
  261.     if (home==NULL) {
  262.       save_fname_root = (char*) alloca(20);
  263.       sprintf(save_fname_root, "/tmp/hv_pics/");
  264.     } else {
  265.       save_fname_root = (char*) alloca(strlen(home)+20);
  266.       sprintf(save_fname_root, "%s/hv_pics", home);
  267.     }
  268.   }
  269.   // create it if it doesn't exist
  270.   struct stat st;
  271.   int error = stat(save_fname_root, &st);
  272. #ifdef HAVE_STAT_EMPTY_STRING_BUG
  273.   if (strlen(save_fname_root)==0) {
  274.     error = -1;
  275.     errno = ENOENT;
  276.   }
  277. #endif
  278.  if (error!=0 && errno==ENOENT || !S_ISDIR(st.st_mode)) {
  279. #if defined(HAVE_MKDIR)
  280.     mode_t mode = S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH;
  281.     error = mkdir(save_fname_root, mode);
  282.     if (error) {
  283.       fprintf(stderr, "can not create directory %sn", save_fname_root);
  284.       exit(-1);
  285.     }
  286.     fprintf(stderr, "created directory for snapshots: %sn", save_fname_root);
  287. #else
  288.     fprintf(stderr, "please create the following directory: %sn", 
  289.     save_fname_root);
  290.     exit(-1);
  291. #endif
  292.   }
  294.   hv_init(conductor_fname, save_fname_root, xsize, ysize);
  295.   initObjs();
  296.   if (reset_world_once) {
  297.     ResetWorld();
  298.   }
  299. }
  302. void contextInit()
  303. {
  304.   argInit( &cparam, 1.0, 0, 0, 0 );
  305.   argInitContext( fullscreen, fsaa );
  307.   SDL_EnableKeyRepeat( 250, 10 );
  309.   if( grabinput )
  310.     SDL_WM_GrabInput( SDL_GRAB_ON );
  312.   // two-sided lighting, use both sides of the polgyons
  313.   glLightModeli( GL_LIGHT_MODEL_TWO_SIDE, GL_FALSE );
  315.   // use infinite viewer for specular lighting - looks good, speedy
  316.   glLightModeli( GL_LIGHT_MODEL_LOCAL_VIEWER, GL_FALSE );
  317. }
  320. void displayInit()
  321. {
  322.   arVideoCapStart();
  324.   arUtilSleep( 500 );
  326.   argDrawMode2D();
  327.   argDrawMode3D();
  328.   argDraw3dCamera( 0, 0 );
  329.   
  330.   glGetIntegerv( GL_VIEWPORT, viewport );
  331.   glGetDoublev( GL_PROJECTION_MATRIX, projectionMatrix );
  333.   glClearColor( 0.0, 0.0, 0.0, 0.0 );
  334.   glClear( GL_COLOR_BUFFER_BIT );
  335.   sdlutSwapBuffers();
  336.   glClear( GL_COLOR_BUFFER_BIT );
  337.   sdlutSwapBuffers();
  339.   sdlutKeyboardFunc( keyboardFunc );
  340.   sdlutMouseFunc( mouseFunc );
  341.   sdlutMotionFunc( motionFunc );
  342.   sdlutIdleFunc( idleFunc );
  343.   sdlutDisplayFunc( displayFunc );
  345.   if( fsaa )
  346.   {
  347.     glEnable( GL_MULTISAMPLE_ARB );
  348.     glHint( GL_MULTISAMPLE_FILTER_HINT_NV, GL_NICEST );
  349.   }
  350. }
  353. //-------------------------------------------------------------------------
  354. // cleanup functions
  355. //-------------------------------------------------------------------------
  358. void cleanup()
  359. {
  360.   for( vector< Marker * >::iterator i = floater_markers.begin() ;
  361.        i != floater_markers.end() ; ++i )
  362.   {
  363.     arFreePatt( ( *i )->id );
  364.     delete *i;
  365.   }
  367.   if( multi_markers )
  368.   {
  369.     if( multi_markers->marker )
  370.       free( multi_markers->marker );
  372.     free( multi_markers );
  373.   }
  375.   // destroy cg context?
  377.   arVideoCapStop();
  378.   arVideoClose();
  380.   argCleanup();
  381.   sdlutDestroyWindow();
  382. }
  385. void quit()
  386. {
  387.   cleanup();
  388.   exit( 0 );
  389. }
  392. void reload()
  393. {
  394.   EXIT_WITH_ERROR( "reload" );
  395. }
  398. //-------------------------------------------------------------------------
  399. // marker functions
  400. //-------------------------------------------------------------------------
  403. Marker * findMarker( const char *patt )
  404. {
  405.   for( vector< Marker * >::iterator i = floater_markers.begin() ;
  406.        i != floater_markers.end() ; ++i )
  407.   {
  408.     if( ( *i )->name == string( patt ) )
  409.       return *i;
  410.   }
  412.   return NULL;
  413. }
  416. #ifdef USE_CG
  418. void cgErrorCallback ()
  419. {
  420.   CGerror error = cgGetError();
  422.   if( error ) 
  423.   {
  424.     const char *listing = cgGetLastListing( cgContext );
  425.     fprintf( stderr, "--------------------------------------------------------------------------n" );
  426.     fprintf( stderr, "ERROR: Cg errorn" );
  427.     fprintf( stderr, "%snn", cgGetErrorString( error ) );
  428.     fprintf( stderr, "%sn", listing );
  429.     fprintf( stderr, "--------------------------------------------------------------------------n" );
  430.     exit( 1 );
  431.   }
  432. }
  434. #endif // USE_CG
  437. //-------------------------------------------------------------------------
  438. // config file functions
  439. //-------------------------------------------------------------------------
  441. char * getNextLine( char *line, int size, FILE *fp )
  442. {
  443.   char *retval;
  445.   while( 1 )
  446.   {
  447.     retval = fgets( line, size, fp );
  449.     if( retval == NULL )
  450.       return NULL;
  452.     if( line[0] != 'n' && line[0] != '#' )
  453.       return retval;
  454.   }
  455. }
  457. void readFloaters( const char *filename )
  458. {
  459. #undef CLOSE_AND_RETURN
  460. #define CLOSE_AND_RETURN { floater_markers.clear(); fclose( fp ); return; }
  462.   FILE *fp = fopen( filename, "r" );
  463.   if( fp == NULL )
  464.     CLOSE_AND_RETURN;
  466.   char pattern_filename[1024];
  467.   char line[1024];
  468.   getNextLine( line, 1024, fp );
  470.   int num;
  471.   if( sscanf( line, "%d", &num ) != 1 )
  472.     CLOSE_AND_RETURN;
  474.   if( num < 1 )
  475.     CLOSE_AND_RETURN;
  477.   floater_markers.clear();
  478.   floater_markers.reserve( num );
  480.   for( int i = 0 ; i < num ; ++i )
  481.   {
  482.     Marker *m = new Marker();
  483.     floater_markers.push_back( m );
  485.     getNextLine( line, 1024, fp );
  486.     if( sscanf( line, "%s", pattern_filename ) != 1 )
  487.       CLOSE_AND_RETURN;
  489.     string temp = pattern_filename;
  490.     m->name = temp.substr( temp.find_last_of( '.' ) + 1 );
  492.     m->id = arLoadPatt( pattern_filename );
  493.     if( m->id < 0 )
  494.       CLOSE_AND_RETURN;
  496.     m->visible = false;
  498.     getNextLine( line, 1024, fp );
  499.     if( sscanf( line, "%lf", &m->width ) != 1 )
  500.       CLOSE_AND_RETURN;
  502.     getNextLine( line, 1024, fp );
  503.     if( sscanf( line, "%lf %lf", &m->center[0], &m->center[1] ) != 2 )
  504.       CLOSE_AND_RETURN;
  505.   }
  507.   fclose( fp );
  508. }
  511. inline void EXIT_WITH_ERROR ( char *msg )
  512. {
  513.   fprintf( stderr, "ERROR: %sn", msg );
  514.   fflush( stderr );
  515.   exit( 1 );
  516. }