1 /* 2 * Copyright (C) 2023 Huawei Device Co., Ltd. 3 * Licensed under the Apache License, Version 2.0 (the "License"); 4 * you may not use this file except in compliance with the License. 5 * You may obtain a copy of the License at 6 * 7 * http://www.apache.org/licenses/LICENSE-2.0 8 * 9 * Unless required by applicable law or agreed to in writing, software 10 * distributed under the License is distributed on an "AS IS" BASIS, 11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 * See the License for the specific language governing permissions and 13 * limitations under the License. 14 */ 15 16 #ifndef CODEC_UTILS_H 17 #define CODEC_UTILS_H 18 19 #include "av_common.h" 20 #include "avcodec_errors.h" 21 #include "buffer/avbuffer.h" 22 #include "fsurface_memory.h" 23 #include "meta/format.h" 24 #include "surface.h" 25 extern "C" { 26 #include "libavcodec/avcodec.h" 27 #include "libavutil/imgutils.h" 28 #include "libswscale/swscale.h" 29 }; 30 namespace OHOS { 31 namespace MediaAVCodec { 32 namespace Codec { 33 using AVMemory = Media::AVMemory; 34 using Format = Media::Format; 35 const int32_t VIDEO_ALIGN_SIZE = 16; 36 constexpr uint32_t VIDEO_PIX_DEPTH_RGBA = 4; 37 constexpr int32_t UV_SCALE_FACTOR = 2; 38 struct ScalePara { 39 int32_t srcWidth = 0; 40 int32_t srcHeight = 0; 41 AVPixelFormat srcFfFmt = AVPixelFormat::AV_PIX_FMT_NONE; 42 int32_t dstWidth = 0; 43 int32_t dstHeight = 0; 44 AVPixelFormat dstFfFmt = AVPixelFormat::AV_PIX_FMT_RGBA; 45 int32_t align = VIDEO_ALIGN_SIZE; 46 }; 47 struct Scale { 48 public: 49 int32_t Init(const ScalePara &scalePara, uint8_t **dstData, int32_t *dstLineSize); 50 int32_t Convert(uint8_t **srcData, const int32_t *srcLineSize, uint8_t **dstData, int32_t *dstLineSize); 51 52 private: 53 ScalePara scalePara_; 54 std::shared_ptr<SwsContext> swsCtx_ = nullptr; 55 }; 56 struct SurfaceInfo { 57 uint32_t surfaceStride = 0; 58 sptr<SyncFence> surfaceFence = nullptr; 59 uint8_t **scaleData = nullptr; 60 int32_t *scaleLineSize = nullptr; 61 }; 62 GraphicPixelFormat TranslateSurfaceFormat(const VideoPixelFormat &surfaceFormat); 63 VideoPixelFormat ConvertPixelFormatFromFFmpeg(int32_t ffmpegPixelFormat); 64 AVPixelFormat ConvertPixelFormatToFFmpeg(VideoPixelFormat pixelFormat); 65 GraphicTransformType TranslateSurfaceRotation(const VideoRotation &rotation); 66 int32_t ConvertVideoFrame(std::shared_ptr<Scale> *scale, std::shared_ptr<AVFrame> frame, uint8_t **dstData, 67 int32_t *dstLineSize, AVPixelFormat dstPixFmt); 68 int32_t WriteSurfaceData(const std::shared_ptr<AVMemory> &memory, struct SurfaceInfo &surfaceInfo, 69 const Format &format); 70 int32_t WriteBufferData(const std::shared_ptr<AVMemory> &memory, uint8_t **scaleData, int32_t *scaleLineSize, 71 const Format &format); 72 73 std::string AVStrError(int errnum); 74 bool IsYuvFormat(VideoPixelFormat &format); 75 bool IsRgbFormat(VideoPixelFormat &format); 76 } // namespace Codec 77 } // namespace MediaAVCodec 78 } // namespace OHOS 79 #endif