/**********
 This library is free software; you can redistribute it and/or modify it under
 the terms of the GNU Lesser General Public License as published by the
 Free Software Foundation; either version 2.1 of the License, or (at your
 option) any later version. (See <http://www.gnu.org/copyleft/lesser.html>.)

 This library is distributed in the hope that it will be useful, but WITHOUT
 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for
 more details.

 You should have received a copy of the GNU Lesser General Public License
 along with this library; if not, write to the Free Software Foundation, Inc.,
 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301  USA
 **********/
// "liveMedia"
// Copyright (c) 1996-2016 Live Networks, Inc.  All rights reserved.
// A simplified version of "H264VideoStreamFramer" that takes only complete,
// discrete frames (rather than an arbitrary byte stream) as input.
// This avoids the parsing and data copying overhead of the full
// "H264VideoStreamFramer".
// Implementation
#include "include/H264or5VideoNaluFramer.hh"

#include <iostream>
#include "../IPCProgram/interface/FrameNaluParser.h"
#include "string.h"

using namespace std;
using namespace Byl;

H264or5VideoNaluFramer* H264or5VideoNaluFramer::createNew(UsageEnvironment& env, FramedSource* inputSource, int hNum)
{
    return new H264or5VideoNaluFramer(env, inputSource, hNum);
}

H264or5VideoNaluFramer::H264or5VideoNaluFramer(UsageEnvironment& env, FramedSource* inputSource, int hNum)
    : H264or5VideoStreamDiscreteFramer(hNum, env, inputSource),
      _parser(new FrameNaluParser(FrameNaluParser::NotManagerBuffer))
{
}

H264or5VideoNaluFramer::~H264or5VideoNaluFramer() { delete _parser; }

Boolean H264or5VideoNaluFramer::isH264VideoStreamFramer() const { return True; }

Boolean H264or5VideoNaluFramer::isH265VideoStreamFramer() const { return True; }

void H264or5VideoNaluFramer::doGetNextFrame()
{
    if (_parser->findNalu()) {
        fillData();
    } else {
        fInputSource->getNextFrame(fTo, fMaxSize, H264or5VideoNaluFramer::afterGettingFrame, this,
                                   FramedSource::handleClosure, this);
    }
}

void H264or5VideoNaluFramer::afterGettingFrame(void* clientData, unsigned frameSize, unsigned numTruncatedBytes,
                                               struct timeval presentationTime, unsigned durationInMicroseconds)
{
    H264or5VideoNaluFramer* source = (H264or5VideoNaluFramer*)clientData;
    source->afterGettingFrame1(frameSize, numTruncatedBytes, presentationTime, durationInMicroseconds);
}

void H264or5VideoNaluFramer::afterGettingFrame1(unsigned frameSize, unsigned numTruncatedBytes,
                                                struct timeval presentationTime, unsigned durationInMicroseconds)
{
    _parser->setBuffer(fTo, frameSize);
    fPresentationTime = presentationTime;
    if (_parser->findNalu()) {
        fillData();
    }
}

void H264or5VideoNaluFramer::fillData()
{
    uint8_t* data = NULL;
    uint32_t size = 0;
    bool isLast = false;
    _parser->getNextNaluBuffer(data, size, isLast);
    memmove(fTo, data, size);
    fFrameSize = size;
    fPictureEndMarker = isLast;
#ifdef DUMP_DEBUG
    if (isLast) _parser->dumpStartcode();
#endif
    afterGetting(this);
}