[tests] improve sink_custom tests
[aubio.git] / tests / utils_tests.h
1 #include <time.h>
2 #include <stdlib.h>
3 #include <stdio.h>
4 #include <math.h>
5 #include <assert.h>
6 #include "config.h"
7
8 #ifdef HAVE_STRING_H
9 #include <string.h>
10 #endif
11
12 #ifdef HAVE_UNISTD_H
13 #include <unistd.h> // unlink, close
14 #endif
15
16 #ifdef HAVE_LIMITS_H
17 #include <limits.h> // PATH_MAX
18 #endif /* HAVE_LIMITS_H */
19 #ifndef PATH_MAX
20 #define PATH_MAX 1024
21 #endif
22
23 #if defined(HAVE_WIN_HACKS) && !defined(__GNUC__)
24 #include <io.h> // _access
25 #endif
26
27 // This macro is used to pass a string to msvc compiler: since msvc's -D flag
28 // strips the quotes, we define the string without quotes and re-add them with
29 // this macro.
30
31 #define REDEFINESTRING(x) #x
32 #define DEFINEDSTRING(x) REDEFINESTRING(x)
33
34 #ifndef AUBIO_TESTS_SOURCE
35 #error "AUBIO_TESTS_SOURCE is not defined"
36 #endif
37
38 #ifdef HAVE_C99_VARARGS_MACROS
39 #define PRINT_ERR(...)   fprintf(stderr, "AUBIO-TESTS ERROR: " __VA_ARGS__)
40 #define PRINT_MSG(...)   fprintf(stdout, __VA_ARGS__)
41 #define PRINT_DBG(...)   fprintf(stderr, __VA_ARGS__)
42 #define PRINT_WRN(...)   fprintf(stderr, "AUBIO-TESTS WARNING: " __VA_ARGS__)
43 #else
44 #define PRINT_ERR(format, args...)   fprintf(stderr, "AUBIO-TESTS ERROR: " format , ##args)
45 #define PRINT_MSG(format, args...)   fprintf(stdout, format , ##args)
46 #define PRINT_DBG(format, args...)   fprintf(stderr, format , ##args)
47 #define PRINT_WRN(format, args...)   fprintf(stderr, "AUBIO-TESTS WARNING: " format, ##args)
48 #endif
49
50 #ifndef M_PI
51 #define M_PI         (3.14159265358979323846)
52 #endif
53
54 #ifndef RAND_MAX
55 #define RAND_MAX 32767
56 #endif
57
58 // are we on windows ? or are we using -std=c99 ?
59 #if defined(HAVE_WIN_HACKS) || defined(__STRICT_ANSI__)
60 // http://en.wikipedia.org/wiki/Linear_congruential_generator
61 // no srandom/random on win32
62
63 uint_t srandom_seed = 1029;
64
65 void srandom(uint_t new_seed) {
66     srandom_seed = new_seed;
67 }
68
69 uint_t random(void) {
70     srandom_seed = 1664525 * srandom_seed + 1013904223;
71     return srandom_seed;
72 }
73 #endif
74
75 void utils_init_random (void);
76
77 void utils_init_random (void) {
78   time_t now = time(0);
79   struct tm *tm_struct = localtime(&now);
80   size_t **tm_address = (void*)&tm_struct;
81   int seed = tm_struct->tm_sec + (size_t)tm_address;
82   //PRINT_WRN("current seed: %d\n", seed);
83   srandom ((unsigned int)seed);
84 }
85
86 // create_temp_sink / close_temp_sink
87 #if defined(__GNUC__) // mkstemp
88
89 int check_source(char *source_path)
90 {
91   return access(source_path, R_OK);
92 }
93
94 int create_temp_sink(char *sink_path)
95 {
96   return mkstemp(sink_path);
97 }
98
99 int close_temp_sink(char *sink_path, int sink_fildes)
100 {
101   int err;
102   if ((err = close(sink_fildes)) != 0) return err;
103   if ((err = unlink(sink_path)) != 0) return err;
104   return err;
105 }
106
107 #elif defined(HAVE_WIN_HACKS) //&& !defined(__GNUC__)
108 // windows workaround, where mkstemp does not exist...
109
110 int check_source(char *source_path)
111 {
112   return _access(source_path, 04);
113 }
114
115 int create_temp_sink(char *templ)
116 {
117   int i = 0;
118   static const char letters[] = "abcdefg0123456789";
119   int letters_len = strlen(letters);
120   int templ_len = strlen(templ);
121   if (templ_len == 0) return 0;
122   utils_init_random();
123   for (i = 0; i < 6; i++)
124   {
125     templ[templ_len - i] = letters[rand() % letters_len];
126   }
127   return 1;
128 }
129
130 int close_temp_sink(char* sink_path, int sink_fildes) {
131   // the file should be closed when not using mkstemp, no need to open it
132   if (sink_fildes == 0) return 1;
133   return _unlink(sink_path);
134 }
135
136 #else // windows workaround
137 // otherwise, we don't really know what to do yet
138 #error "mkstemp undefined, but not on windows. additional workaround required."
139 #endif
140
141 // pass progname / default
142 int run_on_default_source( int main(int, char**) )
143 {
144   const int argc = 2;
145   int err = 0;
146   char** argv = (char**)calloc(argc, sizeof(char*));
147   argv[0] = __FILE__;
148   argv[1] = DEFINEDSTRING(AUBIO_TESTS_SOURCE);
149   // check if the file can be read
150   if ( check_source(argv[1]) ) return 1;
151   err = main(argc, argv);
152   if (argv) free(argv);
153   return err;
154 }
155
156 int run_on_default_sink( int main(int, char**) )
157 {
158   const int argc = 2;
159   int err = 0;
160   char** argv = (char**)calloc(argc, sizeof(char*));
161   char sink_path[PATH_MAX] = "tmp_aubio_XXXXXX";
162   int fd = create_temp_sink(sink_path);
163   if (!fd) return 1;
164   argv[0] = __FILE__;
165   argv[1] = sink_path;
166   err = main(argc, argv);
167   close_temp_sink(sink_path, fd);
168   if (argv) free(argv);
169   return err;
170 }
171
172 int run_on_default_source_and_sink( int main(int, char**) )
173 {
174   const int argc = 3;
175   int err = 0;
176   char** argv = (char**)calloc(argc, sizeof(char*));
177   char sink_path[PATH_MAX] = "tmp_aubio_XXXXXX";
178   int fd = create_temp_sink(sink_path);
179   if (!fd) return 1;
180   argv[0] = __FILE__;
181   argv[1] = DEFINEDSTRING(AUBIO_TESTS_SOURCE);
182   argv[2] = sink_path;
183   // check if the file can be read
184   if ( check_source(argv[1]) ) return 1;
185   err = main(argc, argv);
186   close_temp_sink(sink_path, fd);
187   if (argv) free(argv);
188   return err;
189 }