general_connector.py

#!/usr/bin/env python

import pygtk
pygtk.require('2.0')
import gtk
import getopt
import sys
import liblo
import math
import threading
import time

gtk.gdk.threads_init()

def db_to_coeff(db):
	if db > -318.8:
		return math.pow(10.0, db * 0.05)
	else:
		return 0.0

def coeff_to_db(coeff):
	if coeff == 0.0:
		return float('-inf')
	else:
		return 20.0 * math.log10(coeff)

def fader_to_db(fader):
	if fader == 0.0:
		return float('-inf')
	log = math.log10(fader * 1.3422)
	if fader > 0.745:
		return 78.25 * log
	else:
		return 93.44 * log

def db_to_fader(db):
	if db > -318.8:
		if db > 0.0:
			return 0.745 * math.pow(10.0, db*0.01277955272)
		else:
			return 0.745 * math.pow(10.0, db*0.01070205479)
	else:
		return 0.0

def scroll2str(scroll):
	if scroll == gtk.SCROLL_NONE:
		return "SCROLL_NONE"
	elif scroll == gtk.SCROLL_JUMP:
		return "SCROLL_JUMP"
	elif scroll == gtk.SCROLL_STEP_BACKWARD:
		return "SCROLL_STEP_BACKWARD"
	elif scroll == gtk.SCROLL_STEP_FORWARD:
		return "SCROLL_STEP_FORWARD"
	elif scroll == gtk.SCROLL_PAGE_BACKWARD:
		return "SCROLL_PAGE_BACKWARD"
	elif scroll == gtk.SCROLL_PAGE_FORWARD:
		return "SCROLL_PAGE_FORWARD"
	elif scroll == gtk.SCROLL_STEP_UP:
		return "SCROLL_STEP_UP"
	elif scroll == gtk.SCROLL_STEP_DOWN:
		return "SCROLL_STEP_DOWN"
	elif scroll == gtk.SCROLL_PAGE_UP:
		return "SCROLL_PAGE_UP"
	elif scroll == gtk.SCROLL_PAGE_DOWN:
		return "SCROLL_PAGE_DOWN"
	elif scroll == gtk.SCROLL_STEP_LEFT:
		return "SCROLL_STEP_LEFT"
	elif scroll == gtk.SCROLL_STEP_RIGHT:
		return "SCROLL_STEP_RIGHT"
	elif scroll == gtk.SCROLL_PAGE_LEFT:
		return "SCROLL_PAGE_LEFT"
	elif scroll == gtk.SCROLL_PAGE_RIGHT:
		return "SCROLL_PAGE_RIGHT"
	elif scroll == gtk.SCROLL_START:
		return "SCROLL_START"
	elif scroll == gtk.SCROLL_END:
		return "SCROLL_END"


class VolumeOSC():
	def __init__(self, gui, host, port, instance, external_instance, listen_port):
#		threading.Thread.__init__(self)
		self.daemon = True
		self.gui = gui
		self.address = liblo.Address(host, port, protocol)
		self.server_active = False
		try:
			searching_port = True
			ntrials = 0
			while searching_port:
				try:
					self.server = liblo.Server(local_port)
					searching_port = False
				except liblo.ServerError, err:
					local_port = local_port + 1
					ntrials = ntrials + 1
					if ntrials > 1000:
						raise liblo.ServerError(99, "stop searching free port", None)

			self.server.add_method("/net/mhcloud/volume/" + instance + "/master", "f", self.callback_master_gain)
			self.server.add_method("/net/mhcloud/volume/" + instance + "/master/mute", "i", self.callback_master_mute)
			for i in range(channels):
				self.server.add_method("/net/mhcloud/volume/" + instance + "/" + str(i), "f", self.callback_channel_gain, i)
				self.server.add_method("/net/mhcloud/volume/" + instance + "/" + str(i) + "/mute", "i", self.callback_channel_mute, i)
			self.server_active = True
		except liblo.ServerError, err:
			sys.stderr.write("OSC server error occured:\n")
			sys.stderr.write(str(err) + '\n')

	def run(self):
		while self.server_active:
			self.server.recv(100)

	def callback_master_gain(self, path, args):
		gtk.threads_enter()
		self.gui.fader_event_safe(-1, args[0], False)
		gtk.threads_leave()

	def callback_master_mute(self, path, args):
		gtk.threads_enter()
		self.gui.mute_event_safe(-1, args[0]!=0, False)
		gtk.threads_leave()

	def callback_channel_gain(self, path, args, types, src, data):
		gtk.threads_enter()
		self.gui.fader_event_safe(data, args[0], False)
		gtk.threads_leave()

	def callback_channel_mute(self, path, args, types, src, data):
		gtk.threads_enter()
		self.gui.mute_event_safe(data, args[0]!=0, False)
		gtk.threads_leave()

	def send_master(self, val):
		liblo.send(self.address, "/net/mhcloud/volume/" + instance + "/master", val)

	def send_channel(self, channel, val):
		liblo.send(self.address, "/net/mhcloud/volume/" + instance + "/" + str(channel), val)

	def send_master_mute(self, mute):
		liblo.send(self.address, "/net/mhcloud/volume/" + instance + "/master/mute", int(mute))

	def send_channel_mute(self, channel, mute):
		liblo.send(self.address, "/net/mhcloud/volume/" + instance + "/" + str(channel) + "/mute", int(mute))

