/*
 * Copyright (c) 2021 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.
 */

#include "stream_customer.h"
#include "video_key_info.h"

namespace OHOS::Camera {
StreamCustomer::StreamCustomer() {}
StreamCustomer::~StreamCustomer() {}

void StreamCustomer::CamFrame(const std::function<void(void*, uint32_t)> callback)
{
    CAMERA_LOGD("demo test:enter CamFrame thread ++ ");
#ifdef CAMERA_BUILT_ON_OHOS_LITE
    do {
        OHOS::SurfaceBuffer* buffer = consumer_->AcquireBuffer();
        if (buffer != nullptr) {
            if (callback != nullptr) {
                void* addr = buffer->GetVirAddr();
                int32_t size = 0;
                buffer->GetInt32(VIDEO_KEY_INFO_DATA_SIZE, size);
                callback(addr, size);
            }
            consumer_->ReleaseBuffer(buffer);
        }
    } while (camFrameExit_ == 0);
#else
    OHOS::Rect damage;
    int32_t flushFence = 0;
    int64_t timestamp = 0;
    constexpr uint32_t delayTime = 12000;

    do {
        OHOS::sptr<OHOS::SurfaceBuffer> buff = nullptr;
        consumer_->AcquireBuffer(buff, flushFence, timestamp, damage);
        if (buff != nullptr) {
            void* addr = buff->GetVirAddr();
            if (callback != nullptr) {
                int32_t gotSize = 0;
                int32_t frameNum = 0;
                int isKey = 0;
                timestamp = 0;
                buff->GetExtraData()->ExtraGet(OHOS::Camera::dataSize, gotSize);
                buff->GetExtraData()->ExtraGet(OHOS::Camera::isKeyFrame, isKey);
                buff->GetExtraData()->ExtraGet(OHOS::Camera::timeStamp, timestamp);
                CAMERA_LOGE("demo test:CamFrame callback +++++++ Size == %d frameNum = %d timestamp == %lld\n",
                    gotSize, frameNum, timestamp);
                callback(addr, gotSize);
            }
            consumer_->ReleaseBuffer(buff, -1);
        }
        usleep(delayTime);
    } while (camFrameExit_ == 0);
#endif

    CAMERA_LOGD("demo test:Exiting CamFrame thread -- ");
}

#ifdef CAMERA_BUILT_ON_OHOS_LITE
std::shared_ptr<OHOS::Surface> StreamCustomer::CreateProducer()
#else
sptr<OHOS::IBufferProducer> StreamCustomer::CreateProducer()
#endif
{
#ifdef CAMERA_BUILT_ON_OHOS_LITE
    consumer_ = std::shared_ptr<OHOS::Surface>(OHOS::Surface::CreateSurface());
    if (consumer_ == nullptr) {
        return nullptr;
    }
    return consumer_;
#else
    consumer_ = OHOS::IConsumerSurface::Create();
    if (consumer_ == nullptr) {
        return nullptr;
    }
    sptr<IBufferConsumerListener> listener = new(std::nothrow) TestBuffersConsumerListener();
    consumer_->RegisterConsumerListener(listener);

    auto producer = consumer_->GetProducer();
    if (producer == nullptr) {
        return nullptr;
    }

    CAMERA_LOGI("demo test, create a buffer queue producer");
    return producer;
#endif
}

RetCode StreamCustomer::ReceiveFrameOn(const std::function<void(void*, uint32_t)> callback)
{
    CAMERA_LOGD("demo test:ReceiveFrameOn enter");

    if (camFrameExit_ == 1) {
        camFrameExit_ = 0;
        previewThreadId_ = new (std::nothrow) std::thread(&StreamCustomer::CamFrame, this, callback);
        if (previewThreadId_ == nullptr) {
            CAMERA_LOGE("demo test:ReceiveFrameOn failed");
            return OHOS::Camera::RC_ERROR;
        }
    } else {
        CAMERA_LOGD("demo test:ReceiveFrameOn loop thread is running");
    }
    CAMERA_LOGD("demo test:ReceiveFrameOn exit");

    return RC_OK;
}

void StreamCustomer::ReceiveFrameOff()
{
    CAMERA_LOGD("demo test:ReceiveFrameOff enter");

    if (camFrameExit_ == 0) {
        camFrameExit_ = 1;
        if (previewThreadId_ != nullptr) {
            previewThreadId_->join();
            delete previewThreadId_;
            previewThreadId_ = nullptr;
        }
    }

    CAMERA_LOGD("demo test:ReceiveFrameOff exit");
}
}