数学建模社区-数学中国

标题: 目录处理函数 [打印本页]

作者: 韩冰    时间: 2005-1-26 13:18
标题: 目录处理函数

目录处理函数

" U3 N* N9 K+ `# r" C

编程语言:C++ Builder 作者:王行舟

, `5 r* y! a& A" B$ s( A% J9 n

在编程时,经常有一些针对目录的操作,如打开目录对话框选择一个目录,直接创建多级目录,直接删除多级目录,判断某个目录是否存在等。本文就这些问题给出编程实现方法,并给出详细的程序代码,供各位编程爱好者参考。

* o. x! e. ]5 L' b

一、判断目录是否存在:

5 G% ^3 ~! p7 q/ i

C++ Builder中提供了检查文件是否存在的函数FileExists,但没有提供检查目录是否存在的函数,我们可以用Windows API函数FindFirstFile实现这个功能。程序实现如下:

/ b; e2 B6 S' u" S+ X2 L# _9 L

设char *Dir为带判断的目录 + e5 H+ w( K4 ^bool Exist; // 最后结果,表示目录是否存在 / R2 F. b w, \6 `. mif(Dir[strlen(Dir)]=='\\')Dir[strlen(Dir)-1]='\0'; // 先删除最后的“\” ( c! D( X$ F4 V0 q- ~2 B+ W$ i9 oWIN32_FIND_DATA wfd; // 查找$ `) U: M4 X0 [; j, c HANDLE hFind=FindFirstFile(Dir,&wfd); ) M- U3 K; ], L8 Y* u$ Oif(hFind==INVALID_HANDLE_VALUE)Exist=false; // 没有找到配备,目录肯定不存在 5 M5 P( c( K6 S& m, X, }else) W5 l. Q; b- b2 V7 Y3 P( B {0 m5 i2 h9 Y2 ~! H, p' [ if(wfd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) // 检查找到的结果是否目录5 T$ d9 C s- T' Z Exist=true; // 是目录,目录存在/ G- _ e' P) {- M else! R7 @8 S' [7 k Exist=false; // 是目录,目录不存在0 G5 f5 `: G& j1 h4 S/ U FindClose(hFind); ) ^# S" R9 `! S; `}

$ m% r% b' B# _. @8 G

二、打开目录选择对话框选择一个目录:

& Y1 s: D5 Q" R7 i( K$ Y4 z& t

大多专业软件在要求输入目录的编辑框旁都放了一个按钮,点击后打开一个目录窗口,很多编程爱好者也希望能掌握这个方法。实现这个功能要调用Windows API函数SHBrowseForFolder,完整声明为WINSHELLAPI LPITEMIDLIST WINAPI SHBrowseForFolder(LPBROWSEINFO lpbi),返回一个ITEMIDLIST类型的指针,通过这个指针调用函数SHGetPathFromIDList可以确定所选择的目录的全名称。入参为BROWSEINFO结构的指针,这个结构较为复杂,成员如下所示:

. d8 V. p s+ W% g

HWND hwndOwner; // 拥有对话框的窗口,可以设置为Application->Handle 0 d) B4 z9 b8 m9 c0 T! }, kLPCITEMIDLIST pidlRoot; // ITEMIDLIST类型的指针,表示在哪个路径下选择,一般可以设置为NULL * ]$ m- I7 t3 b& j) v3 nLPSTR pszDisplayName; // 选择后,所选目录的名称(不包含父级目录)被拷贝到这个指针指向的位置 - Z2 J% I+ ]* ` LPCSTR lpszTitle; // 作为标题显示在对话框中目录树的上面,可以根据实际情况设置 / g( `* Q4 m* `9 z7 x1 ZUINT ulFlags; // 标志位,有点复杂,一般设置为BIF_RETURNONLYFSDIRS 8 l3 F5 i! B. Y+ j8 H+ U BFFCALLBACK lpfn; // 回调函数,一般不用,设置为NULL # \8 a' b, A" o" p/ uLPARAM lParam; // 预定义的对话框传递给回调函数的值 1 R! d1 r8 R, |3 D$ X" S' Yint iImage; // 与所选目录相关联的图标在系统图标集合中的索引 ) C( c) a6 k' e; m0 t& Z. [5 _可以看出,使用函数SHBrowseForFolder还真麻烦,普通爱好者掌握它确实有一定的难度,现给出完整程序段如下: , G( w! q2 h# [% H& e( V. ?) c# W #include <shlobj.h> // 必须包含的头文件 6 @' f: {/ Y& Fchar SelectedDir[MAX_PATH]; // 最终结果 9 p& r' X3 b& S& mBROWSEINFO bi; // 入参 + C- ~! q2 e4 l; a6 Y8 |$ U0 C& bchar FolderName[MAX_PATH]; // 所选目录名称,例如选择C:\Windows\Font,则为Font 2 D$ O* r0 A1 a2 d1 L1 D; _LPITEMIDLIST ItemID; // 所选目录的系统标志指针

