/*
 * Copyright (c) 2023 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file expected 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 UT_COMMON_H
#define UT_COMMON_H

#include <stdlib.h>
#include <thread>
#include <iostream>
#include <unistd.h>
#include <gtest/gtest.h>
#include <sys/time.h>
#include <map>
#include <vector>
#include <fcntl.h>
#include "camera.h"
#include "v1_0/types.h"
#include "metadata_utils.h"
#include "v1_0/icamera_host.h"
#include "v1_0/icamera_device.h"
#include "v1_0/istream_operator.h"
#include "v1_0/camera_host_proxy.h"
#include "v1_0/ioffline_stream_operator.h"
#include "display_format.h"
#include "iconsumer_surface.h"

namespace OHOS::Camera {
enum CameraUtConstants {
    UT_SLEEP_TIME = 2,
    UT_SECOND_TIMES,
    UT_TUNNEL_MODE = 5,
    UT_DATA_SIZE = 8,
};

enum ImageDataSaveSwitch {
    SWITCH_OFF,
    SWITCH_ON,
};
using namespace OHOS::HDI::Camera::V1_0;
class Test {
public:
    void Init();
    void Open();
    void Close();
    void GetCameraMetadata();
    void StartStream(std::vector<StreamIntent> intents);
    void DefaultPreview(std::shared_ptr<StreamInfo> &infos);
    void DefaultCapture(std::shared_ptr<StreamInfo> &infos);
    void DefaultInfosPreview(std::shared_ptr<StreamInfo> &infos);
    void DefaultInfosCapture(std::shared_ptr<StreamInfo> &infos);
    void DefaultInfosVideo(std::shared_ptr<StreamInfo> &infos);
    void DefaultInfosAnalyze(std::shared_ptr<StreamInfo> &infos);
    uint64_t GetCurrentLocalTimeStamp();
    int32_t DumpImageFile(int streamId, std::string suffix, const void* buffer, int32_t size);
    void StartCapture(int streamId, int captureId, bool shutterCallback, bool isStreaming);
    void StopStream(std::vector<int>& captureIds, std::vector<int>& streamIds);
    OHOS::sptr<OHOS::Camera::ICameraHost> service = nullptr;
    OHOS::sptr<ICameraDevice> cameraDevice = nullptr;
    OHOS::sptr<IStreamOperatorCallback> streamOperatorCallback = nullptr;
    OHOS::sptr<ICameraHostCallback> hostCallback = nullptr;
    OHOS::sptr<IStreamOperator> streamOperator = nullptr;
    OHOS::sptr<ICameraDeviceCallback> deviceCallback = nullptr;
    std::vector<StreamInfo> streamInfos;
    std::shared_ptr<StreamInfo> streamInfo = nullptr;
    std::shared_ptr<StreamInfo> streamInfoSnapshot = nullptr;
    std::shared_ptr<StreamInfo> streamInfoCapture = nullptr;
    std::shared_ptr<StreamInfo> streamInfoAnalyze = nullptr;
    std::shared_ptr<StreamInfo> streamInfoPre = nullptr;
    std::shared_ptr<StreamInfo> streamInfoVideo = nullptr;
    std::shared_ptr<CaptureInfo> captureInfo = nullptr;
    int previewFormat = PIXEL_FMT_YCRCB_420_SP;
    int videoFormat = PIXEL_FMT_YCRCB_420_SP;
    int snapshotFormat = PIXEL_FMT_YCRCB_420_SP;
    int analyzeFormat = PIXEL_FMT_YCRCB_420_SP;
    int videoEncodeType = ENCODE_TYPE_H265;
    int streamIdPreview = 100;
    int streamIdCapture = 101;
    int captureWidth = 1280;
    int captureHeight = 960;
    int captureIdPreview = 2000;
    int previewWidth = 1920;
    int previewHeight = 1080;
    int captureIdCapture = 2010;
    int captureIdVideo = 2020;
    int streamIdVideo = 102;
    int videoHeight = 1080;
    int videoWidth = 1920;
    int analyzeWidth = 1920;
    int analyzeHeight = 1080;
    int snapshotWidth = 4160;
    int snapshotHeight = 3120;
    int streamIdAnalyze = 103;
    int usbCamera_previewWidth = 640;
    int usbCamera_previewHeight = 480;
    int usbCamera_videoWidth = 1280;
    int usbCamera_videoHeight = 960;
    int usbCamera_captureWidth = 1280;
    int usbCamera_captureHeight = 960;
    int usbCamera_analyzeWidth = 640;
    int usbCamera_analyzeHeight = 480;
    int usbCamera_previewFormat = PIXEL_FMT_RGBA_8888;
    int usbCamera_videoFormat = PIXEL_FMT_YCRCB_420_SP;
    int usbCamera_snapshotFormat = PIXEL_FMT_RGBA_8888;
    int usbCamera_analyzeFormat = PIXEL_FMT_RGBA_8888;
    int usbCamera_videoEncodeType = ENCODE_TYPE_H264;
    std::vector<int> captureIds;
    std::vector<int> streamIds;
    int32_t imageDataSaveSwitch = SWITCH_OFF;

    int32_t rc;
    bool status;
    std::vector<std::string> cameraIds;
    std::vector<uint8_t> abilityVec = {};
    std::shared_ptr<CameraMetadata> ability = nullptr;
    std::vector<StreamIntent> intents;
    class StreamConsumer;
    std::map<OHOS::Camera::StreamIntent, std::shared_ptr<StreamConsumer>> consumerMap_ = {};

    class TestBufferConsumerListener : public OHOS::IBufferConsumerListener {
    public:
        TestBufferConsumerListener() {}
        ~TestBufferConsumerListener() {}
        void OnBufferAvailable()
        {
            hasAvailablebuffer = true;
        }
        bool checkBufferAvailable()
        {
            if (hasAvailablebuffer) {
                hasAvailablebuffer = false;
                return true;
            }
            return false;
        }
    private:
        bool hasAvailablebuffer = false;
    };

    class StreamConsumer {
    public:
        void CalculateFps(int64_t timestamp, int32_t streamId);
        OHOS::sptr<OHOS::IBufferProducer> CreateProducer(std::function<void(void*, uint32_t)> callback);
        OHOS::sptr<BufferProducerSequenceable> CreateProducerSeq(std::function<void(void*, uint32_t)> callback);
        void TakeSnapshoe()
        {
            shotCount_++;
        }
        void WaitSnapshotEnd()
        {
            std::cout << "ready to wait" << std::endl;
            std::unique_lock<std::mutex> l(l_);
            cv_.wait(l, [this]() {return shotCount_ == 0; });
        }
        ~StreamConsumer()
        {
            running_ = false;
            if (consumerThread_ != nullptr) {
                consumerThread_->join();
                delete consumerThread_;
            }
        }
    public:
        std::atomic<uint64_t> shotCount_ = 0;
        std::mutex l_;
        std::condition_variable cv_;
        bool running_ = true;
        OHOS::sptr<OHOS::IConsumerSurface> consumer_ = nullptr;
        std::thread* consumerThread_ = nullptr;
        std::function<void(void*, uint32_t)> callback_ = nullptr;
        bool isFirstCalculateFps_ = false;
        int timestampCount_ = 0;
        int64_t intervalTimestamp_ = 0;
        const int64_t ONESECOND_OF_MICROSECOND_UNIT = 1000000000;
        int64_t interval_ = ONESECOND_OF_MICROSECOND_UNIT;
    };

    class DemoCameraDeviceCallback : public ICameraDeviceCallback {
    public:
        DemoCameraDeviceCallback() = default;
        virtual ~DemoCameraDeviceCallback() = default;

        int32_t OnError(ErrorType type, int32_t errorMsg) override;
        int32_t OnResult(uint64_t timestamp, const std::vector<uint8_t> &result) override;
    };

    using ResultCallback = std::function<void (uint64_t, const std::shared_ptr<CameraMetadata>)>;
    static ResultCallback resultCallback_;

    class TestStreamOperatorCallback : public IStreamOperatorCallback {
    public:
        TestStreamOperatorCallback() = default;
        virtual ~TestStreamOperatorCallback() = default;
        int32_t OnCaptureStarted(int32_t captureId, const std::vector<int32_t> &streamId) override;
        int32_t OnCaptureEnded(int32_t captureId, const std::vector<CaptureEndedInfo> &infos) override;
        int32_t OnCaptureError(int32_t captureId, const std::vector<CaptureErrorInfo> &infos) override;
        int32_t OnFrameShutter(int32_t captureId, const std::vector<int32_t> &streamIds, uint64_t timestamp) override;
    };

    class TestCameraHostCallback : public ICameraHostCallback {
    public:
        TestCameraHostCallback() = default;
        virtual ~TestCameraHostCallback() = default;

        int32_t OnCameraStatus(const std::string& cameraId, CameraStatus status) override;
        int32_t OnFlashlightStatus(const std::string& cameraId, FlashlightStatus status) override;
        int32_t OnCameraEvent(const std::string& cameraId, CameraEvent event) override;
    };
};
}
#endif