From 09cc5aeb8c1299126b0196f7bc182989c3425f19 Mon Sep 17 00:00:00 2001
From: "Xue, Maxim" <maxim.xue@intel.com>
Date: Thu, 7 Sep 2023 07:10:11 +0000
Subject: [PATCH] test: Enable AV1 encode test

---
 test/test_data.h                  | 12 +++++++++++-
 test/test_va_api_createbuffer.cpp |  5 +++++
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/test/test_data.h b/test/test_data.h
index c29bbaaa..a04be830 100644
--- a/test/test_data.h
+++ b/test/test_data.h
@@ -59,7 +59,11 @@ static const Profiles g_vaProfiles = {
     VAProfileVP9Profile1,
     VAProfileVP9Profile2,
     VAProfileVP9Profile3,
-#if VA_CHECK_VERSION(1,18,0) 
+#if VA_CHECK_VERSION(1,8,0)
+    VAProfileAV1Profile0,
+    VAProfileAV1Profile1,
+#endif
+#if VA_CHECK_VERSION(1,18,0)
     VAProfileH264High10,
 #endif
 };
@@ -110,6 +114,12 @@ static const Profiles g_vaVP9Profiles = {
     VAProfileVP9Profile3,
 };
 
+static const Profiles g_vaAV1Profiles = {
+#if VA_CHECK_VERSION(1,8,0)
+    VAProfileAV1Profile0, VAProfileAV1Profile1,
+#endif
+};
+
 static const Entrypoints g_vaEntrypoints = {
     VAEntrypointVLD,
     VAEntrypointIZZ,
diff --git a/test/test_va_api_createbuffer.cpp b/test/test_va_api_createbuffer.cpp
index 2dc0f6a7..532ddaf5 100644
--- a/test/test_va_api_createbuffer.cpp
+++ b/test/test_va_api_createbuffer.cpp
@@ -145,6 +145,7 @@ static const BufferSpecsMap encodeBufferSpecs = {
             make_tuple(g_vaVP8Profiles, sizeof(VAEncSequenceParameterBufferVP8)),
             make_tuple(g_vaHEVCProfiles, sizeof(VAEncSequenceParameterBufferHEVC)),
             make_tuple(g_vaVP9Profiles, sizeof(VAEncSequenceParameterBufferVP9)),
+            make_tuple(g_vaAV1Profiles, sizeof(VAEncSequenceParameterBufferAV1)),
         }
     },
     {
@@ -157,6 +158,7 @@ static const BufferSpecsMap encodeBufferSpecs = {
             make_tuple(g_vaVP8Profiles, sizeof(VAEncPictureParameterBufferVP8)),
             make_tuple(g_vaHEVCProfiles, sizeof(VAEncPictureParameterBufferHEVC)),
             make_tuple(g_vaVP9Profiles, sizeof(VAEncPictureParameterBufferVP9)),
+            make_tuple(g_vaAV1Profiles, sizeof(VAEncPictureParameterBufferAV1)),
         }
     },
     {
@@ -165,16 +167,19 @@ static const BufferSpecsMap encodeBufferSpecs = {
             make_tuple(g_vaH264Profiles, sizeof(VAEncSliceParameterBufferH264)),
             make_tuple(g_vaJPEGProfiles, sizeof(VAEncSliceParameterBufferJPEG)),
             make_tuple(g_vaHEVCProfiles, sizeof(VAEncSliceParameterBufferHEVC)),
+            make_tuple(g_vaAV1Profiles, sizeof(VAEncPictureParameterBufferAV1)),            
         }
     },
     {
         VAEncPackedHeaderParameterBufferType, {
             make_tuple(g_vaProfiles, sizeof(VAEncPackedHeaderParameterBuffer)),
+            make_tuple(g_vaAV1Profiles, sizeof(VAEncPictureParameterBufferAV1)),            
         }
     },
     {
         VAEncMiscParameterBufferType, {
             make_tuple(g_vaProfiles, sizeof(VAEncMiscParameterBuffer)),
+            make_tuple(g_vaAV1Profiles, sizeof(VAEncPictureParameterBufferAV1)),            
         }
     },
 };