'''

User interface for stopwatch

'''

with utl.MlUi('ml_stopwatch', 'Stopwatch', width=400, height=175, info='''Press the start button to start recording.

Continue pressing to set marks.

When finished, press the stop button and the report will pop up.''') as win:

mc.checkBoxGrp('ml_stopwatch_round_checkBox',label='Round to nearest frame', value1=True, annotation='Only whole number frames')

mc.text('ml_stopwatch_countdown_text', label='Ready...')

mc.button('ml_stopwatch_main_button', label='Start', height=80)

_setButtonStart()

'''

Run when the start button is pressed, gathers args and starts the stopwatch.

'''

global STOPWATCH

_setButtonMark()

rounded = mc.checkBoxGrp('ml_stopwatch_round_checkBox', query=True, value1=True)

kwargs = dict()

kwargs['startFrame'], null = utl.frameRange()

if rounded:

kwargs['roundTo'] = 0

STOPWATCH = Stopwatch(**kwargs)

'''

Run when the Mark button is pressed, records a mark with the stopwatch.

'''

global STOPWATCH

global STOPWATCH

_setButtonStart()

if not STOPWATCH:

return

'''

Resets the UI text and button labels.

'''

mc.button('ml_stopwatch_main_button', edit=True, label='Start', command=_startButton, annotation='Start the timer. Once running, use this same button to set marks.')

'''

Sets the UI text and button labels during recording.

'''

mc.button('ml_stopwatch_main_button', edit=True, label='Mark', command=_markButton, annotation='Record a mark.')

'''

This is temp and will possibly be rolled into future releases.

'''

start,end = utl.frameRange()

camera = utl.getCurrentCamera()

camShape = mc.listRelatives(camera, shapes=True)[0]

aov = mc.getAttr(camShape+'.horizontalFilmAperture')

name = 'ml_stopwatch_'

numStopwatches = len(mc.ls(name+'*', type='locator'))

top = mc.spaceLocator(name=name+'#')

ename = ':'.join([str(x) for x in marks])

mc.addAttr(top, longName='keyTimes', at='enum', enumName=ename, keyable=True)

markRange = float(marks[-1]-marks[0])

viewWidth = aov*2

viewHeight = -0.4*aov+(numStopwatches*aov*0.08)

depth = 5

for mark in marks[1:-1]:

ann = mc.annotate(top, text=str(mark))

mc.setAttr(ann+'.displayArrow', 0)

#parent

annT = mc.parent(mc.listRelatives(ann, parent=True, path=True), top)[0]

annT = mc.rename(annT, 'mark_'+str(round(mark)))

ann = mc.listRelatives(annT, shapes=True, path=True)[0]

#set the position

normalX = float(mark-marks[0])/markRange-0.5

mc.setAttr(annT+'.translateX', viewWidth*normalX*2)

mc.setAttr(annT+'.translateY', viewHeight)

mc.setAttr(annT+'.translateZ', -depth)

#keyframe for color

mc.setAttr(ann+'.overrideEnabled', 1)

mc.setKeyframe(ann, attribute='overrideColor', value=17, time=(int(marks[0]-1),int(mark+1)))

mc.setKeyframe(ann, attribute='overrideColor', value=13, time=(int(mark),))

mc.keyTangent(ann+'.overrideColor', ott='step')

mc.select(clear=True)

def __init__(self, startFrame=0, endFrame=None, addMarkers=True, roundTo=2):

'''

Initializes the stopwatch.

'''

self.frameRate = utl.getFrameRate()

self.startFrame = int(startFrame)

self.endFrame = endFrame

self.addMarkers = True

self.roundTo = roundTo

self.markTime = list()

def start(self):

'''

Starts the stopwatch.

'''

self.startTime = time.time()

def stop(self):

'''

Stop the stopwatch and convert the time to frames.

'''

if not self.markTime:

OpenMaya.MGlobal.displayWarning('No marks recorded, the "Mark" button needs to be pressed at least once after recording is started.')

return

#don't mark the last frame

#self.mark()

self.elapsed = self.markTime[-1] - self.startTime

#mark frames

self.frameMarks = [0]

for m in self.markTime:

seconds = m - self.startTime

frame = (seconds*self.frameRate)

frame = round(frame, self.roundTo)

if not self.roundTo:

frame = int(frame)

self.frameMarks.append(frame)

self.ui()

def mark(self):

'''

Save the current time.

'''

self.markTime.append(time.time())

def ui(self):

'''

Launch a UI to display the marks that were recorded.

'''

with utl.MlUi('ml_stopwatchReport', 'Stopwatch Report', width=400, height=400, info='''This is the report from the stopwatch you just ran.

Adjust the start frame, and then press the frame buttons to jump to that frame.

The fields on the right can be used for notes.''', menu=False) as win:

self.uiSlider = mc.intSliderGrp(field=True, label='Start Frame', value=self.startFrame,

minValue=((-0.5*self.frameMarks[-1])+self.startFrame), maxValue=(self.frameMarks[-1]/2)+self.startFrame,

fieldMinValue=-1000, fieldMaxValue=1000,

changeCommand=self.uiUpdateStartFrame)

self.frameRateField = mc.intFieldGrp(label='Frame Rate', value1=self.frameRate, enable1=False, extraLabel='fps', annotation='')

mc.scrollLayout()

mc.rowColumnLayout(numberOfColumns=3, columnWidth=[(1, 50), (2, 80), (3, 340)])

mc.text('Frame')

mc.text('Duration')

mc.text('Notes')

for i in range(3):

mc.separator(style='single', height=15)

self.uiButton = list()

for i in range(len(self.frameMarks)):

#frame button

frame = self.frameMarks[i]+self.startFrame

self.uiButton.append(mc.button(label=str(frame), annotation='Go to frame %i' % frame,command='import maya.cmds;maya.cmds.currentTime(%i,edit=True)' % frame))

#duration text

if i:

mc.text(label=str(self.frameMarks[i]-self.frameMarks[i-1]))

else:

mc.text(label='Start')

#notes field

mc.textField()

#add the stop

mc.text(label='')

mc.text(label='Stop')

mc.setParent('..')

mc.setParent('..')

#next and prev buttons!

mc.paneLayout(configuration='vertical2',separatorThickness=1)

mc.button(label='<< Previous', command=self.previousFrame, annotation='Go to the previous frame in the list.')

mc.button(label='Next >>', command=self.nextFrame, annotation='Go to the next frame in the list.')

def previousFrame(self, *args):

'''

Go to the previous frame based on the current time and frame list.

'''

current = mc.currentTime(query=True)

for f in [x+self.startFrame for x in reversed(self.frameMarks)]:

if current <= f:

continue

mc.currentTime(f)

break

def nextFrame(self, *args):

'''

Go to the next frame based on the current time and frame list.

'''

current = mc.currentTime(query=True)

for f in [x+self.startFrame for x in self.frameMarks]:

if current >= f:

continue

mc.currentTime(f)

break

def uiUpdateStartFrame(self, *args):

'''

Update the UI text with the new frames when the slider is updated.

'''

self.startFrame = mc.intSliderGrp(self.uiSlider, query=True, value=True)

for b,f in zip(self.uiButton, self.frameMarks):

frame = f+self.startFrame