#include "pch.h"
#include "gm_sbc.h"
#include "defaults.h"

#ifdef HW_HELLEN
#include "hellen_meta.h"
#endif // HW_HELLEN

#include "proteus_meta.h"

void setStepperHw() {
	engineConfiguration->useStepperIdle = true;
	engineConfiguration->useHbridgesToDriveIdleStepper = true;
	engineConfiguration->stepper_dc_use_two_wires = false;
	engineConfiguration->stepperDcInvertedPins = false; // or true?

    // for instance IWP069
	engineConfiguration->injector.flow = 482.5;
  setPPSInputs(EFI_ADC_NONE, EFI_ADC_NONE);

#ifdef HW_HELLEN_8CHAN
	// using 8chan pinout for DC1: A26 (OUT_DC1+ AH pin "D") and A27 (OUT_DC1- AL pin "C")
	engineConfiguration->stepperDcIo[0].controlPin = Gpio::H144_GP_IO4; // DC1_PWM
	engineConfiguration->stepperDcIo[0].directionPin1 = Gpio::H144_GP_IO3; // DC1_DIR
	engineConfiguration->stepperDcIo[0].directionPin2 = Gpio::Unassigned;
	engineConfiguration->stepperDcIo[0].disablePin = Gpio::H144_GP_IO5; // ETB_DIS
	// using 8chan pinout for DC2: A28 (OUT_DC2+ BL pin "A") and A29 (OUT_DC2- BH pin "B")
	engineConfiguration->stepperDcIo[1].controlPin = H144_UART8_TX; // DC2_PWM
	engineConfiguration->stepperDcIo[1].directionPin1 = H_SPI3_CS; // DC2_DIR
	engineConfiguration->stepperDcIo[1].directionPin2 = Gpio::Unassigned;
#endif // HW_HELLEN

#ifdef HW_HELLEN_UAEFI
  // TODO: all the copy-pasting here begs the question: "shall we rename etbIo to hBridgeIo and reuse for stepper"?
	engineConfiguration->stepperDcIo[0].controlPin = Gpio::MM100_OUT_PWM3;
	engineConfiguration->stepperDcIo[0].directionPin1 = Gpio::MM100_OUT_PWM4;
	engineConfiguration->stepperDcIo[0].directionPin2 = Gpio::Unassigned;
	engineConfiguration->stepperDcIo[0].disablePin = Gpio::MM100_SPI2_MISO;

	engineConfiguration->stepperDcIo[1].controlPin = Gpio::MM100_OUT_PWM5;
	engineConfiguration->stepperDcIo[1].directionPin1 = Gpio::MM100_SPI2_MOSI;
	engineConfiguration->stepperDcIo[1].directionPin2 = Gpio::MM100_USB1ID;
#endif // HW_HELLEN_UAEFI

#ifdef HW_PROTEUS
	// coil #1 = proteus pin 15 (DC2 positive) to BL pin "A", pin B to pin 8 (DC2 negative)
	// PWM pin
	engineConfiguration->stepperDcIo[0].controlPin = Gpio::D13;
	// DIR pin
	engineConfiguration->stepperDcIo[0].directionPin1 = Gpio::D9;
	// Disable pin
	engineConfiguration->stepperDcIo[0].disablePin = Gpio::D8;
	// Unused
	engineConfiguration->stepperDcIo[0].directionPin2 = Gpio::Unassigned;

	// coil #2 - proteus pin 7 (DC1 positive) to AH pin "D", pin C to pin 6 (DC1 negative)
	// PWM pin
	engineConfiguration->stepperDcIo[1].controlPin = Gpio::D12;
	// DIR pin
	engineConfiguration->stepperDcIo[1].directionPin1 = Gpio::D10;
	// Disable pin
	engineConfiguration->stepperDcIo[1].disablePin = Gpio::D11;
	// Unused
	engineConfiguration->stepperDcIo[1].directionPin2 = Gpio::Unassigned;

#endif // HW_PROTEUS
}

static void gmRailSensor() {
	engineConfiguration->highPressureFuel.v1 = 0.5; /* volts */;
	engineConfiguration->highPressureFuel.value1 = 0;
	engineConfiguration->highPressureFuel.v2 = 4.5; /* volts */;
	// fun fact: twice the range of VAG B6?
	engineConfiguration->highPressureFuel.value2 = PSI2KPA(400);
}

