数学建模社区-数学中国
标题: 目录处理函数 [打印本页]
作者: 韩冰 时间: 2005-1-26 13:18
标题: 目录处理函数
目录处理函数
. u& h- h: D# F' m7 |- l
编程语言：C++ Builder 作者：王行舟
- g' h1 i; [) H* H 在编程时,经常有一些针对目录的操作,如打开目录对话框选择一个目录,直接创建多级目录,直接删除多级目录,判断某个目录是否存在等。本文就这些问题给出编程实现方法，并给出详细的程序代码，供各位编程爱好者参考。
& ?0 K- `9 q( c- [7 a/ |- s; l一、判断目录是否存在：
9 {9 U. c4 R( Y, N D- t C++ Builder中提供了检查文件是否存在的函数FileExists，但没有提供检查目录是否存在的函数，我们可以用Windows API函数FindFirstFile实现这个功能。程序实现如下：
/ x: `/ n \& T0 X0 N
设char *Dir为带判断的目录! P- k' J, r Y4 X1 O1 F! a
bool Exist; // 最后结果，表示目录是否存在' Y, T+ K- ^* k F
if(Dir[strlen(Dir)]=='\\')Dir[strlen(Dir)-1]='\0'; // 先删除最后的“\”% B( {! B& |2 Y
WIN32_FIND_DATA wfd; // 查找9 T5 ?- e% [, _9 O9 J# \: P6 O* M
HANDLE hFind=FindFirstFile(Dir,&wfd);
U3 o8 R3 N) V; ~# yif(hFind==INVALID_HANDLE_VALUE)Exist=false; // 没有找到配备，目录肯定不存在& P6 x: n. `8 o1 j/ l4 Q) p
else
, U; u3 x. f+ L8 j. J{
n( h* d& i2 h. i9 L. ?0 | if(wfd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) // 检查找到的结果是否目录
$ Z, F: U% X! d9 h5 Y: g Exist=true; // 是目录,目录存在! Q, f) {! O1 c/ [
else
1 K q( g8 M# B Exist=false; // 是目录,目录不存在; X% e- `- y8 z5 }" f$ Z
FindClose(hFind);" U, n" Z% m* q+ }: B- z
}
+ [1 J: h3 C+ n2 G1 G6 k# f* w
二、打开目录选择对话框选择一个目录：
[" X6 G+ D. Q+ s& B$ Q 大多专业软件在要求输入目录的编辑框旁都放了一个按钮，点击后打开一个目录窗口，很多编程爱好者也希望能掌握这个方法。实现这个功能要调用Windows API函数SHBrowseForFolder，完整声明为WINSHELLAPI LPITEMIDLIST WINAPI SHBrowseForFolder(LPBROWSEINFO lpbi)，返回一个ITEMIDLIST类型的指针，通过这个指针调用函数SHGetPathFromIDList可以确定所选择的目录的全名称。入参为BROWSEINFO结构的指针，这个结构较为复杂，成员如下所示：
. }$ B( ^( O, t c. W1 _+ s! _6 s
HWND hwndOwner; // 拥有对话框的窗口，可以设置为Application->Handle 7 c& T9 E, k4 d3 _( W7 s
LPCITEMIDLIST pidlRoot; // ITEMIDLIST类型的指针,表示在哪个路径下选择，一般可以设置为NULL# G/ J3 v) q+ r
LPSTR pszDisplayName; // 选择后，所选目录的名称(不包含父级目录)被拷贝到这个指针指向的位置 9 T. A+ s1 I7 ]0 C7 e4 B
LPCSTR lpszTitle; // 作为标题显示在对话框中目录树的上面，可以根据实际情况设置
) d8 d, ~+ d4 u) @) jUINT ulFlags; // 标志位，有点复杂，一般设置为BIF_RETURNONLYFSDIRS
4 l7 n; Z7 k3 w: p/ }* `( f# eBFFCALLBACK lpfn; // 回调函数，一般不用，设置为NULL
9 g; k3 h: \" S4 j! G! ]- d+ PLPARAM lParam; // 预定义的对话框传递给回调函数的值
3 g" |1 t. b* h4 L. Xint iImage; // 与所选目录相关联的图标在系统图标集合中的索引 0 J N' ~: R1 ?) p/ G# }; p7 ^
可以看出，使用函数SHBrowseForFolder还真麻烦，普通爱好者掌握它确实有一定的难度，现给出完整程序段如下：
. O& m) O8 F0 H1 q#include <shlobj.h> // 必须包含的头文件
! M; b- H: K' f/ e0 h# d: |char SelectedDir[MAX_PATH]; // 最终结果, \- s! G: @0 P( V6 u
BROWSEINFO bi; // 入参
6 U$ k( t, G( Qchar FolderName[MAX_PATH]; // 所选目录名称，例如选择C:\Windows\Font,则为Font r& o. G/ S9 k9 x9 j" {
LPITEMIDLIST ItemID; // 所选目录的系统标志指针
# e: ~0 P) M2 ^! q8 l$ O8 @8 I8 H
memset(SelectedDir, 0, MAX_PATH); // 初始化最终结果& P3 E _: N3 ]
memset(&bi, 0, sizeof(BROWSEINFO)); // 初始化入参所有数据
% p# ^% o" s6 k1 L9 kbi.hwndOwner = Application->Handle;
& f' v! V' n, X9 i- j$ P; m4 D1 m2 Ybi.pszDisplayName = FolderName;
4 z( U3 {% g, }- L5 _! d/ mbi.lpszTitle = "请选择目录"; // 改成自己希望的
8 _6 }7 K! _6 [9 M( d3 R2 D3 Tbi.ulFlags=BIF_RETURNONLYFSDIRS;8 Q# B5 w/ t& @9 G) a& v: j9 m8 n
ItemID = SHBrowseForFolder(&bi); // 调用函数，打开目录选择对话框
. ^4 P$ U4 X8 o: Z. j, h: |if(ItemID)/ l( M4 I0 U* ]9 s5 ]) W1 a8 l+ z
{
* p* N/ R: j/ J2 T! g( U9 x SHGetPathFromIDList(ItemID, SelectedDir); // 获取所选目录的全名
2 p- ]7 S5 B; t' z3 L+ K GlobalFree(ItemID); // 返回的ItemID占用了系统资源，不要忘了释放* f- D, Z7 C6 g( c* `' |: \
}
+ y! v8 o, k- K
三、直接建立多级目录：
2 J. P! ?, d4 j! [2 e
Windows API提供了建立目录的函数CreateDirectory，但是调用前要保证父目录必须存在，否则会失败。其实，有时越级建立多级目录很有用，因为在建立目录特别是建立多层目录时，层层加以判断会大大地增加程序的复杂程度。如何实现这个功能呢？本人用递归方法设计了一个可以直接建立多级目录的函数，现说明如下，供各位朋友参考。
9 F: [# J1 ?5 e" i2 \ _8 C
bool MakeDirectoryEx(const AnsiString &
) // 入参为打算创建的目录名,根据操作结果返回"true"或"false"2 W. n2 s! r$ s5 ^, H/ [: J! z. \
{( C( x7 l7 _5 o
if(P.IsEmpty())return false;4 G3 X. H2 N8 }( a( D
int len=P.Length();3 A2 h/ h. K9 A8 E& N, t" W( [# D2 O
char *Path=P.c_str();
4 L6 |; ]& P" r$ ^$ M" [% J6 Z if(Path[len-1]=='\\')8 f/ g+ ]: H* `& o
{
8 h+ k7 q4 E/ _5 J+ { len--;
% U3 L( q& w$ {6 [ Path[len]='\0';
) E, A- Y- `5 m/ D% U } // 删除末尾的"\"3 y" _& y Y4 o
AnsiString Dir=Path;
0 ]! z! {' h' ]4 P- O5 c5 H // 分开父目录和本身目录名称$ t0 A f" C$ L) v! I
AnsiString Parent;
, m! e: o) E5 K$ d- E for(int i=len-1;i>0;i--)) x: w* ~$ @0 @* F: r% g
{
; |0 _4 x5 C/ P; `+ z if(Dir.IsPathDelimiter(i))$ ^5 D! e: m5 E2 q' x/ z) Y3 u0 K% k/ A
{
* s5 x0 L9 B) V5 r Parent=Dir.SubString(0,i);2 L8 C" p6 x! t, w7 \
break;6 j6 B2 B( S! T+ m# a
}
) d& A, C$ g3 l3 d, @ }+ g( a+ b o1 r2 L+ e
if(Parent.IsEmpty())return false; // 目录名称错误" S2 L# t9 I C9 G$ r5 Z
bool Ret=true;5 v8 y6 m4 F5 l7 u. `
if(Parent.Length()>3) // 如果长度小于3，表示为磁盘根目录
' c) i9 _4 E- [3 m( Q! E9 _ Ret=DirectoryExistEx(Parent.c_str());// 检查父目录是否存在
' V9 R& r7 m- U: g. F3 m if(!Ret)Ret=MakeDirectoryEx(Parent); // 父目录不存在,递归调用创建父目录& n$ K: F" d! ^( B% E# S
if(Ret) // 父目录存在,直接创建目录7 v/ q+ [4 a# V# t. u
{/ k& z# }& W! q( b1 `/ |
SECURITY_ATTRIBUTES sa;
; d! y4 _) ]0 K+ E2 n5 P sa.nLength=sizeof(SECURITY_ATTRIBUTES); J, C9 k8 L% V
sa.lpSecurityDescriptor=NULL;
& E% t- h# Q1 Y8 p5 J. {/ D sa.bInheritHandle=0;
) O7 M3 K( V! l6 R" _ Ret=CreateDirectory(Path,&sa);
( F, ^ Y2 ]% Q) y# I7 } }
( u: q4 j3 w5 \0 O( ^8 g4 c% {: z8 H! H return Ret;
/ S) i8 p3 b4 K* U} 3 Y; s X3 V/ \! ?. V
可以看出基本方法是:
6 ]8 J7 C% L0 G# Q5 l) h先检查父目录是否存在，这里用到的函数DirectoryExistEx可以按照前面介绍的方法设计； ' B6 R9 o8 z/ Y9 K3 G9 A5 V9 ^
如果父目录存在，则直接创建目录,否则自我调用创建父目录。
9 V) j |2 u0 ?. @, Q: z! H# Y
, S( }/ a: g9 A四、直接删除整个目录：
2 t) q# ^# y# t7 D; X# y
在DOS下有一个Deltree命令，用来删除整个目录，这是一个很有用的功能，可惜，Windows API提供的函数RemoveDirectory只能删除控目录，就像DOS的RD命令一样。编程实现这个功能同样需要递归方法，基本流程是：
' b& N% ~' W8 |3 d查找目录下的所有文件和目录，即调用API函数FindFirstFile、FindNextFile(*.*)
8 n. X4 O1 F0 m6 \' e5 w- X n' Z如果找到文件，则强制删除。所谓强制删除，即删除前先调用SetFileAttributes把它的属性设置为Normal，然后调用DeleteFile删除它。 0 F( A' F: ?1 k( l
如果找到目录，则进行自我调用，即开始递归过程。 . U) p% Z) q+ E/ s! ^
如果没有找到目录，即表示为控目录，调用RemoveDirectory直接删除。
1 U/ X0 m' a# i( q, n具体程序代码如下：
& r/ |* X9 k0 z3 g2 f! F' L0 g( G* g gbool DeleteDirectoryEx(const AnsiString &)# Y7 e$ m2 u) `0 @! ~9 j# g
{
# r' u+ _0 j; D9 d' k3 A& W if(P.IsEmpty() || P.Length()<4)return false; // 参数长度必须大于3，即不能为磁盘根目录或空白5 s3 p* z+ c" ]0 x* h! ?, N
int len=P.Length();
5 {4 m& t! N* M. x1 R char *Path=P.c_str();0 A, h1 ?0 C' F) C$ K% ]
AnsiString Dir=Path;* `: Y0 V% z2 p2 n# p
if(Path[len-1]!='\\')Dir=Dir+'\\';
7 l9 D1 R$ Y y5 i% w& y AnsiString Files=Dir+"*.*";
( K0 H: i" [5 B WIN32_FIND_DATA wfd;
: r/ K; A5 J# g/ x$ V4 @) b& O HANDLE hFind=FindFirstFile(Files.c_str(),&wfd);! Z7 q3 P7 D Q' s S' W: q5 z/ V
bool Ret=true;
8 z8 h* K' M8 {4 B c+ y AnsiString Tmp;/ s! p0 U2 ^$ i2 V0 Q
if(hFind!=INVALID_HANDLE_VALUE)
7 t# q3 O1 H* p' r. j7 Z1 T% G {/ e8 q8 R) t, L7 a/ d; `! n
bool bFind=true;% I2 W/ y; ~8 W! f* P& m
while(bFind)
6 g" q7 ?: A4 d/ ?' d {
' m7 b' i* B' b3 G4 q& r# J: H3 ~ if(wfd.cFileName[0]!='.') // . ..
9 m9 O; @+ v' t3 q% n% x {
. R7 Y# Z+ j( B* a" N8 [ Tmp=Dir+wfd.cFileName;' u: \9 s& y _/ J
if(wfd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
; G0 ]% M' L0 ?( g, P) t { // 删除所有子目录
+ W2 R2 W" t" c# l8 q P8 q Ret=Ret&&DeleteDirectoryEx(Tmp.c_str(),false);
$ C1 P& K1 K$ h1 {6 m9 ?% d }else( U8 t1 u ] {3 L
{ // 删除所有文件
) c# K* X/ k* {. p( F SetFileAttributes(Tmp.c_str(),FILE_ATTRIBUTE_NORMAL);
2 a1 A0 @3 ~$ ~, D& l) P: t, ^ Ret=Ret&&DeleteFile(Tmp.c_str());5 q8 L: Y, \) n P" m& q- B% \ d; `
}
5 g/ R7 \, Y x! ~8 F. P; D+ Y& P% H }$ M% q8 |3 \8 R" \7 w
bFind=FindNextFile(hFind,&wfd);1 R* v% B0 n4 Z) D
}2 Y& e! O3 V$ ^
FindClose(hFind); t, u, b9 L/ l5 D$ ?! H9 K; D3 M# x
}
. _& \, i3 z" \6 | if(Ret)return RemoveDirectory(Path);
! Z% R$ Z, o/ X! D( e' m4 b& N return false;
4 u- F# E6 {- r% ]" t- _' P}
$ y3 P& l. k) W# L3 c9 P R+ c( n0 A完
7 Q) @# {8 U+ [ u
% W2 a' z2 \7 D
| 欢迎光临 数学建模社区-数学中国 (http://www.madio.net/) |
Powered by Discuz! X2.5 |