Chris Lattner | a2e9363 | 2003-12-31 05:40:02 +0000 | [diff] [blame] | 1 | //===-- SlowOperationInformer.cpp - Keep the user informed ----------------===// |
Misha Brukman | f976c85 | 2005-04-21 22:55:34 +0000 | [diff] [blame] | 2 | // |
Chris Lattner | a2e9363 | 2003-12-31 05:40:02 +0000 | [diff] [blame] | 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
Chris Lattner | 4ee451d | 2007-12-29 20:36:04 +0000 | [diff] [blame^] | 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
Misha Brukman | f976c85 | 2005-04-21 22:55:34 +0000 | [diff] [blame] | 7 | // |
Chris Lattner | a2e9363 | 2003-12-31 05:40:02 +0000 | [diff] [blame] | 8 | //===----------------------------------------------------------------------===// |
Misha Brukman | f976c85 | 2005-04-21 22:55:34 +0000 | [diff] [blame] | 9 | // |
Chris Lattner | a2e9363 | 2003-12-31 05:40:02 +0000 | [diff] [blame] | 10 | // This file implements the SlowOperationInformer class for the LLVM debugger. |
| 11 | // |
| 12 | //===----------------------------------------------------------------------===// |
| 13 | |
Reid Spencer | 551ccae | 2004-09-01 22:55:40 +0000 | [diff] [blame] | 14 | #include "llvm/Support/SlowOperationInformer.h" |
Bill Wendling | fe6b146 | 2006-11-26 10:52:51 +0000 | [diff] [blame] | 15 | #include "llvm/Support/Streams.h" |
Reid Spencer | a6d50d4 | 2005-12-22 03:31:26 +0000 | [diff] [blame] | 16 | #include "llvm/System/Alarm.h" |
Chris Lattner | a2e9363 | 2003-12-31 05:40:02 +0000 | [diff] [blame] | 17 | #include <sstream> |
Chris Lattner | 0a5f033 | 2003-12-31 07:31:10 +0000 | [diff] [blame] | 18 | #include <cassert> |
Chris Lattner | a2e9363 | 2003-12-31 05:40:02 +0000 | [diff] [blame] | 19 | using namespace llvm; |
| 20 | |
Chris Lattner | a2e9363 | 2003-12-31 05:40:02 +0000 | [diff] [blame] | 21 | SlowOperationInformer::SlowOperationInformer(const std::string &Name) |
| 22 | : OperationName(Name), LastPrintAmount(0) { |
Reid Spencer | a6d50d4 | 2005-12-22 03:31:26 +0000 | [diff] [blame] | 23 | sys::SetupAlarm(1); |
Chris Lattner | a2e9363 | 2003-12-31 05:40:02 +0000 | [diff] [blame] | 24 | } |
| 25 | |
| 26 | SlowOperationInformer::~SlowOperationInformer() { |
Reid Spencer | a6d50d4 | 2005-12-22 03:31:26 +0000 | [diff] [blame] | 27 | sys::TerminateAlarm(); |
Chris Lattner | 9e26027 | 2003-12-31 10:20:38 +0000 | [diff] [blame] | 28 | if (LastPrintAmount) { |
| 29 | // If we have printed something, make _sure_ we print the 100% amount, and |
| 30 | // also print a newline. |
Bill Wendling | e815619 | 2006-12-07 01:30:32 +0000 | [diff] [blame] | 31 | cout << std::string(LastPrintAmount, '\b') << "Progress " |
| 32 | << OperationName << ": 100% \n"; |
Chris Lattner | 9e26027 | 2003-12-31 10:20:38 +0000 | [diff] [blame] | 33 | } |
Chris Lattner | a2e9363 | 2003-12-31 05:40:02 +0000 | [diff] [blame] | 34 | } |
| 35 | |
| 36 | /// progress - Clients should periodically call this method when they are in |
| 37 | /// an exception-safe state. The Amount variable should indicate how far |
| 38 | /// along the operation is, given in 1/10ths of a percent (in other words, |
| 39 | /// Amount should range from 0 to 1000). |
Chris Lattner | 148f440 | 2006-07-06 22:34:06 +0000 | [diff] [blame] | 40 | bool SlowOperationInformer::progress(unsigned Amount) { |
Reid Spencer | a6d50d4 | 2005-12-22 03:31:26 +0000 | [diff] [blame] | 41 | int status = sys::AlarmStatus(); |
| 42 | if (status == -1) { |
Bill Wendling | e815619 | 2006-12-07 01:30:32 +0000 | [diff] [blame] | 43 | cout << "\n"; |
Chris Lattner | a2e9363 | 2003-12-31 05:40:02 +0000 | [diff] [blame] | 44 | LastPrintAmount = 0; |
Chris Lattner | 148f440 | 2006-07-06 22:34:06 +0000 | [diff] [blame] | 45 | return true; |
Chris Lattner | a2e9363 | 2003-12-31 05:40:02 +0000 | [diff] [blame] | 46 | } |
| 47 | |
| 48 | // If we haven't spent enough time in this operation to warrant displaying the |
| 49 | // progress bar, don't do so yet. |
Reid Spencer | a6d50d4 | 2005-12-22 03:31:26 +0000 | [diff] [blame] | 50 | if (status == 0) |
Chris Lattner | 148f440 | 2006-07-06 22:34:06 +0000 | [diff] [blame] | 51 | return false; |
Chris Lattner | a2e9363 | 2003-12-31 05:40:02 +0000 | [diff] [blame] | 52 | |
| 53 | // Delete whatever we printed last time. |
| 54 | std::string ToPrint = std::string(LastPrintAmount, '\b'); |
| 55 | |
| 56 | std::ostringstream OS; |
Chris Lattner | 9e26027 | 2003-12-31 10:20:38 +0000 | [diff] [blame] | 57 | OS << "Progress " << OperationName << ": " << Amount/10; |
| 58 | if (unsigned Rem = Amount % 10) |
| 59 | OS << "." << Rem << "%"; |
| 60 | else |
| 61 | OS << "% "; |
Chris Lattner | a2e9363 | 2003-12-31 05:40:02 +0000 | [diff] [blame] | 62 | |
| 63 | LastPrintAmount = OS.str().size(); |
Bill Wendling | e815619 | 2006-12-07 01:30:32 +0000 | [diff] [blame] | 64 | cout << ToPrint+OS.str() << std::flush; |
Chris Lattner | 148f440 | 2006-07-06 22:34:06 +0000 | [diff] [blame] | 65 | return false; |
Chris Lattner | a2e9363 | 2003-12-31 05:40:02 +0000 | [diff] [blame] | 66 | } |