Introduce a new tool_output_file class, which extends raw_ostream with

functionality that most command-line tools need: ensuring that the
output file gets deleted if the tool is interrupted or encounters an
error.


git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@111595 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Dan Gohman
2010-08-20 00:48:10 +00:00
parent f762fbe4fa
commit af0636f4d6
4 changed files with 75 additions and 1 deletions

View File

@@ -19,10 +19,14 @@
namespace llvm
{
class formatted_tool_output_file;
/// formatted_raw_ostream - Formatted raw_fd_ostream to handle
/// asm-specific constructs.
///
class formatted_raw_ostream : public raw_ostream {
friend class formatted_tool_output_file;
public:
/// DELETE_STREAM - Tell the destructor to delete the held stream.
///
@@ -136,6 +140,25 @@ namespace llvm
}
};
/// formatted_tool_output_file - This is a subclass of formatted_raw_ostream
/// for use when the underlying stream is a tool_output_file. It exposes
/// the keep() member function.
class formatted_tool_output_file : public formatted_raw_ostream {
public:
formatted_tool_output_file(tool_output_file &Stream, bool Delete = false)
: formatted_raw_ostream(Stream, Delete) {}
formatted_tool_output_file() {}
~formatted_tool_output_file();
void setStream(tool_output_file &Stream, bool Delete = false) {
return formatted_raw_ostream::setStream(Stream, Delete);
}
void keep() { return static_cast<tool_output_file *>(TheStream)->keep(); }
};
/// fouts() - This returns a reference to a formatted_raw_ostream for
/// standard output. Use it like: fouts() << "foo" << "bar";
formatted_raw_ostream &fouts();