develop/utils/rocvideodecode/roc_video_dec.h Source File

develop/utils/rocvideodecode/roc_video_dec.h Source File#

rocDecode: develop/utils/rocvideodecode/roc_video_dec.h Source File
 /*
 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 <assert.h>
 #include <stdint.h>
 #include <mutex>
 #include <vector>
 #include <string>
 #include <iostream>
 #include <sstream>
 #include <string.h>
 #include <queue>
 #include <stdexcept>
 #include <exception>
 #include <cstring>
 #include <unordered_map>
 #include <chrono>
 #include <hip/hip_runtime.h>
 #include "rocdecode.h"
 #include "rocparser.h"
  
 #define MAX_FRAME_NUM       16
  
 typedef int (ROCDECAPI *PFNRECONFIGUEFLUSHCALLBACK)(void *, uint32_t, void *);
  
 typedef enum SeiAvcHevcPayloadType_enum {
     SEI_TYPE_TIME_CODE = 136,
     SEI_TYPE_USER_DATA_UNREGISTERED = 5
 } SeiAvcHevcPayloadType;
  
 typedef enum OutputSurfaceMemoryType_enum {
     OUT_SURFACE_MEM_DEV_INTERNAL = 0,      
     OUT_SURFACE_MEM_DEV_COPIED = 1,        
     OUT_SURFACE_MEM_HOST_COPIED = 2,        
     OUT_SURFACE_MEM_NOT_MAPPED  = 3         
 } OutputSurfaceMemoryType;
  
 #define TOSTR(X) std::to_string(static_cast<int>(X))
 #define STR(X) std::string(X)
  
 #if DBGINFO
 #define INFO(X) std::clog << "[INF] " << " {" << __func__ <<"} " << " " << X << std::endl;
 #else
 #define INFO(X) ;
 #endif
 #define ERR(X) std::cerr << "[ERR] "  << " {" << __func__ <<"} " << " " << X << std::endl;
  
 inline int GetChromaPlaneCount(rocDecVideoSurfaceFormat surface_format) {
     int num_planes = 1;
     switch (surface_format) {
     case rocDecVideoSurfaceFormat_NV12:
     case rocDecVideoSurfaceFormat_P016:
         num_planes = 1;
         break;
     case rocDecVideoSurfaceFormat_YUV444:
     case rocDecVideoSurfaceFormat_YUV444_16Bit:
         num_planes = 2;
         break;
     case rocDecVideoSurfaceFormat_YUV420:
     case rocDecVideoSurfaceFormat_YUV420_16Bit:
         num_planes = 2;
         break;
     }
  
     return num_planes;
 };
  
 inline float GetChromaHeightFactor(rocDecVideoSurfaceFormat surface_format) {
     float factor = 0.5;
     switch (surface_format) {
     case rocDecVideoSurfaceFormat_NV12:
     case rocDecVideoSurfaceFormat_P016:
     case rocDecVideoSurfaceFormat_YUV420:
     case rocDecVideoSurfaceFormat_YUV420_16Bit:
         factor = 0.5;
         break;
     case rocDecVideoSurfaceFormat_YUV444:
     case rocDecVideoSurfaceFormat_YUV444_16Bit:
         factor = 1.0;
         break;
     }
  
     return factor;
 };
  
  
 class RocVideoDecodeException : public std::exception {
 public:
  
     explicit RocVideoDecodeException(const std::string& message, const int err_code):_message(message), _err_code(err_code) {}
     explicit RocVideoDecodeException(const std::string& message):_message(message), _err_code(-1) {}
     virtual const char* what() const throw() override {
         return _message.c_str();
     }
     int Geterror_code() const { return _err_code; }
 private:
     std::string _message;
     int _err_code;
 };
  
 #define ROCDEC_THROW(X, CODE) throw RocVideoDecodeException(" { " + std::string(__func__) + " } " + X , CODE);
 #define THROW(X) throw RocVideoDecodeException(" { " + std::string(__func__) + " } " + X);
  
 #define ROCDEC_API_CALL( rocDecAPI )                                                                         \
     do {                                                                                                     \
         rocDecStatus error_code = rocDecAPI;                                                                 \
         if( error_code != ROCDEC_SUCCESS) {                                                                  \
             std::ostringstream error_log;                                                                    \
             error_log << #rocDecAPI << " returned " << rocDecGetErrorName(error_code) << " at " <<__FILE__ <<":" << __LINE__;\
             ROCDEC_THROW(error_log.str(), error_code); \
         }                                                                                                     \
     } while (0)
  
 #define HIP_API_CALL( call )                                                                                  \
     do {                                                                                                      \
         hipError_t hip_status = call;                                                                         \
         if (hip_status != hipSuccess) {                                                                       \
             const char *sz_err_name = NULL;                                                                     \
             sz_err_name = hipGetErrorName(hip_status);                                                          \
             std::ostringstream error_log;                                                                     \
             error_log << "hip API error " << sz_err_name ;                                                      \
             ROCDEC_THROW(error_log.str(), hip_status);                   \
         }                                                                                                     \
     }                                                                                                         \
     while (0)
  
  
 struct Rect {
     int left;
     int top;
     int right;
     int bottom;
 };
  
 struct Dim {
     int w, h;
 };
  
 static inline int align(int value, int alignment) {
    return (value + alignment - 1) & ~(alignment - 1);
 }
  
 typedef struct DecFrameBuffer_ {
     uint8_t *frame_ptr;       
     int64_t  pts;             
     int picture_index;         
 } DecFrameBuffer;
  
  
 typedef struct OutputSurfaceInfoType {
     uint32_t output_width;                      
     uint32_t output_height;                     
     uint32_t output_pitch;                      
     uint32_t output_vstride;                    
     uint32_t chroma_height;                     
     Rect     disp_rect;                         
     uint32_t bytes_per_pixel;                   
     uint32_t bit_depth;                         
     uint32_t num_chroma_planes;                 
     uint64_t output_surface_size_in_bytes;      
     rocDecVideoSurfaceFormat surface_format;    
     OutputSurfaceMemoryType mem_type;           
 } OutputSurfaceInfo;
  
 typedef struct ReconfigParams_t {
     PFNRECONFIGUEFLUSHCALLBACK p_fn_reconfigure_flush;
     void *p_reconfig_user_struct;
     uint32_t reconfig_flush_mode;
 } ReconfigParams;
  
 class RocVideoDecoder {
     public:
         RocVideoDecoder(int device_id,  OutputSurfaceMemoryType out_mem_type, rocDecVideoCodec codec, bool force_zero_latency = false,
                           const Rect *p_crop_rect = nullptr, bool extract_user_SEI_Message = false, uint32_t disp_delay = 0, int max_width = 0, int max_height = 0,
                           uint32_t clk_rate = 1000);
         ~RocVideoDecoder();
         
         rocDecVideoCodec GetCodecId() { return codec_id_; }
  
         hipStream_t GetStream() {return hip_stream_;}
  
         uint32_t GetWidth() { assert(disp_width_); return disp_width_;}
  
         int GetDecodeWidth() { assert(coded_width_); return coded_width_; }
  
         uint32_t GetHeight() { assert(disp_height_); return disp_height_; }
  
         int GetChromaHeight() { assert(chroma_height_); return chroma_height_; }
  
         int GetNumChromaPlanes() { assert(num_chroma_planes_); return num_chroma_planes_; }
  
         virtual int GetFrameSize() { assert(disp_width_); return disp_width_ * (disp_height_ + (chroma_height_ * num_chroma_planes_)) * byte_per_pixel_; }
  
  
         uint32_t GetBitDepth() { assert(bitdepth_minus_8_); return (bitdepth_minus_8_ + 8); }
         uint32_t GetBytePerPixel() { assert(byte_per_pixel_); return byte_per_pixel_; }
         size_t GetSurfaceSize() { assert(surface_size_); return surface_size_; }
         uint32_t GetSurfaceStride() { assert(surface_stride_); return surface_stride_; }
         //RocDecImageFormat GetSubsampling() { return subsampling_; }
         const char *GetCodecFmtName(rocDecVideoCodec codec_id);
  
         const char *GetSurfaceFmtName(rocDecVideoSurfaceFormat surface_format_id);
  
         bool GetOutputSurfaceInfo(OutputSurfaceInfo **surface_info);
  
         bool SetReconfigParams(ReconfigParams *p_reconfig_params, bool b_force_reconfig_flush = false);
         
         int FlushAndReconfigure();
         virtual int DecodeFrame(const uint8_t *data, size_t size, int pkt_flags, int64_t pts = 0, int *num_decoded_pics = nullptr);
         virtual uint8_t* GetFrame(int64_t *pts);
  
         virtual bool ReleaseFrame(int64_t pTimestamp, bool b_flushing = false);
  
         //void SaveImage(std::string output_file_name, void* dev_mem, OutputImageInfo* image_info, bool is_output_RGB = 0);
  
         void GetDeviceinfo(std::string &device_name, std::string &gcn_arch_name, int &pci_bus_id, int &pci_domain_id, int &pci_device_id);
         
         virtual void SaveFrameToFile(std::string output_file_name, void *surf_mem, OutputSurfaceInfo *surf_info, size_t rgb_image_size = 0);
  
         virtual void ResetSaveFrameToFile();
  
         int32_t GetNumOfFlushedFrames() { return num_frames_flushed_during_reconfig_;}
  
         void WaitForDecodeCompletion();
  
         // Session overhead refers to decoder initialization and deinitialization time
         void AddDecoderSessionOverHead(std::thread::id session_id, double duration) { session_overhead_[session_id] += duration; }
         double GetDecoderSessionOverHead(std::thread::id session_id) {
             if (session_overhead_.find(session_id) != session_overhead_.end()) {
                 return session_overhead_[session_id];
             } else {
                 return 0;
             }
          }
  
         bool CodecSupported(int device_id, rocDecVideoCodec codec_id, uint32_t bit_depth);
  
         virtual int ReconfigureDecoder(RocdecVideoFormat *p_video_format);
  
     protected:
         static int ROCDECAPI HandleVideoSequenceProc(void *p_user_data, RocdecVideoFormat *p_video_format) { return ((RocVideoDecoder *)p_user_data)->HandleVideoSequence(p_video_format); }
  
         static int ROCDECAPI HandlePictureDecodeProc(void *p_user_data, RocdecPicParams *p_pic_params) { return ((RocVideoDecoder *)p_user_data)->HandlePictureDecode(p_pic_params); }
  
         static int ROCDECAPI HandlePictureDisplayProc(void *p_user_data, RocdecParserDispInfo *p_disp_info) { return ((RocVideoDecoder *)p_user_data)->HandlePictureDisplay(p_disp_info); }
  
         static int ROCDECAPI HandleSEIMessagesProc(void *p_user_data, RocdecSeiMessageInfo *p_sei_message_info) { return ((RocVideoDecoder *)p_user_data)->GetSEIMessage(p_sei_message_info); } 
  
         int HandleVideoSequence(RocdecVideoFormat *p_video_format);
  
         int HandlePictureDecode(RocdecPicParams *p_pic_params);
  
         int HandlePictureDisplay(RocdecParserDispInfo *p_disp_info);
         int GetSEIMessage(RocdecSeiMessageInfo *p_sei_message_info);
         
         bool ReleaseInternalFrames();
  
         bool InitHIP(int device_id);
  
         std::chrono::_V2::system_clock::time_point StartTimer();
  
         double StopTimer(const std::chrono::_V2::system_clock::time_point &start_time);
  
         int num_devices_;
         int device_id_;
         RocdecVideoParser rocdec_parser_ = nullptr;
         rocDecDecoderHandle roc_decoder_ = nullptr;
         OutputSurfaceMemoryType out_mem_type_ = OUT_SURFACE_MEM_DEV_INTERNAL;
         bool b_extract_sei_message_ = false;
         bool b_force_zero_latency_ = false;
         uint32_t disp_delay_;
         ReconfigParams *p_reconfig_params_ = nullptr;
         bool b_force_recofig_flush_ = false;
         int32_t num_frames_flushed_during_reconfig_ = 0;
         hipDeviceProp_t hip_dev_prop_;
         hipStream_t hip_stream_;
         rocDecVideoCodec codec_id_ = rocDecVideoCodec_NumCodecs;
         rocDecVideoChromaFormat video_chroma_format_ = rocDecVideoChromaFormat_420;
         rocDecVideoSurfaceFormat video_surface_format_ = rocDecVideoSurfaceFormat_NV12;
         RocdecSeiMessageInfo *curr_sei_message_ptr_ = nullptr;
         RocdecSeiMessageInfo sei_message_display_q_[MAX_FRAME_NUM];
         RocdecVideoFormat *curr_video_format_ptr_ = nullptr;
         int output_frame_cnt_ = 0, output_frame_cnt_ret_ = 0;
         int decoded_pic_cnt_ = 0;
         int decode_poc_ = 0, pic_num_in_dec_order_[MAX_FRAME_NUM];
         int num_alloced_frames_ = 0;
         int last_decode_surf_idx_ = 0;
         std::ostringstream input_video_info_str_;
         int bitdepth_minus_8_ = 0;
         uint32_t byte_per_pixel_ = 1;
         uint32_t coded_width_ = 0;
         uint32_t disp_width_ = 0;
         uint32_t coded_height_ = 0;
         uint32_t disp_height_ = 0;
         uint32_t target_width_ = 0;
         uint32_t target_height_ = 0;
         int max_width_ = 0, max_height_ = 0;
         uint32_t chroma_height_ = 0, chroma_width_ = 0;
         uint32_t num_chroma_planes_ = 0;
         uint32_t num_components_ = 0;
         uint32_t surface_stride_ = 0;
         uint32_t surface_vstride_ = 0, chroma_vstride_ = 0;      // vertical stride between planes: used when using internal dev memory
         size_t surface_size_ = 0;
         OutputSurfaceInfo output_surface_info_ = {};
         std::mutex mtx_vp_frame_;
         std::vector<DecFrameBuffer> vp_frames_;      // vector of decoded frames
         std::queue<DecFrameBuffer> vp_frames_q_;
         Rect disp_rect_ = {}; // displayable area specified in the bitstream
         Rect crop_rect_ = {}; // user specified region of interest within diplayable area disp_rect_
         FILE *fp_sei_ = NULL;
         FILE *fp_out_ = NULL;
         bool is_decoder_reconfigured_ = false;
         std::string current_output_filename = "";
         uint32_t extra_output_file_count_ = 0;
         std::thread::id decoder_session_id_; // Decoder session identifier. Used to gather session level stats.
         std::unordered_map<std::thread::id, double> session_overhead_; // Records session overhead of initialization+deinitialization time. Format is (thread id, duration)
 };