controllermode.lua

--todo: situation based controls
function startControllerMode()
  --[[local--]] windowlist={MainForm} --visible windowlist
  formhistory={}
  local formChangeNotificationID,formRemovalNotificationID
  currentForm=MainForm  
  getLuaEngine().cbShowOnPrint.Checked=false
  
  
  function showForm(f)
    if getApplication().ModalLevel>0 then
      print("Modal mode")
      return
    end

    if f~=currentForm then
      table.insert(formhistory,currentForm)
    end
    
    while #formhistory>10 do
      table.remove(formhistory,1)      
    end
       
    
    printf("showForm:%s",tostringx(f))
    if currentForm and currentForm.formState:find('fsModal') then return end --the current form is modal
    
    local isnew=true
    for i=1,#windowlist do
      if f==windowlist[i] then
        isnew=false
        break
      end
    end
    
    if isnew then
      print("unknown form. adding it to the list")
      table.insert(windowlist,f) 
    end
    
    for i=1,#windowlist do
      windowlist[i].visible=windowlist[i]==f --hide (not close) all other windows, except this one        
    end
    f.show()
    currentForm=f
  end
  
  formRemovalNotificationID=registerFormRemoveNotification(function(f)
    printf("removeForm:%s",tostringx(f))
    
    local i=1
    while i<=#formhistory do    
      if formhistory[i]==f then
        table.remove(formhistory,i)
      else
        i=i+1
      end
    end
    
    for i=2,#windowlist do
      if windowlist[i]==f then
        table.remove(windowlist,i)
      
        local nextform
        if f==currentForm then
          currentForm=nil
          nextform=windowlist[i-1]
          showForm(nextform)          
        end      
        
          
        return
      end
    end
  end)
  

  local laststate

  function startPress()
    local a=getApplication()
    if a.ProcessWindow==nil or a.ProcessWindow.Visible==false then
      MainForm.sbOpenProcess.doClick()
    end
  end

  if f then
    f.destroy()
    f=nil
  end

  local dpadrepeater

  function handleDPAD(v,state)
    if (state==false) then --key got released

      if dpadrepeater then
        dpadrepeater.destroy()
        dpadrepeater=nil
      end
    end

    if dpadrepeater then
      dpadrepeater.enabled=false
      dpadrepeater=nil
    end

    if state then

      local f=getFocusedForm()
      
     
      
      if f then
        
        local c=f.ActiveControl
        if c then
          print("dpad "..v.." f="..f.Name..':'..f.ClassName..' - '..c.Name..':'..c.ClassName)

          if c.inheritsFrom('TCustomTreeView') then
            dpadrepeater=createTimer(200, function() handleDPAD(v, true) end)          
          else
            dpadrepeater=createTimer(120, function() handleDPAD(v, true) end)
          end
          

          if v==0 then --up
            --print("up")
            if c.ItemIndex then
              if c.ItemIndex>0 then
                c.ItemIndex=c.ItemIndex-1
              end
            elseif c.CaretY then
              c.CaretY=c.CaretY-1
            else
              if c.inheritsFrom('TCustomTreeView') then
                --treeview inherited
                if c.Selected==nil then
                  c.Selected=c.items.getFirstVisibleNode()                  
                else
                  local nextnode=c.Selected.getPrevVisible()
                  if nextnode then
                    c.Selected=nextnode
                  end
                end
                
                if c.Selected then
                  c.Selected.makeVisible()
                end
                
              elseif c.ClassName=='THexView' then
                c.Address=c.Address-c.BytesPerLine
              end
            end

          elseif v==1 then --down
            --print("down")
            if c.ItemIndex then
              if c.ItemIndex<c.Items.Count-1 then
                c.ItemIndex=c.ItemIndex+1
              end
            elseif c.CaretY then
              c.CaretY=c.CaretY+1
            else
              if c.inheritsFrom('TCustomTreeView') then
                --treeview inherited
                if c.Selected==nil then
                  c.Selected=c.Items.getLastVisibleNode()                 
                else
                  local nextnode=c.Selected.getNextVisible()
                  if nextnode then
                    c.Selected=nextnode
                  end
                end
                
                if c.Selected then
                  c.Selected.makeVisible()
                end
                
              elseif c.ClassName=='THexView' then
                c.Address=c.Address+c.BytesPerLine
              elseif c.ClassName=='TDisassemblerview' then
                c.SelectedAddress=c.SelectedAddress+getInstructionSize(c.SelectedAddress)
              end
            end
          elseif v==2 then --left
            if c.inheritsFrom('TCustomTreeView') then
              --treeview inherited
              if c.Selected==nil then
                c.Selected=c.Items.getFirstVisibleNode()    
              else
                local nextnode=c.Selected.Parent
                if nextnode then
                  c.Selected=nextnode
                end
              end
                
              if c.Selected then
                c.Selected.makeVisible()
              end                
            elseif c.ClassName=='THexView' then
              c.Address=c.Address-1
            end
          elseif v==3 then --right
            if c.inheritsFrom('TCustomTreeView') then
              --treeview inherited
              if c.Selected==nil then
                c.Selected=c.Items.getLastVisibleNode()    
              else
                if c.Selected.HasChildren then
                  local nextnode=c.Selected.getFirstVisibleChild()
                  if nextnode then
                    c.Selected=nextnode
                  end
                end
              end
                
              if c.Selected then
                c.Selected.makeVisible()
              end                
            elseif c.ClassName=='THexView' then
              c.Address=c.Address+1
            end
          end
        end
      end
    end
  end

  controllerKeyHandler={}
  controllerKeyHandler.GAMEPAD_DPAD_UP=function(state) handleDPAD(0,state) end
  controllerKeyHandler.GAMEPAD_DPAD_DOWN=function(state) handleDPAD(1,state) end
  controllerKeyHandler.GAMEPAD_DPAD_LEFT=function(state) handleDPAD(2,state) end
  controllerKeyHandler.GAMEPAD_DPAD_RIGHT=function(state) handleDPAD(3,state) end

  controllerKeyHandler.GAMEPAD_LEFT_SHOULDER=function(state)   
    if getApplication().ModalLevel>0 then return end
    
    if state then      
      local windowIndex
      for i=1,#windowlist do
        if windowlist[i]==currentForm then
          windowIndex=i
          break
        end
      end
      
      if windowIndex==nil then
        print("left shoulder: currentForm was not found")
        showForm(MainForm)
        return        
      end
      
      
      if windowIndex==1 then 
        windowIndex=#windowlist
      else
        windowIndex=windowIndex-1
      end
      showForm(windowlist[windowIndex])
    end
  end

  controllerKeyHandler.GAMEPAD_RIGHT_SHOULDER=function(state)
    if getApplication().ModalLevel>0 then return end
      
    if state then      
      local windowIndex
      for i=1,#windowlist do
        if windowlist[i]==currentForm then
          windowIndex=i
          break
        end
      end
      
      if windowIndex==nil then
        print("right shoulder: currentForm was not found")
        showForm(MainForm)
        return
      end      
      
      if windowIndex==#windowlist then
        windowIndex=1
      else
        windowIndex=windowIndex+1
      end
      showForm(windowlist[windowIndex])
    end
  end

  controllerKeyHandler.GAMEPAD_A=function(state)
    if state then
      --print("A")
      local f=getFocusedForm()
      if f then
        local c=f.ActiveControl
        
        if c==nil then return end

        if c.ClassName=='TTreeviewWithScroll' then --addresslist
          if c.Selected and c.Selected.memrec then
            c.Selected.memrec.Active=not c.Selected.memrec.Active
            return
          end
        end
        

        if c.ClassName=='TEdit' then
          --todo
          --spawn a keyboard. On steamdeck use the steamdeck keyboard. todo: create my own keyboard ?
          showMessage('editing here')

          --if the result doesn't include Enter/Return then
          return
        end

        if c.OnClick  then
          queue(function()
            c.doClick()
            if c.ModalResult and c.ModalResult~=0 then
              f.ModalResult=c.ModalResult
            end
          end)
          return
        end

        if c.OnDblClick then
          queue(function()
            c.OnDblClick(c)
            if c.ModalResult and c.ModalResult~=0 then
              f.ModalResult=c.ModalResult
            end
          end)
          return
        end

        if c.ModalResult and c.ModalResult~=0 then
          f.ModalResult=c.ModalResult
          return
        end

        if f.FormState:find('fsModal') then
          if f.DefaultControl then
            f.DefaultControl.doClick()
            if f.ModalResult==0 then
              f.ModalResult=f.DefaultControl.ModalResult
              if f.ModalResult then return end
            else
              return
            end
          end
        end

        local c=f.ActiveControl
        if c then
          queue(function()
            c.doClick()
          end)

        else

        end
      else
        --print("no form")
      end
    end
  end

  controllerKeyHandler.GAMEPAD_B=function(state)
    if state then
      --print("B")
      local f=getFocusedForm()
      if f then
        c=f.ActiveControl
        
        

        if f.CancelControl then
          f.CancelControl.doClick()
          if f.ModalResult==0 then
            f.ModalResult=f.CancelControl.ModalResult
            if f.ModalResult then return end
          else
            return
          end
        else
          if f.FormState:find('fsModal') then
            f.ModalResult=mrCancel
          end
        end
      end
    end
  end

  controllerKeyHandler.GAMEPAD_X=function(state)
    if state then
      local f=getFocusedForm()
      if f then
        local c=f.ActiveControl
        if c then
          --c.color=clRed
          c.performTab(laststate.RightTrigger<127)

          c=f.ActiveControl
          --c.color=clGreen
        end
      end
    end
  end

  controllerKeyHandler.GAMEPAD_Y=function(state)
    if state then
      if newAddressList then
        if getFocusedForm()==newAddressList then
          --pressing Y when already in the addresslist shows the opentable dialog
          queue(function()
            local f=createOpenTableDialog()
            if f then
              loadTable(f)
            end
          end)
        else            
          newAddressList.show()
        end
      end
    end
  end

  --[[

  x=tab
  y=addresslist
  left trigger+y=open table
  --]]


  controllerKeyHandler.GAMEPAD_START={function() startPress() end}


  f=createThread(function(t)
    local startwasdown=false
    local upwasdown=false
    local lastpacket=0
    local currentControl, currentControlClass
    laststate=getXBox360ControllerState()

    t.freeOnTerminate(false)
    while not t.Terminated do
      local hasfocus
      if getOperatingSystem()==0 then
        hasfocus=getForegroundProcess()==getCheatEnginePreturn rocessID()
      else
        hasfocus=getForegroundWindow()~=0
      end

      if hasfocus then
        local s=getXBox360ControllerState() --todo, add a waitForPacket param so the loop isn't in here
        if s.PacketNumber~=laststate.PacketNumber then

          for controllerfield,value in pairs(s) do
            if value~=laststate[controllerfield] then

              --check if there is a handler for this field
              local handler=controllerKeyHandler[controllerfield]
              if handler then
                if type(handler)=='table' then
                  queue(handler[1])
                else
                  synchronize(handler,value)
                end
              end

            end
          end

          laststate=s

        else
          sleep(16)
        end
      else
        sleep(32)
      end
    end
  end)

  local w=getScreenWidth() / 2
  if newAddressList then
    newAddressList.close()
  end
  newAddressList=createForm()
  newAddressList.Name='newAddressList'
  newAddressList.Caption='Cheat Engine Address List'
  newAddressList.BorderStyle=bsSizeable
  newAddressList.Width=800 --getScreenWidth()
  newAddressList.Height=600 --getScreenHeight()
  newAddressList.Position=poScreenCenter
  newAddressList.PopupMode='pmNone'
  
  local oldheight=MainForm.Height
  MainForm.Height=MainForm.Height-AddressList.Height
  
  
  showForm(newAddressList)

  local al=MainForm.AddressList
  al.Font.Size=20
  al.Parent=newAddressList  
  al.List.setFocus()

  newAddressList.OnResize=function(c)
    local w=newAddressList.clientWidth
    al.Header.Sections[0].Width=w * 0.10
    al.Header.Sections[1].Width=w * 0.30
    al.Header.Sections[2].Width=w * 0.20
    al.Header.Sections[3].Width=w * 0.15 --type
    al.Header.Sections[4].Width=w * 0.25 --value
  end

  newAddressList.OnClose=function()
    local al=MainForm.AddressList
    local w=MainForm.ClientWidth
    MainForm.Height=oldheight
    al.Parent=MainForm.Panel1
    al.Font.Size=0

    al.Header.Sections[0].Width=w * 0.10
    al.Header.Sections[1].Width=w * 0.30
    al.Header.Sections[2].Width=w * 0.20
    al.Header.Sections[3].Width=w * 0.15 --type
    al.Header.Sections[4].Width=w * 0.25 --value

    newAddressList=nil
    
    if formChangeNotificationID then
      unregisterActiveFormChangeNotification(formChangeNotificationID)
      formChangeNotificationID=nil
    end
    
    if formRemovalNotificationID then
      unregisterFormRemoveNotification(formRemovalNotificationID)
      formRemovalNotificationID=nil
    end
    
   -- print("Cleaning the panels")
    
    for i=1,#windowlist do
      windowlist[i].visible=true
      local c=windowlist[i].ControllerPanel
      if c then
        --printf("%s has controller panel. destroying it", tostringx(windowlist[i]))
        c.destroy()
      else
        --printf("%s has no controller panel", tostringx(windowlist[i]))
      end
    end
    
    MainForm.show()
    
    return caFree
  end
  
  local oldmfclose=MainForm.OnCloseQuery
  MainForm.OnCloseQuery=function(...)
    if newAddressList then
      newAddressList.close()
      newAddressList=nil
    end
    return oldmfclose(...)
  end


  function getControllerPanelText(f)
    local s
    local A,B,X,Y
    local rest=''

    if f.Name=='newAddressList' then
      A='Activate/Deactivate'
      Y='Open Table'
      rest='Shoulder=switch window'
    else
      A='Select/Confirm'
      B='Cancel/Back'
      X='Tab'
      
      if f.FormState:find('fsModal')==nil then
        Y='AddressList'
        rest='Shoulder=switch window'
      end
    end

    local c=f.ActiveControl
    if c then
      --add extra buttons
      if c.ClassName=='TEdit' then
        A='Open editor' --if steamdeck
      end
    end

    s=''

    if A then s=s..'A='..A..' ' end
    if B then s=s..'B='..B..' ' end
    if X then s=s..'X='..X..' ' end
    if Y then s=s..'Y='..Y..' ' end
    if rest then s=s..rest end
    return s
  end


  formChangeNotificationID=registerActiveFormChangeNotification(function(f)    
    if f.ControllerPanel==nil then
      print(f.Name..':'..f.ClassName)
      if f.ControlCount>0 then
        lowest=f.Control[0]
        for i=1, f.ControlCount-1 do
          local c=f.Control[i]
          if lowest.Top+lowest.Height<c.Top+c.Height then
            lowest=c
          end
        end
      end

      if lowest then
        p=createPanel(f)
        p.Name='ControllerPanel'
        p.Caption=getControllerPanelText(f)
        p.Color=clRed

        if f.ClassName=='TProcessWindow' then
           --place the panel where it looks best           
           p.Parent=f.Panel1

           p.AnchorSideTop.Control=f.Panel5
           p.AnchorSideTop.Side=asrBottom
           p.BorderSpacing.Top=4

           p.AnchorSideLeft.Control=f.Panel1
           p.AnchorSideLeft.Side=asrLeft

           p.AnchorSideRight.Control=f.Panel1
           p.AnchorSideRight.Side=asrRight

           p.Anchors='[akLeft, akRight, akTop]'
        elseif f.ClassName=='TformSettings' then
           p.Parent=f.Panel6
           p.AnchorSideTop.Control=f.Panel9
           p.AnchorSideTop.Side=asrBottom
           p.BorderSpacing.Top=4

           p.AnchorSideLeft.Control=f.Panel6
           p.AnchorSideLeft.Side=asrLeft

           p.AnchorSideRight.Control=f.Panel6
           p.AnchorSideRight.Side=asrRight

           p.Anchors='[akLeft, akRight, akTop]'
        elseif f.ClassName=='TformAddressChange' then
           p.Parent=f
           p.AnchorSideTop.Control=f.btnOK
           p.AnchorSideTop.Side=asrBottom
           p.BorderSpacing.Top=4

           p.AnchorSideLeft.Control=f
           p.AnchorSideLeft.Side=asrLeft

           p.AnchorSideRight.Control=f
           p.AnchorSideRight.Side=asrRight

           p.Anchors='[akLeft, akRight, akTop]'  
           
           f.Constraints.OnChange=function()
             --print("poo")
             local oldf=f.Constraints.OnChange
             f.Constraints.OnChange=nil
             f.Constraints.MinHeight=p.Top+p.Height             
             f.Constraints.OnChange=oldf
           end
        elseif f.ClassName=='TPromptDialog' then
          p.Parent=f
          p.AnchorSideTop.Control=lowest
          p.AnchorSideTop.Side=asrBottom
          p.BorderSpacing.Top=4

          p.AnchorSideLeft.Control=f
          p.AnchorSideLeft.Side=asrLeft

          p.AnchorSideRight.Control=f
          p.AnchorSideRight.Side=asrRight

          p.Anchors='[akLeft, akRight, akTop]'  
          f.Constraints.MinHeight=p.Top+p.Height
        else
          --unknown/default works fine
          if inheritsFromWinControl(lowest) then
          
            p.Parent=lowest
          end
          p.Align=alBottom
        end
        
        f.Constraints.MinWidth=f.Canvas.getTextWidth(p.Caption)+5


      end
      
      if f.ControllerPanel then
        f.Height=f.Height+f.ControllerPanel.Height
      end
      
    else
      f.ControllerPanel.Caption=getControllerPanelText(f)
    end
    
    
    
    if f.inheritsFrom('THintWindow') then
      print("hintwindow")
      f.visible=false
      CurrentForm.show()
    else
      printf("active form is %s",tostringx(f))
      CurrentForm=f
      local found=false
      for i=1,#windowlist do
        if windowlist[i]==f then
          found=true
          break
        end
      end
      if not found then    
        print("new form. adding it to the windowlist")
        
        table.insert(windowlist,f)
        f.registerCloseCallback(function(frm, ca)
          for i=1,#windowlist do
            if windowlist[i]==frm then
              print("found the form in the windowlist. Removing it. it's not visible anymore") --keep in history
              table.remove(windowlist,i)
              break              
            end
          end
          
          if ca==caFree then
            local i=1
            while i<=#formhistory do
              if formhistory[i]==f then
                table.remove(formhistory,i)
              else
                i=i+1
              end
            end
          end
          
          while (#formhistory>0) and (formhistory[#formhistory]==f) do --make sure it's never the last one
            formhistory[#formhistory]=nil
          end
          
          if CurrentForm==f then
            print("this was the focused form. Finding previous form")
            if #formhistory then
              printf("Previous form was %s", tostringx(formhistory[#formhistory]))
              showForm(formhistory[#formhistory])
            else
              printf("Unknown previous form. Going for MainForm")            
              showForm(MainForm)
            end
          end
        end)
        
        showForm(f) --show it again
      else
        print("known form")
      end

    end
    
    
  end)

  registerActiveControlChangeNotification(function(c)
    local f
    if c then
      while c.Parent do
        c=c.Parent
      end
      f=c
    end

    if f==nil then
      f=getActiveForm()
    end

    if f and f.ControllerPanel then
      f.ControllerPanel.Caption=getControllerPanelText(f)
    end
  end)

end

if getControllerState() then
  if MainForm.miControllerModeBase then
    MainForm.miControllerModeBase.destroy()
  end

  if MainForm.tControllerModeLauncherWatcher then
    MainForm.tControllerModeLauncherWatcher.destroy()
  end



  local miControllerModeBase=createMenuItem(MainForm)
  miControllerModeBase.Name='miControllerModeBase'
  miControllerModeBase.Caption='Controller Mode (Press A+B)'
  MainForm.Menu.Items.insert(MainForm.Menu.Items.Count-1, miControllerModeBase)

  local miControllerModeStart=createMenuItem(MainForm)
  miControllerModeStart.Name='miControllerModeStart'
  miControllerModeStart.Caption='Start Controller mode (A+B)'
  miControllerModeStart.OnClick=function()
    startControllerMode()
  end

  miControllerModeBase.add(miControllerModeStart)

  tControllerModeLauncherWatcher=createTimer(MainForm)
  tControllerModeLauncherWatcher.Name='tControllerModeLauncherWatcher'

  tControllerModeLauncherWatcher.Interval=100
  tControllerModeLauncherWatcher.OnTimer=function(t)
    if getForegroundWindow()==MainForm.Handle then
      local s=getControllerState()
      if s==nil then
        print("error:no controllerstate result")
        t.Enabled=false
        t.destroy()
        return
      end

      if s.GAMEPAD_A and s.GAMEPAD_B then
        print("launching controllermode")
        t.Enabled=false
        t.destroy()

        miControllerModeBase.destroy()

        startControllerMode()
      end
    end
  end
end


--[[


--]]