av1 initial commit (#298)

[ROCm/rocdecode commit: 6265f675c1]
2024-03-26 15:59:15 -07:00
@@ -0,0 +1,27 @@
+/*
+Copyright (c) 2024 Advanced Micro Devices, Inc. All rights reserved.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.
+*/
+
+#pragma once
+
+#include <stdint.h>
+
+//todo: add the defines
@@ -0,0 +1,44 @@
+/*
+Copyright (c) 2024 Advanced Micro Devices, Inc. All rights reserved.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.
+*/
+
+#include <algorithm>
+#include "av1_parser.h"
+
+Av1VideoParser::Av1VideoParser() {
+    //todo: add constructor code
+}
+
+Av1VideoParser::~Av1VideoParser() {
+}
+
+rocDecStatus Av1VideoParser::Initialize(RocdecParserParams *p_params) {
+    return RocVideoParser::Initialize(p_params);
+}
+
+rocDecStatus Av1VideoParser::UnInitialize() {
+    return ROCDEC_SUCCESS;
+}
+
+rocDecStatus Av1VideoParser::ParseVideoData(RocdecSourceDataPacket *p_data) { 
+    //to be implemented
+    return ROCDEC_NOT_IMPLEMENTED;
+}
@@ -0,0 +1,53 @@
+/*
+Copyright (c) 2023 - 2024 Advanced Micro Devices, Inc. All rights reserved.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.
+*/
+#pragma once
+
+#include "av1_defines.h"
+#include "roc_video_parser.h"
+
+class Av1VideoParser : public RocVideoParser {
+public:
+    /*! \brief Av1VideoParser constructor
+     */
+    Av1VideoParser();
+
+    /*! \brief Av1VideoParser destructor
+     */
+    virtual ~Av1VideoParser();
+
+    /*! \brief Function to Initialize the parser
+     * \param [in] p_params Input of <tt>RocdecParserParams</tt> with codec type to initialize parser.
+     * \return <tt>rocDecStatus</tt> Returns success on completion, else error code for failure
+     */
+    virtual rocDecStatus Initialize(RocdecParserParams *p_params);
+
+    /*! \brief Function to Parse video data: Typically called from application when a demuxed picture is ready to be parsed
+     * \param [in] p_data Pointer to picture data of type <tt>RocdecSourceDataPacket</tt>
+     * @return <tt>rocDecStatus</tt> Returns success on completion, else error_code for failure
+     */
+    virtual rocDecStatus ParseVideoData(RocdecSourceDataPacket *p_data);
+
+    /*! \brief function to uninitialize AV1 parser
+     * @return rocDecStatus 
+     */
+    virtual rocDecStatus UnInitialize();     // derived method
+};
@@ -26,6 +26,7 @@ THE SOFTWARE.
 #include "rocparser.h"
 #include "roc_video_parser.h"
 #include "avc_parser.h"
+#include "av1_parser.h"
 #include "hevc_parser.h"

 class RocParserHandle {
@@ -49,6 +50,9 @@ private:
            case rocDecVideoCodec_HEVC:
                roc_parser_ = std::make_shared<HevcVideoParser>();
                break;
+            case rocDecVideoCodec_AV1:
+                roc_parser_ = std::make_shared<Av1VideoParser>();
+                break;
            default:
                THROW("Unsupported parser type "+ TOSTR(params->codec_type));
                break;
@@ -35,8 +35,10 @@ rocDecCreateVideoParser(RocdecVideoParser *parser_handle, RocdecParserParams *pa
    }

    const char *is_avc_enabled = std::getenv("ROCDECODE_ENABLE_AVC");
+    const char *is_av1_enabled = std::getenv("ROCDECODE_ENABLE_AV1");
    if (parser_params->codec_type != rocDecVideoCodec_HEVC &&
-        (parser_params->codec_type == rocDecVideoCodec_AVC && (is_avc_enabled == nullptr || std::string(is_avc_enabled) != "1"))) {
+        (parser_params->codec_type == rocDecVideoCodec_AVC && (is_avc_enabled == nullptr || std::string(is_avc_enabled) != "1")) &&
+        (parser_params->codec_type == rocDecVideoCodec_AV1 && (is_av1_enabled == nullptr || std::string(is_av1_enabled) != "1"))) {
        ERR("The current version of rocDecode officially supports only the H.265 (HEVC) codec.");
        return ROCDEC_NOT_IMPLEMENTED;
    }