5 G# ^( R9 ?+ S3 I8 q

memset(SelectedDir, 0, MAX_PATH); // 初始化最终结果( R x. @) x8 i! p; P memset(&bi, 0, sizeof(BROWSEINFO)); // 初始化入参所有数据6 |6 {" X, t7 w bi.hwndOwner = Application->Handle;/ C2 |2 u) z) P( m6 A" X! k bi.pszDisplayName = FolderName;; s/ T6 B: E1 K. r- V. }! ^. A bi.lpszTitle = "请选择目录"; // 改成自己希望的7 X. Z) U; x, ^3 k- P3 D bi.ulFlags=BIF_RETURNONLYFSDIRS; " N. ~2 ]9 O2 S0 Q- h6 c: NItemID = SHBrowseForFolder(&bi); // 调用函数,打开目录选择对话框 6 h8 D# x' q; s! ~# c0 Q' k" X2 sif(ItemID)1 F3 Y3 ^( J$ j. Z- i4 | { ) z& M' x9 _, k7 T' E SHGetPathFromIDList(ItemID, SelectedDir); // 获取所选目录的全名 7 E% K2 ]( c% H W- q- o GlobalFree(ItemID); // 返回的ItemID占用了系统资源,不要忘了释放5 ~1 m& I* X9 E! E- M }

