output.cc revision 2665
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 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 */ 30 31#include <errno.h> 32#include <limits.h> 33#include <stdlib.h> 34#include <sys/stat.h> 35#include <sys/types.h> 36 37#include <fstream> 38 39#include "base/misc.hh" 40#include "base/output.hh" 41 42using namespace std; 43 44OutputDirectory simout; 45 46/** 47 * 48 */ 49OutputDirectory::OutputDirectory() 50{} 51 52OutputDirectory::~OutputDirectory() 53{} 54 55void 56OutputDirectory::setDirectory(const string &d) 57{ 58 if (!dir.empty()) 59 panic("Output directory already set!\n"); 60 61 dir = d; 62 63 if (dir != ".") { 64 if (mkdir(dir.c_str(), 0777) < 0 && errno != EEXIST) 65 panic("couldn't make output dir %s: %s\n", 66 dir, strerror(errno)); 67 } 68 69 // guarantee that directory ends with a '/' 70 if (dir[dir.size() - 1] != '/') 71 dir += "/"; 72} 73 74const string & 75OutputDirectory::directory() 76{ 77 if (dir.empty()) 78 panic("Output directory not set!"); 79 80 return dir; 81} 82 83string 84OutputDirectory::resolve(const string &name) 85{ 86 return (name[0] != '/') ? dir + name : name; 87} 88 89ostream * 90OutputDirectory::create(const string &name) 91{ 92 if (name == "cerr" || name == "stderr") 93 return &cerr; 94 95 if (name == "cout" || name == "stdout") 96 return &cout; 97 98 ofstream *file = new ofstream(resolve(name).c_str(), ios::trunc); 99 if (!file->is_open()) 100 panic("Cannot open file %s", name); 101 102 return file; 103} 104 105ostream * 106OutputDirectory::find(const string &name) 107{ 108 if (name == "cerr" || name == "stderr") 109 return &cerr; 110 111 if (name == "cout" || name == "stdout") 112 return &cout; 113 114 string filename = resolve(name); 115 map_t::iterator i = files.find(filename); 116 if (i != files.end()) 117 return (*i).second; 118 119 ofstream *file = new ofstream(filename.c_str(), ios::trunc); 120 if (!file->is_open()) 121 panic("Cannot open file %s", filename); 122 123 files[filename] = file; 124 return file; 125} 126 127bool 128OutputDirectory::isFile(const std::ostream *os) 129{ 130 return os && os != &cerr && os != &cout; 131} 132