diff options
author | Vito Caputo <vcaputo@pengaru.com> | 2020-07-02 23:03:25 -0700 |
---|---|---|
committer | Vito Caputo <vcaputo@pengaru.com> | 2020-07-11 13:36:41 -0700 |
commit | 2c97c3ab76f2107fb5c30534d30b05f03c2983c4 (patch) | |
tree | 7ce1ba1a702829312e996af55d744fadd1551440 /recordmydesktop/src | |
parent | 7f116975cb9f8405feedb727b4c963bb4c4e74df (diff) |
cache_frame: more formatting cleanups
Mostly mindless whitespace fixups but also changed some yuvblocks
indexing that seemed brain damaged to just be simpler without
actually changing anything functionally
Diffstat (limited to 'recordmydesktop/src')
-rw-r--r-- | recordmydesktop/src/rmd_cache_frame.c | 195 |
1 files changed, 100 insertions, 95 deletions
diff --git a/recordmydesktop/src/rmd_cache_frame.c b/recordmydesktop/src/rmd_cache_frame.c index 8173080..79570f3 100644 --- a/recordmydesktop/src/rmd_cache_frame.c +++ b/recordmydesktop/src/rmd_cache_frame.c @@ -54,33 +54,33 @@ static int rmdFlushBlock( FILE *ucfp, int flush) { - int j,i, - bytes_written=0, - block_i=(!blockwidth)?0:(blockno/(width/blockwidth)),//place on the grid - block_k=(!blockwidth)?0:(blockno%(width/blockwidth)); + int bytes_written = 0, + block_i = (!blockwidth) ? 0 : (blockno / (width / blockwidth)),//place on the grid + block_k = (!blockwidth) ? 0 : (blockno % (width / blockwidth)); register unsigned char *buf_reg = (&buf[(block_i * width + block_k) * blockwidth]); static unsigned char out_buffer[CACHE_OUT_BUFFER_SIZE]; - static unsigned int out_buffer_bytes=0; + static unsigned int out_buffer_bytes = 0; - if (out_buffer_bytes+pow(blockwidth,2)>=CACHE_OUT_BUFFER_SIZE || (flush && out_buffer_bytes)) { - if (ucfp==NULL) - gzwrite(fp,(void *)out_buffer,out_buffer_bytes); + if (out_buffer_bytes + pow(blockwidth, 2) >= CACHE_OUT_BUFFER_SIZE || (flush && out_buffer_bytes)) { + if (ucfp == NULL) + gzwrite(fp, (void *)out_buffer, out_buffer_bytes); else - fwrite((void *)out_buffer,1,out_buffer_bytes,ucfp); + fwrite((void *)out_buffer, 1, out_buffer_bytes,ucfp); - bytes_written=out_buffer_bytes; - out_buffer_bytes=0; + bytes_written = out_buffer_bytes; + out_buffer_bytes = 0; } + if (!flush) { - register unsigned char *out_buf_reg=&out_buffer[out_buffer_bytes]; + register unsigned char *out_buf_reg = &out_buffer[out_buffer_bytes]; - for(j=0;j<blockwidth;j++) { + for (int j = 0;j < blockwidth; j++) { - for(i=0;i<blockwidth;i++) - (*out_buf_reg++)=(*buf_reg++); + for(int i = 0;i < blockwidth; i++) + (*out_buf_reg++) = (*buf_reg++); - out_buffer_bytes+=blockwidth; - buf_reg+=width-blockwidth; + out_buffer_bytes += blockwidth; + buf_reg += width - blockwidth; } } @@ -89,45 +89,46 @@ static int rmdFlushBlock( void *rmdCacheImageBuffer(ProgData *pdata) { - gzFile fp=NULL; - FILE *ucfp=NULL; - int index_entry_size=sizeof(u_int32_t), - blocknum_x=pdata->enc_data->yuv.y_width/Y_UNIT_WIDTH, - blocknum_y=pdata->enc_data->yuv.y_height/Y_UNIT_WIDTH, - firstrun=1, - frameno=0, - nbytes=0, - nth_cache=1; - u_int32_t ynum,unum,vnum, - y_short_blocks[blocknum_x*blocknum_y], - u_short_blocks[blocknum_x*blocknum_y], - v_short_blocks[blocknum_x*blocknum_y]; + gzFile fp = NULL; + FILE *ucfp = NULL; + int index_entry_size = sizeof(u_int32_t), + blocknum_x = pdata->enc_data->yuv.y_width / Y_UNIT_WIDTH, + blocknum_y = pdata->enc_data->yuv.y_height / Y_UNIT_WIDTH, + firstrun = 1, + frameno = 0, + nbytes = 0, + nth_cache = 1; + u_int32_t ynum, unum, vnum, + y_short_blocks[blocknum_x * blocknum_y], + u_short_blocks[blocknum_x * blocknum_y], + v_short_blocks[blocknum_x * blocknum_y]; unsigned long long int total_bytes = 0; unsigned long long int total_received_bytes = 0; + unsigned int capture_frameno = 0; if (!pdata->args.zerocompression) { - fp=pdata->cache_data->ifp; + fp = pdata->cache_data->ifp; - if (fp==NULL) + if (fp == NULL) exit(13); } else { - ucfp=pdata->cache_data->uncifp; + ucfp = pdata->cache_data->uncifp; - if (ucfp==NULL) + if (ucfp == NULL) exit(13); } while (pdata->running) { - int j; + FrameHeader fheader; + + ynum = unum = vnum = 0; - FrameHeader fheader; - ynum=unum=vnum=0; + pthread_mutex_lock(&pdata->time_mutex); + while (pdata->running && capture_frameno >= pdata->capture_frameno) + pthread_cond_wait(&pdata->time_cond, &pdata->time_mutex); - pthread_mutex_lock(&pdata->img_buff_ready_mutex); - pdata->th_enc_thread_waiting = 1; - while (pdata->th_enc_thread_waiting) - pthread_cond_wait(&pdata->image_buffer_ready, &pdata->img_buff_ready_mutex); - pthread_mutex_unlock(&pdata->img_buff_ready_mutex); + capture_frameno = pdata->capture_frameno; + pthread_mutex_unlock(&pdata->time_mutex); pthread_mutex_lock(&pdata->pause_mutex); while (pdata->paused) @@ -138,80 +139,82 @@ void *rmdCacheImageBuffer(ProgData *pdata) { //find and flush different blocks if (firstrun) { - firstrun=0; - for(j=0;j<blocknum_x*blocknum_y;j++) { + firstrun = 0; + for(int j = 0; j < blocknum_x * blocknum_y; j++) { + yblocks[ynum] = 1; + y_short_blocks[ynum] = j; ynum++; - yblocks[ynum-1]=1; - y_short_blocks[ynum-1]=j; + ublocks[unum] = 1; + u_short_blocks[unum] = j; unum++; - ublocks[unum-1]=1; - u_short_blocks[ynum-1]=j; + vblocks[vnum] = 1; + v_short_blocks[vnum] = j; vnum++; - vblocks[vnum-1]=1; - v_short_blocks[ynum-1]=j; } } else { /**COMPRESS ARRAYS*/ - for(j=0;j<blocknum_x*blocknum_y;j++) { + for(int j = 0; j < blocknum_x * blocknum_y; j++) { if (yblocks[j]) { + y_short_blocks[ynum] = j; ynum++; - y_short_blocks[ynum-1]=j; } if (ublocks[j]) { + u_short_blocks[unum] = j; unum++; - u_short_blocks[unum-1]=j; } if (vblocks[j]) { + v_short_blocks[vnum] = j; vnum++; - v_short_blocks[vnum-1]=j; } } } /**WRITE FRAME TO DISK*/ if (!pdata->args.zerocompression) { - if (ynum*4+unum+vnum>(blocknum_x*blocknum_y*6)/10) - gzsetparams (fp,1,Z_FILTERED); + if (ynum * 4 + unum + vnum > (blocknum_x * blocknum_y * 6) / 10) + gzsetparams(fp, 1, Z_FILTERED); else - gzsetparams (fp,0,Z_FILTERED); + gzsetparams(fp, 0, Z_FILTERED); } - strncpy(fheader.frame_prefix,"FRAM",4); - fheader.frameno=++frameno; + strncpy(fheader.frame_prefix, "FRAM", 4); + fheader.frameno = ++frameno; fheader.current_total = pdata->frames_total; - fheader.Ynum=ynum; - fheader.Unum=unum; - fheader.Vnum=vnum; + fheader.Ynum = ynum; + fheader.Unum = unum; + fheader.Vnum = vnum; if (!pdata->args.zerocompression) { - nbytes+=gzwrite(fp,(void*)&fheader,sizeof(FrameHeader)); + nbytes += gzwrite(fp, (void*)&fheader, sizeof(FrameHeader)); //flush indexes if (ynum) - nbytes+=gzwrite(fp, (void*)y_short_blocks, ynum*index_entry_size); + nbytes += gzwrite(fp, (void*)y_short_blocks, ynum * index_entry_size); if (unum) - nbytes+=gzwrite(fp, (void*)u_short_blocks, unum*index_entry_size); + nbytes += gzwrite(fp, (void*)u_short_blocks, unum * index_entry_size); if (vnum) - nbytes+=gzwrite(fp, (void*)v_short_blocks, vnum*index_entry_size); + nbytes += gzwrite(fp, (void*)v_short_blocks, vnum * index_entry_size); } else { - nbytes+=sizeof(FrameHeader)* - fwrite((void*)&fheader,sizeof(FrameHeader),1,ucfp); + nbytes += sizeof(FrameHeader)* + fwrite((void*)&fheader, sizeof(FrameHeader), 1, ucfp); //flush indexes if (ynum) - nbytes+=index_entry_size* fwrite(y_short_blocks,index_entry_size,ynum,ucfp); + nbytes += index_entry_size * fwrite(y_short_blocks, index_entry_size, ynum, ucfp); if (unum) - nbytes+=index_entry_size* fwrite(u_short_blocks,index_entry_size,unum,ucfp); + nbytes += index_entry_size * fwrite(u_short_blocks, index_entry_size, unum, ucfp); if (vnum) - nbytes+=index_entry_size* fwrite(v_short_blocks,index_entry_size,vnum,ucfp); + nbytes += index_entry_size * fwrite(v_short_blocks, index_entry_size, vnum, ucfp); } + //flush the blocks for each buffer if (ynum) { - for(j=0;j<ynum;j++) - nbytes+=rmdFlushBlock( pdata->enc_data->yuv.y,y_short_blocks[j], + for(int j = 0; j < ynum; j++) + nbytes += rmdFlushBlock(pdata->enc_data->yuv.y, + y_short_blocks[j], pdata->enc_data->yuv.y_width, pdata->enc_data->yuv.y_height, Y_UNIT_WIDTH, @@ -221,8 +224,9 @@ void *rmdCacheImageBuffer(ProgData *pdata) { } if (unum) { - for(j=0;j<unum;j++) - nbytes+=rmdFlushBlock( pdata->enc_data->yuv.u,u_short_blocks[j], + for(int j = 0; j < unum; j++) + nbytes += rmdFlushBlock(pdata->enc_data->yuv.u, + u_short_blocks[j], pdata->enc_data->yuv.uv_width, pdata->enc_data->yuv.uv_height, UV_UNIT_WIDTH, @@ -232,8 +236,9 @@ void *rmdCacheImageBuffer(ProgData *pdata) { } if (vnum) { - for(j=0;j<vnum;j++) - nbytes+=rmdFlushBlock( pdata->enc_data->yuv.v,v_short_blocks[j], + for(int j = 0; j < vnum; j++) + nbytes += rmdFlushBlock(pdata->enc_data->yuv.v, + v_short_blocks[j], pdata->enc_data->yuv.uv_width, pdata->enc_data->yuv.uv_height, UV_UNIT_WIDTH, @@ -245,37 +250,37 @@ void *rmdCacheImageBuffer(ProgData *pdata) { //release main buffer pthread_mutex_unlock(&pdata->yuv_mutex); - nbytes+=rmdFlushBlock(NULL,0,0,0,0,fp,ucfp,1); + nbytes += rmdFlushBlock(NULL, 0, 0, 0, 0, fp, ucfp, 1); /**@________________@**/ - pdata->avd+=pdata->frametime; + pdata->avd += pdata->frametime; - if (nbytes>CACHE_FILE_SIZE_LIMIT) { - if (rmdSwapCacheFilesWrite(pdata->cache_data->imgdata, nth_cache,&fp,&ucfp)) { + if (nbytes > CACHE_FILE_SIZE_LIMIT) { + if (rmdSwapCacheFilesWrite(pdata->cache_data->imgdata, nth_cache, &fp, &ucfp)) { fprintf(stderr,"New cache file could not be created.\n" "Ending recording...\n"); fflush(stderr); raise(SIGINT); //if for some reason we cannot make a new file - //we have to stop. If we are out of space, - //which means - //that encoding cannot happen either, - //InitEncoder will cause an abrupt end with an - //error code and the cache will remain intact. - //If we've chosen separate two-stages, - //the program will make a - //clean exit. - //In either case data will be preserved so if - //space is freed the recording - //can be proccessed later. + //we have to stop. If we are out of space, + //which means + //that encoding cannot happen either, + //InitEncoder will cause an abrupt end with an + //error code and the cache will remain intact. + //If we've chosen separate two-stages, + //the program will make a + //clean exit. + //In either case data will be preserved so if + //space is freed the recording + //can be proccessed later. } total_bytes += nbytes; nth_cache++; - nbytes=0; + nbytes = 0; } } total_bytes += nbytes; { - unsigned int bytes_per_pixel = pdata->specs.depth >= 24 ? 4 : 2; + unsigned int bytes_per_pixel = pdata->specs.depth >= 24 ? 4 : 2; unsigned int pixels_per_frame = pdata->brwin.rrect.width * pdata->brwin.rrect.height; total_received_bytes = ((unsigned long long int)frameno) * bytes_per_pixel * pixels_per_frame; @@ -303,7 +308,7 @@ void *rmdCacheImageBuffer(ProgData *pdata) { fflush(stderr); if (!pdata->args.zerocompression) { - gzflush(fp,Z_FINISH); + gzflush(fp, Z_FINISH); gzclose(fp); } else { fflush(ucfp); |