tests/record_raw.cpp

   1 /******************************************/
   2 /*
   3   record_raw.c
   4   by Gary P. Scavone, 2001
   5
   6   Records from default input.  Takes
   7   number of channels and sample rate
   8   as input arguments. Uses blocking calls.
   9 */
  10 /******************************************/
  11
  12 #include "RtAudio.h"
  13 #include <stdio.h>
  14
  15 /*
  16 typedef char  MY_TYPE;
  17 #define FORMAT RtAudio::RTAUDIO_SINT8
  18
  19 typedef signed short  MY_TYPE;
  20 #define FORMAT RtAudio::RTAUDIO_SINT16
  21
  22 typedef signed long  MY_TYPE;
  23 #define FORMAT RtAudio::RTAUDIO_SINT24
  24
  25 typedef signed long  MY_TYPE;
  26 #define FORMAT RtAudio::RTAUDIO_SINT32
  27
  28 typedef float  MY_TYPE;
  29 #define FORMAT RtAudio::RTAUDIO_FLOAT32
  30 */
  31
  32 typedef double  MY_TYPE;
  33 #define FORMAT RtAudio::RTAUDIO_FLOAT64
  34
  35 #define TIME   2.0
  36
  37 void usage(void) {
  38   /* Error function in case of incorrect command-line
  39      argument specifications
  40   */
  41   cout << "\nuseage: record_raw N fs\n";
  42   cout << "    where N = number of channels,\n";
  43   cout << "    and fs = the sample rate.\n\n";
  44   exit(0);
  45 }
  46
  47 int main(int argc, char *argv[])
  48 {
  49   int chans, fs, device, buffer_size, stream;
  50   long frames, counter = 0;
  51   MY_TYPE *buffer;
  52   FILE *fd;
  53   RtAudio *audio;
  54
  55   // minimal command-line checking
  56   if (argc != 3) usage();
  57
  58   chans = (int) atoi(argv[1]);
  59   fs = (int) atoi(argv[2]);
  60
  61   // Open the realtime output device
  62   buffer_size = 512;
  63   device = 0; // default device
  64   try {
  65     audio = new RtAudio(&stream, 0, 0, device, chans,
  66                         FORMAT, fs, &buffer_size, 8);
  67   }
  68   catch (RtError &) {
  69     exit(EXIT_FAILURE);
  70   }
  71
  72   fd = fopen("test.raw","wb");
  73   frames = (long) (fs * TIME);
  74
  75   try {
  76     buffer = (MY_TYPE *) audio->getStreamBuffer(stream);
  77     audio->startStream(stream);
  78   }
  79   catch (RtError &) {
  80     goto cleanup;
  81   }
  82
  83   cout << "\nRecording for " << TIME << " seconds ... writing file test.raw." << endl;
  84   while (counter < frames) {
  85
  86     try {
  87       audio->tickStream(stream);
  88     }
  89     catch (RtError &) {
  90       goto cleanup;
  91     }
  92
  93     fwrite(buffer, sizeof(MY_TYPE), chans * buffer_size, fd);
  94     counter += buffer_size;
  95   }
  96
  97   try {
  98     audio->stopStream(stream);
  99   }
 100   catch (RtError &) {
 101   }
 102
 103  cleanup:
 104   audio->closeStream(stream);
 105   delete audio;
 106   fclose(fd);
 107
 108   return 0;
 109 }