1/*
2 * Copyright (c) 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

--- 12 unchanged lines hidden (view full) ---

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 * Chris Emmons
30 */
31
32#ifndef __BASE_OUTPUT_HH__
33#define __BASE_OUTPUT_HH__
34
35#include <ios>
36#include <map>
37#include <string>
38
39/** Interface for creating files in a gem5 output directory. */
40class OutputDirectory
41{
42 private:
43 /** File names and associated stream handles */
44 typedef std::map<std::string, std::ostream *> map_t;
45
46 /** Open file streams within this directory */
47 map_t files;
48
49 /** Name of this directory */
50 std::string dir;
51
52 /** System-specific path separator character */
53 static const char PATH_SEPARATOR = '/';
54
55 protected:
56 /**
57 * Determines whether given file name corresponds to standard output
58 * streams.
59 *
60 * @param name name of file to check
61 * @return output stream for standard output or error stream if name
62 * corresponds to one or the other; NULL otherwise
63 */
64 std::ostream *checkForStdio(const std::string &name) const;
65
66 public:
67 /** Constructor. */
68 OutputDirectory();
69
70 /** Destructor. */
71 ~OutputDirectory();
72
73 /**
74 * Returns relative file names prepended with name of this directory.
75 * Returns absolute file names unaltered.
76 *
77 * @param name file name to prepend with directory name
78 * @return file name prepended with base directory name or unaltered
79 * absolute file name
80 */
81 std::string resolve(const std::string &name) const;
82
83 /** Opens a file (optionally compressed).
84 *
85 * Will open a file as a compressed stream if filename ends in .gz.
86 *
87 * @param filename file to open
88 * @param mode attributes to open file with
89 * @param no_gz true to disable opening the file as a gzip compressed output
90 * stream; false otherwise
91 * @return stream pointer to opened file; will cause sim fail on error
92 */
93 std::ostream *openFile(const std::string &filename,
94 std::ios_base::openmode mode = std::ios::trunc,
95 bool no_gz = false);
96
97 /**
98 * Sets name of this directory.
99 * @param dir name of this directory
100 */
101 void setDirectory(const std::string &dir);
102
103 /**
104 * Gets name of this directory.
105 * @return name of this directory
106 */
107 const std::string &directory() const;
108
109 /**
110 * Creates a file in this directory (optionally compressed).
111 *
112 * Will open a file as a compressed stream if filename ends in .gz.
113 *
114 * @param name name of file to create (without this directory's name
115 * leading it)
116 * @param binary true to create a binary file; false otherwise
117 * @param no_gz true to disable creating a gzip compressed output stream;
118 * false otherwise
119 * @return stream to the opened file
120 */
121 std::ostream *create(const std::string &name, bool binary = false,
122 bool no_gz = false);
123
124 /**
125 * Closes a file stream.
126 *
127 * Stream must have been opened through this interface, or sim will fail.
128 *
129 * @param openStream open stream to close
130 */
131 void close(std::ostream *openStream);
132
133 /**
134 * Finds stream associated with a file.
135 * @param name of file
136 * @return stream to specified file or NULL if file does not exist
137 */
138 std::ostream *find(const std::string &name) const;
139
140 /**
141 * Returns true if stream is open and not standard output or error.
142 * @param os output stream to evaluate
143 * @return true if os is non-NULL and not cout or cerr
144 */
145 static bool isFile(const std::ostream *os);
146
147 /**
148 * Determines whether a file name corresponds to a file in this directory.
149 * @param name name of file to evaluate
150 * @return true iff file has been opened in this directory or exists on the
151 * file system within this directory
152 */
153 bool isFile(const std::string &name) const;
154
155 /**
156 * Returns true if stream is open and not standard output or error.
157 * @param os output stream to evaluate
158 * @return true if os is non-NULL and not cout or cerr
159 */
160 static inline bool isFile(const std::ostream &os) {
161 return isFile(&os);
162 }
163
164 /**
165 * Creates a subdirectory within this directory.
166 * @param name name of subdirectory
167 * @return the new subdirectory's name suffixed with a path separator
168 */
169 std::string createSubdirectory(const std::string &name) const;
170
171 /**
172 * Removes a specified file or subdirectory.
173 *
174 * Will cause sim to fail for most errors. However, it will only warn the
175 * user if a directory could not be removed. This is in place to
176 * accommodate slow file systems where file deletions within a subdirectory
177 * may not be recognized quickly enough thereby causing the subsequent call

--- 13 unchanged lines hidden ---