public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-05 16:49:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add Window.is_key_pressed method

Browse source
This commit is contained in:
Grant Sanderson 2021-01-06 12:47:13 -08:00
parent b13b94938a
commit c65a7fb340
1 changed files with 11 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
manimlib/window.py
View file

@ -21,6 +21,8 @@ class Window(PygletWindow):
if "position" in kwargs:
self.position = kwargs["position"]
self.pressed_keys = set()
mglw.activate_context(window=self)
self.timer = Timer()
self.config = mglw.WindowConfig(ctx=self.ctx, wnd=self, timer=self.timer)
@ -58,14 +60,16 @@ class Window(PygletWindow):
offset = self.pixel_coords_to_space_coords(x_offset, y_offset, relative=True)
self.scene.on_mouse_scroll(point, offset)
def on_key_release(self, symbol, modifiers):
super().on_key_release(symbol, modifiers)
self.scene.on_key_release(symbol, modifiers)
def on_key_press(self, symbol, modifiers):
self.pressed_keys.add(symbol) # Modifiers?
super().on_key_press(symbol, modifiers)
self.scene.on_key_press(symbol, modifiers)
def on_key_release(self, symbol, modifiers):
self.pressed_keys.difference_update({symbol}) # Modifiers?
super().on_key_release(symbol, modifiers)
self.scene.on_key_release(symbol, modifiers)
def on_resize(self, width: int, height: int):
super().on_resize(width, height)
self.scene.on_resize(width, height)
@ -81,3 +85,6 @@ class Window(PygletWindow):
def on_close(self):
super().on_close()
self.scene.on_close()
def is_key_pressed(self, symbol):
return (symbol in self.pressed_keys)