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