X-Git-Url: https://git.ralfj.de/lilass.git/blobdiff_plain/6f78b3a6bb1715b8c742a124f696b6bb8b36751a..11d554b0e9fa2fa36f3949424018d652ff994400:/external_screen.py

diff --git a/external_screen.py b/external_screen.py
index e365ef5..9d70b82 100755
--- a/external_screen.py
+++ b/external_screen.py
@@ -10,46 +10,82 @@ def getXrandrInformation():
 	connector = None # current connector
 	for line in p.stdout:
 		# new connector?
-		m = re.search('^([\w]+) connected ', line)
+		m = re.search(r'^([\w]+) connected ', line)
 		if m is not None:
 			connector = m.groups()[0]
 			assert connector not in connectors
 			connectors[connector] = []
 			continue
 		# new resolution?
-		m = re.search('^   ([\d]+)x([\d]+) +', line)
+		m = re.search(r'^   ([\d]+)x([\d]+) +', line)
 		if m is not None:
 			assert connector is not None
 			connectors[connector].append((int(m.groups()[0]), int(m.groups()[1])))
+	p.communicate()
+	if p.returncode != 0: raise Exception("Querying xrandr for data failed")
 	return connectors
 
-def res2str(res):
+def res2xrandr(res):
 	(w, h) = res
 	return str(w)+'x'+str(h)
 
+def res2user(res):
+	(w, h) = res
+	# get ratio
+	ratio = int(round(16.0*h/w))
+	if ratio == 12: # 16:12 = 4:3
+		strRatio = '4:3'
+	elif ratio == 13: # 16:12.8 = 5:4
+		strRatio = '5:4'
+	else: # let's just hope this will never be 14 or more...
+		strRatio = '16:%d' % ratio
+	return '%dx%d (%s)' %(w, h, strRatio)
+
 # Check screen setup
-internalName = "LVDS"
-externalName = "CRT1"
+internalName = "LVDS1"
+externalNames = ["HDMI1", "DP1", "VGA1"]
 connectors = getXrandrInformation()
 internalResolutions = connectors[internalName] # there must be a screen assoicated to the internal connector
-externalResolutions = connectors.get(externalName)
+externalName = None # *the* external connector which is actually used
+externalResolutions = None # resultions of the external connector
+args = {} # maps connector names to xrand arguments
+
+# look for enabled external screen, disable all the others
+for name in externalNames:
+	if externalResolutions is None and name in connectors:
+		externalName = name
+		externalResolutions = connectors[name]
+	else:
+		args[name] = ["--off"]
 
 # Check what to do
-internalArgs = ["--mode", res2str(internalResolutions[0])] # there must be a resolution for the internal screen
-externalArgs = ["--off"]
-if externalResolutions is not None: # we need to ask what to do
-	extPosition = PositionSelection(map(res2str, externalResolutions))
+if externalName is not None: # we need to ask what to do
+	extPosition = PositionSelection(externalName, map(res2user, internalResolutions), map(res2user, externalResolutions))
 	extPosition.exec_()
 	if not extPosition.result(): sys.exit(1) # the user canceled
+	extResolution = res2xrandr(externalResolutions[extPosition.extResolutions.currentIndex()])
+	intResolution = res2xrandr(internalResolutions[extPosition.intResolutions.currentIndex()])
 	# build command-line
-	externalArgs = ["--mode", extPosition.resolution] # we definitely want an external screen
-	if extPosition.position == PositionSelection.EXTERNAL_ONLY:
-		internalArgs = ["--off"]
-	elif extPosition.position == PositionSelection.LEFT:
-		externalArgs += ["--left-of", internalName]
+	args[externalName] = ["--mode", extResolution] # we definitely want an external screen
+	if extPosition.extOnly.isChecked():
+		args[internalName] = ["--off"]
+		args[externalName] += ["--primary"]
 	else:
-		externalArgs += ["--right-of", internalName]
+		# there are two screens
+		args[internalName] = ["--mode", intResolution]
+		if extPosition.posLeft.isChecked():
+			args[externalName] += ["--left-of", internalName]
+		else:
+			args[externalName] += ["--right-of", internalName]
+		if extPosition.primExt.isChecked():
+			args[externalName] += ["--primary"]
+		else:
+			args[internalName] += ["--primary"]
+else:
+	args[internalName] = ["--mode", res2xrandr(internalResolutions[0]), "--primary"]
 # and do it
-args = ["--output", internalName] + internalArgs + ["--output", externalName] + externalArgs
-print args
-subprocess.check_call(["xrandr"] + args)
+call = ["xrandr"]
+for name in args:
+	call += ["--output", name] + args[name]
+print "Call that will be made:",call
+subprocess.check_call(call)