int main(int argc, char **argv) { char *input_img1 = NULL, *input_img2 = NULL; double deg = 0.0, scale_fac = 0.0, sigma = 0.0; int num_threads = 1, chunk_size = 1; // check input parameters if (argc <= 6){ if (argc < 2){ printUsage(); exit(1); } // parse the input parameters if (!strcmp(argv[1], "-m") || !strcmp(argv[1], "-mp")){ printf("Motion Estimation in process ...\n"); if (argc < 4 || argc > 6){ printUsage(); exit(1); } // assign parameters input_img1 = argv[2]; input_img2 = argv[3]; if(strcmp(argv[1], "-m") == 0) { motion_estimation(input_img1, input_img2); } else { num_threads = 8; if (argc == 5){ num_threads = atoi(argv[4]); } motion_estimation_parallel(input_img1, input_img2, num_threads); } } else if (!strcmp(argv[1], "-c") || !strcmp(argv[1], "-cp")){ printf("Corner Detection in process ...\n"); if (argc < 3 || argc > 5){ printUsage(); exit(1); } // assign parameters input_img1 = argv[2]; if(strcmp(argv[1], "-c") == 0) { cornerDetectionSequential(input_img1); } else { num_threads = 8; if (argc == 4){ num_threads = atoi(argv[3]); } cornerDetectionParallel(input_img1, num_threads); } } else if (!strcmp(argv[1], "-r") || !strcmp(argv[1], "-rp")){ printf("Rotation in process ...\n"); if (argc < 4 || argc > 6){ printUsage(); exit(1); } // assign parameters input_img1 = argv[2]; deg = atof(argv[3]); if (!strcmp(argv[1], "-r")){ rotation(input_img1, deg); } else { // assigne the number of threads and chunk size if (argc == 5){ num_threads = atoi(argv[4]); } else { num_threads = atoi(argv[4]); chunk_size = atoi(argv[5]); } rotation_parallel(input_img1, deg, num_threads, chunk_size); } } else if (!strcmp(argv[1], "-s") || !strcmp(argv[1], "-sp")){ printf("Scaling in process ...\n"); if (argc < 4 || argc > 6){ printUsage(); exit(1); } // --------------------Done by Mengyi Zhu---------------------------------- //---------------------Here is scaling operation--------------------------- //--------------------it will store the scaled version into scaling.bmp---- // assign parameters input_img1 = argv[2]; scale_fac = atof(argv[3]); num_threads = 1; //---------------------implemention the operation-------------------------- if (!strcmp(argv[1], "-s")) image_scaling(scale_fac, input_img1,"scaling.bmp"); else if(!strcmp(argv[1], "-sp")) { if (argc == 5) num_threads = atoi(argv[4]); image_scaling_parallel(scale_fac, input_img1,"scaling.bmp", num_threads); } //------------------------------------------------------------------------- //------------------------------------------------------------------------- } else if (!strcmp(argv[1], "-g") || !strcmp(argv[1], "-gp")){ printf("Gasussian Blur in process ...\n"); if (argc < 4 || argc > 6){ printUsage(); exit(1); } // assign parameters input_img1 = argv[2]; sigma = atof(argv[3]); if (!strcmp(argv[1], "-g")){ gaussian_blur(input_img1, sigma); } else { // assigne the number of threads and chunk size if (argc == 5){ num_threads = atoi(argv[4]); } else { num_threads = atoi(argv[4]); chunk_size = atoi(argv[5]); } gaussian_blur_parallel(input_img1, sigma, num_threads, chunk_size); } } else if (!strcmp(argv[1], "-o") || !strcmp(argv[1], "-op")){ printf("Outline Detection in process ...\n"); if (argc < 3 || argc > 5){ printUsage(); exit(1); } // assign parameters input_img1 = argv[2]; // TODO: Ali, add your code here } else { if (argc == 6){ printf("All-in-one in process ...\n"); // assign parameters input_img1 = argv[1]; input_img2 = argv[2]; deg = atof(argv[3]); scale_fac = atof(argv[4]); sigma = atof(argv[5]); // Ding, add your sequential code here motion_estimation(input_img1, input_img2); // Michael's sequential code here cornerDetectionSequential(input_img1); // Haokun, add your sequential code here rotation(input_img1, deg); // --------------------Done by Mengyi Zhu--------------------------------- //---------------------Here is scaling operation-------------------------- //--------------------it will store the scaled version into scaling.bmp--- image_scaling(scale_fac, input_img1,"scaling.bmp"); //------------------------------------------------------------------------ //------------------------------------------------------------------------ // Xin, add your sequential code here gaussian_blur(input_img1, sigma); // TODO: Ali, add your sequential code here } else if (argc >= 7 && argc <= 9 && !strcmp(argv[1], "-p")){ // assign parameters input_img1 = argv[2]; input_img2 = argv[3]; deg = atof(argv[4]); scale_fac = atof(argv[5]); sigma = atof(argv[6]); if (argc == 8){ num_threads = atoi(argv[7]); } else if (argc == 9){ num_threads = atoi(argv[7]); chunk_size = atoi(argv[8]); } // Ding, add your parallel code here motion_estimation_parallel(input_img1, input_img2, num_threads); // Michael, add your parallel code here cornerDetectionParallel(input_img1,num_threads); // Haokun, add your parallel code here rotation_parallel(input_img1, deg, num_threads, chunk_size); // --------------------Done by Mengyi Zhu--------------------------------- //---------------------Here is scaling operation------------------------- //--------------------it will store the scaled version into scaling.bmp---- image_scaling_parallel(scale_fac, input_img1,"scaling.bmp", num_threads); //---------------------------------------------------------------------- //---------------------------------------------------------------------- // Xin, add your parallel code here gaussian_blur_parallel(input_img1, sigma, num_threads, chunk_size); // TODO: Ali, add your parallel code here } else { printf("Wrong option passed in or not enough parameters. Please refer the following usage.\n"); printUsage(); exit(1); } } } else { printUsage(); exit(1); } return 0; }
void putseq() { /* this routine assumes (N % M) == 0 */ int i, j, k, f, f0, n, np, nb, sxf, syf, sxb, syb; int ipflag; FILE *fd; char name[256]; unsigned char *neworg[3], *newref[3]; static char ipb[5] = {' ','I','P','B','D'}; rc_init_seq(); /* initialize rate control */ /* sequence header, sequence extension and sequence display extension */ putseqhdr(); if (!mpeg1) { putseqext(); putseqdispext(); } /* optionally output some text data (description, copyright or whatever) */ if (strlen(id_string) > 1) putuserdata(id_string); /* loop through all frames in encoding/decoding order */ for (i=0; i<nframes; i++) { if (!quiet) { fprintf(stderr,"Encoding frame %d ",i); fflush(stderr); } /* f0: lowest frame number in current GOP * * first GOP contains N-(M-1) frames, * all other GOPs contain N frames */ f0 = N*((i+(M-1))/N) - (M-1); if (f0<0) f0=0; if (i==0 || (i-1)%M==0) { /* I or P frame */ for (j=0; j<3; j++) { /* shuffle reference frames */ neworg[j] = oldorgframe[j]; newref[j] = oldrefframe[j]; oldorgframe[j] = neworgframe[j]; oldrefframe[j] = newrefframe[j]; neworgframe[j] = neworg[j]; newrefframe[j] = newref[j]; } /* f: frame number in display order */ f = (i==0) ? 0 : i+M-1; if (f>=nframes) f = nframes - 1; if (i==f0) /* first displayed frame in GOP is I */ { /* I frame */ pict_type = I_TYPE; forw_hor_f_code = forw_vert_f_code = 15; back_hor_f_code = back_vert_f_code = 15; /* n: number of frames in current GOP * * first GOP contains (M-1) less (B) frames */ n = (i==0) ? N-(M-1) : N; /* last GOP may contain less frames */ if (n > nframes-f0) n = nframes-f0; /* number of P frames */ if (i==0) np = (n + 2*(M-1))/M - 1; /* first GOP */ else np = (n + (M-1))/M - 1; /* number of B frames */ nb = n - np - 1; rc_init_GOP(np,nb); putgophdr(f0,i==0); /* set closed_GOP in first GOP only */ } else { /* P frame */ pict_type = P_TYPE; forw_hor_f_code = motion_data[0].forw_hor_f_code; forw_vert_f_code = motion_data[0].forw_vert_f_code; back_hor_f_code = back_vert_f_code = 15; sxf = motion_data[0].sxf; syf = motion_data[0].syf; } } else { /* B frame */ for (j=0; j<3; j++) { neworg[j] = auxorgframe[j]; newref[j] = auxframe[j]; } /* f: frame number in display order */ f = i - 1; pict_type = B_TYPE; n = (i-2)%M + 1; /* first B: n=1, second B: n=2, ... */ forw_hor_f_code = motion_data[n].forw_hor_f_code; forw_vert_f_code = motion_data[n].forw_vert_f_code; back_hor_f_code = motion_data[n].back_hor_f_code; back_vert_f_code = motion_data[n].back_vert_f_code; sxf = motion_data[n].sxf; syf = motion_data[n].syf; sxb = motion_data[n].sxb; syb = motion_data[n].syb; } temp_ref = f - f0; frame_pred_dct = frame_pred_dct_tab[pict_type-1]; q_scale_type = qscale_tab[pict_type-1]; intravlc = intravlc_tab[pict_type-1]; altscan = altscan_tab[pict_type-1]; fprintf(statfile,"\nFrame %d (#%d in display order):\n",i,f); fprintf(statfile," picture_type=%c\n",ipb[pict_type]); fprintf(statfile," temporal_reference=%d\n",temp_ref); fprintf(statfile," frame_pred_frame_dct=%d\n",frame_pred_dct); fprintf(statfile," q_scale_type=%d\n",q_scale_type); fprintf(statfile," intra_vlc_format=%d\n",intravlc); fprintf(statfile," alternate_scan=%d\n",altscan); if (pict_type!=I_TYPE) { fprintf(statfile," forward search window: %d...%d / %d...%d\n", -sxf,sxf,-syf,syf); fprintf(statfile," forward vector range: %d...%d.5 / %d...%d.5\n", -(4<<forw_hor_f_code),(4<<forw_hor_f_code)-1, -(4<<forw_vert_f_code),(4<<forw_vert_f_code)-1); } if (pict_type==B_TYPE) { fprintf(statfile," backward search window: %d...%d / %d...%d\n", -sxb,sxb,-syb,syb); fprintf(statfile," backward vector range: %d...%d.5 / %d...%d.5\n", -(4<<back_hor_f_code),(4<<back_hor_f_code)-1, -(4<<back_vert_f_code),(4<<back_vert_f_code)-1); } sprintf(name,tplorg,f+frame0); readframe(name,neworg); if (fieldpic) { if (!quiet) { fprintf(stderr,"\nfirst field (%s) ",topfirst ? "top" : "bot"); fflush(stderr); } pict_struct = topfirst ? TOP_FIELD : BOTTOM_FIELD; #ifndef SKIP_PREDICTION motion_estimation(oldorgframe[0],neworgframe[0], oldrefframe[0],newrefframe[0], neworg[0],newref[0], sxf,syf,sxb,syb,mbinfo,0,0); predict(oldrefframe,newrefframe,predframe,0,mbinfo); #endif #ifndef SKIP_BLOCK_ENCODE dct_type_estimation(predframe[0],neworg[0],mbinfo); transform(predframe,neworg,mbinfo,blocks); #endif putpict(neworg[0]); #ifndef SKIP_BLOCK_DECODE for (k=0; k<mb_height2*mb_width; k++) { if (mbinfo[k].mb_type & MB_INTRA) for (j=0; j<block_count; j++) iquant_intra(blocks[k*block_count+j],blocks[k*block_count+j], dc_prec,intra_q,mbinfo[k].mquant); else for (j=0;j<block_count;j++) iquant_non_intra(blocks[k*block_count+j],blocks[k*block_count+j], inter_q,mbinfo[k].mquant); } itransform(predframe,newref,mbinfo,blocks); #endif calcSNR(neworg,newref); stats(); if (!quiet) { fprintf(stderr,"second field (%s) ",topfirst ? "bot" : "top"); fflush(stderr); } pict_struct = topfirst ? BOTTOM_FIELD : TOP_FIELD; ipflag = (pict_type==I_TYPE); if (ipflag) { /* first field = I, second field = P */ pict_type = P_TYPE; forw_hor_f_code = motion_data[0].forw_hor_f_code; forw_vert_f_code = motion_data[0].forw_vert_f_code; back_hor_f_code = back_vert_f_code = 15; sxf = motion_data[0].sxf; syf = motion_data[0].syf; } #ifndef SKIP_PREDICTION motion_estimation(oldorgframe[0],neworgframe[0], oldrefframe[0],newrefframe[0], neworg[0],newref[0], sxf,syf,sxb,syb,mbinfo,1,ipflag); predict(oldrefframe,newrefframe,predframe,1,mbinfo); #endif #ifndef SKIP_BLOCK_ENCOED dct_type_estimation(predframe[0],neworg[0],mbinfo); transform(predframe,neworg,mbinfo,blocks); #endif putpict(neworg[0]); #ifndef SKIP_BLOCK_DECODE for (k=0; k<mb_height2*mb_width; k++) { if (mbinfo[k].mb_type & MB_INTRA) for (j=0; j<block_count; j++) iquant_intra(blocks[k*block_count+j],blocks[k*block_count+j], dc_prec,intra_q,mbinfo[k].mquant); else for (j=0;j<block_count;j++) iquant_non_intra(blocks[k*block_count+j],blocks[k*block_count+j], inter_q,mbinfo[k].mquant); } itransform(predframe,newref,mbinfo,blocks); #endif calcSNR(neworg,newref); stats(); } else { pict_struct = FRAME_PICTURE; /* do motion_estimation * * uses source frames (...orgframe) for full pel search * and reconstructed frames (...refframe) for half pel search */ #ifndef SKIP_PREDICTION motion_estimation(oldorgframe[0],neworgframe[0], oldrefframe[0],newrefframe[0], neworg[0],newref[0], sxf,syf,sxb,syb,mbinfo,0,0); predict(oldrefframe,newrefframe,predframe,0,mbinfo); #endif #ifndef SKIP_BLOCK_ENCODE dct_type_estimation(predframe[0],neworg[0],mbinfo); transform(predframe,neworg,mbinfo,blocks); #endif putpict(neworg[0]); #ifndef SKIP_BLOCK_DECODE for (k=0; k<mb_height*mb_width; k++) { if (mbinfo[k].mb_type & MB_INTRA) for (j=0; j<block_count; j++) iquant_intra(blocks[k*block_count+j],blocks[k*block_count+j], dc_prec,intra_q,mbinfo[k].mquant); else for (j=0;j<block_count;j++) iquant_non_intra(blocks[k*block_count+j],blocks[k*block_count+j], inter_q,mbinfo[k].mquant); } itransform(predframe,newref,mbinfo,blocks); #endif calcSNR(neworg,newref); stats(); } sprintf(name,tplref,f+frame0); writeframe(name,newref); } putseqend(); }