/* *****************************************************************************
 * Copyright(C) Maxim Integrated Products, Inc., All Rights Reserved.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files(the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included
 * in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
 * IN NO EVENT SHALL MAXIM INTEGRATED BE LIABLE FOR ANY CLAIM, DAMAGES
 * OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
 * OTHER DEALINGS IN THE SOFTWARE.
 *
 * Except as contained in this notice, the name of Maxim Integrated
 * Products, Inc. shall not be used except as stated in the Maxim Integrated
 * Products, Inc. Branding Policy.
 *
 * The mere transfer of this software does not imply any licenses
 * of trade secrets, proprietary technology, copyrights, patents,
 * trademarks, maskwork rights, or any other form of intellectual
 * property whatsoever. Maxim Integrated Products, Inc. retains all
 * ownership rights.
 *
 **************************************************************************** */

/* **** Includes **** */
#include <stddef.h>
#include "mxc_assert.h"
#include "tmr.h"
#include "tmr_reva.h"
#include "gpio.h"
#include "mxc_pins.h"
#include "mxc_lock.h"

/* **** Functions **** */
void MXC_TMR_RevA_Init(mxc_tmr_reva_regs_t *tmr, mxc_tmr_cfg_t* cfg)
{
    // Clear interrupt flag
    tmr->intr = MXC_F_TMR_REVA_INTR_IRQ;
    
    // Set the prescaler
    switch(cfg->pres) {
    case TMR_PRES_1:
        tmr->cn |= (MXC_S_TMR_REVA_CN_PRES_DIV1);
        break;
        
    case TMR_PRES_2:
        tmr->cn |= (MXC_S_TMR_REVA_CN_PRES_DIV2);
        break;
        
    case TMR_PRES_4:
        tmr->cn |= (MXC_S_TMR_REVA_CN_PRES_DIV4);
        break;
        
    case TMR_PRES_8:
        tmr->cn |= (MXC_S_TMR_REVA_CN_PRES_DIV8);
        break;
        
    case TMR_PRES_16:
        tmr->cn |= (MXC_S_TMR_REVA_CN_PRES_DIV16);
        break;
        
    case TMR_PRES_32:
        tmr->cn |= (MXC_S_TMR_REVA_CN_PRES_DIV32);
        break;
        
    case TMR_PRES_64:
        tmr->cn |= (MXC_S_TMR_REVA_CN_PRES_DIV64);
        break;
        
    case TMR_PRES_128:
        tmr->cn |= (MXC_S_TMR_REVA_CN_PRES_DIV128);
        break;
        
    case TMR_PRES_256:
        tmr->cn |= (MXC_F_TMR_REVA_CN_PRES3);
        break;
        
    case TMR_PRES_512:
        tmr->cn |= (MXC_F_TMR_REVA_CN_PRES3);
        tmr->cn |= (MXC_S_TMR_REVA_CN_PRES_DIV4);
        break;
        
    case TMR_PRES_1024:
        tmr->cn |= (MXC_F_TMR_REVA_CN_PRES3);
        tmr->cn |= (MXC_S_TMR_REVA_CN_PRES_DIV8);
        break;
        
    case TMR_PRES_2048:
        tmr->cn |= (MXC_F_TMR_REVA_CN_PRES3);
        tmr->cn |= (MXC_S_TMR_REVA_CN_PRES_DIV16);
        break;
        
    case TMR_PRES_4096:
        tmr->cn |= (MXC_F_TMR_REVA_CN_PRES3);
        tmr->cn |= (MXC_S_TMR_REVA_CN_PRES_DIV128);
        break;

    default:
        tmr->cn |= (MXC_F_TMR_REVA_CN_PRES3);
        tmr->cn |= (MXC_S_TMR_REVA_CN_PRES_DIV128);
        break;
    }

    tmr->cn |= cfg->mode << MXC_F_TMR_REVA_CN_TMODE_POS;
    tmr->cn |= (cfg->pol) << MXC_F_TMR_REVA_CN_TPOL_POS;
    //enable timer interrupt if needed
    tmr->cnt = 0x1;
    tmr->cmp = cfg->cmp_cnt;
}

void MXC_TMR_RevA_Shutdown(mxc_tmr_reva_regs_t *tmr)
{
    int tmr_id = MXC_TMR_GET_IDX((mxc_tmr_regs_t*) tmr);
    if (tmr_id < 0) {
        MXC_ASSERT(0);
    }

    // Disable timer and clear settings
    tmr->cn = 0;
}

void MXC_TMR_RevA_Start(mxc_tmr_reva_regs_t* tmr)
{
    int tmr_id = MXC_TMR_GET_IDX((mxc_tmr_regs_t*) tmr);
    if (tmr_id < 0) {
        MXC_ASSERT(0);
    }

    tmr->cn |= MXC_F_TMR_REVA_CN_TEN;
}

void MXC_TMR_RevA_Stop(mxc_tmr_reva_regs_t* tmr)
{
    int tmr_id = MXC_TMR_GET_IDX((mxc_tmr_regs_t*) tmr);
    if (tmr_id < 0) {
        MXC_ASSERT(0);
    }

    tmr->cn &= ~MXC_F_TMR_REVA_CN_TEN;
}

int MXC_TMR_RevA_SetPWM(mxc_tmr_reva_regs_t* tmr, uint32_t pwm)
{
    int tmr_id = MXC_TMR_GET_IDX((mxc_tmr_regs_t*) tmr);
    if (tmr_id < 0) {
        MXC_ASSERT(0);
    }
    
    if(pwm > (tmr->cmp)) {
        return E_BAD_PARAM;
    }
    
    tmr->pwm = pwm;
    
    return E_NO_ERROR;
}

