Adding support for USB Code trigger assignment

[kiibohd-kll.git] / kll_lib / containers.py

diff --git a/kll_lib/containers.py b/kll_lib/containers.py
index d843574564b63c37ffce995bd44a1c13e277b3bc..751f94947d9a224f788fa9ce1b8153e041a57e84 100644 (file)
--- a/kll_lib/containers.py
+++ b/kll_lib/containers.py
@@ -94,6 +94,9 @@ class Macros:
                self.triggerList = []
                self.maxScanCode = []
 
+               # USBCode Assignment Cache
+               self.assignmentCache = []
+
        def __repr__( self ):
                return "{0}".format( self.macros )
 
@@ -124,11 +127,35 @@ class Macros:
 
                # Scan current layer for USB Codes
                for macro in self.macros[ self.layer ].keys():
-                       if usbCode == self.macros[ self.layer ][ macro ]:
+                       if usbCode in self.macros[ self.layer ][ macro ]:
                                scanCodeList.append( macro )
 
                return scanCodeList
 
+       # Cache USBCode Assignment
+       def cacheAssignment( self, operator, scanCode, result ):
+               self.assignmentCache.append( [ operator, scanCode, result ] )
+
+       # Assign cached USBCode Assignments
+       def replayCachedAssignments( self ):
+               # Iterate over each item in the assignment cache
+               for item in self.assignmentCache:
+                       # Check operator, and choose the specified assignment action
+                       # Append Case
+                       if item[0] == ":+":
+                               self.appendScanCode( item[1], item[2] )
+
+                       # Remove Case
+                       elif item[0] == ":-":
+                               self.removeScanCode( item[1], item[2] )
+
+                       # Replace Case
+                       elif item[0] == ":":
+                               self.replaceScanCode( item[1], item[2] )
+
+               # Clear assignment cache
+               self.assignmentCache = []
+
        # Generate/Correlate Layers
        def generate( self ):
                self.generateIndices()