1 /* 2 * Copyright (c) 2022 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 OHOS_IMAGE_SOURCE_ENCODER_H 17 #define OHOS_IMAGE_SOURCE_ENCODER_H 18 19 #include <memory> 20 21 #include "avsharedmemory.h" 22 #include "avcodec_video_encoder.h" 23 #include "avcodec_common.h" 24 #include "media_errors.h" 25 #include "format.h" 26 #include "surface.h" 27 28 #include "iimage_source_processor_listener.h" 29 #include "image_encoder_callback.h" 30 #include "video_param.h" 31 32 namespace OHOS { 33 namespace DistributedHardware { 34 class ImageSourceEncoder : public std::enable_shared_from_this<ImageSourceEncoder> { 35 public: ImageSourceEncoder(const std::shared_ptr<IImageSourceProcessorListener> & imageListener)36 explicit ImageSourceEncoder(const std::shared_ptr<IImageSourceProcessorListener> &imageListener) 37 : imageProcessorListener_(imageListener) {}; 38 ~ImageSourceEncoder() = default; 39 40 int32_t ConfigureEncoder(const VideoParam &cofigParam); 41 int32_t ReleaseEncoder(); 42 int32_t StartEncoder(); 43 int32_t StopEncoder(); GetInputSurface()44 sptr<Surface> &GetInputSurface() 45 { 46 return videoSurface_; 47 } 48 49 void OnError(Media::AVCodecErrorType errorType, int32_t errorCode); 50 void OnOutputBufferAvailable(uint32_t index, Media::AVCodecBufferInfo info, Media::AVCodecBufferFlag flag); 51 void OnInputBufferAvailable(uint32_t index); 52 void OnOutputFormatChanged(const Media::Format &format); 53 54 private: 55 int32_t InitVideoEncoder(const VideoParam &configParam); 56 int32_t SetEncoderFormat(const VideoParam &cofigParam); 57 58 private: 59 static const constexpr char *LOG_TAG = "ImageSourceEncoder"; 60 61 Media::Format imageFormat_; 62 Media::AVCodecBufferInfo encoderBufferInfo_; 63 64 sptr<Surface> videoSurface_; 65 std::shared_ptr<Media::AVCodecVideoEncoder> videoEncoder_; 66 std::shared_ptr<Media::AVSharedMemory> videoSharedMemory_; 67 std::shared_ptr<Media::AVCodecCallback> encodeVideoCallback_; 68 std::weak_ptr<IImageSourceProcessorListener> imageProcessorListener_; 69 const size_t DATA_BUFFER_MAX_SIZE = 10 * 1024 * 1024; 70 }; 71 } // namespace DistributedHardware 72 } // namespace OHOS 73 #endif