как использовать художников с патч-прямоугольником для перемещения 2 retcngale одновременно

Question

Я использую прямоугольный патч в двух панелях зума1 и панели зума 2 Идея состоит в том, чтобы наложить прямоугольник на панель, чтобы два прямоугольника могли двигаться одновременно, когда я перемещаю прямоугольник в zooom 1 - это перемещение в то же время и в той же области на панели масштабирования 2

как я могу использовать художник прямоугольника с matplotlib в этом примере

import wx
from numpy import arange, sin, pi,cos
import numpy as np
import matplotlib
import matplotlib.pyplot as plt
from matplotlib.backends.backend_wxagg import FigureCanvasWxAgg as FigureCanvas
from matplotlib.figure import Figure
import matplotlib.patches as patches

class MainFrame(wx.Frame):
    def __init__(self, parent ):
        wx.Panel.__init__(self, parent,name="Main", size = (800,800))
        self.Panel = Panel(self)


class Panel(wx.Panel):
    def __init__(self,parent):
        super().__init__(parent)
        panel = wx.Panel(self)
        self.canvas_panel = CanvasPanel(self)
        self.zoom_panel=Zoom(parent=self)
        self.zoom_panel2=Zoom2(parent=self)
        canvas_sizer = wx.BoxSizer(wx.HORIZONTAL)
        canvas_sizer.Add(self.canvas_panel,1,wx.EXPAND)
        canvas_sizer.Add(self.zoom_panel,1,wx.EXPAND)
        canvas_sizer.Add(self.zoom_panel2,1,wx.EXPAND)

        sizer = wx.BoxSizer(wx.VERTICAL)
        sizer.Add(panel)
        sizer.Add(canvas_sizer)
        self.SetSizerAndFit(sizer)
        self.Show()

class CanvasPanel(wx.Panel):
    """ Panel du graphique matplotlib """
    def __init__(self, parent , size=(200,350)):
        super().__init__(parent)
        self.figure = Figure(figsize =(4,4))
        self.canvas = FigureCanvas(self, -1, self.figure)
        self.axes = self.figure.add_subplot(111)
        self.Size = self.canvas.Size
        self.parent = parent
        t = arange(0.5, 3.0, 0.01)
        s = cos(2 * pi * t)
        self.axes.plot(t, s)
        self.axes.get_xaxis().set_visible(False)
        self.axes.get_yaxis().set_visible(False)
        self.canvas.mpl_connect('button_press_event', self.on_press)
        x = y = 0.02
        self.rect = patches.Rectangle((x, y), 0.4,0.4,edgecolor='g', alpha=1, fill=None, label='Label')
        self.axes.add_patch(self.rect)
        self.axes.plot()

    def on_press(self, click):
        x1, y1 = click.xdata, click.ydata
        zx1 = x1 - 0.2
        zy1 = y1 - 0.2
        zx2 = x1 + 0.2
        zy2 = y1 + 0.2
        self.rect.set_x(x1 - 0.2) #Move the rectangle and centre it on the X click point
        self.rect.set_y(y1 - 0.2) #Move the rectangle and centre it on the Y click point


        self.axes.plot()
        self.canvas.draw()
        self.zoom_axes=[zx1,zx2,zy1,zy2]
        self.parent.zoom_panel.Update(self)
        self.parent.zoom_panel2.Update(self)



class Zoom(wx.Panel):
    def __init__(self,parent):
        wx.Panel.__init__(self,parent,size=(200,200))
        self.Show()
    def Update(self,parent):
        #Load axis values of the selected rectangle
        zoom_axes=parent.zoom_axes

        #duplicate the plot from the main panel
        self.figure = Figure(figsize =(4,4))
        self.canvas = FigureCanvas(self, -1, self.figure)
        self.axes = self.figure.add_subplot(111)
        #Apply axis of drawn rectangle to the plot
        self.axes.axis(zoom_axes)
        t = arange(0.5, 3.0, 0.01)
        s = cos(2 * pi * t)
        self.axes.plot(t, s)
        self.axes.get_xaxis().set_visible(False)
        self.axes.get_yaxis().set_visible(False)
        self.canvas.mpl_connect('button_press_event', self.on_press)
        x = y = 0.01
        self.rect = patches.Rectangle((x, y), 0.02, 0.02,edgecolor='g', alpha=1, fill=None, label='Label')
        self.axes.add_patch(self.rect)
        self.axes.plot()



    def on_press(self, click):
        x1, y1 = click.xdata, click.ydata
        zx1 = x1 - 0.01
        zy1 = y1 - 0.01
        zx2 = x1 + 0.01
        zy2 = y1 + 0.01
        self.rect.set_x(x1 - 0.01) #Move the rectangle and centre it on the X click point
        self.rect.set_y(y1 - 0.01) #Move the rectangle and centre it on the Y click point


        self.axes.plot()
        self.canvas.draw()
        self.zoom_axes=[zx1,zx2,zy1,zy2]

