| 1 | < | /* | 
| 1 | > | /* | 
| 2 |  | * Copyright (c) 2005 The University of Notre Dame. All Rights Reserved. | 
| 3 |  | * | 
| 4 |  | * The University of Notre Dame grants you ("Licensee") a | 
| 45 |  | #include <vector> | 
| 46 |  | namespace oopse { | 
| 47 |  |  | 
| 48 | < | /** | 
| 49 | < | * @class basic_teebuf basic_teebuf.hpp "utils/basic_teebuf.hpp" | 
| 50 | < | * @brief As a subclass of basic_streambuf,  basic_teebuf can operate on multiple stream simultaneously. | 
| 51 | < | * @code | 
| 52 | < | *    std::ofstream file1("file1"); | 
| 53 | < | *    std::ofstream file2("file22"); | 
| 54 | < | *    std::vector<std::streambuf*> buffers; | 
| 55 | < | *    buffers.push_back(file1.rdbuf()); | 
| 56 | < | *    buffers.push_back(file2.rdbuf()); | 
| 57 | < | *    teebuf tmp(buffers.begin(), buffers.end()); | 
| 58 | < | *    std::ostream myOs(&tmp); | 
| 59 | < | *    myOs << "hello world"; | 
| 60 | < | * @endcode | 
| 61 | < | */ | 
| 48 | > | /** | 
| 49 | > | * @class basic_teebuf basic_teebuf.hpp "utils/basic_teebuf.hpp" | 
| 50 | > | * @brief As a subclass of basic_streambuf,  basic_teebuf can operate on multiple stream simultaneously. | 
| 51 | > | * @code | 
| 52 | > | *    std::ofstream file1("file1"); | 
| 53 | > | *    std::ofstream file2("file22"); | 
| 54 | > | *    std::vector<std::streambuf*> buffers; | 
| 55 | > | *    buffers.push_back(file1.rdbuf()); | 
| 56 | > | *    buffers.push_back(file2.rdbuf()); | 
| 57 | > | *    teebuf tmp(buffers.begin(), buffers.end()); | 
| 58 | > | *    std::ostream myOs(&tmp); | 
| 59 | > | *    myOs << "hello world"; | 
| 60 | > | * @endcode | 
| 61 | > | */ | 
| 62 |  |  | 
| 63 | < | template <class CharT, class Traits = std::char_traits<CharT> > | 
| 64 | < | class basic_teebuf: public std::basic_streambuf<CharT, Traits> { | 
| 65 | < | public: | 
| 66 | < | typedef std::basic_streambuf<CharT, Traits> streambuf_type; | 
| 67 | < | typedef Traits traits_type; | 
| 68 | < | typedef CharT char_type; | 
| 69 | < | typedef typename traits_type::int_type int_type; | 
| 63 | > | template <class CharT, class Traits = std::char_traits<CharT> > | 
| 64 | > | class basic_teebuf: public std::basic_streambuf<CharT, Traits> { | 
| 65 | > | public: | 
| 66 | > | typedef std::basic_streambuf<CharT, Traits> streambuf_type; | 
| 67 | > | typedef Traits traits_type; | 
| 68 | > | typedef CharT char_type; | 
| 69 | > | typedef typename traits_type::int_type int_type; | 
| 70 |  |  | 
| 71 | < | template <typename ForwardIterator> | 
| 72 | < | basic_teebuf(ForwardIterator begin, ForwardIterator end) : buffers_(begin, end){ | 
| 71 | > | template <typename ForwardIterator> | 
| 72 | > | basic_teebuf(ForwardIterator begin, ForwardIterator end) : buffers_(begin, end){ | 
| 73 |  |  | 
| 74 | < | } | 
| 74 | > | } | 
| 75 |  |  | 
| 76 | < | protected: | 
| 77 | < | int_type overflow(int_type c = traits_type::eof()) { | 
| 76 | > | protected: | 
| 77 | > | int_type overflow(int_type c = traits_type::eof()) { | 
| 78 |  |  | 
| 79 | < | //avoid writing eof to stream | 
| 80 | < | if (c == traits_type::eof()) { | 
| 81 | < | return traits_type::eof(); | 
| 82 | < | } | 
| 79 | > | //avoid writing eof to stream | 
| 80 | > | if (c == traits_type::eof()) { | 
| 81 | > | return traits_type::eof(); | 
| 82 | > | } | 
| 83 |  |  | 
| 84 | < | typename std::vector<streambuf_type*>::iterator iter; //typename is needed since it's a dependant name | 
| 85 | < | for (iter = buffers_.begin(); iter != buffers_.end(); ++iter) { | 
| 86 | < | if ((*iter)->sputc(c) ==  traits_type::eof()) { | 
| 87 | < | return  traits_type::eof(); | 
| 88 | < | } | 
| 89 | < | } | 
| 84 | > | typename std::vector<streambuf_type*>::iterator iter; //typename is needed since it's a dependant name | 
| 85 | > | for (iter = buffers_.begin(); iter != buffers_.end(); ++iter) { | 
| 86 | > | if ((*iter)->sputc(c) ==  traits_type::eof()) { | 
| 87 | > | return  traits_type::eof(); | 
| 88 | > | } | 
| 89 | > | } | 
| 90 |  |  | 
| 91 | < | return traits_type::not_eof(c); | 
| 92 | < | } | 
| 91 | > | return traits_type::not_eof(c); | 
| 92 | > | } | 
| 93 |  |  | 
| 94 | < | int sync() { | 
| 94 | > | int sync() { | 
| 95 |  |  | 
| 96 | < | //flush buffer, checking return for eof. | 
| 97 | < | if (traits_type::eq_int_type(overflow(traits_type::eof()), traits_type::eof())) { | 
| 98 | < | return -1; | 
| 99 | < | } | 
| 96 | > | //flush buffer, checking return for eof. | 
| 97 | > | if (traits_type::eq_int_type(overflow(traits_type::eof()), traits_type::eof())) { | 
| 98 | > | return -1; | 
| 99 | > | } | 
| 100 |  |  | 
| 101 | < | //flush streams | 
| 102 | < | typename std::vector<streambuf_type*>::iterator iter; | 
| 103 | < | for (iter = buffers_.begin(); iter != buffers_.end(); ++iter) { | 
| 104 | < | if ((*iter)->pubsync() == -1) { | 
| 105 | < | return -1; | 
| 106 | < | } | 
| 107 | < | } | 
| 101 | > | //flush streams | 
| 102 | > | typename std::vector<streambuf_type*>::iterator iter; | 
| 103 | > | for (iter = buffers_.begin(); iter != buffers_.end(); ++iter) { | 
| 104 | > | if ((*iter)->pubsync() == -1) { | 
| 105 | > | return -1; | 
| 106 | > | } | 
| 107 | > | } | 
| 108 |  |  | 
| 109 | < | return 0; | 
| 110 | < | } | 
| 109 | > | return 0; | 
| 110 | > | } | 
| 111 |  |  | 
| 112 | < | private: | 
| 113 | < | std::vector<streambuf_type*> buffers_; | 
| 114 | < | }; | 
| 112 | > | private: | 
| 113 | > | std::vector<streambuf_type*> buffers_; | 
| 114 | > | }; | 
| 115 |  |  | 
| 116 | < | typedef basic_teebuf<char> TeeBuf; | 
| 116 | > | typedef basic_teebuf<char> TeeBuf; | 
| 117 |  |  | 
| 118 |  | } | 
| 119 |  | #endif |