base/cprintf.hh

   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
  37 namespace cp {
  38
  39 #include "base/cprintf_formats.hh"
  40
  41 class 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
 114 template<class T>
 115 inline ArgList &
 116 operator,(ArgList &alist, const T &data)
 117 {
 118     alist.append(data);
 119     return alist;
 120 }
 121
 122 class ArgListNull {
 123 };
 124
 125 inline ArgList &
 126 operator,(ArgList &alist, ArgListNull)
 127 { return alist; }
 128
 129 //
 130 // cprintf(format, args, ...) prints to cout
 131 // (analogous to printf())
 132 //
 133 inline 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 //
 145 inline 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 //
 157 inline 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
 165 template<class T>
 166 inline ArgList &
 167 operator<<(ArgList &list, const T &data)
 168 {
 169     list.append(data);
 170     return list;
 171 }
 172
 173 inline ArgList &
 174 operator<<(std::ostream &str, ArgList &list)
 175 {
 176     list.stream = &str;
 177     return list;
 178 }
 179
 180 class 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__