• Win32非递归遍历和搜索文件以及目录算法


    转载请注明来源:http://www.cnblogs.com/xuesongshu

    要点:

    1、搜索的顶层目录在进入循环之前进栈

    2、栈元素存储字符串指针,出栈时释放资源

    3、每次循环开始,栈顶元素出栈

    4、在遍历一个文件夹时,遇到子文件夹则进栈

    5、外层循环以堆栈是否为空为标志,内层循环以FindNextFile返回值为标志

    6、本搜索算法适用于按关键字搜索,当文件夹名称符合搜索条件时文件夹作为搜索结果通知调用者,不会入栈。

    7、回调函数地址使用long型传递

    8、本程序功能是批量增加、修改或者删除指定文件夹下所有文件前缀,附带遍历功能。在参照流程图编写实现代码时与流程图所画逻辑会稍有出入。

    流程图如下:

    需要引入的头文件:

    #include <stack>
    #include <iostream>
    #include <fstream>
    #include <list>
    #include <set>
    

      

    回调函数定义:

    typedef HRESULT (__stdcall *XCallbackMethodType)(HWND,BSTR,int);
    

      

    实现代码如下:

    STDMETHODIMP CFileTool::SearchFiles(BSTR topDirectoryName, BSTR key, HWND mainWnd,long callbackAddress)
    {
    	HINSTANCE dllInst=(HINSTANCE)GetModuleHandle(L"xssfj.dll");
    
    	WCHAR finishedMsg[256]={0};
    	LoadString(dllInst,IDS_FILE_DELETESPECIFY_SEARCHSUCCESS,finishedMsg,256);
    	XCallbackMethodType callbackMsg=(XCallbackMethodType)callbackAddress;
    	DWORD attribute=GetFileAttributesW(topDirectoryName);
    	if(!(attribute&FILE_ATTRIBUTE_DIRECTORY))
    		return E_INVALIDARG;
    	if(!callbackMsg)
    		return E_INVALIDARG;
    	if(StrStrI(topDirectoryName,key))
    	{
    		callbackMsg(mainWnd,topDirectoryName,2);
    		callbackMsg(mainWnd,finishedMsg,0);
    		return S_OK;
    	}
    	std::stack<WCHAR*> directoryNameStack;
    	WCHAR* tmpStackData=(WCHAR*)calloc(MAX_PATH,sizeof(WCHAR));
    	lstrcpyW(tmpStackData,topDirectoryName);
    	directoryNameStack.push(tmpStackData);
    	while(!directoryNameStack.empty())
    	{
    		WCHAR* currentDirName=directoryNameStack.top();
    		directoryNameStack.pop();
    		callbackMsg(mainWnd,currentDirName,1);
    		WCHAR tmpParentPath[MAX_PATH]={0};
    		lstrcpy(tmpParentPath,currentDirName);
    		if(currentDirName[lstrlen(currentDirName)-1]!='\')
    		{
    			lstrcat(tmpParentPath,L"\");
    			lstrcat(currentDirName,L"\*.*");
    		}
    		else
    			lstrcat(currentDirName,L"*.*");
    		WIN32_FIND_DATA meiju={0};
    
    		HANDLE tmpHandle=FindFirstFile(currentDirName,&meiju);
    		if(tmpHandle!=INVALID_HANDLE_VALUE&&tmpHandle)
    		{
    			if(lstrcmpW(meiju.cFileName,L".")!=0&&lstrcmpW(meiju.cFileName,L"..")!=0)
    			{
    				if(StrStrI(meiju.cFileName,key))
    				{
    					WCHAR tmpFsObj[MAX_PATH]={0};
    					lstrcpyW(tmpFsObj,tmpParentPath);
    					lstrcat(tmpFsObj,meiju.cFileName);
    					callbackMsg(mainWnd,tmpFsObj,2);
    				}
    			}
    			do
    			{
    				DWORD tmpFileResult=FindNextFile(tmpHandle,&meiju);
    				if(tmpFileResult==ERROR_NO_MORE_FILES||tmpFileResult==0)
    					break;
    				if(lstrcmpW(meiju.cFileName,L".")==0||lstrcmpW(meiju.cFileName,L"..")==0)
    					continue;
    				if(StrStrI(meiju.cFileName,key))
    				{
    					WCHAR tmpFsObj[MAX_PATH]={0};
    					lstrcpyW(tmpFsObj,tmpParentPath);
    					lstrcat(tmpFsObj,meiju.cFileName);
    					callbackMsg(mainWnd,tmpFsObj,2);
    				}
    				else
    				{
    					WCHAR tmpFsobjPath[MAX_PATH]={0};
    					lstrcpy(tmpFsobjPath,tmpParentPath);
    					lstrcat(tmpFsobjPath,meiju.cFileName);
    					callbackMsg(mainWnd,tmpFsobjPath,1);
    					DWORD tmpFileAttribute=GetFileAttributes(tmpFsobjPath);
    					if(tmpFileAttribute&FILE_ATTRIBUTE_DIRECTORY)
    					{
    						WCHAR* tmpData=(WCHAR*)calloc(MAX_PATH,sizeof(WCHAR));
    						lstrcpy(tmpData,tmpFsobjPath);
    						directoryNameStack.push(tmpData);
    					}
    				}
    			}
    			while(true);
    			FindClose(tmpHandle);
    		}
    		free(currentDirName);
    	}
    	callbackMsg(mainWnd,finishedMsg,0);
    	return S_OK;
    }
    

      

  • 相关阅读:
    我的第一次JAVA实训——校园公用房管理系统
    打我吧,不写了
    我有我的方向
    JAVA 算法练习(二)
    JAVA 算法练习(一)
    php设置时区和strtotime转化为时间戳函数
    dos中文乱码怎么办?
    织梦怎么调用栏目SEO标题
    for in循环介绍以及陷阱
    浅谈js for循环输出i为同一值的问题(闭包解决)
  • 原文地址:https://www.cnblogs.com/xuesongshu/p/3893728.html
Copyright © 2020-2023  润新知