/* FreqCount Library, for measuring frequencies
 * http://www.pjrc.com/teensy/td_libs_FreqCount.html
 * Copyright (c) 2014 PJRC.COM, LLC - Paul Stoffregen <paul@pjrc.com>
 *
 * Version 1.1
 * 
 * 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 THE
 * AUTHORS OR COPYRIGHT HOLDERS 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.
 */

#include "FreqCount.h"
#include "util/FreqCountTimers.h"

#if(!defined(__IMXRT1062__))
static uint16_t count_msw;
static uint16_t gate_length;
static uint16_t gate_index;

void FreqCountClass::begin(uint16_t msec)
{
	if (msec < 1) return;
	gate_index = 0;
	count_msw = 0;
	count_prev = 0;
	count_ready = 0;
	counter_init();
	gate_length = timer_init(msec);
	uint8_t status = SREG;
	cli();
	timer_start();
	timer_isr_latency_delay();
	counter_start();
	SREG = status;
}

uint8_t FreqCountClass::available(void)
{
	return count_ready;
}

uint32_t FreqCountClass::read(void)
{
	uint32_t count;
	uint8_t status;

	status = SREG;
	cli();
#if defined(TIMER_USE_TIMER2) && F_CPU == 12000000L
	float correct = count_output * 0.996155;
	count = (uint32_t) (correct+0.5);
#else
	count = count_output;
#endif
	count_ready = 0;
	SREG = status;
	return count;
}

void FreqCountClass::end(void)
{
	timer_shutdown();
	counter_shutdown();
}


ISR(TIMER_ISR_VECTOR)
{
	uint16_t lsw, msw;
	uint32_t count;
	uint16_t index, length;

	lsw = counter_read();
	msw = count_msw;
	if (counter_overflow()) {
		counter_overflow_reset();
		if (lsw < 0xFA00) {
			msw = msw + 1;
			count_msw = msw;
		} else {
			count_msw = msw + 1;
		}
	}
	index = gate_index + 1;
	length = gate_length;
	if (index >= length) {
		gate_index = 0;
		count = ((uint32_t)msw << 16) + lsw;
		count_output = count - count_prev;
		count_prev = count;
		count_ready = 1;
		restore_other_interrupts();
	} else {
		if (index == length - 1) disable_other_interrupts();
		gate_index = index;
	}
}

#else
void FreqCountClass::begin(uint32_t usec)
{
  counter_init();
  timer_init(usec);
  //counter_start();

}

uint8_t FreqCountClass::available(void)
{ 
	return count_ready;
}

uint32_t FreqCountClass::read(void)
{
    count_output = count - count_prev;
    count_prev = count;
    //Serial.println(count - count_prev);
    count_prev = count;
    count_ready = 0;
  
  return count_output;
}

#endif

FreqCountClass FreqCount;