Add X91 TPMS training scenario

pyren/mod_ecu_scenario.py

@@ -22,7 +22,11 @@ def playScenario(command, ecu, elm):
   showable = False
   if scenarioName.lower().startswith('scm'):
     showable = True
-    scenarioName = scenarioName.split(':')[1].lower()
+  
+  if showable:
+    ecuNumberPattern = re.compile(r'\d{5}')
+    ecuNumberIndex = ecuNumberPattern.search(scenarioData)
+    scenarioName = scenarioData[:scenarioData.find(ecuNumberIndex.group(0)) - 1].lower()
     
   if os.path.isfile('./'+scenarioName+'.py'):
     scen = __import__( scenarioName )

pyren/scen_lect_ssppx91.py

@@ -0,0 +1,417 @@
+#!/usr/bin/env python
+'''
+Scenarium usage example
+
+Name of this script should be exactly the same as in scenaruim URL but with '.py' extension
+
+URL  -  scm:scen_ecri_calinj1#scen_ecri_calinj1_xxxxx.xml
+
+'run' procedure will be executed by pyren script
+ 
+'''
+
+import os
+import sys
+import re
+import time
+import string
+import mod_globals
+import mod_utils
+import mod_ecu
+import mod_db_manager
+from   mod_utils     import KBHit
+from   mod_utils     import pyren_encode
+from   mod_utils     import clearScreen
+from   mod_utils     import ASCIITOHEX
+from   mod_utils     import StringToIntToHex
+from   mod_utils     import Choice
+from   collections   import OrderedDict
+import xml.dom.minidom
+
+def run( elm, ecu, command, data ):
+  '''
+  MAIN function of scenarium
+  
+  Parameters:
+      elm     - refernce to adapter class
+      ecu     - reference to ecu class
+      command - refernce to the command this scenarium belongs to
+      data    - name of xml file with parameters from scenarium URL
+  '''
+  
+  clearScreen()
+  header =  '['+command.codeMR+'] '+command.label
+  
+  ScmSet   = {}
+  ScmParam = OrderedDict()
+
+  def get_message( msg, encode = True ):
+    if msg in ScmParam.keys():
+      value = ScmParam[msg]
+    else:
+      value = msg
+    if value.isdigit() and value in mod_globals.language_dict.keys():
+      if encode:
+        value = pyren_encode(mod_globals.language_dict[value])
+      else:
+        value = mod_globals.language_dict[value]
+    return value
+
+  def get_message_by_id( id, encode = True ):
+    if id.isdigit() and id in mod_globals.language_dict.keys():
+      if encode:
+        value = pyren_encode(mod_globals.language_dict[id])
+      else:
+        value = mod_globals.language_dict[id]
+    return value
+  
+  #
+  #      Data file parsing
+  #  
+  DOMTree = xml.dom.minidom.parse(mod_db_manager.get_file_from_clip(data))
+  ScmRoom = DOMTree.documentElement
+
+  ScmParams = ScmRoom.getElementsByTagName("ScmParam")
+ 
+  for Param in ScmParams:
+    name  = pyren_encode( Param.getAttribute("name") )
+    value = pyren_encode( Param.getAttribute("value") )
+        
+    ScmParam[name] = value
+    
+  ScmSets = ScmRoom.getElementsByTagName("ScmSet")
+ 
+  for Set in ScmSets:
+    if len(Set.attributes) != 1:           
+      setname  = pyren_encode(mod_globals.language_dict[Set.getAttribute("name")])
+      ScmParams = Set.getElementsByTagName("ScmParam")
+    
+      for Param in ScmParams:
+        name  = pyren_encode( Param.getAttribute("name")  )
+        value = pyren_encode( Param.getAttribute("value") )
+        
+        ScmSet[setname]= value
+        ScmParam[name] = value
+
+  kb = KBHit()
+
+  confirm = get_message_by_id('19800')
+  confirmCodes = get_message_by_id('17571')
+  resultMessage = get_message('TxtResult')
+  title = get_message('TextTitre', False)
+  stateLabel = get_message_by_id('804')
+  finishedInfo = get_message_by_id('15902')
+  messageInfo = get_message('MessageScr1', False)
+  readCodeMessage = get_message_by_id('17884')
+  codeInfo = get_message('TxtInformation')
+  codeRed = codeInfo.split('-')[1].strip()
+  codeOrange = codeInfo.split('-')[2].strip()
+  codeGreen = codeInfo.split('-')[3].strip()
+  oneValveText = get_message_by_id('14964')
+  fourValveText = get_message_by_id('14966', False)
+
+  NR10 = get_message('Nack10')
+  NR11 = get_message('NackServiceNotSupported')
+  NR12 = get_message('NackSubFunctionNotSupported')
+  NR21 = get_message('NackBusyRepeatRequest')
+  NR22 = get_message('NackConditionsNotCorrect')
+  NR23 = get_message('Nack23')
+  NR33 = get_message('NackSecurityAccessDenied')
+  NR35 = get_message('NackInvalidKey')
+  NR78 = get_message('NackResponsePending')
+  NR80 = get_message('Nack80')
+  NRFF = get_message('NackUnknown')
+
+  negrsp = {
+    '10' : NR10,
+    '11' : NR11,
+    '12' : NR12,
+    '21' : NR21,
+    '22' : NR22,
+    '23' : NR23,
+    '33' : NR33,
+    '35' : NR35,
+    '78' : NR78,
+    '80' : NR80,
+    'FF' : NRFF
+  }
+
+  searchInProgress = []
+  searchInProgress.append(get_message('Label_AVG_En_Cours', False))
+  searchInProgress.append(get_message('Label_AVD_En_Cours', False))
+  searchInProgress.append(get_message('Label_ARD_En_Cours', False))
+  searchInProgress.append(get_message('Label_ARG_En_Cours', False))
+  searchFinished = []
+  searchFinished.append(get_message('Label_AVG_Terminee', False))
+  searchFinished.append(get_message('Label_AVD_Terminee', False))
+  searchFinished.append(get_message('Label_ARD_Terminee', False))
+  searchFinished.append(get_message('Label_ARG_Terminee', False))
+  summerFLCode = ecu.get_id(ScmParam['ID_AVG_ETE'], 1)
+  summerFRCode = ecu.get_id(ScmParam['ID_AVD_ETE'], 1)
+  summerRRCode = ecu.get_id(ScmParam['ID_ARD_ETE'], 1)
+  summerRLCode = ecu.get_id(ScmParam['ID_ARG_ETE'], 1)
+  winterFLCode = ecu.get_id(ScmParam['ID_AVG_HIVER'], 1)
+  winterFRCode = ecu.get_id(ScmParam['ID_AVD_HIVER'], 1)
+  winterRRCode = ecu.get_id(ScmParam['ID_ARD_HIVER'], 1)
+  winterRLCode = ecu.get_id(ScmParam['ID_ARG_HIVER'], 1)
+  summerTyreCodes = [summerFLCode,summerFRCode,summerRRCode,summerRLCode]
+  winterTyreCodes = [winterFLCode,winterFRCode,winterRRCode,winterRLCode]
+  tyreCodes = []
+  tyreCodes.extend(summerTyreCodes)
+  tyreCodes.extend(winterTyreCodes)
+  tyresLabels = []
+  tyresLabels.append(get_message('Label_CodeAVG', False))
+  tyresLabels.append(get_message('Label_CodeAVD', False))
+  tyresLabels.append(get_message('Label_CodeARD', False))
+  tyresLabels.append(get_message('Label_CodeARG', False))
+  value1, datastr1 = ecu.get_st(ScmParam['State_ET469'])
+  tariningCmdResp = ecu.Services[ecu.get_ref_cmd(ScmParam['CmndRoutineTraining']).serviceID[0]].simpleRsp
+
+  currentTyreSet = ''
+  lastCodeStatus = ''
+
+  notUpdatedText = []
+  notUpdatedText.append(title)
+  notUpdatedText.append('*'*80)
+  notUpdatedText.append(messageInfo)
+  notUpdatedText.append('*'*80)
+  notUpdatedText.append(datastr1)
+  notUpdatedText.append('*'*80)
+  tyreCodesTable = []
+  if get_message('Text_ET469_ETE') in datastr1:
+    summerTyresSet = True
+    tyreCodesTable.append("%-50s %-20s"%(tyresLabels[0], summerTyreCodes[0]))
+    tyreCodesTable.append("%-50s %-20s"%(tyresLabels[1], summerTyreCodes[1]))
+    tyreCodesTable.append("%-50s %-20s"%(tyresLabels[2], summerTyreCodes[2]))
+    tyreCodesTable.append("%-50s %-20s"%(tyresLabels[3], summerTyreCodes[3]))
+    currentTyreSet = summerTyreCodes
+  else:
+    summerTyresSet = False
+    tyreCodesTable.append("%-50s %-20s"%(tyresLabels[0], winterTyreCodes[0]))
+    tyreCodesTable.append("%-50s %-20s"%(tyresLabels[1], winterTyreCodes[1]))
+    tyreCodesTable.append("%-50s %-20s"%(tyresLabels[2], winterTyreCodes[2]))
+    tyreCodesTable.append("%-50s %-20s"%(tyresLabels[3], winterTyreCodes[3]))
+    currentTyreSet = winterTyreCodes
+  notUpdatedText = notUpdatedText + tyreCodesTable
+  notUpdatedText.append('*'*80)
+
+  buttons = OrderedDict()
+  buttons[1] = get_message_by_id('14964', False)
+  buttons[2] = get_message_by_id('14966', False)
+  buttons[3] = title
+  buttons["exit"] = '<exit>'
+  
+  def sendTrainingCmd():
+    resp = ecu.run_cmd(ScmParam['CmndRoutineTraining'])
+    clearScreen()
+    if tariningCmdResp not in resp and not mod_globals.opt_demo:
+      nrCode = resp[resp.find('NR') - 3: resp.find('NR')]
+      print
+      if 'NR' in resp:
+        print negrsp[resp[resp.find('NR') - 3: resp.find('NR')]]
+        print ''
+      else:
+        print negrsp['FF']
+      print
+      raw_input('Press any key to exit')
+      return False
+    else:
+      return True
+  
+  def writeCodes(codes):
+    ch = raw_input(confirmCodes + ' <YES/NO>: ')
+    while (ch.upper()!='YES') and (ch.upper()!='NO'):
+      ch = raw_input(confirmCodes + ' <YES/NO>: ')
+    if ch.upper()!='YES':
+      print
+      return
+    print
+    print '*'*80
+    rsp = ecu.run_cmd('VP003', codes)
+    print '*'*80
+    print
+
+  def generateScreen(title, codes):
+    screen = ''
+    screenPartOne = []
+    screenPartOne.append(title)
+    screenPartOne.append('*'*80)
+    screenPartOne.append(datastr1)
+    screenPartOne.append('*'*80)
+    integratedLines = []
+    for num in range(len(tyreCodesTable)):
+      try:
+        integratedLines.append(tyreCodesTable[num] + codes[num])
+      except:
+        integratedLines.append(tyreCodesTable[num])
+    integratedScreen = screenPartOne + integratedLines
+    for lines in integratedScreen:
+      screen = screen + pyren_encode(lines) + '\n'
+    return screen
+
+  def writeOneValve():
+    selectedValve = ''
+    selectedValveKey = ''
+    screen = generateScreen(oneValveText, [])
+    valveLabelsDict = OrderedDict()
+    for lb in range(4):
+      valveLabelsDict[lb] = tyresLabels[lb]
+    valveLabelsDict["exit"] = '<exit>'
+    
+    clearScreen()
+    print screen
+    print
+    choice = Choice(valveLabelsDict.values(), get_message_by_id('14127', False).replace('.',':'))
+
+    clearScreen()
+    print screen
+    print
+    for key, value in valveLabelsDict.iteritems():
+      if choice[0] =='<exit>': return
+      if value == choice[0]:
+        selectedValve = valveLabelsDict[key]
+        selectedValveKey = key
+      
+    userCode = raw_input(selectedValve + ': ').upper()
+    while not len(userCode) == 6 or not all(c in string.hexdigits for c in userCode):
+      if not len(userCode) == 6:
+        print 'Valve code should be 6 characters long.'
+      else:
+        print 'Illegal characters in the valve code'
+      userCode = raw_input(selectedValve + ': ').upper()
+
+    paramToSend = ''
+    if summerTyresSet:
+      for code in summerTyreCodes:
+        paramToSend +=  code
+      paramToSend = paramToSend.replace(summerTyreCodes[selectedValveKey], userCode)
+    else:
+      for code in winterTyreCodes:
+        paramToSend +=  code
+      paramToSend = paramToSend.replace(winterTyreCodes[selectedValveKey], userCode)
+
+    clearScreen()
+    print screen
+    print
+    print tyreCodesTable[selectedValveKey], userCode
+    print
+    writeCodes(paramToSend)
+
+  def writeFourValves():
+    userCodes = []
+    screen = generateScreen(fourValveText, [])
+
+    clearScreen()
+    print screen
+    ch = raw_input(confirm + ' <YES/NO>: ')
+    while (ch.upper()!='YES') and (ch.upper()!='NO'):
+      ch = raw_input(confirm + ' <YES/NO>: ')
+    if ch.upper()!='YES':
+      print
+      return
+    
+    clearScreen()
+    print screen
+    for num in range(len(tyresLabels)):
+      userCode = raw_input(tyresLabels[num] + ': ').upper()
+      while not len(userCode) == 6 or not all(c in string.hexdigits for c in userCode):
+        if not len(userCode) == 6:
+          print 'Valve code should be 6 characters long.'
+        else:
+          print 'Illegal characters in the valve code'
+        userCode = raw_input(tyresLabels[num] + ': ').upper()
+      userCodes.append(userCode)
+      clearScreen()
+      screen = generateScreen(fourValveText, userCodes)
+      print screen
+    
+    paramToSend = ''
+    for code in userCodes:
+      paramToSend += code
+    writeCodes(paramToSend)
+
+  def valvesTraining():
+    readCodes = OrderedDict()
+    if not sendTrainingCmd():
+      return
+    
+    tb = time.time()
+    while(1):
+      readCode = ecu.get_id(ScmParam['ID_Code_Valves'], 1)
+      value2, datastr2 = ecu.get_st(ScmParam['State_ET002'])
+      oldScreen = ''
+      for lines in notUpdatedText:
+        oldScreen = oldScreen + pyren_encode(lines) + '\n'
+      
+      clearScreen()
+      print oldScreen
+      try:
+        print stateLabel + ': ' + searchInProgress[len(readCodes)]
+      except:
+        raw_input('More than 4 tyres codes, aborting.')
+        return
+      print
+      print '*'*80
+
+      if value2 == 3 and len(readCodes) < 4:
+        print "%-50s %-20s"%(readCodeMessage, readCode)
+        print
+        print "No codes read"
+      elif value2 == 2 and len(readCodes) < 4:
+        print "%-50s %-20s"%(readCodeMessage, readCode)
+        print
+
+        if readCode != '000000' and readCode not in readCodes.keys():
+          if not readCode in tyreCodes:
+            lastCodeStatus = codeRed
+          elif readCode == currentTyreSet[len(readCodes)]:
+            lastCodeStatus = codeGreen
+          else:
+            lastCodeStatus = codeOrange
+          readCodes[readCode] = readCode
+        
+        print lastCodeStatus
+        print
+        
+        for code in range(len(readCodes)):
+          print "%-60s %-8s"%(searchFinished[code], readCodes[readCodes.keys()[code]])
+      
+      print '*'*80
+      print
+      print 'Press any key to exit'
+      print
+      time.sleep(int(ScmParam['Tempo2']) / 1000)
+      tc = time.time()
+      if (tc - tb) > int(ScmParam['Tempo1']) / 1000:
+        tb = time.time()
+        if not sendTrainingCmd():
+          return
+      if len(readCodes) == 4:
+        break
+      if kb.kbhit():
+        return
+
+    print finishedInfo
+    print
+    paramToSend = ''
+    for code in readCodes.keys():
+      paramToSend += code
+    if "ERROR" in paramToSend:
+      raw_input("Data downloading went wrong. Aborting.")
+      return
+    writeCodes(paramToSend)
+
+  print get_message_by_id('19830')
+  print
+  choice = Choice(buttons.values(), "Choose :")
+  for key, value in buttons.iteritems():
+    if choice[0] =='<exit>': return
+    if value == choice[0]:
+      if key == 1:
+        writeOneValve()
+      elif key == 2:
+        writeFourValves()
+      elif key == 3:
+        valvesTraining()
+      raw_input('Press any key to exit')
+      return