Browse Source

Removal of boost progress bar, replaced with simple RAII class

master
Trevor Irons 4 years ago
parent
commit
f4fb29b94b

+ 0
- 4
CMakeLists.txt View File

@@ -244,10 +244,6 @@ if (LEMMA_USE_BOOST)
244 244
 			add_definitions(-DHAVE_BOOST_SPECIAL_FUNCTIONS)
245 245
 		endif()
246 246
 	endif()
247
-	find_path( HAVE_BOOST_PROGRESS "boost/progress.hpp" ) 
248
-		if(HAVE_BOOST_PROGRESS)
249
-		add_definitions(-DHAVE_BOOST_PROGRESS)
250
-	endif()
251 247
 endif()
252 248
 
253 249
 if (LEMMA_PYTHON3_BINDINGS)

+ 1
- 3
Modules/FDEM1D/include/EMEarth1D.h View File

@@ -33,9 +33,7 @@
33 33
 #include "QWEKey.h"
34 34
 #include "CubicSplineInterpolator.h"
35 35
 
36
-#ifdef HAVE_BOOST_PROGRESS
37
-#include "boost/progress.hpp"
38
-#endif
36
+#include "ProgressBar.h"
39 37
 
40 38
 namespace Lemma {
41 39
 

+ 14
- 26
Modules/FDEM1D/src/EMEarth1D.cpp View File

@@ -73,9 +73,6 @@ namespace Lemma {
73 73
     EMEarth1D::EMEarth1D( const ctor_key& key ) : LemmaObject( key ),
74 74
             Dipole(nullptr), Earth(nullptr), Receivers(nullptr), Antenna(nullptr),
75 75
             FieldsToCalculate(BOTH), HankelType(ANDERSON801), icalcinner(0), icalc(0)
76
-        //#ifdef HAVE_BOOST_PROGRESS
77
-        //    , disp(0)
78
-        //#endif
79 76
         {
80 77
     }
81 78
 
@@ -191,9 +188,7 @@ namespace Lemma {
191 188
 
192 189
     void EMEarth1D::CalculateWireAntennaFields(bool progressbar) {
193 190
 
194
-        #ifdef HAVE_BOOST_PROGRESS
195
-        boost::progress_display *disp;
196
-        #endif
191
+        ProgressBar *mdisp;
197 192
 
198 193
         if (Earth == nullptr) {
199 194
             throw NullEarth();
@@ -234,11 +229,10 @@ namespace Lemma {
234 229
             if ( Antenna->IsHorizontallyPlanar() && ( HankelType == ANDERSON801 || HankelType == FHTKEY201  || HankelType==FHTKEY101 ||
235 230
                                                       HankelType == FHTKEY51    || HankelType == FHTKONG61  || HankelType == FHTKONG121 ||
236 231
                                                       HankelType == FHTKONG241  || HankelType == IRONS )) {
237
-                #ifdef HAVE_BOOST_PROGRESS
238 232
                 if (progressbar) {
239
-                    disp = new boost::progress_display( Receivers->GetNumberOfPoints()*Antenna->GetNumberOfFrequencies() );
233
+                    mdisp = new ProgressBar( Receivers->GetNumberOfPoints()*Antenna->GetNumberOfFrequencies(), "Progress");
240 234
                 }
241
-                #endif
235
+
242 236
                 for (int ifreq=0; ifreq<Antenna->GetNumberOfFrequencies();++ifreq) {
243 237
                     Real wavef = 2.*PI* Antenna->GetFrequency(ifreq);
244 238
                     #ifdef LEMMAUSEOMP
@@ -252,9 +246,9 @@ namespace Lemma {
252 246
                     #endif
253 247
                     for (int irec=0; irec<Receivers->GetNumberOfPoints(); ++irec) {
254 248
                         SolveLaggedTxRxPair(irec, Hankel.get(), wavef, ifreq, AntCopy.get());
255
-                        #ifdef HAVE_BOOST_PROGRESS
256
-                        if (progressbar) ++(*disp);
257
-                        #endif
249
+                        if (progressbar) {
250
+                            ++(*mdisp);
251
+                        }
258 252
                     }
259 253
                     #pragma omp barrier
260 254
                     #ifdef LEMMAUSEOMP
@@ -265,11 +259,10 @@ namespace Lemma {
265 259
 
266 260
                 //std::cout << "freq parallel #1" << std::endl;
267 261
                 //** Progress display bar for long calculations */
268
-                #ifdef HAVE_BOOST_PROGRESS
269 262
                 if (progressbar) {
270
-                    disp = new boost::progress_display( Receivers->GetNumberOfPoints()*Antenna->GetNumberOfFrequencies() );
263
+                    mdisp = new ProgressBar( Receivers->GetNumberOfPoints()*Antenna->GetNumberOfFrequencies() );
271 264
                 }
272
-                #endif
265
+
273 266
                 // parallelise across receivers
274 267
                 #ifdef LEMMAUSEOMP
275 268
                 #pragma omp parallel
@@ -336,9 +329,9 @@ namespace Lemma {
336 329
                         //std::cout << "Normal Path\n";
337 330
                         //std::cout << Receivers->GetHfield(0, irec) << std::endl;
338 331
                         //if (irec == 1) exit(0);
339
-                        #ifdef HAVE_BOOST_PROGRESS
340
-                        if (progressbar) ++(*disp);
341
-                        #endif
332
+                        if (progressbar) {
333
+                            ++(*mdisp);
334
+                        }
342 335
                     } // receiver loop
343 336
                 } // OMP_PARALLEL BLOCK
344 337
             } else if (Antenna->GetNumberOfFrequencies() > 8) {
@@ -390,11 +383,9 @@ namespace Lemma {
390 383
                             } // frequency loop
391 384
                         } // OMP_PARALLEL BLOCK
392 385
                     } // mask loop
393
-                    #ifdef HAVE_BOOST_PROGRESS
394 386
                     //if (Receivers->GetNumberOfPoints() > 100) {
395
-                    //    ++ disp;
387
+                    //    ++ mdisp;
396 388
                     //}
397
-                    #endif
398 389
                 } // receiver loop
399 390
                 //std::cout << "End freq parallel " << std::endl;
400 391
             } // Frequency Parallel
@@ -457,11 +448,9 @@ namespace Lemma {
457 448
                             } // frequency loop
458 449
                         } // OMP_PARALLEL BLOCK
459 450
                     } // mask loop
460
-                    #ifdef HAVE_BOOST_PROGRESS
461 451
                     //if (Receivers->GetNumberOfPoints() > 100) {
462 452
                     //    ++ disp;
463 453
                     //}
464
-                    #endif
465 454
                 } // receiver loop
466 455
            } // Polygonal parallel logic
467 456
         } else {
@@ -479,11 +468,10 @@ namespace Lemma {
479 468
             this->Dipole = nullptr;
480 469
         }
481 470
 
482
-        #ifdef HAVE_BOOST_PROGRESS
483 471
         if (progressbar) {
484
-            delete disp;
472
+            delete mdisp;
485 473
         }
486
-        #endif
474
+
487 475
     }
