/*
 * Copyright (C) 2024 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 IMAGE_EFFECT_NATIVE_BASE_H
#define IMAGE_EFFECT_NATIVE_BASE_H

#include <memory>
#include <utility>
#include <unordered_set>

#include "efilter.h"
#include "image_effect_inner.h"
#include "image_effect_filter.h"

struct OH_EffectFilter {
    std::shared_ptr<OHOS::Media::Effect::EFilter> filter_ = nullptr;
    void SetParameter(const std::string &key, OHOS::Media::Plugin::Any &param);
    OHOS::Media::Effect::ErrorCode GetParameter(const std::string &key, OHOS::Media::Plugin::Any &param);
    void RemoveParameter(const std::string &key);
    bool isCreatedBySystem_ = false;
private:
    std::unordered_map<std::string, OHOS::Media::Plugin::Any&> params_;
};

struct OH_ImageEffect {
    OH_ImageEffect() = default;
    std::shared_ptr<OHOS::Media::Effect::ImageEffect> imageEffect_ = nullptr;
    char *saveJson = nullptr;
    std::vector<std::pair<OH_EffectFilter *, std::string>> filters_;
    ~OH_ImageEffect()
    {
        if (saveJson != nullptr) {
            delete saveJson;
            saveJson = nullptr;
        }
        for (const auto &filter : filters_) {
            auto ohEFilter = filter.first;
            if (ohEFilter != nullptr && ohEFilter->isCreatedBySystem_) {
                OH_EffectFilter_Release(filter.first);
            }
        }
        filters_.clear();
    }
};

struct OH_EffectFilterInfo {
    OH_EffectFilterInfo() = default;
    ~OH_EffectFilterInfo();
    std::string filterName = "";
    std::unordered_set<ImageEffect_BufferType> supportedBufferTypes;
    std::unordered_set<ImageEffect_Format> supportedFormats;

    ImageEffect_BufferType *effectBufferType = nullptr;
    uint32_t bufferTypeArraySize = 0;
    ImageEffect_Format *effectFormat = nullptr;
    uint32_t formatArraySize = 0;
};

struct OH_EffectBufferInfo {
    void *addr = nullptr;
    int32_t width = 0;
    int32_t height = 0;
    int32_t rowSize = 0;
    ImageEffect_Format format = ImageEffect_Format::EFFECT_PIXEL_FORMAT_UNKNOWN;
    int64_t timestamp = 0;
};

#endif // IMAGE_EFFECT_NATIVE_BASE_H