Skip to content

Instantly share code, notes, and snippets.

@Earlopain
Last active December 2, 2019 15:09
Show Gist options
  • Save Earlopain/075d336badc0a94b9240cf4778a791ca to your computer and use it in GitHub Desktop.
Save Earlopain/075d336badc0a94b9240cf4778a791ca to your computer and use it in GitHub Desktop.
Modified version of https://github.com/pmb6tz/windows-desktop-switcher to fit my needs
global Wallpapers := []
Loop, C:\tools\ahk\wallpaper\*.*
{
Wallpapers.Push(A_LoopFileFullPath)
}
#SingleInstance Force ; The script will Reload if launched while already running
#NoEnv ; Recommended for performance and compatibility with future AutoHotkey releases
#KeyHistory 0 ; Ensures user privacy when debugging is not needed
SetWorkingDir %A_ScriptDir% ; Ensures a consistent starting directory
SendMode Input ; Recommended for new scripts due to its superior speed and reliability
; Globals
DesktopCount := 2 ; Windows starts with 2 desktops at boot
CurrentDesktop := 1 ; Desktop count is 1-indexed (Microsoft numbers them this way)
LastOpenedDesktop := 1
; DLL
hVirtualDesktopAccessor := DllCall("LoadLibrary", "Str", A_ScriptDir . "\VirtualDesktopAccessor.dll", "Ptr")
global IsWindowOnDesktopNumberProc := DllCall("GetProcAddress", Ptr, hVirtualDesktopAccessor, AStr, "IsWindowOnDesktopNumber", "Ptr")
global MoveWindowToDesktopNumberProc := DllCall("GetProcAddress", Ptr, hVirtualDesktopAccessor, AStr, "MoveWindowToDesktopNumber", "Ptr")
; Main
SetKeyDelay, 75
mapDesktopsFromRegistry()
OutputDebug, [loading] desktops: %DesktopCount% current: %CurrentDesktop%
#Include %A_ScriptDir%\hotkeys.ahk
return
;
; This function examines the registry to build an accurate list of the current virtual desktops and which one we're currently on.
; Current desktop UUID appears to be in HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\CurrentVersion\Explorer\SessionInfo\1\VirtualDesktops
; List of desktops appears to be in HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\CurrentVersion\Explorer\VirtualDesktops
;
mapDesktopsFromRegistry()
{
global CurrentDesktop, DesktopCount
; Get the current desktop UUID. Length should be 32 always, but there's no guarantee this couldn't change in a later Windows release so we check.
IdLength := 32
SessionId := getSessionId()
if (SessionId) {
RegRead, CurrentDesktopId, HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\CurrentVersion\Explorer\SessionInfo\%SessionId%\VirtualDesktops, CurrentVirtualDesktop
if (CurrentDesktopId) {
IdLength := StrLen(CurrentDesktopId)
}
}
; Get a list of the UUIDs for all virtual desktops on the system
RegRead, DesktopList, HKEY_CURRENT_USER, SOFTWARE\Microsoft\Windows\CurrentVersion\Explorer\VirtualDesktops, VirtualDesktopIDs
if (DesktopList) {
DesktopListLength := StrLen(DesktopList)
; Figure out how many virtual desktops there are
DesktopCount := floor(DesktopListLength / IdLength)
}
else {
DesktopCount := 1
}
; Parse the REG_DATA string that stores the array of UUID's for virtual desktops in the registry.
i := 0
while (CurrentDesktopId and i < DesktopCount) {
StartPos := (i * IdLength) + 1
DesktopIter := SubStr(DesktopList, StartPos, IdLength)
OutputDebug, The iterator is pointing at %DesktopIter% and count is %i%.
; Break out if we find a match in the list. If we didn't find anything, keep the
; old guess and pray we're still correct :-D.
if (DesktopIter = CurrentDesktopId) {
CurrentDesktop := i + 1
OutputDebug, Current desktop number is %CurrentDesktop% with an ID of %DesktopIter%.
break
}
i++
}
}
;
; This functions finds out ID of current session.
;
getSessionId()
{
ProcessId := DllCall("GetCurrentProcessId", "UInt")
if ErrorLevel {
OutputDebug, Error getting current process id: %ErrorLevel%
return
}
OutputDebug, Current Process Id: %ProcessId%
DllCall("ProcessIdToSessionId", "UInt", ProcessId, "UInt*", SessionId)
if ErrorLevel {
OutputDebug, Error getting session id: %ErrorLevel%
return
}
OutputDebug, Current Session Id: %SessionId%
return SessionId
}
_switchDesktopToTarget(targetDesktop)
{
; Globals variables should have been updated via updateGlobalVariables() prior to entering this function
global CurrentDesktop, DesktopCount, LastOpenedDesktop
; Don't attempt to switch to an invalid desktop
if (targetDesktop > DesktopCount || targetDesktop < 1 || targetDesktop == CurrentDesktop) {
OutputDebug, [invalid] target: %targetDesktop% current: %CurrentDesktop%
return
}
LastOpenedDesktop := CurrentDesktop
; Fixes the issue of active windows in intermediate desktops capturing the switch shortcut and therefore delaying or stopping the switching sequence. This also fixes the flashing window button after switching in the taskbar. More info: https://github.com/pmb6tz/windows-desktop-switcher/pull/19
WinActivate, ahk_class Shell_TrayWnd
; Go right until we reach the desktop we want
while(CurrentDesktop < targetDesktop) {
Send {LWin down}{LCtrl down}{Right down}{LWin up}{LCtrl up}{Right up}
CurrentDesktop++
OutputDebug, [right] target: %targetDesktop% current: %CurrentDesktop%
}
; Go left until we reach the desktop we want
while(CurrentDesktop > targetDesktop) {
Send {LWin down}{LCtrl down}{Left down}{Lwin up}{LCtrl up}{Left up}
CurrentDesktop--
OutputDebug, [left] target: %targetDesktop% current: %CurrentDesktop%
}
Random, var, 1, Wallpapers.Count()
DllCall("SystemParametersInfo", UInt, 0x14, UInt, 0, Str, Wallpapers[var], UInt, 3)
; Makes the WinActivate fix less intrusive
Sleep, 50
focusTheForemostWindow(targetDesktop)
}
updateGlobalVariables()
{
; Re-generate the list of desktops and where we fit in that. We do this because
; the user may have switched desktops via some other means than the script.
mapDesktopsFromRegistry()
}
switchDesktopByNumber(targetDesktop)
{
global CurrentDesktop, DesktopCount
updateGlobalVariables()
_switchDesktopToTarget(targetDesktop)
}
switchDesktopToLastOpened()
{
global CurrentDesktop, DesktopCount, LastOpenedDesktop
updateGlobalVariables()
_switchDesktopToTarget(LastOpenedDesktop)
}
switchDesktopToRight()
{
global CurrentDesktop, DesktopCount
updateGlobalVariables()
if(CurrentDesktop == DesktopCount)
{
return
}
updateGlobalVariables()
_switchDesktopToTarget(CurrentDesktop == DesktopCount ? 1 : CurrentDesktop + 1)
}
switchDesktopToLeft()
{
global CurrentDesktop, DesktopCount
updateGlobalVariables()
if(CurrentDesktop == 1)
{
return
}
updateGlobalVariables()
_switchDesktopToTarget(CurrentDesktop == 1 ? DesktopCount : CurrentDesktop - 1)
}
focusTheForemostWindow(targetDesktop)
{
foremostWindowId := getForemostWindowIdOnDesktop(targetDesktop)
WinActivate, ahk_id %foremostWindowId%
}
getForemostWindowIdOnDesktop(n)
{
n := n - 1 ; Desktops start at 0, while in script it's 1
; winIDList contains a list of windows IDs ordered from the top to the bottom for each desktop.
WinGet winIDList, list
Loop % winIDList {
windowID := % winIDList%A_Index%
windowIsOnDesktop := DllCall(IsWindowOnDesktopNumberProc, UInt, windowID, UInt, n)
; Select the first (and foremost) window which is in the specified desktop.
if (windowIsOnDesktop == 1) {
return windowID
}
}
}
MoveCurrentWindowToDesktop(desktopNumber) {
updateGlobalVariables()
WinGet, activeHwnd, ID, A
DllCall(MoveWindowToDesktopNumberProc, UInt, activeHwnd, UInt, desktopNumber - 1)
switchDesktopByNumber(desktopNumber)
}
;
; This function creates a new virtual desktop and switches to it
;
createVirtualDesktop()
{
global CurrentDesktop, DesktopCount
Send, #^d
DesktopCount++
CurrentDesktop := DesktopCount
OutputDebug, [create] desktops: %DesktopCount% current: %CurrentDesktop%
}
;
; This function deletes the current virtual desktop
;
deleteVirtualDesktop()
{
global CurrentDesktop, DesktopCount, LastOpenedDesktop
Send, #^{F4}
if (LastOpenedDesktop >= CurrentDesktop) {
LastOpenedDesktop--
}
DesktopCount--
CurrentDesktop--
OutputDebug, [delete] desktops: %DesktopCount% current: %CurrentDesktop%
}
MoveCurrentWindowRight(){
global CurrentDesktop, DesktopCount
updateGlobalVariables()
if(CurrentDesktop == DesktopCount)
{
return
}
MoveCurrentWindowToDesktop(CurrentDesktop + 1)
}
MoveCurrentWindowLeft(){
global CurrentDesktop, DesktopCount
updateGlobalVariables()
if(CurrentDesktop == 1)
{
return
}
MoveCurrentWindowToDesktop(CurrentDesktop - 1)
}
^!Right::switchDesktopToRight()
^!Left::switchDesktopToLeft()
^!+Right::MoveCurrentWindowRight()
^!+Left::MoveCurrentWindowLeft()
^!l::DllCall("LockWorkStation")
^+!SPACE:: Winset, Alwaysontop, , A
Pause::Media_Play_Pause
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment