/*
 * Copyright (C) 2023 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef AVBUFFER_H
#define AVBUFFER_H

#include <memory>
#include <string>
#include "buffer/avallocator.h"
#include "common/status.h"

namespace OHOS {
namespace Media {
/**
 * @brief Class that encapsulates some types of media buffer.
 */
class __attribute__((visibility("default"))) AVBuffer {
public:
    ~AVBuffer();
    /**
     * @brief Create the AVBuffer by configuration.
     * @param config The configuration of AVBuffer, refer to {@link AVBufferConfig}
     * @return The shared pointer of AVBuffer.
     * @since 4.1
     */
    static std::shared_ptr<AVBuffer> CreateAVBuffer(const AVBufferConfig &config);

    /**
     * @brief Create the AVBuffer by allocator.
     * @param allocator The instance of AVAllocator, refer to {@link AVAllocator}
     * @param capacity The capacity of the memory, bytes.
     * @param align The align of AVBuffer, bytes.
     * @return The shared pointer of AVBuffer.
     * @since 4.1
     */
    static std::shared_ptr<AVBuffer> CreateAVBuffer(std::shared_ptr<AVAllocator> allocator, int32_t capacity = 0,
                                                    int32_t align = 0);

    /**
     * @brief Create the AVBuffer by alloced memory.
     * @param ptr The pointer of alloced memory, it requires users to manage the lifecycle.
     * @param capacity The capacity of the memory, bytes.
     * @param size The size of the memory, bytes. If it can not greater than capacity.
     * @return The shared pointer of AVBuffer.
     * @since 4.1
     */
    static std::shared_ptr<AVBuffer> CreateAVBuffer(uint8_t *ptr, int32_t capacity, int32_t size = 0);

    /**
     * @brief Create the AVBuffer by surface buffer.
     * @param surfaceBuffer The sptr of SurfaceBuffer, refer to {@link SurfaceBuffer}
     * @return The shared pointer of AVBuffer.
     * @since 4.1
     */
    static std::shared_ptr<AVBuffer> CreateAVBuffer(sptr<SurfaceBuffer> surfaceBuffer);

    /**
     * @brief Create the AVBuffer.
     * @return The shared pointer of AVBuffer.
     * @since 4.1
     */
    static std::shared_ptr<AVBuffer> CreateAVBuffer();

    /**
     * @brief Get the AVBufferConfig.
     * @return The config struct of AVBuffer.
     * @since 4.1
     */
    const AVBufferConfig &GetConfig();

    /**
     * @brief Get the unique identifier of buffer.
     * @return The unique identifier of buffer.
     * @since 4.1
     */
    uint64_t GetUniqueId();

    /**
     * @brief Wirte buffer info to MessageParcel.
     * @param parcel The MessageParcel wirtten by buffer, refer to {@link MessageParcel}.
     * @return Whether the writing was successful.
     * @since 4.1
     */
    bool WriteToMessageParcel(MessageParcel &parcel);

    /**
     * @brief Read buffer info from MessageParcel.
     * @param parcel The MessageParcel that wirtten by remote buffer, refer to {@link MessageParcel}.
     * @param isSurfaceBuffer Whether the parcel was obtained directly through SurfaceBuffer's function, {@link
     * SurfaceBuffer}.
     * @return Whether the reading was successful.
     * @since 4.1
     */
    bool ReadFromMessageParcel(MessageParcel &parcel, bool isSurfaceBuffer = false);

    using MetaData = std::vector<uint8_t>;

    int64_t pts_;
    int64_t dts_;
    int64_t duration_;
    uint32_t flag_;
    std::shared_ptr<Meta> meta_;
    std::shared_ptr<AVMemory> memory_;

private:
    AVBuffer();
    Status Init(std::shared_ptr<AVAllocator> allocator, int32_t capacity = 0, int32_t align = 0);
    Status Init(uint8_t *ptr, int32_t capacity, int32_t size = 0);
    Status Init(sptr<SurfaceBuffer> surfaceBuffer);
    AVBufferConfig config_;
};

/**
 * @brief AVBuffer's memory.
 */
class __attribute__((visibility("default"))) AVMemory {
public:
    friend class AVBuffer;
    virtual ~AVMemory();
    /**
     * @brief Get the memory's types set by the allocator, refer to {@link MemoryType}
     * @return the memory's types if the memory is valid, otherwise {@link VIRTUAL_MEMORY}.
     * @since 4.1
     */
    virtual MemoryType GetMemoryType();

