1 #!/usr/local/bin/python 2 # -*- coding: iso-8859-1 -*- 3 4 # Id 5 6 # Copyright (c) 2004 Kungliga Tekniska Hgskolan 7 # (Royal Institute of Technology, Stockholm, Sweden). 8 # All rights reserved. 9 # 10 # Redistribution and use in source and binary forms, with or without 11 # modification, are permitted provided that the following conditions 12 # are met: 13 # 14 # 1. Redistributions of source code must retain the above copyright 15 # notice, this list of conditions and the following disclaimer. 16 # 17 # 2. Redistributions in binary form must reproduce the above copyright 18 # notice, this list of conditions and the following disclaimer in the 19 # documentation and/or other materials provided with the distribution. 20 # 21 # 3. Neither the name of the Institute nor the names of its contributors 22 # may be used to endorse or promote products derived from this software 23 # without specific prior written permission. 24 # 25 # THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND 26 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 27 # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 28 # ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE 29 # FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 30 # DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 31 # OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 32 # HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 33 # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 34 # OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 35 # SUCH DAMAGE. 36 37 import datetime 38 import string 39 import os 40 import sys 41 42 class GeneratedFile : 43 "Represents a generated file" 44 def __init__(self, name) : 45 "Create a new GeneratedFile with name" 46 self.name = os.path.basename(name) 47 self.file = open(name, 'w') 48 self.file.write('/* ' + name + ' */\n') 49 self.file.write('/* Automatically generated at ' + 50 datetime.datetime.now().isoformat() + 51 ' */\n\n') 52 53 def close(self) : 54 """End and close the file header""" 55 self.file.close() 56 57 58 class Header(GeneratedFile) : 59 "Represents a generated header file" 60 if sys.hexversion>0x300000f: 61 guardTrans = bytes.maketrans(b'-.', b'__') 62 else: 63 import string 64 guardTrans = string.maketrans('-.', '__') 65 def makeGuard(self) : 66 """Return a name to be used as ifdef guard""" 67 return self.name.translate(self.guardTrans).upper() 68 69 def __init__(self, name) : 70 "Create a new Header with name" 71 GeneratedFile.__init__(self, name) 72 self.guard = self.makeGuard() 73 self.file.write('#ifndef ' + self.guard + '\n') 74 self.file.write('#define ' + self.guard + ' 1\n') 75 76 def close(self) : 77 """End and close the file header""" 78 self.file.write('#endif /* ' + self.guard + ' */\n') 79 GeneratedFile.close(self) 80 81 82 class Implementation(GeneratedFile) : 83 "Represents a generated implementation file" 84 def __init__(self, name) : 85 "Create a new Implementation with name" 86 GeneratedFile.__init__(self, name) 87