multifstream.h
Go to the documentation of this file.
1 /*
2  * Copyright (C) 2003 Tommi Maekitalo
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * As a special exception, you may use this file as part of a free
10  * software library without restriction. Specifically, if other files
11  * instantiate templates or use macros or inline functions from this
12  * file, or you compile this file and link it with other files to
13  * produce an executable, this file does not by itself cause the
14  * resulting executable to be covered by the GNU General Public
15  * License. This exception does not however invalidate any other
16  * reasons why the executable file might be covered by the GNU Library
17  * General Public License.
18  *
19  * This library is distributed in the hope that it will be useful,
20  * but WITHOUT ANY WARRANTY; without even the implied warranty of
21  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
22  * Lesser General Public License for more details.
23  *
24  * You should have received a copy of the GNU Lesser General Public
25  * License along with this library; if not, write to the Free Software
26  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
27  */
28 
29 #ifndef CXXTOOLS_MULTIFSTREAM_H
30 #define CXXTOOLS_MULTIFSTREAM_H
31 
32 #include <fstream>
33 #include <glob.h>
34 #include <string>
35 #include <queue>
36 
37 namespace cxxtools
38 {
39 
46 class multifstreambuf : public std::streambuf
47 {
48  glob_t mglob;
49  unsigned current;
50  std::filebuf file;
51  char ch;
52  typedef std::queue<std::pair<std::string, int> > patterns_type;
53  patterns_type patterns;
54 
55  public:
57  explicit multifstreambuf(const char* pattern, int flags = 0);
59 
60  bool open_next();
61 
62  int_type overflow(int_type c);
63  int_type underflow();
64  int sync();
65 
66  const char* current_filename() const
67  { return mglob.gl_pathv[current]; }
68 
69  void add_pattern(const std::string& pattern, int flags = 0)
70  { patterns.push(patterns_type::value_type(pattern, flags)); }
71 };
72 
73 class multi_ifstream : public std::istream
74 {
75  multifstreambuf buffer;
76 
77  public:
79  : std::istream(0)
80  {
81  init(&buffer);
82  }
83  multi_ifstream(const char* pattern, int flags = 0)
84  : std::istream(0),
85  buffer(pattern, flags)
86  {
87  init(&buffer);
88  }
89 
90  const char* current_filename() const
91  { return buffer.current_filename(); }
92 
93  void add_pattern(const std::string& pattern, int flags = 0)
94  { buffer.add_pattern(pattern, flags); }
95 
98  : public std::iterator<std::output_iterator_tag, const std::string>
99  {
100  multi_ifstream& m;
101  public:
103  : m(m_) { }
104  pattern_inserter& operator= (const std::string& p)
105  { m.add_pattern(p); return *this; }
107  { return *this; }
109  { return *this; }
111  { return *this; }
112  };
114  { return pattern_inserter(*this); }
115 };
116 
117 }
118 
119 #endif // CXXTOOLS_MULTIFSTREAM_H
120