source: tests/utils_tests.h @ 46a1e34

feature/autosinkfeature/cnnfeature/cnn_orgfeature/constantqfeature/crepefeature/crepe_orgfeature/pitchshiftfeature/timestretchfix/ffmpeg5
Last change on this file since 46a1e34 was 8386dbe, checked in by Paul Brossier <piem@piem.org>, 5 years ago

[tests] add run_on_default_source_and_sink helper

  • Property mode set to 100644
File size: 4.7 KB
RevLine 
[474a573]1#include <time.h>
[248da64]2#include <stdlib.h>
3#include <stdio.h>
[474a573]4#include <math.h>
5#include <assert.h>
[06dba46]6#include "config.h"
[248da64]7
[d22b568]8#ifdef HAVE_STRING_H
9#include <string.h>
10#endif
11
[a6065b9]12#ifdef HAVE_UNISTD_H
13#include <unistd.h> // unlink, close
14#endif
15
[abe67e1]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
[752ce6d]23#if defined(HAVE_WIN_HACKS) && !defined(__GNUC__)
24#include <io.h> // _access
25#endif
26
[586383d]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
[c6bb567a]34#ifndef AUBIO_TESTS_SOURCE
35#error "AUBIO_TESTS_SOURCE is not defined"
36#endif
37
[dc08c4f]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
[474a573]44#define PRINT_ERR(format, args...)   fprintf(stderr, "AUBIO-TESTS ERROR: " format , ##args)
[248da64]45#define PRINT_MSG(format, args...)   fprintf(stdout, format , ##args)
46#define PRINT_DBG(format, args...)   fprintf(stderr, format , ##args)
[474a573]47#define PRINT_WRN(format, args...)   fprintf(stderr, "AUBIO-TESTS WARNING: " format, ##args)
[dc08c4f]48#endif
[474a573]49
[8b42c7d]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
[da632d3]58// are we on windows ? or are we using -std=c99 ?
59#if defined(HAVE_WIN_HACKS) || defined(__STRICT_ANSI__)
[06dba46]60// http://en.wikipedia.org/wiki/Linear_congruential_generator
61// no srandom/random on win32
62
63uint_t srandom_seed = 1029;
64
65void srandom(uint_t new_seed) {
66    srandom_seed = new_seed;
67}
68
69uint_t random(void) {
70    srandom_seed = 1664525 * srandom_seed + 1013904223;
71    return srandom_seed;
72}
73#endif
74
[158e031]75void utils_init_random (void);
76
77void utils_init_random (void) {
[474a573]78  time_t now = time(0);
79  struct tm *tm_struct = localtime(&now);
[cb288f2]80  size_t **tm_address = (void*)&tm_struct;
81  int seed = tm_struct->tm_sec + (size_t)tm_address;
[474a573]82  //PRINT_WRN("current seed: %d\n", seed);
[e7edf92]83  srandom ((unsigned int)seed);
[474a573]84}
[abe67e1]85
[aa15080]86// create_temp_sink / close_temp_sink
87#if defined(__GNUC__) // mkstemp
88
[c6bb567a]89int check_source(char *source_path)
90{
91  return access(source_path, R_OK);
92}
93
[aa15080]94int create_temp_sink(char *sink_path)
95{
96  return mkstemp(sink_path);
97}
98
99int 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
[6a91a9d]107#elif defined(HAVE_WIN_HACKS) //&& !defined(__GNUC__)
[aa15080]108// windows workaround, where mkstemp does not exist...
[c6bb567a]109
110int check_source(char *source_path)
111{
112  return _access(source_path, 04);
113}
114
[aa15080]115int 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
130int 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
[c6bb567a]141// pass progname / default
142int 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__;
[586383d]148  argv[1] = DEFINEDSTRING(AUBIO_TESTS_SOURCE);
[c6bb567a]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}
[aa15080]155
[abe67e1]156int run_on_default_sink( int main(int, char**) )
157{
[d22b568]158  const int argc = 2;
159  int err = 0;
[d94afb3]160  char** argv = (char**)calloc(argc, sizeof(char*));
[abe67e1]161  char sink_path[PATH_MAX] = "tmp_aubio_XXXXXX";
[aa15080]162  int fd = create_temp_sink(sink_path);
[abe67e1]163  if (!fd) return 1;
[aa15080]164  argv[0] = __FILE__;
[abe67e1]165  argv[1] = sink_path;
166  err = main(argc, argv);
[aa15080]167  close_temp_sink(sink_path, fd);
[d94afb3]168  if (argv) free(argv);
[abe67e1]169  return err;
170}
[8386dbe]171
172int 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}
Note: See TracBrowser for help on using the repository browser.