--[[
 Driver for INF_Inject EFI system

 https://innoflighttechnology.com/efi/
--]]

local PARAM_TABLE_KEY = 43
local PARAM_TABLE_PREFIX = "EFI_INF_"

local MAV_SEVERITY = {EMERGENCY=0, ALERT=1, CRITICAL=2, ERROR=3, WARNING=4, NOTICE=5, INFO=6, DEBUG=7}

local CMD_CDI1 = 1
local CMD_CDI2 = 2
local CMD_OIL_PUMP = 3
-- local CMD_SHUTDOWN = 4
-- local CMD_PRE_INJECTION = 5
local CMD_THROTTLE = 6

local K_THROTTLE = 70

-- bind a parameter to a variable given
local function bind_param(name)
    local p = Parameter()
    assert(p:init(name), string.format('could not find %s parameter', name))
    return p
end

-- add a parameter and bind it to a variable
local function bind_add_param(name, idx, default_value)
    assert(param:add_param(PARAM_TABLE_KEY, idx, name, default_value), string.format('could not add param %s', name))
    return bind_param(PARAM_TABLE_PREFIX .. name)
end

-- setup script specific parameters
assert(param:add_table(PARAM_TABLE_KEY, PARAM_TABLE_PREFIX, 8), 'could not add param table')

--[[
  // @Param: EFI_INF_ENABLE
  // @DisplayName: EFI INF-Inject enable
  // @Description: Enable EFI INF-Inject driver
  // @Values: 0:Disabled,1:Enabled
  // @User: Standard
--]]
EFI_INF_ENABLE = bind_add_param("ENABLE", 1, 1)

--[[
  // @Param: EFI_INF_OPTIONS
  // @DisplayName: EFI INF-Inject options
  // @Description: EFI INF driver options
  // @Bitmask: 0:EnableLogging
  // @User: Standard
--]]
EFI_INF_OPTIONS = bind_add_param("OPTIONS", 2, 0)

--[[
  // @Param: EFI_INF_THR_HZ
  // @DisplayName: EFI INF-Inject throttle rate
  // @Description: EFI INF throttle output rate
  // @Range: 0 50
  // @Units: Hz
  // @User: Standard
--]]
EFI_INF_THR_HZ = bind_add_param("THR_HZ", 3, 0)

--[[
  // @Param: EFI_INF_IGN_AUX
  // @DisplayName: EFI INF-Inject ignition aux function
  // @Description: EFI INF throttle ignition aux function
  // @User: Standard
--]]
EFI_INF_IGN_AUX = bind_add_param("IGN_AUX", 4, 300)

local OPTION_LOGGING = (1<<0)

--[[
   return true if an option is enabled
--]]
local function option_enabled(option)
   return (EFI_INF_OPTIONS:get() & option) ~= 0
end

if EFI_INF_ENABLE:get() ~= 1 then
   return
end

local EFI_FUEL_DENS = bind_param("EFI_FUEL_DENS")
local SCR_VM_I_COUNT = bind_param("SCR_VM_I_COUNT")

local uart = serial:find_serial(0) -- first scripting serial
if not uart then
   gcs:send_text(MAV_SEVERITY.ERROR, "EFI_INF: unable to find scripting serial")
   return
end
uart:begin(9600)

local efi_backend = efi:get_backend(0)
if not efi_backend then
   gcs:send_text(MAV_SEVERITY.ERROR, "EFI_INF: unable to find EFI backend")
   return
end

--[[
   we need a bit more time in this driver
--]]
if SCR_VM_I_COUNT:get() < 50000 then
   gcs:send_text(MAV_SEVERITY.INFO, "EFI_INF: raising SCR_VM_I_COUNT to 50000")
   SCR_VM_I_COUNT:set_and_save(50000)
end

local state = {}
state.last_read_us = uint32_t(0)
state.chk0 = 0
state.chk1 = 0
state.total_fuel_g = 0.0

local last_throttle_send_ms = uint32_t(0)
local last_ignition_send_ms = uint32_t(0)
local last_ign_sw_pos = -1

local file_handle = nil
local efi_device_id = nil

--[[
   log a set of bytes
--]]
local function log_bytes(s)
   if not file_handle then
      file_handle = io.open("INF_Inject.log", "w")
   end
   if file_handle then
      local magic = 0x7fe53b04
      local now_ms = millis():toint()
      local hdr = string.pack("<III", magic, now_ms, string.len(s))
      file_handle:write(hdr)
      file_handle:write(s)
   end
end

local function read_bytes(n)
   local ret = ""
   for _ = 1, n do
      local b = uart:read()
      state.chk0 = state.chk0 ~ b
      state.chk1 = state.chk1 ~ state.chk0
      ret = ret .. string.char(b)
   end
   if option_enabled(OPTION_LOGGING) then
      log_bytes(ret)
   end
   return ret
end

