def __init__(self, position, movesets):
UIObjectBase.__init__(self)
self.set_layout_data(position, MovesetSelector.HEIGHT, MovesetSelector.WIDTH)
self.selected_thumbnail_index = 0
self.fixed_dimensions = True
MovesetLoader.__init__(self, movesets)
self.selected_moveset = self.thumbnails[self.selected_thumbnail_index].moveset
self.left_scroll_button = ScrollButton(SCROLL_LEFT)
self.right_scroll_button = ScrollButton(SCROLL_RIGHT)
self.set_scroll_button_layout_data()
self.scroll_increment = 10
class MovesetSelector(UIObjectBase, MovesetLoader):
WIDTH = 300
HEIGHT = 120
def __init__(self, position, movesets):
UIObjectBase.__init__(self)
self.set_layout_data(position, MovesetSelector.HEIGHT, MovesetSelector.WIDTH)
self.selected_thumbnail_index = 0
self.fixed_dimensions = True
MovesetLoader.__init__(self, movesets)
self.selected_moveset = self.thumbnails[self.selected_thumbnail_index].moveset
self.left_scroll_button = ScrollButton(SCROLL_LEFT)
self.right_scroll_button = ScrollButton(SCROLL_RIGHT)
self.set_scroll_button_layout_data()
self.scroll_increment = 10
def load_movesets(self, movesets):
self.thumbnails = []
if len(movesets) > 0:
moveset_thumbnails = []
for moveset in movesets:
moveset_thumbnails.append(MovesetThumbnail(moveset))
self.thumbnails.extend(moveset_thumbnails)
self.layout_thumbnails()
def set_scroll_button_layout_data(self):
self.left_scroll_button.set_layout_data((0,0))
left_button_position = (
self.position[0],
self.center()[1] - (self.left_scroll_button.height / 2)
)
self.left_scroll_button.set_layout_data(left_button_position)
self.right_scroll_button.set_layout_data((0,0))
right_button_position = (
self.position[0] + self.width - self.right_scroll_button.width,
self.center()[1] - (self.right_scroll_button.height / 2)
)
self.right_scroll_button.set_layout_data(right_button_position)
def draw(self, surface):
container_surface = pygame.Surface((self.width, self.height))
for thumbnail in self.thumbnails:
thumbnail.draw_relative(container_surface, self.position)
self.draw_scroll_button(self.left_scroll_button, container_surface)
self.draw_scroll_button(self.right_scroll_button, container_surface)
surface.blit(container_surface, self.position)
def draw_scroll_button(self, scroll_button, container_surface):
scroll_button_surface = pygame.Surface((scroll_button.width, self.height))
scroll_button_surface_position = (
scroll_button.position[0],
self.position[1]
)
scroll_button.draw_relative(
scroll_button_surface,
scroll_button_surface_position
)
relative_position = scroll_button.get_relative_position(self.position)
container_surface.blit(
scroll_button_surface,
(relative_position[0], 0)
)
def get_selected_thumbnail_position(self):
selected_thumbnail = self.thumbnails[self.selected_thumbnail_index]
return (
self.center()[0] - (selected_thumbnail.width / 2),
self.position[1]
)
def layout_thumbnails(self):
thumbnails = self.thumbnails
selected_thumbnail = thumbnails[self.selected_thumbnail_index]
selected_thumbnail_position = self.get_selected_thumbnail_position()
selected_thumbnail.set_position(selected_thumbnail_position)
previous_position = selected_thumbnail_position
for i in range(self.selected_thumbnail_index - 1, -1, -1):
current_thumbnail = thumbnails[i]
x_position = previous_position[0] - current_thumbnail.width - 10
y_position = previous
current_thumbnail.set_position((x_position, y_position))
previous_position = (x_position, y_position)
previous_thumbnail = selected_thumbnail
for i in range(self.selected_thumbnail_index + 1, len(thumbnails)):
current_thumbnail = thumbnails[i]
x_position = previous_thumbnail.position[0] + previous_thumbnail.width + 10
y_position = previous_thumbnail.position[1]
current_thumbnail.set_position((x_position, y_position))
previous_thumbnail = current_thumbnail
[thumbnail.inactivate()
for thumbnail in thumbnails
if thumbnail != selected_thumbnail]
def handle_events(self):
self.thumbnails[self.selected_thumbnail_index].handle_events()
if self.is_selected_thumbnail_in_place() == False:
self.shift_thumbnails()
if pygame.MOUSEBUTTONDOWN in wotsuievents.event_types:
if (self.left_scroll_button.contains(wotsuievents.mouse_pos) and
self.left_scroll_button.selected == False):
self.left_scroll_button.handle_selected()
if self.selected_thumbnail_index > 0:
self.select_next_moveset(self.selected_thumbnail_index - 1)
elif (self.right_scroll_button.contains(wotsuievents.mouse_pos) and
self.right_scroll_button.selected == False):
self.right_scroll_button.handle_selected()
if self.selected_thumbnail_index < len(self.thumbnails) - 1:
self.select_next_moveset(self.selected_thumbnail_index + 1)
elif self.contains(wotsuievents.mouse_pos):
for i in range(len(self.thumbnails)):
if self.thumbnails[i].contains(wotsuievents.mouse_pos):
self.select_next_moveset(i)
elif pygame.MOUSEBUTTONUP in wotsuievents.event_types:
if self.left_scroll_button.selected:
self.left_scroll_button.handle_deselected()
if self.right_scroll_button.selected:
self.right_scroll_button.handle_deselected()
def is_selected_thumbnail_in_place(self):
target_x_position = self.get_selected_thumbnail_position()[0]
index = self.selected_thumbnail_index
return target_x_position == self.thumbnails[index].position[0]
def set_moveset_by_name(self, moveset_name):
for i in range(len(self.thumbnails)):
if self.thumbnails[i].moveset.name == moveset_name:
self.select_next_moveset(i)
break
def get_thumbnail_x_displacement(self):
target_x_position = self.get_selected_thumbnail_position()[0]
index = self.selected_thumbnail_index
selected_thumbnail_x_position = self.thumbnails[index].position[0]
distance = target_x_position - selected_thumbnail_x_position
if distance > 0:
return min(self.scroll_increment, distance)
else:
return max(-self.scroll_increment, distance)
def shift_thumbnails(self):
x_displacement = self.get_thumbnail_x_displacement()
[thumbnail.shift(x_displacement, 0) for thumbnail in self.thumbnails]
def select_next_moveset(self, next_moveset_index):
self.thumbnails[self.selected_thumbnail_index].inactivate()
self.thumbnails[next_moveset_index].activate()
self.selected_moveset = self.thumbnails[next_moveset_index].moveset
self.selected_thumbnail_index = next_moveset_index