#!/usr/bin/env python

import sys, os
import time
import mod_utils
import mod_db_manager
    
import mod_globals
import mod_ecu
import pyren

os.chdir(os.path.dirname(os.path.realpath(sys.argv[0])))

try:
  import androidhelper as android
  mod_globals.os = 'android'
except:
  try:
    import android
    mod_globals.os = 'android'
  except:
    pass

if mod_globals.os != 'android':    
  try:
    import serial
    from serial.tools  import list_ports
  except ImportError:
    sys.exit()
    
from mod_elm       import ELM
from mod_scan_ecus import ScanEcus
from mod_ecu       import ECU
from mod_optfile   import *
from mod_utils     import *

def prepareECUs():
  '''This function loads data for ECUs'''
  
  global elm
  global ecu
  global se
  global lang
  
  pyren.optParser()

  mod_utils.chkDirTree()
  mod_db_manager.find_DBs()

  if len(mod_globals.opt_log)==0:
    mod_globals.opt_log = 'commander_log.txt'
  
  print('Opening ELM')
  elm = ELM( mod_globals.opt_port, mod_globals.opt_speed, mod_globals.opt_log )

  print('Loading ECUs list')
  se  = ScanEcus(elm)                    #Prepare list of all ecus

  if not os.path.isfile("savedEcus.p") or mod_globals.opt_scan: 
    # choosing model 
    se.chooseModel( mod_globals.opt_car )  #choose model of car for doing full scan
  
  # Do this check every time
  se.scanAllEcus()                         #First scan of all ecus
  
  print("Loading language ")
  sys.stdout.flush()
                                         #loading language data
  lang = optfile("Location/DiagOnCAN_"+mod_globals.opt_lang+".bqm",True)
  mod_globals.language_dict = lang.dict
  print("Done")
     
  return se.detectedEcus
  
def chooseEcu( ecu_number ):

  global elm
  global ecu
  global se
  global lang
  
  choosen_ecu = se.chooseECU( ecu_number ) 
  if choosen_ecu==-1:
    print("#\n"*3,"#   Unknown ECU defined!!!\n","#\n"*3)
    exit(1)
  
  ecucashfile = "./cache/"+choosen_ecu['ModelId']+'_'+mod_globals.opt_lang+".p"
  
  if os.path.isfile(ecucashfile):                         #if cache exists
    ecu = pickle.load( open( ecucashfile, "rb" ) )        #load it
  else:                                                   #else
    ecu = ECU(choosen_ecu, lang.dict )                    #load original data for chosen ECU
    pickle.dump( ecu, open( ecucashfile, "wb" ) )         #and save data to cache for next time
  
  ecu.initELM( elm )                                      #init ELM for chosen ECU 
  
  #ecu.show_screens()                                      # show ECU screens

def main():
  list = prepareECUs()

  tot = ''
  
  for l in list:
    if l['idf']=='1':  #family 01
      print("### Connecting to Engine ###")
      chooseEcu(l['ecuname'])
      tot += "%-15s : " % "Engine    PR025" 
      num, string = ecu.get_pr('PR025')
      print(pyren_encode(string))
      tot += str(num); tot += '\n'
      tot += "%-15s : " % "Engine    PR992" 
      num, string = ecu.get_pr('PR992')
      print(pyren_encode(string))
      tot += str(num); tot += '\n'
      num, string = ecu.get_pr('PR391')
      print(pyren_encode(string))
      num, string = ecu.get_pr('PR412')
      print(pyren_encode(string))
      #num, string = ecu.get_pr('PR804')
      #print pyren_encode(string)
      #num, string = ecu.get_pr('PR869')
      #print pyren_encode(string)
      #num, string = ecu.get_pr('PR870')
      #print pyren_encode(string)
      print()
    if l['idf']=='2':  #family 02
      print("### Connecting to ABS ###") 
      chooseEcu(l['ecuname'])
      tot += "%-15s : " % "ABS       PR121" 
      num, string = ecu.get_pr('PR121')
      print(pyren_encode(string))
      tot += str(num); tot += '\n'
      print()
    if l['idf']=='3':  #family 03
      print("### Connecting to TDB ###") 
      chooseEcu(l['ecuname'])
      tot += "%-15s : " % "TDB       PR009" 
      num, string = ecu.get_pr('PR009')
      print(pyren_encode(string))
      tot += str(num); tot += '\n'
      tot += "%-15s : " % "TDB (km)  PR025" 
      num, string = ecu.get_pr('PR025')
      print(pyren_encode(string))
      tot += str(num); tot += '\n'
      tot += "%-15s : " % "TDB (mil) PR026" 
      num, string = ecu.get_pr('PR026')
      print(pyren_encode(string))
      tot += str(num); tot += '\n'
      print()
  if mod_globals.os != 'android':  
    print(pyren_encode('Listening to CAN. Please wait a bit...')) 
    elm.cmd('at z')
    elm.cmd("at e1")
    elm.cmd("at l1")
    elm.cmd("at h1")
    elm.cmd("at d1")
    elm.cmd("at caf0")
    elm.cmd("at sp 6")
    elm.cmd("at al")
    elm.portTimeout = 1

    elm.cmd("at cf 5C5")
    elm.cmd("at cm 7FF")
    elm.cmd("at cra 5C5")
    resp = elm.cmd("atma")
    elm.cmd("at")
    for l in resp.split('\n'):
      if l.upper().startswith('5C5'):
        kmt = l[9:18].replace(' ','')
        tot += "%-10s : " % "Frame 5C5" 
        tot = tot + str(int(kmt,16)); tot += '\n'
        break
        
    elm.cmd("at cf 715")
    elm.cmd("at cm 7FF")
    elm.cmd("at cra 715")
    elm.portTimeout = 5
    resp = elm.cmd("atma")
    elm.portTimeout = 1
    elm.cmd("at")
    for l in resp.split('\n'):
      if l.upper().startswith('715'):
        kmt = l[6:15].replace(' ','')
        tot += "%-10s : " % "Frame 715" 
        tot = tot + str(int(kmt,16)); tot += '\n'
        break

    elm.cmd("at cf 5FD")
    elm.cmd("at cm 7FF")
    elm.cmd("at cra 5FD")
    elm.portTimeout = 5
    resp = elm.cmd("atma")
    elm.portTimeout = 1
    elm.cmd("at")
    for l in resp.split('\n'):
      if l.upper().startswith('5FD'):
        kmt = l[6:15].replace(' ','')
        tot += "%-10s : " % "Frame 5FD" 
        tot = tot + str(int(kmt,16)); tot += '\n'
        break
            
  #print tot
  elm.lastMessage = tot
    
if __name__ == '__main__':  
  main()