class VolumeConnector:

	def delete_event(self, widget, event, data=None):
		print "delete event occurred"
		return False

	def destroy(self, widget, data=None):
		print "destroy signal occurred"
		gtk.main_quit()

	def send_master_osc(self, value):
		self.osc.send_master(value)

	def send_channel_osc(self, channel, value):
		self.osc.send_channel(channel, value)

	def send_master_mute(self, mute):
		self.osc.send_master_mute(mute)

	def send_channel_mute(self, channel, mute):
		self.osc.send_channel_mute(channel, mute)

	def mute_event(self, ch, mute, send_osc):
		self.set_mute(ch, mute)
	
		if send_osc:
			if ch == self.nchannels-1:
				self.send_master_mute(self.muted[ch])
			else:
				self.send_channel_mute(ch, self.muted[ch])

	def mute_event_safe(self, ch, mute, send_osc):
		self.lock.acquire()
		self.mute_event(ch, mute, send_osc)
		self.lock.release()

	def fader_event(self, ch, gain_abs, send_osc):
		if send_osc:
			if ch == self.nchannels-1:
				self.send_master_osc(gain_abs)
			else:
				self.send_channel_osc(ch, gain_abs)
#		print "fader_event ch=" + str(ch) + " gain=" + str(gain_abs)
		gain_db = coeff_to_db(gain_abs)
		if ch == -1:
			ch = self.nchannels-1
#		print "fader_event ch=" + str(ch) + " gain=" + str(gain_abs)
		self.vscales[ch].set_value(db_to_fader(gain_db))
		self.dbs[ch].set_text(str(gain_db))

	def fader_event_safe(self, ch, gain_abs, send_osc):
		self.lock.acquire()
		self.fader_event(ch, gain_abs, send_osc)
		self.lock.release()

	def scale_event(self, scale, scroll, value):
		#print "scroll=" + scroll2str(scroll)
		if scroll == gtk.SCROLL_STEP_FORWARD:
			value += 0.01
		elif scroll == gtk.SCROLL_STEP_BACKWARD:
			value -= 0.01
		elif scroll == gtk.SCROLL_PAGE_FORWARD:
			value += 0.1
		elif scroll == gtk.SCROLL_PAGE_BACKWARD:
			value -= 0.1
		value = min(value, 1.0)
		value = max(value, 0.0)

		channel = 0
		for i, val in enumerate(self.vscales):
			if scale == self.vscales[i]:
				#print "scale found"
				channel = i
		self.fader_event_safe(channel, db_to_coeff(fader_to_db(value)), True)
		return True

	def activate_event(self, entry):
		gain_db = float(entry.get_text().replace(',', '.'))
		gain_db = min(gain_db, 10.0)

		channel = 0
		for i, item in enumerate(self.dbs):
			if entry == self.dbs[i]:
				channel = i

		gain_abs = db_to_coeff(gain_db)
		self.fader_event_safe(channel, gain_abs, True)

	def click_mute(self, button):
		channel = 0
		for i, item in enumerate(self.mutes):
			if button == self.mutes[i]:
				channel = i

		self.mute_event_safe(channel, not self.muted[channel], True)

	def set_mute(self, channel, mute):
		if (self.muted[channel] != mute):
			self.muted[channel] = mute
			if mute:
				self.mutes[channel].modify_bg(gtk.STATE_NORMAL, gtk.gdk.Color(65535,0,0))
				self.mutes[channel].modify_bg(gtk.STATE_ACTIVE, gtk.gdk.Color(55535,0,0))
				self.mutes[channel].modify_bg(gtk.STATE_PRELIGHT, gtk.gdk.Color(65535,0,0))
			else:
				self.mutes[channel].modify_bg(gtk.STATE_NORMAL, gtk.gdk.Color(0,65535,0))
				self.mutes[channel].modify_bg(gtk.STATE_ACTIVE, gtk.gdk.Color(0,55535,0))
				self.mutes[channel].modify_bg(gtk.STATE_PRELIGHT, gtk.gdk.Color(0,65535,0))

	def __init__(self, nchannels, host, port, instance, external_instance, local_port):
		self.lock = threading.Lock()
		self.osc = VolumeOSC(self, host, port, instance, external_instance, local_port)
		self.nchannels = nchannels
		print "channels=" + str(self.nchannels)

		if self.nchannels == 1:
			self.nchannels = 0

		self.nchannels += 1
		self.vscales = [None]*self.nchannels
		self.dbs = [None]*self.nchannels
		self.mutes = [None]*self.nchannels
		self.muted = [None]*self.nchannels


		for i, val in enumerate(self.vscales):
			self.vscales[i] = gtk.VScale()
			self.vscales[i].set_digits(2)
			self.vscales[i].set_draw_value(False)
			self.vscales[i].set_range(0, 1)
			self.vscales[i].set_value(db_to_fader(0.0))
			self.vscales[i].set_inverted(True)
