Coding styles

[perso/Immae/Projets/Python/MusicSampler.git] / helpers / __init__.py

diff --git a/helpers/__init__.py b/helpers/__init__.py
index 3b97f2fb435d5837c803895e479184b9c16b694d..9d6663893bc32634b1163f7a525b8b04a7bdf9a5 100644 (file)
--- a/helpers/__init__.py
+++ b/helpers/__init__.py
@@ -2,13 +2,12 @@
 import argparse
 import sys
 import os
+import math
+import sounddevice as sd
+import logging
 
 class Config:
-    def __init__(self, **kwargs):
-        for arg in kwargs:
-            setattr(self, arg, kwargs[arg])
-
-config = Config(yml_file="config.yml")
+    pass
 
 def path():
     if getattr(sys, 'frozen', False):
@@ -18,30 +17,102 @@ def path():
         return path + "/../"
 
 def parse_args():
-    argv = sys.argv[1:]
-    sys.argv = sys.argv[:1]
+    argv = sys.argv[1 :]
+    sys.argv = sys.argv[: 1]
     if "--" in argv:
         index = argv.index("--")
-        kivy_args = argv[index+1:]
-        argv = argv[:index]
+        kivy_args = argv[index+1 :]
+        argv = argv[: index]
 
         sys.argv.extend(kivy_args)
 
-    parser = argparse.ArgumentParser(description="A Music Sampler application.")
+    parser = argparse.ArgumentParser(
+            description="A Music Sampler application.",
+            formatter_class=argparse.ArgumentDefaultsHelpFormatter)
     parser.add_argument("-c", "--config",
             default="config.yml",
             required=False,
             help="Config file to load")
+    parser.add_argument("-d", "--debug",
+            nargs=0,
+            action=DebugModeAction,
+            help="Print messages in console")
+    parser.add_argument("-m", "--builtin-mixing",
+            action="store_true",
+            help="Make the mixing of sounds manually\
+                    (do it if the system cannot handle it correctly)")
+    parser.add_argument("-l", "--latency",
+            default="high",
+            required=False,
+            help="Latency: low, high or number of seconds")
+    parser.add_argument("-b", "--blocksize",
+            default=0,
+            type=int,
+            required=False,
+            help="Blocksize: If not 0, the number of frames to take\
+                    at each step for the mixer")
+    parser.add_argument("-f", "--frame-rate",
+            default=44100,
+            type=int,
+            required=False,
+            help="Frame rate to play the musics")
+    parser.add_argument("-x", "--channels",
+            default=2,
+            type=int,
+            required=False,
+            help="Number of channels to use")
+    parser.add_argument("-s", "--sample-width",
+            default=2,
+            type=int,
+            required=False,
+            help="Sample width (number of bytes for each frame)")
     parser.add_argument("-V", "--version",
             action="version",
-            help="Displays the current version and exits. Only use in bundled package",
+            help="Displays the current version and exits. Only use\
+                    in bundled package",
             version=show_version())
+    parser.add_argument("--device",
+            action=SelectDeviceAction,
+            help="Select this sound device"
+            )
+    parser.add_argument("--list-devices",
+            nargs=0,
+            action=ListDevicesAction,
+            help="List available sound devices"
+            )
     parser.add_argument('--',
             dest="args",
-            help="Kivy arguments. All arguments after this are interpreted by Kivy. Pass \"-- --help\" to get Kivy's usage.")
+            help="Kivy arguments. All arguments after this are interpreted\
+                    by Kivy. Pass \"-- --help\" to get Kivy's usage.")
+
+    from kivy.logger import Logger
+    Logger.setLevel(logging.ERROR)
+
     args = parser.parse_args(argv)
 
-    config.yml_file = args.config
+    Config.yml_file = args.config
+
+    Config.latency = args.latency
+    Config.blocksize = args.blocksize
+    Config.frame_rate = args.frame_rate
+    Config.channels = args.channels
+    Config.sample_width = args.sample_width
+    Config.builtin_mixing = args.builtin_mixing
+
+class DebugModeAction(argparse.Action):
+    def __call__(self, parser, namespace, values, option_string=None):
+        from kivy.logger import Logger
+        Logger.setLevel(logging.DEBUG)
+
+class SelectDeviceAction(argparse.Action):
+    def __call__(self, parser, namespace, values, option_string=None):
+        sd.default.device = values
+
+class ListDevicesAction(argparse.Action):
+    nargs = 0
+    def __call__(self, parser, namespace, values, option_string=None):
+        print(sd.query_devices())
+        sys.exit()
 
 def show_version():
     if getattr(sys, 'frozen', False):
@@ -50,9 +121,6 @@ def show_version():
     else:
         return "option '-v' can only be used in bundled package"
 
-def yml_file():
-    return config.yml_file
-
 def duration_to_min_sec(duration):
     minutes = int(duration / 60)
     seconds = int(duration) % 60
@@ -60,3 +128,19 @@ def duration_to_min_sec(duration):
         return "{:2}:{:0>2}".format(minutes, seconds)
     else:
         return "{}:{:0>2}".format(minutes, seconds)
+
+def gain(volume, old_volume=None):
+    if old_volume is None:
+        return 20 * math.log10(volume / 100)
+    else:
+        return [
+                20 * math.log10(max(volume, 0.1) / max(old_volume, 0.1)),
+                max(volume, 0)]
+
+def debug_print(message):
+    from kivy.logger import Logger
+    Logger.debug('MusicSampler: ' + message)
+
+def error_print(message):
+    from kivy.logger import Logger
+    Logger.error('MusicSampler: ' + message)