123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242 |
- #include "common_utils.h"
- #include "common_directx11.h"
- #include <util/windows/device-enum.h>
- #include <util/config-file.h>
- #include <util/platform.h>
- #include <util/pipe.h>
- #include <util/dstr.h>
- #include <intrin.h>
- #include <inttypes.h>
- #include <obs-module.h>
- /* =======================================================
- * Windows implementation of OS-specific utility functions
- */
- mfxStatus Initialize(mfxVersion ver, mfxSession *pSession, mfxFrameAllocator *pmfxAllocator, mfxHDL *deviceHandle,
- bool bCreateSharedHandles, enum qsv_codec codec, void **data)
- {
- UNUSED_PARAMETER(codec);
- UNUSED_PARAMETER(data);
- obs_video_info ovi;
- obs_get_video_info(&ovi);
- mfxU32 adapter_idx = ovi.adapter;
- mfxU32 idx_adjustment = 0;
- // Select current adapter - will be iGPU if exists due to adapter reordering
- if (codec == QSV_CODEC_AV1 && !adapters[adapter_idx].supports_av1) {
- for (mfxU32 i = 0; i < MAX_ADAPTERS; i++) {
- if (!adapters[i].is_intel) {
- idx_adjustment++;
- continue;
- }
- if (adapters[i].supports_av1) {
- adapter_idx = i;
- break;
- }
- }
- } else if (!adapters[adapter_idx].is_intel) {
- for (mfxU32 i = 0; i < MAX_ADAPTERS; i++) {
- if (adapters[i].is_intel) {
- adapter_idx = i;
- break;
- }
- idx_adjustment++;
- }
- }
- adapter_idx -= idx_adjustment;
- mfxStatus sts = MFX_ERR_NONE;
- mfxVariant impl;
- // If mfxFrameAllocator is provided it means we need to setup DirectX device and memory allocator
- if (pmfxAllocator) {
- // Initialize Intel VPL Session
- mfxLoader loader = MFXLoad();
- mfxConfig cfg = MFXCreateConfig(loader);
- impl.Type = MFX_VARIANT_TYPE_U32;
- impl.Data.U32 = MFX_IMPL_TYPE_HARDWARE;
- MFXSetConfigFilterProperty(cfg, (const mfxU8 *)"mfxImplDescription.Impl", impl);
- impl.Type = MFX_VARIANT_TYPE_U32;
- impl.Data.U32 = INTEL_VENDOR_ID;
- MFXSetConfigFilterProperty(cfg, (const mfxU8 *)"mfxImplDescription.VendorID", impl);
- impl.Type = MFX_VARIANT_TYPE_U32;
- impl.Data.U32 = MFX_ACCEL_MODE_VIA_D3D11;
- MFXSetConfigFilterProperty(cfg, (const mfxU8 *)"mfxImplDescription.AccelerationMode", impl);
- sts = MFXCreateSession(loader, adapter_idx, pSession);
- MSDK_CHECK_RESULT(sts, MFX_ERR_NONE, sts);
- // Create DirectX device context
- if (deviceHandle == NULL || *deviceHandle == NULL) {
- sts = CreateHWDevice(*pSession, deviceHandle, NULL, bCreateSharedHandles);
- MSDK_CHECK_RESULT(sts, MFX_ERR_NONE, sts);
- }
- if (deviceHandle == NULL || *deviceHandle == NULL)
- return MFX_ERR_DEVICE_FAILED;
- // Provide device manager to VPL
- sts = MFXVideoCORE_SetHandle(*pSession, DEVICE_MGR_TYPE, *deviceHandle);
- MSDK_CHECK_RESULT(sts, MFX_ERR_NONE, sts);
- pmfxAllocator->pthis = *pSession; // We use VPL session ID as the allocation identifier
- pmfxAllocator->Alloc = simple_alloc;
- pmfxAllocator->Free = simple_free;
- pmfxAllocator->Lock = simple_lock;
- pmfxAllocator->Unlock = simple_unlock;
- pmfxAllocator->GetHDL = simple_gethdl;
- // Since we are using video memory we must provide VPL with an external allocator
- sts = MFXVideoCORE_SetFrameAllocator(*pSession, pmfxAllocator);
- MSDK_CHECK_RESULT(sts, MFX_ERR_NONE, sts);
- } else {
- // Initialize Intel VPL Session
- mfxLoader loader = MFXLoad();
- mfxConfig cfg = MFXCreateConfig(loader);
- impl.Type = MFX_VARIANT_TYPE_U32;
- impl.Data.U32 = MFX_IMPL_TYPE_HARDWARE;
- MFXSetConfigFilterProperty(cfg, (const mfxU8 *)"mfxImplDescription.Impl", impl);
- impl.Type = MFX_VARIANT_TYPE_U32;
- impl.Data.U32 = INTEL_VENDOR_ID;
- MFXSetConfigFilterProperty(cfg, (const mfxU8 *)"mfxImplDescription.VendorID", impl);
- impl.Type = MFX_VARIANT_TYPE_U32;
- impl.Data.U32 = MFX_ACCEL_MODE_VIA_D3D9;
- MFXSetConfigFilterProperty(cfg, (const mfxU8 *)"mfxImplDescription.AccelerationMode", impl);
- sts = MFXCreateSession(loader, adapter_idx, pSession);
- MSDK_CHECK_RESULT(sts, MFX_ERR_NONE, sts);
- }
- return sts;
- }
- void Release()
- {
- CleanupHWDevice();
- }
- void ReleaseSessionData(void *) {}
- void mfxGetTime(mfxTime *timestamp)
- {
- QueryPerformanceCounter(timestamp);
- }
- double TimeDiffMsec(mfxTime tfinish, mfxTime tstart)
- {
- static LARGE_INTEGER tFreq = {0};
- if (!tFreq.QuadPart)
- QueryPerformanceFrequency(&tFreq);
- double freq = (double)tFreq.QuadPart;
- return 1000.0 * ((double)tfinish.QuadPart - (double)tstart.QuadPart) / freq;
- }
- void util_cpuid(int cpuinfo[4], int flags)
- {
- return __cpuid(cpuinfo, flags);
- }
- static bool enum_luids(void *param, uint32_t idx, uint64_t luid)
- {
- struct dstr *cmd = (struct dstr *)param;
- dstr_catf(cmd, " %" PRIx64, luid);
- UNUSED_PARAMETER(idx);
- return true;
- }
- void check_adapters(struct adapter_info *adapters, size_t *adapter_count)
- {
- char *test_exe = os_get_executable_path_ptr("obs-qsv-test.exe");
- struct dstr cmd = {0};
- struct dstr caps_str = {0};
- os_process_pipe_t *pp = nullptr;
- config_t *config = nullptr;
- const char *error = nullptr;
- size_t config_adapter_count;
- dstr_init_move_array(&cmd, test_exe);
- dstr_insert_ch(&cmd, 0, '\"');
- dstr_cat(&cmd, "\"");
- enum_graphics_device_luids(enum_luids, &cmd);
- pp = os_process_pipe_create(cmd.array, "r");
- if (!pp) {
- blog(LOG_INFO, "Failed to launch the QSV test process I guess");
- goto fail;
- }
- for (;;) {
- char data[2048];
- size_t len = os_process_pipe_read(pp, (uint8_t *)data, sizeof(data));
- if (!len)
- break;
- dstr_ncat(&caps_str, data, len);
- }
- if (dstr_is_empty(&caps_str)) {
- blog(LOG_INFO, "Seems the QSV test subprocess crashed. "
- "Better there than here I guess. "
- "Let's just skip loading QSV then I suppose.");
- goto fail;
- }
- if (config_open_string(&config, caps_str.array) != 0) {
- blog(LOG_INFO, "Couldn't open QSV configuration string");
- goto fail;
- }
- error = config_get_string(config, "error", "string");
- if (error) {
- blog(LOG_INFO, "Error querying QSV support: %s", error);
- goto fail;
- }
- config_adapter_count = config_num_sections(config);
- if (config_adapter_count < *adapter_count)
- *adapter_count = config_adapter_count;
- for (size_t i = 0; i < *adapter_count; i++) {
- char section[16];
- snprintf(section, sizeof(section), "%d", (int)i);
- struct adapter_info *adapter = &adapters[i];
- adapter->is_intel = config_get_bool(config, section, "is_intel");
- adapter->is_dgpu = config_get_bool(config, section, "is_dgpu");
- adapter->supports_av1 = config_get_bool(config, section, "supports_av1");
- adapter->supports_hevc = config_get_bool(config, section, "supports_hevc");
- }
- fail:
- config_close(config);
- dstr_free(&caps_str);
- dstr_free(&cmd);
- os_process_pipe_destroy(pp);
- }
- /* (Lain) Functions currently unused */
- #if 0
- void ClearYUVSurfaceVMem(mfxMemId memId)
- {
- ClearYUVSurfaceD3D(memId);
- }
- void ClearRGBSurfaceVMem(mfxMemId memId)
- {
- ClearRGBSurfaceD3D(memId);
- }
- #endif
|