1 |
/*************************************************************************** |
2 |
* * |
3 |
* libgig - C++ cross-platform Gigasampler format file access library * |
4 |
* * |
5 |
* Copyright (C) 2003-2007 by Christian Schoenebeck * |
6 |
* <cuse@users.sourceforge.net> * |
7 |
* * |
8 |
* This program is part of libgig. * |
9 |
* * |
10 |
* This program is free software; you can redistribute it and/or modify * |
11 |
* it under the terms of the GNU General Public License as published by * |
12 |
* the Free Software Foundation; either version 2 of the License, or * |
13 |
* (at your option) any later version. * |
14 |
* * |
15 |
* This program is distributed in the hope that it will be useful, * |
16 |
* but WITHOUT ANY WARRANTY; without even the implied warranty of * |
17 |
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * |
18 |
* GNU General Public License for more details. * |
19 |
* * |
20 |
* You should have received a copy of the GNU General Public License * |
21 |
* along with this program; if not, write to the Free Software * |
22 |
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, * |
23 |
* MA 02111-1307 USA * |
24 |
***************************************************************************/ |
25 |
|
26 |
#ifdef HAVE_CONFIG_H |
27 |
#include <config.h> |
28 |
#endif |
29 |
|
30 |
// just for testing the disk streaming capability of libgig; |
31 |
// you will also need to set this to 1 at the moment if you want to |
32 |
// extract compressed samples, as I haven't implemented the |
33 |
// decompression algorithm in gig::Sample::LoadSampleData() yet |
34 |
#define USE_DISK_STREAMING 1 |
35 |
|
36 |
// only when USE_DISK_STREAMING is set to 1: |
37 |
// just for testing the disk stream capability; with this option set |
38 |
// gigextract will read the samples in smaller pieces, just to stress |
39 |
// gig::Sample::Read() method a bit |
40 |
#define HASHED_READS_TEST 1 |
41 |
|
42 |
#include <iostream> |
43 |
#include <cstdlib> |
44 |
#include <string.h> |
45 |
#include <string> |
46 |
#include <stdlib.h> |
47 |
#include <sys/types.h> |
48 |
#include <sys/stat.h> |
49 |
#include <errno.h> |
50 |
|
51 |
#include "gig.h" |
52 |
|
53 |
#ifdef _MSC_VER |
54 |
#define S_ISDIR(x) (S_IFDIR & (x)) |
55 |
#define S_IWUSR S_IWRITE |
56 |
#define S_IXUSR S_IEXEC |
57 |
#endif |
58 |
|
59 |
#if POSIX |
60 |
# include <dlfcn.h> |
61 |
#endif |
62 |
|
63 |
// only libsndfile is available for Windows, so we use that for writing the sound files |
64 |
#ifdef WIN32 |
65 |
# define HAVE_SNDFILE 1 |
66 |
#endif // WIN32 |
67 |
|
68 |
// abort compilation here if neither libsndfile nor libaudiofile are available |
69 |
#if !HAVE_SNDFILE && !HAVE_AUDIOFILE |
70 |
# error "Neither libsndfile nor libaudiofile seem to be available!" |
71 |
# error "(HAVE_SNDFILE and HAVE_AUDIOFILE are both false)" |
72 |
#endif |
73 |
|
74 |
// we prefer libsndfile before libaudiofile |
75 |
#if HAVE_SNDFILE |
76 |
# include <sndfile.h> |
77 |
#else |
78 |
# include <audiofile.h> |
79 |
#endif // HAVE_SNDFILE |
80 |
|
81 |
using namespace std; |
82 |
|
83 |
typedef map<unsigned int, bool> OrderMap; |
84 |
OrderMap* pOrderedSamples = NULL; |
85 |
|
86 |
string Revision(); |
87 |
void PrintVersion(); |
88 |
void PrintUsage(); |
89 |
void ExtractSamples(gig::File* gig, char* destdir, OrderMap* ordered); |
90 |
int writeWav(const char* filename, void* samples, long samplecount, int channels, int bitdepth, long rate); |
91 |
string ToString(int i); |
92 |
|
93 |
#if !HAVE_SNDFILE // use libaudiofile |
94 |
void* hAFlib; // handle to libaudiofile |
95 |
void openAFlib(void); |
96 |
void closeAFlib(void); |
97 |
// pointers to libaudiofile functions |
98 |
AFfilesetup(*_afNewFileSetup)(void); |
99 |
void(*_afFreeFileSetup)(AFfilesetup); |
100 |
void(*_afInitChannels)(AFfilesetup,int,int); |
101 |
void(*_afInitSampleFormat)(AFfilesetup,int,int,int); |
102 |
void(*_afInitFileFormat)(AFfilesetup,int); |
103 |
void(*_afInitRate)(AFfilesetup,int,double); |
104 |
int(*_afWriteFrames)(AFfilehandle,int,const void*,int); |
105 |
AFfilehandle(*_afOpenFile)(const char*,const char*,AFfilesetup); |
106 |
int(*_afCloseFile)(AFfilehandle file); |
107 |
#endif // !HAVE_SNDFILE |
108 |
|
109 |
int main(int argc, char *argv[]) { |
110 |
if (argc >= 2) { |
111 |
if (argv[1][0] == '-') { |
112 |
switch (argv[1][1]) { |
113 |
case 'v': |
114 |
PrintVersion(); |
115 |
return EXIT_SUCCESS; |
116 |
} |
117 |
} |
118 |
} |
119 |
if (argc < 3) { |
120 |
PrintUsage(); |
121 |
return EXIT_FAILURE; |
122 |
} |
123 |
if (argc > 3) { // extracting specific samples |
124 |
pOrderedSamples = new OrderMap; |
125 |
for (int i = 3; i < argc; i++) { |
126 |
unsigned int index = atoi(argv[i]); |
127 |
(*pOrderedSamples)[index] = true; |
128 |
} |
129 |
} |
130 |
FILE* hFile = fopen(argv[1], "r"); |
131 |
if (!hFile) { |
132 |
cout << "Invalid input file argument!" << endl; |
133 |
return EXIT_FAILURE; |
134 |
} |
135 |
fclose(hFile); |
136 |
struct stat buf; |
137 |
if (stat(argv[2], &buf) == -1) { |
138 |
cout << "Unable to open DESTDIR: "; |
139 |
switch (errno) { |
140 |
case EACCES: cout << "Permission denied." << endl; |
141 |
break; |
142 |
case ENOENT: cout << "Directory does not exist, or name is an empty string." << endl; |
143 |
break; |
144 |
case ENOMEM: cout << "Insufficient memory to complete the operation." << endl; |
145 |
break; |
146 |
case ENOTDIR: cout << "Is not a directory." << endl; |
147 |
break; |
148 |
default: cout << "Unknown error" << endl; |
149 |
} |
150 |
return EXIT_FAILURE; |
151 |
} else if (!S_ISDIR(buf.st_mode)) { |
152 |
cout << "Unable to open DESTDIR: Is not a directory." << endl; |
153 |
return EXIT_FAILURE; |
154 |
} else if (!(S_IWUSR & buf.st_mode) || !(S_IXUSR & buf.st_mode)) { |
155 |
cout << "Unable to open DESTDIR: Permission denied." << endl; |
156 |
return EXIT_FAILURE; |
157 |
} |
158 |
try { |
159 |
RIFF::File* riff = new RIFF::File(argv[1]); |
160 |
gig::File* gig = new gig::File(riff); |
161 |
cout << "Extracting samples from \"" << argv[1] << "\" to directory \"" << argv[2] << "\"." << endl << flush; |
162 |
ExtractSamples(gig, argv[2], pOrderedSamples); |
163 |
cout << "Extraction finished." << endl << flush; |
164 |
delete gig; |
165 |
delete riff; |
166 |
if (pOrderedSamples) delete pOrderedSamples; |
167 |
} |
168 |
catch (RIFF::Exception e) { |
169 |
e.PrintMessage(); |
170 |
return EXIT_FAILURE; |
171 |
} |
172 |
catch (...) { |
173 |
cout << "Unknown exception while trying to parse file." << endl; |
174 |
return EXIT_FAILURE; |
175 |
} |
176 |
|
177 |
return EXIT_SUCCESS; |
178 |
} |
179 |
|
180 |
void ExtractSamples(gig::File* gig, char* destdir, OrderMap* ordered) { |
181 |
#if !HAVE_SNDFILE // use libaudiofile |
182 |
hAFlib = NULL; |
183 |
openAFlib(); |
184 |
#endif // !HAVE_SNDFILE |
185 |
uint8_t* pWave = NULL; |
186 |
int* pIntWave = NULL; |
187 |
long BufferSize = 0; |
188 |
int samples = 0; |
189 |
gig::buffer_t decompressionBuffer; |
190 |
decompressionBuffer.Size = 0; |
191 |
cout << "Seeking for available samples..." << flush; |
192 |
gig::Sample* pSample = gig->GetFirstSample(); |
193 |
cout << "OK" << endl << flush; |
194 |
while (pSample) { |
195 |
samples++; |
196 |
if (ordered) { |
197 |
if ((*ordered)[samples] == false) { |
198 |
pSample = gig->GetNextSample(); |
199 |
continue; |
200 |
} |
201 |
} |
202 |
string name = pSample->pInfo->Name; |
203 |
string filename = destdir; |
204 |
if (filename[filename.size() - 1] != '/') filename += "/"; |
205 |
filename += ToString(samples); |
206 |
filename += "_"; |
207 |
if (name == "") { |
208 |
name = "(NO NAME)"; |
209 |
filename += "NONAME"; |
210 |
} |
211 |
else { |
212 |
filename += name; |
213 |
name.insert(0, "\""); |
214 |
name += "\""; |
215 |
} |
216 |
filename += ".wav"; |
217 |
if (pSample->Compressed) cout << "Decompressing "; |
218 |
else cout << "Extracting "; |
219 |
cout << "Sample " << samples << ") " << name << " (" << pSample->BitDepth <<"Bits, " << pSample->SamplesPerSecond << "Hz, " << pSample->Channels << " Channels, " << pSample->SamplesTotal << " Samples)..." << flush; |
220 |
|
221 |
|
222 |
#if USE_DISK_STREAMING |
223 |
long neededsize = pSample->BitDepth == 24 ? |
224 |
pSample->SamplesTotal * pSample->Channels * sizeof(int) : |
225 |
pSample->SamplesTotal * pSample->FrameSize; |
226 |
if (BufferSize < neededsize) { |
227 |
if (pWave) delete[] pWave; |
228 |
pWave = new uint8_t[neededsize]; |
229 |
BufferSize = neededsize; |
230 |
} |
231 |
pIntWave = (int*)pWave; |
232 |
# if HASHED_READS_TEST |
233 |
unsigned long readinthisrun = 0, |
234 |
samplepiecesize = 2000; |
235 |
uint8_t* pSamplePiece = pWave; |
236 |
do { // we read the sample in small pieces and increment the size with each run just to test streaming capability |
237 |
readinthisrun = pSample->Read(pSamplePiece, ++samplepiecesize); |
238 |
pSamplePiece += readinthisrun * pSample->FrameSize; |
239 |
} while (readinthisrun == samplepiecesize); |
240 |
|
241 |
# else // read in one piece |
242 |
if (pSample->Compressed) { |
243 |
if (decompressionBufferSize < pSample->SamplesTotal) { |
244 |
gig::Sample::DestroyDecompressionBuffer(decompressionBuffer); |
245 |
decompressionBuffer = gig::Sample::CreateDecompressionBuffer(pSample->SamplesTotal); |
246 |
decompressionBufferSize = pSample->SamplesTotal; |
247 |
} |
248 |
pSample->Read(pWave, pSample->SamplesTotal, &decompressionBuffer); |
249 |
} else { |
250 |
pSample->Read(pWave, pSample->SamplesTotal); |
251 |
} |
252 |
# endif // HASHED_READS_TEST |
253 |
#else // no disk streaming |
254 |
if (pSample->Compressed) { |
255 |
cout << "Sorry, sample is compressed and Sample::LoadSampleData() only decompresses the beginning of the sample - Solution: set USE_DISK_STREAMING in gigextract.cpp (line 32) to 1 and recompile!" << endl; |
256 |
} else { |
257 |
gig::buffer_t buffer = pSample->LoadSampleData(); // load wave into RAM |
258 |
pWave = static_cast<uint8_t*>(buffer.pStart); |
259 |
if (pSample->BitDepth == 24) { |
260 |
long neededsize = pSample->SamplesTotal * pSample->Channels; |
261 |
if (BufferSize < neededsize) { |
262 |
if (pIntWave) delete[] pIntWave; |
263 |
pIntWave = new int[neededsize]; |
264 |
BufferSize = neededsize; |
265 |
} |
266 |
} |
267 |
} |
268 |
#endif // USE_DISK_STREAMING |
269 |
if (pWave) { |
270 |
|
271 |
// Both libsndfile and libaudiofile uses int for 24 bit |
272 |
// samples. libgig however returns 3 bytes per sample, so |
273 |
// we have to convert the wave data before writing. |
274 |
if (pSample->BitDepth == 24) { |
275 |
int n = pSample->SamplesTotal * pSample->Channels; |
276 |
for (int i = n - 1 ; i >= 0 ; i--) { |
277 |
#if HAVE_SNDFILE |
278 |
pIntWave[i] = pWave[i * 3] << 8 | pWave[i * 3 + 1] << 16 | pWave[i * 3 + 2] << 24; |
279 |
#else |
280 |
pIntWave[i] = pWave[i * 3] | pWave[i * 3 + 1] << 8 | pWave[i * 3 + 2] << 16; |
281 |
#endif |
282 |
} |
283 |
} |
284 |
|
285 |
int res = writeWav(filename.c_str(), |
286 |
pSample->BitDepth == 24 ? static_cast<void*>(pIntWave) : pWave, |
287 |
pSample->SamplesTotal, |
288 |
pSample->Channels, |
289 |
pSample->BitDepth, |
290 |
pSample->SamplesPerSecond); |
291 |
if (res < 0) cout << "Couldn't write sample data." << endl; |
292 |
else cout << "ok" << endl; |
293 |
pSample->ReleaseSampleData(); // free wave from RAM |
294 |
} |
295 |
else cout << "Failed to load sample data." << endl; |
296 |
|
297 |
pSample = gig->GetNextSample(); |
298 |
} |
299 |
gig::Sample::DestroyDecompressionBuffer(decompressionBuffer); |
300 |
#if USE_DISK_STREAMING |
301 |
if (pWave) delete[] pWave; |
302 |
#else |
303 |
if (pIntWave) delete[] pIntWave; |
304 |
#endif |
305 |
#if !HAVE_SNDFILE // use libaudiofile |
306 |
closeAFlib(); |
307 |
#endif // !HAVE_SNDFILE |
308 |
} |
309 |
|
310 |
int writeWav(const char* filename, void* samples, long samplecount, int channels, int bitdepth, long rate) { |
311 |
#if HAVE_SNDFILE |
312 |
SNDFILE* hfile; |
313 |
SF_INFO sfinfo; |
314 |
int format = SF_FORMAT_WAV; |
315 |
switch (bitdepth) { |
316 |
case 8: |
317 |
format |= SF_FORMAT_PCM_S8; |
318 |
break; |
319 |
case 16: |
320 |
format |= SF_FORMAT_PCM_16; |
321 |
break; |
322 |
case 24: |
323 |
format |= SF_FORMAT_PCM_24; |
324 |
break; |
325 |
case 32: |
326 |
format |= SF_FORMAT_PCM_32; |
327 |
break; |
328 |
default: |
329 |
cerr << "Error: Bithdepth " << ToString(bitdepth) << " not supported by libsndfile, ignoring sample!\n" << flush; |
330 |
return -1; |
331 |
} |
332 |
memset(&sfinfo, 0, sizeof (sfinfo)); |
333 |
sfinfo.samplerate = rate; |
334 |
sfinfo.frames = samplecount; |
335 |
sfinfo.channels = channels; |
336 |
sfinfo.format = format; |
337 |
if (!(hfile = sf_open(filename, SFM_WRITE, &sfinfo))) { |
338 |
cerr << "Error: Unable to open output file \'" << filename << "\'.\n" << flush; |
339 |
return -1; |
340 |
} |
341 |
sf_count_t res = bitdepth == 24 ? |
342 |
sf_write_int(hfile, static_cast<int*>(samples), channels * samplecount) : |
343 |
sf_write_short(hfile, static_cast<short*>(samples), channels * samplecount); |
344 |
if (res != channels * samplecount) { |
345 |
cerr << sf_strerror(hfile) << endl << flush; |
346 |
sf_close(hfile); |
347 |
return -1; |
348 |
} |
349 |
sf_close(hfile); |
350 |
#else // use libaudiofile |
351 |
AFfilesetup setup = _afNewFileSetup(); |
352 |
_afInitFileFormat(setup, AF_FILE_WAVE); |
353 |
_afInitChannels(setup, AF_DEFAULT_TRACK, channels); |
354 |
_afInitSampleFormat(setup, AF_DEFAULT_TRACK, AF_SAMPFMT_TWOSCOMP, bitdepth); |
355 |
_afInitRate(setup, AF_DEFAULT_TRACK, rate); |
356 |
if (setup == AF_NULL_FILESETUP) return -1; |
357 |
AFfilehandle hFile = _afOpenFile(filename, "w", setup); |
358 |
if (hFile == AF_NULL_FILEHANDLE) return -1; |
359 |
if (_afWriteFrames(hFile, AF_DEFAULT_TRACK, samples, samplecount) < 0) return -1; |
360 |
_afCloseFile(hFile); |
361 |
_afFreeFileSetup(setup); |
362 |
#endif // HAVE_SNDFILE |
363 |
|
364 |
return 0; // success |
365 |
} |
366 |
|
367 |
#if !HAVE_SNDFILE // use libaudiofile |
368 |
void openAFlib() { |
369 |
hAFlib = dlopen("libaudiofile.so", RTLD_NOW); |
370 |
if (!hAFlib) { |
371 |
cout << "Unable to load library libaudiofile.so: " << dlerror() << endl; |
372 |
return; |
373 |
} |
374 |
_afNewFileSetup = (AFfilesetup(*)(void)) dlsym(hAFlib, "afNewFileSetup"); |
375 |
_afFreeFileSetup = (void(*)(AFfilesetup)) dlsym(hAFlib, "afFreeFileSetup"); |
376 |
_afInitChannels = (void(*)(AFfilesetup,int,int)) dlsym(hAFlib, "afInitChannels"); |
377 |
_afInitSampleFormat = (void(*)(AFfilesetup,int,int,int)) dlsym(hAFlib, "afInitSampleFormat"); |
378 |
_afInitFileFormat = (void(*)(AFfilesetup,int)) dlsym(hAFlib, "afInitFileFormat"); |
379 |
_afInitRate = (void(*)(AFfilesetup,int,double)) dlsym(hAFlib, "afInitRate"); |
380 |
_afWriteFrames = (int(*)(AFfilehandle,int,const void*,int)) dlsym(hAFlib, "afWriteFrames"); |
381 |
_afOpenFile = (AFfilehandle(*)(const char*,const char*,AFfilesetup)) dlsym(hAFlib, "afOpenFile"); |
382 |
_afCloseFile = (int(*)(AFfilehandle file)) dlsym(hAFlib, "afCloseFile"); |
383 |
if (dlerror()) cout << "Failed to load function from libaudiofile.so: " << dlerror() << endl; |
384 |
} |
385 |
|
386 |
void closeAFlib() { |
387 |
if (hAFlib) dlclose(hAFlib); |
388 |
} |
389 |
#endif // !HAVE_SNDFILE |
390 |
|
391 |
string Revision() { |
392 |
string s = "$Revision: 1.11 $"; |
393 |
return s.substr(11, s.size() - 13); // cut dollar signs, spaces and CVS macro keyword |
394 |
} |
395 |
|
396 |
void PrintVersion() { |
397 |
cout << "gigextract revision " << Revision() << endl; |
398 |
cout << "using " << gig::libraryName() << " " << gig::libraryVersion(); |
399 |
#if HAVE_SNDFILE |
400 |
char versionBuffer[128]; |
401 |
sf_command(NULL, SFC_GET_LIB_VERSION, versionBuffer, 128); |
402 |
cout << ", " << versionBuffer; |
403 |
#else // use libaudiofile |
404 |
cout << "\nbuilt against libaudiofile " |
405 |
<< LIBAUDIOFILE_MAJOR_VERSION << "." << LIBAUDIOFILE_MINOR_VERSION; |
406 |
# ifdef LIBAUDIOFILE_MICRO_VERSION |
407 |
cout << "." << LIBAUDIOFILE_MICRO_VERSION; |
408 |
# endif // LIBAUDIOFILE_MICRO_VERSION |
409 |
#endif // HAVE_SNDFILE |
410 |
cout << endl; |
411 |
} |
412 |
|
413 |
void PrintUsage() { |
414 |
cout << "gigextract - extracts samples from a Gigasampler file." << endl; |
415 |
cout << endl; |
416 |
cout << "Usage: gigextract [-v] GIGFILE DESTDIR [SAMPLENR] [ [SAMPLENR] ...]" << endl; |
417 |
cout << endl; |
418 |
cout << " GIGFILE Input Gigasampler (.gig) file." << endl; |
419 |
cout << endl; |
420 |
cout << " DESTDIR Destination directory where all .wav files will be written to." << endl; |
421 |
cout << endl; |
422 |
cout << " SAMPLENR Index (/indices) of Sample(s) which should be extracted." << endl; |
423 |
cout << " If no sample indices are given, all samples will be extracted" << endl; |
424 |
cout << " (use gigdump to look for available samples)." << endl; |
425 |
cout << endl; |
426 |
cout << " -v Print version and exit." << endl; |
427 |
cout << endl; |
428 |
} |
429 |
|
430 |
string ToString(int i) { |
431 |
static char strbuf[1024]; |
432 |
sprintf(strbuf,"%d",i); |
433 |
string s = strbuf; |
434 |
return s; |
435 |
} |