    /**
     * @brief Get the memory's Flag set by the allocator, refer to {@link MemoryType}
     * @return the memory's flag.
     * @since 4.1
     */
    virtual MemoryFlag GetMemoryFlag();

    /**
     * @brief Get the memory's capacity, which was set during creation and alloced by the allocator.
     * @return The memory's capacity, bytes. If the memory is valid, otherwise -1.
     * @since 4.1
     */
    virtual int32_t GetCapacity();

    /**
     * @brief Get the memory's used size.
     * @return The memory's size, bytes.
     * @since 4.1
     */
    virtual int32_t GetSize();

    /**
     * @brief Set the memory's used size.
     * @param size The memory's used size. If the size is greater than the capacity, it will be set to equal the
     * capacity.
     * @return Returns Status::OK if the execution is successful, otherwise returns a specific error code, refer to
     * {@link Status}
     * @since 4.1
     */
    virtual Status SetSize(int32_t size);

    /**
     * @brief Get the memory's used size.
     * @return The memory's used size, bytes.
     * @since 4.1
     */
    virtual int32_t GetOffset();

    /**
     * @brief Set the memory's offset.
     * @param offset The memory's offset, bytes.
     * @return Returns Status::OK if the execution is successful, otherwise returns a specific error code, refer to
     * {@link Status}
     * @since 4.1
     */
    virtual Status SetOffset(int32_t offset);

    /**
     * @brief Get the memory's file descriptor.
     * @return The memory's file descriptor. If the memory type is {@link SURFACE_MEMORY} or {@link VIRTUAL_MEMORY}, it
     * will return -1.
     * @since 4.1
     */
    virtual int32_t GetFileDescriptor();

    /**
     * @brief Get the memory's address.
     * @return The pointer of memory's address.
     * @since 4.1
     */
    virtual uint8_t *GetAddr();

    /**
     * @brief Writing data to memory.
     * @param in The pointer to the data being written.
     * @param writeSize The size of writing data, bytes.
     * @param position The position of writing data in memory, if equal to INVALID_POSITION, write continuously after
     * existing data, bytes.
     * @return The length of the actual written data.
     * @since 4.1
     */
    virtual int32_t Write(const uint8_t *in, int32_t writeSize, int32_t position = INVALID_POSITION);

    /**
     * @brief Reading data from memory.
     * @param out The pointer to save the read data.
     * @param readSize The size of reading data, bytes.
     * @param position The position of reading data in memory, if equal to INVALID_POSITION, read from begin, bytes.
     * @return The length of the actual read data.
     * @since 4.1
     */
    virtual int32_t Read(uint8_t *out, int32_t readSize, int32_t position = INVALID_POSITION);

    /**
     * @brief Set the memory's used size to zero.
     * @since 4.1
     */
    void Reset();

    /**
     * @brief Get the surface buffer of memory.
     * @return Returns the surface buffer if the memory type is {@link SURFACE_MEMORY},
     * otherwise returns nullptr.
     * @since 4.1
     */
    virtual sptr<SurfaceBuffer> GetSurfaceBuffer();

protected:
    AVMemory();
    virtual Status Init();
    virtual Status Init(MessageParcel &parcel);
    virtual Status InitSurfaceBuffer(MessageParcel &parcel);
    virtual Status InitSurfaceBuffer(sptr<SurfaceBuffer> surfaceBuffer);
    virtual bool WriteToMessageParcel(MessageParcel &parcel);
    virtual bool ReadFromMessageParcel(MessageParcel &parcel);

    bool ReadCommonFromMessageParcel(MessageParcel &parcel);
    bool SkipCommonFromMessageParcel(MessageParcel &parcel);
    bool WriteCommonToMessageParcel(MessageParcel &parcel);

    int32_t capacity_ = 0;
    int32_t align_;

    int32_t offset_;
    int32_t size_;
    uint8_t *base_;
    uint64_t uid_;
    std::shared_ptr<AVAllocator> allocator_;

private:
    static std::shared_ptr<AVMemory> CreateAVMemory(std::shared_ptr<AVAllocator> allocator,
                                                    int32_t capacity = 0, int32_t align = 0);
    static std::shared_ptr<AVMemory> CreateAVMemory(uint8_t *ptr, int32_t capacity, int32_t size);
    static std::shared_ptr<AVMemory> CreateAVMemory(MessageParcel &parcel, bool isSurfaceBuffer = false);
    static std::shared_ptr<AVMemory> CreateAVMemory(sptr<SurfaceBuffer> surfaceBuffer);
};
} // namespace Media
} // namespace OHOS
#endif // AVBUFFER_H