Function d3d12va_encode_setup_roi

libavcodec/d3d12va_encode.c:166–263

}

static int d3d12va_encode_setup_roi(AVCodecContext *avctx,
                                    D3D12VAEncodePicture *pic,
                                    const uint8_t *data, size_t size)
{
    D3D12VAEncodeContext *ctx = avctx->priv_data;
    const AVRegionOfInterest *roi;
    uint32_t roi_size;
    int nb_roi, i;
    int block_width, block_height;
    int block_size, qp_range;
    int is_av1 = 0;

    // Use the QP map region size reported by the driver
    block_size = ctx->qp_map_region_size;

    // Determine QP range and element size based on codec
    switch (ctx->codec->d3d12_codec) {
        case D3D12_VIDEO_ENCODER_CODEC_H264:
        case D3D12_VIDEO_ENCODER_CODEC_HEVC:
            qp_range = 51;
            is_av1 = 0;
            break;
#if CONFIG_AV1_D3D12VA_ENCODER
        case D3D12_VIDEO_ENCODER_CODEC_AV1:
            qp_range = 255;
            is_av1 = 1;
            break;
#endif
        default:
            av_log(avctx, AV_LOG_ERROR, "Unsupported codec for ROI.\n");
            return AVERROR(EINVAL);
    }

    // Calculate map dimensions using ceil division as required by D3D12
    block_width  = (avctx->width + block_size - 1) / block_size;
    block_height = (avctx->height + block_size - 1) / block_size;

    // Allocate QP map with correct type based on codec
    if (is_av1) {
        pic->qp_map = av_calloc(block_width * block_height, sizeof(int16_t));
    } else {
        pic->qp_map = av_calloc(block_width * block_height, sizeof(int8_t));
    }
    if (!pic->qp_map)
        return AVERROR(ENOMEM);

    // Process ROI regions
    roi = (const AVRegionOfInterest *)data;
    roi_size = roi->self_size;
    av_assert0(roi_size && size % roi_size == 0);
    nb_roi = size / roi_size;

    // Iterate in reverse for priority (first region in array takes priority on overlap)
    for (i = nb_roi - 1; i >= 0; i--) {
        int startx, endx, starty, endy;
        int delta_qp;
        int x, y;