Updates for release 4.0.8, including new python binding, new teststops.cpp program...
[rtaudio-cdist.git] / tests / playsaw.cpp
index 772881c1657c63cc62fb73a2d69980f251ba99e7..51f8ad996bf33f5333bb040c149c7193d669e4df 100644 (file)
@@ -10,6 +10,7 @@
 
 #include "RtAudio.h"
 #include <iostream>
+#include <cstdlib>
 
 /*
 typedef signed long  MY_TYPE;
@@ -19,42 +20,56 @@ typedef signed long  MY_TYPE;
 typedef char  MY_TYPE;
 #define FORMAT RTAUDIO_SINT8
 #define SCALE  127.0
+*/
 
 typedef signed short  MY_TYPE;
 #define FORMAT RTAUDIO_SINT16
 #define SCALE  32767.0
 
+/*
 typedef signed long  MY_TYPE;
 #define FORMAT RTAUDIO_SINT32
 #define SCALE  2147483647.0
-*/
 
 typedef float  MY_TYPE;
 #define FORMAT RTAUDIO_FLOAT32
 #define SCALE  1.0
 
-/*
 typedef double  MY_TYPE;
 #define FORMAT RTAUDIO_FLOAT64
 #define SCALE  1.0
 */
 
+// Platform-dependent sleep routines.
+#if defined( __WINDOWS_ASIO__ ) || defined( __WINDOWS_DS__ )
+  #include <windows.h>
+  #define SLEEP( milliseconds ) Sleep( (DWORD) milliseconds ) 
+#else // Unix variants
+  #include <unistd.h>
+  #define SLEEP( milliseconds ) usleep( (unsigned long) (milliseconds * 1000.0) )
+#endif
+
 #define BASE_RATE 0.005
 #define TIME   1.0
 
 void usage( void ) {
   // Error function in case of incorrect command-line
   // argument specifications
-  std::cout << "\nuseage: playsaw N fs <device> <channelOffset>\n";
+  std::cout << "\nuseage: playsaw N fs <device> <channelOffset> <time>\n";
   std::cout << "    where N = number of channels,\n";
   std::cout << "    fs = the sample rate,\n";
   std::cout << "    device = optional device to use (default = 0),\n";
-  std::cout << "    and channelOffset = an optional channel offset on the device (default = 0).\n\n";
+  std::cout << "    channelOffset = an optional channel offset on the device (default = 0),\n";
+  std::cout << "    and time = an optional time duration in seconds (default = no limit).\n\n";
   exit( 0 );
 }
 
 unsigned int channels;
 RtAudio::StreamOptions options;
+unsigned int frameCounter = 0;
+bool checkCount = false;
+unsigned int nFrames = 0;
+const unsigned int callbackReturnValue = 1;
 
 //#define USE_INTERLEAVED
 #if defined( USE_INTERLEAVED )
@@ -79,6 +94,8 @@ int saw( void *outputBuffer, void *inputBuffer, unsigned int nBufferFrames,
     }
   }
 
+  frameCounter += nBufferFrames;
+  if ( checkCount && ( frameCounter >= nFrames ) ) return callbackReturnValue;
   return 0;
 }
 
@@ -105,6 +122,8 @@ int saw( void *outputBuffer, void *inputBuffer, unsigned int nBufferFrames,
     }
   }
 
+  frameCounter += nBufferFrames;
+  if ( checkCount && ( frameCounter >= nFrames ) ) return callbackReturnValue;
   return 0;
 }
 #endif
@@ -114,7 +133,7 @@ int main( int argc, char *argv[] )
   unsigned int bufferFrames, fs, device = 0, offset = 0;
 
   // minimal command-line checking
-  if (argc < 3 || argc > 5 ) usage();
+  if (argc < 3 || argc > 6 ) usage();
 
   RtAudio dac;
   if ( dac.getDeviceCount() < 1 ) {
@@ -128,9 +147,11 @@ int main( int argc, char *argv[] )
     device = (unsigned int) atoi( argv[3] );
   if ( argc > 4 )
     offset = (unsigned int) atoi( argv[4] );
+  if ( argc > 5 )
+    nFrames = (unsigned int) (fs * atof( argv[5] ));
+  if ( nFrames > 0 ) checkCount = true;
 
-  double *data;
-  data = (double *) calloc( channels, sizeof( double ) );
+  double *data = (double *) calloc( channels, sizeof( double ) );
 
   // Let RtAudio print messages to stderr.
   dac.showWarnings( true );
@@ -143,6 +164,7 @@ int main( int argc, char *argv[] )
   oParams.firstChannel = offset;
 
   options.flags |= RTAUDIO_HOG_DEVICE;
+  options.flags |= RTAUDIO_SCHEDULE_REALTIME;
 #if !defined( USE_INTERLEAVED )
   options.flags |= RTAUDIO_NONINTERLEAVED;
 #endif
@@ -155,17 +177,22 @@ int main( int argc, char *argv[] )
     goto cleanup;
   }
 
-  char input;
-  std::cout << "\nPlaying ... press <enter> to quit (buffer size = " << bufferFrames << ").\n";
-  std::cin.get( input );
-
-  try {
-    // Stop the stream
-    std::cout << "Stream latency = " << dac.getStreamLatency() << "\n" << std::endl;
-    dac.stopStream();
+  if ( checkCount ) {
+    while ( dac.isStreamRunning() == true ) SLEEP( 100 );
   }
-  catch ( RtError& e ) {
-    e.printMessage();
+  else {
+    char input;
+    //std::cout << "Stream latency = " << dac.getStreamLatency() << "\n" << std::endl;
+    std::cout << "\nPlaying ... press <enter> to quit (buffer size = " << bufferFrames << ").\n";
+    std::cin.get( input );
+
+    try {
+      // Stop the stream
+      dac.stopStream();
+    }
+    catch ( RtError& e ) {
+      e.printMessage();
+    }
   }
 
  cleanup: