Adding define support to KLL compiler.

[kiibohd-kll.git] / backends / kiibohd.py

diff --git a/backends/kiibohd.py b/backends/kiibohd.py
index 71eb6dac9b15974ab937cb73a1569e9112d00435..c035fb28f39b1fa31bf27b38779966a864c06165 100644 (file)
--- a/backends/kiibohd.py
+++ b/backends/kiibohd.py
@@ -34,6 +34,7 @@ from kll_lib.containers import *
 
  ## Print Decorator Variables
 ERROR = '\033[5;1;31mERROR\033[0m:'
+WARNING = '\033[5;1;33mWARNING\033[0m:'
 
 
 
@@ -42,12 +43,13 @@ ERROR = '\033[5;1;31mERROR\033[0m:'
 class Backend:
        # Initializes backend
        # Looks for template file and builds list of fill tags
-       def __init__( self, templatePath ):
+       def __init__( self, templatePath, definesTemplatePath ):
                # Does template exist?
                if not os.path.isfile( templatePath ):
                        print ( "{0} '{1}' does not exist...".format( ERROR, templatePath ) )
                        sys.exit( 1 )
 
+               self.definesTemplatePath = definesTemplatePath
                self.templatePath = templatePath
                self.fill_dict = dict()
 
@@ -58,6 +60,11 @@ class Backend:
                                match = re.findall( '<\|([^|>]+)\|>', line )
                                for item in match:
                                        self.tagList.append( item )
+               with open( definesTemplatePath, 'r' ) as openFile:
+                       for line in openFile:
+                               match = re.findall( '<\|([^|>]+)\|>', line )
+                               for item in match:
+                                       self.tagList.append( item )
 
 
        # USB Code Capability Name
@@ -73,13 +80,13 @@ class Backend:
 
 
        # Processes content for fill tags and does any needed dataset calculations
-       def process( self, capabilities, macros ):
+       def process( self, capabilities, macros, variables ):
                ## Information ##
                # TODO
                self.fill_dict['Information']  = "// This file was generated by the kll compiler, DO NOT EDIT.\n"
                self.fill_dict['Information'] += "// Generation Date:    {0}\n".format( "TODO" )
                self.fill_dict['Information'] += "// Compiler arguments: {0}\n".format( "TODO" )
-               self.fill_dict['Information'] += "// KLL Backend:        {0}\n".format( "TODO" )
+               self.fill_dict['Information'] += "// KLL Backend:        {0}\n".format( "kiibohd" )
                self.fill_dict['Information'] += "// KLL Git Rev:        {0}\n".format( "TODO" )
                self.fill_dict['Information'] += "//\n"
                self.fill_dict['Information'] += "// - Base Layer -\n"
@@ -87,6 +94,25 @@ class Backend:
                self.fill_dict['Information'] += "// - Partial Layers -\n"
 
 
+               ## Variable Information ##
+               self.fill_dict['VariableInformation'] = ""
+
+               # Iterate through the variables, output, and indicate the last file that modified it's value
+               # Output separate tables per file, per table and overall
+               # TODO
+
+
+               ## Defines ##
+               self.fill_dict['Defines'] = ""
+
+               # Iterate through defines and lookup the variables
+               for define in variables.defines.keys():
+                       if define in variables.overallVariables.keys():
+                               self.fill_dict['Defines'] += "\n#define {0} {1}".format( variables.defines[ define ], variables.overallVariables[ define ] )
+                       else:
+                               print( "{0} '{1}' not defined...".format( WARNING, define ) )
+
+
                ## Capabilities ##
                self.fill_dict['CapabilitiesList'] = "const Capability CapabilitiesList[] = {\n"
 
@@ -272,9 +298,9 @@ class Backend:
 
 
        # Generates the output keymap with fill tags filled
-       def generate( self, filepath ):
+       def generate( self, outputPath, definesOutputPath ):
                # Process each line of the template, outputting to the target path
-               with open( filepath, 'w' ) as outputFile:
+               with open( outputPath, 'w' ) as outputFile:
                        with open( self.templatePath, 'r' ) as templateFile:
                                for line in templateFile:
                                        # TODO Support multiple replacements per line
@@ -290,3 +316,20 @@ class Backend:
                                        else:
                                                outputFile.write( line )
 
+               # Process each line of the defines template, outputting to the target path
+               with open( definesOutputPath, 'w' ) as outputFile:
+                       with open( self.definesTemplatePath, 'r' ) as templateFile:
+                               for line in templateFile:
+                                       # TODO Support multiple replacements per line
+                                       # TODO Support replacement with other text inline
+                                       match = re.findall( '<\|([^|>]+)\|>', line )
+
+                                       # If match, replace with processed variable
+                                       if match:
+                                               outputFile.write( self.fill_dict[ match[ 0 ] ] )
+                                               outputFile.write("\n")
+
+                                       # Otherwise, just append template to output file
+                                       else:
+                                               outputFile.write( line )
+