diff --git a/CMakeLists.txt b/CMakeLists.txt index a0acece85..a489ad468 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -470,6 +470,7 @@ set(LIBFASTFETCH_SRC src/modules/custom/custom.c src/modules/command/command.c src/modules/datetime/datetime.c + src/modules/decoder/decoder.c src/modules/de/de.c src/modules/disk/disk.c src/modules/diskio/diskio.c diff --git a/src/detection/decoder/decoder.h b/src/detection/decoder/decoder.h new file mode 100644 index 000000000..bd62077ca --- /dev/null +++ b/src/detection/decoder/decoder.h @@ -0,0 +1,32 @@ +#pragma once + +#include "fastfetch.h" +#include "modules/decoder/option.h" + +typedef enum FF_A_PACKED FFDecoderType { + FF_DECODER_TYPE_NONE = 0, + FF_DECODER_TYPE_UNKNOWN = UINT32_C(1) << 0, + FF_DECODER_TYPE_H261 = UINT32_C(1) << 1, + FF_DECODER_TYPE_H263 = UINT32_C(1) << 2, + FF_DECODER_TYPE_MJPEG = UINT32_C(1) << 3, + FF_DECODER_TYPE_MPEG1 = UINT32_C(1) << 4, + FF_DECODER_TYPE_MPEG2 = UINT32_C(1) << 5, + FF_DECODER_TYPE_DIVX_XVID = UINT32_C(1) << 6, + FF_DECODER_TYPE_H264 = UINT32_C(1) << 7, + FF_DECODER_TYPE_WMV8 = UINT32_C(1) << 8, + FF_DECODER_TYPE_WMV9 = UINT32_C(1) << 9, + FF_DECODER_TYPE_VC1 = UINT32_C(1) << 10, + FF_DECODER_TYPE_VP8 = UINT32_C(1) << 11, + FF_DECODER_TYPE_HEVC = UINT32_C(1) << 12, + FF_DECODER_TYPE_VP9 = UINT32_C(1) << 13, + FF_DECODER_TYPE_AV1 = UINT32_C(1) << 14, + FF_DECODER_TYPE_MAX = FF_DECODER_TYPE_AV1, + FF_DECODER_TYPE_FORCE_UNSIGNED = UINT32_MAX, +} FFDecoderType; + +typedef struct FFDecoderResult { + FFstrbuf gpu; + FFDecoderType types; +} FFDecoderResult; + +const char* ffDetectDecoder(FFDecoderOptions* options, FFlist* result /*list of FFDecoderResult*/); diff --git a/src/modules/decoder/decoder.c b/src/modules/decoder/decoder.c new file mode 100644 index 000000000..57a8d213e --- /dev/null +++ b/src/modules/decoder/decoder.c @@ -0,0 +1,184 @@ +#include "common/printing.h" +#include "common/jsonconfig.h" +#include "detection/decoder/decoder.h" +#include "modules/decoder/decoder.h" + +static const char* ffDecoderTypeToString(FFDecoderType type) { + switch (type) { + case FF_DECODER_TYPE_H261: + return "H.261"; + case FF_DECODER_TYPE_H263: + return "H.263"; + case FF_DECODER_TYPE_MJPEG: + return "MJPEG"; + case FF_DECODER_TYPE_MPEG1: + return "MPEG-1"; + case FF_DECODER_TYPE_MPEG2: + return "MPEG-2"; + case FF_DECODER_TYPE_DIVX_XVID: + return "DivX / Xvid"; + case FF_DECODER_TYPE_H264: + return "H.264"; + case FF_DECODER_TYPE_WMV8: + return "WMV-8"; + case FF_DECODER_TYPE_WMV9: + return "WMV-9"; + case FF_DECODER_TYPE_VC1: + return "VC-1"; + case FF_DECODER_TYPE_VP8: + return "VP8"; + case FF_DECODER_TYPE_HEVC: + return "HEVC / H.265"; + case FF_DECODER_TYPE_VP9: + return "VP9"; + case FF_DECODER_TYPE_AV1: + return "AV1"; + case FF_DECODER_TYPE_UNKNOWN: + default: + return "Unknown"; + } +} + +static void ffDestroyDecoderResults(FFlist* result) { + FF_LIST_FOR_EACH (FFDecoderResult, item, *result) { + ffStrbufDestroy(&item->gpu); + } +} + +static void printDecoder(const FFDecoderOptions* options, const FFDecoderResult* decoder, uint8_t index) { + FF_STRBUF_AUTO_DESTROY key = ffStrbufCreate(); + + if (options->moduleArgs.key.length == 0) { + ffStrbufSetF(&key, "%s (%s)", FF_DECODER_MODULE_NAME, decoder->gpu.chars); + } else { + FF_PARSE_FORMAT_STRING_CHECKED(&key, &options->moduleArgs.key, ((FFformatarg[]){ + FF_ARG(decoder->gpu, "gpu"), + FF_ARG(options->moduleArgs.keyIcon, "icon"), + FF_ARG(index, "index"), + })); + } + + if (options->moduleArgs.outputFormat.length == 0) { + FF_STRBUF_AUTO_DESTROY typesJoined = ffStrbufCreate(); + for (FFDecoderType type = FF_DECODER_TYPE_UNKNOWN; type <= FF_DECODER_TYPE_MAX; type <<= 1) { + if ((decoder->types & type) == 0) { + continue; + } + if (typesJoined.length > 0) { + ffStrbufAppendS(&typesJoined, ", "); + } + ffStrbufAppendS(&typesJoined, ffDecoderTypeToString(type)); + } + + ffPrintLogoAndKey(key.chars, 0, &options->moduleArgs, FF_PRINT_TYPE_NO_CUSTOM_KEY); + + if (typesJoined.length > 0) { + ffStrbufWriteTo(&typesJoined, stdout); + } + putchar('\n'); + } else { + FF_LIST_AUTO_DESTROY types = ffListCreate(); // Use list instead of pre-joined string for qjs and lua + for (FFDecoderType type = FF_DECODER_TYPE_UNKNOWN; type <= FF_DECODER_TYPE_MAX; type <<= 1) { + if ((decoder->types & type) == 0) { + continue; + } + ffStrbufInitStatic(FF_LIST_ADD(FFstrbuf, types), ffDecoderTypeToString(type)); + } + + FF_PRINT_FORMAT_CHECKED(key.chars, 0, &options->moduleArgs, FF_PRINT_TYPE_NO_CUSTOM_KEY, ((FFformatarg[]){ + FF_ARG(decoder->gpu, "gpu"), + FF_ARG(types, "types"), + })); + // No need to destroy strings in types list, as they are static strings + } +} + +bool ffPrintDecoder(FFDecoderOptions* options) { + FF_LIST_AUTO_DESTROY result = ffListCreate(); + const char* error = ffDetectDecoder(NULL, &result); + + if (error) { + ffPrintError(FF_DECODER_MODULE_NAME, 0, &options->moduleArgs, FF_PRINT_TYPE_DEFAULT, "%s", error); + return false; + } + + if (result.length == 0) { + ffPrintError(FF_DECODER_MODULE_NAME, 0, &options->moduleArgs, FF_PRINT_TYPE_DEFAULT, "No hardware decoder found"); + return false; + } + + for (uint32_t i = 0; i < result.length; ++i) { + uint8_t index = (uint8_t) (result.length == 1 ? 0 : i + 1); + printDecoder(options, FF_LIST_GET(FFDecoderResult, result, i), index); + } + + ffDestroyDecoderResults(&result); + return true; +} + +void ffParseDecoderJsonObject(FFDecoderOptions* options, yyjson_val* module) { + yyjson_val *key, *val; + size_t idx, max; + yyjson_obj_foreach (module, idx, max, key, val) { + if (ffJsonConfigParseModuleArgs(key, val, &options->moduleArgs)) { + continue; + } + + ffPrintError(FF_DECODER_MODULE_NAME, 0, &options->moduleArgs, FF_PRINT_TYPE_DEFAULT, "Unknown JSON key %s", unsafe_yyjson_get_str(key)); + } +} + +void ffGenerateDecoderJsonConfig(FFDecoderOptions* options, yyjson_mut_doc* doc, yyjson_mut_val* module) { + ffJsonConfigGenerateModuleArgsConfig(doc, module, &options->moduleArgs); +} + +bool ffGenerateDecoderJsonResult(FF_A_UNUSED FFDecoderOptions* options, yyjson_mut_doc* doc, yyjson_mut_val* module) { + FF_LIST_AUTO_DESTROY result = ffListCreate(); + const char* error = ffDetectDecoder(NULL, &result); + + if (error) { + yyjson_mut_obj_add_str(doc, module, "error", error); + return false; + } + + yyjson_mut_val* arr = yyjson_mut_obj_add_arr(doc, module, "result"); + + FF_LIST_FOR_EACH (FFDecoderResult, item, result) { + yyjson_mut_val* obj = yyjson_mut_arr_add_obj(doc, arr); + yyjson_mut_obj_add_strbuf(doc, obj, "gpu", &item->gpu); + + yyjson_mut_val* types = yyjson_mut_obj_add_arr(doc, obj, "types"); + for (FFDecoderType type = FF_DECODER_TYPE_UNKNOWN; type <= FF_DECODER_TYPE_MAX; type <<= 1) { + if ((item->types & type) == 0) { + continue; + } + yyjson_mut_arr_add_str(doc, types, ffDecoderTypeToString(type)); + } + } + + ffDestroyDecoderResults(&result); + return true; +} + +void ffInitDecoderOptions(FFDecoderOptions* options) { + ffOptionInitModuleArg(&options->moduleArgs, "󰚔"); +} + +void ffDestroyDecoderOptions(FFDecoderOptions* options) { + ffOptionDestroyModuleArg(&options->moduleArgs); +} + +FFModuleBaseInfo ffDecoderModuleInfo = { + .name = FF_DECODER_MODULE_NAME, + .description = "Print hardware decoder types grouped by GPU", + .initOptions = (void*) ffInitDecoderOptions, + .destroyOptions = (void*) ffDestroyDecoderOptions, + .parseJsonObject = (void*) ffParseDecoderJsonObject, + .printModule = (void*) ffPrintDecoder, + .generateJsonResult = (void*) ffGenerateDecoderJsonResult, + .generateJsonConfig = (void*) ffGenerateDecoderJsonConfig, + .formatArgs = FF_FORMAT_ARG_LIST(((FFModuleFormatArg[]){ + { "GPU name", "gpu" }, + { "Decoder types, comma-separated", "types" }, + })) +}; diff --git a/src/modules/decoder/decoder.h b/src/modules/decoder/decoder.h new file mode 100644 index 000000000..b21177860 --- /dev/null +++ b/src/modules/decoder/decoder.h @@ -0,0 +1,11 @@ +#pragma once + +#include "option.h" + +#define FF_DECODER_MODULE_NAME "Decoder" + +bool ffPrintDecoder(FFDecoderOptions* options); +void ffInitDecoderOptions(FFDecoderOptions* options); +void ffDestroyDecoderOptions(FFDecoderOptions* options); + +extern FFModuleBaseInfo ffDecoderModuleInfo; diff --git a/src/modules/decoder/option.h b/src/modules/decoder/option.h new file mode 100644 index 000000000..137d86c61 --- /dev/null +++ b/src/modules/decoder/option.h @@ -0,0 +1,9 @@ +#pragma once + +#include "common/option.h" + +typedef struct FFDecoderOptions { + FFModuleArgs moduleArgs; +} FFDecoderOptions; + +static_assert(sizeof(FFDecoderOptions) <= FF_OPTION_MAX_SIZE, "FFDecoderOptions size exceeds maximum allowed size"); diff --git a/src/modules/modules.c b/src/modules/modules.c index 2139ec7a3..436d61df0 100644 --- a/src/modules/modules.c +++ b/src/modules/modules.c @@ -78,6 +78,9 @@ static FFModuleBaseInfo* D[] = { #if !FF_MODULE_DISABLE_DE &ffDEModuleInfo, #endif +#if !FF_MODULE_DISABLE_DECODER + &ffDecoderModuleInfo, +#endif #if !FF_MODULE_DISABLE_DISPLAY &ffDisplayModuleInfo, #endif diff --git a/src/modules/modules.h b/src/modules/modules.h index 2579ad90e..52c9f4a67 100644 --- a/src/modules/modules.h +++ b/src/modules/modules.h @@ -19,6 +19,7 @@ #include "modules/cursor/cursor.h" #include "modules/custom/custom.h" #include "modules/datetime/datetime.h" +#include "modules/decoder/decoder.h" #include "modules/disk/disk.h" #include "modules/diskio/diskio.h" #include "modules/display/display.h"