Coding Life

  博客园 :: 首页 :: 博问 :: 闪存 :: 新随笔 :: 联系 :: 订阅 订阅 :: 管理 ::

上一篇中,数组的使用正常,接下来就可以将我们需要的信息存储到一个数组中了。

这次为了方便测试,先在对话框上加个按钮来指定要搜索的文件夹。

测试效果还不错,速度非常快,但刷新的老问题依然没有解决(拖动一个column的时候闪烁就非常明显)。

 

 

  ;
  ; 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.
  ;
 
  #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.
 
  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)
 
  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_nMatchCount           := 0
  g_strAppName            := "Smart Open File"
  g_strVersion            := "2010.5.10"
  g_strTitle              := g_strAppName A_Space g_strVersion
 
 
  ; 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
  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_nMatchCount%]
 
      GuiControl, Disable, BtnLoadFolder
      SetTimer, FileLoadProgressTimer, 100
     
      g_arrResult := ; Free the memory
      g_nMatchCount := LoadFolder(Folder, "g_arrResult")
     
      ListView_SetItemCount(g_hMyListView, g_nMatchCount)
      WinSet, ReDraw      ; Invalidate the list-view
 
      GuiControl, Enable, BtnLoadFolder
      SetTimer, FileLoadProgressTimer, Off
     
      LV_ModifyCol()  ; Auto-size each column to fit its contents.
 
      Gui +LastFound
      WinSetTitle, , , %g_strTitle% [%g_nMatchCount% of %g_nMatchCount%]
  return
 
  OnChangeSearchString:
  Return
 
  FileLoadProgressTimer:
      Gui +LastFound
      WinSetTitle, , , %g_strTitle% [%g_nMatchCount% of %g_nMatchCount%]
  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)
         
          nIndex := iItem + 1
         
          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_nMatchCount+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)
         
          ++nIndex
          ++nTotalFiles
          ++g_nMatchCount
      }
      return nTotalFiles
  }
 
  ;;;;;;;;;;;;;;;;;;; Helper Functions ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 
  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)
  }

posted on 2010-05-10 13:24  yonken  阅读(1873)  评论(0编辑  收藏  举报