/**
 * @file init_sensorss.cpp
 */

#include "pch.h"

#include "init.h"
#include "cli_registry.h"
#include "io_pins.h"

static void initSensorCli();

void initIfValid(const char* msg, adc_channel_e channel) {
	if (!isAdcChannelValid(channel)) {
		return;
	}

#if EFI_PROD_CODE
	brain_pin_e pin = getAdcChannelBrainPin(msg, channel);
	efiSetPadMode(msg, pin, PAL_MODE_INPUT_ANALOG);
#endif
}

void deInitIfValid(const char* msg, adc_channel_e channel) {
	if (!isAdcChannelValid(channel)) {
		return;
	}

#if EFI_PROD_CODE
	brain_pin_e pin = getAdcChannelBrainPin(msg, channel);
	efiSetPadUnused(pin);
#endif
}

static void initOldAnalogInputs() {
	initIfValid("AFR", engineConfiguration->afr.hwChannel);
	initIfValid("AUXF#1", engineConfiguration->auxFastSensor1_adcChannel);
	initIfValid("CJ125 UR", engineConfiguration->cj125ur);
	initIfValid("CJ125 UA", engineConfiguration->cj125ua);
}

static void deInitOldAnalogInputs() {
	deInitIfValid("AFR", activeConfiguration.afr.hwChannel);
	deInitIfValid("AUXF#1", activeConfiguration.auxFastSensor1_adcChannel);
	deInitIfValid("CJ125 UR", activeConfiguration.cj125ur);
	deInitIfValid("CJ125 UA", activeConfiguration.cj125ua);
}

static void initAuxDigital() {
	for (size_t i = 0;i<efi::size(engineConfiguration->luaDigitalInputPins);i++) {
		efiSetPadMode("Lua Digital", engineConfiguration->luaDigitalInputPins[i], getInputMode(PI_DEFAULT));
	}
}

static void deInitAuxDigital() {
	for (size_t i = 0;i<efi::size(activeConfiguration.luaDigitalInputPins);i++) {
		brain_pin_markUnused(activeConfiguration.luaDigitalInputPins[i]);
	}
}

void initNewSensors() {
#if EFI_PROD_CODE && EFI_CAN_SUPPORT
	initCanSensors();
#endif

	initVbatt();
	initMap();
	initTps();
	initFluidPressure();
	initThermistors();
	initLambda();
	initFlexSensor();
	initBaro();
	initAuxSensors();
	initVehicleSpeedSensor();
	initTurbochargerSpeedSensor();
	initAuxSpeedSensors();
	initInputShaftSpeedSensor();

	#if !EFI_UNIT_TEST
		initFuelLevel();
		initMaf();
	#endif

	initOldAnalogInputs();
	initAuxDigital();

	// Init CLI functionality for sensors (mocking)
	initSensorCli();

#if defined(HARDWARE_CI) && !defined(HW_PROTEUS)
	chThdSleepMilliseconds(100);

	if (Sensor::getOrZero(SensorType::BatteryVoltage) < 8) {
		// Fake that we have battery voltage, some tests rely on it
		Sensor::setMockValue(SensorType::BatteryVoltage, 10);
	}
#endif
}

void stopSensors() {
	deInitAuxDigital();
	deInitOldAnalogInputs();

	deinitTps();
	deinitFluidPressure();
	deinitVbatt();
	deinitThermistors();
	deInitFlexSensor();
	deInitVehicleSpeedSensor();
	deinitTurbochargerSpeedSensor();
	deinitAuxSpeedSensors();
	deinitMap();
	deinitInputShaftSpeedSensor();
}

void reconfigureSensors() {
	initMap();
	initTps();
	initFluidPressure();
	initVbatt();
	initThermistors();
	initFlexSensor();
	initVehicleSpeedSensor();
	initTurbochargerSpeedSensor();
	initInputShaftSpeedSensor();

	initOldAnalogInputs();
}

// Mocking/testing helpers
static void initSensorCli() {
	addConsoleActionIF("set_sensor_mock", Sensor::setMockValue);
	addConsoleAction("reset_sensor_mocks", Sensor::resetAllMocks);
	addConsoleAction("show_sensors", Sensor::showAllSensorInfo);
	addConsoleActionI("show_sensor",
		[](int idx) {
			Sensor::showInfo(static_cast<SensorType>(idx));
		});
}