Tools/bgen/bgen/bgenGeneratorGroup.py

   1 from bgenOutput import *
   2
   3 class GeneratorGroup:
   4
   5         def __init__(self, prefix):
   6                 self.prefix = prefix
   7                 self.generators = []
   8
   9         def add(self, g, dupcheck=0):
  10                 if dupcheck:
  11                         if g in self.generators:
  12                                 print 'DUP', g.name
  13                                 return
  14                 g.setprefix(self.prefix)
  15                 self.generators.append(g)
  16
  17         def generate(self):
  18                 for g in self.generators:
  19                         g.generate()
  20                 Output()
  21                 Output("static PyMethodDef %s_methods[] = {", self.prefix)
  22                 IndentLevel()
  23                 for g in self.generators:
  24                         g.reference()
  25                 Output("{NULL, NULL, 0}")
  26                 DedentLevel()
  27                 Output("};")
  28
  29
  30 def _test():
  31         void = None
  32         from bgenGenerator import FunctionGenerator
  33         group = GeneratorGroup("spam")
  34         eggs = FunctionGenerator(void, "eggs")
  35         group.add(eggs)
  36         print "/* START */"
  37         group.generate()
  38
  39 if __name__ == "__main__":
  40         _test()