streamtools/output-drv-rawaudio.cpp

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