我想在程序终止之前用 ctrl-c 中断时向文件中写入几行额外的行。但是,文件的位置不是硬编码的,因此我需要的不仅仅是正常的中断处理,如此处所述。做这个的最好方式是什么?
动机:
我正在做时间相关的有限元模拟。有时我忘记或错误估计了一个合理的 tmax,模拟需要很长时间才能完成,所以我用 ctrl-c 中断它。但是我仍然想使用到目前为止生成的数据。特别是有一个 xml 文件需要在模拟完成后附加几行来关闭标签。当然可以手动完成,但很痛苦,所以我尝试将其自动化。
基于对“如何捕获 ctrl-c 事件?(C++) ”的回答,我们可以安装一个简单地引发异常的信号处理程序。然后我们可以捕获异常并运行我们喜欢的任何标准 C++ 代码。
这是一个工作示例:
#include <iostream>
#include <exception>
#include <signal.h>
#include <stdlib.h>
class InterruptException : public std::exception
{
public:
InterruptException(int s) : S(s) {}
int S;
};
void sig_to_exception(int s)
{
throw InterruptException(s);
}
int main()
{
// Taken from answer to "How can I catch a ctrl-c event? (C++)"
struct sigaction sigIntHandler;
sigIntHandler.sa_handler = sig_to_exception;
sigemptyset(&sigIntHandler.sa_mask);
sigIntHandler.sa_flags = 0;
sigaction(SIGINT, &sigIntHandler, NULL);
try
{
std::cout << "Inside try" << std::endl;
// Do something with output so the loop doesn't get optimised out
double i = 0;
while(i < 1e30) {i++;} // loop until interrupted
std::cout << i << std::endl;
}
catch(InterruptException& e)
{
std::cout << "Write something to file" << std::endl;
std::cout << "Caught signal " << e.S << std::endl;
return 1;
}
return 0;
}
这种方法至少有一个缺点:我们必须在处理程序安装后包装所有内容,否则中断信号会导致 abort()。
基于循环的科学代码的一种常见方案是使用全局 volatile sig_atomic_t 布尔值来指示信号是否被捕获并将其添加到循环条件中。
例如
volatile sig_atomic_t interrupted=false;
...
void signal_handler(int s)
{
// ...
interrupted=true;
}
...
while (!converged && !interrupted)
{
// Perform computations
}
// Save file properly