from itertools import chain
from collections import OrderedDict
+import posix
+
+
virtual_profiles = [
# (name, description, callback)
("common", "Clone all connected outputs at the largest common resolution", None),
The following virtual configurations are available:
""".strip()
+class AutorandrException(Exception):
+ def __init__(self, message, original_exception=None, report_bug=False):
+ self.message = message
+ self.report_bug = report_bug
+ if original_exception:
+ self.original_exception = original_exception
+ trace = sys.exc_info()[2]
+ while trace.tb_next:
+ trace = trace.tb_next
+ self.line = trace.tb_lineno
+ else:
+ try:
+ import inspect
+ self.line = inspect.currentframe().f_back.f_lineno
+ except:
+ self.line = None
+ self.original_exception = None
+
+ def __str__(self):
+ retval = [ self.message ]
+ if self.line:
+ retval.append(" (line %d)" % self.line)
+ if self.original_exception:
+ retval.append(":\n ")
+ retval.append(str(self.original_exception).replace("\n", "\n "))
+ if self.report_bug:
+ retval.append("\nThis appears to be a bug. Please help improving autorandr by reporting it upstream."
+ "\nPlease attach the output of `xrandr --verbose` to your bug report if appropriate.")
+ return "".join(retval)
+
class XrandrOutput(object):
"Represents an XRandR output"
(?: # Differentiate disconnected and connected in first line
disconnected |
unknown\ connection |
- (?P<connected>connected)\s+ # If connected:
- (?:
- (?P<primary>primary\ )? # Might be primary screen
+ (?P<connected>connected)
+ )
+ \s*
+ (?P<primary>primary\ )? # Might be primary screen
+ (?:\s*
(?P<width>[0-9]+)x(?P<height>[0-9]+) # Resolution (might be overridden below!)
- \+(?P<x>[0-9]+)\+(?P<y>[0-9]+)\s+ # Position
+ \+(?P<x>-?[0-9]+)\+(?P<y>-?[0-9]+)\s+ # Position
(?:\(0x[0-9a-fA-F]+\)\s+)? # XID
(?P<rotate>(?:normal|left|right|inverted))\s+ # Rotation
(?:(?P<reflect>X\ and\ Y|X|Y)\ axis)? # Reflection
- )? # .. but everything of the above only if the screen is in use.
- ).*
+ )? # .. but everything of the above only if the screen is in use.
+ (?:[\ \t]*\([^\)]+\))(?:\s*[0-9]+mm\sx\s[0-9]+mm)?
+ (?:[\ \t]*panning\ (?P<panning>[0-9]+x[0-9]+\+[0-9]+\+[0-9]+))? # Panning information
+ (?:[\ \t]*tracking\ (?P<tracking>[0-9]+x[0-9]+\+[0-9]+\+[0-9]+))? # Tracking information
+ (?:[\ \t]*border\ (?P<border>(?:[0-9]+/){3}[0-9]+))? # Border information
(?:\s*(?: # Properties of the output
Gamma: (?P<gamma>[0-9\.: ]+) | # Gamma value
Transform: (?P<transform>(?:[\-0-9\. ]+\s+){3}) | # Transformation matrix
))+
\s*
(?P<modes>(?:
- (?P<mode_width>[0-9]+)x(?P<mode_height>[0-9]+).+?\*current.+\s+
- h:.+\s+v:.+clock\s+(?P<rate>[0-9\.]+)Hz\s* | # Interesting (current) resolution: Extract rate
- [0-9]+x[0-9]+.+\s+h:.+\s+v:.+\s* # Other resolutions
+ (?P<mode_name>\S+).+?\*current.*\s+ # Interesting (current) resolution: Extract rate
+ h:\s+width\s+(?P<mode_width>[0-9]+).+\s+
+ v:\s+height\s+(?P<mode_height>[0-9]+).+clock\s+(?P<rate>[0-9\.]+)Hz\s* |
+ \S+(?:(?!\*current).)+\s+h:.+\s+v:.+\s* # Other resolutions
)*)
"""
XRANDR_OUTPUT_MODES_REGEXP = """(?x)
- (?P<width>[0-9]+)x(?P<height>[0-9]+)
- .*?(?P<preferred>\+preferred)?
- \s+h:.+
- \s+v:.+clock\s+(?P<rate>[0-9\.]+)Hz
+ (?P<name>\S+).+?(?P<preferred>\+preferred)?\s+
+ h:\s+width\s+(?P<width>[0-9]+).+\s+
+ v:\s+height\s+(?P<height>[0-9]+).+clock\s+(?P<rate>[0-9\.]+)Hz\s* |
"""
XRANDR_13_DEFAULTS = {
"transform": "1,0,0,0,1,0,0,0,1",
+ "panning": "0x0",
}
XRANDR_12_DEFAULTS = {
def sort_key(self):
"Return a key to sort the outputs for xrandr invocation"
if not self.edid:
+ return -2
+ if "off" in self.options:
return -1
if "pos" in self.options:
x, y = map(float, self.options["pos"].split("x"))
xrandr_output = xrandr_output.replace("\r\n", "\n")
match_object = re.search(XrandrOutput.XRANDR_OUTPUT_REGEXP, xrandr_output)
except:
- raise RuntimeError("Parsing XRandR output failed, there is an error in the regular expression.")
+ raise AutorandrException("Parsing XRandR output failed, there is an error in the regular expression.", report_bug = True)
if not match_object:
debug = debug_regexp(XrandrOutput.XRANDR_OUTPUT_REGEXP, xrandr_output)
- raise RuntimeError("Parsing XRandR output failed, the regular expression did not match: %s" % debug)
+ raise AutorandrException("Parsing XRandR output failed, the regular expression did not match: %s" % debug, report_bug = True)
remainder = xrandr_output[len(match_object.group(0)):]
if remainder:
- raise RuntimeError(("Parsing XRandR output failed, %d bytes left unmatched after regular expression, "
- "starting at byte %d with ..'%s'.") % (len(remainder), len(len(match_object.group(0))), remainder[:10]))
+ raise AutorandrException(("Parsing XRandR output failed, %d bytes left unmatched after regular expression, "
+ "starting at byte %d with ..'%s'.") % (len(remainder), len(match_object.group(0)), remainder[:10]), report_bug=True)
match = match_object.groupdict()
modes = []
if match["modes"]:
- modes = [ x.groupdict() for x in re.finditer(XrandrOutput.XRANDR_OUTPUT_MODES_REGEXP, match["modes"]) ]
+ modes = [ x.groupdict() for x in re.finditer(XrandrOutput.XRANDR_OUTPUT_MODES_REGEXP, match["modes"]) if x.group("name") ]
if not modes:
- raise RuntimeError("Parsing XRandR output failed, couldn't find any display modes")
+ raise AutorandrException("Parsing XRandR output failed, couldn't find any display modes", report_bug=True)
options = {}
if not match["connected"]:
- options["off"] = None
edid = None
- elif not match["width"]:
- options["off"] = None
+ else:
edid = "".join(match["edid"].strip().split()) if match["edid"] else "%s-%s" % (XrandrOutput.EDID_UNAVAILABLE, match["output"])
+
+ if not match["width"]:
+ options["off"] = None
else:
- if match["mode_width"]:
+ if match["mode_name"]:
+ options["mode"] = match["mode_name"]
+ elif match["mode_width"]:
options["mode"] = "%sx%s" % (match["mode_width"], match["mode_height"])
else:
if match["rotate"] not in ("left", "right"):
elif match["reflect"] == "X and Y":
options["reflect"] = "xy"
options["pos"] = "%sx%s" % (match["x"], match["y"])
+ if match["panning"]:
+ panning = [ match["panning"] ]
+ if match["tracking"]:
+ panning += [ "/", match["tracking"] ]
+ if match["border"]:
+ panning += [ "/", match["border"] ]
+ options["panning"] = "".join(panning)
if match["transform"]:
transformation = ",".join(match["transform"].strip().split())
if transformation != "1.000000,0.000000,0.000000,0.000000,1.000000,0.000000,0.000000,0.000000,1.000000":
options["transform"] = transformation
- if not match["mode_width"]:
+ if not match["mode_name"]:
# TODO We'd need to apply the reverse transformation here. Let's see if someone complains, I doubt that this
# special case is actually required.
- print("Warning: Output %s has a transformation applied. Could not determine correct mode!", file=sys.stderr)
+ print("Warning: Output %s has a transformation applied. Could not determine correct mode! Using `%s'." % (match["output"], options["mode"]), file=sys.stderr)
if match["gamma"]:
gamma = match["gamma"].strip()
options["gamma"] = gamma
if match["rate"]:
options["rate"] = match["rate"]
- edid = "".join(match["edid"].strip().split()) if match["edid"] else "%s-%s" % (XrandrOutput.EDID_UNAVAILABLE, match["output"])
return XrandrOutput(match["output"], edid, options), modes
if fuzzy_output in fuzzy_edid_map:
edid = edid_map[list(edid_map.keys())[fuzzy_edid_map.index(fuzzy_output)]]
elif "off" not in options:
- raise RuntimeError("Failed to find an EDID for output `%s' in setup file, required as `%s' is not off in config file."
- % (options["output"], options["output"]))
+ raise AutorandrException("Failed to find an EDID for output `%s' in setup file, required as `%s' is not off in config file." % (options["output"], options["output"]))
output = options["output"]
del options["output"]
string[partial_length:partial_length+10]))
except ImportError:
pass
- return "Debug information available if `regex' module is installed."
+ return "Debug information would be available if the `regex' module was installed."
def parse_xrandr_output():
"Parse the output of `xrandr --verbose' into a list of outputs"
xrandr_output = os.popen("xrandr -q --verbose").read()
if not xrandr_output:
- raise RuntimeError("Failed to run xrandr")
+ raise AutorandrException("Failed to run xrandr")
# We are not interested in screens
xrandr_output = re.sub("(?m)^Screen [0-9].+", "", xrandr_output).strip()
# Split at output boundaries and instanciate an XrandrOutput per output
split_xrandr_output = re.split("(?m)^([^ ]+ (?:(?:dis)?connected|unknown connection).*)$", xrandr_output)
if len(split_xrandr_output) < 2:
- raise RuntimeError("No output boundaries found")
+ raise AutorandrException("No output boundaries found", report_bug=True)
outputs = OrderedDict()
modes = OrderedDict()
for i in range(1, len(split_xrandr_output), 2):
except:
return False
-def apply_configuration(configuration, dry_run=False):
+def apply_configuration(new_configuration, current_configuration, dry_run=False):
"Apply a configuration"
- outputs = sorted(configuration.keys(), key=lambda x: configuration[x].sort_key)
+ outputs = sorted(new_configuration.keys(), key=lambda x: new_configuration[x].sort_key)
if dry_run:
base_argv = [ "echo", "xrandr" ]
else:
# a limit on the number of enabled screens
# - We must make sure that the screen at 0x0 is activated first,
# or the other (first) screen to be activated would be moved there.
+ # - If an active screen already has a transformation and remains active,
+ # the xrandr call fails with an invalid RRSetScreenSize parameter error.
+ # Update the configuration in 3 passes in that case. (On Haswell graphics,
+ # at least.)
+ auxiliary_changes_pre = []
disable_outputs = []
enable_outputs = []
+ remain_active_count = 0
for output in outputs:
- if not configuration[output].edid or "off" in configuration[output].options:
- disable_outputs.append(configuration[output].option_vector)
+ if not new_configuration[output].edid or "off" in new_configuration[output].options:
+ disable_outputs.append(new_configuration[output].option_vector)
else:
- enable_outputs.append(configuration[output].option_vector)
+ if "off" not in current_configuration[output].options:
+ remain_active_count += 1
+ enable_outputs.append(new_configuration[output].option_vector)
+ if xrandr_version() >= Version("1.3.0") and "transform" in current_configuration[output].options:
+ auxiliary_changes_pre.append(["--output", output, "--transform", "none"])
+
+ # Perform pe-change auxiliary changes
+ if auxiliary_changes_pre:
+ argv = base_argv + list(chain.from_iterable(auxiliary_changes_pre))
+ if subprocess.call(argv) != 0:
+ raise AutorandrException("Command failed: %s" % " ".join(argv))
- # Disable all but the last of the outputs to be disabled
- if len(disable_outputs) > 1:
- if subprocess.call(base_argv + list(chain.from_iterable(disable_outputs[:-1]))) != 0:
+ # Disable unused outputs, but make sure that there always is at least one active screen
+ disable_keep = 0 if remain_active_count else 1
+ if len(disable_outputs) > disable_keep:
+ if subprocess.call(base_argv + list(chain.from_iterable(disable_outputs[:-1] if disable_keep else disable_outputs))) != 0:
# Disabling the outputs failed. Retry with the next command:
# Sometimes disabling of outputs fails due to an invalid RRSetScreenSize.
# This does not occur if simultaneously the primary screen is reset.
pass
else:
- disable_outputs = disable_outputs[-1:]
+ disable_outputs = disable_outputs[-1:] if disable_keep else []
# If disable_outputs still has more than one output in it, one of the xrandr-calls below would
# disable the last two screens. This is a problem, so if this would happen, instead disable only
for index in range(0, len(operations), 2):
argv = base_argv + list(chain.from_iterable(operations[index:index+2]))
if subprocess.call(argv) != 0:
- raise RuntimeError("Command failed: %s" % " ".join(argv))
+ raise AutorandrException("Command failed: %s" % " ".join(argv))
def add_unused_outputs(source_configuration, target_configuration):
"Add outputs that are missing in target to target, in 'off' state"
for output in configuration:
configuration[output].options = {}
if output in modes:
- configuration[output].options["mode"] = "%sx%s" % common_resolution[-1]
+ configuration[output].options["mode"] = [ x["name"] for x in sorted(modes[output], key=lambda x: 0 if x["preferred"] else 1) if x["width"] == common_resolution[-1][0] and x["height"] == common_resolution[-1][1] ][0]
configuration[output].options["pos"] = "0x0"
else:
configuration[output].options["off"] = None
configuration[output].options = {}
if output in modes:
mode = sorted(modes[output], key=lambda a: int(a["width"])*int(a["height"]) + (10**6 if a["preferred"] else 0))[-1]
- configuration[output].options["mode"] = "%sx%s" % (mode["width"], mode["height"])
+ configuration[output].options["mode"] = mode["name"]
configuration[output].options["rate"] = mode["rate"]
configuration[output].options["pos"] = pos_specifier % shift
shift += int(mode[shift_index])
try:
options = dict(getopt.getopt(argv[1:], "s:l:d:cfh", [ "dry-run", "change", "default=", "save=", "load=", "force", "fingerprint", "config", "help" ])[0])
except getopt.GetoptError as e:
- print(str(e))
- options = { "--help": True }
+ print("Failed to parse options: {0}.\n"
+ "Use --help to get usage information.".format(str(e)),
+ file=sys.stderr)
+ sys.exit(posix.EX_USAGE)
profiles = {}
try:
# Sort by descending mtime
profiles = OrderedDict(sorted(profiles.items(), key=lambda x: -x[1]["config-mtime"]))
except Exception as e:
- print("Failed to load profiles:\n%s" % str(e), file=sys.stderr)
- sys.exit(1)
+ raise AutorandrException("Failed to load profiles", e)
- try:
- config, modes = parse_xrandr_output()
- except Exception as e:
- print("Failed to parse current configuration from XRandR:\n%s" % str(e), file=sys.stderr)
- sys.exit(1)
+ config, modes = parse_xrandr_output()
if "--fingerprint" in options:
output_setup(config, sys.stdout)
options["--save"] = options["-s"]
if "--save" in options:
if options["--save"] in ( x[0] for x in virtual_profiles ):
- print("Cannot save current configuration as profile '%s':\nThis configuration name is a reserved virtual configuration." % options["--save"])
- sys.exit(1)
+ raise AutorandrException("Cannot save current configuration as profile '%s':\nThis configuration name is a reserved virtual configuration." % options["--save"])
try:
save_configuration(os.path.join(profile_path, options["--save"]), config)
except Exception as e:
- print("Failed to save current configuration as profile '%s':\n%s" % (options["--save"], str(e)), file=sys.stderr)
- sys.exit(1)
+ raise AutorandrException("Failed to save current configuration as profile '%s'" % (options["--save"],), e)
print("Saved current configuration as profile '%s'" % options["--save"])
sys.exit(0)
load_config = profile["config"]
scripts_path = profile["path"]
except KeyError:
- print("Failed to load profile '%s':\nProfile not found" % load_profile, file=sys.stderr)
- sys.exit(1)
+ raise AutorandrException("Failed to load profile '%s': Profile not found" % load_profile)
if load_profile in detected_profiles and detected_profiles[0] != load_profile:
update_mtime(os.path.join(scripts_path, "config"))
add_unused_outputs(config, load_config)
try:
if "--dry-run" in options:
- apply_configuration(load_config, True)
+ apply_configuration(load_config, config, True)
else:
exec_scripts(scripts_path, "preswitch")
- apply_configuration(load_config, False)
+ apply_configuration(load_config, config, False)
exec_scripts(scripts_path, "postswitch")
except Exception as e:
- print("Failed to apply profile '%s':\n%s" % (load_profile, str(e)), file=sys.stderr)
- sys.exit(1)
+ raise AutorandrException("Failed to apply profile '%s'" % load_profile, e, True)
sys.exit(0)
if __name__ == '__main__':
try:
main(sys.argv)
- except Exception as e:
- print("General failure. Please report this as a bug:\n%s" % (str(e),), file=sys.stderr)
+ except AutorandrException as e:
+ print(e, file=sys.stderr)
sys.exit(1)
+ except Exception as e:
+ if not len(str(e)): # BdbQuit
+ print("Exception: {0}".format(e.__class__.__name__))
+ sys.exit(2)
+
+ print("Unhandled exception ({0}). Please report this as a bug.".format(e), file=sys.stderr)
+ raise