598 lines
17 KiB
C++
598 lines
17 KiB
C++
/*
|
|
|
|
This file is provided under a dual BSD/GPLv2 license. When using or
|
|
redistributing this file, you may do so under either license.
|
|
|
|
GPL LICENSE SUMMARY
|
|
|
|
Copyright(c) Oct. 2015 Intel Corporation.
|
|
|
|
This program is free software; you can redistribute it and/or modify
|
|
it under the terms of version 2 of the GNU General Public License as
|
|
published by the Free Software Foundation.
|
|
|
|
This program 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
|
|
General Public License for more details.
|
|
|
|
Contact Information:
|
|
|
|
Seung-Woo Kim, seung-woo.kim@intel.com
|
|
705 5th Ave S #500, Seattle, WA 98104
|
|
|
|
BSD LICENSE
|
|
|
|
Copyright(c) <date> Intel Corporation.
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
modification, are permitted provided that the following conditions
|
|
are met:
|
|
|
|
* Redistributions of source code must retain the above copyright
|
|
notice, this list of conditions and the following disclaimer.
|
|
|
|
* Redistributions in binary form must reproduce the above copyright
|
|
notice, this list of conditions and the following disclaimer in
|
|
the documentation and/or other materials provided with the
|
|
distribution.
|
|
|
|
* Neither the name of Intel Corporation nor the names of its
|
|
contributors may be used to endorse or promote products derived
|
|
from this software without specific prior written permission.
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#include "QSV_Encoder_Internal.h"
|
|
#include "QSV_Encoder.h"
|
|
#include "mfxastructures.h"
|
|
#include "mfxvideo++.h"
|
|
#include <VersionHelpers.h>
|
|
#include <obs-module.h>
|
|
|
|
#define do_log(level, format, ...) \
|
|
blog(level, "[qsv encoder: '%s'] " format, \
|
|
"msdk_impl", ##__VA_ARGS__)
|
|
|
|
#define warn(format, ...) do_log(LOG_WARNING, format, ##__VA_ARGS__)
|
|
#define info(format, ...) do_log(LOG_INFO, format, ##__VA_ARGS__)
|
|
#define debug(format, ...) do_log(LOG_DEBUG, format, ##__VA_ARGS__)
|
|
|
|
QSV_Encoder_Internal::QSV_Encoder_Internal(mfxIMPL& impl, mfxVersion& version) :
|
|
m_pmfxENC(NULL),
|
|
m_nSPSBufferSize(100),
|
|
m_nPPSBufferSize(100),
|
|
m_nTaskPool(0),
|
|
m_pTaskPool(NULL),
|
|
m_nTaskIdx(0),
|
|
m_nFirstSyncTask(0)
|
|
{
|
|
mfxIMPL tempImpl;
|
|
mfxStatus sts;
|
|
|
|
m_bIsWindows8OrGreater = IsWindows8OrGreater();
|
|
m_bUseD3D11 = false;
|
|
|
|
if (m_bIsWindows8OrGreater) {
|
|
tempImpl = impl | MFX_IMPL_VIA_D3D11;
|
|
sts = m_session.Init(tempImpl, &version);
|
|
if (sts == MFX_ERR_NONE) {
|
|
m_session.QueryVersion(&version);
|
|
m_session.Close();
|
|
|
|
// Use D3D11 surface
|
|
// m_bUseD3D11 = ((version.Major > 1) ||
|
|
// (version.Major == 1 && version.Minor >= 8));
|
|
m_bUseD3D11 = true;
|
|
if (m_bUseD3D11)
|
|
blog(LOG_INFO, "\timpl: D3D11\n"
|
|
"\tsurf: D3D11");
|
|
else
|
|
blog(LOG_INFO, "\timpl: D3D11\n"
|
|
"\tsurf: SysMem");
|
|
|
|
m_impl = tempImpl;
|
|
m_ver = version;
|
|
return;
|
|
}
|
|
}
|
|
|
|
// Either windows 7 or D3D11 failed at this point.
|
|
tempImpl = impl | MFX_IMPL_VIA_D3D9;
|
|
sts = m_session.Init(tempImpl, &version);
|
|
if (sts == MFX_ERR_NONE) {
|
|
m_session.QueryVersion(&version);
|
|
m_session.Close();
|
|
|
|
blog(LOG_INFO, "\timpl: D3D09\n"
|
|
"\tsurf: SysMem");
|
|
|
|
m_impl = tempImpl;
|
|
m_ver = version;
|
|
}
|
|
|
|
}
|
|
|
|
QSV_Encoder_Internal::~QSV_Encoder_Internal()
|
|
{
|
|
if (m_pmfxENC)
|
|
ClearData();
|
|
}
|
|
|
|
mfxStatus QSV_Encoder_Internal::Open(qsv_param_t * pParams)
|
|
{
|
|
mfxStatus sts = MFX_ERR_NONE;
|
|
|
|
if (m_bUseD3D11)
|
|
// Use D3D11 surface
|
|
sts = Initialize(m_impl, m_ver, &m_session, &m_mfxAllocator);
|
|
else
|
|
// Use system memory
|
|
sts = Initialize(m_impl, m_ver, &m_session, NULL);
|
|
|
|
MSDK_CHECK_RESULT(sts, MFX_ERR_NONE, sts);
|
|
|
|
m_pmfxENC = new MFXVideoENCODE(m_session);
|
|
|
|
InitParams(pParams);
|
|
|
|
sts = m_pmfxENC->Query(&m_mfxEncParams, &m_mfxEncParams);
|
|
MSDK_IGNORE_MFX_STS(sts, MFX_WRN_INCOMPATIBLE_VIDEO_PARAM);
|
|
MSDK_CHECK_RESULT(sts, MFX_ERR_NONE, sts);
|
|
|
|
sts = AllocateSurfaces();
|
|
MSDK_CHECK_RESULT(sts, MFX_ERR_NONE, sts);
|
|
|
|
sts = m_pmfxENC->Init(&m_mfxEncParams);
|
|
MSDK_CHECK_RESULT(sts, MFX_ERR_NONE, sts);
|
|
|
|
sts = GetVideoParam();
|
|
MSDK_CHECK_RESULT(sts, MFX_ERR_NONE, sts);
|
|
|
|
sts = InitBitstream();
|
|
MSDK_CHECK_RESULT(sts, MFX_ERR_NONE, sts);
|
|
|
|
return sts;
|
|
}
|
|
|
|
|
|
bool QSV_Encoder_Internal::InitParams(qsv_param_t * pParams)
|
|
{
|
|
memset(&m_mfxEncParams, 0, sizeof(m_mfxEncParams));
|
|
|
|
m_mfxEncParams.mfx.CodecId = MFX_CODEC_AVC;
|
|
m_mfxEncParams.mfx.GopOptFlag = MFX_GOP_STRICT;
|
|
m_mfxEncParams.mfx.NumSlice = 1;
|
|
m_mfxEncParams.mfx.TargetUsage = pParams->nTargetUsage;
|
|
m_mfxEncParams.mfx.CodecProfile = pParams->nCodecProfile;
|
|
m_mfxEncParams.mfx.FrameInfo.FrameRateExtN = pParams->nFpsNum;
|
|
m_mfxEncParams.mfx.FrameInfo.FrameRateExtD = pParams->nFpsDen;
|
|
m_mfxEncParams.mfx.FrameInfo.FourCC = MFX_FOURCC_NV12;
|
|
m_mfxEncParams.mfx.FrameInfo.ChromaFormat = MFX_CHROMAFORMAT_YUV420;
|
|
m_mfxEncParams.mfx.FrameInfo.PicStruct = MFX_PICSTRUCT_PROGRESSIVE;
|
|
m_mfxEncParams.mfx.FrameInfo.CropX = 0;
|
|
m_mfxEncParams.mfx.FrameInfo.CropY = 0;
|
|
m_mfxEncParams.mfx.FrameInfo.CropW = pParams->nWidth;
|
|
m_mfxEncParams.mfx.FrameInfo.CropH = pParams->nHeight;
|
|
|
|
m_mfxEncParams.mfx.RateControlMethod = pParams->nRateControl;
|
|
|
|
switch (pParams->nRateControl) {
|
|
case MFX_RATECONTROL_CBR:
|
|
m_mfxEncParams.mfx.TargetKbps = pParams->nTargetBitRate;
|
|
break;
|
|
case MFX_RATECONTROL_VBR:
|
|
case MFX_RATECONTROL_VCM:
|
|
m_mfxEncParams.mfx.TargetKbps = pParams->nTargetBitRate;
|
|
m_mfxEncParams.mfx.MaxKbps = pParams->nMaxBitRate;
|
|
break;
|
|
case MFX_RATECONTROL_CQP:
|
|
m_mfxEncParams.mfx.QPI = pParams->nQPI;
|
|
m_mfxEncParams.mfx.QPB = pParams->nQPB;
|
|
m_mfxEncParams.mfx.QPP = pParams->nQPP;
|
|
break;
|
|
case MFX_RATECONTROL_AVBR:
|
|
m_mfxEncParams.mfx.TargetKbps = pParams->nTargetBitRate;
|
|
m_mfxEncParams.mfx.Accuracy = pParams->nAccuracy;
|
|
m_mfxEncParams.mfx.Convergence = pParams->nConvergence;
|
|
break;
|
|
case MFX_RATECONTROL_ICQ:
|
|
m_mfxEncParams.mfx.ICQQuality = pParams->nICQQuality;
|
|
break;
|
|
case MFX_RATECONTROL_LA:
|
|
m_mfxEncParams.mfx.TargetKbps = pParams->nTargetBitRate;
|
|
break;
|
|
case MFX_RATECONTROL_LA_ICQ:
|
|
m_mfxEncParams.mfx.ICQQuality = pParams->nICQQuality;
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
|
|
m_mfxEncParams.AsyncDepth = pParams->nAsyncDepth;
|
|
m_mfxEncParams.mfx.GopPicSize = (mfxU16)(pParams->nKeyIntSec *
|
|
pParams->nFpsNum / (float)pParams->nFpsDen);
|
|
|
|
static mfxExtBuffer* extendedBuffers[2];
|
|
int iBuffers = 0;
|
|
if (pParams->nAsyncDepth == 1) {
|
|
m_mfxEncParams.mfx.NumRefFrame = 1;
|
|
// low latency, I and P frames only
|
|
m_mfxEncParams.mfx.GopRefDist = 1;
|
|
memset(&m_co, 0, sizeof(mfxExtCodingOption));
|
|
m_co.Header.BufferId = MFX_EXTBUFF_CODING_OPTION;
|
|
m_co.Header.BufferSz = sizeof(mfxExtCodingOption);
|
|
m_co.MaxDecFrameBuffering = 1;
|
|
extendedBuffers[iBuffers++] = (mfxExtBuffer*)&m_co;
|
|
}
|
|
else
|
|
m_mfxEncParams.mfx.GopRefDist = pParams->nbFrames + 1;
|
|
|
|
if (pParams->nRateControl == MFX_RATECONTROL_LA_ICQ ||
|
|
pParams->nRateControl == MFX_RATECONTROL_LA) {
|
|
|
|
memset(&m_co2, 0, sizeof(mfxExtCodingOption2));
|
|
m_co2.Header.BufferId = MFX_EXTBUFF_CODING_OPTION;
|
|
m_co2.Header.BufferSz = sizeof(m_co2);
|
|
m_co2.LookAheadDepth = pParams->nLADEPTH;
|
|
extendedBuffers[iBuffers++] = (mfxExtBuffer*)& m_co2;
|
|
}
|
|
|
|
if (iBuffers > 0) {
|
|
m_mfxEncParams.ExtParam = extendedBuffers;
|
|
m_mfxEncParams.NumExtParam = (mfxU16)iBuffers;
|
|
}
|
|
|
|
// Width must be a multiple of 16
|
|
// Height must be a multiple of 16 in case of frame picture and a
|
|
// multiple of 32 in case of field picture
|
|
m_mfxEncParams.mfx.FrameInfo.Width = MSDK_ALIGN16(pParams->nWidth);
|
|
m_mfxEncParams.mfx.FrameInfo.Height = MSDK_ALIGN16(pParams->nHeight);
|
|
|
|
if (m_bUseD3D11)
|
|
m_mfxEncParams.IOPattern = MFX_IOPATTERN_IN_VIDEO_MEMORY;
|
|
else
|
|
m_mfxEncParams.IOPattern = MFX_IOPATTERN_IN_SYSTEM_MEMORY;
|
|
|
|
return true;
|
|
}
|
|
|
|
mfxStatus QSV_Encoder_Internal::AllocateSurfaces()
|
|
{
|
|
// Query number of required surfaces for encoder
|
|
mfxFrameAllocRequest EncRequest;
|
|
memset(&EncRequest, 0, sizeof(EncRequest));
|
|
mfxStatus sts = m_pmfxENC->QueryIOSurf(&m_mfxEncParams, &EncRequest);
|
|
MSDK_CHECK_RESULT(sts, MFX_ERR_NONE, sts);
|
|
|
|
EncRequest.Type |= WILL_WRITE;
|
|
|
|
// SNB hack. On some SNB, it seems to require more surfaces
|
|
EncRequest.NumFrameSuggested += m_mfxEncParams.AsyncDepth;
|
|
|
|
// Allocate required surfaces
|
|
if (m_bUseD3D11) {
|
|
sts = m_mfxAllocator.Alloc(m_mfxAllocator.pthis, &EncRequest,
|
|
&m_mfxResponse);
|
|
MSDK_CHECK_RESULT(sts, MFX_ERR_NONE, sts);
|
|
|
|
m_nSurfNum = m_mfxResponse.NumFrameActual;
|
|
|
|
m_pmfxSurfaces = new mfxFrameSurface1 *[m_nSurfNum];
|
|
MSDK_CHECK_POINTER(m_pmfxSurfaces, MFX_ERR_MEMORY_ALLOC);
|
|
|
|
for (int i = 0; i < m_nSurfNum; i++) {
|
|
m_pmfxSurfaces[i] = new mfxFrameSurface1;
|
|
memset(m_pmfxSurfaces[i], 0, sizeof(mfxFrameSurface1));
|
|
memcpy(&(m_pmfxSurfaces[i]->Info),
|
|
&(m_mfxEncParams.mfx.FrameInfo),
|
|
sizeof(mfxFrameInfo));
|
|
m_pmfxSurfaces[i]->Data.MemId = m_mfxResponse.mids[i];
|
|
}
|
|
}
|
|
else {
|
|
mfxU16 width = (mfxU16)MSDK_ALIGN32(EncRequest.Info.Width);
|
|
mfxU16 height = (mfxU16)MSDK_ALIGN32(EncRequest.Info.Height);
|
|
mfxU8 bitsPerPixel = 12;
|
|
mfxU32 surfaceSize = width * height * bitsPerPixel / 8;
|
|
m_nSurfNum = EncRequest.NumFrameSuggested;
|
|
|
|
m_pmfxSurfaces = new mfxFrameSurface1 *[m_nSurfNum];
|
|
for (int i = 0; i < m_nSurfNum; i++) {
|
|
m_pmfxSurfaces[i] = new mfxFrameSurface1;
|
|
memset(m_pmfxSurfaces[i], 0, sizeof(mfxFrameSurface1));
|
|
memcpy(&(m_pmfxSurfaces[i]->Info),
|
|
&(m_mfxEncParams.mfx.FrameInfo),
|
|
sizeof(mfxFrameInfo));
|
|
|
|
mfxU8* pSurface = (mfxU8*) new mfxU8[surfaceSize];
|
|
m_pmfxSurfaces[i]->Data.Y = pSurface;
|
|
m_pmfxSurfaces[i]->Data.U = pSurface + width * height;
|
|
m_pmfxSurfaces[i]->Data.V = pSurface + width * height + 1;
|
|
m_pmfxSurfaces[i]->Data.Pitch = width;
|
|
}
|
|
}
|
|
|
|
blog(LOG_INFO, "\tm_nSurfNum: %d", m_nSurfNum);
|
|
|
|
return sts;
|
|
}
|
|
|
|
mfxStatus QSV_Encoder_Internal::GetVideoParam()
|
|
{
|
|
memset(&m_parameter, 0, sizeof(m_parameter));
|
|
mfxExtCodingOptionSPSPPS opt;
|
|
memset(&m_parameter, 0, sizeof(m_parameter));
|
|
opt.Header.BufferId = MFX_EXTBUFF_CODING_OPTION_SPSPPS;
|
|
opt.Header.BufferSz = sizeof(mfxExtCodingOptionSPSPPS);
|
|
|
|
static mfxExtBuffer* extendedBuffers[1];
|
|
extendedBuffers[0] = (mfxExtBuffer*)& opt;
|
|
m_parameter.ExtParam = extendedBuffers;
|
|
m_parameter.NumExtParam = 1;
|
|
|
|
opt.SPSBuffer = m_SPSBuffer;
|
|
opt.PPSBuffer = m_PPSBuffer;
|
|
opt.SPSBufSize = 100; // m_nSPSBufferSize;
|
|
opt.PPSBufSize = 100; // m_nPPSBufferSize;
|
|
|
|
mfxStatus sts = m_pmfxENC->GetVideoParam(&m_parameter);
|
|
MSDK_CHECK_RESULT(sts, MFX_ERR_NONE, sts);
|
|
|
|
m_nSPSBufferSize = opt.SPSBufSize;
|
|
m_nPPSBufferSize = opt.PPSBufSize;
|
|
|
|
return sts;
|
|
}
|
|
|
|
void QSV_Encoder_Internal::GetSPSPPS(mfxU8 **pSPSBuf, mfxU8 **pPPSBuf,
|
|
mfxU16 *pnSPSBuf, mfxU16 *pnPPSBuf)
|
|
{
|
|
*pSPSBuf = m_SPSBuffer;
|
|
*pPPSBuf = m_PPSBuffer;
|
|
*pnSPSBuf = m_nSPSBufferSize;
|
|
*pnPPSBuf = m_nPPSBufferSize;
|
|
}
|
|
|
|
mfxStatus QSV_Encoder_Internal::InitBitstream()
|
|
{
|
|
m_nTaskPool = m_parameter.AsyncDepth;
|
|
m_nFirstSyncTask = 0;
|
|
|
|
m_pTaskPool = new Task[m_nTaskPool];
|
|
memset(m_pTaskPool, 0, sizeof(Task) * m_nTaskPool);
|
|
|
|
for (int i = 0; i < m_nTaskPool; i++) {
|
|
m_pTaskPool[i].mfxBS.MaxLength =
|
|
m_parameter.mfx.BufferSizeInKB * 1000;
|
|
m_pTaskPool[i].mfxBS.Data =
|
|
new mfxU8[m_pTaskPool[i].mfxBS.MaxLength];
|
|
m_pTaskPool[i].mfxBS.DataOffset = 0;
|
|
m_pTaskPool[i].mfxBS.DataLength = 0;
|
|
|
|
MSDK_CHECK_POINTER(m_pTaskPool[i].mfxBS.Data,
|
|
MFX_ERR_MEMORY_ALLOC);
|
|
}
|
|
|
|
memset(&m_outBitstream, 0, sizeof(mfxBitstream));
|
|
m_outBitstream.MaxLength = m_parameter.mfx.BufferSizeInKB * 1000;
|
|
m_outBitstream.Data = new mfxU8[m_outBitstream.MaxLength];
|
|
m_outBitstream.DataOffset = 0;
|
|
m_outBitstream.DataLength = 0;
|
|
|
|
blog(LOG_INFO, "\tm_nTaskPool: %d", m_nTaskPool);
|
|
|
|
return MFX_ERR_NONE;
|
|
}
|
|
|
|
mfxStatus QSV_Encoder_Internal::LoadNV12(mfxFrameSurface1 *pSurface,
|
|
uint8_t *pDataY, uint8_t *pDataUV, uint32_t strideY,
|
|
uint32_t strideUV)
|
|
{
|
|
mfxU16 w, h, i, pitch;
|
|
mfxU8* ptr;
|
|
mfxFrameInfo* pInfo = &pSurface->Info;
|
|
mfxFrameData* pData = &pSurface->Data;
|
|
|
|
if (pInfo->CropH > 0 && pInfo->CropW > 0)
|
|
{
|
|
w = pInfo->CropW;
|
|
h = pInfo->CropH;
|
|
}
|
|
else
|
|
{
|
|
w = pInfo->Width;
|
|
h = pInfo->Height;
|
|
}
|
|
|
|
pitch = pData->Pitch;
|
|
ptr = pData->Y + pInfo->CropX + pInfo->CropY * pData->Pitch;
|
|
|
|
// load Y plane
|
|
for (i = 0; i < h; i++)
|
|
memcpy(ptr + i * pitch, pDataY + i * strideY, w);
|
|
|
|
// load UV plane
|
|
h /= 2;
|
|
ptr = pData->UV + pInfo->CropX + (pInfo->CropY / 2) * pitch;
|
|
|
|
for (i = 0; i < h; i++)
|
|
memcpy(ptr + i * pitch, pDataUV + i * strideUV, w);
|
|
|
|
return MFX_ERR_NONE;
|
|
}
|
|
|
|
int QSV_Encoder_Internal::GetFreeTaskIndex(Task* pTaskPool, mfxU16 nPoolSize)
|
|
{
|
|
if (pTaskPool)
|
|
for (int i = 0; i < nPoolSize; i++)
|
|
if (!pTaskPool[i].syncp)
|
|
return i;
|
|
return MFX_ERR_NOT_FOUND;
|
|
}
|
|
|
|
mfxStatus QSV_Encoder_Internal::Encode(uint64_t ts, uint8_t *pDataY,
|
|
uint8_t *pDataUV, uint32_t strideY, uint32_t strideUV,
|
|
mfxBitstream **pBS)
|
|
{
|
|
mfxStatus sts = MFX_ERR_NONE;
|
|
*pBS = NULL;
|
|
int nTaskIdx = GetFreeTaskIndex(m_pTaskPool, m_nTaskPool);
|
|
|
|
#if 0
|
|
info("MSDK Encode:\n"
|
|
"\tTaskIndex: %d",
|
|
nTaskIdx);
|
|
#endif
|
|
|
|
int nSurfIdx = GetFreeSurfaceIndex(m_pmfxSurfaces, m_nSurfNum);
|
|
#if 0
|
|
info("MSDK Encode:\n"
|
|
"\tnSurfIdx: %d",
|
|
nSurfIdx);
|
|
#endif
|
|
|
|
while (MFX_ERR_NOT_FOUND == nTaskIdx || MFX_ERR_NOT_FOUND == nSurfIdx) {
|
|
// No more free tasks or surfaces, need to sync
|
|
sts = m_session.SyncOperation(m_pTaskPool[m_nFirstSyncTask].syncp,
|
|
60000);
|
|
MSDK_CHECK_RESULT(sts, MFX_ERR_NONE, sts);
|
|
|
|
mfxU8 *pTemp = m_outBitstream.Data;
|
|
memcpy(&m_outBitstream, &m_pTaskPool[m_nFirstSyncTask].mfxBS,
|
|
sizeof(mfxBitstream));
|
|
|
|
m_pTaskPool[m_nFirstSyncTask].mfxBS.Data = pTemp;
|
|
m_pTaskPool[m_nFirstSyncTask].mfxBS.DataLength = 0;
|
|
m_pTaskPool[m_nFirstSyncTask].mfxBS.DataOffset = 0;
|
|
m_pTaskPool[m_nFirstSyncTask].syncp = NULL;
|
|
nTaskIdx = m_nFirstSyncTask;
|
|
m_nFirstSyncTask = (m_nFirstSyncTask + 1) % m_nTaskPool;
|
|
*pBS = &m_outBitstream;
|
|
|
|
#if 0
|
|
info("MSDK Encode:\n"
|
|
"\tnew FirstSyncTask: %d\n"
|
|
"\tTaskIndex: %d",
|
|
m_nFirstSyncTask,
|
|
nTaskIdx);
|
|
#endif
|
|
|
|
nSurfIdx = GetFreeSurfaceIndex(m_pmfxSurfaces, m_nSurfNum);
|
|
#if 0
|
|
info("MSDK Encode:\n"
|
|
"\tnSurfIdx: %d",
|
|
nSurfIdx);
|
|
#endif
|
|
}
|
|
|
|
mfxFrameSurface1 *pSurface = m_pmfxSurfaces[nSurfIdx];
|
|
if (m_bUseD3D11)
|
|
sts = m_mfxAllocator.Lock(m_mfxAllocator.pthis,
|
|
pSurface->Data.MemId, &(pSurface->Data));
|
|
|
|
sts = LoadNV12(pSurface, pDataY, pDataUV, strideY, strideUV);
|
|
pSurface->Data.TimeStamp = ts;
|
|
|
|
if (m_bUseD3D11)
|
|
sts = m_mfxAllocator.Unlock(m_mfxAllocator.pthis,
|
|
pSurface->Data.MemId, &(pSurface->Data));
|
|
|
|
MSDK_CHECK_RESULT(sts, MFX_ERR_NONE, sts);
|
|
|
|
for (;;) {
|
|
// Encode a frame asychronously (returns immediately)
|
|
sts = m_pmfxENC->EncodeFrameAsync(NULL, pSurface,
|
|
&m_pTaskPool[nTaskIdx].mfxBS,
|
|
&m_pTaskPool[nTaskIdx].syncp);
|
|
|
|
if (MFX_ERR_NONE < sts && !m_pTaskPool[nTaskIdx].syncp) {
|
|
// Repeat the call if warning and no output
|
|
if (MFX_WRN_DEVICE_BUSY == sts)
|
|
MSDK_SLEEP(1); // Wait if device is busy, then repeat the same call
|
|
} else if (MFX_ERR_NONE < sts && m_pTaskPool[nTaskIdx].syncp) {
|
|
sts = MFX_ERR_NONE; // Ignore warnings if output is available
|
|
break;
|
|
} else if (MFX_ERR_NOT_ENOUGH_BUFFER == sts) {
|
|
// Allocate more bitstream buffer memory here if needed...
|
|
break;
|
|
} else
|
|
break;
|
|
}
|
|
|
|
return sts;
|
|
}
|
|
|
|
mfxStatus QSV_Encoder_Internal::Drain()
|
|
{
|
|
mfxStatus sts = MFX_ERR_NONE;
|
|
|
|
while (m_pTaskPool[m_nFirstSyncTask].syncp) {
|
|
sts = m_session.SyncOperation(m_pTaskPool[m_nFirstSyncTask].syncp, 60000);
|
|
MSDK_CHECK_RESULT(sts, MFX_ERR_NONE, sts);
|
|
|
|
m_pTaskPool[m_nFirstSyncTask].syncp = NULL;
|
|
m_nFirstSyncTask = (m_nFirstSyncTask + 1) % m_nTaskPool;
|
|
}
|
|
|
|
return sts;
|
|
}
|
|
|
|
mfxStatus QSV_Encoder_Internal::ClearData()
|
|
{
|
|
mfxStatus sts = MFX_ERR_NONE;
|
|
sts = Drain();
|
|
|
|
sts = m_pmfxENC->Close();
|
|
|
|
if (m_bUseD3D11)
|
|
m_mfxAllocator.Free(m_mfxAllocator.pthis, &m_mfxResponse);
|
|
|
|
for (int i = 0; i < m_nSurfNum; i++) {
|
|
if (!m_bUseD3D11)
|
|
delete m_pmfxSurfaces[i]->Data.Y;
|
|
|
|
delete m_pmfxSurfaces[i];
|
|
}
|
|
MSDK_SAFE_DELETE_ARRAY(m_pmfxSurfaces);
|
|
|
|
for (int i = 0; i < m_nTaskPool; i++)
|
|
delete m_pTaskPool[i].mfxBS.Data;
|
|
MSDK_SAFE_DELETE_ARRAY(m_pTaskPool);
|
|
|
|
delete m_outBitstream.Data;
|
|
|
|
if (m_pmfxENC != NULL) {
|
|
delete m_pmfxENC;
|
|
m_pmfxENC = NULL;
|
|
}
|
|
|
|
if (m_bUseD3D11)
|
|
Release();
|
|
|
|
m_session.Close();
|
|
|
|
return sts;
|
|
}
|
|
|
|
mfxStatus QSV_Encoder_Internal::Reset(qsv_param_t *pParams)
|
|
{
|
|
mfxStatus sts = ClearData();
|
|
MSDK_CHECK_RESULT(sts, MFX_ERR_NONE, sts);
|
|
|
|
sts = Open(pParams);
|
|
MSDK_CHECK_RESULT(sts, MFX_ERR_NONE, sts);
|
|
|
|
return sts;
|
|
}
|