class Zoom2(wx.Panel):
    def __init__(self,parent):
        wx.Panel.__init__(self,parent,size=(200,200))
        self.Show()
    def Update(self,parent):
        #Load axis values of the selected rectangle
        zoom_axes=parent.zoom_axes

        #duplicate the plot from the main panel
        self.figure = Figure(figsize =(4,4))
        self.canvas = FigureCanvas(self, -1, self.figure)
        self.axes = self.figure.add_subplot(111)
        #Apply axis of drawn rectangle to the plot
        self.axes.axis(zoom_axes)
        t = arange(0.5, 3.0, 0.01)
        s = cos(2 * pi * t)
        self.axes.plot(t, s)
        self.axes.get_xaxis().set_visible(False)
        self.axes.get_yaxis().set_visible(False)
        self.canvas.mpl_connect('button_press_event', self.on_press)
        x = y = 0.01
        self.rect = patches.Rectangle((x, y), 0.02, 0.02,edgecolor='g', alpha=1, fill=None, label='Label')
        self.axes.add_patch(self.rect)
        self.axes.add_artist(self.rect)

        self.axes.plot()


    def on_press(self, click):
        x1, y1 = click.xdata, click.ydata
        zx1 = x1 - 0.01
        zy1 = y1 - 0.01
        zx2 = x1 + 0.01
        zy2 = y1 + 0.01
        self.rect.set_x(x1 - 0.01) #Move the rectangle and centre it on the X click point
        self.rect.set_y(y1 - 0.01) #Move the rectangle and centre it on the Y click point


        self.axes.plot()
        self.canvas.draw()
        self.zoom_axes=[zx1,zx2,zy1,zy2]



app = wx.App()
frame = MainFrame(None).Show()
app.MainLoop()

спасибо всем

Answer 1

Изменения сделаны:

• Добавлена ​​связь между 2 панелями Zoom.У каждого может быть подписчик, добавленный на add_subscriber(subscriber_name, action_to_make), и каждый раз, когда один из них нажимается, он будет вызывать все action_to_make(click).
• Поскольку Zoom и Zoom2 - это в основном 2 экземпляра одного класса (с идентичным кодом), я 'мы удалили определение 2-го класса и сделали zoom_panel и zoom_panel2 различных экземпляров одного и того же класса.
• Изменено масштабирование, чтобы только создавать экземпляры Figure и Rectangle и всего этого только один раз и обновлять их вместо создания новогоэкземпляр и поместив его поверх старого.

import wx
from numpy import arange, sin, pi,cos
import numpy as np
import matplotlib
import matplotlib.pyplot as plt
from matplotlib.backends.backend_wxagg import FigureCanvasWxAgg as FigureCanvas
from matplotlib.figure import Figure
import matplotlib.patches as patches

class MainFrame(wx.Frame):
    def __init__(self, parent ):
        wx.Panel.__init__(self, parent,name="Main", size = (1000,800))
        self.Panel = Panel(self)


class Panel(wx.Panel):
    def __init__(self,parent):
        super().__init__(parent)
        panel = wx.Panel(self)
        self.canvas_panel = CanvasPanel(self)
        self.zoom_panel=Zoom(parent=self)
        self.zoom_panel2=Zoom(parent=self)

        self.zoom_panel.add_subscriber("zoom_panel2", self.zoom_panel2.on_notified)
        self.zoom_panel2.add_subscriber("zoom_panel", self.zoom_panel.on_notified)

        canvas_sizer = wx.BoxSizer(wx.HORIZONTAL)
        canvas_sizer.Add(self.canvas_panel,1,wx.EXPAND)
        canvas_sizer.Add(self.zoom_panel,1,wx.EXPAND)
        canvas_sizer.Add(self.zoom_panel2,1,wx.EXPAND)

        sizer = wx.BoxSizer(wx.VERTICAL)
        sizer.Add(panel)
        sizer.Add(canvas_sizer)
        self.SetSizerAndFit(sizer)
        self.Show()

