用 AutoHotkey 做一个文件查找工具之四：根据输入实时更新过滤文件列表

http://www.cnblogs.com/yonken/archive/2010/05/10/Smart_Open_Files_Update_File_List_On_Pattern_Changed.html

又是在上一篇中，已经能把文件都给列出来并显示了，但输入关键字还是没有任何反应，我想应该可以着手实行这部分了。

2010-5-22更新：刷新问题似乎只要用AutoHotkey_L 就不再出现了（12/15/2010 更新：后来发现其实是Common Control，即 Comctl32.dll的问题，只要是6.0以上就不会有刷新问题）。

 

 

 

  ;
  ; AutoHotkey Version: 1.0.48.05
  ; Language:       English
  ; Platform:       Win9x/NT
  ; Author:         Yonken
  ;
  ; Script Function:
  ;   To enumerate the files of specified folder(s), filter out specified type(s), list and
  ; update the result in real time as the search pattern changed.
  ;   Use at your own risk.
  ;
  
  #NoEnv  ; Recommended for performance and compatibility with future AutoHotkey releases.
  SendMode Input  ; Recommended for new scripts due to its superior speed and reliability.
  SetWorkingDir %A_ScriptDir%  ; Ensures a consistent starting directory.
  SetBatchLines, -1   ; Never sleep
  
  WM_NOTIFY               := 0x004E
  LVN_FIRST               := -100
  
  LVN_GETDISPINFOA        := (LVN_FIRST-50)   ; For ANSI version
  LVN_GETDISPINFOW        := (LVN_FIRST-77)   ; For unicode version
  LVN_GETDISPINFO         := LVN_GETDISPINFOW
  
  LVM_FIRST               := 0x1000
  LVM_SETITEMCOUNT        := (LVM_FIRST + 47)
  LVM_REDRAWITEMS         := (LVM_FIRST + 21)
  
  LVS_OWNERDATA           := 0x1000
  
  LPSTR_TEXTCALLBACKA     := -1
  
  sizeofNMHDR             := 12
  sizeofLVITEM            := 40
  
  LVSICF_NOINVALIDATEALL  := 0x00000001
  
  LVIF_TEXT               := 0x0001
  LVIF_IMAGE              := 0x0002
  LVIF_STATE              := 0x0008
  
  CP_ACP                  := 0           ; default to ANSI code page
  
  pRtlFillMemory          := GetProcAddressInModule("RtlFillMemory")
  pWideCharToMultiByte    := GetProcAddressInModule("WideCharToMultiByte")
  pMultiByteToWideChar    := GetProcAddressInModule("MultiByteToWideChar")
  pStrCpy                 := GetProcAddressInModule("strcpy", "msvcrt")
  
  g_strAppName            := "Smart Open File"
  g_strVersion            := "2010.5.10"
  g_strTitle              := g_strAppName A_Space g_strVersion
  g_hMainWnd              := 0
  
  g_nFilesCount           := 0
  g_nMatchCount           := 0
  g_MatchIndices          = -1
  
  
  ; Allow the user to maximize or drag-resize the window:
  Gui +Resize
  
  ; Create some buttons:
  Gui, Add, Button, Default vBtnLoadFolder gButtonLoadFolder, &Load a folder
  
  ; Create the ListView with two columns, Name and Size:
  Gui, Add, ListView, Grid xm r20 w700 vMyListView Hwndg_hMyListView +%LVS_OWNERDATA%, Name|In Folder|Modified|Size (KB)|Type
  
  ; Create an ImageList so that the ListView can display some icons:
  ImageListID1 := IL_Create(10)
  ImageListID2 := IL_Create(10, 10, true)  ; A list of large icons to go with the small ones.
  
  ; Attach the ImageLists to the ListView so that it can later display the icons:
  LV_SetImageList(ImageListID1)
  LV_SetImageList(ImageListID2)
  
  ListView_SetItemCount(g_hMyListView, g_nMatchCount)
  
  Gui, Add, Edit, vEditSearchString gOnChangeSearchString,
  
  Gui, Show, , %g_strTitle% [0 of 0]
  
  OnMessage(WM_NOTIFY, "OnNotify")
  
  Gui, +LastFound
  WinSet, ReDraw      ; Invalidate the list-view
  
  g_hMainWnd := WinExist()
  
  return
  
  GuiEscape:
      GuiClose:  ; Indicate that the script should exit automatically when the window is closed.
      ExitApp
  Return
  
  GuiSize:  ; Expand or shrink the ListView in response to the user's resizing of the window.
      if A_EventInfo = 1  ; The window has been minimized.  No action needed.
          return
      ; Otherwise, the window has been resized or maximized. Resize the ListView to match.
      GuiControl, Move, MyListView, % "W" . (A_GuiWidth - 20) . " H" . (A_GuiHeight - 70)
      GuiControl, Move, EditSearchString, % "W" . (A_GuiWidth - 20) . " Y" . (A_GuiHeight - 30)
  return
  
  ButtonLoadFolder:
      Gui +OwnDialogs  ; Forces user to dismiss the following dialog before using main window.
      FileSelectFolder, Folder,, 3, Select a folder to read:
      if not Folder  ; The user canceled the dialog.
          return
          
      g_nMatchCount := 0
      WinSetTitle, , , %g_strTitle% [%g_nMatchCount% of %g_nFilesCount%]
  
      GuiControl, Disable, BtnLoadFolder
      SetTimer, FileLoadProgressTimer, 100
      
      g_arrResult := ; Free the memory
      g_nFilesCount := LoadFolder(Folder, "g_arrResult")
      
      Gosub, OnChangeSearchString
      
      Gui +LastFound
      GuiControl, Enable, BtnLoadFolder
      SetTimer, FileLoadProgressTimer, Off
  return
  
  OnChangeSearchString:
      GuiControl, -Redraw, MyListView  ; Improve performance by disabling redrawing during load.
      nMatchCount := g_nFilesCount
      bNeedSetFocus := false
      If (g_nFilesCount > 0)
