obs-qsv11: Add native obs-studio QSV encoder

2015-10-12 01:22:01 -07:00
parent d56dc4791d
commit 3e4bcf06bc
56 changed files with 13085 additions and 0 deletions
--- a/plugins/obs-qsv11/QSV_Encoder_Internal.h
+++ b/plugins/obs-qsv11/QSV_Encoder_Internal.h
@@ -0,0 +1,112 @@
+/*
+
+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.
+*/
+#pragma once
+#include "mfxastructures.h"
+#include "mfxvideo++.h"
+#include "QSV_Encoder.h"
+#include "common_utils.h"
+
+class QSV_Encoder_Internal
+{
+public:
+	QSV_Encoder_Internal(mfxIMPL& impl, mfxVersion& version);
+	~QSV_Encoder_Internal();
+
+	mfxStatus    Open(qsv_param_t * pParams);
+	void         GetSPSPPS(mfxU8 **pSPSBuf, mfxU8 **pPPSBuf,
+			mfxU16 *pnSPSBuf, mfxU16 *pnPPSBuf);
+	mfxStatus    Encode(uint64_t ts, uint8_t *pDataY, uint8_t *pDataUV,
+			uint32_t strideY, uint32_t strideUV, mfxBitstream
+			**pBS);
+	mfxStatus    ClearData();
+	mfxStatus    Reset(qsv_param_t *pParams);
+
+protected:
+	bool         InitParams(qsv_param_t * pParams);
+	mfxStatus    AllocateSurfaces();
+	mfxStatus    GetVideoParam();
+	mfxStatus    InitBitstream();
+	mfxStatus    LoadNV12(mfxFrameSurface1 *pSurface, uint8_t *pDataY,
+			uint8_t *pDataUV, uint32_t strideY, uint32_t strideUV);
+	mfxStatus    Drain();
+	int          GetFreeTaskIndex(Task* pTaskPool, mfxU16 nPoolSize);
+
+private:
+	mfxIMPL                        m_impl;
+	mfxVersion                     m_ver;
+	MFXVideoSession                m_session;
+	mfxFrameAllocator              m_mfxAllocator;
+	mfxVideoParam                  m_mfxEncParams;
+	mfxFrameAllocResponse          m_mfxResponse;
+	mfxFrameSurface1**             m_pmfxSurfaces;
+	mfxU16                         m_nSurfNum;
+	MFXVideoENCODE*                m_pmfxENC;
+	mfxU8                          m_SPSBuffer[100];
+	mfxU8                          m_PPSBuffer[100];
+	mfxU16                         m_nSPSBufferSize;
+	mfxU16                         m_nPPSBufferSize;
+	mfxVideoParam                  m_parameter;
+	mfxExtCodingOption2            m_co2;
+	mfxExtCodingOption             m_co;
+	mfxU16                         m_nTaskPool;
+	Task*                          m_pTaskPool;
+	int                            m_nTaskIdx;
+	int                            m_nFirstSyncTask;
+	mfxBitstream                   m_outBitstream;
+	bool                           m_bIsWindows8OrGreater;
+	bool                           m_bUseD3D11;
+};
+