I want to add an arrow (some object) upon the press of a button. The code is a modified version of the example given in the vispy repository.
import wx
from vispy import scene
from vispy.scene.visuals import Arrow
import numpy as np
class Canvas(scene.SceneCanvas):
def __init__(self, *a, **k):
sizes = k.pop("sizes", (300, 300)) # Default value is (300, 300)
scene.SceneCanvas.__init__(self, *a, **k, size=sizes)
view = self.central_widget.add_view()
view.bgcolor = 'snow'
view.camera = scene.TurntableCamera(up='+y', azimuth=100, elevation=15, fov=60)
axis = scene.visuals.XYZAxis(parent=view.scene)
arrow1 = np.array([(0, 0, 0, 1, 1, 0)]) # Arrow direction, position
arr = Arrow(pos=np.array([(0, 0, 0), (1, 1, 0)]), color='teal', method='gl', width=5., arrows=arrow1,
arrow_type="angle_30", arrow_size=5.0, arrow_color='teal', antialias=True, parent=view.scene)
self.show()
class mainFrame ( wx.Frame ):
def __init__( self, parent ):
wx.Frame.__init__ ( self, parent, id = wx.ID_ANY, title = "Add Object", pos = wx.DefaultPosition, size = wx.Size( 905,569 ), style = wx.DEFAULT_FRAME_STYLE|wx.TAB_TRAVERSAL )
self.SetSizeHints( wx.DefaultSize, wx.DefaultSize )
bSizer1 = wx.BoxSizer( wx.VERTICAL )
self.panel_vispy = wx.Panel( self, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, wx.TAB_TRAVERSAL )
bSizer1.Add( self.panel_vispy, 9, wx.EXPAND |wx.ALL, 5 )
self.panel_Buttons = wx.Panel( self, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, wx.TAB_TRAVERSAL )
bSizer2 = wx.BoxSizer( wx.HORIZONTAL )
bSizer2.Add( ( 0, 0), 1, wx.EXPAND, 5 )
self.button_AddArrow = wx.Button( self.panel_Buttons, wx.ID_ANY, u"Add Arrow", wx.DefaultPosition, wx.DefaultSize, 0 )
bSizer2.Add( self.button_AddArrow, 0, wx.ALL|wx.ALIGN_CENTER_VERTICAL, 5 )
bSizer2.Add( ( 0, 0), 1, wx.EXPAND, 5 )
self.panel_Buttons.SetSizer( bSizer2 )
self.panel_Buttons.Layout()
bSizer2.Fit( self.panel_Buttons )
bSizer1.Add( self.panel_Buttons, 1, wx.EXPAND |wx.ALL, 5 )
self.SetSizer( bSizer1 )
self.Layout()
self.Centre( wx.BOTH )
# Connect Events
self.button_AddArrow.Bind(wx.EVT_BUTTON, self.button_AddArrow_OnClick)
self.panel_vispy.canvas = Canvas(app="wx", parent=self, sizes=self.panel_vispy.GetSize())
def button_AddArrow_OnClick(self, event):
print("Adding arrow")
arrow2 = np.array([(0, 0, 0, 1, 1, 0)]) # Arrow direction, position
arr = Arrow(pos=np.array([(0, 0, 0), (1, 1, 0)]), color='red', method='gl', width=5., arrows=arrow2,
arrow_type="angle_30", arrow_size=5.0, arrow_color='red', antialias=True)#, parent=view.scene) -Unable to access view from the Canvas class
if __name__ == "__main__":
app = wx.App(False)
GUI = mainFrame(None)
GUI.Show(True)
app.MainLoop()
I have created the Canvas as given in the example and tried to instantiate it from the mainFrame. All the objects that are given in the __init__ of the Canvas class are created. If I want to add any new object after the GUI is created with the press of a button, I am unable to access the view in the Canvas class from the mainFrame. Or is there any other way to achieve this.
Update: Here is the code after following the suggestions given in the comments:
import wx
from vispy import scene, gloo
from vispy.scene.visuals import Arrow
import numpy as np
class Canvas(scene.SceneCanvas):
def __init__(self, *a, **k):
sizes = k.pop("sizes", (300, 300)) # Default value is (300, 300)
scene.SceneCanvas.__init__(self, *a, **k, size=sizes)
self.unfreeze()
self.view = self.central_widget.add_view()
self.view.bgcolor = 'snow'
self.view.camera = scene.TurntableCamera(up='+y', azimuth=100, elevation=15, fov=60)
axis = scene.visuals.XYZAxis(parent=self.view.scene)
arrow1 = np.array([(0, 0, 0, 1, 1, 0)]) # Arrow direction, position
arr = Arrow(pos=np.array([(0, 0, 0), (1, 1, 0)]), color='teal', method='gl', width=5., arrows=arrow1,
arrow_type="angle_30", arrow_size=5.0, arrow_color='teal', antialias=True, parent=self.view.scene)
self.show()
class mainFrame ( wx.Frame ):
def __init__( self, parent ):
wx.Frame.__init__ ( self, parent, id = wx.ID_ANY, title = "Add Object", pos = wx.DefaultPosition, size = wx.Size( 905,569 ), style = wx.DEFAULT_FRAME_STYLE|wx.TAB_TRAVERSAL )
self.SetSizeHints( wx.DefaultSize, wx.DefaultSize )
bSizer1 = wx.BoxSizer( wx.VERTICAL )
self.panel_vispy = wx.Panel( self, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, wx.TAB_TRAVERSAL )
bSizer1.Add( self.panel_vispy, 9, wx.EXPAND |wx.ALL, 5 )
self.panel_Buttons = wx.Panel( self, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, wx.TAB_TRAVERSAL )
bSizer2 = wx.BoxSizer( wx.HORIZONTAL )
bSizer2.Add( ( 0, 0), 1, wx.EXPAND, 5 )
self.button_AddArrow = wx.Button( self.panel_Buttons, wx.ID_ANY, u"Add Arrow", wx.DefaultPosition, wx.DefaultSize, 0 )
bSizer2.Add( self.button_AddArrow, 0, wx.ALL|wx.ALIGN_CENTER_VERTICAL, 5 )
bSizer2.Add( ( 0, 0), 1, wx.EXPAND, 5 )
self.panel_Buttons.SetSizer( bSizer2 )
self.panel_Buttons.Layout()
bSizer2.Fit( self.panel_Buttons )
bSizer1.Add( self.panel_Buttons, 1, wx.EXPAND |wx.ALL, 5 )
self.SetSizer( bSizer1 )
self.Layout()
self.Centre( wx.BOTH )
# Connect Events
self.button_AddArrow.Bind(wx.EVT_BUTTON, self.button_AddArrow_OnClick)
self.panel_vispy.canvas = Canvas(app="wx", parent=self, sizes=self.panel_vispy.GetSize())
def button_AddArrow_OnClick(self, event):
print("Adding arrow")
arrow2 = np.array([(0, 0, 0, 1, 1, 0)]) # Arrow direction, position
arr = Arrow(pos=np.array([(0, 0, 0), (1, 1, 0)]), color='teal', method='gl', width=5., arrows=arrow2,
arrow_type="angle_30", arrow_size=5.0, arrow_color='teal', antialias=True, parent=self.panel_vispy.canvas.view.scene)
self.panel_vispy.canvas.update() # Canvas not getting updated
if __name__ == "__main__":
app = wx.App(False)
GUI = mainFrame(None)
GUI.Show(True)
app.MainLoop()
The code now runs without any error which means the arrow is probably added but the scene is not updated.