streamtools/output-drv-faac.cpp

   1 #include "output-drv.hpp"
   2 #include <cstdio>
   3 #include <stdexcept>
   4 #include <string>
   5 #include <sstream>
   6
   7 namespace
   8 {
   9         std::string expand_options(const std::string& opts)
  10         {
  11                 bool insert = true;
  12                 std::ostringstream ret;
  13                 for(size_t i = 0; i < opts.length(); i++) {
  14                         if(insert)
  15                                 ret << " -";
  16                         insert = false;
  17                         switch(opts[i]) {
  18                         case ',':
  19                                 insert = true;
  20                                 break;
  21                         case '=':
  22                                 ret << " ";
  23                                 break;
  24                         default:
  25                                 ret << opts[i];
  26                         };
  27                 }
  28                 ret << " ";
  29                 return ret.str();
  30         }
  31
  32         class output_driver_faac : public output_driver
  33         {
  34         public:
  35                 output_driver_faac(const std::string& _filename, const std::string& _options)
  36                 {
  37                         filename = _filename;
  38                         options = _options;
  39                         set_audio_callback<output_driver_faac>(*this, &output_driver_faac::audio_callback);
  40                 }
  41
  42                 ~output_driver_faac()
  43                 {
  44                         pclose(out);
  45                 }
  46
  47                 void ready()
  48                 {
  49                         const audio_settings& a = get_audio_settings();
  50
  51                         std::stringstream commandline;
  52                         commandline << "faac -P -C 2 -R " << a.get_rate() << " ";
  53                         commandline << expand_options(options);
  54                         commandline << "-o " << filename << " -";
  55                         std::string s = commandline.str();
  56                         out = popen(s.c_str(), "w");
  57                         if(!out) {
  58                                 std::stringstream str;
  59                                 str << "Can't run faac (" << s << ")";
  60                                 throw std::runtime_error(str.str());
  61                         }
  62                 }
  63
  64                 void audio_callback(short left, short right)
  65                 {
  66                         uint8_t rawdata[4];
  67                         rawdata[1] = ((unsigned short)left >> 8) & 0xFF;
  68                         rawdata[0] = ((unsigned short)left) & 0xFF;
  69                         rawdata[3] = ((unsigned short)right >> 8) & 0xFF;
  70                         rawdata[2] = ((unsigned short)right) & 0xFF;
  71                         if(fwrite(rawdata, 1, 4, out) < 4)
  72                                 throw std::runtime_error("Error writing sample to faac");
  73                 }
  74         private:
  75                 FILE* out;
  76                 std::string filename;
  77                 std::string options;
  78         };
  79
  80         class output_driver_faac_factory : output_driver_factory
  81         {
  82         public:
  83                 output_driver_faac_factory()
  84                         : output_driver_factory("faac")
  85                 {
  86                 }
  87
  88                 output_driver& make(const std::string& type, const std::string& name, const std::string& parameters)
  89                 {
  90                         return *new output_driver_faac(name, parameters);
  91                 }
  92         } factory;
  93 }