1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
|
/*
* Copyright (C) 2013-2015 Intel Corporation
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
*/
#include <alsa/asoundlib.h>
#define TEMP_RECORD_FILE_NAME "/tmp/bat.wav.XXXXXX"
#define DEFAULT_DEV_NAME "default"
#define OPT_BASE 300
#define OPT_LOG (OPT_BASE + 1)
#define OPT_READFILE (OPT_BASE + 2)
#define OPT_SAVEPLAY (OPT_BASE + 3)
#define OPT_LOCAL (OPT_BASE + 4)
#define OPT_STANDALONE (OPT_BASE + 5)
#define COMPOSE(a, b, c, d) ((a) | ((b)<<8) | ((c)<<16) | ((d)<<24))
#define WAV_RIFF COMPOSE('R', 'I', 'F', 'F')
#define WAV_WAVE COMPOSE('W', 'A', 'V', 'E')
#define WAV_FMT COMPOSE('f', 'm', 't', ' ')
#define WAV_DATA COMPOSE('d', 'a', 't', 'a')
#define WAV_FORMAT_PCM 1 /* PCM WAVE file encoding */
#define MAX_CHANNELS 2
#define MIN_CHANNELS 1
#define MAX_PEAKS 10
#define MAX_FRAMES (10 * 1024 * 1024)
/* Given in ms */
#define CAPTURE_DELAY 500
/* signal frequency should be less than samplerate * RATE_FACTOR */
#define RATE_FACTOR 0.4
/* valid range of samplerate: (1 - RATE_RANGE, 1 + RATE_RANGE) * samplerate */
#define RATE_RANGE 0.05
/* Given in us */
#define MAX_BUFFERTIME 500000
/* devide factor, was 4, changed to 8 to remove reduce capture overrun */
#define DIV_BUFFERTIME 8
/* margin to avoid sign inversion when generate sine wav */
#define RANGE_FACTOR 0.95
#define EBATBASE 1000
#define ENOPEAK (EBATBASE + 1)
#define EONLYDC (EBATBASE + 2)
#define EBADPEAK (EBATBASE + 3)
#define DC_THRESHOLD 7.01
/* tolerance of detected peak = max (DELTA_HZ, DELTA_RATE * target_freq).
* If DELTA_RATE is too high, BAT may not be able to recognize negative result;
* if too low, BAT may be too sensitive and results in uncecessary failure. */
#define DELTA_RATE 0.005
#define DELTA_HZ 1
#define FOUND_DC (1<<1)
#define FOUND_WRONG_PEAK (1<<0)
/* Truncate sample frames to (1 << N), for faster FFT analysis process. The
* valid range of N is (SHIFT_MIN, SHIFT_MAX). When N increases, the analysis
* will be more time-consuming, and the result will be more accurate. */
#define SHIFT_MAX (sizeof(int) * 8 - 2)
#define SHIFT_MIN 8
struct wav_header {
unsigned int magic; /* 'RIFF' */
unsigned int length; /* file len */
unsigned int type; /* 'WAVE' */
};
struct wav_chunk_header {
unsigned int type; /* 'data' */
unsigned int length; /* sample count */
};
struct wav_fmt {
unsigned int magic; /* 'FMT '*/
unsigned int fmt_size; /* 16 or 18 */
unsigned short format; /* see WAV_FMT_* */
unsigned short channels;
unsigned int sample_rate; /* Frequency of sample */
unsigned int bytes_p_second;
unsigned short blocks_align; /* sample size; 1 or 2 bytes */
unsigned short sample_length; /* 8, 12 or 16 bit */
};
struct chunk_fmt {
unsigned short format; /* see WAV_FMT_* */
unsigned short channels;
unsigned int sample_rate; /* Frequency of sample */
unsigned int bytes_p_second;
unsigned short blocks_align; /* sample size; 1 or 2 bytes */
unsigned short sample_length; /* 8, 12 or 16 bit */
};
struct wav_container {
struct wav_header header;
struct wav_fmt format;
struct wav_chunk_header chunk;
};
struct bat;
enum _bat_op_mode {
MODE_UNKNOWN = -1,
MODE_SINGLE = 0,
MODE_LOOPBACK,
MODE_LAST
};
struct pcm {
char *device;
char *file;
enum _bat_op_mode mode;
void *(*fct)(struct bat *);
};
struct sin_generator;
struct sin_generator {
double state_real;
double state_imag;
double phasor_real;
double phasor_imag;
float frequency;
float sample_rate;
float magnitude;
};
struct bat {
unsigned int rate; /* sampling rate */
int channels; /* nb of channels */
int frames; /* nb of frames */
int frame_size; /* size of frame */
int sample_size; /* size of sample */
snd_pcm_format_t format; /* PCM format */
float sigma_k; /* threshold for peak detection */
float target_freq[MAX_CHANNELS];
int sinus_duration; /* number of frames for playback */
char *narg; /* argument string of duration */
char *logarg; /* path name of log file */
char *debugplay; /* path name to store playback signal */
bool standalone; /* enable to bypass analysis */
struct pcm playback;
struct pcm capture;
unsigned int periods_played;
unsigned int periods_total;
bool period_is_limited;
FILE *fp;
FILE *log;
FILE *err;
void (*convert_sample_to_double)(void *, double *, int);
void (*convert_float_to_sample)(float *, void *, int, int);
void *buf; /* PCM Buffer */
bool local; /* true for internal test */
};
struct analyze {
void *buf;
double *in;
double *out;
double *mag;
};
void prepare_wav_info(struct wav_container *, struct bat *);
int read_wav_header(struct bat *, char *, FILE *, bool);
int write_wav_header(FILE *, struct wav_container *, struct bat *);
int update_wav_header(struct bat *, FILE *, int);
|
