generate enums for pdf types

experimental/PdfViewer/generate_code.py

 import sys
 
 class PdfName:
   def __init__(self, name, abr=''):
     self.fName = name
     self.fAbr = abr
 
 class PdfInteger:
   def __init__(self, value):
     self.fValue = value
@@ skipping 70 matching lines @@
     return self.fParent
 
 class PdfClass:
   def __init__(self, name, base):
     self.fFields = []
     self.fIncludes = []
     self.fCC = []
     self.fName = name
     self.fBase = base
     
+    self.fEnumSubclasses = [] 
+    
+    self.fEnum = '!UNDEFINED'
+    self.fEnumEnd = '!UNDEFINED'
+    
   def required(self):
     field = PdfClassField(self, True)
     self.fFields.append(field)
     return field
     
   def optional(self):
     field = PdfClassField(self, False)
     self.fFields.append(field)
     return field
     
   def include(self, path):
     self.fIncludes.append(path)
     return self
     
   def carbonCopy(self, cc):
     self.fCC.append(cc)
     return self 
 
 class PdfClassManager:
   def __init__(self):
-    self.fClasses = []
+    self.fClasses = {}
 
   def addClass(self, name, base=''):
     cls = PdfClass(name, base)
-    self.fClasses.append(cls)
+    self.fClasses[name] = cls
     return cls
   
+  def longName(self, name):
+    ret = ''
+    while name != '':
+      cls = self.fClasses[name]
+      ret = name + ret
+      name = cls.fBase
+      
+    return ret
+  
+  
+  def writeEnum(self, enum, enumToCls):
+    print('  ' + enum + ',')
+    cls = enumToCls[enum]
+    cls.fEnumSubclasses.sort()
+    
+    cnt = 0
+    for sub in cls.fEnumSubclasses:
+      self.writeEnum(cls.fEnumSubclasses[cnt], enumToCls)
+      cnt = cnt + 1
+      
+    if cnt != 0:
+       print('  ' + cls.fEnumEnd + ',')
+  
   def write(self):
+    # generate enum
+    enumsRoot = []
+
+    enumToCls = {}
+    
+    for name in self.fClasses:
+      cls = self.fClasses[name]
+      enum = self.longName(name)
+      cls.fEnum = 'k' + enum + '_PdfObjectType'
+      cls.fEnumEnd = 'k' + enum + '__End_PdfObjectType'
+            
+      if cls.fBase != '':
+        self.fClasses[cls.fBase].fEnumSubclasses.append(cls.fEnum)
+
+      if cls.fBase == '':
+        enumsRoot.append(cls.fEnum)
+       
+      enumToCls[cls.fEnum] = cls
+      
+    enumsRoot.sort()
+    
+    # write enums
+    print('enum PdfObjectType {')
+    for enum in enumsRoot:
+      self.writeEnum(enum, enumToCls)
+    print('};')
+      
+    # generate each class
+    # generate parser  
     return
 
 def generateCode():
   all = PdfClassManager()
   
   all.addClass('Object')
   all.addClass('Null')
   all.addClass('Boolean')
   all.addClass('Integer')
   all.addClass('Real')
@@ skipping 20 matching lines @@
   all.addClass('Form', 'XObject').required().hasField('/Type').must('/XObject').done()\
                                  .required().hasField('/Subtype').must('/Form').done()
 
 
   all.write()
   
   return 1
 
 if '__main__' == __name__:
   sys.exit(generateCode())