--[[
   convert grams of fuel to cm3
--]]
local function gram_to_cm3(g)
   local kg = g * 0.001
   local kg_per_m3 = EFI_FUEL_DENS:get()
   if kg_per_m3 <= 0 then
      kg_per_m3 = 742.9
   end
   local m3 = kg / kg_per_m3
   return m3 * 1.0e6
end

--[[
   check for input and parse data
--]]
local function check_input()
   local packet_size = 83
   local n_bytes = uart:available():toint()
   if n_bytes < packet_size then
      return false
   end

   local tus = micros()

   -- sync on header start
   local header_ok = false
   while n_bytes >= packet_size and not header_ok do
      state.chk0 = 0
      state.chk1 = 0
      local header0 = string.unpack("<B", read_bytes(1))
      n_bytes = n_bytes - 1
      if header0 == 0xB5 then
         local header1 = string.unpack("<B", read_bytes(1))
         n_bytes = n_bytes - 1
         if header1 == 0x62 then
            header_ok = true
         end
      end
   end
   if not header_ok or n_bytes < packet_size-2 then
      return false
   end

   -- look for basic data table 2
   local _, _, dtype, num, device_id, ack = string.unpack("<BBBBIB", read_bytes(9))
   if dtype ~= 0x02 then
      return false
   end
   if ack ~= 0x50 then
      return false
   end
   if num < packet_size then
      return false
   end

   --gcs:send_text(MAV_SEVERITY.INFO, string.format("packet start"))

   state.mode, state.sta, state.sta1, state.bus_thr = string.unpack("<BBBB", read_bytes(4))
   state.bus_thr = state.bus_thr * 0.1
   state.svr_pwm, state.rpm_out, state.rpm1, state.rpm2 = string.unpack("<BHHH", read_bytes(7))
   state.svr_pwm = state.svr_pwm * 0.1
   state.tmp_env, state.tmp0, state.tmp1, state.tmp2, state.tmp3 = string.unpack("<hhhhh", read_bytes(10))
   state.vol_power, state.vol_svr, state.vol_pump, state.amp_pump = string.unpack("<HBBB", read_bytes(5))
   state.vol_power = state.vol_power * 0.1
   state.vol_svr = state.vol_svr * 0.1
   state.vol_pump = state.vol_pump * 0.1
   state.amp_pump = state.amp_pump * 0.1
   state.ADC1, state.ADC2, state.pre_gas, state.pre_alt = string.unpack("<BBfh", read_bytes(8))
   state.ADC1 = state.ADC1 * 0.1
   state.ADC2 = state.ADC2 * 0.1
   state.PWM_IN1, state.PWM_IN2, state.PWM_IN3 = string.unpack("<BBB", read_bytes(3))
   state.PWM_IN1 = state.PWM_IN1 * 10
   state.PWM_IN2 = state.PWM_IN2 * 10
   state.PWM_IN3 = state.PWM_IN3 * 10
   state.PWM_OUT1, state.PWM_OUT2, state.PWM_OUT3 = string.unpack("<BBB", read_bytes(3))
   state.pre_oil, state.inj1_ms, state.inj2_ms = string.unpack("<fhh", read_bytes(8))
   state.inj1_mg, state.inj2_mg = string.unpack("<hh", read_bytes(4))
   state.adc1_thr, state.adc2_thr = string.unpack("<BB", read_bytes(2))
   state.adc1_thr = state.adc1_thr * 10
   state.adc2_thr = state.adc2_thr * 10
   state.ecu_run_time, state.err_flg = string.unpack("<IH", read_bytes(6))
   state.oil1_thr, state.oil2_thr = string.unpack("<BB", read_bytes(2))
   state.oil1_thr = state.oil1_thr * 10
   state.oil2_thr = state.oil2_thr * 10
   state.reserve1, state.reserve2 = string.unpack("<BB", read_bytes(2))
   state.tmp4, state.tmp5 = string.unpack("<hh", read_bytes(4))

   local chk0 = state.chk0
   local chk1 = state.chk1
   state.check0, state.check1 = string.unpack("<BB", read_bytes(2))

   --[[
      the device will sometimes use 0 for the 2nd 8 bits of the checksum
      we will accept these packets, relying on the other header checks
   --]]
   local checksum_ok = chk0 == state.check0 and (chk1 == state.check1 or state.check1 == 0)
   if not checksum_ok then
      gcs:send_text(MAV_SEVERITY.INFO, string.format("chksum wrong (0x%02x,0x%02x) (0x%02x,0x%02x)", chk0, chk1, state.check0, state.check1))
      return false
   end
   state.end0, state.end1 = string.unpack("<BB", read_bytes(2))

   if state.end0 ~= 0x0d or state.end1 ~= 0x0a then
      return false
   end

   local dt = (tus - state.last_read_us):tofloat()*1.0e-6

   local fuel_g_rev = (state.inj1_mg + state.inj2_mg)
   state.fuel_g_per_min = fuel_g_rev * state.rpm1
   state.total_fuel_g = state.total_fuel_g + (state.fuel_g_per_min * dt / 60.0)

   state.last_read_us = micros()

   gcs:send_named_float('VOL_SRV', state.vol_svr)
   gcs:send_named_float('VOL_PUMP', state.vol_pump)
   gcs:send_named_float('AMP_PUMP', state.amp_pump)
   gcs:send_named_float('PWM_OUT2', state.PWM_OUT2)
   gcs:send_named_float('INF_ETEMP', state.tmp_env)
   gcs:send_named_float('INF_TEMP1', state.tmp0)
   gcs:send_named_float('INF_TEMP2', state.tmp1)

   if not efi_device_id then
      gcs:send_text(MAV_SEVERITY.INFO, string.format("EFI DeviceID: %u", device_id))
      efi_device_id = device_id
   end

   return true