-     {
          GuiControlGet, strSearchPattern, , EditSearchString
          
          strSearchPattern := RegExReplace(strSearchPattern, "S)[\s]+", "|")  ; Replace multiple whitespaces with a single character
          if(strSearchPattern = "" || strSearchPattern = "|")
-         {
              ; Nothing is entered, select the first one
              g_MatchIndices = -1
              bNeedSetFocus := true
          }
          Else
-         {
              If ( SubStr(strSearchPattern, StrLen(strSearchPattern)) == "|")
                  strSearchPattern := SubStr(strSearchPattern, 1, StrLen(strSearchPattern)-1)
              StringSplit, arrSearchPatterns, strSearchPattern, |
              
              g_MatchIndices =
              VarSetCapacity(g_MatchIndices, g_nFilesCount * 4)
              
              nMatchCount := 0
  
              Loop, %g_nFilesCount%
-             {
                  strName := g_arrResult%A_Index%_NameAnsi
                  strFolder := g_arrResult%A_Index%_FolderAnsi
                  bMatch := true
                  Loop %arrSearchPatterns0%
-                 {
                      if ( !IsMatch(strName, strFolder, arrSearchPatterns%A_Index%) )
-                     {
                          bMatch := false
                          Break
                      }
                  }
                  if (bMatch)
-                 {
                      NumPut(A_Index, g_MatchIndices, nMatchCount * 4)
                      ++nMatchCount
                  }
              }
          }
      }
      Else
-     {
          g_MatchIndices = -1
      }
      g_nMatchCount := nMatchCount
      ListView_SetItemCount(g_hMyListView, g_nMatchCount)
      ListView_RedrawItems(g_hMyListView, 0, -1)
      ;ToolTip, Done searching %strSearchPattern%
      
      LV_ModifyCol()  ; Auto-size each column to fit its contents.
      GuiControl, +Redraw, MyListView  ; Re-enable redrawing (it was disabled above).
      
      If (bNeedSetFocus)
-     {
          GuiControl, Focus, SysListView321, 
          Send, {Home}
          GuiControl, Focus, EditSearchString, 
      }
  
      WinSetTitle, , , %g_strTitle% [%g_nMatchCount% of %g_nFilesCount%]
  Return
  
  IsMatch(ByRef strFileName, ByRef strFolder, strSubPattern)
