Das deutsche Python-Forum

Code: Alles auswählen

#!/usr/bin/env python
# -*- coding: iso-8859-15 -*-

import wx
import wx.lib.plot as plot


wx.SetDefaultPyEncoding("iso-8859-15")


class MyFrame(wx.Frame):
   
    def __init__(
        self, parent = None, id = -1, title = "Example", size = wx.Size(500, 500)
    ):
        wx.Frame.__init__(self, parent, id, title, size = size)
       
        panel = wx.Panel(self)
       
        vbox_main = wx.BoxSizer(wx.VERTICAL)
        panel.SetSizer(vbox_main)
       
        vbox = wx.BoxSizer(wx.VERTICAL)
        vbox_main.Add(vbox, 1, wx.ALL | wx.EXPAND, 5)
       
        canvas = plot.PlotCanvas(panel)
        vbox.Add(canvas, 1, wx.EXPAND | wx.ALL, 5)
       
        coordinates1 = (
            (1, 50), (2, 60), (3, 45), (4, 60), (5, 33), (6, 79),
        )
        coordinates2 = (
            (1, 40), (2, 30), (3, 65), (4, 80), (5, 23), (6, 19),
        )
        lines = (
            plot.PolyLine(
                coordinates1, colour = "green", legend = u" Jänner", width = 1
            ),
            plot.PolyLine(
                coordinates2, colour = "blue", legend = u" Februar", width = 1
            ),
        )
       
        graphics = plot.PlotGraphics(
            objects = lines, title = u"Monatsübersicht", xLabel = u"Monate",
            yLabel = u"Anzahl Verkäufe"
        )
       
        canvas.SetFont(wx.Font(10,wx.SWISS,wx.NORMAL,wx.NORMAL))
        canvas.SetFontSizeAxis(10)
        canvas.SetFontSizeLegend(7)
        canvas.SetEnableLegend(True)
        canvas.SetXSpec('auto')
        canvas.SetYSpec('auto')
        canvas.SetEnableGrid(True)
       
        canvas.Draw(graphics, xAxis = (0, 12), yAxis = (0, 100))
        #canvas.Draw(graphics)


def main():
    """Testing"""
   
    app = wx.PySimpleApp()
    f = MyFrame()
    f.Center()
    f.Show()
    app.MainLoop()


if __name__ == "__main__":
    main()

Code: Alles auswählen

# -*- coding: iso-8859-15 -*-

import wx
import wx.lib.plot as plot


wx.SetDefaultPyEncoding("iso-8859-15")


class MyFrame(wx.Frame):
   
    def __init__(
        self, parent = None, id = -1, title = "Example", size = wx.Size(500, 500)
    ):
        wx.Frame.__init__(self, parent, id, title, size = size)
       
        panel = wx.Panel(self)
       
        vbox_main = wx.BoxSizer(wx.VERTICAL)
        panel.SetSizer(vbox_main)        
       
        vbox = wx.BoxSizer(wx.VERTICAL)
        vbox_main.Add(vbox, 1, wx.ALL | wx.EXPAND, 5)
       
        self.client = plot.PlotCanvas(panel)

        ##new declarations
        
        #define the function for drawing pointLabels
        self.client.SetPointLabelFunc(self.DrawPointLabel)
        # Create mouse event for showing cursor coords in status bar
        self.client.canvas.Bind(wx.EVT_LEFT_DOWN, self.OnMouseLeftDown)
        # Show closest point when enabled
        self.client.canvas.Bind(wx.EVT_MOTION, self.OnMotion)
        ##end of new declarations

        
        vbox.Add(self.client, 1, wx.EXPAND | wx.ALL, 5)

        
       
        coordinates1 = (
            (1, 50), (2, 60), (3, 45), (4, 60), (5, 33), (6, 79),
        )
        coordinates2 = (
            (1, 40), (2, 30), (3, 65), (4, 80), (5, 23), (6, 19),
        )
        lines = (
            plot.PolyLine(
                coordinates1, colour = "green", legend = u" Jänner", width = 1
            ),
            plot.PolyLine(
                coordinates2, colour = "blue", legend = u" Februar", width = 1
            ),
        )
       
        graphics = plot.PlotGraphics(
            objects = lines, title = u"Monatsübersicht", xLabel = u"Monate",
            yLabel = u"Anzahl Verkäufe"
        )
       
        self.client.SetFont(wx.Font(10,wx.SWISS,wx.NORMAL,wx.NORMAL))
        self.client.SetFontSizeAxis(10)
        self.client.SetFontSizeLegend(7)
        self.client.SetEnableLegend(True)
        self.client.SetXSpec('auto')
        self.client.SetYSpec('auto')
        self.client.SetEnableGrid(True)

        ##Activate Labels        
        self.client.SetEnablePointLabel(True)
       
        self.client.Draw(graphics)

        ##show Statusbar for Mouse Position
        self.CreateStatusBar(1)        


    ##Functions from plot.py
    def DrawPointLabel(self, dc, mDataDict):
        """This is the fuction that defines how the pointLabels are plotted
            dc - DC that will be passed
            mDataDict - Dictionary of data that you want to use for the pointLabel

            As an example I have decided I want a box at the curve point
            with some text information about the curve plotted below.
            Any wxDC method can be used.
        """
        # ----------
        dc.SetPen(wx.Pen(wx.BLACK))
        dc.SetBrush(wx.Brush( wx.BLACK, wx.SOLID ) )
        
        sx, sy = mDataDict["scaledXY"] #scaled x,y of closest point
        dc.DrawRectangle( sx-5,sy-5, 10, 10)  #10by10 square centered on point
        px,py = mDataDict["pointXY"]
        cNum = mDataDict["curveNum"]
        pntIn = mDataDict["pIndex"]
        legend = mDataDict["legend"]
        #make a string to display
        s = "Crv# %i, '%s', Pt. (%.2f,%.2f), PtInd %i" %(cNum, legend, px, py, pntIn)
        dc.DrawText(s, sx , sy+1)
        # -----------

    def OnMouseLeftDown(self,event):
        s= "Left Mouse Down at Point: (%.4f, %.4f)" % self.client._getXY(event)
        self.SetStatusText(s)
        event.Skip()            #allows plotCanvas OnMouseLeftDown to be called

    def OnMotion(self, event):
        #show closest point (when enbled)
        if self.client.GetEnablePointLabel() == True:
            #make up dict with info for the pointLabel
            #I've decided to mark the closest point on the closest curve
            dlst= self.client.GetClosestPoint( self.client._getXY(event), pointScaled= True)
            if dlst != []:    #returns [] if none
                curveNum, legend, pIndex, pointXY, scaledXY, distance = dlst
                #make up dictionary to pass to my user function (see DrawPointLabel) 
                mDataDict= {"curveNum":curveNum, "legend":legend, "pIndex":pIndex,\
                            "pointXY":pointXY, "scaledXY":scaledXY}
                #pass dict to update the pointLabel
                self.client.UpdatePointLabel(mDataDict)
        event.Skip()           #go to next handler
        
    def SetPointLabelFunc(self, func):
        """Sets the function with custom code for pointLabel drawing
            ******** more info needed ***************
        """
        self._pointLabelFunc= func
    ##end of new code        
        
def main():
    """Testing"""
   
    app = wx.PySimpleApp()
    f = MyFrame()
    f.Center()
    f.Show()
    app.MainLoop()


if __name__ == "__main__":
    main()