class CanvasPanel(wx.Panel):
    """ Panel du graphique matplotlib """
    def __init__(self, parent , size=(200,350)):
        super().__init__(parent)
        self.figure = Figure(figsize =(4,4))
        self.canvas = FigureCanvas(self, -1, self.figure)
        self.axes = self.figure.add_subplot(111)
        self.Size = self.canvas.Size
        self.parent = parent
        t = arange(0.5, 3.0, 0.01)
        s = cos(2 * pi * t)
        self.axes.plot(t, s)
        self.axes.get_xaxis().set_visible(False)
        self.axes.get_yaxis().set_visible(False)
        self.canvas.mpl_connect('button_press_event', self.on_press)
        x = y = -0.2
        self.rect = patches.Rectangle((x, y), 0.4,0.4,edgecolor='g', alpha=1, fill=None, label='Label')
        self.axes.add_patch(self.rect)
        self.axes.plot()

    def on_press(self, click):
        x1, y1 = click.xdata, click.ydata
        zx1 = x1 - 0.2
        zy1 = y1 - 0.2
        zx2 = x1 + 0.2
        zy2 = y1 + 0.2
        self.rect.set_x(x1 - 0.2) #Move the rectangle and centre it on the X click point
        self.rect.set_y(y1 - 0.2) #Move the rectangle and centre it on the Y click point


        self.axes.plot()
        self.canvas.draw()
        self.zoom_axes=[zx1,zx2,zy1,zy2]
        self.parent.zoom_panel.Update(self)
        self.parent.zoom_panel2.Update(self)



class Zoom(wx.Panel):
    def __init__(self,parent):
        wx.Panel.__init__(self,parent,size=(200,200))
        self.Show()
        self.subscriber_list = {}
        #init figure&canvas
        self.figure = Figure(figsize =(4,4))
        self.canvas = FigureCanvas(self, -1, self.figure)
        self.axes = self.figure.add_subplot(111)
        self.axes.get_xaxis().set_visible(False)
        self.axes.get_yaxis().set_visible(False)
        self.axes.axis([-0.2,0.2,-0.2,0.2]) # set default display window
        self.canvas.mpl_connect('button_press_event', self.on_press)
        #add rectangle
        x = y = 0.01
        self.rect = patches.Rectangle((x, y), 0.02, 0.02,edgecolor='g', alpha=1, fill=None, label='Label')
        self.axes.add_patch(self.rect)

        #first display
        self.axes.plot()
        self.canvas.draw()

    def Update(self,parent):
        #Load axis values of the selected rectangle
        zoom_axes=parent.zoom_axes

        #clear Axes
        self.axes.clear()

        #Apply axis of drawn rectangle to the plot
        self.axes.axis(zoom_axes)



        #draw same plot as on the main panel
        t = arange(0.5, 3.0, 0.01)
        s = cos(2 * pi * t)
        self.axes.plot(t, s, 'C0')

        #draw rectangle
        self.axes.add_patch(self.rect)

        #display
        self.axes.plot()
        self.canvas.draw()

    def on_press(self, click):
        self.perform_press(click)
        self.notify_subscribers(click)


    def perform_press(self, click):
        x1, y1 = click.xdata, click.ydata
        zx1 = x1 - 0.01
        zy1 = y1 - 0.01
        zx2 = x1 + 0.01
        zy2 = y1 + 0.01
        self.rect.set_x(x1 - 0.01) #Move the rectangle and centre it on the X click point
        self.rect.set_y(y1 - 0.01) #Move the rectangle and centre it on the Y click point

        self.axes.plot()
        self.canvas.draw()
        self.zoom_axes=[zx1,zx2,zy1,zy2]

    def notify_subscribers(self, click):
        for subscriber in self.subscriber_list:
            self.subscriber_list.get(subscriber)(click)

    # adds a new subscriber to the list. "name" is a subscriber identifier, "action" is called as "action(name)" every time this element is clicked
    def add_subscriber(self, name, action):
        self.subscriber_list[name]=action

    def on_notified(self, click):
        #Do something here
        #print("Zoom received notification from Zoom2: ", click)
        self.perform_press(click)


app = wx.App()
frame = MainFrame(None).Show()
app.MainLoop()