488 476
 
489 477
     #ifdef KIHALEE_EM1D

+ 67
- 0
Modules/LemmaCore/include/ProgressBar.h View File

@@ -0,0 +1,67 @@
1
+#ifndef PROGRESS_BAR_H
2
+#define PROGRESS_BAR_H
3
+
4
+#include <string>
5
+
6
+/**
7
+ * RAII implementation of a progress bar.
8
+ * available from https://stackoverflow.com/questions/46717885/make-boostprogress-display-work-when-more-information-is-being-printed-to-the
9
+ */
10
+class ProgressBar
11
+{
12
+public:
13
+    /**
14
+     * Constructor.
15
+     * It takes two values: the expected number of iterations whose progress we
16
+     * want to monitor and an initial message to be displayed on top of the bar
17
+     * (which can be updated with updateLastPrintedMessage()).
18
+     */
19
+    ProgressBar(
20
+            uint32_t expectedIterations, const std::string& initialMessage="");
21
+
22
+    /**
23
+     * Destructor to guarantee RAII.
24
+     */
25
+    ~ProgressBar();
26
+
27
+    // Make the object non-copyable
28
+    ProgressBar(const ProgressBar& o) = delete;
29
+    ProgressBar& operator=(const ProgressBar& o) = delete;
30
+
31
+    /**
32
+     * Must be invoked when the progress bar is no longer needed to restore the
33
+     * position of the cursor to the end of the output.
34
+     * It is automatically invoked when the object is destroyed.
35
+     */
36
+    void endProgressBar();
37
+
38
+    /**
39
+     * Prints a new message under the last printed message, without overwriting
40
+     * it. This moves the progress bar down to be placed under the newly
41
+     * written message.
42
+     */
43
+    void printNewMessage(const std::string& message);
44
+
45
+    /**
46
+     * Prints a message while the progress bar is on the screen on top on the
47
+     * last printed message. Since the cursor is right at the beginning of the
48
+     * progress bar, it moves the cursor up by one line before printing, and
49
+     * then returns it to its original position.
50
+     */
51
+    void updateLastPrintedMessage(const std::string& message);
52
+
53
+    /**
54
+     * Overloaded prefix operator, used to indicate that the has been a new
55
+     * iteration.
56
+     */
57
+    void operator++();
58
+
59
+private:
60
+    unsigned int mTotalIterations;
61
+    unsigned int mNumberOfTicks;
62
+    bool mEnded;
63
+    size_t mLengthOfLastPrintedMessage;
64
+};
65
+
66
+#endif /* PROGRESS_BAR_H */
67
+

