Aggiunta opzione per non stampare output e modificato titolo

Leonardo Robol [2010-03-21 10:54]
Aggiunta opzione per non stampare output e modificato titolo
del grafico.
Filename
Filtering/dwt.py
diff --git a/Filtering/dwt.py b/Filtering/dwt.py
index 59df114..7e69ec9 100755
--- a/Filtering/dwt.py
+++ b/Filtering/dwt.py
@@ -5,27 +5,32 @@
 import sys
 from optparse import OptionParser

+quiet = False
+
 def Output (string):
     """Output with colors :)"""
-    print "\033[32;1m===>\033[0m %s" % string
+    if not quiet:
+        print "\033[32;1m===>\033[0m %s" % string

 def StartProgram():
     """Starting banner"""
-    print "\033[31;1m===>\033[0m Discrete Wavelet transform started"
+    if not quiet:
+        print "\033[31;1m===>\033[0m Discrete Wavelet transform started"

 def EndProgram():
     """End banner"""
-    print "",
+    if not quiet:
+        print "",

 def LoadingLibrariesStarted():
     """Loading libraries banner"""
-    if __name__ == "__main__":
+    if __name__ == "__main__" and not quiet:
         print "\033[31;1m===>\033[0m Loading numeric libraries...",
         sys.stdout.flush ()

 def LoadingLibrariesFinished():
     """Loading libraries finished banner"""
-    if __name__ == "__main__":
+    if __name__ == "__main__" and not quiet:
         print "done"


@@ -35,6 +40,8 @@ if __name__ == "__main__":
     parser.add_option("-r", "--rebuild", dest="rebuild",
                       default=False, action="store_true",
                       help="Make DWT and then IDWT")
+    parser.add_option("-q", "--quiet", dest="quiet",
+                      default=False, action="store_true")
     parser.add_option("-w", "--write", dest="filewrite", default='rebuilt.raw',
                       help="Write reconstructed samples to this file")
     parser.add_option("-s", "--show", dest="show",
@@ -56,6 +63,8 @@ if __name__ == "__main__":
     if (not options.show) and (not options.rebuild):
         exit

+    quiet = options.quiet
+
 LoadingLibrariesStarted()

 # Importing libraries
@@ -216,7 +225,7 @@ class DWT():


         # Set some nice text
-        title (r"Decomposition using %s filter bank" % self.filterBankName)
+        title (r"Wavelet decomposition using %s filter bank" % self.filterBankName)
         xlabel (r"time (s)")

         show ()
ViewGit