X-Git-Url: http://git.kpe.io/?a=blobdiff_plain;f=wdq2wav.cpp;h=5c1816913d66dbff48c35107d5fae7b7ca818954;hb=746824cc3ac28397a58be87f75a250dbfe95c53d;hp=bc0ae3b9cc021e719308a6ad5ec7d2ade87cd68b;hpb=760badbe051b46ce9e2ee82a2e941027fc57ed24;p=wdq2wav.git diff --git a/wdq2wav.cpp b/wdq2wav.cpp index bc0ae3b..5c18169 100644 --- a/wdq2wav.cpp +++ b/wdq2wav.cpp @@ -8,16 +8,7 @@ ** ** Copyright (c) 2003 Kevin Rosenberg ** -** NOTES -** Quick & Dirty hack, but efficient and good error checking -** Allocates memory equal to the size of channel (2 * number of samples) -** -** BUGS -** Need to abstract error checking so it is not so redundant -** Need command line options for verbose and debug output -** Should comment the reading and writing of files to document the file formats -** -** $Id: wdq2wav.cpp,v 1.1 2003/01/21 01:58:32 kevin Exp $ +** $Id: wdq2wav.cpp,v 1.9 2003/01/21 09:38:59 kevin Exp $ ** ** This program is free software; you can redistribute it and/or modify ** it under the terms of the GNU General Public License (version 2) as @@ -35,60 +26,195 @@ #include -static bool g_verbose = true; -static bool g_debug = true; +enum { O_VERBOSE, O_QUIET, O_DEBUG, O_HELP, O_VERSION }; + +static struct option my_options[] = +{ + {"verbose", 0, 0, O_VERBOSE}, + {"quiet", 0, 0, O_QUIET}, + {"debug", 0, 0, O_DEBUG}, + {"help", 0, 0, O_HELP}, + {"version", 0, 0, O_VERSION}, + {0, 0, 0, 0} +}; + +static const char* g_szIdStr = "$Id: wdq2wav.cpp,v 1.9 2003/01/21 09:38:59 kevin Exp $"; + +static bool g_quiet = false; +static bool g_verbose = false; +static bool g_debug = false; + +void +error_msg (const char *msg) +{ + std::cerr << msg << "\n"; +} + +void +info_msg (const char* msg) +{ + std::cout << msg << "\n"; +} void -error_msg (char *msg) +info_msg_sans_newline (const char* msg) +{ + std::cout << msg; +} + +const char* +fileBasename (const char* filename) +{ + const char* p = strrchr (filename, '/'); + return (p ? p + 1 : filename); +} + +char * +str_rm_tail (char *str, const char* const charlist) { - fprintf (stderr, "%s\n",msg); + int i; + + for (i = strlen(str) - 1; i >= 0; i--) + if (strchr (charlist, str[i]) != NULL) + str[i] = 0; + else + break; /* found non-specified char, all done */ + + return (str); } +char * +str_wrm_tail (char *str) +{ + return (str_rm_tail(str, "\b\t\n\r")); +} + + void -usage () +usage (const char* progname) { - fprintf (stderr, "usage: wdq2wav \n"); + std::cout << "usage: " << fileBasename (progname) << " \n"; + std::cout << " --quiet Supress all messages\n"; + std::cout << " --verbose Verbose mode\n"; + std::cout << " --debug Debug mode\n"; + std::cout << " --version Print version\n"; + std::cout << " --help Print this help message\n"; } -bool wdq2wav (const char* wdq_fname, const int channel, const char *wav_fname); int main (int argc, char *argv[]) { - if (argc != 4) { - usage(); - exit(1); - } - - char *wdq_fname = argv[1]; - char *channel_endptr; - int channel = (int) strtol (argv[2], &channel_endptr, 10); - char *wav_fname = argv[3]; + while (1) { + int c = getopt_long (argc, argv, "", my_options, NULL); + if (c == -1) + break; + + switch (c) { + case O_VERSION: + std::cout << "Version " << g_szIdStr << std::endl; + break; + case O_QUIET: + g_quiet = true; + break; + case O_VERBOSE: + g_verbose = true; + break; + case O_DEBUG: + g_debug = true; + break; + case O_HELP: + case '?': + usage(argv[0]); + return (0); + default: + usage(argv[0]); + return (1); + } + } + + if (optind + 3 < argc) { + std::cerr << "Too many parameters\n"; + usage (argv[0]); + return (1); + } - if (*channel_endptr != 0) { - fprintf (stderr, "Error: Channel %s is not an integer\n", argv[2]); - usage(); - return(1); - } + char wdq_fname[MAX_INPUT_STR]; + if (optind < argc) + strncpy (wdq_fname, argv [optind], MAX_INPUT_STR); + else { + std::cout << "Enter input WinDAQ filename: "; + std::cin.getline (wdq_fname, MAX_INPUT_STR); + } - if (! wdq2wav (wdq_fname, channel, wav_fname)) - return 1; + char channel_buf [MAX_INPUT_STR]; + if (optind + 1 < argc) + strncpy (channel_buf, argv[optind+1], MAX_INPUT_STR); + else { + std::cout << "Enter channel number: "; + std::cin.getline (channel_buf, MAX_INPUT_STR); + } + + char *channel_endptr; + int channel = static_cast(strtol (channel_buf, &channel_endptr, 10)); + if (*channel_endptr != 0) { + std::ostringstream os; + os << "Error: Channel " << channel_buf << " is not an integer"; + error_msg (os.str().c_str()); + usage (argv[0]); + return (1); + } - return 0; + char wav_fname[MAX_INPUT_STR]; + if (optind + 2 < argc) + strncpy (wav_fname, argv[optind+2], MAX_INPUT_STR); + else { + std::cout << "Enter output wav filename: "; + std::cin.getline (wav_fname, MAX_INPUT_STR); + } + + if (! wdq2wav (wdq_fname, channel, wav_fname)) + return 1; + + return 0; } -bool wdq2wav (const char* wdq_fname, const int channel, const char *wav_fname) +bool +wdq2wav (const char* wdq_fname, const int channel, const char *wav_fname) { WindaqFile wdq (wdq_fname); if (! wdq.ReadHeader()) { - fprintf (stderr, "Error reading header %s\n", wdq_fname); + if (wdq.m_error.size()) { + std::ostringstream os; + os << "Error reading file " << wdq_fname << ": " << wdq.m_error.c_str(); + error_msg (os.str().c_str()); + } else { + std::ostringstream os; + os << "Error reading file " << wdq_fname; + error_msg (os.str().c_str()); + } return false; } - if (g_verbose) { - printf ("Number of channels: %d\n", wdq.m_nChannels); - printf ("Number of Samples: %d\n", wdq.m_nSamples); - printf ("Sample Rate: %.1f\n", wdq.m_sample_rate); + if (! g_quiet || g_verbose || g_debug) { + std::ostringstream os1; + os1 << "File " << wdq_fname; + info_msg (os1.str().c_str()); + std::ostringstream os2; + time_t time = wdq.m_time_acq_start; + struct tm* tm = gmtime (&time); + os2 << " Time File Creation: " << asctime(tm); + info_msg_sans_newline (os2.str().c_str()); + std::ostringstream os3; + time = wdq.m_time_acq_stop; + tm = gmtime (&time); + os3 << " Time File Written: " << asctime(tm); + info_msg_sans_newline (os3.str().c_str()); + std::ostringstream os4; + os4 << " Samples: " << wdq.m_nSamples << + ", Channels: " << wdq.m_nChannels << + ", Sample Rate: " << wdq.m_sample_rate; + info_msg (os4.str().c_str()); } WindaqChannel wdq_channel (wdq, channel); @@ -97,9 +223,27 @@ bool wdq2wav (const char* wdq_fname, const int channel, const char *wav_fname) return false; } - if (g_verbose) { - printf ("Data Min: %f, Data Max: %f\n", - wdq_channel.m_min_raw_data, wdq_channel.m_max_raw_data); + if (! g_quiet || g_verbose || g_debug) { + std::ostringstream os1; + os1 << "Channel " << channel; + info_msg (os1.str().c_str()); + std::ostringstream os2; + os2 << " Units: " << wdq_channel.m_units.c_str(); + info_msg (os2.str().c_str()); + std::ostringstream os3; + os3 << " Raw minimum: " << wdq_channel.m_min_raw_data << + ", maximum: " << wdq_channel.m_max_raw_data; + info_msg (os3.str().c_str()); + } + if (g_debug) { + std::ostringstream os4; + os4 << " Scaled minimum: " << wdq_channel.m_min_scaled_data << + ", maximum: " << wdq_channel.m_max_scaled_data; + info_msg (os4.str().c_str()); + std::ostringstream os5; + os5 << " Slope " << wdq_channel.m_slope << + ", Intercept " << wdq_channel.m_intercept; + info_msg (os5.str().c_str()); } WavFile wav (wdq_channel, wav_fname); @@ -137,53 +281,60 @@ WindaqFile::ReadHeader () unsigned int tmp4; m_valid = false; - if ((m_fd = open (m_strFile.c_str(), O_RDONLY)) == 0) + if ((m_fd = open (m_strFile.c_str(), O_RDONLY)) < 0) { + m_error = "Unable to open file"; return false; + } lseek (0, 0, SEEK_SET); if (read (m_fd, &tmp2, sizeof(tmp2)) != sizeof(tmp2)) { - error_msg ("Error reading file"); return false; } m_nChannels = tmp2 & 0x1f; m_sr_denom = (tmp2 & 0x7fff) >> 5; m_sr_numer = (tmp2 & 0x8000) << 1; - if (read (m_fd, &tmp2, sizeof(tmp2)) != sizeof(tmp2)) { - error_msg ("Error reading file"); + if (read (m_fd, &tmp2, sizeof(tmp2)) != sizeof(tmp2)) return false; - } + m_sr_numer |= tmp2; m_sample_rate = (double) m_sr_numer / (double) (m_sr_denom * m_nChannels); - if (read (m_fd, &tmp2, sizeof(tmp2)) != sizeof(tmp2)) { - error_msg ("Error reading file"); + if (read (m_fd, &tmp2, sizeof(tmp2)) != sizeof(tmp2)) return false; - } + m_channel_offset = tmp2 & 0xFF; m_nBytes_channel_header = tmp2 >> 8; - if (read (m_fd, &tmp2, sizeof(tmp2)) != sizeof(tmp2)) { - error_msg ("Error reading file"); + if (read (m_fd, &tmp2, sizeof(tmp2)) != sizeof(tmp2)) return false; - } + m_nHeader_bytes = tmp2; - if (read (m_fd, &tmp4, sizeof(tmp4)) != sizeof(tmp4)) { - error_msg ("Error reading file"); + if (read (m_fd, &tmp4, sizeof(tmp4)) != sizeof(tmp4)) return false; - } + m_nData_bytes = tmp4; m_nSamples = (m_nData_bytes / m_nChannels) / 2; + lseek (m_fd, 36, SEEK_SET); + if (read (m_fd, &tmp4, sizeof(tmp4)) != sizeof(tmp4)) + return false; + + m_time_acq_start = tmp4; + if (read (m_fd, &tmp4, sizeof(tmp4)) != sizeof(tmp4)) + return false; + + m_time_acq_stop = tmp4; + // Verify Windaq signature lseek (m_fd, m_nHeader_bytes - 2, SEEK_SET); - if (read (m_fd, &tmp2, sizeof(tmp2)) != sizeof(tmp2)) { - error_msg ("Error reading file"); + if (read (m_fd, &tmp2, sizeof(tmp2)) != sizeof(tmp2)) return false; - } + if (tmp2 != 0x8001) { - fprintf (stderr, "%s is not a WinDAQ file\n", m_strFile.c_str()); + std::ostringstream os; + m_error = "File is not a valid WinDAQ file"; return false; } @@ -200,9 +351,12 @@ WindaqChannel::WindaqChannel (WindaqFile& wdq, const int channel) if (channel >= 1 && channel <= wdq.m_nChannels) { m_valid = true; read_channel_data(); - } else - fprintf (stderr, "Channel %d is invalid, valid range 1-%d\n", - channel, wdq.m_nChannels); + } else { + std::ostringstream os; + os << "Channel " << channel << " is invalid, valid range 1-" << + wdq.m_nChannels; + error_msg (os.str().c_str()); + } } } @@ -240,9 +394,6 @@ WindaqChannel::read_channel_data () } m_intercept = float8; - if (g_verbose) - printf ("Slope: %f, Intercept: %f\n", m_slope, m_intercept); - char units[7]; units[6] = 0; if (read (fd, units, 6) != 6) { @@ -250,8 +401,6 @@ WindaqChannel::read_channel_data () return false; } m_units = units; - if (g_verbose) - printf ("Units: %s\n", units); unsigned int row_bytes = 2 * r_wdq.m_nChannels; signed short int sample_row [row_bytes]; @@ -263,7 +412,9 @@ WindaqChannel::read_channel_data () signed short int data_max, data_min; for (i = 0; i < r_wdq.m_nSamples; i++) { if (read (fd, sample_row, row_bytes) != row_bytes) { - fprintf (stderr, "Error reading file at %d\n", i); + std::ostringstream os; + os << "Error reading file at " << i; + error_msg (os.str().c_str()); return false; } @@ -307,8 +458,11 @@ WavFile::WavFile (WindaqChannel& wdq_channel, const char* fname) data_scale = 65535. / (wdq_channel.m_max_scaled_data - wdq_channel.m_min_scaled_data); - if (g_debug) - printf ("data_scale: %f, data_offset: %f\n", data_scale, data_offset); + if (g_debug) { + std::ostringstream os; + os << " Wav data_scale: " << data_scale << ", data_offset: " << data_offset; + info_msg (os.str().c_str()); + } signed short int* input = wdq_channel.m_data; double slope = wdq_channel.m_slope; @@ -350,7 +504,9 @@ WavFile::WriteFile () if (m_fd == 0) if ((m_fd = open (m_strFile.c_str(), O_WRONLY | O_TRUNC | O_CREAT, S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH)) == 0) { - fprintf (stderr, "Error opening output file %s\n", m_strFile.c_str()); + std::ostringstream os; + os << "Error opening output file " << m_strFile.c_str(); + error_msg (os.str().c_str()); return false; }