123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264 |
- import gi
- import subprocess
- gi.require_version('Gtk', '3.0')
- from gi.repository import Gtk, Gdk, GdkPixbuf
- import app.core.presets as presets
- import app.core.state as state
- import app.core.config as config
- import app.core.utils as utils
- from app.core.presets import Preset
- sox_multiplier = 100
- class MainWindow(Gtk.Window):
- '''
- Main window for Lyrebird
- Lyrebird is a simple and powerful voice changer for Linux, written in GTK 3.
- '''
- def __init__(self):
- Gtk.Window.__init__(self, title='Lyrebird')
- self.set_border_width(10)
- self.set_size_request(600, 500)
- self.set_default_size(600, 500)
- headerbar = Gtk.HeaderBar()
- headerbar.set_show_close_button(True)
- headerbar.props.title = 'Lyrebird'
- about_btn = Gtk.Button.new_from_icon_name('help-about-symbolic', Gtk.IconSize.BUTTON);
- about_btn.connect('clicked', self.about_clicked)
- headerbar.pack_start(about_btn)
- self.set_wmclass ('Lyrebird', 'Lyrebird')
- self.set_title('Lyrebird')
- self.set_titlebar(headerbar)
-
- self.set_icon_from_file('icon.png')
-
- lock_file = utils.place_lock()
- if lock_file is None:
- self.show_error_message("Lyrebird Already Running", "Only one instance of Lyrebird can be ran at a time.")
- exit(1)
- else:
- self.lock_file = lock_file
-
- self.sox_process = None
-
-
-
- utils.unload_pa_modules()
-
- state.config = config.load_config()
-
- self.build_ui()
- def show_error_message(self, title, msg):
- '''
- Create an error message dialog with title and string message.
- '''
- dialog = Gtk.MessageDialog(
- parent = self,
- type = Gtk.MessageType.ERROR,
- buttons = Gtk.ButtonsType.OK,
- message_format = msg)
- dialog.set_transient_for(self)
- dialog.set_title(title)
- dialog.show()
- dialog.run()
- dialog.destroy()
- def build_ui(self):
- self.vbox = Gtk.VBox()
-
- self.hbox_toggle = Gtk.HBox()
- self.toggle_label = Gtk.Label('Toggle Lyrebird')
- self.toggle_label.set_halign(Gtk.Align.START)
- self.toggle_switch = Gtk.Switch()
- self.toggle_switch.set_size_request(10, 25)
- self.toggle_switch.connect('notify::active', self.toggle_activated)
- self.hbox_toggle.pack_start(self.toggle_label, False, False, 0)
- self.hbox_toggle.pack_end(self.toggle_switch, False, False, 0)
-
- self.hbox_pitch = Gtk.HBox()
- self.pitch_label = Gtk.Label('Pitch Shift ')
- self.pitch_label.set_halign(Gtk.Align.START)
- self.pitch_adj = Gtk.Adjustment(0, -10, 10, 5, 10, 0)
- self.pitch_scale = Gtk.Scale(orientation=Gtk.Orientation.HORIZONTAL, adjustment=self.pitch_adj)
- self.pitch_scale.set_valign(Gtk.Align.CENTER)
- self.pitch_scale.connect('value-changed', self.pitch_scale_moved)
-
- self.pitch_scale.set_sensitive(False)
- self.hbox_pitch.pack_start(self.pitch_label, False, False, 0)
- self.hbox_pitch.pack_end(self.pitch_scale, True, True, 0)
-
- self.effects_label = Gtk.Label()
- self.effects_label.set_markup('<b>Presets</b>')
- self.effects_label.set_halign(Gtk.Align.START)
- self.flowbox = Gtk.FlowBox()
- self.flowbox.set_valign(Gtk.Align.START)
- self.flowbox.set_max_children_per_line(5)
- self.flowbox.set_selection_mode(Gtk.SelectionMode.NONE)
-
- self.create_flowbox_items(self.flowbox)
- self.vbox.pack_start(self.hbox_toggle, False, False, 5)
- self.vbox.pack_start(self.hbox_pitch, False, False, 5)
- self.vbox.pack_start(self.effects_label, False, False, 5)
- self.vbox.pack_end(self.flowbox, True, True, 0)
- self.add(self.vbox)
- def create_flowbox_items(self, flowbox):
- state.loaded_presets = presets.load_presets()
- for preset in state.loaded_presets:
- button = Gtk.Button()
- button.set_size_request(80, 80)
- button.set_label(preset.name)
- button.connect('clicked', self.preset_clicked)
- flowbox.add(button)
-
- def about_clicked(self, button):
- about = Gtk.AboutDialog()
- about.set_program_name('Lyrebird Voice Changer')
- about.set_version("v1.1.0")
- about.set_copyright('(c) Lyrebird 2020-2022')
- about.set_comments('Simple and powerful voice changer for Linux, written in GTK 3')
- about.set_logo(GdkPixbuf.Pixbuf.new_from_file('icon.png'))
- about.run()
- about.destroy()
- def toggle_activated(self, switch, gparam):
- if switch.get_active():
-
- null_sink = subprocess.check_call(
- 'pactl load-module module-null-sink sink_name=Lyrebird-Output node.description="Lyrebird Output"'.split(' ')
- )
- remap_sink = subprocess.check_call(
- 'pactl load-module module-remap-source source_name=Lyrebird-Input master=Lyrebird-Output.monitor node.description="Lyrebird Virtual Input"'\
- .split(' ')
- )
- print(f'Loaded null output sink and remap sink')
- state.sink = null_sink
-
- self.terminate_sox()
-
- default_preset = state.loaded_presets[0]
- current_preset = state.current_preset or default_preset
- if current_preset.override_pitch:
-
- self.pitch_scale.set_value(float(current_preset.pitch_value))
- self.pitch_scale.set_sensitive(False)
- command = utils.build_sox_command(
- current_preset,
- config_object=state.config
- )
- else:
- self.pitch_scale.set_sensitive(True)
- command = utils.build_sox_command(
- current_preset,
- config_object=state.config,
- scale_object=self.pitch_scale
- )
- self.sox_process = subprocess.Popen(command.split(' '))
- else:
- utils.unload_pa_modules()
- self.terminate_sox()
- def pitch_scale_moved(self, event):
- global sox_multiplier
-
-
-
- if state.current_preset is not None:
-
- self.terminate_sox()
- if not state.current_preset.override_pitch:
-
- command = utils.build_sox_command(
- state.current_preset,
- config_object=state.config,
- scale_object=self.pitch_scale
- )
- self.sox_process = subprocess.Popen(command.split(' '))
- def preset_clicked(self, button):
- global sox_multiplier
- self.terminate_sox()
-
- current_preset = list(filter(lambda p: p.name == button.props.label, state.loaded_presets))[0]
- state.current_preset = current_preset
- if current_preset.override_pitch:
-
- self.pitch_scale.set_value(float(current_preset.pitch_value))
- self.pitch_scale.set_sensitive(False)
- command = utils.build_sox_command(
- state.current_preset,
- config_object=state.config
- )
- else:
- self.pitch_scale.set_sensitive(True)
- command = utils.build_sox_command(
- state.current_preset,
- config_object=state.config,
- scale_object=self.pitch_scale
- )
- self.sox_process = subprocess.Popen(command.split(' '))
- def terminate_sox(self, timeout=1):
- if self.sox_process is not None:
- self.sox_process.terminate()
- try:
- self.sox_process.wait(timeout=timeout)
- except subprocess.TimeoutExpired:
- self.sox_process.kill()
- self.sox_process.wait(timeout=timeout)
- self.sox_process = None
- def close(self, *args):
- self.terminate_sox()
- self.lock_file.close()
- utils.destroy_lock()
- utils.unload_pa_modules()
- Gtk.main_quit()
|