X-Git-Url: http://demsky.eecs.uci.edu/git/?a=blobdiff_plain;f=lib%2FSupport%2FSlowOperationInformer.cpp;h=b4e9430e5fdfeea8555e1165a15e4b847280c636;hb=4459145c2ccb5d063841a5d8c76b8b8ac9adaf2f;hp=c245f3655c2a289246ba1a55f48634a408acbae2;hpb=2657aecfe273b9dfbd4fff77a3ed908197dcd30a;p=oota-llvm.git

diff --git a/lib/Support/SlowOperationInformer.cpp b/lib/Support/SlowOperationInformer.cpp
index c245f3655c2..b4e9430e5fd 100644
--- a/lib/Support/SlowOperationInformer.cpp
+++ b/lib/Support/SlowOperationInformer.cpp
@@ -1,91 +1,54 @@
 //===-- 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 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 "Support/SlowOperationInformer.h"
-#include "Config/config.h"     // Get the signal handler return type
-#include <iostream>
+#include "llvm/Support/SlowOperationInformer.h"
+#include "llvm/Support/raw_ostream.h"
+#include "llvm/System/Alarm.h"
 #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);
+  sys::SetupAlarm(1);
 }
 
 SlowOperationInformer::~SlowOperationInformer() {
-  NestedSOI = false;
+  sys::TerminateAlarm();
   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";
+    outs() << 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";
+bool SlowOperationInformer::progress(unsigned Amount) {
+  int status = sys::AlarmStatus();
+  if (status == -1) {
+    outs() << "\n";
     LastPrintAmount = 0;
-    throw "While " + OperationName + ", operation cancelled.";
+    return true;
   }
 
   // If we haven't spent enough time in this operation to warrant displaying the
   // progress bar, don't do so yet.
-  if (!ShouldShowStatus)
-    return;
+  if (status == 0)
+    return false;
 
   // Delete whatever we printed last time.
   std::string ToPrint = std::string(LastPrintAmount, '\b');
@@ -98,5 +61,7 @@ void SlowOperationInformer::progress(unsigned Amount) {
     OS << "%  ";
 
   LastPrintAmount = OS.str().size();
-  std::cout << ToPrint+OS.str() << std::flush;
+  outs() << ToPrint+OS.str();
+  outs().flush();
+  return false;
 }