/*
 * 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 NATIVE_AVMAGIC_H
#define NATIVE_AVMAGIC_H

#include <refbase.h>
#include "avcodec_info.h"
#include "buffer/avsharedmemory.h"
#include "avcodec_common.h"
#include "meta/format.h"
#include "native_avcapability.h"


#define AV_MAGIC(a, b, c, d) (((a) << 24) + ((b) << 16) + ((c) << 8) + ((d) << 0))

enum class AVMagic {
    AVCODEC_MAGIC_VIDEO_DECODER = AV_MAGIC('V', 'D', 'E', 'C'),
    AVCODEC_MAGIC_VIDEO_ENCODER = AV_MAGIC('V', 'E', 'N', 'C'),
    AVCODEC_MAGIC_AUDIO_DECODER = AV_MAGIC('A', 'D', 'E', 'C'),
    AVCODEC_MAGIC_AUDIO_ENCODER = AV_MAGIC('A', 'E', 'N', 'C'),
    AVCODEC_MAGIC_AVMUXER = AV_MAGIC('M', 'U', 'X', 'R'),
    AVCODEC_MAGIC_AVDEMUXER = AV_MAGIC('D', 'M', 'U', 'X'),
    AVCODEC_MAGIC_AVSOURCE = AV_MAGIC('S', 'O', 'U', 'C'),
    AVCODEC_MAGIC_AVCAPABILITY = AV_MAGIC('C', 'A', 'P', 'A'),
    AVCODEC_MAGIC_PRIMARY_VIDEO_ENCODER = AV_MAGIC('P', 'V', 'E', 'N'),
    AVCODEC_MAGIC_SECONDARY_VIDEO_ENCODER = AV_MAGIC('S', 'V', 'E', 'N'),
    AVCODEC_MAGIC_INVALID_ENTITY = AV_MAGIC('I', 'N', 'V', 'E'),
};

struct AVObjectMagic : public OHOS::RefBase {
    explicit AVObjectMagic(enum AVMagic m) : magic_(m) {}
    virtual ~AVObjectMagic() = default;
    enum AVMagic magic_;
};
struct OH_AVCodec : public AVObjectMagic {
    explicit OH_AVCodec(enum AVMagic m) : AVObjectMagic(m) {}
    virtual ~OH_AVCodec() = default;
};

struct OH_AVCapability : public OHOS::RefBase {
    OH_AVCapability();
    ~OH_AVCapability() override;
    OHOS::MediaAVCodec::CapabilityData *capabilityData_;
    OH_AVRange *sampleRateRanges_ = nullptr;
    enum AVMagic magic_;
};

struct OH_AVMuxer : public AVObjectMagic {
    explicit OH_AVMuxer(enum AVMagic m) : AVObjectMagic(m) {}
    virtual ~OH_AVMuxer() = default;
};

struct OH_AVDemuxer : public AVObjectMagic {
    explicit OH_AVDemuxer(enum AVMagic m) : AVObjectMagic(m) {}
    virtual ~OH_AVDemuxer() = default;
};

struct OH_AVSource : public AVObjectMagic {
    explicit OH_AVSource(enum AVMagic m) : AVObjectMagic(m) {}
    virtual ~OH_AVSource() = default;
};

#endif // NATIVE_AVMAGIC_H