cprintf.hh revision 2
1/* 2 * Copyright (c) 2003 The Regents of The University of Michigan 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions are 7 * met: redistributions of source code must retain the above copyright 8 * notice, this list of conditions and the following disclaimer; 9 * redistributions in binary form must reproduce the above copyright 10 * notice, this list of conditions and the following disclaimer in the 11 * documentation and/or other materials provided with the distribution; 12 * neither the name of the copyright holders nor the names of its 13 * contributors may be used to endorse or promote products derived from 14 * this software without specific prior written permission. 15 * 16 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 17 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 18 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 19 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 20 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 21 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 22 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 23 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 24 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 26 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 27 */ 28 29#ifndef __CPRINTF_HH__ 30#define __CPRINTF_HH__ 31 32#include <iostream> 33#include <list> 34#include <sstream> 35#include <string> 36 37namespace cp { 38 39#include "cprintf_formats.hh" 40 41class ArgList 42{ 43 private: 44 class Base 45 { 46 public: 47 virtual ~Base() {} 48 virtual void process(std::ostream &out, Format &fmt) = 0; 49 }; 50 51 template <typename T> 52 class Node : public Base 53 { 54 public: 55 const T &data; 56 57 public: 58 Node(const T &d) : data(d) {} 59 virtual void process(std::ostream &out, Format &fmt) { 60 switch (fmt.format) { 61 case Format::character: 62 format_char(out, data, fmt); 63 break; 64 65 case Format::integer: 66 format_integer(out, data, fmt); 67 break; 68 69 case Format::floating: 70 format_float(out, data, fmt); 71 break; 72 73 case Format::string: 74 format_string(out, data, fmt); 75 break; 76 77 default: 78 format_invalid(out); 79 break; 80 } 81 } 82 }; 83 84 typedef std::list<Base *> list_t; 85 86 protected: 87 list_t objects; 88 std::ostream *stream; 89 90 public: 91 ArgList() : stream(&std::cout) {} 92 93 template<class T> 94 void append(const T &data) { 95 Base *obj = new ArgList::Node<T>(data); 96 objects.push_back(obj); 97 } 98 99 template<class T> 100 void prepend(const T &data) { 101 Base *obj = new ArgList::Node<T>(data); 102 objects.push_front(obj); 103 } 104 105 void dump(const std::string &format); 106 void dump(std::ostream &strm, const std::string &fmt) 107 { stream = &strm; dump(fmt); } 108 109 std::string dumpToString(const std::string &format); 110 111 friend ArgList &operator<<(std::ostream &str, ArgList &list); 112}; 113 114template<class T> 115inline ArgList & 116operator,(ArgList &alist, const T &data) 117{ 118 alist.append(data); 119 return alist; 120} 121 122class ArgListNull { 123}; 124 125inline ArgList & 126operator,(ArgList &alist, ArgListNull) 127{ return alist; } 128 129// 130// cprintf(format, args, ...) prints to cout 131// (analogous to printf()) 132// 133inline void 134__cprintf(const std::string &format, ArgList &args) 135{ args.dump(format); delete &args; } 136#define __cprintf__(format, args...) \ 137 cp::__cprintf(format, (*(new cp::ArgList), args)) 138#define cprintf(args...) \ 139 __cprintf__(args, cp::ArgListNull()) 140 141// 142// ccprintf(stream, format, args, ...) prints to the specified stream 143// (analogous to fprintf()) 144// 145inline void 146__ccprintf(std::ostream &stream, const std::string &format, ArgList &args) 147{ args.dump(stream, format); delete &args; } 148#define __ccprintf__(stream, format, args...) \ 149 cp::__ccprintf(stream, format, (*(new cp::ArgList), args)) 150#define ccprintf(stream, args...) \ 151 __ccprintf__(stream, args, cp::ArgListNull()) 152 153// 154// csprintf(format, args, ...) returns a string 155// (roughly analogous to sprintf()) 156// 157inline std::string 158__csprintf(const std::string &format, ArgList &args) 159{ std::string s = args.dumpToString(format); delete &args; return s; } 160#define __csprintf__(format, args...) \ 161 cp::__csprintf(format, (*(new cp::ArgList), args)) 162#define csprintf(args...) \ 163 __csprintf__(args, cp::ArgListNull()) 164 165template<class T> 166inline ArgList & 167operator<<(ArgList &list, const T &data) 168{ 169 list.append(data); 170 return list; 171} 172 173inline ArgList & 174operator<<(std::ostream &str, ArgList &list) 175{ 176 list.stream = &str; 177 return list; 178} 179 180class ArgListTemp 181{ 182 private: 183 std::string format; 184 ArgList *args; 185 186 public: 187 ArgListTemp(const std::string &f) : format(f) { args = new ArgList; } 188 ~ArgListTemp() { args->dump(format); delete args; } 189 190 operator ArgList *() { return args; } 191}; 192 193#define cformat(format) \ 194 (*((cp::ArgList *)cp::ArgListTemp(format))) 195} 196 197#endif // __CPRINTF_HH__ 198