/*
 * 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 "vsync_controller.h"

namespace OHOS {
namespace Rosen {
VSyncController::VSyncController(const sptr<VSyncGenerator> &geng, int64_t offset)
    : generator_(geng), callbackMutex_(), callback_(nullptr),
    offsetMutex_(), phaseOffset_(offset), enabled_(false)
{
}

VSyncController::~VSyncController()
{
}

VsyncError VSyncController::SetEnable(bool enbale)
{
    if (generator_ == nullptr) {
        return VSYNC_ERROR_NULLPTR;
    }
    const sptr<VSyncGenerator> generator = generator_.promote();
    if (generator == nullptr) {
        return VSYNC_ERROR_NULLPTR;
    }
    std::lock_guard<std::mutex> locker(offsetMutex_);
    VsyncError ret = VSYNC_ERROR_OK;
    if (enbale) {
        ret = generator->AddListener(phaseOffset_, this);
    } else {
        ret = generator->RemoveListener(this);
    }
    enabled_ = enbale;
    return ret;
}

VsyncError VSyncController::SetCallback(Callback *cb)
{
    if (cb == nullptr) {
        return VSYNC_ERROR_NULLPTR;
    }
    std::lock_guard<std::mutex> locker(callbackMutex_);
    callback_ = cb;
    return VSYNC_ERROR_OK;
}

VsyncError VSyncController::SetPhaseOffset(int64_t offset)
{
    if (generator_ == nullptr) {
        return VSYNC_ERROR_NULLPTR;
    }
    const sptr<VSyncGenerator> generator = generator_.promote();
    if (generator == nullptr) {
        return VSYNC_ERROR_NULLPTR;
    }
    std::lock_guard<std::mutex> locker(offsetMutex_);
    phaseOffset_ = offset;
    return generator->ChangePhaseOffset(this, phaseOffset_);
}

void VSyncController::OnVSyncEvent(int64_t now)
{
    Callback *cb = nullptr;
    {
        std::lock_guard<std::mutex> locker(callbackMutex_);
        cb = callback_;
    }
    if (cb != nullptr) {
        cb->OnVSyncEvent(now);
    }
}
}
}