From 01f133fbc4d491ed1ca4c902e0fc36641c01d51e Mon Sep 17 00:00:00 2001 From: Peter Hutterer Date: Thu, 19 Dec 2024 08:39:22 +1000 Subject: [PATCH] tools/per-slot-delta: use dataclasses and enums Slight modernization of the code Part-of: --- tools/libinput-analyze-per-slot-delta.py | 24 +++++++++++++----------- 1 file changed, 13 insertions(+), 11 deletions(-) diff --git a/tools/libinput-analyze-per-slot-delta.py b/tools/libinput-analyze-per-slot-delta.py index 9fae4620..782d0096 100755 --- a/tools/libinput-analyze-per-slot-delta.py +++ b/tools/libinput-analyze-per-slot-delta.py @@ -29,6 +29,9 @@ # # Input is a libinput record yaml file +from dataclasses import dataclass +from enum import Enum + import argparse import math import sys @@ -122,24 +125,23 @@ class SlotFormatter: self.slots.append(string.ljust(self.width + len(color) + len(reset))) -class SlotState: +class SlotState(Enum): NONE = 0 BEGIN = 1 UPDATE = 2 END = 3 +@dataclass class Slot: - state = SlotState.NONE - x = 0 - y = 0 - dx = 0 - dy = 0 - used = False - dirty = False - - def __init__(self, index): - self.index = index + index: int + state: SlotState = SlotState.NONE + x: float = 0 + y: float = 0 + dx: float = 0 + dy: float = 0 + used: bool = False + dirty: bool = False def main(argv):