cprintf.hh revision 2665:a124942bacb8 
1/*
2 * Copyright (c) 2002-2005 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 * Authors: Nathan Binkert
29 *          Steve Reinhardt
30 */
31
32#ifndef __CPRINTF_HH__
33#define __CPRINTF_HH__
34
35#include <iostream>
36#include <list>
37#include <string>
38
39#include "base/cprintf_formats.hh"
40
41namespace cp {
42
43class ArgList
44{
45  private:
46    class Base
47    {
48      public:
49        virtual ~Base() {}
50        virtual void process(std::ostream &out, Format &fmt) = 0;
51    };
52
53    template <typename T>
54    class Node : public Base
55    {
56      public:
57        const T &data;
58
59      public:
60        Node(const T &d) : data(d) {}
61        virtual void process(std::ostream &out, Format &fmt) {
62            switch (fmt.format) {
63              case Format::character:
64                format_char(out, data, fmt);
65                break;
66
67              case Format::integer:
68                format_integer(out, data, fmt);
69                break;
70
71              case Format::floating:
72                format_float(out, data, fmt);
73                break;
74
75              case Format::string:
76                format_string(out, data, fmt);
77                break;
78
79              default:
80                out << "<bad format>";
81                break;
82            }
83        }
84    };
85
86    typedef std::list<Base *> list_t;
87
88  protected:
89    list_t objects;
90    std::ostream *stream;
91
92  public:
93    ArgList() : stream(&std::cout) {}
94    ~ArgList();
95
96    template<class T>
97    void append(const T &data) {
98        Base *obj = new ArgList::Node<T>(data);
99        objects.push_back(obj);
100    }
101
102    template<class T>
103    void prepend(const T &data) {
104        Base *obj = new ArgList::Node<T>(data);
105        objects.push_front(obj);
106    }
107
108    void dump(const std::string &format);
109    void dump(std::ostream &strm, const std::string &fmt)
110        { stream = &strm; dump(fmt); }
111
112    std::string dumpToString(const std::string &format);
113
114    friend ArgList &operator<<(std::ostream &str, ArgList &list);
115};
116
117template<class T>
118inline ArgList &
119operator,(ArgList &alist, const T &data)
120{
121    alist.append(data);
122    return alist;
123}
124
125class ArgListNull {
126};
127
128inline ArgList &
129operator,(ArgList &alist, ArgListNull)
130{ return alist; }
131
132//
133// cprintf(format, args, ...) prints to cout
134// (analogous to printf())
135//
136inline void
137__cprintf(const std::string &format, ArgList &args)
138{ args.dump(format); delete &args; }
139#define __cprintf__(format, args...) \
140    cp::__cprintf(format, (*(new cp::ArgList), args))
141#define cprintf(args...) \
142    __cprintf__(args, cp::ArgListNull())
143
144//
145// ccprintf(stream, format, args, ...) prints to the specified stream
146// (analogous to fprintf())
147//
148inline void
149__ccprintf(std::ostream &stream, const std::string &format, ArgList &args)
150{ args.dump(stream, format); delete &args; }
151#define __ccprintf__(stream, format, args...) \
152    cp::__ccprintf(stream, format, (*(new cp::ArgList), args))
153#define ccprintf(stream, args...) \
154    __ccprintf__(stream, args, cp::ArgListNull())
155
156//
157// csprintf(format, args, ...) returns a string
158// (roughly analogous to sprintf())
159//
160inline std::string
161__csprintf(const std::string &format, ArgList &args)
162{ std::string s = args.dumpToString(format); delete &args; return s; }
163#define __csprintf__(format, args...) \
164    cp::__csprintf(format, (*(new cp::ArgList), args))
165#define csprintf(args...) \
166    __csprintf__(args, cp::ArgListNull())
167
168template<class T>
169inline ArgList &
170operator<<(ArgList &list, const T &data)
171{
172    list.append(data);
173    return list;
174}
175
176inline ArgList &
177operator<<(std::ostream &str, ArgList &list)
178{
179    list.stream = &str;
180    return list;
181}
182
183class ArgListTemp
184{
185  private:
186    std::string format;
187    ArgList *args;
188
189  public:
190    ArgListTemp(const std::string &f) : format(f) { args = new ArgList; }
191    ~ArgListTemp() { args->dump(format); delete args; }
192
193    operator ArgList *() { return args; }
194};
195
196#define cformat(format) \
197    (*((cp::ArgList *)cp::ArgListTemp(format)))
198}
199
200#endif // __CPRINTF_HH__
201