Imported Upstream version 3.0
[debian/gnuradio] / gnuradio-core / src / tests / benchmark_dotprod_fcc.cc
1 /* -*- c++ -*- */
2 /*
3  * Copyright 2002 Free Software Foundation, Inc.
4  *
5  * This file is part of GNU Radio
6  *
7  * GNU Radio is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License as published by
9  * the Free Software Foundation; either version 2, or (at your option)
10  * any later version.
11  *
12  * GNU Radio is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with GNU Radio; see the file COPYING.  If not, write to
19  * the Free Software Foundation, Inc., 51 Franklin Street,
20  * Boston, MA 02110-1301, USA.
21  */
22 #ifdef HAVE_CONFIG_H
23 #include "config.h"
24 #endif
25 #include <stdio.h>
26 #include <sys/time.h>
27 #ifdef HAVE_SYS_RESOURCE_H
28 #include <sys/resource.h>
29 #endif
30 #include <unistd.h>
31 #include <gr_fir_util.h>
32 #include <gr_fir_fcc.h>
33 #include <random.h>
34
35 #define TOTAL_TEST_SIZE        (40 * 1000 * 1000L)
36 #define NTAPS                                256
37 #define BLOCK_SIZE                   (50 * 1000)        /* fits in cache */
38
39 #if ((TOTAL_TEST_SIZE % BLOCK_SIZE) != 0)
40 #error "TOTAL_TEST_SIZE % BLOCK_SIZE must equal 0"
41 #endif
42
43 typedef gr_fir_fcc* (*fir_maker_t)(const std::vector<gr_complex> &taps);
44 typedef gr_fir_fcc  filter_t;
45
46
47 static double
48 timeval_to_double (const struct timeval *tv)
49 {
50   return (double) tv->tv_sec + (double) tv->tv_usec * 1e-6;
51 }
52
53
54 static void
55 benchmark (fir_maker_t filter_maker, const char *implementation_name)
56 {
57   int   i;
58   gr_complex coeffs[NTAPS];
59   float input[BLOCK_SIZE + NTAPS];
60   long  n;
61   gr_complex    result;
62 #ifdef HAVE_SYS_RESOURCE_H
63   struct rusage rusage_start;
64   struct rusage rusage_stop;
65 #else
66   double clock_start;
67   double clock_end;
68 #endif
69
70
71
72   // setup coefficients and input data
73
74   for (i = 0; i < NTAPS; i++)
75     coeffs[i] = gr_complex(random() - RANDOM_MAX/2, random() - RANDOM_MAX/2);
76
77   for (i = 0; i < BLOCK_SIZE + NTAPS; i++)
78     input[i] = random() - RANDOM_MAX/2;
79
80   std::vector<gr_complex> taps (&coeffs[0], &coeffs[NTAPS]);
81   filter_t *f = filter_maker (taps);
82   
83   // get starting CPU usage
84 #ifdef HAVE_SYS_RESOURCE_H
85   if (getrusage (RUSAGE_SELF, &rusage_start) < 0){
86     perror ("getrusage");
87     exit (1);
88   }
89 #else
90   clock_start = (double) clock() * (1000000. / CLOCKS_PER_SEC);
91 #endif
92
93   // do the actual work
94
95   for (n = 0; n < TOTAL_TEST_SIZE; n += BLOCK_SIZE){
96     int j;
97     for (j = 0; j < BLOCK_SIZE; j++){
98       result = f->filter (&input[j]);
99     }
100   }
101
102   // get ending CPU usage
103
104 #ifdef HAVE_SYS_RESOURCE_H
105   if (getrusage (RUSAGE_SELF, &rusage_stop) < 0){
106     perror ("getrusage");
107     exit (1);
108   }
109
110   // compute results
111
112   double user =
113     timeval_to_double (&rusage_stop.ru_utime)
114     - timeval_to_double (&rusage_start.ru_utime);
115
116   double sys =
117     timeval_to_double (&rusage_stop.ru_stime)
118     - timeval_to_double (&rusage_start.ru_stime);
119
120   double total = user + sys;
121 #else
122   clock_end= (double) clock() * (1000000. / CLOCKS_PER_SEC);
123   double total = clock_end - clock_start;
124 #endif
125
126   double macs = NTAPS * (double) TOTAL_TEST_SIZE;
127
128   printf ("%10s: taps: %4d  input: %4g  cpu: %6.3f  taps/sec: %10.4g  \n",
129           implementation_name, NTAPS, (double) TOTAL_TEST_SIZE, total, macs / total);
130
131   delete f;
132 }
133
134 static void
135 do_all ()
136 {
137   std::vector<gr_fir_fcc_info>          info;
138   gr_fir_util::get_gr_fir_fcc_info (&info);     // get all known FCC implementations
139
140   for (std::vector<gr_fir_fcc_info>::iterator p = info.begin ();
141        p != info.end () ;
142        ++p){
143     benchmark (p->create, p->name);
144   }
145 }
146
147 int
148 main (int argc, char **argv)
149 {
150   do_all ();
151   return 0;
152 }