class Application(Gtk.Application):
def __init__(self):
app_id = 'com.github.wwmm.pulseeffects'
app_flags = Gio.ApplicationFlags.HANDLES_COMMAND_LINE
GLib.set_application_name('PulseEffects')
GLib.setenv('PULSE_PROP_media.role', 'production', True)
GLib.setenv('PULSE_PROP_application.icon_name', 'pulseeffects', True)
Gtk.Application.__init__(self, application_id=app_id, flags=app_flags)
self.add_main_option('no-window', ord('n'), GLib.OptionFlags.NONE,
GLib.OptionArg.NONE, 'do not show window', None)
def do_startup(self):
Gtk.Application.do_startup(self)
self.ui_initialized = False
self.module_path = os.path.dirname(__file__)
log_format = '%(asctime)s.%(msecs)d - %(name)s - %(levelname)s'
log_format = log_format + ' - %(message)s'
logging.basicConfig(format=log_format,
datefmt='%H:%M:%S',
level=logging.INFO)
self.log = logging.getLogger('PulseEffects')
self.gtk_settings = Gtk.Settings.get_default()
self.settings = Gio.Settings('com.github.wwmm.pulseeffects')
# creating user presets folder
self.user_config_dir = os.path.join(GLib.get_user_config_dir(),
'PulseEffects')
os.makedirs(self.user_config_dir, exist_ok=True)
# pulseaudio
self.pm = PulseManager()
self.pm.load_apps_sink()
self.pm.load_mic_sink()
# ui initializations
self.builder = Gtk.Builder()
self.builder.add_from_file(self.module_path + '/ui/main_ui.glade')
self.builder.connect_signals(self)
self.window = self.builder.get_object('MainWindow')
self.server_info_label = self.builder.get_object('server_info_label')
self.window.set_application(self)
self.create_appmenu()
# gstreamer sink input effects
self.sie = SinkInputEffects(self.pm.default_sink_rate)
self.sie.set_source_monitor_name(self.pm.apps_sink_monitor_name)
self.sie.set_output_sink_name(self.pm.default_sink_name)
# gstreamer source outputs effects
self.soe = SourceOutputEffects(self.pm.default_source_rate)
self.soe.set_source_monitor_name(self.pm.default_source_name)
self.soe.set_output_sink_name('PulseEffects_mic')
self.draw_spectrum = DrawSpectrum(self)
self.init_theme()
self.init_settings_menu()
self.init_buffer_time()
self.init_latency_time()
self.list_sink_inputs = ListSinkInputs(self.sie, self.pm)
self.list_source_outputs = ListSourceOutputs(self.soe, self.pm)
self.list_sink_inputs.connect('app_removed', self.on_app_removed)
self.list_source_outputs.connect('app_removed', self.on_app_removed)
self.init_spectrum_widgets()
self.init_stack_widgets()
# this connection is changed inside the stack switch handler
# depending on the selected child. The connection below is not
# permanent but just a default
self.spectrum_handler_id = self.sie.connect(
'new_spectrum', self.draw_spectrum.on_new_spectrum)
# searching for apps
self.pm.find_sink_inputs()
self.pm.find_source_outputs()
def do_activate(self):
self.window.present()
self.ui_initialized = True
def do_command_line(self, command_line):
options = command_line.get_options_dict()
if options.contains("no-window"):
self.log.info('Running as a background service')
else:
self.activate()
return 0
def do_shutdown(self):
Gtk.Application.do_shutdown(self)
self.sie.set_state('null')
self.soe.set_state('null')
self.pm.exit()
def create_appmenu(self):
menu = Gio.Menu()
menu.append('About', 'app.about')
menu.append('Quit', 'app.quit')
self.set_app_menu(menu)
about_action = Gio.SimpleAction.new('about', None)
about_action.connect('activate', self.onAbout)
self.add_action(about_action)
quit_action = Gio.SimpleAction.new('quit', None)
quit_action.connect('activate', lambda action, parameter: self.quit())
self.add_action(quit_action)
def init_theme(self):
switch = self.builder.get_object('theme_switch')
use_dark = self.settings.get_value('use-dark-theme').unpack()
switch.set_active(use_dark)
def on_theme_switch_state_set(self, obj, state):
self.gtk_settings.props.gtk_application_prefer_dark_theme = state
out = GLib.Variant('b', state)
self.settings.set_value('use-dark-theme', out)
def init_stack_widgets(self):
self.stack = self.builder.get_object('stack')
self.stack.add_named(self.sie.ui_window, 'sink_inputs')
self.stack.child_set_property(self.sie.ui_window, 'icon-name',
'audio-speakers-symbolic')
self.stack.add_named(self.soe.ui_window, "source_outputs")
self.stack.child_set_property(self.soe.ui_window, 'icon-name',
'audio-input-microphone-symbolic')
self.stack_current_child_name = 'sink_inputs'
server_info = str(self.pm.default_sink_format) + ', ' + \
'{:.1f}'.format(round(
self.pm.default_sink_rate / 1000.0, 1)) + ' kHz'
self.server_info_label.set_text(server_info)
def on_visible_child_changed(stack, visible_child):
name = stack.get_visible_child_name()
if name == 'sink_inputs':
if self.stack_current_child_name == 'source_outputs':
self.soe.disconnect(self.spectrum_handler_id)
self.spectrum_handler_id = self.sie.connect(
'new_spectrum', self.draw_spectrum.on_new_spectrum)
self.stack_current_child_name = 'sink_inputs'
server_info = str(self.pm.default_sink_format) + ', ' + \
'{:.1f}'.format(round(
self.pm.default_sink_rate / 1000.0, 1)) + ' kHz'
self.server_info_label.set_text(server_info)
elif name == 'source_outputs':
if self.stack_current_child_name == 'sink_inputs':
self.sie.disconnect(self.spectrum_handler_id)
self.spectrum_handler_id = self.soe.connect(
'new_spectrum', self.draw_spectrum.on_new_spectrum)
self.stack_current_child_name = 'source_outputs'
server_info = str(self.pm.default_source_format) + ', ' + \
'{:.1f}'.format(round(
self.pm.default_source_rate / 1000.0, 1)) + ' kHz'
self.server_info_label.set_text(server_info)
self.draw_spectrum.clear()
self.stack.connect("notify::visible-child", on_visible_child_changed)
def init_settings_menu(self):
button = self.builder.get_object('settings_popover_button')
menu = self.builder.get_object('settings_menu')
popover = Gtk.Popover.new(button)
popover.props.transitions_enabled = True
popover.add(menu)
def button_clicked(arg):
if popover.get_visible():
popover.hide()
else:
popover.show_all()
button.connect("clicked", button_clicked)
def init_buffer_time(self):
value = self.settings.get_value('buffer-time').unpack()
buffer_time = self.builder.get_object('buffer_time')
buffer_time.set_value(value)
self.sie.init_buffer_time(value * 1000)
self.soe.init_buffer_time(value * 1000)
def on_buffer_time_value_changed(self, obj):
value = obj.get_value()
out = GLib.Variant('i', value)
self.settings.set_value('buffer-time', out)
if self.ui_initialized:
self.sie.set_buffer_time(value * 1000)
self.soe.set_buffer_time(value * 1000)
else:
self.sie.init_buffer_time(value * 1000)
self.soe.init_buffer_time(value * 1000)
def init_latency_time(self):
value = self.settings.get_value('latency-time').unpack()
latency_time = self.builder.get_object('latency_time')
latency_time.set_value(value)
self.sie.init_latency_time(value * 1000)
# self.soe.init_latency_time(value * 1000)
def on_latency_time_value_changed(self, obj):
value = obj.get_value()
out = GLib.Variant('i', value)
self.settings.set_value('latency-time', out)
if self.ui_initialized:
self.sie.set_latency_time(value * 1000)
# self.soe.set_latency_time(value * 1000)
else:
self.sie.init_latency_time(value * 1000)
# self.soe.init_latency_time(value * 1000)
def init_spectrum_widgets(self):
show_spectrum_switch = self.builder.get_object('show_spectrum')
spectrum_n_points_obj = self.builder.get_object('spectrum_n_points')
show_spectrum = self.settings.get_value('show-spectrum').unpack()
spectrum_n_points = self.settings.get_value(
'spectrum-n-points').unpack()
show_spectrum_switch.set_active(show_spectrum)
spectrum_n_points_obj.set_value(spectrum_n_points)
self.sie.set_spectrum_n_points(spectrum_n_points)
self.soe.set_spectrum_n_points(spectrum_n_points)
if show_spectrum:
self.draw_spectrum.show()
else:
self.draw_spectrum.hide()
def on_show_spectrum_state_set(self, obj, state):
if state:
self.draw_spectrum.show()
self.sie.enable_spectrum(True)
self.soe.enable_spectrum(True)
else:
self.sie.enable_spectrum(False)
self.soe.enable_spectrum(False)
self.draw_spectrum.hide()
out = GLib.Variant('b', state)
self.settings.set_value('show-spectrum', out)
def on_spectrum_n_points_value_changed(self, obj):
value = obj.get_value()
out = GLib.Variant('i', value)
self.settings.set_value('spectrum-n-points', out)
self.sie.set_spectrum_n_points(value)
self.soe.set_spectrum_n_points(value)
def on_app_removed(self, obj):
self.window.resize(1, 1)
def on_reset_all_settings_clicked(self, obj):
self.settings.reset('buffer-time')
self.settings.reset('latency-time')
self.settings.reset('show-spectrum')
self.settings.reset('spectrum-n-points')
self.settings.reset('use-dark-theme')
self.init_theme()
self.init_buffer_time()
self.init_latency_time()
self.init_spectrum_widgets()
self.stack.set_visible_child(self.sie.ui_window)
self.sie.reset()
self.soe.reset()
def add_file_filter(self, dialog):
file_filter = Gtk.FileFilter()
file_filter.set_name("preset")
file_filter.add_mime_type("text/plain")
dialog.add_filter(file_filter)
def on_save_user_preset_clicked(self, obj):
dialog = Gtk.FileChooserDialog(
'', self.window, Gtk.FileChooserAction.SAVE,
(Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL, Gtk.STOCK_SAVE,
Gtk.ResponseType.OK))
dialog.set_current_folder(self.user_config_dir)
dialog.set_current_name('user.preset')
self.add_file_filter(dialog)
response = dialog.run()
if response == Gtk.ResponseType.OK:
path = dialog.get_filename()
if not path.endswith(".preset"):
path += ".preset"
s = SavePresets(path)
s.save_sink_inputs_presets(self.sie.settings)
s.save_source_outputs_presets(self.soe.settings)
s.write_config()
dialog.destroy()
def on_load_user_preset_clicked(self, obj):
dialog = Gtk.FileChooserDialog(
'', self.window, Gtk.FileChooserAction.OPEN,
(Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL, Gtk.STOCK_OPEN,
Gtk.ResponseType.OK))
dialog.set_current_folder(self.user_config_dir)
self.add_file_filter(dialog)
response = dialog.run()
if response == Gtk.ResponseType.OK:
path = dialog.get_filename()
l = LoadPresets(path)
l.load_sink_inputs_presets(self.sie.settings)
l.load_source_outputs_presets(self.soe.settings)
dialog.destroy()
def onAbout(self, action, parameter):
builder = Gtk.Builder()
builder.add_from_file(self.module_path + '/ui/about.glade')
dialog = builder.get_object('about_dialog')
dialog.set_transient_for(self.window)
dialog.run()
dialog.destroy()
class Application(Gtk.Application):
def __init__(self):
app_id = 'com.github.wwmm.pulseeffects'
app_flags = Gio.ApplicationFlags.HANDLES_COMMAND_LINE
GLib.set_application_name('PulseEffects')
GLib.setenv('PULSE_PROP_application.icon_name', 'pulseeffects', True)
GLib.setenv('PULSE_PROP_application.id',
'com.github.wwmm.pulseeffects', True)
GLib.setenv('PULSE_PROP_application.version', '3.2.1', True)
Gtk.Application.__init__(self, application_id=app_id, flags=app_flags)
GLib.unix_signal_add(GLib.PRIORITY_DEFAULT, 2, self.quit) # sigint
help_msg = _('Quit PulseEffects. Useful when running in service mode.')
self.add_main_option('quit', ord('q'), GLib.OptionFlags.NONE,
GLib.OptionArg.NONE, help_msg, None)
help_msg = _('Show available presets.')
self.add_main_option('presets', ord('p'), GLib.OptionFlags.NONE,
GLib.OptionArg.NONE, help_msg, None)
help_msg = _('Load a preset. Example: pulseeffects -l music')
self.add_main_option('load-preset', ord('l'), GLib.OptionFlags.NONE,
GLib.OptionArg.STRING, help_msg, None)
def do_startup(self):
Gtk.Application.do_startup(self)
self.ui_initialized = False
self.running_as_service = False
self.module_path = os.path.dirname(__file__)
log_format = '%(asctime)s.%(msecs)d - %(name)s - %(levelname)s'
log_format = log_format + ' - %(message)s'
logging.basicConfig(format=log_format,
datefmt='%H:%M:%S',
level=logging.INFO)
self.log = logging.getLogger('PulseEffects')
self.gtk_settings = Gtk.Settings.get_default()
self.settings = Gio.Settings('com.github.wwmm.pulseeffects')
# creating user presets folder
self.user_config_dir = os.path.join(GLib.get_user_config_dir(),
'PulseEffects')
os.makedirs(self.user_config_dir, exist_ok=True)
# autostart file path
autostart_file_name = 'autostart/pulseeffects-service.desktop'
self.autostart_file = os.path.join(GLib.get_user_config_dir(),
autostart_file_name)
self.create_appmenu()
# custom css styles
self.apply_css_style('listbox.css')
icon_theme = Gtk.IconTheme.get_default()
icon_theme.append_search_path(self.module_path + '/ui')
# pulseaudio
self.pm = PulseManager()
self.pm.load_apps_sink()
self.pm.load_mic_sink()
self.sie = SinkInputEffects(self.pm)
self.soe = SourceOutputEffects(self.pm)
if self.props.flags & Gio.ApplicationFlags.IS_SERVICE:
self.running_as_service = True
self.init_ui()
self.sie.post_messages(False)
self.soe.post_messages(False)
self.pm.find_sink_inputs()
self.pm.find_source_outputs()
self.log.info(_('Running in background'))
self.hold()
def init_ui(self):
self.builder = Gtk.Builder.new_from_file(self.module_path +
'/ui/main_ui.glade')
self.builder.connect_signals(self)
self.window = self.builder.get_object('MainWindow')
self.window.set_application(self)
self.window.connect('destroy', self.on_window_destroy)
image_test = self.builder.get_object('image_test')
image_test.set_from_icon_name('pulseeffects-sine-symbolic',
Gtk.IconSize.BUTTON)
self.sie.init_ui()
self.soe.init_ui()
self.draw_spectrum = DrawSpectrum(self)
self.init_settings_menu()
self.init_buffer_time()
self.init_latency_time()
self.init_spectrum_widgets()
self.init_stack_widgets()
self.init_autostart_switch()
# Gsettings bindings
flag = Gio.SettingsBindFlags.DEFAULT
switch_apps = self.builder.get_object('enable_all_apps')
switch_theme = self.builder.get_object('theme_switch')
self.settings.bind('use-dark-theme', switch_theme, 'active', flag)
self.settings.bind('use-dark-theme', self.gtk_settings,
'gtk_application_prefer_dark_theme', flag)
self.settings.bind('enable-all-apps', switch_apps, 'active', flag)
self.settings.bind('enable-all-apps', self.sie, 'switch_on_all_apps',
flag)
self.settings.bind('enable-all-apps', self.soe, 'switch_on_all_apps',
flag)
# this connection is changed inside the stack switch handler
# depending on the selected child. The connection below is not
# permanent but just a default
self.spectrum_handler_id = self.sie.connect(
'new_spectrum', self.draw_spectrum.on_new_spectrum)
self.presets = PresetsManager(self)
self.ui_initialized = True
self.sie.there_is_window = True
self.soe.there_is_window = True
def on_window_destroy(self, window):
self.ui_initialized = False
self.sie.there_is_window = False
self.soe.there_is_window = False
if self.running_as_service:
self.sie.post_messages(False)
self.soe.post_messages(False)
def do_activate(self):
if not self.ui_initialized:
self.init_ui()
self.pm.find_sink_inputs()
self.pm.find_source_outputs()
self.window.present()
self.sie.post_messages(True)
self.soe.post_messages(True)
def do_command_line(self, command_line):
options = command_line.get_options_dict()
if options.contains('quit'):
self.quit()
elif options.contains('presets'):
self.presets.list_presets()
elif options.contains('load-preset'):
value = options.lookup_value('load-preset', None).unpack()
self.presets.load_preset(value)
else:
self.do_activate()
return Gtk.Application.do_command_line(self, command_line)
def do_shutdown(self):
Gtk.Application.do_shutdown(self)
self.sie.set_state('null')
self.soe.set_state('null')
self.pm.exit()
def create_appmenu(self):
menu = Gio.Menu()
menu.append('About', 'app.about')
menu.append('Quit', 'app.quit')
self.set_app_menu(menu)
about_action = Gio.SimpleAction.new('about', None)
about_action.connect('activate', self.onAbout)
self.add_action(about_action)
quit_action = Gio.SimpleAction.new('quit', None)
quit_action.connect('activate', lambda action, parameter: self.quit())
self.add_action(quit_action)
def init_stack_widgets(self):
self.stack = self.builder.get_object('stack')
self.stack.add_named(self.sie.ui_window, 'sink_inputs')
self.stack.child_set_property(self.sie.ui_window, 'icon-name',
'audio-speakers-symbolic')
self.stack.add_named(self.soe.ui_window, "source_outputs")
self.stack.child_set_property(self.soe.ui_window, 'icon-name',
'audio-input-microphone-symbolic')
self.stack_current_child_name = 'sink_inputs'
self.stack.connect("notify::visible-child",
self.on_stack_visible_child_changed)
def on_stack_visible_child_changed(self, stack, visible_child):
name = stack.get_visible_child_name()
if name == 'sink_inputs':
if self.stack_current_child_name == 'source_outputs':
self.soe.disconnect(self.spectrum_handler_id)
self.spectrum_handler_id = self.sie.connect(
'new_spectrum', self.draw_spectrum.on_new_spectrum)
self.stack_current_child_name = 'sink_inputs'
elif name == 'source_outputs':
if self.stack_current_child_name == 'sink_inputs':
self.sie.disconnect(self.spectrum_handler_id)
self.spectrum_handler_id = self.soe.connect(
'new_spectrum', self.draw_spectrum.on_new_spectrum)
self.stack_current_child_name = 'source_outputs'
self.draw_spectrum.clear()
def init_settings_menu(self):
button = self.builder.get_object('settings_popover_button')
menu = self.builder.get_object('settings_menu')
popover = Gtk.Popover.new(button)
popover.props.transitions_enabled = True
popover.add(menu)
def button_clicked(arg):
if popover.get_visible():
popover.hide()
else:
popover.show_all()
button.connect("clicked", button_clicked)
def init_buffer_time(self):
value = self.settings.get_value('buffer-time').unpack()
buffer_time = self.builder.get_object('buffer_time')
buffer_time.set_value(value)
self.sie.init_buffer_time(value * 1000)
def on_buffer_time_value_changed(self, obj):
value = obj.get_value()
out = GLib.Variant('i', value)
self.settings.set_value('buffer-time', out)
if self.ui_initialized:
self.sie.set_buffer_time(value * 1000)
else:
self.sie.init_buffer_time(value * 1000)
def init_latency_time(self):
value = self.settings.get_value('latency-time').unpack()
latency_time = self.builder.get_object('latency_time')
latency_time.set_value(value)
self.sie.init_latency_time(value * 1000)
def on_latency_time_value_changed(self, obj):
value = obj.get_value()
out = GLib.Variant('i', value)
self.settings.set_value('latency-time', out)
if self.ui_initialized:
self.sie.set_latency_time(value * 1000)
else:
self.sie.init_latency_time(value * 1000)
def init_autostart_switch(self):
switch = self.builder.get_object('enable_autostart')
if os.path.isfile(self.autostart_file):
switch.set_state(True)
else:
switch.set_state(False)
def on_enable_autostart_state_set(self, obj, state):
if state:
with open(self.autostart_file, "w") as f:
f.write('[Desktop Entry]\n')
f.write('Name=PulseEffects\n')
f.write('Comment=PulseEffects Service\n')
f.write('Exec=pulseeffects --gapplication-service\n')
f.write('Icon=pulseeffects\n')
f.write('StartupNotify=false\n')
f.write('Terminal=false\n')
f.write('Type=Application\n')
else:
os.remove(self.autostart_file)
def init_spectrum_widgets(self):
show_spectrum_switch = self.builder.get_object('show_spectrum')
spectrum_n_points_obj = self.builder.get_object('spectrum_n_points')
show_spectrum = self.settings.get_value('show-spectrum').unpack()
spectrum_n_points = self.settings.get_value(
'spectrum-n-points').unpack()
show_spectrum_switch.set_active(show_spectrum)
spectrum_n_points_obj.set_value(spectrum_n_points)
self.sie.set_spectrum_n_points(spectrum_n_points)
self.soe.set_spectrum_n_points(spectrum_n_points)
if show_spectrum:
self.draw_spectrum.show()
else:
self.draw_spectrum.hide()
def on_show_spectrum_state_set(self, obj, state):
if state:
self.draw_spectrum.show()
self.sie.enable_spectrum(True)
self.soe.enable_spectrum(True)
else:
self.sie.enable_spectrum(False)
self.soe.enable_spectrum(False)
self.draw_spectrum.hide()
out = GLib.Variant('b', state)
self.settings.set_value('show-spectrum', out)
def on_spectrum_n_points_value_changed(self, obj):
value = obj.get_value()
out = GLib.Variant('i', value)
self.settings.set_value('spectrum-n-points', out)
self.sie.set_spectrum_n_points(value)
self.soe.set_spectrum_n_points(value)
def apply_css_style(self, css_file):
provider = Gtk.CssProvider()
css_file = Gio.File.new_for_path(self.module_path + '/ui/' + css_file)
provider.load_from_file(css_file)
screen = Gdk.Screen.get_default()
priority = Gtk.STYLE_PROVIDER_PRIORITY_APPLICATION
Gtk.StyleContext.add_provider_for_screen(screen, provider, priority)
def on_test_clicked(self, obj):
t = Test(self.pm)
t.run()
def on_reset_all_settings_clicked(self, obj):
self.settings.reset('buffer-time')
self.settings.reset('latency-time')
self.settings.reset('show-spectrum')
self.settings.reset('spectrum-n-points')
self.settings.reset('use-dark-theme')
self.settings.reset('enable-all-apps')
self.init_buffer_time()
self.init_latency_time()
self.init_spectrum_widgets()
self.stack.set_visible_child(self.sie.ui_window)
self.sie.reset()
self.soe.reset()
def onAbout(self, action, parameter):
builder = Gtk.Builder()
builder.add_from_file(self.module_path + '/ui/about.glade')
dialog = builder.get_object('about_dialog')
dialog.set_transient_for(self.window)
dialog.run()
dialog.destroy()
