external_screen.py

   1 #!/usr/bin/python
   2 import sys, re, subprocess
   3 from PyQt4 import QtGui
   4 from selector_window import PositionSelection
   5 app = QtGui.QApplication(sys.argv)
   6
   7 def getXrandrInformation():
   8         p = subprocess.Popen(["xrandr", "-q"], stdout=subprocess.PIPE)
   9         connectors = {} # map of connector names to a list of resolutions
  10         connector = None # current connector
  11         for line in p.stdout:
  12                 # new connector?
  13                 m = re.search(r'^([\w]+) connected ', line)
  14                 if m is not None:
  15                         connector = m.groups()[0]
  16                         assert connector not in connectors
  17                         connectors[connector] = []
  18                         continue
  19                 # new resolution?
  20                 m = re.search(r'^   ([\d]+)x([\d]+) +', line)
  21                 if m is not None:
  22                         assert connector is not None
  23                         connectors[connector].append((int(m.groups()[0]), int(m.groups()[1])))
  24         p.communicate()
  25         if p.returncode != 0: raise Exception("Querying xrandr for data failed")
  26         return connectors
  27
  28 def res2xrandr(res):
  29         (w, h) = res
  30         return str(w)+'x'+str(h)
  31
  32 def res2user(res):
  33         (w, h) = res
  34         # get ratio
  35         ratio = int(round(16.0*h/w))
  36         if ratio == 12: # 16:12 = 4:3
  37                 strRatio = '4:3'
  38         elif ratio == 13: # 16:12.8 = 5:4
  39                 strRatio = '5:4'
  40         else: # let's just hope this will never be 14 or more...
  41                 strRatio = '16:%d' % ratio
  42         return '%dx%d (%s)' %(w, h, strRatio)
  43
  44 # Check screen setup
  45 internalName = "LVDS"
  46 externalName = "CRT1"
  47 connectors = getXrandrInformation()
  48 internalResolutions = connectors[internalName] # there must be a screen assoicated to the internal connector
  49 externalResolutions = connectors.get(externalName)
  50
  51 # Check what to do
  52 if externalResolutions is not None: # we need to ask what to do
  53         extPosition = PositionSelection(map(res2user, internalResolutions), map(res2user, externalResolutions))
  54         extPosition.exec_()
  55         if not extPosition.result(): sys.exit(1) # the user canceled
  56         extResolution = res2xrandr(externalResolutions[extPosition.extResolutions.currentIndex()])
  57         intResolution = res2xrandr(internalResolutions[extPosition.intResolutions.currentIndex()])
  58         # build command-line
  59         externalArgs = ["--mode", extResolution] # we definitely want an external screen
  60         if extPosition.extOnly.isChecked():
  61                 internalArgs = ["--off"]
  62                 externalArgs += ["--primary"]
  63         else:
  64                 # there are two screens
  65                 internalArgs = ["--mode", intResolution]
  66                 if extPosition.posLeft.isChecked():
  67                         externalArgs += ["--left-of", internalName]
  68                 else:
  69                         externalArgs += ["--right-of", internalName]
  70                 if extPosition.primExt.isChecked():
  71                         externalArgs += ["--primary"]
  72                 else:
  73                         internalArgs += ["--primary"]
  74 else:
  75         internalArgs = ["--mode", res2str(internalResolutions[0]), "--primary"]
  76         externalArgs = ["--off"]
  77 # and do it
  78 call = ["xrandr", "--output", internalName] + internalArgs + ["--output", externalName] + externalArgs
  79 print "Call that will be made:",call
  80 subprocess.check_call(call)