def TabMenu_OnInit():
 Application.SetGlobal("TAB_PPG",PPG);
 BuildLookupTables()
 
LOOKUP_TABLE=None

# Callback for the siOnKeyDownEvent event.
def siOnKeyDownEvent_OnEvent( in_ctxt ):

 keycode = in_ctxt.GetAttribute("KeyCode")
 
 if Application.GetGlobal("TAB_ACTIVE"): 
  if keycode>127:
   return True
   
  in_ctxt.SetAttribute("Consumed",True) 

  ppg = Application.GetGlobal("TAB_PPG")
  tabprop = Application.GetGlobal("TAB_PROP")
  tabdisplay = Application.GetGlobal("TAB_DISPLAY")  
  typedsofar = Application.GetGlobal("TAB_TYPEDSOFAR")
  curstring = Application.GetGlobal("TAB_CURSTRING")
  treename = Application.GetGlobal("TAB_TREENAME")
  treeid = Application.GetGlobal("TAB_TREEID")
  
  lookupTable = BuildLookupTables()
  dic = lookupTable[treeid]
    
  if curstring==None:
   curstring=""   
  
  #Ignore spaces
  if keycode==32:
   return True 
  
  if keycode==13 or keycode==9:
   if keycode==13:
    result = GetAbbreviatedMatch(curstring,dic)
    if result!=None:
     match=result[0]
     
     execString = match[1]
     if execString=='Application.UpdatePlugins()':
      #Special case for update plugins command due to XSI bug
      #causing it to sometimes fail on first call
      try:
       Application.UpdatePlugins()
      except:
       Application.UpdatePlugins()
     else:
      if treeid>0:
       execString = execString.replace("%TREE%",treename)
       
      exec execString
     
   Application.SetGlobal("TAB_ACTIVE",False)

   if ppg:
    ppg.Close()

   Application.SetGlobal("TAB_PPG",None)
   Application.SetGlobal("TAB_PROP",None)
   Application.SetGlobal("TAB_DISPLAY",None)  
   Application.SetGlobal("TAB_CURSTRING","") 
   return True
  
  if keycode==8:
   curstring=curstring[:-1]
  else:
   curstring+=chr(keycode)
   
  if len(curstring)!=0:
   result = GetAbbreviatedMatch(curstring,dic)
   if result!=None:
    matchName=result[0][0]
    extStr=result[1]
 
    displayString = ExpandMatchToIncludeSpaces(extStr, matchName)
    
    typedsofar.Label = displayString
    tabdisplay.Label = matchName
    Application.SetGlobal("TAB_CURSTRING",curstring)
    if ppg:
     ppg.Refresh();
  else:
   Application.SetGlobal("TAB_CURSTRING","")
   tabdisplay.Label = ""
   typedsofar.Label = ""
   if ppg:
    ppg.Refresh();
  
 elif keycode==9:
  tabprop = XSIFactory.CreateObject("CustomProperty");
  layout = tabprop.PPGLayout

  treeInfo = GetActiveNodeTree()
  treeType = treeInfo[1]
  
  tabtypedsofar = layout.AddStaticText("");
  tabdisplay = layout.AddStaticText("");
  if treeType==0:
   tabprop.Name = "Tab Menu (Main)";
  if treeType==1:
   tabprop.Name = "Tab Menu (ICE Tree)";
  if treeType==2:
   tabprop.Name = "Tab Menu (Render Tree)";
    
  Application.SetGlobal("TAB_ACTIVE",True)
  Application.SetGlobal("TAB_PROP",tabprop)
  Application.SetGlobal("TAB_DISPLAY",tabdisplay)
  Application.SetGlobal("TAB_TYPEDSOFAR",tabtypedsofar)
  Application.SetGlobal("TAB_CURSTRING","")
  Application.SetGlobal("TAB_TREENAME", treeInfo[0])
  Application.SetGlobal("TAB_TREEID", treeInfo[1])
    
  layout.Language = "Python"
  layout.SetAttribute(constants.siUILogicPrefix, "TabMenu_")
  layout.SetAttribute(constants.siUILogicFile, Application.Plugins(g_pluginName).Filename)

  Application.InspectObj(tabprop);

  in_ctxt.SetAttribute("Consumed",True);  
 
 return True
 
