def __init__(self, parent=None):
super(WacomGui, self).__init__(parent)
self.setupUi(self)
self.toggle = False
self.load = False
self.cwd = os.path.dirname(os.path.abspath(__file__))
if self.cwd == '/usr/local/bin':
self.cwd = '/usr/local/wacom-gui'
self.setFocusPolicy(Qt.NoFocus)
# button instances
self.tabletButtons = ButtonGroup()
self.toolButtons = ButtonGroup()
self.configButtons = ButtonGroup()
# button layouts
self.tabletLayout = ButtonLayout()
self.toolLayout = ButtonLayout()
self.configLayout = ButtonLayout()
self.tabletScroll.setWidget(self.tabletLayout.frame)
self.toolScroll.setWidget(self.toolLayout.frame)
self.configScroll.setWidget(self.configLayout.frame)
# config files
self.configs = {}
# hold current device info
self.dev = None
self.dev_id = None
self.config = None
# load widgets
self.pad = Pad()
self.pad.hide()
self.stylus = Stylus()
self.stylus.hide()
self.touch = Touch()
self.touch.hide()
# ui icon
self.tabletRefresh.setIcon(QIcon(os.path.join(self.cwd, os.path.join(self.cwd, 'icons/ui/refresh.png'))))
# get connected tablet info
self.tablet_data = Tablets()
# attach function to refresh tablets
self.tabletRefresh.clicked.connect(self.refreshTablets)
# add buttons for connected tablets
self.initTabletButtons()
# generate tool buttons
self.initToolButtons()
# add/remove button functions
self.addConfig.setEnabled(True)
self.addConfig.clicked.connect(self.newConfig)
self.removeConfig.clicked.connect(self.verifyConfigRemove)
# about/help menu
self.aboutButton.clicked.connect(About.display)
# refresh tablet list, set tools, configs
self.refreshTablets()
self.controlBox.setContentsMargins(0, 0, 0, 0)
# add control widgets to control box
hbox = QHBoxLayout()
hbox.setAlignment(Qt.AlignHCenter)
hbox.addWidget(self.pad)
hbox.addWidget(self.stylus)
hbox.addWidget(self.touch)
self.controlBox.setLayout(hbox)
# save config button
self.saveConfig.clicked.connect(self.updateConfigs)
# configure device reset button
self.deviceDefaults.clicked.connect(self.deviceReset)
# load first tool found
self.toolSelect(self.toolButtons.btn_grp.checkedId())
# init button functions
self.tabletButtons.btn_grp.buttonClicked['int'].connect(self.tabletSelect)
self.toolButtons.btn_grp.buttonClicked['int'].connect(self.toolSelect)
self.configButtons.btn_grp.buttonClicked['int'].connect(self.configSelect)
def toggleDisplay():
# get tablets
tablet_data = Tablets()
# get displays
displays = {}
cmd = "xdpyinfo | grep 'dimensions:'"
p = subprocess.Popen(cmd, shell=True, stderr=subprocess.PIPE, stdout=subprocess.PIPE)
output = p.communicate()[0]
pattern = r"\s+\S+\s+(\d+)x(\d+)"
full = re.match(pattern, output).groups()
displays['Full'] = {'cmd': "%sx%s+0+0" % (full[0], full[1]),
'x': full[0],
'y': full[1],
'xoff': 0,
'yoff': 0}
cmd = "xdpyinfo -ext all | grep head"
p = subprocess.Popen(cmd, shell=True, stderr=subprocess.PIPE, stdout=subprocess.PIPE)
output = p.communicate()[0]
pattern = r"\s+(\S+)\s+#(\d+):\s+(\d+)x(\d+)\D+(\d+),(\d+)"
heads = re.findall(pattern, output)
for head in heads:
info = list(head)
displays['%s-%s' % (info[0].upper(), info[1])] = {'cmd': '%sx%s+%s+%s' %
(info[2], info[3], info[4], info[5]),
'x': int(info[2]),
'y': int(info[3]),
'xoff': int(info[4]),
'yoff': int(info[5])}
displays['Partial...'] = {'cmd': "%sx%s+0+0" % (full[0], full[1]),
'x': full[0],
'y': full[1],
'xoff': 0,
'yoff': 0}
for dev in tablet_data.tablets.keys():
path = os.path.join(expanduser("~"), ".wacom-gui/%s" % dev)
config_name = None
if os.path.exists(os.path.join(path, 'device_default')):
with open(os.path.join(path, 'device_default'), 'r') as f:
config_name = json.load(f)
if config_name is not None:
for dev_id, data in enumerate(tablet_data.tablets[dev]):
if config_name[str(dev_id)]['toggle'] == 1:
if os.path.exists(os.path.join(path, "%s.json" % config_name[str(dev_id)]['config'])):
with open(os.path.join(path, "%s.json" % config_name[str(dev_id)]['config']), 'r') as f:
config = json.load(f)
if 'mapping' in config['stylus'].keys():
if 'maptooutput' in config['stylus']['mapping'].keys():
cur_display = config['stylus']['mapping']['maptooutput']
disp_list = sorted(displays.keys())
idx = disp_list.index(cur_display)
if idx + 1 <= displays.keys().__len__():
idx = idx + 1
if displays[disp_list[idx]]['cmd'] == displays[disp_list[0]]['cmd']:
idx = 0
else:
idx = 0
# update config
config['stylus']['mapping']['maptooutput'] = disp_list[idx]
conf_file = os.path.join(path, "%s.json" % config_name[str(dev_id)]['config'])
if os.path.exists(conf_file):
os.rename(conf_file, "%s.bak" % conf_file)
with open(conf_file, 'w') as outfile:
json.dump(config, outfile, sort_keys=True,
indent=4, separators=(',', ': '))
# update mapping
sid = tablet_data.tablets[dev][dev_id]['stylus']['id']
eid = tablet_data.tablets[dev][dev_id]['eraser']['id']
coords = displays[disp_list[idx]]['cmd']
cmd = "xsetwacom --set %s maptooutput %s" % (sid, coords)
os.popen(cmd)
cmd = "xsetwacom --set %s maptooutput %s" % (eid, coords)
os.popen(cmd)
sys.exit()
class WacomGui(QMainWindow, wacom_menu.Ui_MainWindow):
buttonClicked = pyqtSignal(int)
def __init__(self, parent=None):
super(WacomGui, self).__init__(parent)
self.setupUi(self)
self.toggle = False
self.load = False
self.cwd = os.path.dirname(os.path.abspath(__file__))
if self.cwd == '/usr/local/bin':
self.cwd = '/usr/local/wacom-gui'
self.setFocusPolicy(Qt.NoFocus)
# button instances
self.tabletButtons = ButtonGroup()
self.toolButtons = ButtonGroup()
self.configButtons = ButtonGroup()
# button layouts
self.tabletLayout = ButtonLayout()
self.toolLayout = ButtonLayout()
self.configLayout = ButtonLayout()
self.tabletScroll.setWidget(self.tabletLayout.frame)
self.toolScroll.setWidget(self.toolLayout.frame)
self.configScroll.setWidget(self.configLayout.frame)
# config files
self.configs = {}
# hold current device info
self.dev = None
self.dev_id = None
self.config = None
# load widgets
self.pad = Pad()
self.pad.hide()
self.stylus = Stylus()
self.stylus.hide()
self.touch = Touch()
self.touch.hide()
# ui icon
self.tabletRefresh.setIcon(QIcon(os.path.join(self.cwd, os.path.join(self.cwd, 'icons/ui/refresh.png'))))
# get connected tablet info
self.tablet_data = Tablets()
# attach function to refresh tablets
self.tabletRefresh.clicked.connect(self.refreshTablets)
# add buttons for connected tablets
self.initTabletButtons()
# generate tool buttons
self.initToolButtons()
# add/remove button functions
self.addConfig.setEnabled(True)
self.addConfig.clicked.connect(self.newConfig)
self.removeConfig.clicked.connect(self.verifyConfigRemove)
# about/help menu
self.aboutButton.clicked.connect(About.display)
# refresh tablet list, set tools, configs
self.refreshTablets()
self.controlBox.setContentsMargins(0, 0, 0, 0)
# add control widgets to control box
hbox = QHBoxLayout()
hbox.setAlignment(Qt.AlignHCenter)
hbox.addWidget(self.pad)
hbox.addWidget(self.stylus)
hbox.addWidget(self.touch)
self.controlBox.setLayout(hbox)
# save config button
self.saveConfig.clicked.connect(self.updateConfigs)
# configure device reset button
self.deviceDefaults.clicked.connect(self.deviceReset)
# load first tool found
self.toolSelect(self.toolButtons.btn_grp.checkedId())
# init button functions
self.tabletButtons.btn_grp.buttonClicked['int'].connect(self.tabletSelect)
self.toolButtons.btn_grp.buttonClicked['int'].connect(self.toolSelect)
self.configButtons.btn_grp.buttonClicked['int'].connect(self.configSelect)
def initTabletButtons(self):
for dev, data in self.tablet_data.tablets.items():
for dev_id, tablet in enumerate(data):
icon = os.path.join(self.cwd, "icons/devices/%spng" % tablet['svg'][:-3])
if not os.path.isfile(os.path.join(os.getcwd(), icon)):
icon = os.path.join(self.cwd, 'icons/devices/generic.png')
self.tabletLayout.addButton(self.tabletButtons.addButton(tablet['cname'], tablet['pad']['id'],
str(dev), dev_id, icon))
def refreshTablets(self):
self.tablet_data.get_connected_tablets()
# clear current tablets from layout
self.tabletLayout.removeButtons()
self.tabletButtons.removeButton()
self.initTabletButtons()
if self.tabletButtons.buttons.items().__len__() == 0:
self.toolButtons.hideButtons()
else:
self.tabletSelect(0)
def initToolButtons(self):
# Functions/Stylus/Touch/Cursor
self.toolLayout.addButton(self.toolButtons.addButton("Functions", 0, 0, 0, os.path.join(self.cwd, 'icons/ui/functions.png'), 48, True))
self.toolLayout.addButton(self.toolButtons.addButton("Stylus", 0, 0, 0, os.path.join(self.cwd, 'icons/ui/stylus.png'), 48, True))
self.toolLayout.addButton(self.toolButtons.addButton("Touch", 0, 0, 0, os.path.join(self.cwd, 'icons/ui/touch.png'), 48, True))
self.toolLayout.addButton(self.toolButtons.addButton("Mouse", 0, 0, 0, os.path.join(self.cwd, 'icons/ui/mouse.png'), 48, True))
def setToolsAvail(self, idx):
self.dev = self.tabletButtons.buttons[(idx, 1)]
self.dev_id = self.tabletButtons.buttons[(idx, 3)]
if 'pad' in self.tablet_data.tablets[self.dev][self.dev_id].keys():
self.toolButtons.buttons[(0, 0)].setVisible(True)
self.toolButtons.buttons[(0, 1)] = self.dev
self.toolButtons.buttons[(0, 2)] = self.tablet_data.tablets[self.dev][self.dev_id]['pad']['id']
self.toolButtons.buttons[(0, 3)] = self.dev_id
else:
self.toolButtons.buttons[(0, 0)].setVisible(False)
self.toolButtons.buttons[(0, 1)] = 0
self.toolButtons.buttons[(0, 2)] = 0
self.toolButtons.buttons[(0, 3)] = 0
if 'stylus' in self.tablet_data.tablets[self.dev][self.dev_id].keys():
self.toolButtons.buttons[(1, 0)].setVisible(True)
self.toolButtons.buttons[(1, 1)] = self.dev
self.toolButtons.buttons[(1, 2)] = [self.tablet_data.tablets[self.dev][self.dev_id]['stylus']['id'],
self.tablet_data.tablets[self.dev][self.dev_id]['eraser']['id']]
self.toolButtons.buttons[(1, 3)] = self.dev_id
else:
self.toolButtons.buttons[(1, 0)].setVisible(False)
self.toolButtons.buttons[(1, 1)] = 0
self.toolButtons.buttons[(1, 2)] = 0
self.toolButtons.buttons[(1, 3)] = 0
if 'touch' in self.tablet_data.tablets[self.dev][self.dev_id].keys():
self.toolButtons.buttons[(2, 0)].setVisible(True)
self.toolButtons.buttons[(2, 1)] = self.dev
self.toolButtons.buttons[(2, 2)] = self.tablet_data.tablets[self.dev][self.dev_id]['touch']['id']
self.toolButtons.buttons[(2, 3)] = self.dev_id
else:
self.toolButtons.buttons[(2, 0)].setVisible(False)
self.toolButtons.buttons[(2, 1)] = 0
self.toolButtons.buttons[(2, 2)] = 0
self.toolButtons.buttons[(2, 3)] = 0
if 'cursor' in self.tablet_data.tablets[self.dev][self.dev_id].keys():
self.toolButtons.buttons[(3, 0)].setVisible(True)
self.toolButtons.buttons[(3, 1)] = self.dev
self.toolButtons.buttons[(3, 2)] = self.tablet_data.tablets[self.dev][self.dev_id]['cursor']['id']
self.toolButtons.buttons[(3, 3)] = self.dev_id
else:
self.toolButtons.buttons[(3, 0)].setVisible(False)
self.toolButtons.buttons[(3, 1)] = 0
self.toolButtons.buttons[(3, 2)] = 0
self.toolButtons.buttons[(3, 3)] = 0
def deviceReset(self):
self.pad.set_default()
self.stylus.resetPen()
self.stylus.resetEraser()
self.stylus.mapping.resetMapping()
self.touch.reset()
# TODO: add other resets
def emptyConfig(self, dev, dev_id):
config = {}
for input in self.tablet_data.tablets[dev][dev_id].keys():
if input in ['pad', 'stylus', 'eraser', 'touch', 'cursor']:
config[input] = {}
if input in ['pad', 'stylus', 'eraser']:
config[input]['buttons'] = {}
if input == 'pad':
buttons = self.tablet_data.tablets[dev][dev_id][input]['buttons']
for button in buttons.keys():
config['pad']['buttons'][button] = 'Default'
return config
def verifyConfigRemove(self):
reply = QMessageBox.question(self, 'Remove Config',
"Delete \"%s\" config file?" % self.config,
QMessageBox.Yes, QMessageBox.No)
if reply == QMessageBox.Yes:
home = "%s/.wacom-gui" % expanduser("~")
conf_path = os.path.join(home, "%s/%s.json" % (self.dev, self.config))
# delete file
try:
os.remove(conf_path)
except Exception as e:
print e
del self.configs[self.dev][self.config]
self.getConfigs(0)
def newConfig(self):
config = AddConfig.add_config(self.configs[self.dev].keys())
if config is not None:
# empty config
self.configs[self.dev][config] = self.emptyConfig(self.dev, self.dev_id)
# add button
self.configLayout.addButton(
self.configButtons.addButton(config, 0, 0, 0, os.path.join(self.cwd, 'icons/ui/config.png'), 48))
self.config = config
self.loadConfig(self.dev, self.dev_id, self.config)
for idx, button in enumerate(self.configButtons.btn_grp.buttons()):
if button.text() == self.config:
self.configButtons.btn_grp.buttons()[idx].setChecked(True)
self.tablet_data.tablets[self.dev][self.dev_id]['config'] = self.config
self.removeConfig.setEnabled(True)
def getConfigs(self, idx):
self.configLayout.removeButtons()
self.configButtons.removeButton()
dev = self.tabletButtons.buttons[(idx, 1)]
dev_id = self.tabletButtons.buttons[(idx, 3)]
home = "%s/.wacom-gui" % expanduser("~")
conf_path = os.path.join(home, dev)
self.tablet_data.tablets[dev][dev_id]['conf_path'] = os.path.join(home, dev)
if dev not in self.configs.keys():
self.configs[dev] = {}
if os.path.exists(conf_path):
# get configs in path
for config in os.listdir(conf_path):
if os.path.isfile(os.path.join(conf_path, config)) and config.endswith(".json"):
with open(os.path.join(conf_path, config), 'r') as f:
self.configs[dev][os.path.splitext(config)[0]] = json.load(f)
# TODO: allow user to sent alternate config to load by default
# load default config, if it exists
if 'default' not in self.configs[dev].keys():
self.configs[dev]['default'] = self.emptyConfig(dev, dev_id)
# get default config to load for given device, ie last selected config
self.config = 'default'
if os.path.exists("%s/device_default" % conf_path):
with open("%s/device_default" % conf_path, 'r') as f:
device = json.load(f)
# config file exists, use it
if str(dev_id) in device.keys() and device[str(dev_id)]['config'] in self.configs[dev].keys():
self.config = device[str(dev_id)]['config']
# set defaults in tablet_data, if it's detected
for idx in device.keys():
try:
self.tablet_data.tablets[self.dev][int(idx)]['config'] = device[idx]['config']
except Exception as e:
pass
self.configLayout.addButton(
self.configButtons.addButton("default", 0, 0, 0, os.path.join(self.cwd, 'icons/ui/config.png'), 48))
for config in sorted(self.configs[dev].keys()):
if config != 'default':
self.configLayout.addButton(
self.configButtons.addButton(config, 0, 0, 0, os.path.join(self.cwd, 'icons/ui/config.png'), 48))
# we are loading default config for now...
if self.config == 'default':
self.removeConfig.setEnabled(False)
else:
self.removeConfig.setEnabled(True)
self.loadConfig(dev, dev_id, self.config)
for idx, button in enumerate(self.configButtons.btn_grp.buttons()):
if button.text() == self.config:
self.configButtons.btn_grp.buttons()[idx].setChecked(True)
self.tablet_data.tablets[dev][dev_id]['config'] = self.config
else:
os.mkdir(self.tablet_data.tablets[dev][dev_id]['conf_path'])
self.configLayout.addButton(
self.configButtons.addButton("default", 0, 0, 0, os.path.join(self.cwd, 'icons/ui/config.png'), 48))
def loadConfig(self, dev, dev_id, config):
# TODO: load cursor configs
# load pad buttons
self.config = config
self.tablet_data.tablets[dev][dev_id]['config'] = self.config
if not self.toolButtons.buttons[(0, 0)].isHidden():
self.pad.init_keys(self.tablet_data.tablets[dev][dev_id]['pad']['id'],
self.tablet_data.tablets[dev][dev_id]['svg'],
self.tablet_data.tablets[dev][dev_id]['pad']['buttons'],
self.configs[dev][config]['pad']['buttons'])
if not self.toolButtons.buttons[(1, 0)].isHidden():
self.stylus.init_pen(self.tablet_data.tablets[dev][dev_id]['stylus']['id'],
self.configs[dev][config]['stylus'])
self.stylus.init_eraser(self.tablet_data.tablets[dev][dev_id]['eraser']['id'],
self.configs[dev][config]['eraser'])
if self.tablet_data.tablets[dev][dev_id]['stylus']['rotate'] == False:
# turn off rotate, if device doesn't support it
if 'mapping' in self.configs[dev][config]['stylus']:
self.configs[dev][config]['stylus']['mapping']['rotate'] = 'False'
else:
self.configs[dev][config]['stylus']['mapping'] = {'rotate': 'False'}
self.stylus.mapping.initUI(self.tablet_data.tablets[dev][dev_id]['stylus']['id'],
self.tablet_data.tablets[dev][dev_id]['eraser']['id'],
self.configs[dev][config]['stylus'])
if not self.toolButtons.buttons[(2, 0)].isHidden():
self.touch.init(self.tablet_data.tablets[dev][dev_id]['touch']['id'],
self.configs[dev][config]['touch'])
def tabletSelect(self, idx):
self.updateConfigs()
self.setToolsAvail(idx)
self.getConfigs(idx)
# set first available tool as selected
if not self.toolButtons.buttons[(0, 0)].isHidden():
self.toolButtons.buttons[(0, 0)].setChecked(True)
elif not self.toolButtons.buttons[(1, 0)].isHidden():
self.toolButtons.buttons[(1, 0)].setChecked(True)
elif not self.toolButtons.buttons[(2, 0)].isHidden():
self.toolButtons.buttons[(2, 0)].setChecked(True)
elif not self.toolButtons.buttons[(3, 0)].isHidden():
self.toolButtons.buttons[(3, 0)].setChecked(True)
def toolSelect(self, idx):
if idx == 0 and not self.toolButtons.buttons[(idx, 0)].isHidden():
# remove previous layout
self.pad.show()
self.stylus.hide()
self.touch.hide()
# stylus menu
elif idx == 1 and not self.toolButtons.buttons[(idx, 0)].isHidden():
# remove previous layout
self.pad.hide()
self.stylus.show()
self.touch.hide()
# touch menu
elif idx == 2 and not self.toolButtons.buttons[(idx, 0)].isHidden():
# remove previous layout
self.pad.hide()
self.stylus.hide()
self.touch.show()
# cursor menu
elif idx == 3 and not self.toolButtons.buttons[(idx, 0)].isHidden():
# remove previous layout
self.pad.hide()
self.stylus.hide()
self.touch.hide()
def configSelect(self, idx):
config = str(self.configButtons.buttons[(idx, 0)].text())
self.loadConfig(self.dev, self.dev_id, config)
if config == 'default':
self.removeConfig.setEnabled(False)
else:
self.removeConfig.setEnabled(True)
def deleteItemsOfLayout(self, layout):
if layout is not None:
while layout.count():
item = layout.takeAt(0)
widget = item.widget()
if widget is not None:
widget.setParent(None)
else:
self.deleteItemsOfLayout(item.layout())
def boxdelete(self, box):
for i in range(self.keys.count()):
layout_item = self.keys.itemAt(i)
if layout_item.layout() == box:
self.deleteItemsOfLayout(layout_item.layout())
self.vlayout.removeItem(layout_item)
break
def updateConfigs(self):
write = False
if not self.toolButtons.buttons[(0, 0)].isHidden():
pad = self.pad.get_config()
if pad != self.configs[self.dev][self.config]['pad']['buttons']:
write = True
self.configs[self.dev][self.config]['pad']['buttons'] = pad
if not self.toolButtons.buttons[(1, 0)].isHidden():
stylus = self.stylus.get_config()
# if stylus['stylus'] != self.configs[self.dev][self.config]['stylus']:
write = True
self.configs[self.dev][self.config]['stylus'] = stylus['stylus']
# if stylus['eraser'] != self.configs[self.dev][self.config]['eraser']:
write = True
self.configs[self.dev][self.config]['eraser'] = stylus['eraser']
if not self.toolButtons.buttons[(2, 0)].isHidden():
touch = self.touch.get_config()
if touch != self.configs[self.dev][self.config]['touch']:
write = True
self.configs[self.dev][self.config]['touch'] = touch
# TODO: cursor
if not self.toolButtons.buttons[(3, 0)].isHidden():
tmp = 1
if write:
reply = None
if self.toggle is False and self.load is False:
reply = QMessageBox.question(self, 'Save Config',
"Write \"%s\" config file?" % self.config,
QMessageBox.Yes, QMessageBox.No)
# update config if toggle is set or save is called
if self.toggle or reply == QMessageBox.Yes:
home = "%s/.wacom-gui" % expanduser("~")
conf_path = os.path.join(home, self.dev)
conf_file = os.path.join(conf_path, "%s.json" % self.config)
if not os.path.exists(conf_path):
os.mkdir(conf_path)
elif os.path.exists(conf_file):
os.rename(conf_file, "%s.bak" % conf_file)
with open(conf_file, 'w') as outfile:
json.dump(self.configs[self.dev][self.config], outfile, sort_keys=True,
indent=4, separators=(',', ': '))
if os.path.exists(conf_file):
if os.path.exists("%s.bak" % conf_file):
os.remove("%s.bak" % conf_file)
# check if we need to update what config is loaded
default = {}
if self.dev is not None:
for idx, tablet in enumerate(self.tablet_data.tablets[self.dev]):
if 'config' in tablet.keys():
# check if toggle enabled
toggle = 0
if 'pad' in self.configs[self.dev][tablet['config']]:
for button in self.configs[self.dev][tablet['config']]['pad']['buttons'].keys():
if self.configs[self.dev][tablet['config']]['pad']['buttons'][button] == 'lhyper z':
toggle = 1
break
default[idx] = {'config': tablet['config'],
'toggle': toggle}
else:
default[idx] = {'config': 'default',
'toggle': 0}
conf_path = os.path.join("%s/.wacom-gui" % expanduser("~"), self.dev)
conf_file = os.path.join(conf_path, 'device_default')
with open(conf_file, 'w') as outfile:
json.dump(default, outfile, sort_keys=True,
indent=4, separators=(',', ': '))
def quickLoad(self):
self.load = True
for idx, button in enumerate(self.tabletButtons.btn_grp.buttons()):
self.tabletSelect(idx)
sys.exit()
def closeEvent(self, event):
self.updateConfigs()
event.accept()
self.deleteLater()