obs-qsv11: Add native obs-studio QSV encoder

2026-09-24 23:31:36 +08:00 · 2016-04-19 08:29:22 -07:00
parent d56dc4791d
commit 3e4bcf06bc
56 changed files with 13085 additions and 0 deletions
@@ -0,0 +1,211 @@
+/*
+
+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.
+*/
+
+// QSV_Encoder.cpp : Defines the exported functions for the DLL application.
+//
+
+#include "QSV_Encoder.h"
+#include "QSV_Encoder_Internal.h"
+#include <string>
+
+QSV_Encoder_Internal *g_pEncoder = NULL;
+mfxIMPL              impl = MFX_IMPL_HARDWARE_ANY;
+mfxVersion           ver = {{0, 1}}; // for backward compatibility
+
+void qsv_encoder_version(unsigned short *major, unsigned short *minor)
+{
+	*major = ver.Major;
+	*minor = ver.Minor;
+}
+
+qsv_t *qsv_encoder_open(qsv_param_t *pParams)
+{
+	QSV_Encoder_Internal *pEncoder = new QSV_Encoder_Internal(impl, ver);
+	mfxStatus sts = pEncoder->Open(pParams);
+	if (sts != MFX_ERR_NONE) {
+		delete pEncoder;
+		return NULL;
+	}
+
+	return (qsv_t *) pEncoder;
+}
+
+int qsv_encoder_headers(qsv_t *pContext, uint8_t **pSPS, uint8_t **pPPS,
+		uint16_t *pnSPS, uint16_t *pnPPS)
+{
+	QSV_Encoder_Internal *pEncoder = (QSV_Encoder_Internal *)pContext;
+	pEncoder->GetSPSPPS(pSPS, pPPS, pnSPS, pnPPS);
+
+	return 0;
+}
+
+int qsv_encoder_encode(qsv_t * pContext, uint64_t ts, uint8_t *pDataY,
+		uint8_t *pDataUV, uint32_t strideY, uint32_t strideUV,
+		mfxBitstream **pBS)
+{
+	QSV_Encoder_Internal *pEncoder = (QSV_Encoder_Internal *)pContext;
+	mfxStatus sts = MFX_ERR_NONE;
+
+	if (pDataY != NULL && pDataUV != NULL)
+		sts = pEncoder->Encode(ts, pDataY, pDataUV, strideY, strideUV,
+				pBS);
+
+	if (sts == MFX_ERR_NONE)
+		return 0;
+	else if (sts == MFX_ERR_MORE_DATA)
+		return 1;
+	else
+		return -1;
+}
+
+int qsv_encoder_close(qsv_t *pContext)
+{
+	QSV_Encoder_Internal *pEncoder = (QSV_Encoder_Internal *)pContext;
+	delete pEncoder;
+
+	return 0;
+}
+
+/*
+int qsv_param_default_preset(qsv_param_t *pParams, const char *preset,
+		const char *tune)
+{
+	return 0;
+}
+
+int qsv_param_parse(qsv_param_t *, const char *name, const char *value)
+{
+	return 0;
+}
+
+int qsv_param_apply_profile(qsv_param_t *, const char *profile)
+{
+	return 0;
+}
+*/
+
+int qsv_encoder_reconfig(qsv_t *pContext, qsv_param_t *pParams)
+{
+	QSV_Encoder_Internal *pEncoder = (QSV_Encoder_Internal *)pContext;
+	mfxStatus sts = pEncoder->Reset(pParams);
+
+	if (sts == MFX_ERR_NONE)
+		return 0;
+	else
+		return -1;
+}
+
+enum qsv_cpu_platform qsv_get_cpu_platform()
+{
+	using std::string;
+
+	int cpuInfo[4];
+	__cpuid(cpuInfo, 0);
+
+	string vendor;
+	vendor += string((char*)&cpuInfo[1], 4);
+	vendor += string((char*)&cpuInfo[3], 4);
+	vendor += string((char*)&cpuInfo[2], 4);
+
+	if (vendor != "GenuineIntel")
+		return QSV_CPU_PLATFORM_UNKNOWN;
+
+	__cpuid(cpuInfo, 1);
+	BYTE model = ((cpuInfo[0] >> 4) & 0xF) + ((cpuInfo[0] >> 12) & 0xF0);
+	BYTE family = ((cpuInfo[0] >> 8) & 0xF) + ((cpuInfo[0] >> 20) & 0xFF);
+
+	// See Intel 64 and IA-32 Architectures Software Developer's Manual,
+	// Vol 3C Table 35-1
+	if (family != 6)
+		return QSV_CPU_PLATFORM_UNKNOWN;
+
+	switch (model)
+	{
+	case 0x1C:
+	case 0x26:
+	case 0x27:
+	case 0x35:
+	case 0x36:
+		return QSV_CPU_PLATFORM_BNL;
+
+	case 0x2a:
+	case 0x2d:
+		return QSV_CPU_PLATFORM_SNB;
+
+	case 0x3a:
+	case 0x3e:
+		return QSV_CPU_PLATFORM_IVB;
+
+	case 0x37:
+	case 0x4A:
+	case 0x4D:
+	case 0x5A:
+	case 0x5D:
+		return QSV_CPU_PLATFORM_SLM;
+
+	case 0x4C:
+		return QSV_CPU_PLATFORM_CHT;
+
+	case 0x3c:
+	case 0x3f:
+	case 0x45:
+	case 0x46:
+		return QSV_CPU_PLATFORM_HSW;
+	}
+
+	//assume newer revisions are at least as capable as haswell
+	return QSV_CPU_PLATFORM_INTEL;
+}