Merge zizzer.eecs.umich.edu:/z/m5/Bitkeeper/m5
[gem5.git] / sim / universe.cc
1 /*
2 * Copyright (c) 2002-2004 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 #include <sys/types.h>
30 #include <sys/stat.h>
31
32 #include <cstring>
33 #include <fstream>
34 #include <list>
35 #include <string>
36 #include <vector>
37
38 #include "base/misc.hh"
39 #include "sim/universe.hh"
40 #include "sim/host.hh"
41 #include "sim/param.hh"
42
43 using namespace std;
44
45 Tick curTick = 0;
46 Tick ticksPerSecond;
47 double __ticksPerMS;
48 double __ticksPerUS;
49 double __ticksPerNS;
50 double __ticksPerPS;
51
52 string outputDirectory;
53 ostream *outputStream;
54 ostream *configStream;
55
56 class UniverseParamContext : public ParamContext
57 {
58 public:
59 UniverseParamContext(const string &is) : ParamContext(is) {}
60 void checkParams();
61 };
62
63 UniverseParamContext universe("Universe");
64
65 Param<Tick> universe_freq(&universe, "frequency", "tick frequency",
66 200000000);
67
68 Param<string> universe_output_dir(&universe, "output_dir",
69 "directory to output data to",
70 ".");
71 Param<string> universe_output_file(&universe, "output_file",
72 "file to dump simulator output to",
73 "cout");
74 Param<string> universe_config_output_file(&universe, "config_output_file",
75 "file to dump simulator config to",
76 "m5config.out");
77
78 void
79 UniverseParamContext::checkParams()
80 {
81 ticksPerSecond = universe_freq;
82 double freq = double(ticksPerSecond);
83 __ticksPerMS = freq / 1.0e3;
84 __ticksPerUS = freq / 1.0e6;
85 __ticksPerNS = freq / 1.0e9;
86 __ticksPerPS = freq / 1.0e12;
87
88 if (universe_output_dir.isValid()) {
89 outputDirectory = universe_output_dir;
90
91 // guarantee that directory ends with a '/'
92 if (outputDirectory[outputDirectory.size() - 1] != '/')
93 outputDirectory += "/";
94
95 if (mkdir(outputDirectory.c_str(), 0777) < 0) {
96 if (errno != EEXIST) {
97 panic("%s\ncould not make output directory: %s\n",
98 strerror(errno), outputDirectory);
99 }
100 }
101 }
102
103 outputStream = makeOutputStream(universe_output_file);
104 configStream = universe_config_output_file.isValid()
105 ? makeOutputStream(universe_config_output_file)
106 : outputStream;
107 }
108
109
110 std::ostream *
111 makeOutputStream(std::string &name)
112 {
113 if (name == "cerr" || name == "stderr")
114 return &std::cerr;
115
116 if (name == "cout" || name == "stdout")
117 return &std::cout;
118
119 string path = (name[0] != '/') ? outputDirectory + name : name;
120
121 // have to dynamically allocate a stream since we're going to
122 // return it... though the caller can't easily free it since it
123 // may be cerr or cout. need GC!
124 ofstream *s = new ofstream(path.c_str(), ios::trunc);
125
126 if (!s->is_open())
127 fatal("Cannot open file %s", path);
128
129 return s;
130 }
131
132
133 void
134 closeOutputStream(std::ostream *os)
135 {
136 // can't close cerr or cout
137 if (os == &std::cerr || os == &std::cout)
138 return;
139
140 // can only close ofstreams, not generic ostreams, so try to
141 // downcast and close only if the downcast succeeds
142 std::ofstream *ofs = dynamic_cast<std::ofstream *>(os);
143 if (ofs)
144 ofs->close();
145 }
146
147
148