40e30df7e3
Fixes #534. * bin/common_file.hh, bin/common_file.cc: Make it possible to reopen a closed file. * bin/common_output.cc, bin/common_aoutput.cc: Add a heuristic to decide when to close files. * tests/core/serial.test: Add a test case. * NEWS: Mention the issue.
75 lines
2.1 KiB
C++
75 lines
2.1 KiB
C++
// -*- coding: utf-8 -*-
|
|
// Copyright (C) 2015, 2016, 2022, 2023 Laboratoire de Recherche et
|
|
// Développement de l'Epita (LRDE).
|
|
//
|
|
// This file is part of Spot, a model checking library.
|
|
//
|
|
// Spot is free software; you can redistribute it and/or modify it
|
|
// under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation; either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// Spot is distributed in the hope that it will be useful, but WITHOUT
|
|
// ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
|
|
// or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public
|
|
// License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
#include "common_file.hh"
|
|
#include <error.h>
|
|
#include <iostream>
|
|
|
|
output_file::output_file(const char* name, bool force_append)
|
|
{
|
|
std::ios_base::openmode mode = std::ios_base::trunc;
|
|
if (name[0] == '>' && name[1] == '>')
|
|
{
|
|
append_ = true;
|
|
name += 2;
|
|
}
|
|
if (force_append)
|
|
append_ = true;
|
|
if (append_)
|
|
mode = std::ios_base::app;
|
|
if (name[0] == '-' && name[1] == 0)
|
|
{
|
|
os_ = &std::cout;
|
|
return;
|
|
}
|
|
of_ = std::make_unique<std::ofstream>(name, mode);
|
|
if (!*of_)
|
|
error(2, errno, "cannot open '%s'", name);
|
|
os_ = of_.get();
|
|
}
|
|
|
|
void
|
|
output_file::reopen_for_append(const std::string& name)
|
|
{
|
|
if (of_ && of_->is_open()) // nothing to do
|
|
return;
|
|
const char* cname = name.c_str();
|
|
if (cname[0] == '>' && cname[1] == '>')
|
|
cname += 2;
|
|
if (name[0] == '-' && name[1] == 0)
|
|
{
|
|
os_ = &std::cout;
|
|
return;
|
|
}
|
|
of_->open(cname, std::ios_base::app);
|
|
if (!*of_)
|
|
error(2, errno, "cannot reopen '%s'", cname);
|
|
}
|
|
|
|
void output_file::close(const std::string& name)
|
|
{
|
|
// We close of_, not os_, so that we never close std::cout.
|
|
if (os_)
|
|
os_->flush();
|
|
if (of_ && of_->is_open())
|
|
of_->close();
|
|
if (os_ && !*os_)
|
|
error(2, 0, "error writing to %s",
|
|
(name == "-") ? "standard output" : name.c_str());
|
|
}
|