103 lines
3.1 KiB
C++
103 lines
3.1 KiB
C++
//===-- SlowOperationInformer.cpp - Keep the user informed ----------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file was developed by the LLVM research group and is distributed under
|
|
// the University of Illinois Open Source License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file implements the SlowOperationInformer class for the LLVM debugger.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/Support/SlowOperationInformer.h"
|
|
#include "llvm/Config/config.h" // Get the signal handler return type
|
|
#include <iostream>
|
|
#include <sstream>
|
|
#include <signal.h>
|
|
#include <unistd.h>
|
|
#include <cassert>
|
|
using namespace llvm;
|
|
|
|
/// OperationCancelled - This flag is set by the SIGINT signal handler if the
|
|
/// user presses CTRL-C.
|
|
static volatile bool OperationCancelled;
|
|
|
|
/// ShouldShowStatus - This flag gets set if the operation takes a long time.
|
|
///
|
|
static volatile bool ShouldShowStatus;
|
|
|
|
/// NestedSOI - Sanity check. SlowOperationInformers cannot be nested or run in
|
|
/// parallel. This ensures that they never do.
|
|
static bool NestedSOI = false;
|
|
|
|
static RETSIGTYPE SigIntHandler(int Sig) {
|
|
OperationCancelled = true;
|
|
signal(SIGINT, SigIntHandler);
|
|
}
|
|
|
|
static RETSIGTYPE SigAlarmHandler(int Sig) {
|
|
ShouldShowStatus = true;
|
|
}
|
|
|
|
static void (*OldSigIntHandler) (int);
|
|
|
|
|
|
SlowOperationInformer::SlowOperationInformer(const std::string &Name)
|
|
: OperationName(Name), LastPrintAmount(0) {
|
|
assert(!NestedSOI && "SlowerOperationInformer objects cannot be nested!");
|
|
NestedSOI = true;
|
|
|
|
OperationCancelled = 0;
|
|
ShouldShowStatus = 0;
|
|
|
|
signal(SIGALRM, SigAlarmHandler);
|
|
OldSigIntHandler = signal(SIGINT, SigIntHandler);
|
|
alarm(1);
|
|
}
|
|
|
|
SlowOperationInformer::~SlowOperationInformer() {
|
|
NestedSOI = false;
|
|
if (LastPrintAmount) {
|
|
// If we have printed something, make _sure_ we print the 100% amount, and
|
|
// also print a newline.
|
|
std::cout << std::string(LastPrintAmount, '\b') << "Progress "
|
|
<< OperationName << ": 100% \n";
|
|
}
|
|
|
|
alarm(0);
|
|
signal(SIGALRM, SIG_DFL);
|
|
signal(SIGINT, OldSigIntHandler);
|
|
}
|
|
|
|
/// progress - Clients should periodically call this method when they are in
|
|
/// an exception-safe state. The Amount variable should indicate how far
|
|
/// along the operation is, given in 1/10ths of a percent (in other words,
|
|
/// Amount should range from 0 to 1000).
|
|
void SlowOperationInformer::progress(unsigned Amount) {
|
|
if (OperationCancelled) {
|
|
std::cout << "\n";
|
|
LastPrintAmount = 0;
|
|
throw "While " + OperationName + ", operation cancelled.";
|
|
}
|
|
|
|
// If we haven't spent enough time in this operation to warrant displaying the
|
|
// progress bar, don't do so yet.
|
|
if (!ShouldShowStatus)
|
|
return;
|
|
|
|
// Delete whatever we printed last time.
|
|
std::string ToPrint = std::string(LastPrintAmount, '\b');
|
|
|
|
std::ostringstream OS;
|
|
OS << "Progress " << OperationName << ": " << Amount/10;
|
|
if (unsigned Rem = Amount % 10)
|
|
OS << "." << Rem << "%";
|
|
else
|
|
OS << "% ";
|
|
|
|
LastPrintAmount = OS.str().size();
|
|
std::cout << ToPrint+OS.str() << std::flush;
|
|
}
|