/**
 * @file	logic_analyzer.cpp
 * @brief	Initialization of Input Capture pins used for rusEfi console sniffer
 *
 * This file is responsible for sniffing of external digital signals and registering
 * these digital events in WaveChart used by the Engine Sniffer tab of rusEfi Console.
 *
 * this is rusEfi build-in logic analyzer
 *
 * @date Jan 7, 2013
 * @author Andrey Belomutskiy, (c) 2012-2020
 */

#include "global.h"
#include "engine.h"
#include "logic_analyzer.h"
#include "os_access.h"
#include "eficonsole.h"
#include "pin_repository.h"
#include "allsensors.h"
#include "engine_configuration.h"
#include "trigger_central.h"
#include "os_util.h"
#include "engine_math.h"
#include "rpm_calculator.h"
#include "engine_sniffer.h"

#if EFI_LOGIC_ANALYZER

EXTERN_ENGINE;

#define CHART_RESET_DELAY 1
#define MAX_ICU_COUNT 5

#if EFI_ENGINE_SNIFFER
extern WaveChart waveChart;
#endif /* EFI_ENGINE_SNIFFER */

/**
 * Difference between current 1st trigger event and previous 1st trigger event.
 */
static volatile uint32_t engineCycleDurationUs;
static volatile efitimeus_t previousEngineCycleTimeUs = 0;

static int waveReaderCount = 0;
static WaveReader readers[MAX_ICU_COUNT];

static void ensureInitialized(WaveReader *reader) {
	/*may be*/UNUSED(reader);
	efiAssertVoid(CUSTOM_ERR_6654, reader->hw != NULL && reader->hw->started, "wave analyzer NOT INITIALIZED");
}

static void waAnaWidthCallback(WaveReader *reader) {
	efitick_t nowUs = getTimeNowUs();
	reader->riseEventCounter++;
	reader->lastActivityTimeUs = nowUs;
	assertIsrContext(CUSTOM_ERR_6670);
	addEngineSnifferEvent(reader->name, PROTOCOL_ES_UP);

	uint32_t width = nowUs - reader->periodEventTimeUs;
	reader->last_wave_low_widthUs = width;

	reader->signalPeriodUs = nowUs - reader->widthEventTimeUs;
	reader->widthEventTimeUs = nowUs;
}

void WaveReader::onFallEvent() {
	efitick_t nowUs = getTimeNowUs();
	fallEventCounter++;
	lastActivityTimeUs = nowUs;
	assertIsrContext(CUSTOM_ERR_6670);
	addEngineSnifferEvent(name, PROTOCOL_ES_DOWN);

	efitick_t width = nowUs - widthEventTimeUs;
	last_wave_high_widthUs = width;

#if EFI_SHAFT_POSITION_INPUT
	int revolutionCounter = getRevolutionCounter();
#else
	int revolutionCounter = 0;
#endif

	totalOnTimeAccumulatorUs += width;
	if (currentRevolutionCounter != revolutionCounter) {
		/**
		 * We are here in case of a new engine cycle
		 */
		currentRevolutionCounter = revolutionCounter;
		prevTotalOnTimeUs = totalOnTimeAccumulatorUs;
		totalOnTimeAccumulatorUs = 0;

		waveOffsetUs = nowUs - previousEngineCycleTimeUs;
	}

	periodEventTimeUs = nowUs;
}

static void waIcuPeriodCallback(WaveReader *reader) {
	reader->onFallEvent();
}

static void initWave(const char *name, int index) {
	brain_pin_e brainPin = CONFIG(logicAnalyzerPins)[index];

	waveReaderCount++;
	efiAssertVoid(CUSTOM_ERR_6655, index < MAX_ICU_COUNT, "too many ICUs");
	WaveReader *reader = &readers[index];

	if (!isBrainPinValid(brainPin)) {
		/**
		 *  in case we are running, and we select none for a channel that was running, 
		 *  this way we ensure that we do not get false report from that channel 
		 **/
		reader->hw = nullptr;
		return;
	}
		

	reader->name = name;

	reader->hw = startDigitalCapture("wave input", brainPin);

	if (reader->hw != NULL) {
		reader->hw->setWidthCallback((VoidInt)(void*) waAnaWidthCallback, (void*) reader);

		reader->hw->setPeriodCallback((VoidInt)(void*) waIcuPeriodCallback, (void*) reader);
	}

	efiPrintf("wave%d input on %s", index, hwPortname(brainPin));
}

WaveReader::WaveReader() {
	hw = nullptr;
}

void waTriggerEventListener(trigger_event_e ckpSignalType, uint32_t index, efitick_t edgeTimestamp DECLARE_ENGINE_PARAMETER_SUFFIX) {
	(void)ckpSignalType;
	if (index != 0) {
		return;
	}

	efitimeus_t nowUs = NT2US(edgeTimestamp);
	engineCycleDurationUs = nowUs - previousEngineCycleTimeUs;
	previousEngineCycleTimeUs = nowUs;
}

