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('^([\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('^   ([\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         return connectors
  25
  26 def res2str(res):
  27         (w, h) = res
  28         return str(w)+'x'+str(h)
  29
  30 # Check screen setup
  31 internalName = "LVDS"
  32 externalName = "CRT1"
  33 connectors = getXrandrInformation()
  34 internalResolutions = connectors[internalName] # there must be a screen assoicated to the internal connector
  35 externalResolutions = connectors.get(externalName)
  36
  37 # Check what to do
  38 internalArgs = ["--mode", res2str(internalResolutions[0])] # there must be a resolution for the internal screen
  39 externalArgs = ["--off"]
  40 if externalResolutions is not None: # we need to ask what to do
  41         extPosition = PositionSelection(map(res2str, externalResolutions))
  42         extPosition.exec_()
  43         if not extPosition.result(): sys.exit(1) # the user canceled
  44         # build command-line
  45         externalArgs = ["--mode", extPosition.resolution] # we definitely want an external screen
  46         if extPosition.position == PositionSelection.EXTERNAL_ONLY:
  47                 internalArgs = ["--off"]
  48         elif extPosition.position == PositionSelection.LEFT:
  49                 externalArgs += ["--left-of", internalName]
  50         else:
  51                 externalArgs += ["--right-of", internalName]
  52 # and do it
  53 args = ["--output", internalName] + internalArgs + ["--output", externalName] + externalArgs
  54 print args
  55 subprocess.check_call(["xrandr"] + args)