1 /*****************************************************************************
5 ** Purpose: Converts a channel of WinDAQ file to .WAV format
6 ** Programmer: Kevin Rosenberg <kevin@rosenberg.net>
7 ** Date Started: Jan 2003
9 ** Copyright (c) 2003 Kevin Rosenberg
13 ** This program is free software; you can redistribute it and/or modify
14 ** it under the terms of the GNU General Public License (version 2) as
15 ** published by the Free Software Foundation.
17 ** This program is distributed in the hope that it will be useful,
18 ** but WITHOUT ANY WARRANTY; without even the implied warranty of
19 ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
20 ** GNU General Public License for more details.
22 ** You should have received a copy of the GNU General Public License
23 ** along with this program; if not, write to the Free Software
24 ** Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
25 ******************************************************************************/
29 const char* g_szIdStr = "$Id$";
32 bool g_verbose = false;
34 bool g_ignore_zero = false;
35 bool g_dont_demean = false;
44 #define O_BINARY _O_BINARY
45 #define O_RDONLY _O_RDONLY
46 #define O_WRONLY _O_WRONLY
47 #define O_RDWR _O_RDRW
48 #define O_TRUNC _O_TRUNC
49 #define O_CREAT _O_CREAT
50 const int g_fileMode = _S_IWRITE | _S_IREAD;
51 struct fpos_t std::_Fpz = {0,0};
53 const int g_fileMode = S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH;
54 // Define as NULL for non-Windows platforms
62 error_msg (const char *msg)
64 std::cerr << msg << "\n";
68 info_msg (const char* msg)
70 std::cout << msg << "\n";
74 info_msg_sans_newline (const char* msg)
80 fileBasename (const char* filename)
82 const char* pslash = strrchr (filename, '/');
83 const char* pbackslash = strrchr (filename, '\\');
84 const char* p = filename;
85 if (pbackslash && (! pslash || pbackslash >= pslash))
87 else if (pslash && (! pbackslash || pslash >= pbackslash))
94 str_rm_tail (char *str, const char* const charlist)
98 for (i = strlen(str) - 1; i >= 0; i--)
99 if (strchr (charlist, str[i]) != NULL)
102 break; /* found non-specified char, all done */
108 str_wrm_tail (char *str)
110 return (str_rm_tail(str, "\b\t\n\r"));
115 usage (const char* progname)
117 std::cout << "usage: " << fileBasename (progname) << " [OPTIONS] <wdq-file> <channel-number> <wav-file>\n";
118 std::cout << "OPTIONS\n";
119 std::cout << " -p Play channel through audio system\n";
120 std::cout << " -q Supress all messages\n";
121 std::cout << " -z Scale output without preserving zero point\n";
122 std::cout << " -m Do not demean the data (don't subtract the mean value from each sample)\n";
123 std::cout << " -v Verbose mode\n";
124 std::cout << " -d Debug mode\n";
125 std::cout << " -r Print program version and exit\n";
126 std::cout << " -h Print this help message\n";
131 main (int argc, char *argv[])
136 const char* progname = argv[0];
138 while ((c = getopt (argc, argv, "rqvzmdph")) != -1) {
141 std::cout << "Version " << g_szIdStr << std::endl;
148 g_dont_demean = true;
151 g_ignore_zero = true;
172 if (g_verbose || g_debug)
173 std::cout << "Version " << g_szIdStr << std::endl;
178 std::cerr << "Too many parameters\n";
183 char wdq_fname[MAX_INPUT_STR];
185 strncpy (wdq_fname, argv [0], MAX_INPUT_STR);
187 std::cout << "Enter input WinDAQ filename: ";
188 std::cin.getline (wdq_fname, MAX_INPUT_STR);
191 char channel_buf [MAX_INPUT_STR];
193 strncpy (channel_buf, argv[1], MAX_INPUT_STR);
195 std::cout << "Enter channel number: ";
196 std::cin.getline (channel_buf, MAX_INPUT_STR);
199 char *channel_endptr;
200 int channel = static_cast<int>(strtol (channel_buf, &channel_endptr, 10));
201 if (*channel_endptr != 0) {
202 std::ostringstream os;
203 os << "Error: Channel " << channel_buf << " is not an integer";
204 error_msg (os.str().c_str());
209 char wav_fname[MAX_INPUT_STR];
211 strncpy (wav_fname, argv[2], MAX_INPUT_STR);
213 std::cout << "Enter output wav filename: ";
214 std::cin.getline (wav_fname, MAX_INPUT_STR);
217 if (! wdq2wav (wdq_fname, channel, wav_fname, play))
224 wdq2wav (const char* wdq_fname, const int channel, const char *wav_fname, bool play)
226 WindaqFile wdq (wdq_fname);
228 if (! wdq.ReadHeader()) {
229 if (wdq.m_error.size()) {
230 std::ostringstream os;
231 os << "Error reading file " << wdq_fname << ": " << wdq.m_error.c_str();
232 error_msg (os.str().c_str());
234 std::ostringstream os;
235 os << "Error reading file " << wdq_fname;
236 error_msg (os.str().c_str());
240 if (! g_quiet || g_verbose || g_debug) {
241 std::ostringstream os1;
242 os1 << "File: " << wdq_fname;
243 info_msg (os1.str().c_str());
244 std::ostringstream os;
245 os << "Legacy Format: ";
246 if (wdq.m_bLegacy_format)
250 info_msg(os.str().c_str());
251 std::ostringstream os2;
252 time_t time = wdq.m_time_acq_start;
253 struct tm* tm = gmtime (&time);
254 os2 << " Time File Creation: " << asctime(tm);
255 info_msg_sans_newline (os2.str().c_str());
256 std::ostringstream os3;
257 time = wdq.m_time_acq_stop;
259 os3 << " Time File Written: " << asctime(tm);
260 info_msg_sans_newline (os3.str().c_str());
261 std::ostringstream os4;
262 os4 << " Samples: " << wdq.m_nSamples <<
263 ", Channels: " << wdq.m_nChannels <<
264 ", Sample Rate: " << wdq.m_sample_rate;
265 info_msg (os4.str().c_str());
268 WindaqChannel wdq_channel (wdq, channel);
269 if (! wdq_channel.m_valid) {
270 error_msg ("Error reading data from channel");
274 if (! g_quiet || g_verbose || g_debug) {
275 std::ostringstream os1;
276 os1 << "Channel " << channel;
277 info_msg (os1.str().c_str());
278 std::ostringstream os2;
279 os2 << " Units: " << wdq_channel.m_units.c_str();
280 info_msg (os2.str().c_str());
281 std::ostringstream os3;
282 os3 << " Raw minimum: " << wdq_channel.m_min_raw_data <<
283 ", maximum: " << wdq_channel.m_max_raw_data;
284 info_msg (os3.str().c_str());
288 std::ostringstream os4;
289 os4 << " Scaled minimum: " << wdq_channel.m_min_scaled_data <<
290 ", maximum: " << wdq_channel.m_max_scaled_data;
291 info_msg (os4.str().c_str());
292 std::ostringstream os5;
293 os5 << " Slope " << wdq_channel.m_slope <<
294 ", Intercept " << wdq_channel.m_intercept;
295 info_msg (os5.str().c_str());
298 WavFile wav (wdq_channel, wav_fname);
301 error_msg ("Error extracting wav from channel");
305 if (! wav.WriteFile ()) {
306 error_msg ("Error writing file");
317 WindaqFile::WindaqFile (const char* fname)
318 : m_valid(false), m_fd(0), m_nChannels(0), m_nSamples(0), m_sample_rate(0),
323 WindaqFile::~WindaqFile ()
329 bool read_int2 (int fd, unsigned short int& n)
333 if (read (fd, &tmp1, 1) != 1)
336 if (read (fd, &tmp1, 1) != 1)
339 n = tmp2 + (tmp1 * 256);
343 bool read_int4 (int fd, unsigned int& n)
346 unsigned short int tmp2;
347 if (! read_int2 (fd, tmp2))
350 if (! read_int2 (fd, tmp2))
353 n = tmp4 + (tmp2 * 65536);
357 bool read_float8 (int fd, double& f)
359 unsigned char buf[8];
360 if (read (fd, &buf, 8) != 8)
365 c = buf[0]; buf[0] = buf[7]; buf[7] = c;
366 c = buf[1]; buf[1] = buf[6]; buf[6] = c;
367 c = buf[2]; buf[2] = buf[5]; buf[5] = c;
368 c = buf[3]; buf[3] = buf[4]; buf[4] = c;
371 f = *(reinterpret_cast<double*>(buf));
377 WindaqFile::ReadHeader ()
379 unsigned short int tmp2;
382 if ((m_fd = open (m_strFile.c_str(), O_RDONLY | O_BINARY)) < 0) {
383 m_error = "Unable to open file";
387 lseek (0, 0, SEEK_SET);
388 if (! read_int2 (m_fd, tmp2))
391 m_sr_denom = (tmp2 & 0x7fff) >> 5;
392 m_sr_numer = (tmp2 & 0x8000) << 1;
393 short unsigned int byte1 = (tmp2 & 0xFF00) >> 8;
394 short unsigned int byte2 = tmp2 & 0xFF;
395 if (byte1 == 0 || byte1 == 1) {
396 m_bLegacy_format = false;
398 m_bLegacy_format = true;
400 if (! read_int2 (m_fd, tmp2))
404 if (! read_int2 (m_fd, tmp2))
407 m_channel_offset = tmp2 & 0xFF;
408 m_nBytes_channel_header = tmp2 >> 8;
410 if (! read_int2 (m_fd, m_nHeader_bytes))
413 m_nMaxChannels = (m_nHeader_bytes - 112) / 36;
414 if (m_nMaxChannels >= 144)
415 m_nChannels = byte2 & 0xFF;
417 m_nChannels = byte2 & 0x1F;
419 if (! read_int4 (m_fd, m_nData_bytes))
422 m_nSamples = (m_nData_bytes / m_nChannels) / 2;
424 lseek (m_fd, 28, SEEK_SET);
425 if (! read_float8 (m_fd, m_time_between_channel_samples))
428 if (m_bLegacy_format)
429 m_sample_rate = (double) m_sr_numer / (double) (m_sr_denom * m_nChannels);
431 m_sample_rate = (double) m_nChannels / m_time_between_channel_samples;
433 lseek (m_fd, 36, SEEK_SET);
434 if (! read_int4 (m_fd, m_time_acq_start))
437 if (! read_int4 (m_fd, m_time_acq_stop))
440 // Verify Windaq signature
441 lseek (m_fd, m_nHeader_bytes - 2, SEEK_SET);
442 if (! read_int2 (m_fd, tmp2))
445 if (tmp2 != 0x8001) {
446 std::ostringstream os;
447 m_error = "File is not a valid WinDAQ file";
455 WindaqChannel::WindaqChannel (WindaqFile& wdq, const int channel)
456 : r_wdq(wdq), m_data(0), m_slope(0), m_intercept (0), m_channel(channel),
460 if (channel >= 1 && channel <= wdq.m_nChannels) {
461 if (read_channel_data())
464 std::ostringstream os;
465 os << "Channel " << channel << " is invalid, valid range 1-" <<
467 error_msg (os.str().c_str());
472 WindaqChannel::~WindaqChannel ()
479 WindaqChannel::read_channel_data ()
483 m_data = new signed short int [r_wdq.m_nSamples * 2];
485 lseek (fd, r_wdq.m_channel_offset + 8 +
486 (m_channel - 1) * r_wdq.m_nBytes_channel_header,
488 if (! read_float8 (fd, m_slope))
491 if (! read_float8 (fd, m_intercept))
496 if (read (fd, units, 6) != 6) {
497 error_msg ("Error reading file");
502 long int row_bytes = 2 * r_wdq.m_nChannels;
504 signed short int *sample_row = new signed short int [row_bytes];
506 signed short int* psample = &sample_row[m_channel - 1];
508 lseek (fd, r_wdq.m_nHeader_bytes, SEEK_SET);
510 signed short int data_max = 0, data_min = 0;
511 double total_data = 0;
512 for (i = 0; i < r_wdq.m_nSamples; i++) {
513 if (read (fd, sample_row, row_bytes) != row_bytes) {
514 std::ostringstream os;
515 os << "Error reading file at " << i;
516 error_msg (os.str().c_str());
520 signed short int v = *psample;
523 unsigned char* p = reinterpret_cast<unsigned char*>(&v);
524 unsigned char c = p[0]; p[0] = p[1]; p[1] = c;
527 signed short int value = v >> 2;
535 if (value > data_max)
537 else if (value < data_min)
542 m_max_raw_data = data_max;
543 m_min_raw_data = data_min;
544 m_max_scaled_data = (m_slope * data_max) + m_intercept;
545 m_min_scaled_data = (m_slope * data_min) + m_intercept;
547 if (! g_dont_demean) {
548 double dmean = total_data / static_cast<double>(r_wdq.m_nSamples);
549 int mean = nearest<int>(dmean);
550 std::cout << "Removing mean: " << (dmean * m_slope) + m_intercept <<
551 " " << m_units << std::endl;
553 for (i = 0; i < r_wdq.m_nSamples; i++)
562 WavFile::WavFile (WindaqChannel& wdq_channel, const char* fname)
563 : m_valid(false), m_data(0), m_nSamples(0), m_strFile(fname), m_fd(0)
565 if (wdq_channel.m_valid) {
566 m_nSamples = wdq_channel.r_wdq.m_nSamples;
568 m_nBitsPerSample = 16;
569 m_nBytesPerSample = 2;
570 m_rate = wdq_channel.r_wdq.m_sample_rate;
572 double data_offset = 0, data_scale = 0;
574 data_offset = -wdq_channel.m_min_scaled_data;
575 if (wdq_channel.m_max_scaled_data != wdq_channel.m_min_scaled_data)
576 data_scale = 65535. / (wdq_channel.m_max_scaled_data -
577 wdq_channel.m_min_scaled_data);
579 double max_value = fabs(wdq_channel.m_max_scaled_data);
580 if (fabs (wdq_channel.m_min_scaled_data) > max_value)
581 max_value = fabs (wdq_channel.m_min_scaled_data);
583 data_scale = 32767. / max_value;
587 std::ostringstream os;
588 os << " Wav data_scale: " << data_scale << ", data_offset: " << data_offset;
589 info_msg (os.str().c_str());
593 m_nDataBytes = m_nSamples * m_nBytesPerSample * m_nChannels;
594 m_nFileBytes = m_nHeaderBytes + m_nDataBytes;
596 unsigned int nHeaderShortInts = m_nHeaderBytes / sizeof(signed short int);
598 m_data = new signed short int [m_nSamples + nHeaderShortInts];
599 signed short int* input = wdq_channel.m_data;
600 signed short int* output = &m_data[nHeaderShortInts];
601 double slope = wdq_channel.m_slope;
602 double intercept = wdq_channel.m_intercept;
604 if (! fill_header ())
608 for (i = 0; i < m_nSamples; i++) {
609 double value = input[i];
610 value = (slope * value) + intercept;
612 value = (value + data_offset) * data_scale;
613 value += 0.5 - 32768;
615 value = value * data_scale;
618 signed short int v = static_cast<signed short int>(value);
620 unsigned char* p = reinterpret_cast<unsigned char*>(&v);
621 unsigned char c = p[0]; p[0] = p[1]; p[1] = c;
641 put_int4 (char* p, unsigned int n)
644 *(p+1) = 0xFF & (n >> 8);
645 *(p+2) = 0xFF & (n >> 16);
646 *(p+3) = 0xFF & (n >> 24);
650 put_int2 (char* p, unsigned short int n)
653 *(p+1) = 0xFF & (n >> 8);
657 WavFile::fill_header ()
659 char* pData = reinterpret_cast<char*> (m_data);
661 strncpy (pData, "RIFF", 4);
663 // Length of file after 8 byte header
664 put_int4 (pData + 4, 36 + m_nDataBytes);
666 strncpy (pData + 8, "WAVEfmt ", 8);
668 // Length of header block
669 put_int4 (pData + 16, 0x10);
672 put_int2 (pData + 20, 1);
674 /* Number of channels */
675 put_int2 (pData + 22, m_nChannels);
678 put_int4 (pData + 24, static_cast<int> (m_rate + 0.5));
681 put_int4 (pData + 28, static_cast<int> (m_rate * m_nBytesPerSample + 0.5));
684 put_int2 (pData + 32, m_nBytesPerSample * m_nChannels);
687 put_int2 (pData + 34, m_nBitsPerSample);
689 strncpy (pData + 36, "data", 4);
691 put_int4 (pData + 40, m_nDataBytes);
697 WavFile::WriteFile ()
703 if ((m_fd = open (m_strFile.c_str(), O_WRONLY | O_BINARY | O_TRUNC | O_CREAT, g_fileMode)) == 0) {
704 std::ostringstream os;
705 os << "Error opening output file " << m_strFile.c_str();
706 error_msg (os.str().c_str());
710 lseek (m_fd, 0, SEEK_SET);
711 if (write (m_fd, m_data, m_nFileBytes) != m_nFileBytes) {
712 error_msg ("Error writing file");
716 if (close (m_fd) < 0)
717 error_msg ("Error closing output file");
725 #include <mmsystem.h>
727 #include <sys/ioctl.h>
728 #include <sys/soundcard.h>
735 if (PlaySound ((LPCSTR) m_data, 0, SND_MEMORY | SND_NODEFAULT))
739 if ((fd = open ("/dev/dsp",O_WRONLY)) == -1) {
740 error_msg ("Error opening /dev/dsp");
744 int format = AFMT_S16_LE;
745 if (ioctl (fd, SNDCTL_DSP_SETFMT, &format) == -1) {
746 error_msg ("Error setting DSP format");
747 close(fd); return false;
749 if (format != AFMT_S16_LE) {
750 error_msg ("DSP Format not set");
751 close(fd); return false;
754 unsigned int channels = m_nChannels;
755 if (ioctl (fd, SNDCTL_DSP_CHANNELS, &format) == -1) {
756 error_msg ("Error setting number of channels");
757 close(fd); return false;
759 if (channels != m_nChannels) {
760 error_msg ("Number of channels not set");
761 close(fd); return false;
764 unsigned int speed = static_cast<int>(m_rate + 0.5);
765 if (ioctl (fd, SNDCTL_DSP_SPEED, &speed) == -1) {
766 error_msg ("Error setting sample rate");
767 close(fd); return false;
769 if (speed != m_rate && ! g_quiet) {
770 std::ostringstream os;
771 os << "Warning: Sample rate set to " << speed << ", not " << m_rate;
772 error_msg (os.str().c_str());
775 if (write (fd, reinterpret_cast<char*>(m_data) + m_nHeaderBytes, m_nDataBytes) !=
777 error_msg ("Error writing audio samples");
778 close(fd); return false;