1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
/******************************************************************************
* 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_us = 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;
}
|
