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

#include <cstdint>

#include "camera_device_ability_items.h"
#include "camera_log.h"
#include "camera_metadata_operator.h"
#include "display_manager.h"
#include "camera_util.h"
#include "hstream_common.h"
#include "ipc_skeleton.h"
#include "istream_depth_data_callback.h"
#include "metadata_utils.h"
#include "camera_report_uitls.h"

namespace OHOS {
namespace CameraStandard {
using namespace OHOS::HDI::Camera::V1_0;
HStreamDepthData::HStreamDepthData(
    sptr<OHOS::IBufferProducer> producer, int32_t format, int32_t width, int32_t height)
    : HStreamCommon(StreamType::DEPTH, producer, format, width, height)
{
    MEDIA_INFO_LOG("HStreamDepthData::HStreamDepthData construct, format:%{public}d, size:%{public}dx%{public}d, "
        "streamId:%{public}d",
        format, width, height, GetFwkStreamId());
}

HStreamDepthData::~HStreamDepthData()
{
    MEDIA_INFO_LOG("HStreamDepthData::~HStreamDepthData deconstruct, format:%{public}d size:%{public}dx%{public}d "
                   "streamId:%{public}d, hdiStreamId:%{public}d",
        format_, width_, height_, GetFwkStreamId(), GetHdiStreamId());
}

int32_t HStreamDepthData::LinkInput(wptr<OHOS::HDI::Camera::V1_0::IStreamOperator> streamOperator,
    std::shared_ptr<OHOS::Camera::CameraMetadata> cameraAbility)
{
    // LCOV_EXCL_START
    MEDIA_INFO_LOG("HStreamDepthData::LinkInput streamId:%{public}d", GetFwkStreamId());
    int32_t ret = HStreamCommon::LinkInput(streamOperator, cameraAbility);
    CHECK_RETURN_RET_ELOG(ret != CAMERA_OK, ret, "HStreamDepthData::LinkInput err, streamId:%{public}d ,err:%{public}d",
        GetFwkStreamId(), ret);
    return CAMERA_OK;
    // LCOV_EXCL_STOP
}

void HStreamDepthData::SetStreamInfo(StreamInfo_V1_5& streamInfo)
{
    // LCOV_EXCL_START
    HStreamCommon::SetStreamInfo(streamInfo);
    streamInfo.v1_0.intent_ =
        static_cast<OHOS::HDI::Camera::V1_0::StreamIntent>(OHOS::HDI::Camera::V1_3::StreamType::STREAM_TYPE_DEPTH);
    // LCOV_EXCL_STOP
}

int32_t HStreamDepthData::SetDataAccuracy(int32_t accuracy)
{
    // LCOV_EXCL_START
    MEDIA_INFO_LOG("HStreamDepthData::SetDataAccuracy accuracy: %{public}d", accuracy);
    streamDepthDataAccuracy_ = {accuracy};
    std::vector<uint8_t> ability;
    std::vector<uint8_t> depthSettings;
    {
        std::lock_guard<std::mutex> lock(cameraAbilityLock_);
        OHOS::Camera::MetadataUtils::ConvertMetadataToVec(cameraAbility_, ability);
        std::shared_ptr<OHOS::Camera::CameraMetadata> dynamicSetting = nullptr;
        OHOS::Camera::MetadataUtils::ConvertVecToMetadata(ability, dynamicSetting);
        if (dynamicSetting == nullptr) {
            dynamicSetting = std::make_shared<OHOS::Camera::CameraMetadata>(0, 0);
        }
        camera_metadata_item_t item;
        CHECK_RETURN_RET_ELOG(dynamicSetting == nullptr, CAMERA_INVALID_ARG,
            "HStreamDepthData::SetDataAccuracy dynamicSetting is nullptr.");
        int ret = OHOS::Camera::FindCameraMetadataItem(dynamicSetting->get(), OHOS_CONTROL_DEPTH_DATA_ACCURACY, &item);
        bool status = false;
        if (ret == CAM_META_ITEM_NOT_FOUND) {
            MEDIA_DEBUG_LOG("HStreamDepthData::SetDataAccuracy Failed to find data accuracy");
            status = dynamicSetting->addEntry(
                OHOS_CONTROL_DEPTH_DATA_ACCURACY, streamDepthDataAccuracy_.data(), streamDepthDataAccuracy_.size());
        } else if (ret == CAM_META_SUCCESS) {
            MEDIA_DEBUG_LOG("HStreamDepthData::SetDataAccuracy success to find data accuracy");
            status = dynamicSetting->updateEntry(
                OHOS_CONTROL_DEPTH_DATA_ACCURACY, streamDepthDataAccuracy_.data(), streamDepthDataAccuracy_.size());
        }
        CHECK_PRINT_ELOG(!status, "HStreamDepthData::SetDataAccuracy Failed to set data accuracy");
        OHOS::Camera::MetadataUtils::ConvertMetadataToVec(dynamicSetting, depthSettings);
    }

    auto streamOperator = GetStreamOperator();

    CamRetCode rc = HDI::Camera::V1_0::NO_ERROR;
    if (streamOperator != nullptr) {
        std::lock_guard<std::mutex> startStopLock(streamStartStopLock_);
        CaptureInfo captureInfo;
        captureInfo.streamIds_ = {GetHdiStreamId()};
        captureInfo.captureSetting_ = depthSettings;
        captureInfo.enableShutterCallback_ = false;
        int32_t currentCaptureId = GetPreparedCaptureId();
        MEDIA_INFO_LOG("HStreamDepthData::SetDataAccuracy stream:%{public}d, with settingCapture ID:%{public}d",
            GetFwkStreamId(), currentCaptureId);
        rc = (CamRetCode)(streamOperator->Capture(currentCaptureId, captureInfo, true));
        CHECK_PRINT_ELOG(rc != HDI::Camera::V1_0::NO_ERROR,
            "HStreamDepthData::SetDataAccuracy Failed with error Code:%{public}d", rc);
    }
    return rc;
    // LCOV_EXCL_STOP
}

int32_t HStreamDepthData::Start()
{
    CAMERA_SYNC_TRACE;
    auto streamOperator = GetStreamOperator();
    CHECK_RETURN_RET(streamOperator == nullptr, CAMERA_INVALID_STATE);
    // LCOV_EXCL_START
    auto preparedCaptureId = GetPreparedCaptureId();
    CHECK_RETURN_RET_ELOG(preparedCaptureId != CAPTURE_ID_UNSET, CAMERA_INVALID_STATE,
        "HStreamDepthData::Start, Already started with captureID: %{public}d", preparedCaptureId);

    int32_t ret = PrepareCaptureId();
    preparedCaptureId = GetPreparedCaptureId();
    CHECK_RETURN_RET_ELOG(ret != CAMERA_OK || preparedCaptureId == CAPTURE_ID_UNSET, ret,
        "HStreamDepthData::Start Failed to allocate a captureId");

    std::vector<uint8_t> ability;
    {
        std::lock_guard<std::mutex> lock(cameraAbilityLock_);
        OHOS::Camera::MetadataUtils::ConvertMetadataToVec(cameraAbility_, ability);
    }

    CaptureInfo captureInfo;
    captureInfo.streamIds_ = { GetHdiStreamId() };
    captureInfo.captureSetting_ = ability;
    captureInfo.enableShutterCallback_ = false;
    MEDIA_INFO_LOG("HStreamDepthData::Start streamId:%{public}d hdiStreamId:%{public}d With capture ID: %{public}d",
        GetFwkStreamId(), GetHdiStreamId(), preparedCaptureId);

    std::lock_guard<std::mutex> startStopLock(streamStartStopLock_);
    HStreamCommon::PrintCaptureDebugLog(cameraAbility_);
    CamRetCode rc = (CamRetCode)(streamOperator->Capture(preparedCaptureId, captureInfo, true));
    if (rc != HDI::Camera::V1_0::NO_ERROR) {
        ResetCaptureId();
        MEDIA_ERR_LOG("HStreamDepthData::Start Failed with error Code:%{public}d", rc);
        CameraReportUtils::ReportCameraError(
            "HStreamDepthData::Start", rc, true, CameraReportUtils::GetCallerInfo());
        ret = HdiToServiceError(rc);
    } else {
        depthDataStreamStatus_ = DepthDataStreamStatus::STARTED;
    }

    return ret;
    // LCOV_EXCL_STOP
}

int32_t HStreamDepthData::Stop()
{
    CAMERA_SYNC_TRACE;
    auto streamOperator = GetStreamOperator();
    CHECK_RETURN_RET_ILOG(
        streamOperator == nullptr, CAMERA_INVALID_STATE, "HStreamDepthData::Stop streamOperator is null");
    // LCOV_EXCL_START
    auto preparedCaptureId = GetPreparedCaptureId();
    MEDIA_INFO_LOG("HStreamDepthData::Start streamId:%{public}d hdiStreamId:%{public}d With capture ID: %{public}d",
        GetFwkStreamId(), GetHdiStreamId(), preparedCaptureId);
    CHECK_RETURN_RET_ELOG(
        preparedCaptureId == CAPTURE_ID_UNSET, CAMERA_INVALID_STATE, "HStreamDepthData::Stop, Stream not started yet");
    int32_t ret = CAMERA_OK;
    {
        std::lock_guard<std::mutex> startStopLock(streamStartStopLock_);
        ret = StopStream();
        if (ret != CAMERA_OK) {
            MEDIA_ERR_LOG("HStreamDepthData::Stop Failed with errorCode:%{public}d, curCaptureID_: %{public}d",
                          ret, preparedCaptureId);
        } else {
            depthDataStreamStatus_ = DepthDataStreamStatus::STOPED;
        }
    }
    return ret;
    // LCOV_EXCL_STOP
}

int32_t HStreamDepthData::Release()
{
    return ReleaseStream(false);
}

int32_t HStreamDepthData::ReleaseStream(bool isDelay)
{
    {
        std::lock_guard<std::mutex> lock(callbackLock_);
        streamDepthDataCallback_ = nullptr;
    }
    return HStreamCommon::ReleaseStream(isDelay);
}

int32_t HStreamDepthData::SetCallback(const sptr<IStreamDepthDataCallback>& callback)
{
    // LCOV_EXCL_START
    CHECK_RETURN_RET_ELOG(callback == nullptr, CAMERA_INVALID_ARG, "HStreamDepthData::SetCallback callback is null");
    std::lock_guard<std::mutex> lock(callbackLock_);
    streamDepthDataCallback_ = callback;
    return CAMERA_OK;
    // LCOV_EXCL_STOP
}

int32_t HStreamDepthData::UnSetCallback()
{
    // LCOV_EXCL_START
    std::lock_guard<std::mutex> lock(callbackLock_);
    streamDepthDataCallback_ = nullptr;
    return CAMERA_OK;
    // LCOV_EXCL_STOP
}

int32_t HStreamDepthData::OnDepthDataError(int32_t errorType)
{
    // LCOV_EXCL_START
    std::lock_guard<std::mutex> lock(callbackLock_);
    if (streamDepthDataCallback_ != nullptr) {
        int32_t depthDataErrorCode;
        if (errorType == BUFFER_LOST) {
            depthDataErrorCode = CAMERA_STREAM_BUFFER_LOST;
        } else {
            depthDataErrorCode = CAMERA_UNKNOWN_ERROR;
        }
        CAMERA_SYSEVENT_FAULT(CreateMsg("Depth OnDepthDataError! errorCode:%d", depthDataErrorCode));
        streamDepthDataCallback_->OnDepthDataError(depthDataErrorCode);
    }
    return CAMERA_OK;
    // LCOV_EXCL_STOP
}

void HStreamDepthData::DumpStreamInfo(CameraInfoDumper& infoDumper)
{
    // LCOV_EXCL_START
    infoDumper.Title("depth stream");
    HStreamCommon::DumpStreamInfo(infoDumper);
    // LCOV_EXCL_STOP
}

int32_t HStreamDepthData::OperatePermissionCheck(uint32_t interfaceCode)
{
    // LCOV_EXCL_START
    switch (static_cast<IStreamDepthDataIpcCode>(interfaceCode)) {
        case IStreamDepthDataIpcCode::COMMAND_START: {
            auto callerToken = IPCSkeleton::GetCallingTokenID();
            CHECK_RETURN_RET_ELOG(callerToken_ != callerToken, CAMERA_OPERATION_NOT_ALLOWED,
                "HStreamDepthData::OperatePermissionCheck fail, callerToken not legal");
            break;
        }
        default:
            break;
    }
    return CAMERA_OK;
    // LCOV_EXCL_STOP
}

int32_t HStreamDepthData::CallbackEnter([[maybe_unused]] uint32_t code)
{
    // LCOV_EXCL_START
    MEDIA_DEBUG_LOG("start, code:%{public}u", code);
    DisableJeMalloc();
    return OperatePermissionCheck(code);
    // LCOV_EXCL_STOP
}
 
int32_t HStreamDepthData::CallbackExit([[maybe_unused]] uint32_t code, [[maybe_unused]] int32_t result)
{
    // LCOV_EXCL_START
    MEDIA_DEBUG_LOG("leave, code:%{public}u, result:%{public}d", code, result);
    return CAMERA_OK;
    // LCOV_EXCL_STOP
}

} // namespace CameraStandard
} // namespace OHOS