def ExpandMatchToIncludeSpaces(match,fullString):
 nchar = len(match)
 result=""
 for c in fullString:
  result+=c
  if c!=" ":
   nchar-=1
  if nchar==0:
   break
 return result
 
def GetAbbreviatedMatch(curstring,dic):
 matchStr = ""
 for c in curstring:
  matchStr += c
  result = GetFirstMatch(matchStr,dic)
  if result==None:
   break
  matchStr = result[1]
 return result

def AddDefinition(name,cmd, dic={}):
 curdic=dic
 for c in name.upper():
  if c==" ":
   continue
  if not curdic.has_key(c):
   curdic[c]={}
  curdic = curdic[c]
 curdic['   node']=[name,cmd]

def GetFirstMatch(name,dic):
 curdic=dic
 for c in name:
  if not curdic.has_key(c):
   return None
  curdic = curdic[c]
 
 if curdic.has_key('   node'):
  return [curdic['   node'],name]

 longestMatch=name
 longestFound=False
 
 #Don't know how deep the tree goes, so set this to a big number instead of
 #using while(true) to make sure we never enter an infinite loop
 #(it should never happen but it's good practice to check)
 i=0
 while i<1000:
  keys = curdic.keys()
  keys.sort()
  
  if keys[0]=='   node':
   return [curdic['   node'], longestMatch]

  curdic = curdic[keys[0]]
  if not longestFound:
   if len(keys)==1:
    longestMatch+=keys[0]
   else:
    longestFound=True
  i+=1

 LogMessage("Infinite loop detected: Aborting")
 return None

def BuildLookupTables():
 global LOOKUP_TABLE

 if LOOKUP_TABLE != None:
  return LOOKUP_TABLE
 
 lookupTables=[{},{},{}]

 pluginPath = Application.Plugins(g_pluginName).OriginPath

 ReadTabDefinitionFile(pluginPath+"/Main_TabDefinitions.txt",lookupTables[0])
 ReadTabDefinitionFile(pluginPath+"/ICE_TabDefinitions.txt",lookupTables[1])
 ReadTabDefinitionFile(pluginPath+"/Render_TabDefinitions.txt",lookupTables[2]) 
 
 LOOKUP_TABLE=lookupTables
 return LOOKUP_TABLE
 
 
def ReadTabDefinitionFile(filename, dic):
 try:
  defFile = open(filename)
 except:
  LogMessage("Can't find Tab Definition File: "+filename,constants.siError)
  return False

 homeDir = Application.GetInstallationPath2(constants.siFactoryPath)  
    
 try:
  for line in defFile:
   curLine = line.strip()
   if curLine!="":
    cmdName = curLine[:curLine.find(",")].strip()
    cmdLine = curLine[curLine.find(",")+1:].strip()
    
    #We need to replace this because some of the commands don't resolve it correctly,
    #and we need to have it work independently of install location
    cmdLine = cmdLine.replace("$XSI_HOME",homeDir)
    
    AddDefinition(cmdName,cmdLine,dic)
 except:
  defFile.close()
  
 return True

def GetActiveNodeTree():
 try:
  vm = Application.Desktop.ActiveLayout.Views.Find( "View Manager" )
  fc = vm.GetAttributeValue("focusedviewport")
 except:
  return ("",0)
  
 activeView = vm.Views(fc)
 iceTreeName = activeView.GetAttributeValue("container")
 
 if iceTreeName=="None":
  return ("",0)
 col = win32com.client.Dispatch( "XSI.Collection" )
 try:
  col.SetAsText(iceTreeName)
 except:
  return ("",0)
 container = col(0)
 
 if container.Type=="ICETree":
  return (container.FullName,1)
 if container.Families.find("NodeOperators")!=-1: #ICE Compound
  return (container.FullName,1) 
 if container.Type=="material":
  return (container.FullName,2) 
 if container.Families.find("Shader Compounds")!=-1: #Render Compound 
  return (container.FullName,2)
 return ("",0)