← Back to branch summary

~jrjohansson/qutip/master

« back to all changes in this revision

Viewing changes to qutip/mac/ProgressBar.py

  • Committer: Paul Nation
  • Date: 2011-04-21 04:46:56 UTC
  • Revision ID: git-v1:dd4c966b490aa468dfbd28cef66694df4bf235c8

Show diffs side-by-side

added added

removed removed

Lines of Context:
qutip/mac/ProgressBar.py
 
1
#This file is part of QuTIP.
 
2
#
 
3
#    QuTIP is free software: you can redistribute it and/or modify
 
4
#    it under the terms of the GNU General Public License as published by
 
5
#    the Free Software Foundation, either version 3 of the License, or
 
6
#   (at your option) any later version.
 
7
#
 
8
#    QuTIP is distributed in the hope that it will be useful,
 
9
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
 
10
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
11
#    GNU General Public License for more details.
 
12
#
 
13
#    You should have received a copy of the GNU General Public License
 
14
#    along with QuTIP.  If not, see <http://www.gnu.org/licenses/>.
 
15
#
 
16
# Copyright (C) 2011, Paul D. Nation & Robert J. Johansson
 
17
#
 
18
###########################################################################
 
19
import os
 
20
 
 
21
class ProgressBar:
 
22
    """Simple class for displaying progress bars using CocoaDialog"""
 
23
 
 
24
    # Change CD_BASE to reflect the location of Cocoadialog on your system
 
25
    CD_BASE = os.path.dirname(__file__)
 
26
    CD_PATH = os.path.join(CD_BASE, "CocoaDialog.app/Contents/MacOS/CocoaDialog")
 
27
    
 
28
    def __init__(self, title="Progress", message="", percent=0):
 
29
        """Create progress bar dialog"""
 
30
        template = "%s progressbar --title '%s' --text '%s' --percent %d"
 
31
        self.percent = percent
 
32
        self.pipe = os.popen(template % (ProgressBar.CD_PATH, title, message, percent), "w")
 
33
        self.message = message
 
34
            
 
35
    def update(self, percent, message=False):
 
36
        """Update progress bar (and message if desired)"""
 
37
        if message:
 
38
            self.message = message  # store message for persistence
 
39
        self.pipe.write("%d %s\n" % (percent, self.message))
 
40
        self.pipe.flush()
 
41
        
 
42
    def finish(self):
 
43
        """Close progress bar window"""
 
44
        self.pipe.close()
 
45
 
 
46
 
 
47
if __name__ == "__main__":
 
48
    # Sample usage
 
49
    import time
 
50
    bar = ProgressBar(title="Running Monte-Carlo Trajectories:")
 
51
    
 
52
    for percent in range(25):
 
53
        time.sleep(.2)
 
54
        bar.update(percent, "Trajectories completed: "+str(percent)+"/10000")
 
55
        
 
56
    for percent in range(25,100):
 
57
        time.sleep(.02)
 
58
        bar.update(percent, "Trajectories completed: "+str(percent)+"/10000")
 
59
     
 
60
    time.sleep(.75)
 
61
    bar.finish()
 
 
b'\\ No newline at end of file'