streamcounter.h
Go to the documentation of this file.
1 /*
2  * Copyright (C) 2006 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_STREAMCOUNTER_H
30 #define CXXTOOLS_STREAMCOUNTER_H
31 
32 #include <iostream>
33 
34 namespace cxxtools
35 {
36  template <typename CharType, typename Traits>
37  class BasicStreamcounterBuf : public std::basic_streambuf<CharType, Traits>
38  {
39  typedef Traits traits_type;
40  typedef typename Traits::int_type int_type;
41 
42  unsigned count;
43 
44  public:
46  : count(0)
47  { }
48 
49  unsigned getCount() const { return count; }
50  void resetCount(unsigned count_) { count = count_; }
51 
52  private:
53  int_type overflow(int_type ch)
54  { ++count; return 0; }
55 
56  int_type underflow()
57  { return traits_type::eof(); }
58 
59  int sync()
60  { return 0; }
61  };
62 
63  template <typename CharType, typename Traits>
64  class BasicStreamcounter : public std::basic_ostream<CharType, Traits>
65  {
67 
68  public:
70  : std::basic_ostream<CharType, Traits>(0)
71  {
72  std::basic_ostream<CharType, Traits>::init(&streambuf);
73  }
74 
75  unsigned getCount() const { return streambuf.getCount(); }
76  void resetCount(unsigned count = 0) { streambuf.resetCount(count); }
77  };
78 
80 }
81 
82 #endif // CXXTOOLS_STREAMCOUNTER_H
83