Initial commit of the rocJPEG library (#1)

* rocJPEG initial commit * Add gitignore file * clean up * code clean up * CMakeLists update * Update README * Update readme and setup script * add docs folder * update rocjepg headers * update the rocjepg header * update the copyright year * Add nullptr check for all APIs * update the hip kernels execution * follow google c++ style guide * code clean up * use google c++ style guide for hip kernels * add support for rocJpegGetErrorName API * Add ParseCommandLine function * update the rocjpeg sample * Update the readme, LIBVA instructions * udate the rocJPEG script * Update CMAKE and libdrm * Add support for ROCJPEG_OUTPUT_UNCHANGED and SaveImage function * code clean up * update the drm dependencies * remove the unused hipstream * code clean up * add -fmt option for selection the output format * update the GetImageInfo API * update the sample * update the save image function based onn the surface and putput format * add new sample images * Put hip interop mem into separate functions and some code clean up * Add support for CopyLuma and CopyChroma * Add support for the yuv output format for NV12 surface * Update HIP kernels * add support for rgbi format conversion for NV12 and YUv444 * Update the status * Add support for converting YUYV to RGB * Add support for unpacking YUYV format * Fix unpacking Y fron YUYV hip kernel * Add suppoort for extractign packed YUYV to YUV planar * hip code clean up * code clean up * Update the headers * code clean up * change hip kernel names * hip kernel clean up * hip clean up * code clean up * code clean up * code clean up * code clean up * code clean up * code clean up for jpege decoder class * code clean up * code clean up * code clean up * code clean up * update the dockers * code clean up * code clean up * hip kernels clean up * remove unused hip kernels * add additional test cases * update the APIs * add new hip yuv400torgbi kernel * update yuv400torgbi kernel * restructure files * code clean up * code clean up * add jenkins * code clean up * code clean up * update readme * update docker's README * make changes based on the reviewers comments * make changes based on the reviewers comments * return ROCJPEG_STATUS_JPEG_NOT_SUPPORTED if the resolution of the jpeg is not supported for HW decoding [ROCm/rocjpeg commit: 6ad4a76177]
2024-03-25 17:05:31 -04:00
parent 52b4d4dd5f
commit 929a892e8d
33 changed files with 4673 additions and 3 deletions
@@ -0,0 +1,75 @@
+/*
+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.
+*/
+
+#ifndef ROC_JPEG_VAAPI_DECODER_H_
+#define ROC_JPEG_VAAPI_DECODER_H_
+
+#pragma once
+
+#include <iostream>
+#include <vector>
+#include <string>
+#include <fcntl.h>
+#include <unistd.h>
+#include <va/va.h>
+#include <va/va_drm.h>
+#include <va/va_drmcommon.h>
+#include "rocjpeg_commons.h"
+#include "rocjpeg_parser.h"
+#include "../api/rocjpeg.h"
+
+/*Note: va.h doesn't have VA_FOURCC_YUYV defined but vaExportSurfaceHandle returns 0x56595559 for packed YUYV for YUV 4:2:2*/
+#define ROCJPEG_FOURCC_YUYV 0x56595559
+
+class RocJpegVappiDecoder {
+public:
+    RocJpegVappiDecoder(int device_id = 0);
+    ~RocJpegVappiDecoder();
+    RocJpegStatus InitializeDecoder(std::string gcn_arch_name);
+    RocJpegStatus SubmitDecode(const JpegStreamParameters *jpeg_stream_params, uint32_t &surface_id);
+    RocJpegStatus ExportSurface(VASurfaceID surface_id, VADRMPRIMESurfaceDescriptor &va_drm_prime_surface_desc);
+    RocJpegStatus SyncSurface(VASurfaceID surface_id);
+    RocJpegStatus ReleaseSurface(VASurfaceID surface_id);
+private:
+    int device_id_;
+    int drm_fd_;
+    uint32_t min_picture_width_;
+    uint32_t min_picture_height_;
+    uint32_t max_picture_width_;
+    uint32_t max_picture_height_;
+    VADisplay va_display_;
+    std::vector<VAConfigAttrib> va_config_attrib_;
+    VAConfigID va_config_id_;
+    VAProfile va_profile_;
+    VAContextID va_context_id_;
+    std::vector<VASurfaceID> va_surface_ids_;
+    VABufferID va_picture_parameter_buf_id_;
+    VABufferID va_quantization_matrix_buf_id_;
+    VABufferID va_huffmantable_buf_id_;
+    VABufferID va_slice_param_buf_id_;
+    VABufferID va_slice_data_buf_id_;
+    RocJpegStatus InitVAAPI(std::string drm_node);
+    RocJpegStatus CreateDecoderConfig();
+    RocJpegStatus DestroyDataBuffers();
+};
+
+#endif // ROC_JPEG_VAAPI_DECODER_H_