数学建模社区-数学中国

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

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

目录处理函数

( G5 G1 E2 K# H A' d# F6 W

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

6 G. o7 a: L- V/ H2 `

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

7 P7 `4 _# l! r# c8 l% f/ W

一、判断目录是否存在:

. j/ R$ Y( b5 g4 G3 u: V

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

/ {$ D1 r$ Z1 R/ C9 |2 a# Z/ f

设char *Dir为带判断的目录+ k! P4 j: n/ E1 z' X ~2 Q( m bool Exist; // 最后结果,表示目录是否存在& K& f) M. T* B2 _4 O. y+ B3 ~ if(Dir[strlen(Dir)]=='\\')Dir[strlen(Dir)-1]='\0'; // 先删除最后的“\”9 A& T$ ^1 Q. N, P& t WIN32_FIND_DATA wfd; // 查找: L% O/ ^ V( |( x' Y* | HANDLE hFind=FindFirstFile(Dir,&wfd); 8 L+ x3 I8 F) N' c if(hFind==INVALID_HANDLE_VALUE)Exist=false; // 没有找到配备,目录肯定不存在' g/ k9 i6 ^/ _! f( X: J" P% H else0 ~1 p0 U# `$ S1 N3 z3 b. \ {+ u4 H! E) n8 r: Q+ {) R5 Z+ K if(wfd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) // 检查找到的结果是否目录 0 ~4 K' B; j0 } j$ ?8 Q6 c Exist=true; // 是目录,目录存在 2 j# N( A- `+ c" t5 J) N# A+ m else1 l) W. W3 H: A0 g/ A$ j4 q, m Exist=false; // 是目录,目录不存在 % d2 I' {& g3 K4 J0 {) c1 k FindClose(hFind); M& i8 P( D& E/ ~4 M: O}

' t( o: C! M4 L& \

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

* L; l$ h! w( `0 l7 x1 J

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

- J% E1 t- |5 ?4 R

HWND hwndOwner; // 拥有对话框的窗口,可以设置为Application->Handle ' Q0 {! J6 f) r; w$ r& N/ FLPCITEMIDLIST pidlRoot; // ITEMIDLIST类型的指针,表示在哪个路径下选择,一般可以设置为NULL6 F' ?6 ]- P: `9 U% h& k5 {( t0 q; S5 D LPSTR pszDisplayName; // 选择后,所选目录的名称(不包含父级目录)被拷贝到这个指针指向的位置 0 M1 z8 {- U3 q6 XLPCSTR lpszTitle; // 作为标题显示在对话框中目录树的上面,可以根据实际情况设置 + g/ L' e F3 Z4 ~UINT ulFlags; // 标志位,有点复杂,一般设置为BIF_RETURNONLYFSDIRS 4 ]5 s! p, _: Q: KBFFCALLBACK lpfn; // 回调函数,一般不用,设置为NULL % y# V5 e6 N; R# SLPARAM lParam; // 预定义的对话框传递给回调函数的值 4 z' B( X9 [' W, n5 R! Z! _int iImage; // 与所选目录相关联的图标在系统图标集合中的索引 7 f$ n6 x5 F8 a) m2 ] 可以看出,使用函数SHBrowseForFolder还真麻烦,普通爱好者掌握它确实有一定的难度,现给出完整程序段如下: $ F8 V, B. ?& T2 M# p#include <shlobj.h> // 必须包含的头文件0 M- [' b; N# |" H char SelectedDir[MAX_PATH]; // 最终结果) w. b" t! o2 _5 F# ~& h9 J' O1 Q: q BROWSEINFO bi; // 入参 1 J2 g7 L$ h+ B6 O9 O" s+ `! S% p0 qchar FolderName[MAX_PATH]; // 所选目录名称,例如选择C:\Windows\Font,则为Font 5 g+ ]9 w. M% U' L LPITEMIDLIST ItemID; // 所选目录的系统标志指针

& X9 M5 L" ~. K" c1 \

memset(SelectedDir, 0, MAX_PATH); // 初始化最终结果- y o9 b+ A4 Q8 g( m( G$ e memset(&bi, 0, sizeof(BROWSEINFO)); // 初始化入参所有数据 2 W1 M( L5 q( I9 q" n& n" ?bi.hwndOwner = Application->Handle;- u x3 c+ l( O$ V! s: n5 v bi.pszDisplayName = FolderName;5 G6 v$ m$ w( f7 Y6 E l; N bi.lpszTitle = "请选择目录"; // 改成自己希望的 9 ? S' Z+ t; H/ ~- E8 X6 }3 Hbi.ulFlags=BIF_RETURNONLYFSDIRS; : y2 L2 G4 T. A$ M: n GItemID = SHBrowseForFolder(&bi); // 调用函数,打开目录选择对话框) `$ `$ W4 M+ p/ d3 F' N' d E if(ItemID) 7 r5 Z( ~8 Q! \6 t# {{ 8 X- m1 i& Z; r- z SHGetPathFromIDList(ItemID, SelectedDir); // 获取所选目录的全名 4 P/ q" w8 P( ] GlobalFree(ItemID); // 返回的ItemID占用了系统资源,不要忘了释放' Q' ~( |( s. G- M }

6 y0 B: d' S8 K, V: W

三、直接建立多级目录:

$ Z$ H" O! N4 U7 p, {# u$ d4 W

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

3 J p! h9 B T P, t" `

bool MakeDirectoryEx(const AnsiString &) // 入参为打算创建的目录名,根据操作结果返回"true"或"false" $ o2 E( u- ]7 k" J l: b{ - {0 j v& ?" J/ U% u if(P.IsEmpty())return false;3 o+ i: w# j$ Z' i5 } int len=P.Length(); 6 U8 z% M. i% ~2 C& W2 v2 X: T char *Path=P.c_str(); ) y; ?) I% i- P- n3 C) y* c* l if(Path[len-1]=='\\') 5 v* ~; ]5 I: m! F { $ A* e5 i& x4 Z" C$ b, S0 _ len--;4 r: l- N' Z% `3 G, K Path[len]='\0'; 1 B, b4 K4 G* z: Y: I( D! ` } // 删除末尾的"\"7 W0 F3 M$ v: h2 a: J) X# Q AnsiString Dir=Path;3 y2 e+ V/ Z+ Y. q/ t. A // 分开父目录和本身目录名称 3 a% q5 O& P9 P' o1 l: S AnsiString Parent;3 Z( K( |: {" X0 f for(int i=len-1;i>0;i--) ( L6 u) I3 @4 [1 T3 t# X1 ^ { 6 {$ S, d! G% Z5 \$ A# ~/ X if(Dir.IsPathDelimiter(i))- z' v1 L7 J- F5 h/ R: i7 X- Q { 1 L9 W _8 ?1 F9 P6 K Parent=Dir.SubString(0,i); 4 r$ {& n8 v( q( l1 |9 U; s break;$ E, ?3 G0 x, v3 P! S0 A3 z1 I t } % `3 K) Y) ] T1 C4 i2 j } ' j4 m5 v k6 F2 |3 C5 g L$ W if(Parent.IsEmpty())return false; // 目录名称错误 7 a: ~" c7 `$ w2 { bool Ret=true; + W! z) @" `$ n% M; E, a- n' e if(Parent.Length()>3) // 如果长度小于3,表示为磁盘根目录2 o' J7 _+ u$ v3 H1 R Ret=DirectoryExistEx(Parent.c_str());// 检查父目录是否存在' o5 N1 ]) Z& {/ x6 I if(!Ret)Ret=MakeDirectoryEx(Parent); // 父目录不存在,递归调用创建父目录* k% v, A0 Z) B) [& z P if(Ret) // 父目录存在,直接创建目录0 Z2 P/ X3 R7 n { ) y* B$ J; x r4 B( L3 o* m Z+ q3 D. o SECURITY_ATTRIBUTES sa;% i4 s7 l* v( B+ p' ^( Y sa.nLength=sizeof(SECURITY_ATTRIBUTES);2 P: S& Z/ B+ c sa.lpSecurityDescriptor=NULL; $ f X R! Y- T5 V" N* w sa.bInheritHandle=0; ' @* l% N! A* p+ ~3 @ Ret=CreateDirectory(Path,&sa); : j9 a4 q% E' y4 K% E, i! R. Q# Q3 M }: c7 L* F$ }9 {! q O6 X return Ret;' K2 y) g9 q, U! Q4 s) }$ N* @ } 7 M6 u- [" i( ?5 g: Z 可以看出基本方法是: ) h5 \2 {! _2 V* o% m* j4 Z1 u* Z先检查父目录是否存在,这里用到的函数DirectoryExistEx可以按照前面介绍的方法设计; 5 T# H$ g: t0 u! U3 t如果父目录存在,则直接创建目录,否则自我调用创建父目录。

5 I2 {# t. X& d% A: z% X

3 V. j9 T3 r+ Y$ k8 X6 s 四、直接删除整个目录:

' l$ ?- k/ s# J: U

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

6 C4 t# {: C$ q; y) `

查找目录下的所有文件和目录,即调用API函数FindFirstFile、FindNextFile(*.*) 0 V6 {+ {% U- t7 ` 如果找到文件,则强制删除。所谓强制删除,即删除前先调用SetFileAttributes把它的属性设置为Normal,然后调用DeleteFile删除它。 $ p+ F+ w' ]% i, v8 }; F如果找到目录,则进行自我调用,即开始递归过程。 # r: ]) D1 A, y, c/ q如果没有找到目录,即表示为控目录,调用RemoveDirectory直接删除。 0 N4 @/ y: T5 S* j1 v% C 具体程序代码如下:

G* s9 H5 M) f

bool DeleteDirectoryEx(const AnsiString &) [. [0 i- V9 G( m2 K {. Q: ~( ?) r5 \9 x [" ?3 y+ J d if(P.IsEmpty() || P.Length()<4)return false; // 参数长度必须大于3,即不能为磁盘根目录或空白 |2 O% `2 `0 t5 @6 K+ W" [ int len=P.Length();2 v6 A2 Z6 n. X- J; X! F7 d char *Path=P.c_str(); 8 Y2 X, J3 H" v" {. u; b AnsiString Dir=Path;& G& n3 h: O' N9 P3 E* [9 i if(Path[len-1]!='\\')Dir=Dir+'\\';: S# ^, t7 v5 l AnsiString Files=Dir+"*.*"; 7 O& e7 N* K K% y WIN32_FIND_DATA wfd; , @4 E. K J. U7 I' t+ i HANDLE hFind=FindFirstFile(Files.c_str(),&wfd); 1 o7 y+ h6 @4 Q6 V. p bool Ret=true; 5 |' A, T$ L) f$ o4 w, F4 J2 O5 ^ AnsiString Tmp;/ @! m9 {( Y; z1 U if(hFind!=INVALID_HANDLE_VALUE)# J3 j6 R! ]7 y7 z4 t {' N) [6 E; t0 u+ m) d0 s bool bFind=true; ) S- L7 B) r, R while(bFind) " _9 G6 b1 G) t' L1 M N+ T2 \ {# ~/ e F1 y; w if(wfd.cFileName[0]!='.') // . ..! c" R( R, U$ B& @+ u- ^ { : D0 `8 M |+ o4 O: o# W4 _2 v/ o Tmp=Dir+wfd.cFileName; 3 I; n- B# {% F) R: H) n if(wfd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)9 a9 v9 ^( L' s6 C7 G" J9 y { // 删除所有子目录- p9 j3 b5 M6 P, _# ~% v7 I Ret=Ret&&DeleteDirectoryEx(Tmp.c_str(),false); / z' L- a1 @' |# e }else6 j! s' }) x/ |1 Z. y/ q { // 删除所有文件0 f8 @' o! s+ q1 n: V6 e SetFileAttributes(Tmp.c_str(),FILE_ATTRIBUTE_NORMAL);5 ?5 M( q4 | V- n# e Ret=Ret&&DeleteFile(Tmp.c_str()); % z5 O: V( t2 w. z }( J+ Z$ M: P2 { } ( s- l( S9 P5 C! ` bFind=FindNextFile(hFind,&wfd); - h) R! H' y" v3 W9 W1 ?/ _ }3 B a5 z) ?* F5 a( x FindClose(hFind); ) p! ~, W1 R9 d. u; d [ }) r2 y, u B5 K if(Ret)return RemoveDirectory(Path);9 z& ]. A; c2 p return false; $ s( F/ ?2 B! M/ T* m} ; @6 i: R. ?3 Y+ A

) D8 H" Q6 Q4 Y( H3 ]2 F- f

2 \2 Z. ]- l: z6 Y* r+ L






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