# Pymel vs OpenMaya speed testing

# Pymel vs OpenMaya speed testing
import maya.api.OpenMaya as om
from pymel.core import *
import time


sl = selected()
itterAmnt = 100000 # 10,000

#### OpenMaya
startTimeOM = time.clock()
randIOM = om.MVector(1,2,3)
p1OM = om.MVector(sl[0].getTranslation())
p2OM = om.MVector(sl[1].getTranslation())
omList = []
for i in range(0,itterAmnt):
    #dist = om.MVector(p1OM-p2OM).length()
    dist = math.sqrt((p1OM[0] - p2OM[0])**2 + (p1OM[1] - p2OM[1])**2 + (p1OM[2] - p2OM[2])**2)
    someMath = om.MVector(p1OM + randIOM)
    om.MVector.normalize(someMath)
    omList.append(someMath)
elapsedOM  = (time.clock() - startTimeOM)
###



#### Pymel
startTimePM = time.clock()
randIPM = dt.Vector(1,2,3)
p1PM = dt.Vector(sl[0].getTranslation())
p2PM = dt.Vector(sl[1].getTranslation())
pmList = []
for i in range(0,itterAmnt):
    #dist = dt.Vector(p1PM-p2PM).length()
    dist = math.sqrt((p1PM[0] - p2PM[0])**2 + (p1PM[1] - p2PM[1])**2 + (p1PM[2] - p2PM[2])**2)
    someMath = dt.Vector(p1PM + randIPM)
    dt.Vector.normalize(someMath)
    pmList.append(someMath)
elapsedPM  = (time.clock() - startTimePM)
###



print "\n"," Total itterations:", itterAmnt
print "\n","Run time OpenMaya:", elapsedOM
print "Run time Pymel:   ", elapsedPM