out.write('}\n')
if __name__ == '__main__':
- import getopt
+ import os, getopt
mad_debug = 0
(options, arguments) = getopt.getopt(sys.argv[1:], "d:")
for (opt, val) in options:
if arguments == ['-']:
generate(sys.stdout)
elif len(arguments) == 1:
- out = open(arguments[0][:-4] + ".c", "w")
+ stem = arguments[0][:-4]
+ out = open(stem + ".c", "w")
generate(out)
out.close()
+ compile = "cc %s.c -lm -o %s" % (stem, stem)
+ if os.system(compile) == 0:
+ os.remove(stem + ".c")
+ else:
+ print "Compilation failed"
+ out.close()
else:
sys.stderr.write("mad: multiple file args are not yet supported\n")
sys.exit(1)