* minor update
[ffmpeg.git] / tests / dsptest.c
1 /*
2  * MMX optimized DSP utils
3  * Copyright (c) 2000, 2001 Gerard Lantau.
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License as published by
7  * the Free Software Foundation; either version 2 of the License, or
8  * (at your option) any later version.
9  *
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU General Public License for more details.
14  *
15  * You should have received a copy of the GNU General Public License
16  * along with this program; if not, write to the Free Software
17  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
18  *
19  */
20
21 #define TESTCPU_MAIN
22 #include "dsputil.h"
23 //#include "../libavcodec/dsputil.c"
24 #include "../libavcodec/i386/cputest.c"
25 #include "../libavcodec/i386/dsputil_mmx.c"
26 #undef TESTCPU_MAIN
27
28 #define PAD 0x10000
29 /*
30  * for testing speed of various routine - should be probably extended
31  * for a general purpose regression test later
32  *
33  * currently only for i386 - FIXME
34  */
35
36 #define PIX_FUNC_C(a) \
37     { #a "_c", a ## _c, 0 }, \
38     { #a "_mmx", a ## _mmx, MM_MMX }, \
39     { #a "_mmx2", a ## _mmx2, MM_MMXEXT | PAD }
40
41 #define PIX_FUNC(a) \
42     { #a "_mmx", a ## _mmx, MM_MMX }, \
43     { #a "_3dnow", a ## _3dnow, MM_3DNOW }, \
44     { #a "_mmx2", a ## _mmx2, MM_MMXEXT | PAD }
45
46 #define PIX_FUNC_MMX(a) \
47     { #a "_mmx", a ## _mmx, MM_MMX | PAD }
48
49 /*
50     PIX_FUNC_C(pix_abs16x16),
51     PIX_FUNC_C(pix_abs16x16_x2),
52     PIX_FUNC_C(pix_abs16x16_y2),
53     PIX_FUNC_C(pix_abs16x16_xy2),
54     PIX_FUNC_C(pix_abs8x8),
55     PIX_FUNC_C(pix_abs8x8_x2),
56     PIX_FUNC_C(pix_abs8x8_y2),
57     PIX_FUNC_C(pix_abs8x8_xy2),
58 */
59
60 static const struct pix_func {
61     char* name;
62     op_pixels_func func;
63     int mm_flags;
64 } pix_func[] = {
65
66     PIX_FUNC_MMX(put_pixels),
67 #if 1
68     PIX_FUNC(put_pixels_x2),
69     PIX_FUNC(put_pixels_y2),
70     PIX_FUNC_MMX(put_pixels_xy2),
71
72     PIX_FUNC(put_no_rnd_pixels_x2),
73     PIX_FUNC(put_no_rnd_pixels_y2),
74     PIX_FUNC_MMX(put_no_rnd_pixels_xy2),
75
76     PIX_FUNC(avg_pixels),
77     PIX_FUNC(avg_pixels_x2),
78     PIX_FUNC(avg_pixels_y2),
79     PIX_FUNC(avg_pixels_xy2),
80 #endif
81     { 0, 0 }
82 };
83
84 static inline long long rdtsc()
85 {
86     long long l;
87     asm volatile(   "rdtsc\n\t"
88                     : "=A" (l)
89                 );
90     return l;
91 }
92
93 static test_speed(int step)
94 {
95     const struct pix_func* pix = pix_func;
96     const int linesize = 720;
97     char empty[32768];
98     char* bu =(char*)(((long)empty + 32) & ~0xf);
99
100     int sum = 0;
101
102     while (pix->name)
103     {
104         int i;
105         uint64_t te, ts;
106         op_pixels_func func = pix->func;
107         char* im = bu;
108
109         if (!(pix->mm_flags & mm_flags))
110             continue;
111
112         printf("%30s... ", pix->name);
113         fflush(stdout);
114         ts = rdtsc();
115         for(i=0; i<100000; i++){
116             func(im, im + 1000, linesize, 16);
117             im += step;
118             if (im > bu + 20000)
119                 im = bu;
120         }
121         te = rdtsc();
122         emms();
123         printf("% 9d\n", (int)(te - ts));
124         sum += (te - ts) / 100000;
125         if (pix->mm_flags & PAD)
126             puts("");
127         pix++;
128     }
129
130     printf("Total sum: %d\n", sum);
131 }
132
133 int main(int argc, char* argv[])
134 {
135     int step = 16;
136
137     if (argc > 1)
138     {
139         // something simple for now
140         if (argc > 2 && (strcmp("-s", argv[1]) == 0
141                          || strcmp("-step", argv[1]) == 0))
142             step = atoi(argv[2]);
143     }
144
145     mm_flags = mm_support();
146     printf("dsptest: CPU flags:");
147     if (mm_flags & MM_MMX)
148         printf(" mmx");
149     if (mm_flags & MM_MMXEXT)
150         printf(" mmxext");
151     if (mm_flags & MM_3DNOW)
152         printf(" 3dnow");
153     if (mm_flags & MM_SSE)
154         printf(" sse");
155     if (mm_flags & MM_SSE2)
156         printf(" sse2");
157     printf("\n");
158
159     printf("Using step: %d\n", step);
160     test_speed(step);
161 }