AI/rocm-systems
2
0
Forkuj 0
Kod Zgłoszenia Oczekujące zmiany Actions Packages Projekty Wydania Wiki Aktywność

AV1: VA-API submission (#362)

Przeglądaj źródła 
* av1 va-api submission

* adding check for anchor_frames and current_diplay_picture

* check limits for memory errors

* review comment - remove unnecessary check
This commit is contained in:
Lakshmi Kumar
2024-06-05 13:23:47 -07:00
zatwierdzone przez GitHub
rodzic 641044d079
commit c70e98abb6
1 zmienionych plików z 45 dodań i 0 usunięć
Ściągnij plik aktualizacji Ściągnij plik porównania Expand all files Collapse all files
src/rocdecode/vaapi/vaapi_videodecoder.cpp
+45
Wyświetl plik
@@ -152,6 +152,9 @@ rocDecStatus VaapiVideoDecoder::CreateDecoderConfig() {
case rocDecVideoCodec_AVC:
va_profile_ = VAProfileH264Main;
break;
case rocDecVideoCodec_AV1:
va_profile_ = VAProfileAV1Profile0;
break;
default:
ERR("The codec type is not supported.");
return ROCDEC_NOT_SUPPORTED;
@@ -301,6 +304,48 @@ rocDecStatus VaapiVideoDecoder::SubmitDecode(RocdecPicParams *pPicParams) {
break;
}
case rocDecVideoCodec_AV1: {
pPicParams->pic_params.av1.current_frame = curr_surface_id;
if (pPicParams->pic_params.av1.current_display_picture != 0xFF) {
if (pPicParams->pic_params.av1.current_display_picture >= va_surface_ids_.size() || pPicParams->pic_params.av1.current_display_picture < 0) {
ERR("Current display picture index exceeded the VAAPI surface pool limit.");
return ROCDEC_INVALID_PARAMETER;
}
pPicParams->pic_params.av1.current_display_picture = va_surface_ids_[pPicParams->pic_params.av1.current_display_picture];
}
for (int i = 0; i < pPicParams->pic_params.av1.anchor_frames_num; i++) {
if (pPicParams->pic_params.av1.anchor_frames_list[i] >= va_surface_ids_.size() || pPicParams->pic_params.av1.anchor_frames_list[i] < 0) {
ERR("Anchor frame index exceeded the VAAPI surface pool limit.");
return ROCDEC_INVALID_PARAMETER;
}
pPicParams->pic_params.av1.anchor_frames_list[i] = va_surface_ids_[pPicParams->pic_params.av1.anchor_frames_list[i]];
}
for (int i = 0; i < 8; i++) {
if (pPicParams->pic_params.av1.ref_frame_map[i] != 0xFF) {
if (pPicParams->pic_params.av1.ref_frame_map[i] >= va_surface_ids_.size() || pPicParams->pic_params.av1.ref_frame_map[i] < 0) {
ERR("Reference frame index exceeded the VAAPI surface pool limit.");
return ROCDEC_INVALID_PARAMETER;
}
pPicParams->pic_params.av1.ref_frame_map[i] = va_surface_ids_[pPicParams->pic_params.av1.ref_frame_map[i]];
}
}
pic_params_ptr = (void*)&pPicParams->pic_params.av1;
pic_params_size = sizeof(RocdecAv1PicParams);
slice_params_ptr = (void*)pPicParams->slice_params.av1;
slice_params_size = sizeof(RocdecAv1SliceParams);
if ((pic_params_size != sizeof(VADecPictureParameterBufferAV1)) || (slice_params_size != sizeof(VASliceParameterBufferAV1))) {
ERR("AV1 data_buffer parameter_size not matching vaapi parameter buffer size.");
return ROCDEC_RUNTIME_ERROR;
}
break;
}
default: {
ERR("The codec type is not supported.");
return ROCDEC_NOT_SUPPORTED;