def mouse_left_up(self, event): if not self._tilemap: return if not self._select_end: return x1, y1 = self._select_start x2, y2 = self._select_end if x2 < x1: x2, x1 = x1, x2 if y2 < y1: y2, y1 = y1, y2 width = x2 - x1 + 1 height = y2 - y1 + 1 self._select_start = None self._select_end = None self.Viewport.Refresh(False) if width and height: selection = Tilemap.from_tilemap(self._tilemap, x1, y1, x2 + 1, y2 + 1) else: selection = None event = TileSelector.SelectEvent(selection=selection) wx.PostEvent(self.GetEventHandler(), event)
def populate_tiles(self): if not self._tileset: self._tilemap.clear() surface = self._presenter.surface tiles_width = int(math.floor(surface.width / float(Tilemap.TILE_SIZE))) tiles_height = int(math.ceil(len(self._tileset.tiles) / float(tiles_width))) tiles = range(0, len(self._tileset.tiles)) self._tilemap = Tilemap(tiles, tiles_width, tiles_height) self._camera.set_max(tiles_width * Tilemap.TILE_SIZE, tiles_height * Tilemap.TILE_SIZE) self.Scrollbar.SetScrollbar(wx.VERTICAL, 0, surface.height, tiles_height, True) self.Scrollbar.SetThumbPosition(0)
def mouse_left_up(self): if self._state == State.SELECT: x, y, width, height = self.get_selection_rectangle(self._select_start, self._select_end) if width and height: if self._select_type == SelectType.SELECT: self._selection = Tilemap.from_tilemap(self._level.tilemap, x, y, x + width, y + height) elif self._select_type == SelectType.FILL: self._frame.undo_add() self._level.tilemap.fill_with(self._selection, x, y, x + width, y + height) self._frame.set_level_modified(True) self._state = State.NONE self._frame.set_viewport_cursor(wx.CURSOR_DEFAULT) elif self._state == State.DRAW: self._state = State.NONE
class TileSelector(wx.Panel): SelectEvent, EVT_SELECT_EVENT = wx.lib.newevent.NewEvent() def __init__(self, parent, id=wx.ID_ANY, pos=wx.DefaultPosition, size=wx.DefaultSize, style=wx.NO_BORDER): wx.Panel.__init__(self, parent, id=id, pos=pos, size=size, style=style) self.Viewport = wx.Panel(self, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, wx.NO_BORDER) self.Scrollbar = wx.ScrollBar(self, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, wx.SB_VERTICAL) sizer = wx.BoxSizer(wx.HORIZONTAL) sizer.Add(self.Viewport, 1, wx.EXPAND, 0) sizer.Add(self.Scrollbar, 0, wx.EXPAND, 0) self.SetSizer(sizer) self.Layout() sizer.Fit(self) self.Scrollbar.SetScrollbar(0, 0, 0, 0) self._presenter = Presenter.from_window(self.Viewport.GetHandle(), config.SCALE) self._camera = Camera(0, 0, 0, 0) self._tilemap = None self._tileset = None self._select_start = None self._select_end = None self._show_collision = False self.Viewport.Bind(wx.EVT_PAINT, self.paint) self.Viewport.Bind(wx.EVT_SIZE, self.resize) self.Viewport.Bind(wx.EVT_MOUSEWHEEL, self.mouse_wheel) self.Viewport.Bind(wx.EVT_LEFT_DOWN, self.mouse_left_down) self.Viewport.Bind(wx.EVT_LEFT_UP, self.mouse_left_up) self.Viewport.Bind(wx.EVT_MOTION, self.mouse_move) self.Scrollbar.Bind(wx.EVT_SCROLL, self.scroll) def set_tileset(self, tileset): self._tileset = tileset self.populate_tiles() self.Viewport.Refresh(False) def populate_tiles(self): if not self._tileset: self._tilemap.clear() surface = self._presenter.surface tiles_width = int(math.floor(surface.width / float(Tilemap.TILE_SIZE))) tiles_height = int(math.ceil(len(self._tileset.tiles) / float(tiles_width))) tiles = range(0, len(self._tileset.tiles)) self._tilemap = Tilemap(tiles, tiles_width, tiles_height) self._camera.set_max(tiles_width * Tilemap.TILE_SIZE, tiles_height * Tilemap.TILE_SIZE) self.Scrollbar.SetScrollbar(wx.VERTICAL, 0, surface.height, tiles_height, True) self.Scrollbar.SetThumbPosition(0) def resize(self, event): self._presenter.resize() surface = self._presenter.surface self._camera.set_size(surface.width, surface.height) if self._tileset: tiles_width = int(math.floor(surface.width / float(Tilemap.TILE_SIZE))) tiles_height = int(math.ceil((len(self._tileset.tiles) / tiles_width) * Tilemap.TILE_SIZE)) diff = int(tiles_height - surface.height) if diff: position = self.Scrollbar.GetThumbPosition() self.Scrollbar.SetScrollbar(position, surface.height, tiles_height, Tilemap.TILE_SIZE) else: self.Scrollbar.SetScrollbar(0, 0, 0, 0) def paint(self, event): if not self._tileset: return surface = self._presenter.surface surface.clear() self._camera.move_absolute(0, self.Scrollbar.GetThumbPosition()) self._tilemap.render(surface, self._camera, self._tileset, self._show_collision) if self._select_start: x1 = self._select_start[0] * Tilemap.TILE_SIZE y1 = self._select_start[1] * Tilemap.TILE_SIZE x2 = self._select_end[0] * Tilemap.TILE_SIZE y2 = self._select_end[1] * Tilemap.TILE_SIZE x1, y1 = self._camera.world_to_camera(x1, y1) x2, y2 = self._camera.world_to_camera(x2, y2) if x2 < x1: x2, x1 = x1, x2 if y2 < y1: y2, y1 = y1, y2 width = x2 - x1 + Tilemap.TILE_SIZE height = y2 - y1 + Tilemap.TILE_SIZE surface.box_fill(x1, y1, width, height, 0xFFFFFFFF, BlendOp.ALPHA50) self._presenter.present() def mouse_wheel(self, event): if not self._tileset: return position = self.Scrollbar.GetThumbPosition() position -= (event.GetWheelRotation() / 20) * Tilemap.TILE_SIZE self.Scrollbar.SetThumbPosition(position) self.Viewport.Refresh(False) def mouse_left_down(self, event): if not self._tileset: return if self._select_start: return pos = event.GetPosition() x, y = self.get_tile_position(pos) self._select_start = (x, y) self._select_end = self._select_start self.Viewport.Refresh(False) def mouse_left_up(self, event): if not self._tilemap: return if not self._select_end: return x1, y1 = self._select_start x2, y2 = self._select_end if x2 < x1: x2, x1 = x1, x2 if y2 < y1: y2, y1 = y1, y2 width = x2 - x1 + 1 height = y2 - y1 + 1 self._select_start = None self._select_end = None self.Viewport.Refresh(False) if width and height: selection = Tilemap.from_tilemap(self._tilemap, x1, y1, x2 + 1, y2 + 1) else: selection = None event = TileSelector.SelectEvent(selection=selection) wx.PostEvent(self.GetEventHandler(), event) def mouse_move(self, event): if not self._select_start: return pos = event.GetPosition() x, y = self.get_tile_position(pos) self._select_end = (x, y) self.Viewport.Refresh(False) def scroll(self, event): self.Viewport.Refresh(False) def get_tile_position(self, pos): x = pos[0] / self._presenter.scale y = pos[1] / self._presenter.scale x, y = self._camera.camera_to_world(x, y) x /= Tilemap.TILE_SIZE y /= Tilemap.TILE_SIZE return x, y def show_collision(self, show): self._show_collision = show self.Viewport.Refresh(False)