#!/usr/bin/env python3

from mod_ecu_mnemonic   import *
from mod_utils          import Choice
from xml.dom.minidom    import parse
from xml.dom.minidom    import parseString
import xml.dom.minidom
import mod_globals

def get_state( st, mn, se, elm, calc, dataids = {} ):
  comp = st.computation
  comp = comp.replace("&","&")
  for m in sorted(st.mnemolist, key=len, reverse=True):
    if dataids and mn[m].request.startswith("22"):
      hex_val = get_SnapShotMnemonic(mn[m], se, elm, dataids )
    else:
      hex_val = get_mnemonic( mn[m], se, elm )
    comp = comp.replace(m, "0x"+hex_val) 
  tmp_val = calc.calculate(comp)
  
  if str(tmp_val) in list(st.caracter.keys()):
    st.value = st.caracter[str(tmp_val)]
  else:
    st.value = str(tmp_val)

  if mod_globals.opt_csv and mod_globals.opt_csv_only:
    csv_val = str(tmp_val)
  else:  
    csv_val = str(st.value)
    
  st.value = " "*(16-len(st.value)//2) + str(st.value)
  return "%-6s %-50s %-20s"%(st.codeMR,st.label,st.value), st.helps, csv_val

class ecu_state:

  name        = ""
  agcdRef     = ""
  codeMR      = ""
  mask        = ""
  label       = ""
  value       = ""
  type        = ""
  helps       = []
  caracter    = {}
  computation = ""
  mnemolist   = []
  
  def __str__(self):
    hlps = '['
    for h in self.helps: hlps += '\'' + h + '\'\n'
    hlps += ']'

    chrc = '{\n'
    for c in list(self.caracter.keys()): chrc += c + ' : \'' + self.caracter[c] + '\'\n'
    chrc += '}'
  
    out = '''
  name        = %s
  agcdRef     = %s
  codeMR      = %s
  mask        = %s
  label       = %s
  value       = %s
  type        = %s
  helps       = %s
  caracter    = %s
  computation = %s
  mnemolist   = %s
    ''' % (
           self.name, self.agcdRef, self.codeMR, self.mask, self.label, self.value, self.type,
           hlps, chrc, self.computation, self.mnemolist)
    return pyren_encode(out)  

  def __init__(self, st, opt, tran ):
    self.name = st.getAttribute("name")
    self.agcdRef = st.getAttribute("agcdRef")
    self.codeMR  = st.getAttribute("codeMR")

    Mask = st.getElementsByTagName("Mask")
    if Mask:
      self.mask = Mask.item(0).getAttribute("value")
    
    Label = st.getElementsByTagName("Label")
    codetext = Label.item(0).getAttribute("codetext")
    defaultText = Label.item(0).getAttribute("defaultText")
    self.label = ""
    if codetext:
      if codetext in list(tran.keys()):
        self.label = tran[codetext]
      elif defaultText:
        self.label = defaultText
      
    self.helps = []
    Helps = st.getElementsByTagName("Helps")
    if Helps:
      for hl in Helps:
        Lines = hl.getElementsByTagName("Line")
        if Lines:
          for ln in Lines:
            line = ""
            Label = ln.getElementsByTagName("Label")
            if Label:
              for la in Label:
                codetext = la.getAttribute("codetext")
                defaultText = la.getAttribute("defaultText")
                if codetext:
                  if codetext in list(tran.keys()):
                    line = line + tran[codetext]
                  elif defaultText:
                    line = line + defaultText
            self.helps.append(line+'\n')

    self.caracter = {}
    Interpretation = st.getElementsByTagName("Interpretation")
    if Interpretation:
      for cor in Interpretation:
        Correspondance = cor.getElementsByTagName("Correspondance")
        if Correspondance:
          for co in Correspondance:
            ivalue = co.getAttribute("value")
            codetext  = co.getAttribute("codetext")
            defaultText = co.getAttribute("defaultText")
            itext = ""
            if codetext:
              if codetext in list(tran.keys()):
                itext = tran[codetext]
              elif defaultText:
                itext = defaultText
              self.caracter[ivalue]=itext
    
    try:
      xmlstr = opt["State\\"+self.name]
    except:
      return

    odom = xml.dom.minidom.parseString( xmlstr.encode( "utf-8" ) )
    odoc = odom.documentElement
    
    self.computation = ""
    Computation = odoc.getElementsByTagName("Computation")
    if Computation:
      for cmpt in Computation:
        self.type = cmpt.getAttribute("type")
        tmp = cmpt.getElementsByTagName("Value").item(0).firstChild.nodeValue
        self.computation = tmp.replace(" ","").replace("&","&")
        
        self.mnemolist = []
        Mnemo = cmpt.getElementsByTagName("Mnemo")
        if Mnemo:
          for mn in Mnemo:
            self.mnemolist.append(mn.getAttribute("name"))
  
  def __str__(self) -> str:
    return self.label
  
  def __repr__(self) -> str:
    return str(self)

class ecu_states:
 
  def __init__(self, state_list, mdoc, opt, tran ):
    States = mdoc.getElementsByTagName("State")
    if States:
      for st in States:
        state = ecu_state( st, opt, tran )
        state_list[state.name] = state
    
    Masks = mdoc.getElementsByTagName("MaskList")
    if Masks:
      for ms in Masks:
        DataRef = ms.getElementsByTagName("DataRef")
        for dr in DataRef:
          if dr.getAttribute("type") == 'State':
            name =  dr.getAttribute("name")
            tempStateXml = """<State name="{}" agcdRef="{}" codeMR="{}">
                              <Label codetext="51354" defaultText="FAULT FINDING VERSION"/></State>""".format(name, name, name)
            mdom = xml.dom.minidom.parseString(tempStateXml)
            mdocElem = mdom.documentElement
            state = ecu_state( mdocElem, opt, tran )
            if state.computation != '':
              state_list[state.name] = state