Shared memory rewrite

src/input/input_flv.cpp

@@ -0,0 +1,130 @@
+#include <iostream>
+#include <fstream>
+#include <cstring>
+#include <cerrno>
+#include <cstdlib>
+#include <cstdio>
+#include <string>
+#include <mist/stream.h>
+#include <mist/flv_tag.h>
+#include <mist/defines.h>
+
+#include "input_flv.h"
+
+namespace Mist {
+  inputFLV::inputFLV(Util::Config * cfg) : Input(cfg) {
+    capa["decs"] = "Enables FLV Input";
+    capa["codecs"][0u][0u].append("H264");
+    capa["codecs"][0u][0u].append("H263");
+    capa["codecs"][0u][0u].append("VP6");
+    capa["codecs"][0u][1u].append("AAC");
+    capa["codecs"][0u][1u].append("MP3");
+  }
+
+  bool inputFLV::setup() {
+    if (config->getString("input") == "-") {
+      std::cerr << "Input from stream not yet supported" << std::endl;
+      return false;
+    }
+    if (config->getString("output") != "-") {
+      std::cerr << "Output to non-stdout not yet supported" << std::endl;
+    }
+    
+    //open File
+    inFile = fopen(config->getString("input").c_str(), "r");
+    if (!inFile) {
+      return false;
+    }
+    return true;
+  }
+
+  bool inputFLV::readHeader() {
+    JSON::Value lastPack;
+    if (!inFile) {
+      return false;
+    }
+    //See whether a separate header file exists.
+    DTSC::File tmp(config->getString("input") + ".dtsh");
+    if (tmp){
+      myMeta = tmp.getMeta();
+      return true;
+    }
+    //Create header file from FLV data
+    fseek(inFile, 13, SEEK_SET);
+    FLV::Tag tmpTag;
+    long long int lastBytePos = 13;
+    while (!feof(inFile) && !FLV::Parse_Error){
+      if (tmpTag.FileLoader(inFile)){
+        lastPack.null();
+        lastPack = tmpTag.toJSON(myMeta);
+        lastPack["bpos"] = lastBytePos;
+        myMeta.update(lastPack);
+        lastBytePos = ftell(inFile);
+      }
+    }
+    if (FLV::Parse_Error){
+      std::cerr << FLV::Error_Str << std::endl;
+      return false;
+    }
+    std::ofstream oFile(std::string(config->getString("input") + ".dtsh").c_str());
+    oFile << myMeta.toJSON().toNetPacked();
+    oFile.close();
+    return true;
+  }
+  
+  void inputFLV::getNext(bool smart) {
+    static JSON::Value thisPack;
+    thisPack.null();
+    long long int lastBytePos = ftell(inFile);
+    FLV::Tag tmpTag;
+    while (!feof(inFile) && !FLV::Parse_Error){
+      if (tmpTag.FileLoader(inFile)){
+        thisPack = tmpTag.toJSON(myMeta);
+        thisPack["bpos"] = lastBytePos;
+        if ( !selectedTracks.count(thisPack["trackid"].asInt())){
+          getNext();
+        }
+        break;
+      }
+    }
+    if (FLV::Parse_Error){
+      std::cerr << FLV::Error_Str << std::endl;
+      thisPack.null();
+      lastPack.null();
+      return;
+    }
+    std::string tmpStr = thisPack.toNetPacked();
+    lastPack.reInit(tmpStr.data(), tmpStr.size());
+  }
+
+  void inputFLV::seek(int seekTime) {
+    //We will seek to the corresponding keyframe of the video track if selected, otherwise audio keyframe.
+    //Flv files are never multi-track, so track 1 is video, track 2 is audio.
+    int trackSeek = (selectedTracks.count(1) ? 1 : 2);
+    size_t seekPos = myMeta.tracks[trackSeek].keys[0].getBpos();
+    for (int i = 0; i < myMeta.tracks[trackSeek].keys.size(); i++){
+      if (myMeta.tracks[trackSeek].keys[i].getTime() > seekTime){
+        DEBUG_MSG(DLVL_WARN, "Seeking to keyframe %d on track %d, timestamp %ld, bytepos %lu", i, trackSeek, myMeta.tracks[trackSeek].keys[i].getTime(), seekPos);
+        break;
+      }
+      seekPos = myMeta.tracks[trackSeek].keys[i].getBpos();
+    }
+    fseek(inFile, seekPos, SEEK_SET);
+  }
+
+  void inputFLV::trackSelect(std::string trackSpec) {
+    selectedTracks.clear();
+    long long int index;
+    while (trackSpec != "") {
+      index = trackSpec.find(' ');
+      selectedTracks.insert(atoi(trackSpec.substr(0, index).c_str()));
+      DEBUG_MSG(DLVL_WARN, "Added track %d, index = %lld, (index == npos) = %d", atoi(trackSpec.substr(0, index).c_str()), index, index == std::string::npos);
+      if (index != std::string::npos) {
+        trackSpec.erase(0, index + 1);
+      } else {
+        trackSpec = "";
+      }
+    }
+  }
+}
+