git.sesse.net Git - casparcg/blob - modules/ffmpeg/producer/format/flv.cpp

   1 #include "../../stdafx.h"\r
   2 \r
   3 #include "flv.h"\r
   4 \r
   5 #include <common/exception/exceptions.h>\r
   6 \r
   7 #include <boost/filesystem.hpp>\r
   8 \r
   9 #include <iostream>\r
  10 \r
  11 #include <unordered_map>\r
  12 \r
  13 namespace caspar {\r
  14         \r
  15 double next_double(std::fstream& fileStream)\r
  16 {\r
  17         std::vector<char> bytes(8);\r
  18     fileStream.read(bytes.data(), bytes.size());\r
  19         fileStream.seekg(1, std::ios::cur);\r
  20 \r
  21         std::reverse(bytes.begin(), bytes.end());\r
  22         double* tmp = (double*)bytes.data();\r
  23 \r
  24     return *tmp;\r
  25 } \r
  26 \r
  27 bool next_bool(std::fstream& fileStream)\r
  28 {\r
  29         std::vector<char> bytes(1);\r
  30     fileStream.read(bytes.data(), bytes.size());\r
  31         fileStream.seekg(1, std::ios::cur);\r
  32     return bytes[0] != 0;\r
  33 }\r
  34 \r
  35 std::map<std::string, std::string> read_flv_meta_info(const std::string& filename)\r
  36 {\r
  37         std::map<std::string, std::string>  values;\r
  38 \r
  39         if(boost::filesystem2::path(filename).extension() != ".flv")\r
  40                 return values;\r
  41         \r
  42         try\r
  43         {\r
  44                 if(!boost::filesystem2::exists(filename))\r
  45                         BOOST_THROW_EXCEPTION(caspar_exception());\r
  46         \r
  47                 std::fstream fileStream = std::fstream(filename, std::fstream::in);\r
  48                 fileStream.seekg(27, std::ios::beg);\r
  49         \r
  50                 std::vector<char> bytes(10);\r
  51                 fileStream.read(bytes.data(), bytes.size());\r
  52                                 \r
  53                 if(std::string(bytes.begin(), bytes.end()) == "onMetaData")\r
  54                 {\r
  55                         fileStream.seekg(6, std::ios::cur);\r
  56 \r
  57                         for(int n = 0; n < 16; ++n)\r
  58                         {\r
  59                                 char name_size = 0;\r
  60                                 fileStream.read(&name_size, 1);\r
  61 \r
  62                                 if(name_size == 0)\r
  63                                         break;\r
  64 \r
  65                                 std::vector<char> name(name_size);\r
  66                                 fileStream.read(name.data(), name.size());\r
  67                                 auto name_str = std::string(name.begin(), name.end());\r
  68 \r
  69                                 char data_type = 0;\r
  70                                 fileStream.read(&data_type, 1);\r
  71 \r
  72                                 switch(data_type)\r
  73                                 {\r
  74                                 case 0:\r
  75                                         values[name_str] = boost::lexical_cast<std::string>(next_double(fileStream));\r
  76                                         break;\r
  77                                 case 1:\r
  78                                         values[name_str] = boost::lexical_cast<std::string>(next_bool(fileStream));\r
  79                                         break;\r
  80                                 }\r
  81                         }\r
  82                 }\r
  83         }\r
  84         catch(...)\r
  85         {\r
  86                 CASPAR_LOG_CURRENT_EXCEPTION();\r
  87         }\r
  88 \r
  89     return values;\r
  90 }\r
  91 \r
  92 }