| """ | |
| File: | |
| JetSegGraph.py | |
| Contents and purpose: | |
| Draws the event graph and progress bar | |
| Copyright (c) 2008 Android Open Source Project | |
| Licensed under the Apache License, Version 2.0 (the "License"); | |
| you may not use this file except in compliance with the License. | |
| You may obtain a copy of the License at | |
| http://www.apache.org/licenses/LICENSE-2.0 | |
| Unless required by applicable law or agreed to in writing, software | |
| distributed under the License is distributed on an "AS IS" BASIS, | |
| WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
| See the License for the specific language governing permissions and | |
| limitations under the License. | |
| """ | |
| import wx | |
| import logging | |
| from JetUtils import * | |
| from JetDefs import * | |
| GRAPH_COLORS = [ | |
| '#C0E272', | |
| '#85CF89', | |
| '#CF9683', | |
| '#749EDE', | |
| '#9FB5B1', | |
| '#B095BF', | |
| '#FE546D', | |
| '#B3BB97', | |
| '#FFFFB8', | |
| ] | |
| PROGRESS_BAR = '#0000CC' | |
| EOS_BAR = '#095000' | |
| APP_BAR = '#B3BB97' | |
| class Marker(): | |
| """ Defines portions of the graph for events """ | |
| def __init__(self, sEventType, iEventId, sName, sStartMbt, sEndMbt, iStartMeasure, ppqn): | |
| self.sEventType = sEventType | |
| self.iEventId = iEventId | |
| self.sName = sName | |
| self.StartMbt = ConvertStrTimeToTuple(sStartMbt) | |
| self.EndMbt = ConvertStrTimeToTuple(sEndMbt) | |
| self.iStartMeasure = iStartMeasure | |
| self.iStart = 0 | |
| self.iEnd = 0 | |
| self.iWidth = 0 | |
| self.iHeight = 0 | |
| self.iTop = 0 | |
| self.iUpdate = False | |
| self.sColor = '#FFFFB8' | |
| self.ppqn = ppqn | |
| self.isDirty = False | |
| def CalcCoord(self, step, height, ColorFct): | |
| """ Calculates the coordinates in pixels for graphing the shaded regions """ | |
| #measures | |
| iStartM = self.StartMbt[0] - self.iStartMeasure | |
| iEndM = self.EndMbt[0] - self.iStartMeasure | |
| self.iStart = step * iStartM | |
| self.iEnd = step * iEndM | |
| #beats | |
| self.iStart = self.iStart + ((step / 4.0) * (self.StartMbt[1]-1)) | |
| self.iEnd = self.iEnd + ((step / 4.0) * (self.EndMbt[1]-1)) | |
| #ticks | |
| pctTickOfBeat = (float(self.StartMbt[2]) / float(self.ppqn)) | |
| self.iStart = self.iStart + ((pctTickOfBeat * (step / 4.0))) | |
| pctTickOfBeat = (float(self.EndMbt[2]) / float(self.ppqn)) | |
| self.iEnd = self.iEnd + ((pctTickOfBeat * (step / 4.0))) | |
| self.iWidth = self.iEnd - self.iStart | |
| self.iHeight = height | |
| self.sColor = ColorFct() | |
| self.iUpdate = False | |
| class SegmentGraph(wx.Panel): | |
| """ Draws the player graph bar """ | |
| def __init__(self, parent, pos=wx.DefaultPosition, size=wx.DefaultSize, ClickCallbackFct=None, showLabels=True, showClips=True, showAppEvts=True): | |
| wx.Panel.__init__(self, parent, -1, pos=pos, size=size, style=wx.BORDER_STATIC) | |
| self.iLocationInMs = 0 | |
| self.iLengthInMs = 0 | |
| self.iLengthInMeasures = 0 | |
| self.iMarkerTop = 15 | |
| self.iScaleTop = 0 | |
| self.iEdges = 5 | |
| self.iStartMeasure = 0 | |
| self.iMidiMode = False | |
| self.ClickCallbackFct = ClickCallbackFct | |
| self.iColor = 0 | |
| self.showLabels = showLabels | |
| self.showClips = showClips | |
| self.showAppEvts = showAppEvts | |
| self.font = wx.Font(8, wx.FONTFAMILY_DEFAULT, wx.FONTSTYLE_NORMAL, wx.FONTWEIGHT_NORMAL, False, 'Courier') | |
| self.Markers = [] | |
| self.SetBackgroundStyle(wx.BG_STYLE_CUSTOM) | |
| self.Bind(wx.EVT_PAINT, self.OnPaint) | |
| self.Bind(wx.EVT_SIZE, self.OnSize) | |
| self.Bind(wx.EVT_LEFT_DOWN, self.OnLeftDown) | |
| #initialize buffer | |
| self.OnSize(None) | |
| def ClearGraph(self): | |
| """ Clears the graph values """ | |
| self.iLocationInMs = 0 | |
| self.iLengthInMs = 0 | |
| self.iLengthInMeasures = 0 | |
| self.iMarkerTop = 15 | |
| self.iScaleTop = 0 | |
| self.iEdges = 5 | |
| self.iStartMeasure = 0 | |
| self.iMidiMode = False | |
| self.iColor = 0 | |
| self.Markers = [] | |
| self.iLocationInMs = 0 | |
| self.DoDrawing() | |
| def LoadSegment(self, segment, segMarker=None, iMidiMode=False, showLabels=True, showClips=True, showAppEvts=True): | |
| """ Loads up the segment drawing the graph """ | |
| if segment is None: | |
| self.ClearGraph() | |
| return None | |
| self.iMidiMode = iMidiMode | |
| self.showLabels = showLabels | |
| self.showClips = showClips | |
| self.showAppEvts = showAppEvts | |
| self.Markers = [] | |
| self.iLocationInMs = 0 | |
| info = MidiSegInfo(segment) | |
| #disable graph for debugging | |
| #return info | |
| self.iLengthInMs = info.iLengthInMs | |
| self.ppqn = info.ppqn | |
| self.StartMbt = mbtFct(ConvertStrTimeToTuple(segment.start), 1) | |
| self.EndMbt = mbtFct(ConvertStrTimeToTuple(segment.end), 1) | |
| self.LengthMbt = None | |
| self.iStartMeasure = self.StartMbt[0] | |
| self.iLengthInMeasures = self.EndMbt[0] - self.StartMbt[0] | |
| for jet_event in segment.jetevents: | |
| if self.showClips and jet_event.event_type == JetDefs.E_CLIP: | |
| self.AddMarker(JetDefs.E_CLIP, jet_event.event_id, jet_event.event_name, mbtFct(jet_event.event_start,1), mbtFct(jet_event.event_end,1), self.iStartMeasure, self.ppqn) | |
| elif jet_event.event_type == JetDefs.E_EOS: | |
| self.AddMarker(JetDefs.E_EOS, jet_event.event_id, jet_event.event_name, mbtFct(jet_event.event_end,1), mbtFct(jet_event.event_end,1), self.iStartMeasure, self.ppqn) | |
| elif self.showAppEvts and jet_event.event_type == JetDefs.E_APP: | |
| self.AddMarker(JetDefs.E_APP, jet_event.event_id, jet_event.event_name, mbtFct(jet_event.event_start,1), mbtFct(jet_event.event_end,1), self.iStartMeasure, self.ppqn) | |
| if segMarker is not None: | |
| self.AddMarker(JetDefs.E_CLIP, 0, segMarker[0], mbtFct(segMarker[1],1), mbtFct(segMarker[2],1), self.iStartMeasure, self.ppqn) | |
| self.DoDrawing() | |
| return info | |
| def AddMarker(self, sEventType, iEventId, sName, sStartMbt, sEndMbt, iStartMeasure, ppqn): | |
| """ Adds a marker to the list """ | |
| if not CompareMbt(sStartMbt, sEndMbt): | |
| sEndMbt = sStartMbt | |
| self.Markers.append(Marker(sEventType, iEventId, sName, sStartMbt, sEndMbt, iStartMeasure, ppqn)) | |
| def OnLeftDown(self, event): | |
| """ Calls the function assicated with an event """ | |
| pt = event.GetPosition() | |
| for Marker in self.Markers: | |
| if pt[0] >= Marker.iStart and pt[0] <= Marker.iEnd and pt[1] >= Marker.iTop and pt[1] <= Marker.iTop + Marker.iHeight: | |
| if self.ClickCallbackFct != None: | |
| self.ClickCallbackFct(Marker.sName, Marker.iEventId) | |
| def GetAColor(self): | |
| """ Gets a color """ | |
| color = GRAPH_COLORS[self.iColor] | |
| self.iColor = self.iColor + 1 | |
| if self.iColor >= len(GRAPH_COLORS): | |
| self.iColor = 0 | |
| return color | |
| def OnSize(self, event=None): | |
| """ Repaints for resizing of screen """ | |
| if OsWindows(): | |
| # The Buffer init is done here, to make sure the buffer is always | |
| # the same size as the Window | |
| Size = self.GetClientSizeTuple() | |
| # Make new offscreen bitmap: this bitmap will always have the | |
| # current drawing in it, so it can be used to save the image to | |
| # a file, or whatever. | |
| self._Buffer = wx.EmptyBitmap(*Size) | |
| self.DoDrawing(None) | |
| if event is not None: | |
| event.Skip() | |
| def OnPaint(self, event=None): | |
| """ Painting of windows """ | |
| if OsWindows(): | |
| dc = wx.BufferedPaintDC(self, self._Buffer) | |
| else: | |
| dc = wx.AutoBufferedPaintDC(self) | |
| dc.Background = wx.Brush(wx.WHITE) | |
| self.DoDrawing(dc) | |
| def DoDrawing(self, dc=None): | |
| """ Does the actual drawing of the control """ | |
| if dc is None: | |
| if OsWindows(): | |
| dc = wx.BufferedDC(wx.ClientDC(self), self._Buffer) | |
| else: | |
| dc = wx.AutoBufferedPaintDC(self) | |
| dc.Background = wx.Brush(wx.WHITE) | |
| dc.Clear() | |
| self.iColor = 0 | |
| gWidth, gHeight = self.GetSize() | |
| gWidth = gWidth - (self.iEdges * 2) | |
| step = int(gWidth / (self.iLengthInMeasures + .01)) | |
| for Marker in self.Markers: | |
| Marker.CalcCoord(step, gHeight, self.GetAColor) | |
| """ eliminate overlaps; establish colors """ | |
| iClips = 0 | |
| iMarkers = 0 | |
| for index, Marker in enumerate(self.Markers): | |
| if Marker.sEventType == JetDefs.E_CLIP: | |
| iClips = iClips + 1 | |
| iOverlaps = 1 | |
| for index1, Marker1 in enumerate(self.Markers): | |
| if Marker.sEventType == JetDefs.E_CLIP: | |
| if index != index1 and not Marker1.iUpdate: | |
| if Marker.iStart <= Marker1.iStart and Marker.iEnd <= Marker1.iEnd and Marker.iEnd >= Marker1.iStart: | |
| iOverlaps = iOverlaps + 1 | |
| Marker.iUpdate = True | |
| Marker1.iUpdate = True | |
| if not Marker.iUpdate and Marker.iStart >= Marker1.iStart and Marker.iEnd >= Marker1.iEnd and Marker.iStart <= Marker1.iEnd: | |
| iOverlaps = iOverlaps + 1 | |
| Marker.iUpdate = True | |
| Marker1.iUpdate = True | |
| if iOverlaps > 1: | |
| iTop = 0 | |
| for index1, Marker1 in enumerate(self.Markers): | |
| if Marker.sEventType == JetDefs.E_CLIP: | |
| if Marker1.iUpdate: | |
| Marker1.iHeight = gHeight / iOverlaps | |
| Marker1.iTop = iTop * Marker1.iHeight | |
| iTop = iTop + 1 | |
| elif Marker.sEventType == JetDefs.E_APP: | |
| iMarkers = iMarkers + 1 | |
| for Marker in self.Markers: | |
| if Marker.sEventType == JetDefs.E_CLIP: | |
| dc.SetPen(wx.Pen(Marker.sColor)) | |
| dc.SetBrush(wx.Brush(Marker.sColor)) | |
| dc.DrawRectangle(Marker.iStart + self.iEdges, Marker.iTop, Marker.iWidth, Marker.iHeight) | |
| width, height = dc.GetTextExtent(Marker.sName) | |
| k = ((Marker.iStart + Marker.iEnd) / 2) - (width/2) + self.iEdges | |
| if self.showLabels or self.iMidiMode: | |
| dc.DrawText(Marker.sName, k, ((Marker.iTop+Marker.iHeight/2) - (height*.5))) | |
| if self.iMidiMode: | |
| self.iMidiModeStart = Marker.iStart | |
| elif Marker.sEventType == JetDefs.E_EOS: | |
| dc.SetPen(wx.Pen(EOS_BAR)) | |
| dc.SetBrush(wx.Brush(EOS_BAR)) | |
| dc.DrawRectangle(Marker.iStart + self.iEdges, Marker.iTop, 1, Marker.iHeight) | |
| width, height = dc.GetTextExtent(Marker.sName) | |
| k = Marker.iStart - (width/2) + self.iEdges | |
| dc.DrawText(Marker.sName, k, ((Marker.iTop+Marker.iHeight/2) - (height*.5))) | |
| elif Marker.sEventType == JetDefs.E_APP: | |
| dc.SetPen(wx.Pen(APP_BAR)) | |
| dc.SetBrush(wx.Brush(APP_BAR)) | |
| dc.DrawRectangle(Marker.iStart + self.iEdges, Marker.iTop, 1, Marker.iHeight) | |
| width, height = dc.GetTextExtent(Marker.sName) | |
| k = Marker.iStart - (width/2) + self.iEdges | |
| if self.showLabels or self.iMidiMode: | |
| dc.DrawText(Marker.sName, k, ((Marker.iTop+Marker.iHeight/2) - (height*.5))) | |
| """ Draw scale """ | |
| if gWidth == 0: | |
| iDiv = 50 | |
| else: | |
| iDiv = (gWidth)/18 | |
| if iDiv == 0: | |
| iDiv = 50 | |
| scale = ((self.iLengthInMeasures / iDiv) + 1) | |
| if scale == 0: | |
| scale = 1 | |
| beatStep = step / 4.0 | |
| dc.SetFont(self.font) | |
| j = 0 | |
| lastEnd = 0 | |
| num = range(self.iStartMeasure, self.iStartMeasure + self.iLengthInMeasures + 1, 1) | |
| dc.SetPen(wx.Pen('#5C5142')) | |
| for i in range(0, (self.iLengthInMeasures+1)*step, step): | |
| k = i + self.iEdges | |
| dc.DrawLine(k, self.iScaleTop, k, self.iScaleTop+8) | |
| if i != (self.iLengthInMeasures)*step: | |
| for iBeat in range(1,4): | |
| k = i+(iBeat * beatStep) + self.iEdges | |
| dc.DrawLine(k, self.iScaleTop, k, self.iScaleTop+4) | |
| width, height = dc.GetTextExtent(str(num[j])) | |
| k = i-(width/2) + self.iEdges | |
| if k > lastEnd: | |
| if j == 0 or (j % scale) == 0: | |
| dc.DrawText(str(num[j]), k, self.iScaleTop+8) | |
| lastEnd = k + width | |
| j = j + 1 | |
| """ Updates the location bar in case screen moved or resized """ | |
| if self.iLocationInMs > 0 and self.iLengthInMs > 0: | |
| iOffset = 0 | |
| if self.iMidiMode: | |
| iOffset = self.iMidiModeStart | |
| till = gWidth * (self.iLocationInMs / self.iLengthInMs) | |
| dc.SetPen(wx.Pen(PROGRESS_BAR)) | |
| dc.SetBrush(wx.Brush(PROGRESS_BAR)) | |
| dc.DrawRectangle(self.iEdges + iOffset, gHeight-6, till, 3) | |
| def UpdateLocation(self, iLocationInMs): | |
| """ Updates the location bar """ | |
| #disable graph for debugging | |
| #return info | |
| self.iLocationInMs = iLocationInMs | |
| if self.iLocationInMs > 0 and self.iLengthInMs > 0: | |
| if OsWindows(): | |
| dc = wx.BufferedDC(wx.ClientDC(self), self._Buffer) | |
| else: | |
| dc = wx.AutoBufferedPaintDC(self) | |
| dc.Background = wx.Brush(wx.WHITE) | |
| iOffset = 0 | |
| if self.iMidiMode: | |
| iOffset = self.iMidiModeStart | |
| gWidth, gHeight = self.GetSize() | |
| gWidth = gWidth - (self.iEdges * 2) | |
| till = gWidth * (self.iLocationInMs / self.iLengthInMs) | |
| dc.SetPen(wx.Pen(PROGRESS_BAR)) | |
| dc.SetBrush(wx.Brush(PROGRESS_BAR)) | |
| dc.DrawRectangle(self.iEdges + iOffset, gHeight-6, till, 3) | |
| self.isDirty = True | |
| else: | |
| if self.isDirty: | |
| self.DoDrawing() | |
| self.isDirty = False |