static float getSignalOnTime(int index) {
	WaveReader *reader = &readers[index];
	ensureInitialized(reader);
	if (getTimeNowUs() - reader->lastActivityTimeUs > 4 * US_PER_SECOND) {
		return 0.0f; // dwell time has expired
	}
	return reader->last_wave_high_widthUs / 1000.0f;
}

static efitime_t getWaveOffset(int index) {
	WaveReader *reader = &readers[index];
	ensureInitialized(reader);
	return reader->waveOffsetUs;
}

static float getSignalPeriodMs(int index) {
	WaveReader *reader = &readers[index];
	ensureInitialized(reader);
	return reader->signalPeriodUs / 1000.0f;
}

static void reportWave(Logging *logging, int index) {
	if (readers[index].hw == nullptr) {
		return;
	}
	if (readers[index].hw->started) {
//	int counter = getEventCounter(index);
//	debugInt2(logging, "ev", index, counter);

		float dwellMs = getSignalOnTime(index);
		float periodMs = getSignalPeriodMs(index);

		logging->appendPrintf("duty%d%s", index, DELIMETER);
		logging->appendFloat(100.0f * dwellMs / periodMs, 2);
		logging->appendPrintf("%s", DELIMETER);

		/**
		 * that's the ON time of the LAST signal
		 */
		logging->appendPrintf("dwell%d%s", index, DELIMETER);
		logging->appendFloat(dwellMs, 2);
		logging->appendPrintf("%s", DELIMETER);

		/**
		 * that's the total ON time during the previous engine cycle
		 */
		logging->appendPrintf("total_dwell%d%s", index, DELIMETER);
		logging->appendFloat(readers[index].prevTotalOnTimeUs / 1000.0f, 2);
		logging->appendPrintf("%s", DELIMETER);

		logging->appendPrintf("period%d%s", index, DELIMETER);
		logging->appendFloat(periodMs, 2);
		logging->appendPrintf("%s", DELIMETER);

		uint32_t offsetUs = getWaveOffset(index);
		int rpm = GET_RPM();
		if (rpm != 0) {
			float oneDegreeUs = getOneDegreeTimeUs(rpm);

			logging->appendPrintf("advance%d%s", index, DELIMETER);
			float angle = (offsetUs / oneDegreeUs) - tdcPosition();
			fixAngle(angle, "waveAn", CUSTOM_ERR_6564);
			logging->appendFloat(angle, 3);
			logging->appendPrintf("%s", DELIMETER);
		}
	}
}

void printWave(Logging *logging) {
	reportWave(logging, 0);
	reportWave(logging, 1);
}

void showWaveInfo(void) {
	efiPrintf("logic input #1: %d/%d", readers[0].fallEventCounter, readers[0].riseEventCounter);
}

void initWaveAnalyzer() {
	if (hasFirmwareError()) {
		return;
	}

	addConsoleAction("waveinfo", showWaveInfo);
}

void startLogicAnalyzerPins() {
	initWave(PROTOCOL_WA_CHANNEL_1, 0);
	initWave(PROTOCOL_WA_CHANNEL_2, 1);
	initWave(PROTOCOL_WA_CHANNEL_3, 2);
	initWave(PROTOCOL_WA_CHANNEL_4, 3);
}

void stopLogicAnalyzerPins() {
	for (int index = 0; index < LOGIC_ANALYZER_CHANNEL_COUNT; index++) {
		brain_pin_e brainPin = activeConfiguration.logicAnalyzerPins[index];

		if (isBrainPinValid(brainPin)) {
			stopDigitalCapture("wave input", brainPin);
		}
	}
}

void getChannelFreqAndDuty(int index, float *duty, int *freq) {

	float high,period;

	if ((duty == nullptr) || (freq == nullptr)) {
		return;
	}

	if (readers[index].hw == nullptr) {
		*duty = 0.0;
		*freq = 0;
	} else {
		high = getSignalOnTime(index);
		period = getSignalPeriodMs(index);

		if ((period != 0) && (readers[index].hw->started)) {

			*duty = (high * 1000.0f) /(period * 10.0f);
			*freq = (int)(1 / (period / 1000.0f));
		} else {		
			*duty = 0.0;
			*freq = 0;
		}
	}

}

void reportLogicAnalyzerToTS() {
#if EFI_TUNER_STUDIO	
	int tmp;
	getChannelFreqAndDuty(0,&tsOutputChannels.debugFloatField1, &tsOutputChannels.debugIntField1);
	getChannelFreqAndDuty(1,&tsOutputChannels.debugFloatField2, &tsOutputChannels.debugIntField2);
	getChannelFreqAndDuty(2,&tsOutputChannels.debugFloatField3, &tsOutputChannels.debugIntField3);
	getChannelFreqAndDuty(3,&tsOutputChannels.debugFloatField4, &tmp);
	tsOutputChannels.debugIntField4 = (int16_t)tmp;
#endif	
}

#endif /* EFI_LOGIC_ANALYZER */