streamtools/output-drv-rawaudio.cpp

   1 #include "output-drv.hpp"
   2 #include <cstdio>
   3 #include <stdexcept>
   4 #include <string>
   5
   6 namespace
   7 {
   8         class output_driver_rawaudio : public output_driver
   9         {
  10         public:
  11                 output_driver_rawaudio(const std::string& filename)
  12                 {
  13                         if(filename != "-")
  14                                 out = fopen(filename.c_str(), "wb");
  15                         else
  16                                 out = stdout;
  17                         if(!out)
  18                                 throw std::runtime_error("Unable to open output file");
  19                         set_audio_callback<output_driver_rawaudio>(*this, &output_driver_rawaudio::audio_callback);
  20                 }
  21
  22                 ~output_driver_rawaudio()
  23                 {
  24                         fclose(out);
  25                 }
  26
  27                 void ready()
  28                 {
  29                 }
  30
  31                 void audio_callback(short left, short right)
  32                 {
  33                         uint8_t rawdata[4];
  34                         rawdata[1] = ((unsigned short)left >> 8) & 0xFF;
  35                         rawdata[0] = ((unsigned short)left) & 0xFF;
  36                         rawdata[3] = ((unsigned short)right >> 8) & 0xFF;
  37                         rawdata[2] = ((unsigned short)right) & 0xFF;
  38                         if(fwrite(rawdata, 1, 4, out) < 4)
  39                                 throw std::runtime_error("Error writing sample to file");
  40                 }
  41         private:
  42                 FILE* out;
  43         };
  44
  45         class output_driver_rawaudio_factory : output_driver_factory
  46         {
  47         public:
  48                 output_driver_rawaudio_factory()
  49                         : output_driver_factory("rawaudio")
  50                 {
  51                 }
  52
  53                 output_driver& make(const std::string& type, const std::string& name, const std::string& parameters)
  54                 {
  55                         if(parameters != "")
  56                                 throw std::runtime_error("rawaudio output does not take parameters");
  57                         return *new output_driver_rawaudio(name);
  58                 }
  59         } factory;
  60 }