Homogenize end markers (#6)

* Homogenize end marks

hdlparse/vhdl_parser.py

@@ -32,13 +32,13 @@
     (r'subtype\s+(\w+)\s+is\s+(\w+)', 'subtype'),
     (r'constant\s+(\w+)\s+:\s+(\w+)', 'constant'),
     (r'type\s+(\w+)\s*is', 'type', 'type_decl'),
-    (r'end\s+package', None, '#pop'),
+    (r'end\s+\w+\s*;', None, '#pop'),
     (r'--#(.*)\n', 'metacomment'),
     (r'/\*', 'block_comment', 'block_comment'),
     (r'--.*\n', None),
   ],
   'package_body': [
-    (r'end\s+package\s+body', None, '#pop'),
+    (r'end\s+\w+\s*;', None, '#pop'),
     (r'--#(.*)\n', 'metacomment'),
     (r'/\*', 'block_comment', 'block_comment'),
     (r'--.*\n', None),
@@ -87,12 +87,11 @@
     (r'generic\s*\(', None, 'generic_list'),
     (r'port\s*\(', None, 'port_list'),
     (r'end\s+\w+\s*;', 'end_entity', '#pop'),
-    (r'end\s+entity\s+\w+\s*;', 'end_entity', '#pop'),
     (r'/\*', 'block_comment', 'block_comment'),
     (r'--.*\n', None),
   ],
   'architecture': [
-    (r'end\s+architecture\s*;', 'end_arch', '#pop'),
+    (r'end\s+\w+\s*;', 'end_arch', '#pop'),
     (r'/\*', 'block_comment', 'block_comment'),
     (r'--.*\n', None),
   ],
@@ -152,7 +151,7 @@
     (r'\*/', 'end_comment', '#pop'),
   ],
 }
-      
+
 VhdlLexer = MiniLexer(vhdl_tokens, flags=re.MULTILINE | re.IGNORECASE)
 
 
@@ -170,7 +169,7 @@ def __init__(self, name, desc=None):
 
 class VhdlParameter(object):
   '''Parameter to subprograms, ports, and generics
-  
+
   Args:
     name (str): Name of the object
     mode (str): Direction mode for the parameter
@@ -197,7 +196,7 @@ def __str__(self):
     if self.param_desc is not None:
       param = '{} --{}'.format(param, self.param_desc)
     return param
-      
+
   def __repr__(self):
     return "VhdlParameter('{}', '{}', '{}')".format(self.name, self.mode, self.data_type.name + self.data_type.arange)
 
@@ -235,7 +234,7 @@ def __init__(self, name, desc=None):
 class VhdlType(VhdlObject):
   '''Type definition
 
-  Args:  
+  Args:
     name (str): Name of the type
     package (str): Package containing the type
     type_of (str): Object type of this type definition
@@ -252,7 +251,7 @@ def __repr__(self):
 
 class VhdlSubtype(VhdlObject):
   '''Subtype definition
-  
+
   Args:
     name (str): Name of the subtype
     package (str): Package containing the subtype
@@ -270,7 +269,7 @@ def __repr__(self):
 
 class VhdlConstant(VhdlObject):
   '''Constant definition
-  
+
   Args:
     name (str): Name of the constant
     package (str): Package containing the constant
@@ -288,7 +287,7 @@ def __repr__(self):
 
 class VhdlFunction(VhdlObject):
   '''Function declaration
-  
+
   Args:
     name (str): Name of the function
     package (str): Package containing the function
@@ -351,7 +350,7 @@ def dump(self):
 
 class VhdlComponent(VhdlObject):
   '''Component declaration
-  
+
   Args:
     name (str): Name of the component
     package (str): Package containing the component
@@ -379,7 +378,7 @@ def dump(self):
 
 def parse_vhdl_file(fname):
   '''Parse a named VHDL file
-  
+
   Args:
     fname(str): Name of file to parse
   Returns:
@@ -398,7 +397,7 @@ def parse_vhdl(text):
     Parsed objects.
   '''
   lex = VhdlLexer
