diff options
Diffstat (limited to 'recordmydesktop/src/recordmydesktop.c')
| -rw-r--r-- | recordmydesktop/src/recordmydesktop.c | 215 |
1 files changed, 215 insertions, 0 deletions
diff --git a/recordmydesktop/src/recordmydesktop.c b/recordmydesktop/src/recordmydesktop.c new file mode 100644 index 0000000..f0f1a3d --- /dev/null +++ b/recordmydesktop/src/recordmydesktop.c @@ -0,0 +1,215 @@ +/********************************************************************************* +* recordMyDesktop * +********************************************************************************** +* * +* Copyright (C) 2006 John Varouhakis * +* * +* * +* This program is free software; you can redistribute it and/or modify * +* it under the terms of the GNU General Public License as published by * +* the Free Software Foundation; either version 2 of the License, or * +* (at your option) any later version. * +* * +* This program is distributed in the hope that it will be useful, * +* but WITHOUT ANY WARRANTY; without even the implied warranty of * +* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * +* GNU General Public License for more details. * +* * +* You should have received a copy of the GNU General Public License * +* along with this program; if not, write to the Free Software * +* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA * +* * +* * +* * +* For further information contact me at biocrasher@gmail.com * +**********************************************************************************/ + + +#include <recordmydesktop.h> + + +int main(int argc,char **argv){ + ProgData pdata; + + if(XInitThreads ()==0){ + fprintf(stderr,"Couldn't initialize thread support!\n"); + exit(1); + } + DEFAULT_ARGS(&pdata.args); + if(ParseArgs(argc,argv,&pdata.args)){ + exit(1); + } + pdata.dpy = XOpenDisplay(pdata.args.display); + + if (pdata.dpy == NULL) { + fprintf(stderr, "Cannot connect to X server %s\n",pdata.args.display); + exit(1); + } + else{ + EncData enc_data; + pthread_t poll_damage_t, + image_capture_t, + image_encode_t, + sound_capture_t, + sound_encode_t, + flush_to_ogg_t; + XShmSegmentInfo shminfo; + + QUERY_DISPLAY_SPECS(pdata.dpy,&pdata.specs); + if(pdata.specs.depth!=24){ + fprintf(stderr,"Only 24bpp color depth mode is currently supported.\n"); + exit(1); + } + if(SetBRWindow(pdata.dpy,&pdata.brwin,&pdata.specs,&pdata.args)) + exit(1); + if(QueryExtensions(pdata.dpy,&pdata.args,&pdata.damage_event, &pdata.damage_error)) + exit(1); + + //init data + + if(!pdata.args.scshot){ + fprintf(stderr,"Initializing...\n"); + + if(pdata.args.have_dummy_cursor){ + pdata.dummy_pointer=MakeDummyPointer(&pdata.specs,16,pdata.args.cursor_color,0,&pdata.npxl); + pdata.dummy_p_size=16; + } + } + if((pdata.args.noshared)||(pdata.args.scshot)) + pdata.datamain=(char *)malloc(pdata.brwin.nbytes); + if(!pdata.args.scshot){ + if(pdata.args.noshared) + pdata.datatemp=(char *)malloc(pdata.brwin.nbytes); + pdata.rect_root[0]=pdata.rect_root[1]=NULL; + pthread_mutex_init(&pdata.list_mutex[0],NULL); + pthread_mutex_init(&pdata.list_mutex[1],NULL); + pthread_mutex_init(&pdata.sound_buffer_mutex,NULL); + pthread_mutex_init(&pdata.yuv_mutex,NULL); + + pthread_cond_init(&pdata.time_cond,NULL); + pthread_cond_init(&pdata.pause_cond,NULL); + pthread_cond_init(&pdata.image_buffer_ready,NULL); + pthread_cond_init(&pdata.sound_buffer_ready,NULL); + pthread_cond_init(&pdata.sound_data_read,NULL); + pdata.list_selector=Paused=Aborted=avd=0; + pdata.running=1; + time_cond=&pdata.time_cond; + pause_cond=&pdata.pause_cond; + Running=&pdata.running; + } + if((pdata.args.noshared)||(pdata.args.scshot)){ + pdata.image=XCreateImage(pdata.dpy, pdata.specs.visual, pdata.specs.depth, ZPixmap, 0,pdata.datamain,pdata.brwin.rgeom.width, + pdata.brwin.rgeom.height, 8, 0); + XInitImage(pdata.image); + GetZPixmap(pdata.dpy,pdata.specs.root,pdata.image->data,pdata.brwin.rgeom.x,pdata.brwin.rgeom.y, + pdata.brwin.rgeom.width,pdata.brwin.rgeom.height); + } + else{ + pdata.image=XShmCreateImage (pdata.dpy,pdata.specs.visual,pdata.specs.depth,ZPixmap,pdata.datamain, + &shminfo, pdata.brwin.rgeom.width,pdata.brwin.rgeom.height); + shminfo.shmid = shmget (IPC_PRIVATE, + pdata.image->bytes_per_line * pdata.image->height, + IPC_CREAT|0777); + shminfo.shmaddr = pdata.image->data = shmat (shminfo.shmid, 0, 0); + shminfo.readOnly = False; + if(!XShmAttach(pdata.dpy,&shminfo)){ + fprintf(stderr,"Failed to attach shared memory to proccess.\n"); + exit(1); + } + XShmGetImage(pdata.dpy,pdata.specs.root,pdata.image,0,0,AllPlanes); + + } + if(pdata.args.scshot){ + if(pdata.args.delay>0){ + fprintf(stderr,"Will sleep for %d seconds now.\n",pdata.args.delay); + sleep(pdata.args.delay); + } + //get a new screenshot + GetZPixmap(pdata.dpy,pdata.specs.root,pdata.image->data,pdata.brwin.rgeom.x,pdata.brwin.rgeom.y, + pdata.brwin.rgeom.width,pdata.brwin.rgeom.height); + ZPixmapToBMP(pdata.image,&pdata.brwin,((!strcmp(pdata.args.filename,"out.ogg"))?"rmdout.bmp":pdata.args.filename),pdata.brwin.nbytes,pdata.args.scale_shot); + fprintf(stderr,"done!\n"); + exit(0); + } + if(!pdata.args.nosound) + pdata.sound_handle=OpenDev(pdata.args.device,pdata.args.channels,&pdata.args.frequency,&pdata.periodsize, &pdata.periodtime,&pdata.hard_pause); + if(pdata.sound_handle==NULL){ + fprintf(stderr,"Error while opening/configuring soundcard %s\nProcceeding with no sound\n",pdata.args.device); + pdata.args.nosound=1; + } + InitEncoder(&pdata,&enc_data); + XImageToYUV(pdata.image,&pdata.enc_data->yuv); + + pdata.frametime=(1000000)/pdata.args.fps; + + if(pdata.args.delay>0){ + fprintf(stderr,"Will sleep for %d seconds now.\n",pdata.args.delay); + sleep(pdata.args.delay); + } + + /*start threads*/ + if(!pdata.args.full_shots) + pthread_create(&poll_damage_t,NULL,PollDamage,(void *)&pdata); + pthread_create(&image_capture_t,NULL,GetFrame,(void *)&pdata); + pthread_create(&image_encode_t,NULL,EncodeImageBuffer,(void *)&pdata); + if(!pdata.args.nosound){ + pthread_create(&sound_capture_t,NULL,CaptureSound,(void *)&pdata); + pthread_create(&sound_encode_t,NULL,EncodeSoundBuffer,(void *)&pdata); + } + pthread_create(&flush_to_ogg_t,NULL,FlushToOgg,(void *)&pdata); + + + RegisterCallbacks(&pdata.args); + fprintf(stderr,"Capturing!\n"); + + //wait all threads to finish + + pthread_join(image_capture_t,NULL); + fprintf(stderr,"Shutting down."); + pthread_join(image_encode_t,NULL); + fprintf(stderr,"."); + if(!pdata.args.nosound){ + pthread_join(sound_capture_t,NULL); + fprintf(stderr,"."); + pthread_join(sound_encode_t,NULL); + fprintf(stderr,"."); + } + else + fprintf(stderr,".."); + pthread_join(flush_to_ogg_t,NULL); + fprintf(stderr,"."); + if(!pdata.args.full_shots) + pthread_join(poll_damage_t,NULL); + fprintf(stderr,"."); + fprintf(stderr,"\n"); + XCloseDisplay(pdata.dpy); + if(Aborted){ + if(remove(pdata.args.filename)){ + perror("Error while removing file:\n"); + return 1; + } + else{ + fprintf(stderr,"SIGABRT received,file %s removed\n",pdata.args.filename); + return 0; + } + } + else + fprintf(stderr,"Goodbye!\n"); + } + return 0; +} + + + + + + + + + + + + + + + |