/*
 * Copyright 2021 Rockchip Electronics 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.
 */

#define MODULE_TAG "mpp_lock"

#include "mpp_env.h"
#include "mpp_log.h"
#include "mpp_lock.h"
#include "mpp_time.h"

#define LOCK_IDLE 0
#define LOCK_BUSY 1

void mpp_spinlock_init(spinlock_t *lock)
{
 MPP_SYNC_CLR(&lock->lock);
 lock->count = 0;
 lock->time = 0;

 mpp_env_get_u32("mpp_lock_debug", &lock->debug, 0);
}

void mpp_spinlock_lock(spinlock_t *lock)
{
 RK_S64 time = 0;

 if (lock->debug)
 time = mpp_time();

 while (!MPP_BOOL_CAS(&lock->lock, LOCK_IDLE, LOCK_BUSY)) {
 asm("NOP");
 asm("NOP");
 }

 if (lock->debug && time) {
 lock->time += mpp_time() - time;
 lock->count++;
 }
}

void mpp_spinlock_deinit(spinlock_t *lock, const char *name)
{
 if (lock->debug && lock->count) {
 mpp_log("lock %s lock %lld times take time %lld avg %d", name,
 lock->count, lock->time, (RK_S32)(lock->time / lock->count));
 }
}

void mpp_spinlock_unlock(spinlock_t *lock)
{
 MPP_SYNC_CLR(&lock->lock);
}

bool mpp_spinlock_trylock(spinlock_t *lock)
{
 RK_S64 time = 0;
 bool ret;

 if (lock->debug)
 time = mpp_time();

 ret = MPP_BOOL_CAS(&lock->lock, LOCK_IDLE, LOCK_BUSY);

 if (ret && lock->debug && time) {
 lock->time += mpp_time() - time;
 lock->count++;
 }

 return ret;
}