#include "output_sanitycheck.h"
#include <mist/bitfields.h>
#include <mist/checksum.h>
#include <mist/defines.h>

namespace Mist{
  OutSanityCheck::OutSanityCheck(Socket::Connection &conn) : Output(conn){
    streamName = config->getString("streamname");
    parseData = true;
    wantRequest = false;
    initialize();
    initialSeek();
    sortSet.clear();
    for (std::set<long unsigned int>::iterator subIt = selectedTracks.begin();
         subIt != selectedTracks.end(); subIt++){
      keyPart temp;
      temp.trackID = *subIt;
      temp.time = myMeta.tracks[*subIt].firstms; // timeplace of frame
      temp.endTime = myMeta.tracks[*subIt].firstms + myMeta.tracks[*subIt].parts[0].getDuration();
      temp.size = myMeta.tracks[*subIt].parts[0].getSize(); // bytesize of frame (alle parts all together)
      temp.index = 0;
      sortSet.insert(temp);
    }
    realTime = 0;

    if (config->getInteger("seek")){
      uint64_t seekPoint = config->getInteger("seek");

      while (!sortSet.empty() && sortSet.begin()->time < seekPoint){
        keyPart temp;
        temp.index = sortSet.begin()->index + 1;
        temp.trackID = sortSet.begin()->trackID;
        if (temp.index < myMeta.tracks[temp.trackID].parts.size()){// only insert when there are parts left
          temp.time = sortSet.begin()->endTime;                      // timeplace of frame
          temp.endTime =
              sortSet.begin()->endTime + myMeta.tracks[temp.trackID].parts[temp.index].getDuration();
          temp.size = myMeta.tracks[temp.trackID].parts[temp.index].getSize(); // bytesize of frame
          sortSet.insert(temp);
        }
        // remove highest keyPart
        sortSet.erase(sortSet.begin());
      }
      seek(seekPoint);
    }
  }

  void OutSanityCheck::init(Util::Config *cfg){
    Output::init(cfg);
    capa["name"] = "SanityCheck";
    capa["desc"] = "Does sanity check on a stream";
    capa["codecs"][0u][0u].append("+*");
    cfg->addOption("streamname", JSON::fromString("{\"arg\":\"string\",\"short\":\"s\",\"long\":"
                                                  "\"stream\",\"help\":\"The name of the stream "
                                                  "that this connector will transmit.\"}"));
    cfg->addOption(
        "seek", JSON::fromString("{\"arg\":\"string\",\"short\":\"S\",\"long\":\"seek\",\"help\":"
                                 "\"Time in ms to check from - by default start of stream\"}"));
    cfg->addBasicConnectorOptions(capa);
    config = cfg;
  }

  void OutSanityCheck::sendNext(){
    if ((unsigned long)thisPacket.getTrackId() != sortSet.begin()->trackID ||
        thisPacket.getTime() != sortSet.begin()->time){
      while (packets.size()){
        std::cout << packets.front() << std::endl;
        packets.pop_front();
      }
      std::cout << "Input is inconsistent! Expected " << sortSet.begin()->trackID << ":"
                << sortSet.begin()->time << " but got " << thisPacket.getTrackId() << ":"
                << thisPacket.getTime() << " (part " << sortSet.begin()->index << " in "
                << myMeta.tracks[sortSet.begin()->trackID].codec << " track)" << std::endl;
      myConn.close();
      return;
    }

    // Packet is normally sent here
    packets.push_back(thisPacket.toSummary());
    while (packets.size() > 10){packets.pop_front();}

    // keep track of where we are
    if (!sortSet.empty()){
      keyPart temp;
      temp.index = sortSet.begin()->index + 1;
      temp.trackID = sortSet.begin()->trackID;
      if (temp.index < myMeta.tracks[temp.trackID].parts.size()){// only insert when there are parts left
        temp.time = sortSet.begin()->endTime;                      // timeplace of frame
        temp.endTime = sortSet.begin()->endTime + myMeta.tracks[temp.trackID].parts[temp.index].getDuration();
        temp.size = myMeta.tracks[temp.trackID].parts[temp.index].getSize(); // bytesize of frame
        sortSet.insert(temp);
      }
      // remove highest keyPart
      sortSet.erase(sortSet.begin());
    }
  }

}// namespace Mist