end

--[[
   update EFI state
--]]
local function update_EFI()
   if state.last_read_us == uint32_t(0) then
      return
   end
   local cylinder_state = Cylinder_Status()
   local efi_state = EFI_State()

   cylinder_state:cylinder_head_temperature(state.tmp0)
   cylinder_state:exhaust_gas_temperature(state.tmp1)
   cylinder_state:injection_time_ms(state.inj1_ms)

   efi_state:engine_speed_rpm(state.rpm_out)

   efi_state:atmospheric_pressure_kpa(state.pre_gas*0.01)
   efi_state:intake_manifold_temperature(state.tmp2)
   efi_state:throttle_position_percent(math.floor(state.bus_thr*0.1))
   efi_state:ignition_voltage(state.vol_power)

   efi_state:cylinder_status(cylinder_state)
   efi_state:last_updated_ms(millis())
   efi_state:fuel_pressure(state.pre_oil)

   efi_state:fuel_consumption_rate_cm3pm(gram_to_cm3(state.fuel_g_per_min))
   efi_state:estimated_consumed_fuel_volume_cm3(gram_to_cm3(state.total_fuel_g))

   -- Set the EFI_State into the EFI scripting driver
   efi_backend:handle_scripting(efi_state)
end

local function get_checksum(pkt)
   local chk0 = 0
   local chk1 = 0
   for i=1,#pkt do
      local b = string.byte(pkt,i,i)
      chk0 = chk0 ~ b
      chk1 = chk1 ~ chk0
    end
   return (chk1 << 8) | chk0
end

--[[
   send command packet
--]]
local function send_packet(cmd, content)
   local pkt = string.pack("<BBBBBBIBBI",
                           0xB5, 0x62,
                           0xc3, 0xa1, 18, 20,
                           efi_device_id, 0x50,
                           cmd, content)
   local crc = get_checksum(pkt)
   pkt = pkt .. string.pack("<HBB", crc, 0x0d, 0x0a)
   for i=1,#pkt do
      local b = string.byte(pkt,i,i)
      uart:write(b)
   end
end

--[[
   send throttle commands
--]]
local function update_throttle()
   if EFI_INF_THR_HZ:get() <= 0 then
      return
   end
   local now_ms = millis()
   local rate_ms = 1000.0 / EFI_INF_THR_HZ:get()
   if now_ms - last_throttle_send_ms < rate_ms then
      return
   end
   last_throttle_send_ms = now_ms
   local thr_k = SRV_Channels:get_output_scaled(K_THROTTLE)*10

   send_packet(CMD_THROTTLE, thr_k)
end

--[[
   send ignition commands
--]]
local function update_ignition()
   local aux_fn = EFI_INF_IGN_AUX:get()
   if not aux_fn then
      return
   end
   local sw_pos = rc:get_aux_cached(aux_fn)
   if not sw_pos then
      return
   end
   local now_ms = millis()
   if sw_pos == last_ign_sw_pos and now_ms - last_ignition_send_ms < 1000 then
      return
   end
   last_ignition_send_ms = now_ms
   local command = 0
   if sw_pos >= 1 then
      command = 1
   end
   if sw_pos ~= last_ign_sw_pos then
      onoff = "OFF"
      if command == 1 then
         onoff = "ON"
      end
      gcs:send_text(MAV_SEVERITY.INFO, string.format("EFI_INF: ignition %s", onoff))
   end
   last_ign_sw_pos = sw_pos
   send_packet(CMD_CDI1, command)
   send_packet(CMD_CDI2, command)
   send_packet(CMD_OIL_PUMP, command)
end


--[[
   main update function
--]]
local function update()
   if check_input() then
      update_EFI()
   end
   if efi_device_id then
      update_throttle()
      update_ignition()
   end
   return update, 10
end

gcs:send_text(MAV_SEVERITY.INFO, "EFI_INF: loaded")

return update()