/* Copyright (c) 2014-2016, 2018, The Linux Foundation. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are
 * met:
 *   * Redistributions of source code must retain the above copyright
 *     notice, this list of conditions and the following disclaimer.
 *   * Redistributions in binary form must reproduce the above
 *     copyright notice, this list of conditions and the following
 *     disclaimer in the documentation and/or other materials provided
 *     with the distribution.
 *   * Neither the name of The Linux Foundation, Inc. nor the names of its
 *     contributors may be used to endorse or promote products derived
 *     from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESS OR IMPLIED
 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS
 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
 * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
 * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
 * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <stdlib.h>
#include <kernel/timer.h>
#include <kernel/thread.h>
#include <pm_vib.h>
#include <vibrator.h>
#include <pm8x41.h>
#include <platform.h>
#include <smem.h>
#include <target.h>

#define CHECK_VIB_TIMER_FREQUENCY    50

/*
 * USE_VIB_THREAD is a micro that use a thread to turn off the vibrator,
 * USE_VIB_THREAD should be define when use ldo to turn on/off vibrator.
 * Note: define USE_VIB_THREAD will reduce the accuracy of vib time.
 */
#if !USE_VIB_THREAD
static struct timer vib_timer;
#else
static uint32_t vib_time;
#endif
static uint32_t vib_timeout = 1;

/* Function to turn on vibrator */
void vib_turn_on()
{
	uint32_t pmic = target_get_pmic();

	if (pmic == PMIC_IS_PM8916)
		pm8x41_vib_turn_on();
	else
		pm_vib_turn_on();
}

/* Function to turn off vibrator */
void vib_turn_off()
{
	uint32_t pmic = target_get_pmic();

	if (pmic == PMIC_IS_PM8916)
		pm8x41_vib_turn_off();
	else
		pm_vib_turn_off();
}

#if !USE_VIB_THREAD
/* Function to turn off vibrator when the vib_timer is expired. */
static enum handler_return vib_timer_func(struct timer *v_timer, time_t t, void *arg)
{
	timer_cancel(&vib_timer);
	if(!vib_timeout){
		vib_turn_off();
		vib_timeout = 1;
	}

	return INT_RESCHEDULE;
}
#else
int vibrator_thread(void *arg)
{
	while((--vib_time)){
		thread_sleep(CHECK_VIB_TIMER_FREQUENCY);
	}
	if(!vib_timeout){
		vib_turn_off();
		vib_timeout = 1;
	}
	return 0;
}
#endif

/*
 * Function to turn on vibrator.
 * vibrate_time - the time of phone vibrate.
 */
void vib_timed_turn_on(const uint32_t vibrate_time)
{
#if USE_VIB_THREAD
	thread_t *thr;
#endif

	if(!vib_timeout){
		dprintf(CRITICAL,"vibrator already turn on\n");
		return;
	}
	vib_turn_on();
	vib_timeout = 0;
#if !USE_VIB_THREAD
	timer_initialize(&vib_timer);
	timer_set_oneshot(&vib_timer, vibrate_time, vib_timer_func, NULL);
#else
	vib_time = (vibrate_time/CHECK_VIB_TIMER_FREQUENCY)+1;
	thr = thread_create("vibrator_thread", &vibrator_thread,
			NULL, DEFAULT_PRIORITY, DEFAULT_STACK_SIZE);
	if (!thr)
	{
		panic("failed to create vibrator thread\n");
	}
	thread_resume(thr);

#endif
}

/* Wait for vibrator timer expired */
void wait_vib_timeout(void)
{
	while (!vib_timeout) {
		/* every 50ms to check if the vibrator timer is timeout*/
		thread_sleep(CHECK_VIB_TIMER_FREQUENCY);
	}
}

void turn_off_vib_early(void)
{
	if(vib_timeout) {
		return;
	}
	vib_turn_off();
	vib_timeout = 1;
}