+ 1
- 0
Modules/LemmaCore/src/CMakeLists.txt View File

@@ -1,5 +1,6 @@
1 1
 set (SOURCE 
2 2
 	${SOURCE}
3
+	${CMAKE_CURRENT_SOURCE_DIR}/ProgressBar.cpp                # external to Lemma
3 4
 	${CMAKE_CURRENT_SOURCE_DIR}/helper.cpp
4 5
 	${CMAKE_CURRENT_SOURCE_DIR}/LemmaObject.cpp
5 6
 	${CMAKE_CURRENT_SOURCE_DIR}/ASCIIParser.cpp                # key 

+ 103
- 0
Modules/LemmaCore/src/ProgressBar.cpp View File

@@ -0,0 +1,103 @@
1
+#include "ProgressBar.h"
2
+#include <iostream>
3
+#include <iomanip>
4
+#include <sstream>
5
+
6
+#define LENGTH_OF_PROGRESS_BAR 55
7
+#define PERCENTAGE_BIN_SIZE (100.0/LENGTH_OF_PROGRESS_BAR)
8
+
9
+namespace
10
+{
11
+    std::string generateProgressBar(unsigned int percentage)
12
+    {
13
+        const int progress = static_cast<int>(percentage/PERCENTAGE_BIN_SIZE);
14
+        std::ostringstream ss;
15
+        ss << " " << std::setw(3) << std::right << percentage << "% ";
16
+        std::string bar("[" + std::string(LENGTH_OF_PROGRESS_BAR-2, ' ') + "]");
17
+
18
+        unsigned int numberOfSymbols = std::min(
19
+                std::max(0, progress - 1),
20
+                LENGTH_OF_PROGRESS_BAR - 2);
21
+
22
+        bar.replace(1, numberOfSymbols, std::string(numberOfSymbols, '|'));
23
+
24
+        ss << bar;
25
+        return ss.str();
26
+    }
27
+}
28
+
29
+ProgressBar::ProgressBar(
30
+            uint32_t expectedIterations, const std::string& initialMessage)
31
+    : mTotalIterations(expectedIterations),
32
+      mNumberOfTicks(0),
33
+      mEnded(false)
34
+{
35
+    std::cout << initialMessage << "\n";
36
+    mLengthOfLastPrintedMessage = initialMessage.size();
37
+    std::cout << generateProgressBar(0) << "\r" << std::flush;
38
+}
39
+
40
+ProgressBar::~ProgressBar()
41
+{
42
+    endProgressBar();
43
+}
44
+
45
+void ProgressBar::operator++()
46
+{
47
+    if (mEnded)
48
+    {
49
+        throw std::runtime_error(
50
+                "Attempted to use progress bar after having terminated it");
51
+    }
52
+
53
+    mNumberOfTicks = std::min(mTotalIterations, mNumberOfTicks+1);
54
+    const unsigned int percentage = static_cast<unsigned int>(
55
+            mNumberOfTicks*100.0/mTotalIterations);
56
+
57
+    std::cout << generateProgressBar(percentage) << "\r" << std::flush;
58
+}
59
+
60
+void ProgressBar::printNewMessage(const std::string& message)
61
+{
62
+    if (mEnded)
63
+    {
64
+        throw std::runtime_error(
65
+                "Attempted to use progress bar after having terminated it");
66
+    }
67
+
68
+    std::cout << "\r"
69
+        << std::left
70
+        << std::setw(LENGTH_OF_PROGRESS_BAR + 6)
71
+        << message << "\n";
72
+    mLengthOfLastPrintedMessage = message.size();
73
+    const unsigned int percentage = static_cast<unsigned int>(
74
+            mNumberOfTicks*100.0/mTotalIterations);
75
+
76
+    std::cout << generateProgressBar(percentage) << "\r" << std::flush;
77
+
78
+}
79
+
80
+void ProgressBar::updateLastPrintedMessage(const std::string& message)
81
+{
82
+    if (mEnded)
83
+    {
84
+        throw std::runtime_error(
85
+                "Attempted to use progress bar after having terminated it");
86
+    }
87
+
88
+    std::cout << "\r\033[F"
89
+        << std::left
90
+        << std::setw(mLengthOfLastPrintedMessage)
91
+        << message << "\n";
92
+    mLengthOfLastPrintedMessage = message.size();
93
+}
94
+
95
+void ProgressBar::endProgressBar()
96
+{
97
+    if (!mEnded)
98
+    {
99
+        std::cout << std::string(2, '\n');
100
+    }
101
+    mEnded = true;
102
+}
103
+

Loading…
Cancel
Save