#!/usr/bin/env python

import sys

sys.path.append("/home/paul/g/fetched/corepy-1.0")

import time

import array

import corepy.arch.x86.platform as env

import corepy.arch.x86.isa as x86

import corepy.arch.x86.types.registers as r

from corepy.arch.x86.lib.memory import MemRef

from corepy.lib.extarray import extarray

def timecmd(name, cmd, r):

    c = 0

    total = 0

    while total < 2:

        t = time.time()

        cmd()

        d = time.time()

        total += d - t

        c += 1

    print "%s %5.1f" % (name, total*(2.4E9)/(r*c))

reps = (1<<6)

code = env.InstructionStream()

code.add(x86.nop())

def callasm():

    for i in range(reps):

        params = env.ExecParams()

        env.Processor().execute(code)

timecmd("callasm", callasm, reps)

datasize = (1<<14)

def timeaccess(name, items, mask, store):

    def timereps():

        for i in xrange(reps):

            for j in xrange(items):

               store[j] ^= mask

    timecmd(name, timereps, datasize*reps)

timeaccess("snative", datasize>>2, 3, [0] * (datasize >> 2))

timeaccess("native", datasize>>2, 0xffffffff, [0] * (datasize >> 2))

timeaccess("sarray", datasize>>2, 3, array.array('L', [0] * (datasize >> 2)))

timeaccess("array", datasize>>2, 0xfffffff, array.array('L', [0] * (datasize >> 2)))

timeaccess("sulongs", datasize>>2, 3,  extarray('L', datasize >> 2))

timeaccess("ulongs", datasize>>2, 0xffffffff,  extarray('L', datasize >> 2))

timeaccess("ushorts", datasize>>1, 0xffff, extarray('H', datasize >> 1))