-  
+
   name = None
   kind = None
   saved_type = None
@@ -417,7 +416,7 @@ def parse_vhdl(text):
   array_range_start_pos = 0
 
   objects = []
-  
+
   for pos, action, groups in lex.run(text):
     if action == 'metacomment':
       realigned = re.sub(r'^#+', lambda m: ' ' * len(m.group(0)), groups[0])
@@ -450,10 +449,10 @@ def parse_vhdl(text):
       param_items.append(VhdlParameter(groups[1]))
     elif action == 'param_type':
       mode, ptype = groups
-      
+
       if mode is not None:
         mode = mode.strip()
-      
+
       for i in param_items: # Set mode and type for all pending parameters
         i.mode = mode
         i.data_type = ptype
@@ -468,14 +467,14 @@ def parse_vhdl(text):
       # Complete last parameters
       for i in param_items:
         parameters.append(i)
-        
+
       if kind == 'function':
         vobj = VhdlFunction(name, cur_package, parameters, groups[0], metacomments)
       else:
         vobj = VhdlProcedure(name, cur_package, parameters, metacomments)
-      
+
       objects.append(vobj)
-    
+
       metacomments = []
       parameters = []
       param_items = []
@@ -505,7 +504,6 @@ def parse_vhdl(text):
 
     elif action == 'generic_param_type':
       ptype = groups[0]
-
       last_items = []
       for i in param_items:
         p = VhdlParameter(i, 'in', VhdlParameterType(ptype))
@@ -607,7 +605,7 @@ def parse_vhdl(text):
 
 def subprogram_prototype(vo):
   '''Generate a canonical prototype string
-  
+
   Args:
     vo (VhdlFunction, VhdlProcedure): Subprogram object
   Returns:
@@ -629,7 +627,7 @@ def subprogram_prototype(vo):
 
 def subprogram_signature(vo, fullname=None):
   '''Generate a signature string
-  
+
   Args:
     vo (VhdlFunction, VhdlProcedure): Subprogram object
   Returns:
@@ -651,7 +649,7 @@ def subprogram_signature(vo, fullname=None):
 
 def is_vhdl(fname):
   '''Identify file as VHDL by its extension
-  
+
   Args:
     fname (str): File name to check
   Returns:
@@ -675,7 +673,7 @@ def __init__(self, array_types=set()):
 
   def extract_objects(self, fname, type_filter=None):
     '''Extract objects from a source file
-    
+
     Args:
       fname (str): File to parse
       type_filter (class, optional): Object class to filter results
@@ -717,7 +715,7 @@ def extract_objects_from_source(self, text, type_filter=None):
 
   def is_array(self, data_type):
     '''Check if a type is a known array type
-    
+
     Args:
       data_type (str): Name of type to check
     Returns:
@@ -732,7 +730,7 @@ def is_array(self, data_type):
 
   def _add_array_types(self, type_defs):
     '''Add array data types to internal registry
-    
+
     Args:
       type_defs (dict): Dictionary of type definitions
     '''
@@ -741,7 +739,7 @@ def _add_array_types(self, type_defs):
 
   def load_array_types(self, fname):
     '''Load file of previously extracted data types
-    
+
     Args:
       fname (str): Name of file to load array database from
     '''
@@ -758,7 +756,7 @@ def load_array_types(self, fname):
 
   def save_array_types(self, fname):
     '''Save array type registry to a file
-    
+
     Args:
       fname (str): Name of file to save array database to
     '''
@@ -768,7 +766,7 @@ def save_array_types(self, fname):
 
   def _register_array_types(self, objects):
     '''Add array type definitions to internal registry
-    
+
     Args:
       objects (list of VhdlType or VhdlSubtype): Array types to track
     '''