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
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
|
/* $Id$ */
#include <pjmedia.h>
#include <pjlib-util.h>
#include <pjlib.h>
#include <stdio.h>
#include <stdlib.h>
#define app_perror(a,b,c) printf("%s: %s (%d)", a, b, c)
/* For logging purpose. */
#define THIS_FILE "cmp_wav.c"
#define BYTES_PER_FRAME 512
static const char *desc =
" FILE \n"
" \n"
" cmp_wav.c \n"
" \n"
" PURPOSE \n"
" \n"
" Compare two WAV files. \n"
" \n"
" USAGE \n"
" \n"
" cmp_wav ORIGINAL_WAV DEGRADED_WAV [TIME] [DETAIL] \n"
" \n"
" ORIGINAL_WAV The original WAV file as reference. \n"
" DEGRADED_WAV The degraded WAV file. \n"
" TIME Compare only some part of the files \n"
" (in ms, since the beginning). \n"
" Specify 0 (default) to compare the whole time. \n"
" DETAIL Show detail result, 1 or 0 (default=0, means no)\n"
" \n"
" Both files must have same clock rate and must contain \n"
" uncompressed (i.e. 16bit) PCM. \n";
/* Sum of multiplication of corresponding samples in buf1 & buf2 */
double sum_mult_sig(pj_int16_t *buf1, pj_int16_t *buf2, unsigned nsamples)
{
double mag = 0;
while (nsamples--)
mag += (double)*buf1++ * (double)*buf2++;
return mag;
}
/*
* main()
*/
int main(int argc, char *argv[])
{
pj_caching_pool cp;
pjmedia_endpt *med_endpt;
pj_pool_t *pool;
pjmedia_port *file_ori_port;
pjmedia_port *file_deg_port;
pj_status_t status;
unsigned first_nsamples = 0;
unsigned samples_compared = 0;
char buf1[BYTES_PER_FRAME];
char buf2[BYTES_PER_FRAME];
double ref_mag = 0;
double deg_mag = 0;
double mix_mag = 0;
int detail = 0;
int res_deg, res_mix, res_overall;
if (argc < 3) {
puts("Error: original & degraded filename required");
puts(desc);
return 1;
}
/* Set log level. */
pj_log_set_level(3);
/* Must init PJLIB first: */
status = pj_init();
PJ_ASSERT_RETURN(status == PJ_SUCCESS, 1);
/* Must create a pool factory before we can allocate any memory. */
pj_caching_pool_init(&cp, &pj_pool_factory_default_policy, 0);
/*
* Initialize media endpoint.
* This will implicitly initialize PJMEDIA too.
*/
status = pjmedia_endpt_create(&cp.factory, NULL, 1, &med_endpt);
PJ_ASSERT_RETURN(status == PJ_SUCCESS, 1);
/* Create memory pool for our file player */
pool = pj_pool_create( &cp.factory, /* pool factory */
"wav", /* pool name. */
4000, /* init size */
4000, /* increment size */
NULL /* callback on error */
);
/* Create file media port from the original WAV file */
status = pjmedia_wav_player_port_create( pool, /* memory pool */
argv[1], /* file to play */
40, /* ptime. */
PJMEDIA_FILE_NO_LOOP, /* flags */
0, /* default buffer */
&file_ori_port/* returned port */
);
if (status != PJ_SUCCESS) {
app_perror(THIS_FILE, "Unable to use WAV file", status);
return 1;
}
/* Create file media port from the degraded WAV file */
status = pjmedia_wav_player_port_create( pool, /* memory pool */
argv[2], /* file to play */
40, /* ptime. */
PJMEDIA_FILE_NO_LOOP, /* flags */
0, /* default buffer */
&file_deg_port/* returned port */
);
if (status != PJ_SUCCESS) {
app_perror(THIS_FILE, "Unable to use WAV file", status);
return 1;
}
if (file_ori_port->info.clock_rate != file_deg_port->info.clock_rate) {
app_perror(THIS_FILE, "Clock rates must be same.", PJ_EINVAL);
return 1;
}
if (argc > 3)
first_nsamples = atoi(argv[3]) * file_ori_port->info.clock_rate / 1000;
if (argc > 4)
detail = atoi(argv[4]);
while (1) {
pjmedia_frame f1, f2;
f1.buf = buf1;
f1.size = BYTES_PER_FRAME;
f2.buf = buf2;
f2.size = BYTES_PER_FRAME;
status = pjmedia_port_get_frame(file_ori_port, &f1);
if (status == PJ_EEOF) {
break;
} else if (status != PJ_SUCCESS) {
app_perror(THIS_FILE, "Error occured while reading file", status);
break;
}
status = pjmedia_port_get_frame(file_deg_port, &f2);
if (status == PJ_EEOF) {
break;
} else if (status != PJ_SUCCESS) {
app_perror(THIS_FILE, "Error occured while reading file", status);
break;
}
/* Calculate magnitudes */
ref_mag += sum_mult_sig(f1.buf, f1.buf, BYTES_PER_FRAME >> 1);
deg_mag += sum_mult_sig(f2.buf, f2.buf, BYTES_PER_FRAME >> 1);
mix_mag += sum_mult_sig(f1.buf, f2.buf, BYTES_PER_FRAME >> 1);
samples_compared += BYTES_PER_FRAME >> 1;
if (first_nsamples && samples_compared >= first_nsamples)
break;
}
/* Degraded magnitude compared to reference magnitude
*/
res_deg = (int) (deg_mag / ref_mag * 100.0);
if (res_deg < 0)
res_deg = -1;
else if (res_deg >= 81)
res_deg = 9;
else
res_deg = pj_isqrt(res_deg);
/* Mixed magnitude (don't know what this is actually :D) compared to
* reference magnitude
*/
res_mix = (int) (mix_mag / ref_mag * 100.0);
if (res_mix < 0)
res_mix = -1;
else if (res_mix >= 81)
res_mix = 9;
else
res_mix = pj_isqrt(res_mix);
/* Overall score.
* If mixed score is -1, then overall score should be -1 as well.
* Apply no weighting (1:1) for now.
*/
if (res_mix == -1)
res_overall = -1;
else
res_overall = (res_mix*1 + res_deg*1) / 2;
if (detail) {
printf("Reference = %.0f\n", ref_mag);
printf("Degraded = %.0f\n", deg_mag);
printf("Mixed = %.0f\n", mix_mag);
printf("\n");
printf("Score 1 = %d\n", res_deg);
printf("Score 2 = %d\n", res_mix);
printf("\n");
}
printf("Overall = %d\n", res_overall);
/* Destroy file port */
status = pjmedia_port_destroy( file_ori_port );
PJ_ASSERT_RETURN(status == PJ_SUCCESS, 1);
status = pjmedia_port_destroy( file_deg_port );
PJ_ASSERT_RETURN(status == PJ_SUCCESS, 1);
/* Release application pool */
pj_pool_release( pool );
/* Destroy media endpoint. */
pjmedia_endpt_destroy( med_endpt );
/* Destroy pool factory */
pj_caching_pool_destroy( &cp );
/* Shutdown PJLIB */
pj_shutdown();
/* Done. */
return 0;
}
|