void setGmSbc() {
	engineConfiguration->cylindersCount = 8;
	engineConfiguration->firingOrder = FO_1_8_4_3_6_5_7_2;
	engineConfiguration->displacement = 5.2;

    // see engine sniffer at https://github.com/rusefi/rusefi/issues/5353
    // if injectors are wired in groups of four best to use channels 1 and 4
    engineConfiguration->crankingInjectionMode = IM_BATCH;
    engineConfiguration->injectionMode = IM_BATCH;

	engineConfiguration->etbFunctions[0] = DC_None;
	engineConfiguration->etbFunctions[1] = DC_None;

	strcpy(engineConfiguration->engineMake, ENGINE_MAKE_GM);
	strcpy(engineConfiguration->engineCode, "SBC");
	// white wire "HEI E" plug pin D
	engineConfiguration->ignitionMode = IM_ONE_COIL;
	engineConfiguration->globalTriggerAngleOffset = 24;

    gppwm_channel *ignOverride = &engineConfiguration->gppwm[0];
   	ignOverride->pwmFrequency = 0;
   	strcpy(engineConfiguration->gpPwmNote[0], "ign ovrrd B");
   	setTable(ignOverride->table, 100);

#ifdef HW_HELLEN_UAEFI121
    // coil control plug pin A

    // tan wire with a black trace - "HEI B", plug pin B (we use yellow)
    ignOverride->pin = Gpio::MM100_IGN2; // 12a Coil 2
    // tach plug pin C
    engineConfiguration->triggerInputPins[0] = Gpio::MM100_IN_D2; // 35a HALL2 red wire
	  engineConfiguration->camInputs[0] = Gpio::Unassigned;
	  engineConfiguration->camInputs[1] = Gpio::Unassigned;
	  engineConfiguration->ignitionPins[1] = Gpio::Unassigned;
#endif // HW_HELLEN_UAEFI121

#if HW_PROTEUS
    // tan wire with a black trace - "HEI B", plug pin B
    ignOverride->pin = Gpio::PROTEUS_IGN_12;
	engineConfiguration->fanPin = Gpio::Unassigned;
	// reminder about D101
   	engineConfiguration->injectionPins[0] = Gpio::PROTEUS_LS_14; // #1
   	// reminder about D104
   	engineConfiguration->injectionPins[3] = Gpio::PROTEUS_LS_15; // #4

    // wow high side relay control on MEFI1 ELECTRONIC-FUEL-INJECTION-MEFI-1-2-5.7L-350CID-8.2L-502CID.pdf page 46
    engineConfiguration->fuelPumpPin = Gpio::PROTEUS_HS_1;
    // low side on MEFI3 ELECTRONIC-FUEL-INJECTION-MEFI-3-5.7L-350CID-8.2L-502CID.pdf page 487

    setGmCltSensor(&engineConfiguration->clt, PROTEUS_DEFAULT_AT_PULLUP);
#endif // HW_PROTEUS
	engineConfiguration->mainRelayPin = Gpio::Unassigned; // vehicle controls main relay

 	setStepperHw();

	// 8 cylinders one coil HEI trigger
	engineConfiguration->trigger.type = trigger_type_e::TT_TOOTHED_WHEEL;
	engineConfiguration->trigger.customTotalToothCount = 8;
	engineConfiguration->trigger.customSkippedToothCount = 0;
	engineConfiguration->skippedWheelOnCam = true;
	// Proteus Digital 1 is Tach input "HEI R", plug pin C

	engineConfiguration->map.sensor.type = MT_GM_1_BAR;
}

