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
|
/*
* This file is part of the OpenKinect Project. http://www.openkinect.org
*
* Copyright (c) 2011 individual OpenKinect contributors. See the CONTRIB file
* for details.
*
* This code is licensed to you under the terms of the Apache License, version
* 2.0, or, at your option, the terms of the GNU General Public License,
* version 2.0. See the APACHE20 and GPL2 files for the text of the licenses,
* or the following URLs:
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.gnu.org/licenses/gpl-2.0.txt
*
* If you redistribute this file in source form, modified or unmodified, you
* may:
* 1) Leave this header intact and distribute it under the same terms,
* accompanying it with the APACHE20 and GPL20 files, or
* 2) Delete the Apache 2.0 clause and accompany it with the GPL2 file, or
* 3) Delete the GPL v2 clause and accompany it with the APACHE20 file
* In all cases you must keep the copyright notice intact and include a copy
* of the CONTRIB file.
*
* Binary distributions must follow the binary distribution requirements of
* either License.
*/
#include "libfreenect.h"
#include "libfreenect_audio.h"
#include <stdio.h>
#include <signal.h>
static freenect_context* f_ctx;
static freenect_device* f_dev;
int die = 0;
char wavheader[] = {
0x52, 0x49, 0x46, 0x46, // ChunkID = "RIFF"
0x00, 0x00, 0x00, 0x00, // Chunksize (will be overwritten later)
0x57, 0x41, 0x56, 0x45, // Format = "WAVE"
0x66, 0x6d, 0x74, 0x20, // Subchunk1ID = "fmt "
0x10, 0x00, 0x00, 0x00, // Subchunk1Size = 16
0x01, 0x00, 0x01, 0x00, // AudioFormat = 1 (linear quantization) | NumChannels = 1
0x80, 0x3e, 0x00, 0x00, // SampleRate = 16000 Hz
0x00, 0xfa, 0x00, 0x00, // ByteRate = SampleRate * NumChannels * BitsPerSample/8 = 64000
0x04, 0x00, 0x20, 0x00, // BlockAlign = NumChannels * BitsPerSample/8 = 4 | BitsPerSample = 32
0x64, 0x61, 0x74, 0x61, // Subchunk2ID = "data"
0x00, 0x00, 0x00, 0x00, // Subchunk2Size = NumSamples * NumChannels * BitsPerSample / 8 (will be overwritten later)
};
typedef struct {
FILE* logfiles[4];
int samples;
} capture;
void in_callback(freenect_device* dev, int num_samples,
int32_t* mic1, int32_t* mic2,
int32_t* mic3, int32_t* mic4,
int16_t* cancelled, void *unknown) {
capture* c = (capture*)freenect_get_user(dev);
fwrite(mic1, 1, num_samples*sizeof(int32_t), c->logfiles[0]);
fwrite(mic2, 1, num_samples*sizeof(int32_t), c->logfiles[1]);
fwrite(mic3, 1, num_samples*sizeof(int32_t), c->logfiles[2]);
fwrite(mic4, 1, num_samples*sizeof(int32_t), c->logfiles[3]);
c->samples += num_samples;
printf("Sample received. Total samples recorded: %d\n", c->samples);
}
void cleanup(int sig) {
printf("Caught SIGINT, cleaning up\n");
die = 1;
}
int main(int argc, char** argv) {
if (freenect_init(&f_ctx, NULL) < 0) {
printf("freenect_init() failed\n");
return 1;
}
freenect_set_log_level(f_ctx, FREENECT_LOG_SPEW);
freenect_select_subdevices(f_ctx, FREENECT_DEVICE_AUDIO);
int nr_devices = freenect_num_devices (f_ctx);
printf ("Number of devices found: %d\n", nr_devices);
if (nr_devices < 1) {
freenect_shutdown(f_ctx);
return 1;
}
int user_device_number = 0;
if (freenect_open_device(f_ctx, &f_dev, user_device_number) < 0) {
printf("Could not open device\n");
freenect_shutdown(f_ctx);
return 1;
}
capture state;
state.samples = 0;
state.logfiles[0] = fopen("channel1.wav", "wb");
state.logfiles[1] = fopen("channel2.wav", "wb");
state.logfiles[2] = fopen("channel3.wav", "wb");
state.logfiles[3] = fopen("channel4.wav", "wb");
fwrite(wavheader, 1, 44, state.logfiles[0]);
fwrite(wavheader, 1, 44, state.logfiles[1]);
fwrite(wavheader, 1, 44, state.logfiles[2]);
fwrite(wavheader, 1, 44, state.logfiles[3]);
freenect_set_user(f_dev, &state);
freenect_set_audio_in_callback(f_dev, in_callback);
freenect_start_audio(f_dev);
signal(SIGINT, cleanup);
while(!die && freenect_process_events(f_ctx) >= 0) {
// If we did anything else, it might go here.
// Alternately, we might split off another thread
// to do this loop while the main thread did something
// interesting.
}
// Make the WAV header valid for each of the four files
int i;
for(i = 0; i < 4 ; i++) {
char buf[4];
fseek(state.logfiles[i], 4, SEEK_SET);
// Write ChunkSize = 36 + subchunk2size
int chunksize = state.samples * 4 + 36;
buf[0] = (chunksize & 0x000000ff);
buf[1] = (chunksize & 0x0000ff00) >> 8;
buf[2] = (chunksize & 0x00ff0000) >> 16;
buf[3] = (chunksize & 0xff000000) >> 24;
fwrite(buf, 1, 4,state.logfiles[i]);
fseek(state.logfiles[i], 40, SEEK_SET);
// Write Subchunk2Size = NumSamples * NumChannels (1) * BitsPerSample/8 (4)
int subchunk2size = state.samples * 4;
buf[0] = (subchunk2size & 0x000000ff);
buf[1] = (subchunk2size & 0x0000ff00) >> 8;
buf[2] = (subchunk2size & 0x00ff0000) >> 16;
buf[3] = (subchunk2size & 0xff000000) >> 24;
fwrite(buf, 1, 4,state.logfiles[i]);
fclose(state.logfiles[i]);
}
freenect_shutdown(f_ctx);
return 0;
}
|
