output.cc (5524:e5fbd38bc828) | output.cc (5749:7015e400bd1d) |
---|---|
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; --- 22 unchanged lines hidden (view full) --- 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 | 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; --- 22 unchanged lines hidden (view full) --- 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 <gzstream.hh> 40 |
|
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() | 41#include "base/misc.hh" 42#include "base/output.hh" 43 44using namespace std; 45 46OutputDirectory simout; 47 48/** 49 * 50 */ 51OutputDirectory::OutputDirectory() 52{} 53 54OutputDirectory::~OutputDirectory() |
53{} | 55{ 56 for (map_t::iterator i = files.begin(); i != files.end(); i++) { 57 if (i->second) 58 delete i->second; 59 } 60} |
54 | 61 |
62std::ostream * 63OutputDirectory::checkForStdio(const string &name) const 64{ 65 if (name == "cerr" || name == "stderr") 66 return &cerr; 67 68 if (name == "cout" || name == "stdout") 69 return &cout; 70 71 return NULL; 72} 73 74ostream * 75OutputDirectory::openFile(const string &filename, 76 ios_base::openmode mode) const 77{ 78 if (filename.find(".gz", filename.length()-3) < filename.length()) { 79 ogzstream *file = new ogzstream(filename.c_str(), mode); 80 81 if (!file->is_open()) 82 fatal("Cannot open file %s", filename); 83 84 return file; 85 } else { 86 ofstream *file = new ofstream(filename.c_str(), mode); 87 88 if (!file->is_open()) 89 fatal("Cannot open file %s", filename); 90 91 return file; 92 } 93} 94 |
|
55void 56OutputDirectory::setDirectory(const string &d) 57{ 58 if (!dir.empty()) 59 panic("Output directory already set!\n"); 60 61 dir = d; 62 63 // guarantee that directory ends with a '/' 64 if (dir[dir.size() - 1] != '/') 65 dir += "/"; 66} 67 68const string & | 95void 96OutputDirectory::setDirectory(const string &d) 97{ 98 if (!dir.empty()) 99 panic("Output directory already set!\n"); 100 101 dir = d; 102 103 // guarantee that directory ends with a '/' 104 if (dir[dir.size() - 1] != '/') 105 dir += "/"; 106} 107 108const string & |
69OutputDirectory::directory() | 109OutputDirectory::directory() const |
70{ 71 if (dir.empty()) 72 panic("Output directory not set!"); 73 74 return dir; 75} 76 | 110{ 111 if (dir.empty()) 112 panic("Output directory not set!"); 113 114 return dir; 115} 116 |
77string 78OutputDirectory::resolve(const string &name) | 117inline string 118OutputDirectory::resolve(const string &name) const |
79{ 80 return (name[0] != '/') ? dir + name : name; 81} 82 83ostream * 84OutputDirectory::create(const string &name, bool binary) 85{ | 119{ 120 return (name[0] != '/') ? dir + name : name; 121} 122 123ostream * 124OutputDirectory::create(const string &name, bool binary) 125{ |
86 if (name == "cerr" || name == "stderr") 87 return &cerr; | 126 ostream *file = checkForStdio(name); 127 if (file) 128 return file; |
88 | 129 |
89 if (name == "cout" || name == "stdout") 90 return &cout; | 130 string filename = resolve(name); 131 ios_base::openmode mode = 132 ios::trunc | binary ? ios::binary : (ios::openmode)0; 133 file = openFile(filename, mode); |
91 | 134 |
92 ofstream *file = new ofstream(resolve(name).c_str(), 93 ios::trunc | binary ? ios::binary : (ios::openmode)0); 94 if (!file->is_open()) 95 fatal("Cannot open file %s", name); 96 | |
97 return file; 98} 99 100ostream * 101OutputDirectory::find(const string &name) 102{ | 135 return file; 136} 137 138ostream * 139OutputDirectory::find(const string &name) 140{ |
103 if (name == "cerr" || name == "stderr") 104 return &cerr; | 141 ostream *file = checkForStdio(name); 142 if (file) 143 return file; |
105 | 144 |
106 if (name == "cout" || name == "stdout") 107 return &cout; 108 | |
109 string filename = resolve(name); 110 map_t::iterator i = files.find(filename); 111 if (i != files.end()) 112 return (*i).second; 113 | 145 string filename = resolve(name); 146 map_t::iterator i = files.find(filename); 147 if (i != files.end()) 148 return (*i).second; 149 |
114 ofstream *file = new ofstream(filename.c_str(), ios::trunc); 115 if (!file->is_open()) 116 fatal("Cannot open file %s", filename); 117 | 150 file = openFile(filename); |
118 files[filename] = file; 119 return file; 120} 121 122bool 123OutputDirectory::isFile(const std::ostream *os) 124{ 125 return os && os != &cerr && os != &cout; 126} | 151 files[filename] = file; 152 return file; 153} 154 155bool 156OutputDirectory::isFile(const std::ostream *os) 157{ 158 return os && os != &cerr && os != &cout; 159} |