src/base/output.cc

   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
  42 using namespace std;
  43
  44 OutputDirectory simout;
  45
  46 /**
  47  *
  48  */
  49 OutputDirectory::OutputDirectory()
  50 {}
  51
  52 OutputDirectory::~OutputDirectory()
  53 {}
  54
  55 void
  56 OutputDirectory::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
  74 const string &
  75 OutputDirectory::directory()
  76 {
  77     if (dir.empty())
  78         panic("Output directory not set!");
  79
  80     return dir;
  81 }
  82
  83 string
  84 OutputDirectory::resolve(const string &name)
  85 {
  86     return (name[0] != '/') ? dir + name : name;
  87 }
  88
  89 ostream *
  90 OutputDirectory::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
 105 ostream *
 106 OutputDirectory::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
 127 bool
 128 OutputDirectory::isFile(const std::ostream *os)
 129 {
 130     return os && os != &cerr && os != &cout;
 131 }