mirror of
https://github.com/fastfetch-cli/fastfetch.git
synced 2026-10-29 08:13:44 +01:00
Decoder: renames to Codec and adds encoders detection support
This commit is contained in:
Carter Li
@@ -0,0 +1,38 @@
|
||||
#pragma once
|
||||
|
||||
#include "fastfetch.h"
|
||||
#include "modules/codec/option.h"
|
||||
|
||||
typedef enum FF_A_PACKED FFCodecType {
|
||||
FF_CODEC_TYPE_NONE = 0,
|
||||
FF_CODEC_TYPE_UNKNOWN = UINT32_C(1) << 0,
|
||||
FF_CODEC_TYPE_H261 = UINT32_C(1) << 1,
|
||||
FF_CODEC_TYPE_H263 = UINT32_C(1) << 2,
|
||||
FF_CODEC_TYPE_MJPEG = UINT32_C(1) << 3,
|
||||
FF_CODEC_TYPE_MPEG1 = UINT32_C(1) << 4,
|
||||
FF_CODEC_TYPE_MPEG2 = UINT32_C(1) << 5,
|
||||
FF_CODEC_TYPE_DIVX_XVID = UINT32_C(1) << 6,
|
||||
FF_CODEC_TYPE_H264 = UINT32_C(1) << 7,
|
||||
FF_CODEC_TYPE_WMV8 = UINT32_C(1) << 8,
|
||||
FF_CODEC_TYPE_WMV9 = UINT32_C(1) << 9,
|
||||
FF_CODEC_TYPE_VC1 = UINT32_C(1) << 10,
|
||||
FF_CODEC_TYPE_VP8 = UINT32_C(1) << 11,
|
||||
FF_CODEC_TYPE_HEVC = UINT32_C(1) << 12,
|
||||
FF_CODEC_TYPE_VP9 = UINT32_C(1) << 13,
|
||||
FF_CODEC_TYPE_AV1 = UINT32_C(1) << 14,
|
||||
FF_CODEC_TYPE_VVC = UINT32_C(1) << 15,
|
||||
FF_CODEC_TYPE_DOLBY_VISION_HEVC = UINT32_C(1) << 16,
|
||||
FF_CODEC_TYPE_PRORES = UINT32_C(1) << 17,
|
||||
FF_CODEC_TYPE_PRORES_RAW = UINT32_C(1) << 18,
|
||||
FF_CODEC_TYPE_JPEG_XL = UINT32_C(1) << 19,
|
||||
FF_CODEC_TYPE_MAX = FF_CODEC_TYPE_JPEG_XL,
|
||||
FF_CODEC_TYPE_FORCE_UNSIGNED = UINT32_MAX,
|
||||
} FFCodecType;
|
||||
|
||||
typedef struct FFCodecResult {
|
||||
FFstrbuf gpu;
|
||||
FFCodecType decoders;
|
||||
FFCodecType encoders;
|
||||
} FFCodecResult;
|
||||
|
||||
const char* ffDetectCodec(FFCodecOptions* options, FFlist* result /*list of FFCodecResult*/);
|
||||
@@ -0,0 +1,95 @@
|
||||
#include "codec.h"
|
||||
|
||||
#include <VideoToolbox/VideoToolbox.h>
|
||||
#include "common/apple/cf_helpers.h"
|
||||
|
||||
static const struct {
|
||||
CMVideoCodecType codec;
|
||||
FFCodecType type;
|
||||
} FF_CODEC_CODECS[] = {
|
||||
{ kCMVideoCodecType_H263, FF_CODEC_TYPE_H263 },
|
||||
{ kCMVideoCodecType_JPEG, FF_CODEC_TYPE_MJPEG },
|
||||
{ kCMVideoCodecType_JPEG_XL, FF_CODEC_TYPE_JPEG_XL },
|
||||
{ kCMVideoCodecType_MPEG1Video, FF_CODEC_TYPE_MPEG1 },
|
||||
{ kCMVideoCodecType_MPEG2Video, FF_CODEC_TYPE_MPEG2 },
|
||||
{ kCMVideoCodecType_MPEG4Video, FF_CODEC_TYPE_DIVX_XVID },
|
||||
{ kCMVideoCodecType_H264, FF_CODEC_TYPE_H264 },
|
||||
{ kCMVideoCodecType_HEVC, FF_CODEC_TYPE_HEVC },
|
||||
{ kCMVideoCodecType_HEVCWithAlpha, FF_CODEC_TYPE_HEVC },
|
||||
{ kCMVideoCodecType_DolbyVisionHEVC, FF_CODEC_TYPE_DOLBY_VISION_HEVC },
|
||||
{ kCMVideoCodecType_DisparityHEVC, FF_CODEC_TYPE_HEVC },
|
||||
{ kCMVideoCodecType_DepthHEVC, FF_CODEC_TYPE_HEVC },
|
||||
{ kCMVideoCodecType_VP9, FF_CODEC_TYPE_VP9 },
|
||||
{ kCMVideoCodecType_AV1, FF_CODEC_TYPE_AV1 },
|
||||
{ kCMVideoCodecType_AppleProRes4444XQ, FF_CODEC_TYPE_PRORES },
|
||||
{ kCMVideoCodecType_AppleProRes4444, FF_CODEC_TYPE_PRORES },
|
||||
{ kCMVideoCodecType_AppleProRes422HQ, FF_CODEC_TYPE_PRORES },
|
||||
{ kCMVideoCodecType_AppleProRes422, FF_CODEC_TYPE_PRORES },
|
||||
{ kCMVideoCodecType_AppleProRes422LT, FF_CODEC_TYPE_PRORES },
|
||||
{ kCMVideoCodecType_AppleProRes422Proxy, FF_CODEC_TYPE_PRORES },
|
||||
{ kCMVideoCodecType_AppleProResRAW, FF_CODEC_TYPE_PRORES_RAW },
|
||||
{ kCMVideoCodecType_AppleProResRAWHQ, FF_CODEC_TYPE_PRORES_RAW },
|
||||
};
|
||||
|
||||
static FFCodecType ffCodecCodecToType(CMVideoCodecType codec) {
|
||||
for (uint32_t i = 0; i < ARRAY_SIZE(FF_CODEC_CODECS); ++i) {
|
||||
if (FF_CODEC_CODECS[i].codec == codec) {
|
||||
return FF_CODEC_CODECS[i].type;
|
||||
}
|
||||
}
|
||||
return FF_CODEC_TYPE_NONE;
|
||||
}
|
||||
|
||||
static FFCodecType ffCodecDetectEncoders(void) {
|
||||
CFArrayRef encoderList = NULL;
|
||||
if (VTCopyVideoEncoderList(NULL, &encoderList) != noErr || !encoderList) {
|
||||
return FF_CODEC_TYPE_NONE;
|
||||
}
|
||||
|
||||
FFCodecType types = FF_CODEC_TYPE_NONE;
|
||||
for (uint32_t i = 0; i < CFArrayGetCount(encoderList); ++i) {
|
||||
CFDictionaryRef encoder = CFArrayGetValueAtIndex(encoderList, i);
|
||||
bool isHardwareAccelerated;
|
||||
int codec;
|
||||
if (ffCfDictGetBool(encoder, CFSTR("IsHardwareAccelerated"), &isHardwareAccelerated) != NULL ||
|
||||
!isHardwareAccelerated ||
|
||||
ffCfDictGetInt(encoder, CFSTR("CodecType"), &codec) != NULL) {
|
||||
continue;
|
||||
}
|
||||
types |= ffCodecCodecToType((CMVideoCodecType) codec);
|
||||
}
|
||||
|
||||
return types;
|
||||
}
|
||||
|
||||
static FFCodecType ffCodecDetectDecoders() {
|
||||
FFCodecType types = FF_CODEC_TYPE_NONE;
|
||||
for (uint32_t i = 0; i < ARRAY_SIZE(FF_CODEC_CODECS); ++i) {
|
||||
if (types & FF_CODEC_CODECS[i].type) {
|
||||
continue;
|
||||
}
|
||||
|
||||
bool supported = VTIsHardwareDecodeSupported(FF_CODEC_CODECS[i].codec);
|
||||
if (!supported) {
|
||||
continue;
|
||||
}
|
||||
|
||||
types |= FF_CODEC_CODECS[i].type;
|
||||
}
|
||||
|
||||
return types;
|
||||
}
|
||||
|
||||
const char* ffDetectCodec(FF_A_UNUSED FFCodecOptions* options, FFlist* result /* list of FFCodecResult */) {
|
||||
FFCodecType decoders = ffCodecDetectDecoders();
|
||||
FFCodecType encoders = ffCodecDetectEncoders();
|
||||
|
||||
if (decoders != FF_CODEC_TYPE_NONE || encoders != FF_CODEC_TYPE_NONE) {
|
||||
FFCodecResult* item = FF_LIST_ADD(FFCodecResult, *result);
|
||||
ffStrbufInitStatic(&item->gpu, "Default");
|
||||
item->decoders = decoders;
|
||||
item->encoders = encoders;
|
||||
}
|
||||
|
||||
return NULL;
|
||||
}
|
||||
@@ -0,0 +1,314 @@
|
||||
#include "codec.h"
|
||||
|
||||
#if FF_HAVE_DRM && FF_HAVE_VA
|
||||
|
||||
#include "detection/gpu/gpu.h"
|
||||
#include "common/io.h"
|
||||
#include "common/library.h"
|
||||
#include "common/mallocHelper.h"
|
||||
|
||||
#include <fcntl.h>
|
||||
#include <string.h>
|
||||
#include <va/va.h>
|
||||
#include <va/va_drm.h>
|
||||
#include <xf86drm.h>
|
||||
|
||||
static FFCodecType ffCodecProfileToType(VAProfile profile) {
|
||||
switch (profile) {
|
||||
case 11: // VAProfileH263Baseline
|
||||
return FF_CODEC_TYPE_H263;
|
||||
|
||||
case 12: // VAProfileJPEGBaseline
|
||||
return FF_CODEC_TYPE_MJPEG;
|
||||
|
||||
case 0: // VAProfileMPEG2Simple
|
||||
case 1: // VAProfileMPEG2Main
|
||||
return FF_CODEC_TYPE_MPEG2;
|
||||
|
||||
case 2: // VAProfileMPEG4Simple
|
||||
case 3: // VAProfileMPEG4AdvancedSimple
|
||||
case 4: // VAProfileMPEG4Main
|
||||
return FF_CODEC_TYPE_DIVX_XVID;
|
||||
|
||||
case 5: // VAProfileH264Baseline
|
||||
case 6: // VAProfileH264Main
|
||||
case 7: // VAProfileH264High
|
||||
case 13: // VAProfileH264ConstrainedBaseline
|
||||
case 15: // VAProfileH264MultiviewHigh
|
||||
case 16: // VAProfileH264StereoHigh
|
||||
case 36: // VAProfileH264High10
|
||||
case 40: // VAProfileH264High422
|
||||
return FF_CODEC_TYPE_H264;
|
||||
|
||||
case 8: // VAProfileVC1Simple
|
||||
case 9: // VAProfileVC1Main
|
||||
case 10: // VAProfileVC1Advanced
|
||||
return FF_CODEC_TYPE_VC1;
|
||||
|
||||
case 14: // VAProfileVP8Version0_3
|
||||
return FF_CODEC_TYPE_VP8;
|
||||
|
||||
case 17: // VAProfileHEVCMain
|
||||
case 18: // VAProfileHEVCMain10
|
||||
case 23: // VAProfileHEVCMain12
|
||||
case 24: // VAProfileHEVCMain422_10
|
||||
case 25: // VAProfileHEVCMain422_12
|
||||
case 26: // VAProfileHEVCMain444
|
||||
case 27: // VAProfileHEVCMain444_10
|
||||
case 28: // VAProfileHEVCMain444_12
|
||||
case 29: // VAProfileHEVCSccMain
|
||||
case 30: // VAProfileHEVCSccMain10
|
||||
case 31: // VAProfileHEVCSccMain444
|
||||
case 34: // VAProfileHEVCSccMain444_10
|
||||
return FF_CODEC_TYPE_HEVC;
|
||||
|
||||
case 19: // VAProfileVP9Profile0
|
||||
case 20: // VAProfileVP9Profile1
|
||||
case 21: // VAProfileVP9Profile2
|
||||
case 22: // VAProfileVP9Profile3
|
||||
return FF_CODEC_TYPE_VP9;
|
||||
|
||||
case 32: // VAProfileAV1Profile0
|
||||
case 33: // VAProfileAV1Profile1
|
||||
case 39: // VAProfileAV1Profile2
|
||||
return FF_CODEC_TYPE_AV1;
|
||||
|
||||
case 37: // VAProfileVVCMain10
|
||||
case 38: // VAProfileVVCMultilayerMain10
|
||||
return FF_CODEC_TYPE_VVC;
|
||||
|
||||
default:
|
||||
return FF_CODEC_TYPE_UNKNOWN;
|
||||
}
|
||||
}
|
||||
|
||||
static bool ffCodecEntrypointIsDecode(VAEntrypoint entrypoint) {
|
||||
switch (entrypoint) {
|
||||
case VAEntrypointVLD:
|
||||
case VAEntrypointIDCT:
|
||||
case VAEntrypointMoComp:
|
||||
return true;
|
||||
default:
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
static bool ffCodecEntrypointIsEncode(VAEntrypoint entrypoint) {
|
||||
switch (entrypoint) {
|
||||
case VAEntrypointEncSlice:
|
||||
case VAEntrypointEncPicture:
|
||||
case VAEntrypointEncSliceLP:
|
||||
case VAEntrypointEncFEI:
|
||||
return true;
|
||||
default:
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
static bool ffCodecProfileHasOutput(
|
||||
VADisplay display,
|
||||
__typeof__(vaQueryConfigEntrypoints)* ffvaQueryConfigEntrypoints,
|
||||
__typeof__(vaGetConfigAttributes)* ffvaGetConfigAttributes,
|
||||
int maxEntrypoints,
|
||||
VAEntrypoint* entrypoints,
|
||||
VAProfile profile,
|
||||
bool encode
|
||||
) {
|
||||
int numEntrypoints = maxEntrypoints;
|
||||
if (ffvaQueryConfigEntrypoints(display, profile, entrypoints, &numEntrypoints) != VA_STATUS_SUCCESS) {
|
||||
return false;
|
||||
}
|
||||
|
||||
for (int i = 0; i < numEntrypoints; ++i) {
|
||||
if (encode ? !ffCodecEntrypointIsEncode(entrypoints[i]) : !ffCodecEntrypointIsDecode(entrypoints[i])) {
|
||||
continue;
|
||||
}
|
||||
|
||||
VAConfigAttrib attrib = {
|
||||
.type = VAConfigAttribRTFormat,
|
||||
.value = 0,
|
||||
};
|
||||
|
||||
if (ffvaGetConfigAttributes(display, profile, entrypoints[i], &attrib, 1) == VA_STATUS_SUCCESS &&
|
||||
attrib.value != VA_ATTRIB_NOT_SUPPORTED &&
|
||||
attrib.value != 0) {
|
||||
return true;
|
||||
}
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
static void ffCodecFillGpuName(const drmDevice* dev, const char* path, FFstrbuf* name) {
|
||||
ffStrbufInit(name);
|
||||
|
||||
switch (dev->bustype) {
|
||||
case DRM_BUS_PCI: {
|
||||
FFGPUResult gpu = {
|
||||
.vendor = ffStrbufCreateStatic(ffGPUGetVendorString(dev->deviceinfo.pci->vendor_id)),
|
||||
.name = ffStrbufCreate(),
|
||||
};
|
||||
|
||||
ffGPUFillVendorAndName(0, dev->deviceinfo.pci->vendor_id, dev->deviceinfo.pci->device_id, &gpu);
|
||||
ffStrbufSetF(name, "%s %s", gpu.vendor.chars, gpu.name.chars);
|
||||
|
||||
ffStrbufDestroy(&gpu.vendor);
|
||||
ffStrbufDestroy(&gpu.name);
|
||||
break;
|
||||
}
|
||||
case DRM_BUS_PLATFORM:
|
||||
ffStrbufSetS(name, dev->deviceinfo.platform->compatible[0]);
|
||||
return;
|
||||
case DRM_BUS_HOST1X:
|
||||
ffStrbufSetS(name, dev->deviceinfo.host1x->compatible[0]);
|
||||
return;
|
||||
case DRM_BUS_USB:
|
||||
ffStrbufSetF(name, "0x%04X 0x%04X", dev->deviceinfo.usb->vendor, dev->deviceinfo.usb->product);
|
||||
return;
|
||||
default:
|
||||
ffStrbufSetStatic(name, "Unknown GPU");
|
||||
return;
|
||||
}
|
||||
|
||||
if (!name->length && path && *path) {
|
||||
const char* base = strrchr(path, '/');
|
||||
ffStrbufSetS(name, base ? base + 1 : path);
|
||||
}
|
||||
|
||||
if (!name->length) {
|
||||
ffStrbufSetStatic(name, "Unknown GPU");
|
||||
}
|
||||
}
|
||||
|
||||
const char* ffDetectCodec(FF_A_UNUSED FFCodecOptions* options, FFlist* result /* list of FFCodecResult */) {
|
||||
FF_LIBRARY_LOAD_MESSAGE(libdrm, "libdrm" FF_LIBRARY_EXTENSION, 2)
|
||||
FF_LIBRARY_LOAD_SYMBOL_MESSAGE(libdrm, drmGetDevices)
|
||||
FF_LIBRARY_LOAD_SYMBOL_MESSAGE(libdrm, drmFreeDevices)
|
||||
|
||||
FF_LIBRARY_LOAD_MESSAGE(libva, "libva" FF_LIBRARY_EXTENSION, 2)
|
||||
FF_LIBRARY_LOAD_SYMBOL_MESSAGE(libva, vaInitialize)
|
||||
FF_LIBRARY_LOAD_SYMBOL_MESSAGE(libva, vaTerminate)
|
||||
FF_LIBRARY_LOAD_SYMBOL_MESSAGE(libva, vaMaxNumProfiles)
|
||||
FF_LIBRARY_LOAD_SYMBOL_MESSAGE(libva, vaMaxNumEntrypoints)
|
||||
FF_LIBRARY_LOAD_SYMBOL_MESSAGE(libva, vaQueryConfigProfiles)
|
||||
FF_LIBRARY_LOAD_SYMBOL_MESSAGE(libva, vaQueryConfigEntrypoints)
|
||||
FF_LIBRARY_LOAD_SYMBOL_MESSAGE(libva, vaGetConfigAttributes)
|
||||
|
||||
FF_LIBRARY_LOAD_MESSAGE(libvaDrm, "libva-drm" FF_LIBRARY_EXTENSION, 2)
|
||||
FF_LIBRARY_LOAD_SYMBOL_MESSAGE(libvaDrm, vaGetDisplayDRM)
|
||||
|
||||
drmDevicePtr devices[64];
|
||||
int numDevices = ffdrmGetDevices(devices, ARRAY_SIZE(devices));
|
||||
if (numDevices < 0) {
|
||||
return "drmGetDevices() failed";
|
||||
}
|
||||
if (numDevices == 0) {
|
||||
return NULL;
|
||||
}
|
||||
|
||||
for (int i = 0; i < numDevices; ++i) {
|
||||
drmDevice* dev = devices[i];
|
||||
|
||||
const char* path = NULL;
|
||||
if (dev->available_nodes & (1 << DRM_NODE_RENDER)) {
|
||||
path = dev->nodes[DRM_NODE_RENDER];
|
||||
} else if (dev->available_nodes & (1 << DRM_NODE_PRIMARY)) {
|
||||
path = dev->nodes[DRM_NODE_PRIMARY];
|
||||
} else {
|
||||
continue;
|
||||
}
|
||||
|
||||
FF_AUTO_CLOSE_FD int fd = open(path, O_RDWR | O_CLOEXEC);
|
||||
if (fd < 0) {
|
||||
fd = open(path, O_RDONLY | O_CLOEXEC);
|
||||
}
|
||||
if (fd < 0) {
|
||||
continue;
|
||||
}
|
||||
|
||||
VADisplay display = ffvaGetDisplayDRM(fd);
|
||||
if (!display) {
|
||||
continue;
|
||||
}
|
||||
|
||||
int major = 0, minor = 0;
|
||||
if (ffvaInitialize(display, &major, &minor) != VA_STATUS_SUCCESS) {
|
||||
continue;
|
||||
}
|
||||
|
||||
int maxProfiles = ffvaMaxNumProfiles(display);
|
||||
int maxEntrypoints = ffvaMaxNumEntrypoints(display);
|
||||
if (maxProfiles <= 0 || maxEntrypoints <= 0) {
|
||||
ffvaTerminate(display);
|
||||
continue;
|
||||
}
|
||||
|
||||
FF_AUTO_FREE VAProfile* profiles = (VAProfile*) malloc(sizeof(VAProfile) * (size_t) maxProfiles);
|
||||
FF_AUTO_FREE VAEntrypoint* entrypoints = (VAEntrypoint*) malloc(sizeof(VAEntrypoint) * (size_t) maxEntrypoints);
|
||||
if (!profiles || !entrypoints) {
|
||||
ffvaTerminate(display);
|
||||
return "malloc() failed";
|
||||
}
|
||||
|
||||
int numProfiles = maxProfiles;
|
||||
if (ffvaQueryConfigProfiles(display, profiles, &numProfiles) != VA_STATUS_SUCCESS) {
|
||||
ffvaTerminate(display);
|
||||
continue;
|
||||
}
|
||||
|
||||
FFCodecType decoderTypes = FF_CODEC_TYPE_NONE;
|
||||
FFCodecType encoderTypes = FF_CODEC_TYPE_NONE;
|
||||
for (int j = 0; j < numProfiles; ++j) {
|
||||
bool hasDecoder = ffCodecProfileHasOutput(
|
||||
display,
|
||||
ffvaQueryConfigEntrypoints,
|
||||
ffvaGetConfigAttributes,
|
||||
maxEntrypoints,
|
||||
entrypoints,
|
||||
profiles[j],
|
||||
false);
|
||||
|
||||
bool hasEncoder = ffCodecProfileHasOutput(
|
||||
display,
|
||||
ffvaQueryConfigEntrypoints,
|
||||
ffvaGetConfigAttributes,
|
||||
maxEntrypoints,
|
||||
entrypoints,
|
||||
profiles[j],
|
||||
true);
|
||||
|
||||
if (!hasDecoder && !hasEncoder) {
|
||||
continue;
|
||||
}
|
||||
|
||||
FFCodecType type = ffCodecProfileToType(profiles[j]);
|
||||
if (hasDecoder) {
|
||||
decoderTypes = (FFCodecType) ((uint32_t) decoderTypes | (uint32_t) type);
|
||||
}
|
||||
if (hasEncoder) {
|
||||
encoderTypes = (FFCodecType) ((uint32_t) encoderTypes | (uint32_t) type);
|
||||
}
|
||||
}
|
||||
|
||||
if (decoderTypes != FF_CODEC_TYPE_NONE || encoderTypes != FF_CODEC_TYPE_NONE) {
|
||||
FFCodecResult* item = FF_LIST_ADD(FFCodecResult, *result);
|
||||
ffCodecFillGpuName(dev, path, &item->gpu);
|
||||
item->decoders = decoderTypes;
|
||||
item->encoders = encoderTypes;
|
||||
}
|
||||
|
||||
ffvaTerminate(display);
|
||||
}
|
||||
|
||||
ffdrmFreeDevices(devices, numDevices);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
#else
|
||||
|
||||
const char* ffDetectCodec(FFCodecOptions* options, FFlist* result /* list of FFCodecResult */) {
|
||||
FF_UNUSED(options, result);
|
||||
return "Fastfetch was built without DRM / VA-API headers";
|
||||
}
|
||||
|
||||
#endif
|
||||
@@ -0,0 +1,223 @@
|
||||
extern "C" {
|
||||
#include "codec.h"
|
||||
#include "common/library.h"
|
||||
#include "common/windows/com.h"
|
||||
#include "common/windows/unicode.h"
|
||||
}
|
||||
|
||||
#include <d3d11.h>
|
||||
#include <dxgi.h>
|
||||
#include <initguid.h>
|
||||
#include <dxva.h>
|
||||
|
||||
static const DXGI_FORMAT FF_NATIVE_CODEC_FORMATS[] = {
|
||||
DXGI_FORMAT_420_OPAQUE,
|
||||
DXGI_FORMAT_NV12,
|
||||
DXGI_FORMAT_P010,
|
||||
DXGI_FORMAT_P016,
|
||||
DXGI_FORMAT_YUY2,
|
||||
DXGI_FORMAT_Y210,
|
||||
DXGI_FORMAT_Y216,
|
||||
DXGI_FORMAT_AYUV,
|
||||
DXGI_FORMAT_Y410,
|
||||
DXGI_FORMAT_Y416,
|
||||
};
|
||||
|
||||
static FFCodecType ffCodecProfileToType(const GUID& profile) {
|
||||
// clang-format off
|
||||
if (IsEqualGUID(profile, DXVA_ModeH261_A) ||
|
||||
IsEqualGUID(profile, DXVA_ModeH261_B)) {
|
||||
return FF_CODEC_TYPE_H261;
|
||||
}
|
||||
|
||||
if (IsEqualGUID(profile, DXVA_ModeH263_A) ||
|
||||
IsEqualGUID(profile, DXVA_ModeH263_B) ||
|
||||
IsEqualGUID(profile, DXVA_ModeH263_C) ||
|
||||
IsEqualGUID(profile, DXVA_ModeH263_D) ||
|
||||
IsEqualGUID(profile, DXVA_ModeH263_E) ||
|
||||
IsEqualGUID(profile, DXVA_ModeH263_F)) {
|
||||
return FF_CODEC_TYPE_H263;
|
||||
}
|
||||
|
||||
if (IsEqualGUID(profile, DXVA_ModeMJPEG_VLD_420) ||
|
||||
IsEqualGUID(profile, DXVA_ModeMJPEG_VLD_422) ||
|
||||
IsEqualGUID(profile, DXVA_ModeMJPEG_VLD_444) ||
|
||||
IsEqualGUID(profile, DXVA_ModeMJPEG_VLD_4444) ||
|
||||
IsEqualGUID(profile, DXVA_ModeJPEG_VLD_420) ||
|
||||
IsEqualGUID(profile, DXVA_ModeJPEG_VLD_422) ||
|
||||
IsEqualGUID(profile, DXVA_ModeJPEG_VLD_444)) {
|
||||
return FF_CODEC_TYPE_MJPEG;
|
||||
}
|
||||
|
||||
if (IsEqualGUID(profile, DXVA_ModeMPEG1_A) ||
|
||||
IsEqualGUID(profile, DXVA_ModeMPEG1_VLD)) {
|
||||
return FF_CODEC_TYPE_MPEG1;
|
||||
}
|
||||
|
||||
if (IsEqualGUID(profile, DXVA_ModeMPEG2_A) ||
|
||||
IsEqualGUID(profile, DXVA_ModeMPEG2_B) ||
|
||||
IsEqualGUID(profile, DXVA_ModeMPEG2_C) ||
|
||||
IsEqualGUID(profile, DXVA_ModeMPEG2_D) ||
|
||||
IsEqualGUID(profile, DXVA_ModeMPEG2and1_VLD)) {
|
||||
return FF_CODEC_TYPE_MPEG2;
|
||||
}
|
||||
|
||||
if (IsEqualGUID(profile, DXVA_ModeMPEG4pt2_VLD_Simple) ||
|
||||
IsEqualGUID(profile, DXVA_ModeMPEG4pt2_VLD_AdvSimple_NoGMC) ||
|
||||
IsEqualGUID(profile, DXVA_ModeMPEG4pt2_VLD_AdvSimple_GMC)) {
|
||||
return FF_CODEC_TYPE_DIVX_XVID;
|
||||
}
|
||||
|
||||
if (IsEqualGUID(profile, DXVA_ModeH264_E) ||
|
||||
IsEqualGUID(profile, DXVA_ModeH264_F) ||
|
||||
IsEqualGUID(profile, DXVA_ModeH264_VLD_WithFMOASO_NoFGT) ||
|
||||
IsEqualGUID(profile, DXVA_ModeH264_VLD_Stereo_Progressive_NoFGT) ||
|
||||
IsEqualGUID(profile, DXVA_ModeH264_VLD_Stereo_NoFGT) ||
|
||||
IsEqualGUID(profile, DXVA_ModeH264_VLD_Multiview_NoFGT)) {
|
||||
return FF_CODEC_TYPE_H264;
|
||||
}
|
||||
|
||||
if (IsEqualGUID(profile, DXVA_ModeWMV8_A) ||
|
||||
IsEqualGUID(profile, DXVA_ModeWMV8_B)) {
|
||||
return FF_CODEC_TYPE_WMV8;
|
||||
}
|
||||
|
||||
if (IsEqualGUID(profile, DXVA_ModeWMV9_A) ||
|
||||
IsEqualGUID(profile, DXVA_ModeWMV9_B) ||
|
||||
IsEqualGUID(profile, DXVA_ModeWMV9_C)) {
|
||||
return FF_CODEC_TYPE_WMV9;
|
||||
}
|
||||
|
||||
if (IsEqualGUID(profile, DXVA_ModeVC1_A) ||
|
||||
IsEqualGUID(profile, DXVA_ModeVC1_B) ||
|
||||
IsEqualGUID(profile, DXVA_ModeVC1_C) ||
|
||||
IsEqualGUID(profile, DXVA_ModeVC1_D) ||
|
||||
IsEqualGUID(profile, DXVA_ModeVC1_D2010)) {
|
||||
return FF_CODEC_TYPE_VC1;
|
||||
}
|
||||
|
||||
if (IsEqualGUID(profile, DXVA_ModeVP8_VLD)) {
|
||||
return FF_CODEC_TYPE_VP8;
|
||||
}
|
||||
|
||||
if (IsEqualGUID(profile, DXVA_ModeHEVC_VLD_Main) ||
|
||||
IsEqualGUID(profile, DXVA_ModeHEVC_VLD_Main10) ||
|
||||
IsEqualGUID(profile, DXVA_ModeHEVC_VLD_Main12) ||
|
||||
IsEqualGUID(profile, DXVA_ModeHEVC_VLD_Main10_422) ||
|
||||
IsEqualGUID(profile, DXVA_ModeHEVC_VLD_Main12_422) ||
|
||||
IsEqualGUID(profile, DXVA_ModeHEVC_VLD_Main_444) ||
|
||||
IsEqualGUID(profile, DXVA_ModeHEVC_VLD_Main10_Ext) ||
|
||||
IsEqualGUID(profile, DXVA_ModeHEVC_VLD_Main10_444) ||
|
||||
IsEqualGUID(profile, DXVA_ModeHEVC_VLD_Main12_444) ||
|
||||
IsEqualGUID(profile, DXVA_ModeHEVC_VLD_Main16)) {
|
||||
return FF_CODEC_TYPE_HEVC;
|
||||
}
|
||||
|
||||
if (IsEqualGUID(profile, DXVA_ModeVP9_VLD_Profile0) ||
|
||||
IsEqualGUID(profile, DXVA_ModeVP9_VLD_10bit_Profile2)) {
|
||||
return FF_CODEC_TYPE_VP9;
|
||||
}
|
||||
|
||||
if (IsEqualGUID(profile, DXVA_ModeAV1_VLD_Profile0) ||
|
||||
IsEqualGUID(profile, DXVA_ModeAV1_VLD_Profile1) ||
|
||||
IsEqualGUID(profile, DXVA_ModeAV1_VLD_Profile2) ||
|
||||
IsEqualGUID(profile, DXVA_ModeAV1_VLD_12bit_Profile2) ||
|
||||
IsEqualGUID(profile, DXVA_ModeAV1_VLD_12bit_Profile2_420)) {
|
||||
return FF_CODEC_TYPE_AV1;
|
||||
}
|
||||
// clang-format on
|
||||
|
||||
return FF_CODEC_TYPE_UNKNOWN;
|
||||
}
|
||||
|
||||
static bool ffCodecProfileHasNativeOutput(ID3D11VideoDevice* videoDevice, const GUID& profile) {
|
||||
for (uint32_t i = 0; i < ARRAY_SIZE(FF_NATIVE_CODEC_FORMATS); ++i) {
|
||||
BOOL supported = FALSE;
|
||||
if (SUCCEEDED(videoDevice->CheckVideoDecoderFormat(&profile, FF_NATIVE_CODEC_FORMATS[i], &supported)) && supported) {
|
||||
return true;
|
||||
}
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
const char* ffDetectCodec(FF_A_UNUSED FFCodecOptions* options, FFlist* result /*list of FFCodecResult*/) {
|
||||
FF_LIBRARY_LOAD_MESSAGE(dxgi, "dxgi" FF_LIBRARY_EXTENSION, 1)
|
||||
FF_LIBRARY_LOAD_SYMBOL_MESSAGE(dxgi, CreateDXGIFactory1)
|
||||
FF_LIBRARY_LOAD_MESSAGE(d3d11, "d3d11" FF_LIBRARY_EXTENSION, 1)
|
||||
FF_LIBRARY_LOAD_SYMBOL_MESSAGE(d3d11, D3D11CreateDevice)
|
||||
|
||||
const char* error = ffInitCom();
|
||||
if (error) {
|
||||
return error;
|
||||
}
|
||||
|
||||
IDXGIFactory1* FF_AUTO_RELEASE_COM_OBJECT factory = nullptr;
|
||||
if (FAILED(ffCreateDXGIFactory1(__uuidof(IDXGIFactory1), (void**) &factory)) || !factory) {
|
||||
return "CreateDXGIFactory1() failed";
|
||||
}
|
||||
|
||||
for (UINT adapterIndex = 0;; ++adapterIndex) {
|
||||
IDXGIAdapter1* FF_AUTO_RELEASE_COM_OBJECT adapter = nullptr;
|
||||
HRESULT adapterStatus = factory->EnumAdapters1(adapterIndex, &adapter);
|
||||
if (adapterStatus == DXGI_ERROR_NOT_FOUND) {
|
||||
break;
|
||||
}
|
||||
if (FAILED(adapterStatus) || !adapter) {
|
||||
continue;
|
||||
}
|
||||
|
||||
DXGI_ADAPTER_DESC1 desc;
|
||||
if (FAILED(adapter->GetDesc1(&desc))) {
|
||||
continue;
|
||||
}
|
||||
if (desc.Flags & DXGI_ADAPTER_FLAG_SOFTWARE) {
|
||||
continue;
|
||||
}
|
||||
|
||||
ID3D11Device* FF_AUTO_RELEASE_COM_OBJECT d3dDevice = nullptr;
|
||||
D3D_FEATURE_LEVEL featureLevel;
|
||||
HRESULT deviceStatus = ffD3D11CreateDevice(
|
||||
adapter,
|
||||
D3D_DRIVER_TYPE_UNKNOWN,
|
||||
nullptr,
|
||||
D3D11_CREATE_DEVICE_VIDEO_SUPPORT,
|
||||
nullptr,
|
||||
0,
|
||||
D3D11_SDK_VERSION,
|
||||
&d3dDevice,
|
||||
&featureLevel,
|
||||
nullptr);
|
||||
if (FAILED(deviceStatus) || !d3dDevice) {
|
||||
continue;
|
||||
}
|
||||
|
||||
ID3D11VideoDevice* FF_AUTO_RELEASE_COM_OBJECT videoDevice = nullptr;
|
||||
if (FAILED(d3dDevice->QueryInterface(__uuidof(ID3D11VideoDevice), (void**) &videoDevice)) || !videoDevice) {
|
||||
continue;
|
||||
}
|
||||
|
||||
FFCodecResult* gpuResult = nullptr;
|
||||
UINT profileCount = videoDevice->GetVideoDecoderProfileCount();
|
||||
for (UINT profileIndex = 0; profileIndex < profileCount; ++profileIndex) {
|
||||
GUID profile;
|
||||
if (FAILED(videoDevice->GetVideoDecoderProfile(profileIndex, &profile))) {
|
||||
continue;
|
||||
}
|
||||
|
||||
if (!ffCodecProfileHasNativeOutput(videoDevice, profile)) {
|
||||
continue;
|
||||
}
|
||||
|
||||
if (!gpuResult) {
|
||||
gpuResult = FF_LIST_ADD(FFCodecResult, *result);
|
||||
ffStrbufInitWS(&gpuResult->gpu, desc.Description);
|
||||
gpuResult->decoders = FF_CODEC_TYPE_NONE;
|
||||
gpuResult->encoders = FF_CODEC_TYPE_NONE;
|
||||
}
|
||||
gpuResult->decoders = (FFCodecType) (((uint32_t) gpuResult->decoders) | ((uint32_t) ffCodecProfileToType(profile)));
|
||||
}
|
||||
}
|
||||
|
||||
return nullptr;
|
||||
}
|
||||
Reference in New Issue
Block a user