Adds functions to convert to/from strings and a function to iterate over all supported device types. Also adds a new invalid type AV_HWDEVICE_TYPE_NONE, which acts as a sentinel value. --- libavutil/hwcontext.c | 46 ++++++++++++++++++++++++++++++++++++++++++++++ libavutil/hwcontext.h | 28 ++++++++++++++++++++++++++++ 2 files changed, 74 insertions(+)
diff --git a/libavutil/hwcontext.c b/libavutil/hwcontext.c index 252dd32e3..8195a7245 100644 --- a/libavutil/hwcontext.c +++ b/libavutil/hwcontext.c @@ -47,6 +47,52 @@ static const HWContextType * const hw_table[] = { NULL, }; +static const struct { + enum AVHWDeviceType type; + const char *name; +} hw_type_map[] = { + { AV_HWDEVICE_TYPE_CUDA, "cuda" }, + { AV_HWDEVICE_TYPE_DXVA2, "dxva2" }, + { AV_HWDEVICE_TYPE_QSV, "qsv" }, + { AV_HWDEVICE_TYPE_VAAPI, "vaapi" }, + { AV_HWDEVICE_TYPE_VDPAU, "vdpau" }, +}; + +enum AVHWDeviceType av_hwdevice_find_type_by_name(const char *name) +{ + int i; + for (i = 0; i < FF_ARRAY_ELEMS(hw_type_map); i++) { + if (!strcmp(hw_type_map[i].name, name)) + return hw_type_map[i].type; + } + return AV_HWDEVICE_TYPE_NONE; +} + +const char *av_hwdevice_get_type_name(enum AVHWDeviceType type) +{ + int i; + for (i = 0; i < FF_ARRAY_ELEMS(hw_type_map); i++) { + if (hw_type_map[i].type == type) + return hw_type_map[i].name; + } + return NULL; +} + +enum AVHWDeviceType av_hwdevice_iterate_types(enum AVHWDeviceType prev) +{ + enum AVHWDeviceType next; + int i, set = 0; + for (i = 0; hw_table[i]; i++) { + if (prev != AV_HWDEVICE_TYPE_NONE && hw_table[i]->type <= prev) + continue; + if (!set || hw_table[i]->type < next) { + next = hw_table[i]->type; + set = 1; + } + } + return set ? next : AV_HWDEVICE_TYPE_NONE; +} + static const AVClass hwdevice_ctx_class = { .class_name = "AVHWDeviceContext", .item_name = av_default_item_name, diff --git a/libavutil/hwcontext.h b/libavutil/hwcontext.h index 49c7a21c1..cce7cdc7e 100644 --- a/libavutil/hwcontext.h +++ b/libavutil/hwcontext.h @@ -30,6 +30,8 @@ enum AVHWDeviceType { AV_HWDEVICE_TYPE_VAAPI, AV_HWDEVICE_TYPE_DXVA2, AV_HWDEVICE_TYPE_QSV, + + AV_HWDEVICE_TYPE_NONE }; typedef struct AVHWDeviceInternal AVHWDeviceInternal; @@ -223,6 +225,32 @@ typedef struct AVHWFramesContext { } AVHWFramesContext; /** + * Look up an AVHWDeviceType by name. + * + * @param name String name of the device type (case-insensitive). + * @return The type from enum AVHWDeviceType, or AV_HWDEVICE_TYPE_NONE if + * not found. + */ +enum AVHWDeviceType av_hwdevice_find_type_by_name(const char *name); + +/** Get the string name of an AVHWDeviceType. + * + * @param type Type from enum AVHWDeviceType. + * @return Pointer to a static string containing the name, or NULL if the type + * is not valid. + */ +const char *av_hwdevice_get_type_name(enum AVHWDeviceType type); + +/** + * Iterate over supported device types. + * + * @param type AV_HWDEVICE_TYPE_NONE initially, then the previous type + * returned by this function in subsequent iterations. + * @return The next usable device type from enum AVHWDeviceType. + */ +enum AVHWDeviceType av_hwdevice_iterate_types(enum AVHWDeviceType prev); + +/** * Allocate an AVHWDeviceContext for a given pixel format. * * @return a reference to the newly created AVHWDeviceContext on success or NULL -- 2.11.0 _______________________________________________ libav-devel mailing list libav-devel@libav.org https://lists.libav.org/mailman/listinfo/libav-devel