#			self.vscales[i].add_mark(100, gtk.POS_RIGHT, "100")
#			self.vscales[i].add_mark(75, gtk.POS_RIGHT, "75")
#			self.vscales[i].add_mark(50, gtk.POS_RIGHT, "50")
#			self.vscales[i].add_mark(25, gtk.POS_RIGHT, "25")
#			self.vscales[i].add_mark(0, gtk.POS_RIGHT, "0")
			self.vscales[i].connect("change-value", self.scale_event)
			self.table.attach(self.vscales[i], i, i+1, 2, 3)
			self.vscales[i].show()

			self.mutes[i] = gtk.Button("Mute")
			self.set_mute(i, False)
#			self.send_channel_mute(i, False)
			self.mutes[i].connect("clicked", self.click_mute)
			self.table.attach(self.mutes[i], i, i+1, 0, 1, gtk.SHRINK, gtk.SHRINK)
			self.mutes[i].show()

			self.dbs[i] = gtk.Entry(0)
			self.dbs[i].set_text("0.0")
			self.dbs[i].set_width_chars(4)
			self.dbs[i].connect("activate", self.activate_event)
			self.dbs[i].show()
			self.table.attach(self.dbs[i], i, i+1, 1, 2, gtk.SHRINK, gtk.SHRINK)
			if (i != self.nchannels-1):
#				self.send_channel_osc(i, db_to_coeff(0.0))
				label = gtk.Label("Channel "+str(i+1))
				self.table.attach(label, i, i+1, 3, 4, gtk.SHRINK, gtk.FILL)
				label.show()

	def run(self):
		self.osc.run()

	def get_nchannels(self):
		return self.nchannels


if __name__ == "__main__":
	channels = 2
	port = 7600
	host = "localhost"
	instance = "jack-volume"
	external_instance = "touchosc"
	local_port = 7601

	try:
		opts, args = getopt.getopt(sys.argv[1:], "c:p:h:j:e:s:")
	except getopt.GetoptError as err:
		sys.stderr.write(str(err) + '\n')
		sys.stderr.write("usage: " + sys.argv[0] + " [-c <nchannels>] [-p <jv-port>] [-h <host>] [-j <jack-volume_instance>] [-e <external-name>] [-s <listening-port>]\n")
		sys.stderr.write("options:\n")
		sys.stderr.write("-c  number of channels\n")
		sys.stderr.write("-p  udp port of jack-volume\n")
		sys.stderr.write("-h  jack-volume hostname\n")
		sys.stderr.write("-j  name of the jack-volume instance\n")
		sys.stderr.write("-e  name of the external instance\n")
#		sys.stderr.write("-q  udp port of the external instance\n")
		sys.stderr.write("-s  OSC UDP listening port\n")
		sys.exit(2)
	for k, v in opts:
		if k == "-c":
			channels = v
		elif k == "-p":
			port = int(v)
			local_port = port + 1
		elif k == "-h":
			host = v
		elif k == "-j":
			instance = v
		elif k == "-e":
			external_instance = v
		elif k == "-s":
			local_port = int(v)
	channels = int(channels)
	channels = min(channels, 32)
	channels = max(channels, 1)
	try:
		ctrl = VolumeConnector(channels, host, port, instance, external_instance, local_port)
		ctrl.run()
	except IOError:
		sys.stderr.write("IOError occured:\n")
		sys.stderr.write(str(IOError) + '\n')