summaryrefslogtreecommitdiff
path: root/recordmydesktop/src/rmd_init_encoder.c
diff options
context:
space:
mode:
authorVito Caputo <vcaputo@pengaru.com>2020-06-23 17:20:32 -0700
committerVito Caputo <vcaputo@pengaru.com>2020-07-11 13:36:41 -0700
commit8e36165c43072a592b4ac9e122c8fc6e4555b915 (patch)
tree56f04cd5784d1fa5bffc8233eb177b3b8d9da8bd /recordmydesktop/src/rmd_init_encoder.c
parent451548d4c4a1fcab7a95e8678ca63189a0a3cc93 (diff)
*: more random formatting cleanups
some theora init error checking fixups snuck in there as well
Diffstat (limited to 'recordmydesktop/src/rmd_init_encoder.c')
-rw-r--r--recordmydesktop/src/rmd_init_encoder.c183
1 files changed, 92 insertions, 91 deletions
diff --git a/recordmydesktop/src/rmd_init_encoder.c b/recordmydesktop/src/rmd_init_encoder.c
index da442f2..2b2273e 100644
--- a/recordmydesktop/src/rmd_init_encoder.c
+++ b/recordmydesktop/src/rmd_init_encoder.c
@@ -83,7 +83,7 @@ static int rmdIncrementalNaming(char **name) {
return 0;
}
-void rmdInitEncoder(ProgData *pdata,EncData *enc_data_t,int buffer_ready) {
+void rmdInitEncoder(ProgData *pdata, EncData *enc_data_t, int buffer_ready) {
int y0, y1, y2, fname_length;
ogg_stream_state m_ogg_skel;
ogg_page skel_og_pg;
@@ -91,13 +91,13 @@ void rmdInitEncoder(ProgData *pdata,EncData *enc_data_t,int buffer_ready) {
skel_fba; //audio fisbone packet
const char *fname;
- (pdata)->enc_data=enc_data_t;
+ pdata->enc_data = enc_data_t;
fname = pdata->args.filename;
- fname_length=strlen(fname);
+ fname_length = strlen(fname);
if (!(fname_length>4 && !strcasecmp(&fname[fname_length-3], "ogv"))) {
- char *new_name=malloc(fname_length+5);
+ char *new_name = malloc(fname_length + 5);
strcpy(new_name,fname);
strcat(new_name,".ogv");
@@ -110,34 +110,34 @@ void rmdInitEncoder(ProgData *pdata,EncData *enc_data_t,int buffer_ready) {
fprintf(stderr, "Output file: %s\n", pdata->args.filename);
}
- enc_data_t->fp=fopen((pdata)->args.filename,"w");
- if (enc_data_t->fp==NULL) {
+ enc_data_t->fp = fopen((pdata)->args.filename,"w");
+ if (enc_data_t->fp == NULL) {
fprintf(stderr,"Cannot open file %s for writting!\n", (pdata)->args.filename);
exit(13);
}
//each stream must have a unique
srand(time(NULL));
- y0=rand()+1;
- y1=rand()+1;
- y2=rand()+1;
- y2+=(y1==y2);
- y0=(((y0==y1)||(y0==y2))?(y1+y2):y0);
+ y0 = rand() + 1;
+ y1 = rand() + 1;
+ y2 = rand() + 1;
+ y2 += (y1 == y2);
+ y0 = (((y0 == y1) || (y0 == y2)) ? (y1 + y2) : y0);
//init ogg streams
//skeleton first
ogg_stream_init(&m_ogg_skel, y0);
m_add_fishead_packet(&m_ogg_skel);
- if (ogg_stream_pageout(&m_ogg_skel,&skel_og_pg)!= 1) {
+ if (ogg_stream_pageout(&m_ogg_skel, &skel_og_pg) != 1) {
fprintf (stderr, "Internal Ogg library error.\n");
exit (2);
}
- fwrite(skel_og_pg.header,1,skel_og_pg.header_len,enc_data_t->fp);
- fwrite(skel_og_pg.body,1,skel_og_pg.body_len,enc_data_t->fp);
+ fwrite(skel_og_pg.header, 1, skel_og_pg.header_len, enc_data_t->fp);
+ fwrite(skel_og_pg.body, 1, skel_og_pg.body_len, enc_data_t->fp);
- ogg_stream_init(&enc_data_t->m_ogg_ts,y1);
+ ogg_stream_init(&enc_data_t->m_ogg_ts, y1);
if (!pdata->args.nosound)
- ogg_stream_init(&enc_data_t->m_ogg_vs,y2);
+ ogg_stream_init(&enc_data_t->m_ogg_vs, y2);
theora_info_init(&enc_data_t->m_th_inf);
enc_data_t->m_th_inf.frame_width = pdata->brwin.rrect.width;
@@ -168,9 +168,6 @@ void rmdInitEncoder(ProgData *pdata,EncData *enc_data_t,int buffer_ready) {
enc_data_t->m_th_inf.noise_sensitivity = 1;
enc_data_t->m_th_inf.sharpness = 2;
- theora_encode_init(&enc_data_t->m_th_st,&enc_data_t->m_th_inf);
-
-
if (!pdata->args.nosound) {
int ret;
vorbis_info_init(&enc_data_t->m_vo_inf);
@@ -183,134 +180,141 @@ void rmdInitEncoder(ProgData *pdata,EncData *enc_data_t,int buffer_ready) {
exit(2);
}
vorbis_comment_init(&enc_data_t->m_vo_cmmnt);
- vorbis_analysis_init(&enc_data_t->m_vo_dsp,&enc_data_t->m_vo_inf);
- vorbis_block_init(&enc_data_t->m_vo_dsp,&enc_data_t->m_vo_block);
+ vorbis_analysis_init(&enc_data_t->m_vo_dsp, &enc_data_t->m_vo_inf);
+ vorbis_block_init(&enc_data_t->m_vo_dsp, &enc_data_t->m_vo_block);
+ }
+
+ if (theora_encode_init(&enc_data_t->m_th_st, &enc_data_t->m_th_inf)) {
+ fprintf(stderr, "Theora encoder initialization failure.\n");
+ exit(2);
}
+ if (theora_encode_header(&enc_data_t->m_th_st, &enc_data_t->m_ogg_pckt1)) {
+ fprintf(stderr, "Theora enocde header failure.\n");
+ exit(2);
+ }
- theora_encode_header(&enc_data_t->m_th_st,&enc_data_t->m_ogg_pckt1);
- ogg_stream_packetin(&enc_data_t->m_ogg_ts,&enc_data_t->m_ogg_pckt1);
- if (ogg_stream_pageout(&enc_data_t->m_ogg_ts,&enc_data_t->m_ogg_pg)!=1) {
+ ogg_stream_packetin(&enc_data_t->m_ogg_ts, &enc_data_t->m_ogg_pckt1);
+ if (ogg_stream_pageout(&enc_data_t->m_ogg_ts, &enc_data_t->m_ogg_pg) != 1) {
fprintf(stderr,"Internal Ogg library error.\n");
exit(2);
}
- fwrite(enc_data_t->m_ogg_pg.header,1,
+ fwrite(enc_data_t->m_ogg_pg.header, 1,
enc_data_t->m_ogg_pg.header_len,
enc_data_t->fp);
- fwrite(enc_data_t->m_ogg_pg.body,1,
+ fwrite(enc_data_t->m_ogg_pg.body, 1,
enc_data_t->m_ogg_pg.body_len,
enc_data_t->fp);
theora_comment_init(&enc_data_t->m_th_cmmnt);
- theora_comment_add_tag(&enc_data_t->m_th_cmmnt,"recordMyDesktop",VERSION);
- theora_encode_comment(&enc_data_t->m_th_cmmnt,&enc_data_t->m_ogg_pckt1);
- ogg_stream_packetin(&enc_data_t->m_ogg_ts,&enc_data_t->m_ogg_pckt1);
- theora_encode_tables(&enc_data_t->m_th_st,&enc_data_t->m_ogg_pckt1);
- ogg_stream_packetin(&enc_data_t->m_ogg_ts,&enc_data_t->m_ogg_pckt1);
+ theora_comment_add_tag(&enc_data_t->m_th_cmmnt, "recordMyDesktop", VERSION);
+ theora_encode_comment(&enc_data_t->m_th_cmmnt, &enc_data_t->m_ogg_pckt1);
+ ogg_stream_packetin(&enc_data_t->m_ogg_ts, &enc_data_t->m_ogg_pckt1);
+ theora_encode_tables(&enc_data_t->m_th_st, &enc_data_t->m_ogg_pckt1);
+ ogg_stream_packetin(&enc_data_t->m_ogg_ts, &enc_data_t->m_ogg_pckt1);
if (!pdata->args.nosound) {
ogg_packet header;
ogg_packet header_comm;
ogg_packet header_code;
- vorbis_analysis_headerout(&enc_data_t->m_vo_dsp,
- &enc_data_t->m_vo_cmmnt,
- &header,&header_comm,
- &header_code);
- ogg_stream_packetin(&enc_data_t->m_ogg_vs,&header);
- if (ogg_stream_pageout(&enc_data_t->m_ogg_vs,&enc_data_t->m_ogg_pg)!=1) {
+ vorbis_analysis_headerout( &enc_data_t->m_vo_dsp,
+ &enc_data_t->m_vo_cmmnt,
+ &header, &header_comm,
+ &header_code);
+ ogg_stream_packetin(&enc_data_t->m_ogg_vs, &header);
+ if (ogg_stream_pageout(&enc_data_t->m_ogg_vs, &enc_data_t->m_ogg_pg) != 1) {
fprintf(stderr,"Internal Ogg library error.\n");
exit(2);
}
- fwrite(enc_data_t->m_ogg_pg.header,1,
+ fwrite(enc_data_t->m_ogg_pg.header, 1,
enc_data_t->m_ogg_pg.header_len,
enc_data_t->fp);
- fwrite(enc_data_t->m_ogg_pg.body,1,
+ fwrite(enc_data_t->m_ogg_pg.body, 1,
enc_data_t->m_ogg_pg.body_len,
enc_data_t->fp);
- ogg_stream_packetin(&enc_data_t->m_ogg_vs,&header_comm);
- ogg_stream_packetin(&enc_data_t->m_ogg_vs,&header_code);
+ ogg_stream_packetin(&enc_data_t->m_ogg_vs, &header_comm);
+ ogg_stream_packetin(&enc_data_t->m_ogg_vs, &header_code);
}
//fishbone packets go here
- memset(&skel_fbv,0,sizeof(skel_fbv));
- skel_fbv.serial_no=enc_data_t->m_ogg_ts.serialno;
- skel_fbv.nr_header_packet=3;
- skel_fbv.granule_rate_n=enc_data_t->m_th_inf.fps_numerator;
- skel_fbv.granule_rate_d=enc_data_t->m_th_inf.fps_denominator;
- skel_fbv.start_granule=0;
- skel_fbv.preroll=0;
- skel_fbv.granule_shift=theora_granule_shift(&enc_data_t->m_th_inf);
+ memset(&skel_fbv, 0, sizeof(skel_fbv));
+ skel_fbv.serial_no = enc_data_t->m_ogg_ts.serialno;
+ skel_fbv.nr_header_packet = 3;
+ skel_fbv.granule_rate_n = enc_data_t->m_th_inf.fps_numerator;
+ skel_fbv.granule_rate_d = enc_data_t->m_th_inf.fps_denominator;
+ skel_fbv.start_granule = 0;
+ skel_fbv.preroll = 0;
+ skel_fbv.granule_shift = theora_granule_shift(&enc_data_t->m_th_inf);
add_message_header_field(&skel_fbv, "Content-Type", "video/theora");
add_fisbone_to_stream(&m_ogg_skel,&skel_fbv);
if (!pdata->args.nosound) {
- memset(&skel_fba,0,sizeof(skel_fba));
- skel_fba.serial_no=enc_data_t->m_ogg_vs.serialno;
- skel_fba.nr_header_packet=3;
- skel_fba.granule_rate_n=pdata->args.frequency;
- skel_fba.granule_rate_d=(ogg_int64_t)1;
- skel_fba.start_granule=0;
- skel_fba.preroll=2;
- skel_fba.granule_shift=0;
+ memset(&skel_fba, 0, sizeof(skel_fba));
+ skel_fba.serial_no = enc_data_t->m_ogg_vs.serialno;
+ skel_fba.nr_header_packet = 3;
+ skel_fba.granule_rate_n = pdata->args.frequency;
+ skel_fba.granule_rate_d = (ogg_int64_t)1;
+ skel_fba.start_granule = 0;
+ skel_fba.preroll = 2;
+ skel_fba.granule_shift = 0;
add_message_header_field(&skel_fba, "Content-Type", "audio/vorbis");
- add_fisbone_to_stream(&m_ogg_skel,&skel_fba);
+ add_fisbone_to_stream(&m_ogg_skel, &skel_fba);
}
while (1) {
int result = ogg_stream_flush(&m_ogg_skel, &skel_og_pg);
- if (result<0) {
+ if (result < 0) {
fprintf (stderr, "Internal Ogg library error.\n");
exit(2);
}
- if (result==0)
+ if (result == 0)
break;
- fwrite(skel_og_pg.header,1,skel_og_pg.header_len,enc_data_t->fp);
- fwrite(skel_og_pg.body,1,skel_og_pg.body_len,enc_data_t->fp);
+
+ fwrite(skel_og_pg.header, 1, skel_og_pg.header_len, enc_data_t->fp);
+ fwrite(skel_og_pg.body, 1, skel_og_pg.body_len, enc_data_t->fp);
}
while (1) {
- int result = ogg_stream_flush(&enc_data_t->m_ogg_ts,
- &enc_data_t->m_ogg_pg);
- if (result<0) {
+ int result = ogg_stream_flush(&enc_data_t->m_ogg_ts, &enc_data_t->m_ogg_pg);
+ if (result < 0) {
fprintf(stderr,"Internal Ogg library error.\n");
exit(2);
}
- if (result==0)
+ if (result == 0)
break;
- fwrite( enc_data_t->m_ogg_pg.header,1,
+ fwrite( enc_data_t->m_ogg_pg.header, 1,
enc_data_t->m_ogg_pg.header_len,
enc_data_t->fp);
- fwrite( enc_data_t->m_ogg_pg.body,1,
+ fwrite( enc_data_t->m_ogg_pg.body, 1,
enc_data_t->m_ogg_pg.body_len,
enc_data_t->fp);
}
if (!pdata->args.nosound) {
while (1) {
- int result=ogg_stream_flush(&enc_data_t->m_ogg_vs,
- &enc_data_t->m_ogg_pg);
- if (result<0) {
+ int result = ogg_stream_flush(&enc_data_t->m_ogg_vs, &enc_data_t->m_ogg_pg);
+ if (result < 0) {
fprintf(stderr,"Internal Ogg library error.\n");
exit(2);
}
- if (result==0)
+ if (result == 0)
break;
- fwrite( enc_data_t->m_ogg_pg.header,1,
+ fwrite( enc_data_t->m_ogg_pg.header, 1,
enc_data_t->m_ogg_pg.header_len,
enc_data_t->fp);
- fwrite( enc_data_t->m_ogg_pg.body,1,
+ fwrite( enc_data_t->m_ogg_pg.body, 1,
enc_data_t->m_ogg_pg.body_len,
enc_data_t->fp);
}
@@ -318,29 +322,26 @@ void rmdInitEncoder(ProgData *pdata,EncData *enc_data_t,int buffer_ready) {
//skeleton eos
add_eos_packet_to_stream(&m_ogg_skel);
- if (ogg_stream_flush(&m_ogg_skel,&skel_og_pg)<0) {
+ if (ogg_stream_flush(&m_ogg_skel, &skel_og_pg) < 0) {
fprintf(stderr,"Internal Ogg library error.\n");
exit(2);
}
- fwrite(skel_og_pg.header,1,skel_og_pg.header_len,enc_data_t->fp);
+ fwrite(skel_og_pg.header, 1, skel_og_pg.header_len,enc_data_t->fp);
//theora buffer allocation, if any
if (!buffer_ready) {
- enc_data_t->yuv.y=(unsigned char *)malloc(enc_data_t->m_th_inf.height*
- enc_data_t->m_th_inf.width);
- enc_data_t->yuv.u=(unsigned char *)malloc(enc_data_t->m_th_inf.height*
- enc_data_t->m_th_inf.width/4);
- enc_data_t->yuv.v=(unsigned char *)malloc(enc_data_t->m_th_inf.height*
- enc_data_t->m_th_inf.width/4);
- enc_data_t->yuv.y_width=enc_data_t->m_th_inf.width;
- enc_data_t->yuv.y_height=enc_data_t->m_th_inf.height;
- enc_data_t->yuv.y_stride=enc_data_t->m_th_inf.width;
-
- enc_data_t->yuv.uv_width=enc_data_t->m_th_inf.width/2;
- enc_data_t->yuv.uv_height=enc_data_t->m_th_inf.height/2;
- enc_data_t->yuv.uv_stride=enc_data_t->m_th_inf.width/2;
- enc_data_t->x_offset=enc_data_t->m_th_inf.offset_x;
- enc_data_t->y_offset=enc_data_t->m_th_inf.offset_y;
+ enc_data_t->yuv.y = malloc(enc_data_t->m_th_inf.height * enc_data_t->m_th_inf.width);
+ enc_data_t->yuv.u = malloc(enc_data_t->m_th_inf.height * enc_data_t->m_th_inf.width / 4);
+ enc_data_t->yuv.v = malloc(enc_data_t->m_th_inf.height * enc_data_t->m_th_inf.width / 4);
+ enc_data_t->yuv.y_width = enc_data_t->m_th_inf.width;
+ enc_data_t->yuv.y_height = enc_data_t->m_th_inf.height;
+ enc_data_t->yuv.y_stride = enc_data_t->m_th_inf.width;
+
+ enc_data_t->yuv.uv_width = enc_data_t->m_th_inf.width / 2;
+ enc_data_t->yuv.uv_height = enc_data_t->m_th_inf.height / 2;
+ enc_data_t->yuv.uv_stride = enc_data_t->m_th_inf.width / 2;
+ enc_data_t->x_offset = enc_data_t->m_th_inf.offset_x;
+ enc_data_t->y_offset = enc_data_t->m_th_inf.offset_y;
}
theora_info_clear(&enc_data_t->m_th_inf);
© All Rights Reserved