uint32_t MXC_TMR_RevA_GetCompare(mxc_tmr_reva_regs_t* tmr)
{
    int tmr_id = MXC_TMR_GET_IDX((mxc_tmr_regs_t*) tmr);
    if (tmr_id < 0) {
        MXC_ASSERT(0);
    }

    return tmr->cmp;
}

uint32_t MXC_TMR_RevA_GetCapture(mxc_tmr_reva_regs_t* tmr)
{
    int tmr_id = MXC_TMR_GET_IDX((mxc_tmr_regs_t*) tmr);
    if (tmr_id < 0) {
        MXC_ASSERT(0);
    }

    return tmr->pwm;//check this
}

uint32_t MXC_TMR_RevA_GetCount(mxc_tmr_reva_regs_t* tmr)
{
    int tmr_id = MXC_TMR_GET_IDX((mxc_tmr_regs_t*) tmr);
    if (tmr_id < 0) {
        MXC_ASSERT(0);
    }

    return tmr->cnt;
}

void MXC_TMR_RevA_ClearFlags(mxc_tmr_reva_regs_t* tmr)
{
    int tmr_id = MXC_TMR_GET_IDX((mxc_tmr_regs_t*) tmr);
    if (tmr_id < 0) {
        MXC_ASSERT(0);
    }

    tmr->intr = MXC_F_TMR_REVA_INTR_IRQ;
}

uint32_t MXC_TMR_RevA_GetFlags(mxc_tmr_reva_regs_t* tmr)
{
    int tmr_id = MXC_TMR_GET_IDX((mxc_tmr_regs_t*) tmr);
    if (tmr_id < 0) {
        MXC_ASSERT(0);
    }

    return tmr->intr;
}

void MXC_TMR_RevA_SetCompare(mxc_tmr_reva_regs_t *tmr, uint32_t cmp_cnt)
{
    int tmr_id = MXC_TMR_GET_IDX((mxc_tmr_regs_t*) tmr);
    if (tmr_id < 0) {
        MXC_ASSERT(0);
    }

    tmr->cmp = cmp_cnt;
}

void MXC_TMR_RevA_SetCount(mxc_tmr_reva_regs_t *tmr, uint32_t cnt)
{
    int tmr_id = MXC_TMR_GET_IDX((mxc_tmr_regs_t*) tmr);
    if (tmr_id < 0) {
        MXC_ASSERT(0);
    }
    
    tmr->cnt = cnt;
}

void MXC_TMR_RevA_TO_Start(mxc_tmr_reva_regs_t *tmr, unsigned long us)
{
    uint64_t ticks;
    int clk_shift = 0;
    
    ticks = (uint64_t) us * (uint64_t) PeripheralClock / (uint64_t) 1000000;
    
    while(ticks > 0xFFFFFFFFUL) {
        ticks >>= 1;
        ++clk_shift;
    }
    
    mxc_tmr_pres_t prescale = (mxc_tmr_pres_t) clk_shift << MXC_F_TMR_REVA_CN_PRES_POS;
    mxc_tmr_cfg_t cfg = {0, 0, 0, 0}; // = (mxc_tmr_cfg_t) {.pres=0, .mode=0, .cmp_cnt=0, .pol=0};
    
    // Initialize the timer in one-shot mode
    cfg.pres = prescale;
    cfg.mode = TMR_MODE_ONESHOT;
    cfg.cmp_cnt = ticks;
    cfg.pol = 0;
    
    MXC_TMR_Stop((mxc_tmr_regs_t*) tmr);
    MXC_TMR_Init((mxc_tmr_regs_t*) tmr, &cfg);
    MXC_TMR_ClearFlags((mxc_tmr_regs_t*) tmr);
    MXC_TMR_Start((mxc_tmr_regs_t*) tmr);
}

int MXC_TMR_RevA_GetTime(mxc_tmr_reva_regs_t *tmr, uint32_t ticks, uint32_t *time, mxc_tmr_unit_t *units)
{
    uint64_t temp_time = 0;
    uint32_t timerClock = PeripheralClock;
    uint32_t prescale = ((tmr->cn & MXC_F_TMR_REVA_CN_PRES) >> MXC_F_TMR_REVA_CN_PRES_POS)
                        |(((tmr->cn & MXC_F_TMR_REVA_CN_PRES3) >> (MXC_F_TMR_REVA_CN_PRES3_POS)) <<3);
                        
    temp_time = (uint64_t) ticks * 1000 * (1 <<(prescale & 0xF)) / (timerClock / 1000000);
    
    if(!(temp_time & 0xffffffff00000000)) {
        *time = temp_time;
        *units = TMR_UNIT_NANOSEC;
        return E_NO_ERROR;
    }
    
    temp_time = (uint64_t) ticks * 1000 * (1 <<(prescale & 0xF)) / (timerClock / 1000);
    
    if(!(temp_time & 0xffffffff00000000)) {
        *time = temp_time;
        *units = TMR_UNIT_MICROSEC;
        return E_NO_ERROR;
    }
    
    temp_time = (uint64_t) ticks * 1000 * (1 <<(prescale & 0xF)) / timerClock;
    
    if(!(temp_time & 0xffffffff00000000)) {
        *time = temp_time;
        *units = TMR_UNIT_MILLISEC;
        return E_NO_ERROR;
    }
    
    temp_time = (uint64_t) ticks * (1 <<(prescale & 0xF)) / timerClock;
    
    if(!(temp_time & 0xffffffff00000000)) {
        *time = temp_time;
        *units = TMR_UNIT_SEC;
        return E_NO_ERROR;
    }
    
    return E_INVALID;
}