apt-pkg/contrib/progress.cc

   1 // -*- mode: cpp; mode: fold -*-
   2 // Description                                                          /*{{{*/
   3 // $Id: progress.cc,v 1.6 1998/08/26 04:52:28 jgg Exp $
   4 /* ######################################################################
   5
   6    OpProgress - Operation Progress
   7
   8    ##################################################################### */
   9                                                                         /*}}}*/
  10 // Include Files                                                        /*{{{*/
  11 #ifdef __GNUG__
  12 #pragma implementation "apt-pkg/progress.h"
  13 #endif
  14 #include <apt-pkg/progress.h>
  15 #include <apt-pkg/error.h>
  16 #include <stdio.h>
  17                                                                         /*}}}*/
  18
  19 // OpProgress::OpProgress - Constructor                                 /*{{{*/
  20 // ---------------------------------------------------------------------
  21 /* */
  22 OpProgress::OpProgress() : Current(0), Total(0), Size(0), SubTotal(1),
  23                            LastPercent(0), Percent(0)
  24 {
  25    memset(&LastTime,0,sizeof(LastTime));
  26 }
  27                                                                         /*}}}*/
  28 // OpProgress::Progress - Sub progress with no state change             /*{{{*/
  29 // ---------------------------------------------------------------------
  30 /* This assumes that Size is the same as the current sub size */
  31 void OpProgress::Progress(unsigned long Cur)
  32 {
  33    Percent = (Current + Cur/((float)SubTotal)*Size)*100.0/Total;
  34    Update();
  35 }
  36                                                                         /*}}}*/
  37 // OpProgress::OverallProgress - Set the overall progress               /*{{{*/
  38 // ---------------------------------------------------------------------
  39 /* */
  40 void OpProgress::OverallProgress(unsigned long Current, unsigned long Total,
  41                                  unsigned long Size,string Op)
  42 {
  43    this->Current = Current;
  44    this->Total = Total;
  45    this->Size = Size;
  46    this->Op = Op;
  47    SubOp = string();
  48    Percent = Current*100.0/Total;
  49    Update();
  50 }
  51                                                                         /*}}}*/
  52 // OpProgress::SubProgress - Set the sub progress state                 /*{{{*/
  53 // ---------------------------------------------------------------------
  54 /* */
  55 void OpProgress::SubProgress(unsigned long SubTotal,string Op)
  56 {
  57    this->SubTotal = SubTotal;
  58    SubOp = Op;
  59    Percent = Current*100.0/Total;
  60    Update();
  61 }
  62                                                                         /*}}}*/
  63 // OpProgress::SubProgress - Set the sub progress state                 /*{{{*/
  64 // ---------------------------------------------------------------------
  65 /* */
  66 void OpProgress::SubProgress(unsigned long SubTotal)
  67 {
  68    this->SubTotal = SubTotal;
  69    Percent = Current*100.0/Total;
  70    Update();
  71 }
  72                                                                         /*}}}*/
  73 // OpProgress::CheckChange - See if the display should be updated       /*{{{*/
  74 // ---------------------------------------------------------------------
  75 /* Progress calls are made so frequently that if every one resulted in
  76    an update the display would be swamped and the system much slower.
  77    This provides an upper bound on the update rate. */
  78 bool OpProgress::CheckChange(float Interval)
  79 {
  80    // New major progress indication
  81    if (Op != LastOp)
  82    {
  83       MajorChange = true;
  84       LastOp = Op;
  85       return true;
  86    }
  87    MajorChange = false;
  88
  89    if (SubOp != LastSubOp)
  90    {
  91       LastSubOp = SubOp;
  92       return true;
  93    }
  94
  95    if ((int)LastPercent == (int)Percent)
  96       return false;
  97
  98    // Check time delta
  99    struct timeval Now;
 100    gettimeofday(&Now,0);
 101    double Diff = Now.tv_sec - LastTime.tv_sec + (Now.tv_usec - LastTime.tv_usec)/1000000.0;
 102    if (Diff < Interval)
 103       return false;
 104    LastTime = Now;
 105    LastPercent = Percent;
 106    return true;
 107 }
 108                                                                         /*}}}*/
 109 // OpTextProgress::Done - Clean up the display                          /*{{{*/
 110 // ---------------------------------------------------------------------
 111 /* */
 112 void OpTextProgress::Done()
 113 {
 114    if (NoUpdate == false && OldOp.empty() == false)
 115    {
 116       char S[300];
 117       if (_error->PendingError() == true)
 118          snprintf(S,sizeof(S),"\r%s... Error!",OldOp.c_str());
 119       else
 120          snprintf(S,sizeof(S),"\r%s... Done",OldOp.c_str());
 121       Write(S);
 122       cout << endl;
 123       OldOp = string();
 124    }
 125 }
 126                                                                         /*}}}*/
 127 // OpTextProgress::Update - Simple text spinner                         /*{{{*/
 128 // ---------------------------------------------------------------------
 129 /* */
 130 void OpTextProgress::Update()
 131 {
 132    if (CheckChange() == false)
 133       return;
 134
 135    // No percent spinner
 136    if (NoUpdate == true)
 137    {
 138       if (MajorChange == false)
 139          return;
 140       cout << Op << endl;
 141       return;
 142    }
 143
 144    // Erase the old text and 'log' the event
 145    char S[300];
 146    if (MajorChange == true && OldOp.empty() == false)
 147    {
 148       snprintf(S,sizeof(S),"\r%s",OldOp.c_str());
 149       Write(S);
 150       cout << endl;
 151    }
 152
 153    // Print the spinner
 154    snprintf(S,sizeof(S),"\r%s... %u%%",Op.c_str(),(unsigned int)Percent);
 155    Write(S);
 156
 157    OldOp = Op;
 158 }
 159                                                                         /*}}}*/
 160 // OpTextProgress::Write - Write the progress string                    /*{{{*/
 161 // ---------------------------------------------------------------------
 162 /* This space fills the end to overwrite the previous text */
 163 void OpTextProgress::Write(const char *S)
 164 {
 165    cout << S;
 166    for (unsigned int I = strlen(S); I < LastLen; I++)
 167       cout << ' ';
 168    cout << '\r' << flush;
 169    LastLen = strlen(S);
 170 }
 171                                                                         /*}}}*/