r3831: *** empty log message ***
[wdq2wav.git] / wdq2wav.cpp
index bc0ae3b9cc021e719308a6ad5ec7d2ade87cd68b..5c1816913d66dbff48c35107d5fae7b7ca818954 100644 (file)
@@ -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
 
 #include <wdq2wav.h>
 
-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 <wdq-file> <channel-number> <wav-file>\n");
+  std::cout << "usage: " << fileBasename (progname) << " <wdq-file> <channel-number> <wav-file>\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<int>(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;
     }