- {
      chChar1         := SubStr(strSubPattern, 1, 1)
      chChar2         := SubStr(strSubPattern, 2, 1)
      bIsExclude      := chChar1 == "-" || chChar2 == "-"
      bIsInFolder     := chChar1 == "\" || chChar2 == "\"
      
      strSubPattern := RegExReplace(strSubPattern, "S)^[-\\]+")
      If (strSubPattern = "")
          return true
-     /*
      StringReplace, strSubPattern, strSubPattern, -
      StringReplace, strSubPattern, strSubPattern, \
      */
      strSearchText   := bIsInFolder ? strFolder : strFileName
      bMatch          := InStr(strSearchText, strSubPattern)
      if (bIsExclude)
          bMatch := !bMatch
  
      return bMatch
  }
  
  FileLoadProgressTimer:
      Gui +LastFound
      WinSetTitle, , , %g_strTitle% [%g_nMatchCount% of %g_nFilesCount%]
  Return
  
- /*
  typedef struct tagNMHDR
  {
      HWND  hwndFrom;
      UINT  idFrom;
      UINT  code;         // NM_ code
  }   NMHDR;
  
   NMHDR *pnm
  */
  OnNotify(idCtrl, pnmh)
- {
      global g_hMyListView, LVN_GETDISPINFO
      hwndFrom := DecodeInteger( "uint4", pnmh, 0 )
      if ( hwndFrom == g_hMyListView )
-     {
          idFrom := DecodeInteger( "uint4", pnmh, 4)
          code := DecodeInteger( "uint4", pnmh, 8)
          nCode := code > 0x7FFFFFFF ? -(~code) - 1 : code
          if ( nCode == LVN_GETDISPINFO )
-         {
              OnGetDispInfo(pnmh)
          }
      }
  }
  
- /*
  NMLVDISPINFO* pnmv
  
  typedef struct tagNMLVDISPINFO {
      NMHDR hdr;
      LVITEM item;
  } NMLVDISPINFO;
  
  typedef struct _LVITEM { 
      UINT mask;     0
      int iItem;     4
      int iSubItem;  8
      UINT state;    12
      UINT stateMask; 16
      LPTSTR pszText; 20
      int cchTextMax; 24
      int iImage;     28
      LPARAM lParam;
  #if (_WIN32_IE >= 0x0300)
      int iIndent;
  #endif
  #if (_WIN32_WINNT >= 0x560)
      int iGroupId;
      UINT cColumns; // tile view columns
      PUINT puColumns;
  #endif
  #if (_WIN32_WINNT >= 0x0600)
      int* piColFmt;
      int iGroup;
  #endif
  } LVITEM, *LPLVITEM; 
  
  */
  OnGetDispInfo(pnmv)
- {
      global
      iItemOffset     := sizeofNMHDR + 4
      iItem           := DecodeInteger( "uint4", pnmv, iItemOffset)
      
      if (iItem < 0 || iItem > g_nMatchCount)
          return  ; requesting invalid item
      
      maskOffset      := sizeofNMHDR + 0
      mask            := DecodeInteger( "uint4", pnmv, maskOffset)
      
      if (mask & LVIF_TEXT)
-     {
          iSubItemOffset  := sizeofNMHDR + 8
          iSubItem        := DecodeInteger( "uint4", pnmv, iSubItemOffset)
          
          pszTextOffset   := sizeofNMHDR + 20
          pszText         := DecodeInteger( "uint4", pnmv, pszTextOffset)
          
          If (g_MatchIndices = -1)
              nIndex := iItem + 1
          else
              nIndex := NumGet(g_MatchIndices, iItem * 4)
              
          ;ToolTip, Row %iItem% Index in list %nIndex%
          
          pstrText := 0
          if (0 == iSubItem)
              pstrText := &g_arrResult%nIndex%_Name
          Else if (1 == iSubItem)
              pstrText := &g_arrResult%nIndex%_Folder
          Else if (2 == iSubItem)
              pstrText := &g_arrResult%nIndex%_Modified
          Else if (3 == iSubItem)
              pstrText := &g_arrResult%nIndex%_SizeKB
          Else if (4 == iSubItem)
              pstrText := &g_arrResult%nIndex%_Ext
          EncodeInteger( pstrText, 4, pnmv, pszTextOffset )
      }
      if (mask & LVIF_STATE)
-     {
          stateOffset := sizeofNMHDR + 12
          EncodeInteger( 0, 4, pnmv, stateOffset )
      }
      if (mask & LVIF_IMAGE)
-     {
          iImageOffset := sizeofNMHDR + 28
          EncodeInteger( -1, 4, pnmv, iImageOffset )
      }
  }
  
  LoadFolder(ByRef strFolder, arrResultName, ByRef strExtInclude = "", ByRef strExtExclude = "", bRecursive = true)
