diff options
Diffstat (limited to 'src/rmd_rescue.c')
| -rw-r--r-- | src/rmd_rescue.c | 124 |
1 files changed, 124 insertions, 0 deletions
diff --git a/src/rmd_rescue.c b/src/rmd_rescue.c new file mode 100644 index 0000000..84bcf4b --- /dev/null +++ b/src/rmd_rescue.c @@ -0,0 +1,124 @@ +/****************************************************************************** +* recordMyDesktop * +******************************************************************************* +* * +* Copyright (C) 2006,2007,2008 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 johnvarouhakis@gmail.com * +******************************************************************************/ + +#include "config.h" +#include "rmd_rescue.h" + +#include "rmd_yuv_utils.h" +#include "rmd_encode_cache.h" +#include "rmd_initialize_data.h" +#include "rmd_register_callbacks.h" +#include "rmd_specsfile.h" +#include "rmd_types.h" + +#include <pthread.h> + +#include <string.h> +#include <stdlib.h> + + +int rmdRescue(const char *path) { + unsigned short width, height; + ProgData pdata; + EncData enc_data; + CacheData cache_data; + + rmdSetupDefaultArgs(&pdata.args); + + pdata.enc_data = &enc_data; + pdata.cache_data = &cache_data; + + //projname + cache_data.projname = malloc(strlen(path) + 2); + strcpy(cache_data.projname, path); + strcat(cache_data.projname, "/");//having two of these doesn't hurt... + //image data + cache_data.imgdata = malloc(strlen(cache_data.projname) + 11); + strcpy(cache_data.imgdata, cache_data.projname); + strcat(cache_data.imgdata, "img.out"); + //audio data + cache_data.audiodata = malloc(strlen(cache_data.projname) + 10); + strcpy(cache_data.audiodata, cache_data.projname); + strcat(cache_data.audiodata, "audio.pcm"); + //specsfile + cache_data.specsfile = malloc(strlen(cache_data.projname) + 10); + strcpy(cache_data.specsfile, cache_data.projname); + strcat(cache_data.specsfile, "specs.txt"); + + if (rmdReadSpecsFile(&pdata)) + return 1; + + width = ((pdata.brwin.rrect.width + 15) >> 4) << 4; + height = ((pdata.brwin.rrect.height + 15) >> 4) << 4; + + enc_data.yuv.y = malloc(height * width); + enc_data.yuv.u = malloc(height * width / 4); + enc_data.yuv.v = malloc(height * width / 4); + enc_data.yuv.y_width = width; + enc_data.yuv.y_height = height; + enc_data.yuv.y_stride = width; + + enc_data.yuv.uv_width = width / 2; + enc_data.yuv.uv_height = height / 2; + enc_data.yuv.uv_stride = width / 2; + + for (int i = 0; i < (enc_data.yuv.y_width * enc_data.yuv.y_height); i++) + enc_data.yuv.y[i] = 0; + + for (int i = 0; i < (enc_data.yuv.uv_width * enc_data.yuv.uv_height); i++) + enc_data.yuv.v[i] = enc_data.yuv.u[i] = 127; + + yblocks = malloc(sizeof(*yblocks) * (enc_data.yuv.y_width / Y_UNIT_WIDTH) * + (enc_data.yuv.y_height / Y_UNIT_WIDTH)); + ublocks = malloc(sizeof(*ublocks) * (enc_data.yuv.y_width / Y_UNIT_WIDTH) * + (enc_data.yuv.y_height / Y_UNIT_WIDTH)); + vblocks = malloc(sizeof(*vblocks) * (enc_data.yuv.y_width / Y_UNIT_WIDTH) * + (enc_data.yuv.y_height / Y_UNIT_WIDTH)); + + pdata.frametime = 1000000 / pdata.args.fps; + + pthread_mutex_init(&pdata.theora_lib_mutex, NULL); + pthread_mutex_init(&pdata.vorbis_lib_mutex, NULL); + pthread_mutex_init(&pdata.libogg_mutex, NULL); + pthread_cond_init(&pdata.theora_lib_clean, NULL); + pthread_cond_init(&pdata.vorbis_lib_clean, NULL); + pdata.th_encoding_clean = pdata.v_encoding_clean = 1; + pdata.avd = 0; + pdata.sound_buffer = NULL; + pdata.running = TRUE; + pdata.aborted = FALSE; + + rmdRegisterCallbacks(&pdata); + fprintf(stderr, "Restoring %s!!!\n", path); + + rmdEncodeCache(&pdata); + + fprintf(stderr, "Done!!!\n"); + fprintf(stderr, "Goodbye!\n"); + rmdCleanUp(); + + return 0; +} |