Ticket #8036: FFMpegEncodingAPI.h

#pragma once
#include "FFMPEG_API.h"
#include "LinkResolveHack.h"
#include "IAESemaphore.h"
#include "HealthData.h"
#include "FrameGrabber.h"

using namespace IAE::VideoProcessing;

namespace IAE
{
        namespace VideoProcessing
        {
                /// C++ Processing Modules Namespace
                namespace FFMPEG
                {
                        class FrameProcessingThread;    // Forward declaration
                        /// <summary>
                        /// Unmanaged interface for interacting with the FFMPEG API.
                        /// </summary>
                        class FFMpegEncodingAPI : public FrameGrabber
                        {
                        private:
                                const AVPixelFormat INTERNAL_FORMAT = AV_PIX_FMT_YUV420P;
                                const int DEFAULT_RTP_PAYLOAD_SIZE = 30000;
                                const int DEFAULT_MAX_SLICE_SIZE = 20000;
                                const int MIN_MP4_MAIN_PROFILE_BR = 2000000;
                                const int MIN_MP4_CORE_PROFILE_BR = 384000;
                                const int MEM_ALIGNMENT = 32;

                                AVCodec *_codec = nullptr;
                                AVFrame *_avFrame = nullptr;
                                SwsContext *_swsContext = nullptr;

                                AVCodecID _selectedCodec;
                                AVPixelFormat _incomingFormat;
                                AVPixelFormat _encodingFormat;

                                int errCode = 0;
                                int _iLastError = 0;
                                uint64_t _ulLastPTS = 0;
                                uint64_t _ulLastFrame = 0;

                                bool _bIsInitialized = false;
                                bool _bIsStopping = false;
                                bool _bIsStopped = false;
                                bool _bCodecOpen = false;

                                double _dTimeScaleFactor = 1;
                                double _dActualFPS = 30;
                                double _dRealPTS = 0;
                                char _codecName [80];

                                void InitializeAPI ();
                                void InitializeContext (AVCodecID codec, AVPixelFormat pixFmt, int64_t lBitRate, const char* accel);
                                void SetAVOptions (int64_t lBitRate);

                                void SetCommonOptions (void* options);
                                void SetH26xOptions (int64_t lBitRate);
                                void SetQSV264Options (void* options);
                                const char* GetEncoderName (const char* accel);

                                void PrintError (char* devNote);
                                uint64_t GetPresentationTime (uint64_t lFrameNo);

                        public:
                                int m_iFramesPerSecond = 30;
                                int m_iGroupOfPictureSize = 10;
                                double m_dScaleFactor = 1.0;

                                bool m_bUseInternalBuffer = false;

                                AVCodecContext *m_encodingContext = nullptr;


                                FFMpegEncodingAPI (int iFrameHeight, int iFrameWidth, int iBytesPerPixel, int iFPS);
                                ~FFMpegEncodingAPI ();

                                int LastError () { return errCode; };
                                bool Initialize (AVCodecID codec, AVPixelFormat pixFmt, int64_t lBitRate, const char* accel);
                                void CleanUp (AVPacket *avPacket);
                                void SetFrameRate (double dFPS);
                        };
                }
        }
}