Tools/modulator/modulator.py

   1 #! /usr/bin/env python
   2 #
   3 # Modulator - Generate skeleton modules.
   4 #
   5 # The user fills out some forms with information about what the module
   6 # should support (methods, objects), names of these things, prefixes to
   7 # use for C code, whether the objects should also support access as numbers,
   8 # etc etc etc.
   9 # When the user presses 'Generate code' we generate a complete skeleton
  10 # module in C.
  11 #
  12 # Alternatively, the selections made can be save to a python sourcefile and
  13 # this sourcefile can be passed on the command line (resulting in the same
  14 # skeleton C code).
  15 #
  16 # Jack Jansen, CWI, October 1994.
  17 #
  18
  19 import sys, os
  20 if os.name <> 'mac':
  21     sys.path.append(os.path.join(os.environ['HOME'],
  22                                  'src/python/Tools/modulator'))
  23
  24 from Tkinter import *
  25 from Tkextra import *
  26 from ScrolledListbox import ScrolledListbox
  27 import sys
  28 import genmodule
  29 import string
  30
  31 oops = 'oops'
  32
  33 # Check that string is a legal C identifier
  34 def checkid(str):
  35     if not str: return 0
  36     if not str[0] in string.letters+'_':
  37         return 0
  38     for c in str[1:]:
  39         if not c in string.letters+string.digits+'_':
  40             return 0
  41     return 1
  42
  43 def getlistlist(list):
  44     rv = []
  45     n = list.size()
  46     for i in range(n):
  47         rv.append(list.get(i))
  48     return rv
  49
  50 class UI:
  51     def __init__(self):
  52         self.main = Frame()
  53         self.main.pack()
  54         self.main.master.title('Modulator: Module view')
  55         self.cmdframe = Frame(self.main, {'relief':'raised', 'bd':'0.5m',
  56                                           Pack:{'side':'top',
  57                                                  'fill':'x'}})
  58         self.objframe = Frame(self.main, {Pack:{'side':'top', 'fill':'x',
  59                                                 'expand':1}})
  60
  61
  62         self.check_button = Button(self.cmdframe,
  63                                   {'text':'Check', 'command':self.cb_check,
  64                                    Pack:{'side':'left', 'padx':'0.5m'}})
  65         self.save_button = Button(self.cmdframe,
  66                                   {'text':'Save...', 'command':self.cb_save,
  67                                    Pack:{'side':'left', 'padx':'0.5m'}})
  68         self.code_button = Button(self.cmdframe,
  69                                   {'text':'Generate code...',
  70                                    'command':self.cb_gencode,
  71                                    Pack:{'side':'left', 'padx':'0.5m'}})
  72         self.quit_button = Button(self.cmdframe,
  73                                   {'text':'Quit',
  74                                    'command':self.cb_quit,
  75                                    Pack:{'side':'right', 'padx':'0.5m'}})
  76
  77         self.module = UI_module(self)
  78         self.objects = []
  79         self.modified = 0
  80
  81     def run(self):
  82         self.main.mainloop()
  83
  84     def cb_quit(self, *args):
  85         if self.modified:
  86             if not askyn('You have not saved\nAre you sure you want to quit?'):
  87                 return
  88         sys.exit(0)
  89
  90     def cb_check(self, *args):
  91         try:
  92             self.module.synchronize()
  93             for o in self.objects:
  94                 o.synchronize()
  95         except oops:
  96             pass
  97
  98     def cb_save(self, *args):
  99         try:
 100             pycode = self.module.gencode('m', self.objects)
 101         except oops:
 102             return
 103
 104         fn = askfile('Python file name: ')
 105         if not fn:
 106             return
 107
 108         fp = open(fn, 'w')
 109
 110         fp.write(pycode)
 111         fp.close()
 112
 113     def cb_gencode(self, *args):
 114         try:
 115             pycode = self.module.gencode('m', self.objects)
 116         except oops:
 117             pass
 118
 119         fn = askfile('C file name: ')
 120         if not fn:
 121             return
 122
 123         fp = open(fn, 'w')
 124
 125         try:
 126             exec pycode
 127         except:
 128             message('An error occurred:-)')
 129             return
 130         genmodule.write(fp, m)
 131         fp.close()
 132
 133 class UI_module:
 134     def __init__(self, parent):
 135         self.parent = parent
 136         self.frame = Frame(parent.objframe, {'relief':'raised', 'bd':'0.2m',
 137                                              Pack:{'side':'top',
 138                                                    'fill':'x'}})
 139         self.f1 = Frame(self.frame, {Pack:{'side':'top', 'pady':'0.5m',
 140                                            'fill':'x'}})
 141         self.f2 = Frame(self.frame, {Pack:{'side':'top', 'pady':'0.5m',
 142                                            'fill':'x'}})
 143         self.f3 = Frame(self.frame, {Pack:{'side':'top', 'pady':'0.5m',
 144                                            'fill':'x'}})
 145         self.f4 = Frame(self.frame, {Pack:{'side':'top', 'pady':'0.5m',
 146                                            'fill':'x'}})
 147
 148         self.l1 = Label(self.f1, {'text':'Module:', Pack:{'side':'left',
 149                                                         'padx':'0.5m'}})
 150         self.name_entry = Entry(self.f1, {'relief':'sunken',
 151                               Pack:{'side':'left', 'padx':'0.5m', 'expand':1}})
 152         self.l2 = Label(self.f1, {'text':'Abbrev:', Pack:{'side':'left',
 153                                                         'padx':'0.5m'}})
 154         self.abbrev_entry = Entry(self.f1, {'relief':'sunken', 'width':5,
 155                               Pack:{'side':'left', 'padx':'0.5m'}})
 156
 157         self.l3 = Label(self.f2, {'text':'Methods:', Pack:{'side':'left',
 158                                                         'padx':'0.5m'}})
 159         self.method_list = ScrolledListbox(self.f2, {'relief':'sunken','bd':2,
 160                                       Pack:{'side':'left', 'expand':1,
 161                                             'padx':'0.5m', 'fill':'both'}})
 162
 163         self.l4 = Label(self.f3, {'text':'Add method:', Pack:{'side':'left',
 164                                                               'padx':'0.5m'}})
 165         self.method_entry = Entry(self.f3, {'relief':'sunken',
 166                              Pack:{'side':'left', 'padx':'0.5m', 'expand':1}})
 167         self.method_entry.bind('<Return>', self.cb_method)
 168         self.delete_button = Button(self.f3, {'text':'Delete method',
 169                                               'command':self.cb_delmethod,
 170                                               Pack:{'side':'left',
 171                                                     'padx':'0.5m'}})
 172
 173         self.newobj_button = Button(self.f4, {'text':'new object',
 174                                               'command':self.cb_newobj,
 175                                               Pack:{'side':'left',
 176                                                     'padx':'0.5m'}})
 177
 178     def cb_delmethod(self, *args):
 179         list = self.method_list.curselection()
 180         for i in list:
 181             self.method_list.delete(i)
 182
 183     def cb_newobj(self, *arg):
 184         self.parent.objects.append(UI_object(self.parent))
 185
 186     def cb_method(self, *arg):
 187         name = self.method_entry.get()
 188         if not name:
 189             return
 190         self.method_entry.delete('0', 'end')
 191         self.method_list.insert('end', name)
 192
 193     def synchronize(self):
 194         n = self.name_entry.get()
 195         if not n:
 196             message('Module name not set')
 197             raise oops
 198         if not checkid(n):
 199             message('Module name not an identifier:\n'+n)
 200             raise oops
 201         if not self.abbrev_entry.get():
 202             self.abbrev_entry.insert('end', n)
 203         m = getlistlist(self.method_list)
 204         for n in m:
 205             if not checkid(n):
 206                 message('Method name not an identifier:\n'+n)
 207                 raise oops
 208
 209     def gencode(self, name, objects):
 210         rv = ''
 211         self.synchronize()
 212         for o in objects:
 213             o.synchronize()
 214         onames = []
 215         for i in range(len(objects)):
 216             oname = 'o'+`i+1`
 217             rv = rv + objects[i].gencode(oname)
 218             onames.append(oname)
 219         rv = rv + (name+' = genmodule.module()\n')
 220         rv = rv + (name+'.name = '+`self.name_entry.get()`+'\n')
 221         rv = rv + (name+'.abbrev = '+`self.abbrev_entry.get()`+'\n')
 222         rv = rv + (name+'.methodlist = '+`getlistlist(self.method_list)`+'\n')
 223         rv = rv + (name+'.objects = ['+string.joinfields(onames, ',')+']\n')
 224         rv = rv + ('\n')
 225         return rv
 226
 227 object_number = 0
 228
 229 class UI_object:
 230     def __init__(self, parent):
 231         global object_number
 232
 233         object_number = object_number + 1
 234         self.num = object_number
 235         self.vpref = 'o'+`self.num`+'_'
 236         self.frame = Toplevel(parent.objframe)
 237 #       self.frame.pack()
 238         self.frame.title('Modulator: object view')
 239 #       self.frame = Frame(parent.objframe, {'relief':'raised', 'bd':'0.2m',
 240 #                                            Pack:{'side':'top',
 241 #                                                  'fill':'x'}})
 242         self.f1 = Frame(self.frame, {Pack:{'side':'top', 'pady':'0.5m',
 243                                            'fill':'x'}})
 244         self.f2 = Frame(self.frame, {Pack:{'side':'top', 'pady':'0.5m',
 245                                            'fill':'x'}})
 246         self.f3 = Frame(self.frame, {Pack:{'side':'top', 'pady':'0.5m',
 247                                            'fill':'x'}})
 248         self.f4 = Frame(self.frame, {Pack:{'side':'top', 'pady':'0.5m',
 249                                            'fill':'x'}})
 250
 251
 252         self.l1 = Label(self.f1, {'text':'Object:', Pack:{'side':'left',
 253                                                         'padx':'0.5m'}})
 254         self.name_entry = Entry(self.f1, {'relief':'sunken',
 255                               Pack:{'side':'left', 'padx':'0.5m', 'expand':1}})
 256         self.l2 = Label(self.f1, {'text':'Abbrev:', Pack:{'side':'left',
 257                                                         'padx':'0.5m'}})
 258         self.abbrev_entry = Entry(self.f1, {'relief':'sunken', 'width':5,
 259                               Pack:{'side':'left', 'padx':'0.5m'}})
 260
 261         self.l3 = Label(self.f2, {'text':'Methods:', Pack:{'side':'left',
 262                                                         'padx':'0.5m'}})
 263         self.method_list = ScrolledListbox(self.f2, {'relief':'sunken','bd':2,
 264                                       Pack:{'side':'left', 'expand':1,
 265                                             'padx':'0.5m', 'fill':'both'}})
 266
 267         self.l4 = Label(self.f3, {'text':'Add method:', Pack:{'side':'left',
 268                                                               'padx':'0.5m'}})
 269         self.method_entry = Entry(self.f3, {'relief':'sunken',
 270                              Pack:{'side':'left', 'padx':'0.5m', 'expand':1}})
 271         self.method_entry.bind('<Return>', self.cb_method)
 272         self.delete_button = Button(self.f3, {'text':'Delete method',
 273                                               'command':self.cb_delmethod,
 274                                               Pack:{'side':'left',
 275                                                     'padx':'0.5m'}})
 276
 277
 278         self.l5 = Label(self.f4, {'text':'functions:',
 279                                   Pack:{'side':'left',
 280                                         'padx':'0.5m'}})
 281         self.f5 = Frame(self.f4, {Pack:{'side':'left', 'pady':'0.5m',
 282                                            'fill':'both'}})
 283         self.l6 = Label(self.f4, {'text':'Types:',
 284                                   Pack:{'side':'left', 'padx':'0.5m'}})
 285         self.f6 = Frame(self.f4, {Pack:{'side':'left', 'pady':'0.5m',
 286                                            'fill':'x'}})
 287         self.funcs = {}
 288         for i in genmodule.FUNCLIST:
 289             vname = self.vpref+i
 290             self.f5.setvar(vname, 0)
 291             b = Checkbutton(self.f5, {'variable':vname, 'text':i,
 292                                       Pack:{'side':'top', 'pady':'0.5m',
 293                                             'anchor':'w','expand':1}})
 294             self.funcs[i] = b
 295         self.f5.setvar(self.vpref+'new', 1)
 296
 297         self.types = {}
 298         for i in genmodule.TYPELIST:
 299             vname = self.vpref + i
 300             self.f6.setvar(vname, 0)
 301             b = Checkbutton(self.f6, {'variable':vname, 'text':i,
 302                                       Pack:{'side':'top', 'pady':'0.5m',
 303                                             'anchor':'w'}})
 304             self.types[i] = b
 305
 306     def cb_method(self, *arg):
 307         name = self.method_entry.get()
 308         if not name:
 309             return
 310         self.method_entry.delete('0', 'end')
 311         self.method_list.insert('end', name)
 312
 313     def cb_delmethod(self, *args):
 314         list = self.method_list.curselection()
 315         for i in list:
 316             self.method_list.delete(i)
 317
 318     def synchronize(self):
 319         n = self.name_entry.get()
 320         if not n:
 321             message('Object name not set')
 322             raise oops
 323         if not self.abbrev_entry.get():
 324             self.abbrev_entry.insert('end', n)
 325         n = self.abbrev_entry.get()
 326         if not checkid(n):
 327             message('Abbreviation not an identifier:\n'+n)
 328             raise oops
 329         m = getlistlist(self.method_list)
 330         for n in m:
 331             if not checkid(n):
 332                 message('Method name not an identifier:\n'+n)
 333                 raise oops
 334         if m:
 335             self.f5.setvar(self.vpref+'tp_getattr', 1)
 336         pass
 337
 338     def gencode(self, name):
 339         rv = ''
 340         rv = rv + (name+' = genmodule.object()\n')
 341         rv = rv + (name+'.name = '+`self.name_entry.get()`+'\n')
 342         rv = rv + (name+'.abbrev = '+`self.abbrev_entry.get()`+'\n')
 343         rv = rv + (name+'.methodlist = '+`getlistlist(self.method_list)`+'\n')
 344         fl = []
 345         for fn in genmodule.FUNCLIST:
 346             vname = self.vpref + fn
 347             if self.f5.getvar(vname) == '1':
 348                 fl.append(fn)
 349         rv = rv + (name+'.funclist = '+`fl`+'\n')
 350
 351         fl = []
 352         for fn in genmodule.TYPELIST:
 353             vname = self.vpref + fn
 354             if self.f5.getvar(vname) == '1':
 355                 fl.append(fn)
 356
 357         rv = rv + (name+'.typelist = '+`fl`+'\n')
 358
 359         rv = rv + ('\n')
 360         return rv
 361
 362
 363 def main():
 364     if len(sys.argv) < 2:
 365         ui = UI()
 366         ui.run()
 367     elif len(sys.argv) == 2:
 368         fp = open(sys.argv[1])
 369         pycode = fp.read()
 370         try:
 371             exec pycode
 372         except:
 373             sys.stderr.write('An error occurred:-)\n')
 374             sys.exit(1)
 375         ##genmodule.write(sys.stdout, m)
 376     else:
 377         sys.stderr.write('Usage: modulator [file]\n')
 378         sys.exit(1)
 379
 380 main()