1 | #ifndef MARS_MDirIter
|
---|
2 | #define MARS_MDirIter
|
---|
3 |
|
---|
4 | #ifndef ROOT_TObjArray
|
---|
5 | #include <TObjArray.h>
|
---|
6 | #endif
|
---|
7 |
|
---|
8 | class MDirIter : public TObject
|
---|
9 | {
|
---|
10 | private:
|
---|
11 | TObjArray fList;
|
---|
12 | TString fFilter;
|
---|
13 |
|
---|
14 | TIter fNext; //!
|
---|
15 | void *fDirPtr; //!
|
---|
16 | TObject *fCurrentPath; //!
|
---|
17 |
|
---|
18 | void *Open();
|
---|
19 | void Close();
|
---|
20 | Bool_t CheckEntry(const TString n) const;
|
---|
21 | Int_t IsDir(const char *dir) const;
|
---|
22 | Bool_t MatchFilter(const TString &name, TString filter) const;
|
---|
23 | TString ConcatFileName(const char *dir, const char *name) const;
|
---|
24 | void PrintEntry(const TObject &o) const;
|
---|
25 |
|
---|
26 | public:
|
---|
27 | MDirIter() : fNext(&fList), fDirPtr(NULL)
|
---|
28 | {
|
---|
29 | fList.SetOwner();
|
---|
30 | }
|
---|
31 | MDirIter(const MDirIter &dir) : TObject(), fNext(&fList), fDirPtr(NULL)
|
---|
32 | {
|
---|
33 | fList.SetOwner();
|
---|
34 |
|
---|
35 | TObject *o=NULL;
|
---|
36 | TIter NextD(&dir.fList);
|
---|
37 | while ((o=NextD()))
|
---|
38 | AddDirectory(o->GetName(), o->GetTitle());
|
---|
39 | }
|
---|
40 | MDirIter(const char *dir, const char *filter="", Int_t rec=0) : fNext(&fList), fDirPtr(NULL)
|
---|
41 | {
|
---|
42 | fList.SetOwner();
|
---|
43 | AddDirectory(dir, filter, rec);
|
---|
44 | }
|
---|
45 | ~MDirIter()
|
---|
46 | {
|
---|
47 | Close();
|
---|
48 | }
|
---|
49 |
|
---|
50 | Int_t ReadList(const char *fname, const TString &path="");
|
---|
51 | void Sort();
|
---|
52 |
|
---|
53 | Int_t AddDirectory(const char *dir, const char *filter="", Int_t recursive=0);
|
---|
54 | Int_t AddFile(const char *name);
|
---|
55 | void Add(const MDirIter &iter);
|
---|
56 | void Reset();
|
---|
57 |
|
---|
58 | TString Next(Bool_t nocheck=kFALSE);
|
---|
59 | TString operator()(Bool_t nocheck=kFALSE) { return Next(nocheck); }
|
---|
60 |
|
---|
61 | void SetFilter(const char *f="") { fFilter = f; }
|
---|
62 | UInt_t GetNumEntries() const
|
---|
63 | {
|
---|
64 | UInt_t n = 0;
|
---|
65 | MDirIter NextD(*this);
|
---|
66 | while (!NextD().IsNull()) n++;
|
---|
67 | return n;
|
---|
68 | }
|
---|
69 |
|
---|
70 | void Print(const Option_t *o="") const;
|
---|
71 |
|
---|
72 | ClassDef(MDirIter, 1) // Iterator for files in several directories (with filters)
|
---|
73 | };
|
---|
74 |
|
---|
75 | #endif
|
---|