Merge commit 'a6a750c7ef240b72ce01e9653343a0ddf247d196'
[ffmpeg.git] / libavutil / tests / lzo.c
1 /*
2  * Copyright (c) 2006 Reimar Doeffinger
3  *
4  * This file is part of FFmpeg.
5  *
6  * FFmpeg is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * FFmpeg is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with FFmpeg; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19  */
20
21 #include <stdio.h>
22 #include <lzo/lzo1x.h>
23
24 #include "libavutil/log.h"
25 #include "libavutil/lzo.h"
26 #include "libavutil/mem.h"
27
28 #define MAXSZ (10*1024*1024)
29
30 /* Define one of these to 1 if you wish to benchmark liblzo
31  * instead of our native implementation. */
32 #define BENCHMARK_LIBLZO_SAFE   0
33 #define BENCHMARK_LIBLZO_UNSAFE 0
34
35 int main(int argc, char *argv[]) {
36     FILE *in = fopen(argv[1], "rb");
37     int comp_level = argc > 2 ? atoi(argv[2]) : 0;
38     uint8_t *orig = av_malloc(MAXSZ + 16);
39     uint8_t *comp = av_malloc(2*MAXSZ + 16);
40     uint8_t *decomp = av_malloc(MAXSZ + 16);
41     size_t s = fread(orig, 1, MAXSZ, in);
42     lzo_uint clen = 0;
43     long tmp[LZO1X_MEM_COMPRESS];
44     int inlen, outlen;
45     int i;
46     av_log_set_level(AV_LOG_DEBUG);
47     if (comp_level == 0) {
48         lzo1x_1_compress(orig, s, comp, &clen, tmp);
49     } else if (comp_level == 11) {
50         lzo1x_1_11_compress(orig, s, comp, &clen, tmp);
51     } else if (comp_level == 12) {
52         lzo1x_1_12_compress(orig, s, comp, &clen, tmp);
53     } else if (comp_level == 15) {
54         lzo1x_1_15_compress(orig, s, comp, &clen, tmp);
55     } else
56         lzo1x_999_compress(orig, s, comp, &clen, tmp);
57     for (i = 0; i < 300; i++) {
58 START_TIMER
59         inlen = clen; outlen = MAXSZ;
60 #if BENCHMARK_LIBLZO_SAFE
61         if (lzo1x_decompress_safe(comp, inlen, decomp, &outlen, NULL))
62 #elif BENCHMARK_LIBLZO_UNSAFE
63         if (lzo1x_decompress(comp, inlen, decomp, &outlen, NULL))
64 #else
65         if (av_lzo1x_decode(decomp, &outlen, comp, &inlen))
66 #endif
67             av_log(NULL, AV_LOG_ERROR, "decompression error\n");
68 STOP_TIMER("lzod")
69     }
70     if (memcmp(orig, decomp, s))
71         av_log(NULL, AV_LOG_ERROR, "decompression incorrect\n");
72     else
73         av_log(NULL, AV_LOG_ERROR, "decompression OK\n");
74     fclose(in);
75     av_free(orig);
76     av_free(comp);
77     av_free(decomp);
78     return 0;
79 }