// Copyright 2019 Fuzhou Rockchip Electronics Co., Ltd. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "mjpeg_video_source.hh"
#include "utils.h"

namespace easymedia
{

    MJPEGVideoSource* MJPEGVideoSource::createNew(UsageEnvironment& env, FramedSource* source)
    {
        return new MJPEGVideoSource(env, source);
    }

    MJPEGVideoSource::MJPEGVideoSource(UsageEnvironment& env, FramedSource* source)
        : JPEGVideoSource(env), m_inputSource(source), m_width(0), m_height(0), m_qTable0Init(false),
          m_qTable1Init(false)
    {
        memset(&m_qTable, 0, sizeof(m_qTable));
    }
    MJPEGVideoSource::~MJPEGVideoSource()
    {
        Medium::close(m_inputSource);
    }
    void MJPEGVideoSource::doGetNextFrame()
    {
        if (m_inputSource) {
            m_inputSource->getNextFrame(fTo, fMaxSize, afterGettingFrame, this, FramedSource::handleClosure, this);
        }
    }

    void MJPEGVideoSource::doStopGettingFrames()
    {
        FramedSource::doStopGettingFrames();
        if (m_inputSource) {
            m_inputSource->stopGettingFrames();
        }
    }
    void MJPEGVideoSource::afterGettingFrame(void* clientData, unsigned frameSize, unsigned numTruncatedBytes,
                                             struct timeval presentationTime, unsigned durationInMicroseconds)
    {
        MJPEGVideoSource* source = (MJPEGVideoSource*)clientData;
        source->afterGettingFrame(frameSize, numTruncatedBytes, presentationTime, durationInMicroseconds);
    }
    void MJPEGVideoSource::afterGettingFrame(unsigned frameSize, unsigned numTruncatedBytes,
                                             struct timeval presentationTime, unsigned durationInMicroseconds)
    {
        int headerSize = 0;
        bool headerOk = false;
        fFrameSize = 0;

        for (unsigned int i = 0; i < frameSize; ++i) {
            // SOF
            if ((i + 8) < frameSize && fTo[i] == 0xFF && fTo[i + 1] == 0xC0) {
                m_height = (fTo[i + 5] << 5) | (fTo[i + 6] >> 3);
                m_width = (fTo[i + 7] << 5) | (fTo[i + 8] >> 3);
            }
            // DQT
            if ((i + 5 + 64) < frameSize && fTo[i] == 0xFF && fTo[i + 1] == 0xDB) {
                if (fTo[i + 4] == 0) {
                    memcpy(m_qTable, fTo + i + 5, 64);
                    m_qTable0Init = true;
                } else if (fTo[i + 4] == 1) {
                    memcpy(m_qTable + 64, fTo + i + 5, 64);
                    m_qTable1Init = true;
                }
            }
            // End of header
            if ((i + 1) < frameSize && fTo[i] == 0x3F && fTo[i + 1] == 0x00) {
                headerOk = true;
                headerSize = i + 2;
                break;
            }
        }

        if (headerOk) {
            fFrameSize = frameSize - headerSize;
            memmove(fTo, fTo + headerSize, fFrameSize);
        }

        fNumTruncatedBytes = numTruncatedBytes;
        fPresentationTime = presentationTime;
        fDurationInMicroseconds = durationInMicroseconds;
        afterGetting(this);
    }

    u_int8_t const* MJPEGVideoSource::quantizationTables(u_int8_t& precision, u_int16_t& length)
    {
        length = 0;
        precision = 0;
        if (m_qTable0Init && m_qTable1Init) {
            precision = 8;
            length = sizeof(m_qTable);
        }
        return m_qTable;
    }

    unsigned char MJPEGVideoSource::type()
    {
        return 1;
    };
    unsigned char MJPEGVideoSource::qFactor()
    {
        return 128;
    };
    unsigned char MJPEGVideoSource::width()
    {
        return m_width;
    };
    unsigned char MJPEGVideoSource::height()
    {
        return m_height;
    };

} // namespace easymedia