( f4 p0 o% y7 P8 y/ E$ C2 A

三、直接建立多级目录:

( n9 q2 k9 C( ~ W8 H0 }( I; I

Windows API提供了建立目录的函数CreateDirectory,但是调用前要保证父目录必须存在,否则会失败。其实,有时越级建立多级目录很有用,因为在建立目录特别是建立多层目录时,层层加以判断会大大地增加程序的复杂程度。如何实现这个功能呢?本人用递归方法设计了一个可以直接建立多级目录的函数,现说明如下,供各位朋友参考。

. V& v9 R# R% Y& Y

bool MakeDirectoryEx(const AnsiString &) // 入参为打算创建的目录名,根据操作结果返回"true"或"false"4 Z4 i0 k( {- v% k; u9 K+ B {6 |0 \: u5 Q# S! b/ f if(P.IsEmpty())return false;" ?& r5 z% I |7 r9 C+ T int len=P.Length(); ) ?( o/ b$ h7 Z! `+ v) U char *Path=P.c_str();7 |3 f' O. a D( b if(Path[len-1]=='\\') ( l1 p8 b$ ^: [1 z2 a3 { { ) D( j& _' x3 P% D" `: O7 y! ` len--;* z! H& ^) g+ B7 C+ w& O Path[len]='\0';( Z# i! e/ ]3 P( N; c+ S: m4 S } // 删除末尾的"\" 7 }2 b6 @3 h1 G( A# \ AnsiString Dir=Path;- I0 v' w$ L7 x& F/ K0 \0 l // 分开父目录和本身目录名称5 D% w2 y! _$ r [/ B: j AnsiString Parent; ' \+ _: \+ x* ~6 e for(int i=len-1;i>0;i--)- {: f9 T0 F0 A: T$ I: a+ I { Y* Q# G/ W+ D6 \: ~ if(Dir.IsPathDelimiter(i)) 3 Y6 E0 o" v* w1 a- q { # g1 \% i% e, s% f Parent=Dir.SubString(0,i);$ _$ \* W+ v8 l break; 9 ]/ E0 m/ a0 y, G- k } / w4 ]6 \# x* A) r) w) [; w, a }9 E; i! k8 g! U- u4 S if(Parent.IsEmpty())return false; // 目录名称错误 & _& I$ ?" }! d- w) c3 o bool Ret=true; ( {* h# m+ k5 g if(Parent.Length()>3) // 如果长度小于3,表示为磁盘根目录8 @ k; W% j5 R Y+ w* u) j! ? Ret=DirectoryExistEx(Parent.c_str());// 检查父目录是否存在) a7 J6 G5 B9 J3 a) X2 N if(!Ret)Ret=MakeDirectoryEx(Parent); // 父目录不存在,递归调用创建父目录 , D; m7 z; W% i- n) W4 W if(Ret) // 父目录存在,直接创建目录5 N5 X- D. K0 D, u3 |+ Y { , t6 t, \- P% b' s SECURITY_ATTRIBUTES sa;% M1 [# L( d4 n+ S sa.nLength=sizeof(SECURITY_ATTRIBUTES);% `3 T F: `; M& q( ]/ |- b sa.lpSecurityDescriptor=NULL;) J8 m3 V- C: ]$ u/ h sa.bInheritHandle=0;- {- }6 ?; ^, }& q! K7 g$ S Ret=CreateDirectory(Path,&sa); & Z0 Q# a9 f, Y' Q }! \7 Z X% Y b6 \9 V return Ret; m s5 ?- z7 {+ j! X0 e( g } " ]- Y& [& X Q( X7 g% u: Q8 h G 可以看出基本方法是:: W+ t- }6 K% U; d 先检查父目录是否存在,这里用到的函数DirectoryExistEx可以按照前面介绍的方法设计; 7 S5 Z% ]6 Z; { q5 X; v5 |如果父目录存在,则直接创建目录,否则自我调用创建父目录。

$ u6 D% k! @$ W: p a, k# j

( }0 F& ]/ |& R1 ]7 `+ Z. z5 n% ^$ {. g四、直接删除整个目录:

/ v- n- \! ^) a6 p. ~( H

在DOS下有一个Deltree命令,用来删除整个目录,这是一个很有用的功能,可惜,Windows API提供的函数RemoveDirectory只能删除控目录,就像DOS的RD命令一样。编程实现这个功能同样需要递归方法,基本流程是:

3 j$ U3 d9 C6 U9 C. c W+ p: P

查找目录下的所有文件和目录,即调用API函数FindFirstFile、FindNextFile(*.*) ( L/ R7 M' z2 H! Z+ V如果找到文件,则强制删除。所谓强制删除,即删除前先调用SetFileAttributes把它的属性设置为Normal,然后调用DeleteFile删除它。 2 |" d1 m3 ~9 K: U3 \: Q7 w如果找到目录,则进行自我调用,即开始递归过程。 - J1 m/ Q; k( H" S3 T$ y3 e 如果没有找到目录,即表示为控目录,调用RemoveDirectory直接删除。 $ V* b3 F" w" y X具体程序代码如下:

. [5 H5 X/ `: E" D4 i+ n

bool DeleteDirectoryEx(const AnsiString &) 6 {! N$ l2 x3 C{ ) }" A; ^. q4 J6 b if(P.IsEmpty() || P.Length()<4)return false; // 参数长度必须大于3,即不能为磁盘根目录或空白3 {$ A% }& h4 ? int len=P.Length(); / `7 W) C1 p7 M0 L: l char *Path=P.c_str(); . A. h- {1 ^4 B9 t; B5 q& ?9 p AnsiString Dir=Path; U; H1 \% N K$ f- B if(Path[len-1]!='\\')Dir=Dir+'\\'; * n) z% S/ a+ X- F: L9 r9 ? AnsiString Files=Dir+"*.*"; 1 p$ S3 q$ v$ Y& u+ }, _1 j WIN32_FIND_DATA wfd;2 k8 u4 {# N5 Y) n HANDLE hFind=FindFirstFile(Files.c_str(),&wfd); 3 {6 q9 ~2 y z/ N, }* } bool Ret=true; 7 I J; L% } X1 x4 Z6 z4 S AnsiString Tmp; : B; o, K+ k/ g$ W$ L+ f- ?. m if(hFind!=INVALID_HANDLE_VALUE)1 u" V/ s4 q1 Y/ p { $ B) ~( a! d( i; A* k1 y bool bFind=true;: |9 C! L; y. o( L( d while(bFind) , E' w5 i7 w; E {3 R# ?( w/ R" E if(wfd.cFileName[0]!='.') // . .. ; i. U0 F- F- m! s# i$ g* Y6 e. m { 9 b( D/ S+ l/ v0 X; z9 W3 W0 K; S Tmp=Dir+wfd.cFileName; # ?. T4 ?3 R- j6 z* d: s! |0 q* L# u/ [ if(wfd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)$ E, R" Y. O! L5 L- S { // 删除所有子目录 ; c# X/ E, R0 s' B. T6 d9 c Ret=Ret&&DeleteDirectoryEx(Tmp.c_str(),false); |: ]; k1 u1 P# }6 i- } }else% I9 {) Z2 n- `# E8 E% j { // 删除所有文件( k `/ x. F& u- f0 s6 N! X SetFileAttributes(Tmp.c_str(),FILE_ATTRIBUTE_NORMAL); 5 w1 o9 ]7 o8 m: Q2 V" a1 @% o Ret=Ret&&DeleteFile(Tmp.c_str());8 F- d2 j4 Y! g/ C } 7 V- \$ G& q( ^; S& B' a }# u' x: `$ u- L1 @ bFind=FindNextFile(hFind,&wfd); , m7 u7 s1 o" B7 a- y4 U4 L }$ L0 Y, G) a( h7 ^ FindClose(hFind);- K) R- Y8 i+ Q }( l' y( \* F+ ?/ E/ a5 T6 h if(Ret)return RemoveDirectory(Path);/ s, D6 }+ [; D+ B# r return false;. O* A- D' f/ N' j) f+ b } # @6 |- K2 k. d

& ] Q- Y9 _, w I$ T& ]

5 h1 e% |0 J7 C- K0 l+ y" |; f3 \+ Q






欢迎光临 数学建模社区-数学中国 (http://www.madio.net/) Powered by Discuz! X2.5