- {
      global  ; This is important for creating/accessing array
      nTotalFiles := 0
      nIndex := g_nFilesCount+1
      
      ; Check if the last character of the folder name is a backslash, which happens for root
      ; directories such as C:\. If it is, remove it to prevent a double-backslash later on.
      StringRight, LastChar, strFolder, 1
      if LastChar = \
          StringTrimRight, strFolder, strFolder, 1  ; Remove the trailing backslash.
  
      ; Ensure the variable has enough capacity to hold the longest file path. This is done
      ; because ExtractAssociatedIconA() needs to be able to store a new filename in it.
      VarSetCapacity(Filename, 260)
      sfi_size = 352
      VarSetCapacity(sfi, sfi_size)
  
      ; Gather a list of file names from the folder
      Loop %strFolder%\*.*, 0, %bRecursive%
-     {
          ;MsgBox %A_LoopFileName%
          
          ; filter out
          If (strExtExclude != "")
-         {
              if A_LoopFileExt in %strExtExclude%
-             {
                  Continue
              }
          }
          If (strExtInclude != "")
-         {
              if A_LoopFileExt not in %strExtInclude%
-             {
                  Continue
              }
          }
          
          ; Name|In Folder|Modified|Size (KB)|Type
          FormatTime, FileTimeModified, %A_LoopFileTimeModified% LSys R D1    ;, MM/dd/yyyy HH:mm
          
          ToWideChar(A_LoopFileName, %arrResultName%%nIndex%_Name)
          ToWideChar(A_LoopFileExt, %arrResultName%%nIndex%_Ext)
          ToWideChar(A_LoopFileDir, %arrResultName%%nIndex%_Folder)
          ToWideChar(FileTimeModified, %arrResultName%%nIndex%_Modified)
          ToWideChar(A_LoopFileSizeKB, %arrResultName%%nIndex%_SizeKB)
          
          ; We need this for pattern matching
          %arrResultName%%nIndex%_NameAnsi := A_LoopFileName
          %arrResultName%%nIndex%_FolderAnsi := A_LoopFileDir
          
          ++nIndex
          ++nTotalFiles
          ++g_nFilesCount
      }
      return nTotalFiles
  }
  
  #IfWinActive ahk_class AutoHotkeyGUI
  ~Up::
  ~+Up::
      IfWinNotActive, ahk_id %g_hMainWnd%,
-     {
          Send, {Up}
          Return
      }
      
      GuiControl, Focus, SysListView321, 
      ;ControlSend, SysListView321, {Up}, %g_hMainWnd%
  Return
  
  ~Down::
  ~+Down::
      IfWinNotActive, ahk_id %g_hMainWnd%,
-     {
          Send, {Down}
          Return
      }
      
      GuiControl, Focus, SysListView321, 
      ;ControlSend, SysListView321, {Down}, %g_hMainWnd%
  Return
  #IfWinActive
  
  ;;;;;;;;;;;;;;;;;;; Helper Functions ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
  
  ListView_RedrawItems(hwndLV, iFirst, iLast)
- {
      global LVM_REDRAWITEMS
      SendMessage, LVM_REDRAWITEMS, iFirst, iLast, , ahk_id %hwndLV%
      return %ErrorLevel%
  }
  
  ListView_SetItemCount(hwndLV, cItems)
