step3.py

import sys

def write_prepare(f):
    f.write("#include <stdio.h>\n#include <stdlib.h>\n#include <time.h>\n#include <stdint.h>\n\n")
    f.write("union ticks {\n")
    f.write("\tuint64_t t64;\n")
    f.write("\tstruct s32{\n")
    f.write("\t\tuint32_t th,tl;\n")
    f.write("\t} t32;\n")
    f.write("};\n\n")


def write_main(f,L1,L2):
    f.write("void main() {\n")
    f.write("\tunion ticks start, end;\n")
    f.write("\tint long unsigned i, liter = 10000;\n")
    f.write("\tint a,b,c;\n")
    f.write("\tfor (i = 0; i < liter; ++i){\n")
    f.write("\t\tif ((i % "+L1+")==0)a=1;\n")
    f.write("\t\telse a=0;\n")
    f.write("\t\tif ((i % "+L2+")==0)b=1;\n")
    f.write("\t\telse b=0;\n")
    f.write("\t\t__asm__(\"rdtsc\\n\\t\":\"=a\"(start.t32.th),\"=d\"(start.t32.tl));\n")
    f.write("\t\tif ((a*b)==1) c=1;")
    f.write("\t\t__asm__(\"rdtsc\\n\\t\":\"=a\"(end.t32.th),\"=d\"(end.t32.tl));\n")
    #f.write("\t\tif ((i % ("+str(L)+"-1))==0)\n")
    f.write("\tprintf(\"%u %lu\\n\", i, end.t64-start.t64);\n")
    f.write("\t}\n")
    f.write("}")

def main():
    if (len(sys.argv)!=3):
        print("Invalid number of arguments\n")
        return

    f = open("step3.c", "w")
    write_prepare(f)
    write_main(f, sys.argv[1], sys.argv[2])

if __name__ == "__main__":
    main()