static void setGmEcotec3() {
  engineConfiguration->globalTriggerAngleOffset = 360 + 90;
  engineConfiguration->hpfpCamLobes = 4;
  engineConfiguration->vvtMode[0] = VVT_BOSCH_QUICK_START;
  engineConfiguration->lowPressureFuel.hwChannel = EFI_ADC_NONE;
  gmRailSensor();
  engineConfiguration->EtbSentInput = SentInput::INPUT1;

#ifdef HW_HELLEN_4K_GDI
  engineConfiguration->invertCamVVTSignal = true;
  engineConfiguration->invertExhaustCamVVTSignal = true;

  engineConfiguration->triggerInputPins[0] = Gpio::H144_IN_SENS1; // Digital Input 1
  engineConfiguration->camInputs[0] = Gpio::H144_IN_SENS3; // Digital Input 4
  engineConfiguration->sentInputPins[0] = Gpio::H144_IN_AUX2_DIGITAL;

  // engineConfiguration->starterControlPin = high side :()
#endif // HW_HELLEN_4K_GDI

#ifdef HW_HELLEN_8CHAN
  engineConfiguration->sentInputPins[0] = Gpio::MM176_IN_D3;
  engineConfiguration->starterControlPin = Gpio::MM176_IGN5; // 14C

	engineConfiguration->camInputs[2] = Gpio::Unassigned;
	engineConfiguration->camInputs[3] = Gpio::Unassigned;
	config->boardUseCrankPullUp = true;
#endif // HW_HELLEN_8CHAN

  engineConfiguration->sentEtbType = SentEtbType::GM_TYPE_1;
  setTPS1Inputs(EFI_ADC_NONE, EFI_ADC_NONE);
  setPPSCalibration(1, 4.25, 0.5, 2.14);

	strcpy(engineConfiguration->engineMake, ENGINE_MAKE_GM);
	setGDIFueling();
}

static void set4CylGmEcotec3() {
  setGmEcotec3();
  engineConfiguration->vvtMode[1] = VVT_BOSCH_QUICK_START;
	setInline4();
}

void setGmLcv() {
  set4CylGmEcotec3();
  engineConfiguration->displacement = 2.5;
	engineConfiguration->map.sensor.type = MT_GM_1_BAR;
  strcpy(engineConfiguration->engineCode, "LCV");
}

void setGmLtg() {
  set4CylGmEcotec3();
  engineConfiguration->displacement = 2.0;
  strcpy(engineConfiguration->engineCode, "LTG");
}

void setGmSbcGen5() {
  setGmEcotec3();
  engineConfiguration->displacement = 5.3;
	engineConfiguration->cylindersCount = 8;
	engineConfiguration->firingOrder = FO_1_8_7_2_6_5_4_3;
	engineConfiguration->map.sensor.type = MT_GM_1_BAR;

#ifdef HW_HELLEN_8CHAN
	engineConfiguration->injectionPins[4] = Gpio::MM176_INJ5;
	engineConfiguration->injectionPins[5] = Gpio::MM176_INJ6;
	engineConfiguration->injectionPins[6] = Gpio::MM176_INJ7;
	engineConfiguration->injectionPins[7] = Gpio::MM176_INJ8;

	engineConfiguration->ignitionPins[4] = Gpio::MM176_IGN5;
	engineConfiguration->ignitionPins[5] = Gpio::MM176_IGN6;
	engineConfiguration->ignitionPins[6] = Gpio::MM176_IGN7;
	engineConfiguration->ignitionPins[7] = Gpio::MM176_IGN8;

	engineConfiguration->starterControlPin = Gpio::MM176_SPI3_SCK; // 29C - Ignition 9

  engineConfiguration->vvtMode[1] = VVT_INACTIVE;
  engineConfiguration->camInputs[1] = Gpio::Unassigned;

  engineConfiguration->fuelPumpPin = Gpio::MM176_OUT_IO2; // 6C - High Side 2

  engineConfiguration->oilPressure.hwChannel = MM176_IN_MAP1_ANALOG; // 2A Aux Analog 4 / External MAP
  engineConfiguration->fuelLevelSensor = MM176_IN_O2S_ANALOG; // 3A - Aux Analog 2
  engineConfiguration->lowPressureFuel.hwChannel = MM176_IN_AUX2_ANALOG; // 	4A - Aux Analog 7
  engineConfiguration->acPressure.hwChannel = MM176_IN_MAP2_ANALOG;// 10A - Aux Analog 5
  engineConfiguration->flexSensorPin = Gpio::MM176_IN_SENS3;
  engineConfiguration->map.sensor.hwChannel = MM176_IN_CRANK_ANALOG; // 1A Aux Analog 1. Important to use analog without muxing!
#endif // HW_HELLEN_8CHAN
}