- {
      global LVM_SETITEMCOUNT, LVSICF_NOINVALIDATEALL
      SendMessage, LVM_SETITEMCOUNT, cItems, LVSICF_NOINVALIDATEALL, , ahk_id %hwndLV%
      return %ErrorLevel%
  }
  
  TransformWideCharToMultiByte(pWideChar, nWideCharNumber)
- {
      Global CP_ACP
      nRequiredSize := WideCharToMultiByte(CP_ACP, 0, pWideChar, nWideCharNumber, 0, 0, 0, 0)
      if(nRequiredSize > 0)
-     {
          VarSetCapacity( pMultiByteBuffer, nRequiredSize, 0 )
          nBytesWritten := WideCharToMultiByte(CP_ACP, 0, pWideChar, nWideCharNumber, &pMultiByteBuffer, nRequiredSize, 0, 0)
          return pMultiByteBuffer
      }
      return 0
  }
  
  TransformMultiByteToWideChar(pMultiByte, ByRef pWideCharBuffer)
- {
      Global CP_ACP
      nRequiredSize := MultiByteToWideChar(CP_ACP, 0, pMultiByte, -1, 0, 0)
      if(nRequiredSize > 0)
-     {
          VarSetCapacity( pWideCharBuffer, nRequiredSize<<1, 0 )
          nBytesWritten := MultiByteToWideChar(CP_ACP, 0, pMultiByte, -1, &pWideCharBuffer, nRequiredSize)
          return nBytesWritten
      }
      return -1
  }
  
  ToWideChar(strAnsiText, ByRef strUnicodeText)
- {
      nStrLen := StrLen(strAnsiText)
      VarSetCapacity(strUnicodeText, (nStrLen+1)*2, 0)
      TransformMultiByteToWideChar(&strAnsiText, strUnicodeText)
      Return strUnicodeText
  }
  
  WideCharToMultiByte(CodePage, dwFlags, lpWideCharStr, cchWideChar, lpMultiByteStr, cbMultiByte, lpDefaultChar, lpUsedDefaultChar)
- {
      global pWideCharToMultiByte
      return DllCall(pWideCharToMultiByte, "UInt", CodePage, "UInt", dwFlags, "UInt", lpWideCharStr, "Int", cchWideChar, "UInt",lpMultiByteStr, "Int", cbMultiByte, "UInt", lpDefaultChar, "UInt", lpUsedDefaultChar)
  }
  
  MultiByteToWideChar(CodePage, dwFlags, lpMultiByteStr, cbMultiByte, lpWideCharStr, cchWideChar)
- {
      global pMultiByteToWideChar
      return DllCall(pMultiByteToWideChar, "UInt", CodePage, "UInt", dwFlags, "UInt", lpMultiByteStr, "Int", cbMultiByte, "UInt",lpWideCharStr, "Int", cchWideChar)
  }
  
  DecodeInteger( p_type, p_address, p_offset, p_hex=true )
- {
    old_FormatInteger := A_FormatInteger
    ifEqual, p_hex, 1, SetFormat, Integer, hex
    else, SetFormat, Integer, dec
    StringRight, size, p_type, 1
    loop, %size%
        value += *( ( p_address+p_offset )+( A_Index-1 ) ) << ( 8*( A_Index-1 ) )
    if ( size <= 4 and InStr( p_type, "u" ) != 1 and *( p_address+p_offset+( size-1 ) ) & 0x80 )
        value := -( ( ~value+1 ) & ( ( 2**( 8*size ) )-1 ) )
    SetFormat, Integer, %old_FormatInteger%
    return, value
  }
  
  EncodeInteger( p_value, p_size, p_address, p_offset )
- {
      global pRtlFillMemory
      loop, %p_size%
          DllCall( pRtlFillMemory, "uint", p_address+p_offset+A_Index-1, "uint", 1, "uchar", p_value >> ( 8*( A_Index-1 ) ) )
  }
  
  GetProcAddressInModule(strFuncName, strModuleName = "kernel32")
- {
      return DllCall("GetProcAddress", uint, DllCall("GetModuleHandle", str, strModuleName), str, strFuncName)
  }

分类:  Script