def start(self):
# start with menu open
self.gfx = command_main_menu(self.gfx)
# main loop
while 1:
self.event = pygame.event.poll()
pygame.time.wait(1) # so pygame doesn't use 100% CPU
cursor = pygame.mouse.get_pos()
self.last_rect = Rect(self.gfx['rect'])
# drag image code:
if gl.HAND_TOOL:
if left_click(self.event): # calculate drag coordinates:
if gl.IMG_BORDER:
self.border_fix() # erase the current border
grab_hand_cursor()
self.minus1 = cursor[0] - self.gfx['rect'][0]
self.minus2 = cursor[1] - self.gfx['rect'][1]
gl.DO_DRAG = 1
if self.event.type == MOUSEMOTION and gl.DO_DRAG: # move the image when dragged:
grab_hand_cursor()
self.gfx['rect'][0] = cursor[0] - self.minus1
self.gfx['rect'][1] = cursor[1] - self.minus2
self.gfx['screen'].fill(gl.IMGV_COLOR, self.last_rect)
self.gfx['screen'].blit(self.gfx['new_img'], self.gfx['rect'])
update(self.gfx['rect'].union(self.last_rect))
if self.event.type == MOUSEBUTTONUP: # released mouse button, redisplay status bars:
drag_hand_cursor()
my_update_screen(self.gfx['new_img'], self.gfx['rect'], self.gfx['file'])
gl.DO_DRAG = 0
if self.event.type == VIDEORESIZE:
self.gfx['screen'] = pygame.display.set_mode(self.event.dict['size'], RESIZABLE)
self.gfx['rect'] = get_center(self.gfx['screen'], self.gfx['new_img'])
my_update_screen(self.gfx['new_img'], self.gfx['rect'], self.gfx['file'])
if self.event.type == KEYDOWN:
gl.HAND_TOOL = 0
if self.event.key not in (K_DOWN, K_UP, K_RIGHT, K_LEFT):
normal_cursor() # stop displaying hand tool
(self.gfx, self.last_rect) = handle_keyboard(self.event, self.gfx, self.last_rect)
if self.event.type == KEYUP:
self.stop_auto_repeat()
check_quit(self.event)
if self.event.type == MOUSEBUTTONDOWN and right_click(self.event):
self.open_main_menu()
# Re-open the purposely closed window that frees up RAM
if (gl.KEEP_MENU_OPEN == "1" and gl.COUNT_CLICKS == 1) or gl.JUST_RESIZED:
gl.COUNT_CLICKS = 0
gl.JUST_RESIZED = 0
self.gfx = command_main_menu(self.gfx)
self.start_auto_repeat()
def four(screen, file, new_img, ns):
paint_screen(screen, gl.IMGV_COLOR) # so transparent status bars don't mess up on VIDEOEXPOSE repaints
old_file = file
(img_one_rect, img_two_rect, img_three_rect, img_four_rect) = (0, 0, 0, 0)
(img_one_name, img_two_name, img_three_name, img_four_name) = (0, 0, 0, 0)
(show_img_one, show_img_two, show_img_three, show_img_four) = (0, 0, 0, 0)
rect = show_message(screen, "", "bottom", 9, ("bold")) # needed to not paint on esc_rect first time
(file, img_one_rect, img_one_name, img_one_file) = square_one(screen, file)
(file, img_two_rect, img_two_name, img_two_file) = square_two(screen, file)
(file, img_three_rect, img_three_name, img_three_file) = square_three(screen, file)
(file, img_four_rect, img_four_name, img_four_file) = square_four(screen, file)
(esc_rect, close_font) = close_button(screen)
start = ns
while 1:
flag = 0
event = pygame.event.poll()
pygame.time.wait(1)
cursor = pygame.mouse.get_pos()
if event.type == VIDEORESIZE:
pygame.event.set_blocked(VIDEOEXPOSE)
screen = pygame.display.set_mode(event.dict['size'], RESIZABLE)
file = file - 4
pygame.event.set_allowed(VIDEOEXPOSE)
(file, new_img, start) = four(screen, file, new_img, ns)
flag = 1
break
if hit_key(event, K_ESCAPE):
gl.ESCAPED = 1
file = old_file
break
check_quit(event)
if hit_key(event, K_SPACE) or hit_key(event, K_n) or hit_key(event, K_4) or right_click(event): # show next 4 images
paint_screen(screen, gl.IMGV_COLOR)
flag = 1
if hit_key(event, K_BACKSPACE) or hit_key(event, K_b) or middle_click(event): # show previous 4 images
paint_screen(screen, gl.IMGV_COLOR)
file = file - 8
flag = 1
if flag == 1:
(file, img_one_rect, img_one_name, img_one_file) = square_one(screen, file)
if flag == 1:
(file, img_two_rect, img_two_name, img_two_file) = square_two(screen, file)
if flag == 1:
(file, img_three_rect, img_three_name, img_three_file) = square_three(screen,\
file)
if flag == 1:
(file, img_four_rect, img_four_name, img_four_file) = square_four(screen, file)
(show_img_one, show_img_two, show_img_three, show_img_four, rect) = hover_square(\
screen, show_img_one, show_img_two, show_img_three, show_img_four, img_one_rect,\
img_two_rect, img_three_rect, img_four_rect, img_one_name, img_two_name, img_three_name,\
img_four_name, img_one_file, img_two_file, img_three_file, img_four_file, rect, event)
hover_fx(screen, img_one_name, img_two_name, img_three_name, img_four_name, img_one_rect, img_two_rect, img_three_rect, img_four_rect, cursor)
if show_img_one == None:
file = old_file
break
if hit_key(event, K_w):
if len(gl.files) <= 1: # nothin to slideshow
file = old_file
break # kick 'em out
(file, img_one_file, img_two_file, img_three_file, img_four_file) =\
my_fourslideshow(screen, new_img, rect, gl.files[file], file - 4,\
len(gl.files), img_one_file, img_two_file, img_three_file, img_four_file, ns)
paint_screen(screen, gl.IMGV_COLOR)
(file, new_img, start) = four(screen, file - 4, new_img, ns)
flag = 1
break
if left_click(event):
start = start_timer()
if img_one_rect.collidepoint(cursor):
wait_cursor()
new_img = load_img(gl.files[img_one_file], screen)
return (img_one_file, new_img, start)
if img_two_rect.collidepoint(cursor):
wait_cursor()
new_img = load_img(gl.files[img_two_file], screen)
return (img_two_file, new_img, start)
if img_three_rect.collidepoint(cursor):
wait_cursor()
new_img = load_img(gl.files[img_three_file], screen)
return (img_three_file, new_img, start)
if img_four_rect.collidepoint(cursor):
wait_cursor()
new_img = load_img(gl.files[img_four_file], screen)
return (img_four_file, new_img, start)
if esc_rect.collidepoint(cursor):
file = old_file
gl.ESCAPED = 1
break
if event.type == VIDEOEXPOSE:
# if event.type == VIDEOEXPOSE and not pygame.mouse.get_focused():#
# repaint the screen in case other windows painted over it:
file = file - 4
(file, new_img, start) = four(screen, file, new_img, ns)
flag = 1
break
return (file, new_img, start)
def command_file_master(screen, file_names, msg, down, button_op, disable_right_click, again):
set_caption("Image Browser - imgv")
screen_pause = place = marker = 0
menu_items = []
edit_rect = back_rect = forward_rect = sort_rect = junk_rect()
(esc_rect, font) = close_button(screen)
create_rect = imgv_button(screen, " Create New List ", 0, 18, "midtop")
if len(file_names) < 1:
my_string = ask(screen, "Create playlist first (Enter a name)")
if my_string == None or my_string == []:
return ([], [], [], []) # don't create a list
if my_string != []: # create list
if (len(my_string) > 0) and my_string != "\n":
return (file_names, None, None, my_string)
pygame.event.set_blocked(MOUSEMOTION)
while 1:
event = pygame.event.poll()
pygame.time.wait(1)
if screen_pause == 1:
while 1:
event = pygame.event.poll()
pygame.time.wait(1)
cursor = pygame.mouse.get_pos()
hover_fx(screen, menu_items, cursor)
hover_cursor(cursor, [esc_rect, edit_rect, sort_rect, back_rect, forward_rect, create_rect] + [x[0] for x in menu_items])
if button_op:
hover_button(create_rect, cursor, screen, " Create New List ", 0, 18, "midtop")
if (place + 1) < len(file_names):
hover_button(forward_rect, cursor, screen, " Next ", 10, 18, "topright")
if (((place + 1) - gl.MAX_SCREEN_FILES) > 1):
hover_button(back_rect, cursor, screen, " Previous ", 10, 18, "topleft")
if not gl.SORT_HIT:
hover_button(sort_rect, cursor, screen, " Sort ", 13, 42, "midtop")
check_quit(event)
if hit_key(event, K_ESCAPE):
return (None, None, None, None)
if left_click(event):
if esc_rect.collidepoint(cursor):
return (None, None, None, None)
if left_click(event):
for item in menu_items:
if item[0].collidepoint(cursor):
if pygame.mouse.get_pressed()[0] and (pygame.key.get_pressed()[K_LCTRL] or\
pygame.key.get_pressed()[K_RCTRL]):
return (file_names, item[1], "deleteit", None)
if again == "do again":
return (file_names, item[1], "do again", None)
return (file_names, item[1], menu_items, None)
if right_click(event):
if not disable_right_click:
for item in menu_items:
if item[0].collidepoint(cursor):
if not os.path.isfile(gl.DATA_DIR + item[1]):
if edit_rect != junk_rect():
paint_screen(gl.BLACK)
edit_rect = show_message(
"%s doesn't exist in %s" % (item[1], gl.DATA_DIR), "top", 9, ("bold", "transparent"))
else:
return (None, item[1], "rclicked", None)
if hit_key(event, K_SPACE) or right_click(event):
if not place >= len(file_names):
screen_pause = 0
marker = 0
menu_items = []
break
if left_click(event):
if forward_rect.collidepoint(cursor):
if not place >= len(file_names):
screen_pause = 0
marker = 0
menu_items = []
break
if hit_key(event, K_BACKSPACE) or middle_click(event):
if ((place - marker) > 0):
paint_screen(gl.BLACK)
screen_pause = 0
place = place - (gl.MAX_SCREEN_FILES + marker)
marker = 0
menu_items = []
break
if left_click(event):
if back_rect.collidepoint(cursor):
if ((place - marker) > 0):
paint_screen(gl.BLACK)
screen_pause = 0
place = place - (gl.MAX_SCREEN_FILES + marker)
marker = 0
menu_items = []
break
if left_click(event):
if sort_rect.collidepoint(cursor):
gl.SORT_HIT = 1
file_names = basename_sort(file_names)
(file_names, menu_items, screen_pause, place, marker, forward_rect, back_rect, sort_rect) = file_master(screen, file_names, place, marker, menu_items, msg, down, button_op)
screen_pause = place = marker = 0
menu_items = []
break
if left_click(event):
if create_rect.collidepoint(cursor):
my_string = ask(screen, "Enter name of list")
if my_string != None:
if (len(my_string) > 0) and my_string != "\n":
return (file_names, None, menu_items, my_string)
(file_names, menu_items, screen_pause, place, marker, forward_rect, back_rect, sort_rect) =\
file_master(screen, file_names, place, marker, menu_items, msg, down, button_op)
pygame.time.delay(5)
def thumbs_engine(screen, new_img, file):
screen_pause = 0
SPACER = 5
x = []
place = file # start thumbing from current image position
marker = 0
(i, j) = (SPACER, SPACER)
(esc_rect, close_font) = close_button(screen)
font_size = 9
font = pygame.font.Font(gl.FONT_NAME, font_size)
pygame.event.set_blocked(MOUSEMOTION)
while 1:
event = pygame.event.poll()
pygame.time.wait(1)
check_quit(event)
cursor = pygame.mouse.get_pos()
hover_cursor(cursor, (esc_rect,))
gl.PAUSED = 0 # critical
if hit_key(event, K_ESCAPE):
gl.ESCAPED = 1
return (new_img, new_img, new_img, file)
if left_click(event):
if esc_rect.collidepoint(cursor):
gl.ESCAPED = 1
break
if hit_key(event, K_t) or hit_key(event, K_p) or hit_key(event, K_PAUSE):
# pause
set_caption("Thumbnails [Paused]")
try:
if place % gl.MAX_THUMBS_SET != 0:
# only pause if it's not the last thumbnail on the page
gl.PAUSED = 1
screen_pause = 1
except:
gl.PAUSED = 1
screen_pause = 1
if screen_pause == 1:
normal_cursor()
while 1:
event = pygame.event.poll()
pygame.time.wait(1)
cursor = pygame.mouse.get_pos()
(esc_rect, close_font) = close_button(screen)
hover_fx(screen, x, cursor, marker)
hover_cursor(cursor, [esc_rect] + [y[0] for y in x])
if left_click(event):
if esc_rect.collidepoint(cursor):
gl.ESCAPED = 1
return (new_img, new_img, new_img, file)
for item in x: # load clicked image:
if item[0].collidepoint(cursor):
wait_cursor()
new_img = load_img(item[1])
file = gl.files.index(item[1])
return (new_img, new_img, new_img, file)
check_quit(event)
if (
hit_key(event, K_SPACE)
or hit_key(event, K_t)
or hit_key(event, K_n)
or hit_key(event, K_p)
or hit_key(event, K_PAUSE)
or right_click(event)
):
if not place >= len(gl.files):
if not gl.PAUSED: # go to next thumb page:
paint_screen(gl.IMGV_COLOR)
close_button(screen)
set_caption("Thumbnails [Paused]")
x = []
screen_pause = 0
marker = 0
break
# unpause:
gl.PAUSED = 0
screen_pause = 0
break
if hit_key(event, K_BACKSPACE) or hit_key(event, K_b) or middle_click(event):
# go back to previous thumb page, even if paused:
if (place - marker) > 0:
i = j = SPACER
if gl.PAUSED:
gl.PAUSED = 0
paint_screen(gl.IMGV_COLOR)
close_button(screen)
screen_pause = 0
place = place - (marker + gl.MAX_THUMBS)
marker = 0
x = []
break
if hit_key(event, K_ESCAPE):
gl.ESCAPED = 1
return (new_img, new_img, new_img, file)
else:
set_caption("Loading Thumbnails [%d] - imgv" % marker)
(x, i, j, place, screen_pause, marker) = show_thumbs(
screen, SPACER, x, i, j, place, marker, font, font_size
)
pygame.time.delay(5)
return (new_img, new_img, new_img, file)
def command_main_menu(refresh_img, screen, file, num_imgs, rect, new_img, img, new_img_width, new_img_height, ns):
menu_items = []
i = 23
cursor = pygame.mouse.get_pos()
screen_width = screen.get_width()
screen_height = screen.get_height()
if screen_height < 600:
font_size = 10
gl.MENU_DIVIDER_AMOUNT = 12
else:
gl.MENU_DIVIDER_AMOUNT = 14
font_size = 11
font = pygame.font.Font(gl.FONT_NAME, font_size)
font.set_bold(1)
if gl.MENU_POS == -1:
gl.MENU_POS = cursor[0]
main_menu_fg(screen, font, i, menu_items)
normal_cursor()
last_rect = Rect(rect)
new_img_width = new_img.get_width()
new_img_height = new_img.get_height()
if gl.REMOTE and not gl.ALREADY_DOWNLOADED:
download_rect = imgv_button(screen, " Downlo(a)d Image ", None, None, "topright")
while 1:
event = pygame.event.poll()
pygame.time.wait(1)
check_quit(event)
cursor2 = pygame.mouse.get_pos()
if event.type == VIDEORESIZE:#
gl.JUST_RESIZED = 1
screen = pygame.display.set_mode(event.dict['size'], RESIZABLE)
rect = get_center(screen, new_img)
my_update_screen(new_img, screen, rect, file, len(gl.files))
return (refresh_img, screen, file, num_imgs, new_img, img, new_img_width, new_img_height, rect)
if gl.REMOTE and not gl.ALREADY_DOWNLOADED:
hover_button(download_rect, cursor2, screen, " Downlo(a)d Image ", None, None, "topright")
if event.type == KEYDOWN:
if event.key in (K_c, K_h, K_UP, K_DOWN, K_RIGHT, K_LEFT):
if event.key != K_c:
pygame.event.set_allowed(MOUSEMOTION)
gl.HAND_TOOL = 1
drag_hand_cursor()
# close the menu
gl.MENU_POS = -1
my_update_screen(new_img, screen, rect, file, len(gl.files))
if event.key == K_c:
normal_cursor()
return (refresh_img, screen, file, num_imgs, new_img, img, new_img_width, new_img_height, rect)
(screen, rect, new_img, img, refresh_img, file, num_imgs,\
screen_width, screen_height, new_img_width, new_img_height, last_rect) =\
handle_keyboard(event, screen, rect, new_img, img, refresh_img, file, len(gl.files),\
screen_width, screen_height, new_img_width, new_img_height, last_rect, ns)
break
hover_cursor(cursor2, [x[0] for x in menu_items])
if left_click(event):
if gl.REMOTE and not gl.ALREADY_DOWNLOADED:
if download_rect.collidepoint(cursor2):
wait_cursor()
save_remote_img(screen, file)
break
for it in menu_items:
if it[0].collidepoint(cursor2):
if it[1] == " Next Image ":
(new_img, img, refresh_img, file, rect) = command_next_img(new_img, screen, file, len(gl.files), rect)
elif it[1] == " Previous Image ":
(new_img, img, refresh_img, file, rect) = command_prev_img(new_img, screen, file, len(gl.files), rect)
elif it[1] == " Directory Browser ":
gl.USING_SCROLL_MENU = 1
# save current things in case the user ESCAPES out of show_dirs()
gl.LAST_DIR = getcwd()
last_files = gl.files
(last_new_img, last_img, last_refresh_img, last_num_imgs, last_file, last_rect) =\
(new_img, img, refresh_img, num_imgs, file, rect)
(new_img, img, refresh_img, num_imgs, file, rect) = command_show_dirs(new_img, img, screen, rect,\
file, num_imgs)
# user ESCAPED from show_dirs, reset last values
if gl.ESCAPED:
(new_img, img, refresh_img, num_imgs, file, rect) =\
(last_new_img, last_img, last_refresh_img, last_num_imgs, last_file, last_rect)
chdir(gl.LAST_DIR)
gl.files = last_files
gl.USING_SCROLL_MENU = 0
my_update_screen(new_img, screen, rect, file, num_imgs)
else:
gl.REFRESH_IMG_COUNT = 0
gl.ESCAPED = 0
gl.USING_SCROLL_MENU = 0
elif it[1] == " Image Browser ":
(new_img, img, refresh_img, file, rect) = command_img_names(screen, new_img, img, file,\
num_imgs, rect)
elif it[1] == " Thumbnails ":
gl.THUMBING = 1
gl.CALC_ZOOM = 0
zoom_percent = gl.CURRENT_ZOOM_PERCENT
real_width = gl.REAL_WIDTH
(new_img, img, refresh_img, file, rect) = command_thumbs(screen, new_img, file, ns)
if gl.ESCAPED:
gl.CURRENT_ZOOM_PERCENT = zoom_percent
gl.REAL_WIDTH = real_width
gl.ESCAPED = 0
gl.THUMBING = 0
my_update_screen(new_img, screen, rect, file, num_imgs)
elif it[1] == " Image Properties ":
transparency = 0
if not gl.TOGGLE_TRANSPARENT:
gl.TOGGLE_TRANSPARENT = 1
transparency = 1
command_verbose_info(screen, new_img, rect, file, num_imgs)
if transparency:
gl.TOGGLE_TRANSPARENT = 0
my_update_screen(new_img, screen, rect, file, num_imgs)
elif it[1] == " Zoom Out ":
if int(gl.N_MILLISECONDS) < gl.MAX_ZOOM_MAX_MS and gl.CURRENT_ZOOM_PERCENT< gl.ZOOM_PERCENT_MAX:
try:
(new_img, img, rect) = command_zoom_out(new_img, new_img_width, new_img_height, img, screen, file, num_imgs, rect, "normal")
except:
print 'Out of memory.'
else:
print "Can't zoom out. Out of memory. Resetting the image."
gl.SKIP_FIT = 1
gl.ZOOM_EXP = 0
start = start_timer()
wait_cursor()
new_img = load_img(gl.files[file], screen)
img = refresh_img = new_img
rect = get_center(screen, new_img)
ns = check_timer(start)
my_update_screen(new_img, screen, rect, file, num_imgs, ns)
normal_cursor()
gl.N_MILLISECONDS = "0"
elif it[1] == " Zoom In ":
if int(gl.N_MILLISECONDS) < gl.MAX_ZOOM_MAX_MS and gl.CURRENT_ZOOM_PERCENT < gl.ZOOM_PERCENT_MAX:
try: # triple zoom crash protection
(new_img, img, rect) = command_zoom_in(new_img, new_img_width, new_img_height, img, screen, gl.files, file, num_imgs, rect, "normal")
except:
print 'Zoom max reached.'
else:
print 'Zoom max reached.'
elif it[1] == " Fit to Window ":
gl.RESET_FIT = 0
gl.SCALE_UP = 1
if gl.FIT_IMAGE_VAL:
gl.RESET_FIT = 0
else:
gl.RESET_FIT = 1
gl.FIT_IMAGE_VAL = 1
start = start_timer()
wait_cursor()
new_img = load_img(gl.files[file], screen)
img = refresh_img = new_img
rect = get_center(screen, new_img)
ns = check_timer(start)
my_update_screen(new_img, screen, rect, file, num_imgs, ns)
if gl.RESET_FIT == 1:
gl.FIT_IMAGE_VAL = 0
normal_cursor()
elif it[1] == " Lock Zoom ":
gl.PERSISTENT_ZOOM_VAL ^= 1
if not gl.PERSISTENT_ZOOM_VAL:
gl.ZOOM_EXP = 0
my_update_screen(new_img, screen, rect, file, num_imgs, ns)
elif it[1] == " Actual Size ":
gl.SKIP_FIT = 1
gl.ZOOM_EXP = 0
start = start_timer()
wait_cursor()
new_img = load_img(gl.files[file], screen)
img = refresh_img = new_img
rect = get_center(screen, new_img)
ns = check_timer(start)
my_update_screen(new_img, screen, rect, file, num_imgs, ns)
normal_cursor()
elif it[1] == " Close Image ":
(new_img, img, refresh_img, file, num_imgs, rect) = command_remove_img(new_img, screen, file, rect)
elif it[1] == " Rotate Right ":
if int(gl.N_MILLISECONDS) < gl.MAX_ZOOM_MAX_MS and gl.CURRENT_ZOOM_PERCENT < gl.ZOOM_PERCENT_MAX:
gl.CALC_ZOOM = 0
(new_img, img, rect) = command_rotate_right(new_img, screen, file, num_imgs, rect)
else:
print "Can't rotate. Out of memory."
elif it[1] == " Rotate Left ":
if int(gl.N_MILLISECONDS) < gl.MAX_ZOOM_MAX_MS and gl.CURRENT_ZOOM_PERCENT < gl.ZOOM_PERCENT_MAX:
gl.CALC_ZOOM = 0
(new_img, img, rect) = command_rotate_left(new_img, screen, file, num_imgs, rect)
else:
print "Can't rotate. Out of memory."
elif it[1] == " Four at a Time ":
gl.CALC_ZOOM = 0
zoom_percent = gl.CURRENT_ZOOM_PERCENT
real_width = gl.REAL_WIDTH
(file, new_img, img, refresh_img, rect) = command_four(screen, file, new_img, ns)
if gl.ESCAPED:
gl.CURRENT_ZOOM_PERCENT = zoom_percent
gl.REAL_WIDTH = real_width
gl.ESCAPED = 0
my_update_screen(new_img, screen, rect, file, num_imgs)
normal_cursor()
# elif it[1] == " Resize Options ":##
# gl.USING_SCROLL_MENU = 1
# gl.CALC_ZOOM = 0
# zoom_percent = gl.CURRENT_ZOOM_PERCENT
# real_width = gl.REAL_WIDTH
# rect = command_show_res_modes(screen, new_img, file, num_imgs, rect)
# gl.CURRENT_ZOOM_PERCENT = zoom_percent
# gl.REAL_WIDTH = real_width
# gl.USING_SCROLL_MENU = 0
elif it[1] == " Refresh ":
(new_img, img, rect, file) = command_refresh(refresh_img, screen, gl.files, file, num_imgs)
elif it[1] == " First Image ":
(new_img, img, refresh_img, file, rect) = command_first_img(new_img, screen, file, len(gl.files), rect)
elif it[1] == " Last Image ":
(new_img, img, refresh_img, file, rect) = command_last_img(new_img, screen, file, len(gl.files), rect)
elif it[1] == " Shuffle ":
(new_img, img, refresh_img, rect) = command_shuffle(new_img, img, screen, rect, file, num_imgs)
elif it[1] == " Unshuffle ":
(new_img, img, refresh_img, rect, file) = command_unshuffle(new_img, img, screen, rect, file, num_imgs)
elif it[1] == " Flip Horizontal ":
(new_img, img, rect) = command_horiz(new_img, screen, file, num_imgs, rect)
elif it[1] == " Flip Vertical ":
(new_img, img, rect) = command_vert(new_img, screen, file, num_imgs, rect)
elif it[1] == " Slideshow ":
(new_img, img, refresh_img, file, rect) = my_slideshow(new_img, img, screen, file, num_imgs, rect)
my_update_screen(new_img, screen, rect, file, len(gl.files))
elif it[1] == " Playlists ":
(new_img, new_img, new_img, file, rect, num_imgs) = command_play_list_options(screen, file)
gl.SORT_HIT = 0
elif it[1] == " Add to Playlist ":
command_add_to_play_list(screen, gl.files[file])
gl.SORT_HIT = 0
my_update_screen(new_img, screen, rect, file, len(gl.files))
elif it[1] == " Edit ":
(screen, before_winsize, not_accepted) = adjust_screen(screen)
gl.USING_SCROLL_MENU = 1
(new_img, img, refresh_img, file, num_imgs, rect) = command_edit_menu(screen, file, new_img, num_imgs, rect)
screen = restore_screen(screen, before_winsize, not_accepted, new_img, file, num_imgs, rect)
my_update_screen(new_img, screen, rect, file, num_imgs)
gl.USING_SCROLL_MENU = 0
elif it[1] == " Hide Image ":
command_hide(screen, new_img, rect, file, num_imgs)
elif it[1] == " Extract from Web ":
(screen, before_winsize, not_accepted) = adjust_screen(screen)
(new_img, num_imgs) = open_url(screen, img)
gl.URL_ERROR = False
file = 0
img = refresh_img = new_img
screen = restore_screen(screen, before_winsize, not_accepted, new_img, file, num_imgs, rect)
rect = get_center(screen, new_img)
my_update_screen(new_img, screen, rect, file, len(gl.files))
normal_cursor()
elif it[1] == " Help ":
gl.CALC_ZOOM = 0
zoom_percent = gl.CURRENT_ZOOM_PERCENT
real_width = gl.REAL_WIDTH
command_help(screen, new_img, file, rect, num_imgs)
if gl.ESCAPED:
gl.CURRENT_ZOOM_PERCENT = zoom_percent
gl.REAL_WIDTH = real_width
gl.ESCAPED = 0
elif it[1] == " Close Menu " or it[1] == " Hand Tool ":
if it[1] == " Hand Tool ":
pygame.event.set_allowed(MOUSEMOTION)
gl.HAND_TOOL = 1
drag_hand_cursor()
gl.MENU_POS = -1
my_update_screen(new_img, screen, rect, file, len(gl.files))
normal_cursor()
return (refresh_img, screen, file, num_imgs, new_img, img, new_img_width, new_img_height, rect)
elif it[1] == " Exit ":
clean_screen()
raise SystemExit
break
if event.type == KEYDOWN and event.key not in (K_LALT, K_RALT, K_LCTRL, K_RCTRL, K_TAB):
return (refresh_img, screen, file, num_imgs, new_img, img, new_img_width, new_img_height, rect)
if middle_click(event):
"close the menu upon middle click"
gl.MENU_POS = -1
my_update_screen(new_img, screen, rect, file, len(gl.files))
normal_cursor()
return (refresh_img, screen, file, num_imgs, new_img, img, new_img_width, new_img_height, rect)
if right_click(event):
wait_cursor()
gl.MENU_POS = -1
my_update_screen(new_img, screen, rect, file, len(gl.files))
(refresh_img, screen, file, num_imgs, new_img, img, new_img_width, new_img_height, rect) =\
command_main_menu(refresh_img, screen, file, num_imgs, rect, new_img, img, new_img_width, new_img_height, ns)
return (refresh_img, screen, file, num_imgs, new_img, img, new_img_width, new_img_height, rect)
if event.type == MOUSEBUTTONDOWN: # this needs to be down here to work
if event.dict['button'] in (4, 5): # scroll wheel activated
# allow for mouse dragging:
pygame.event.set_allowed(MOUSEMOTION)
gl.HAND_TOOL = 1
drag_hand_cursor()
# close menu:
gl.MENU_POS = -1
my_update_screen(new_img, screen, rect, file, len(gl.files))
return (refresh_img, screen, file, num_imgs, new_img, img, new_img_width, new_img_height, rect)
if gl.KEEP_MENU_OPEN == "1":
# this code purposely closes the main menu by breaking the recursion to free up RAM memory
gl.COUNT_CLICKS += 1
if gl.COUNT_CLICKS == 1: # free up ram every click
return (refresh_img, screen, file, num_imgs, new_img, img, new_img_width, new_img_height, rect)
(refresh_img, screen, file, num_imgs, new_img, img, new_img_width, new_img_height, rect) =\
command_main_menu(refresh_img, screen, file, num_imgs, rect, new_img, img, new_img_width, new_img_height, ns)
normal_cursor()
return (refresh_img, screen, file, num_imgs, new_img, img, new_img_width, new_img_height, rect)
def main():
pygame.time.delay(5) # to make start_timer() work initially
start = start_timer()
num_imgs = len(gl.files)
pygame.init() # needed for Mac OSX?
init_screen()
wait_cursor()
screen = pygame.display.set_mode(gl.DEFAULT_RES, RESIZABLE)
set_caption(gl.TITLE)
if gl.REMOTE == 1:
show_message(screen, "Loading image. Please wait..", 34, 42)
file = 0
if num_imgs < 1:
gl.files = [gl.IMGV_LOGO]
num_imgs = 1
img = load_img(gl.files[file], screen)
else:
img = load_img(gl.files[file], screen)
wait_cursor()
img_width = img.get_width()
img_height = img.get_height()
refresh_img = img
new_img = img
rect = get_center(screen, new_img)
ns = check_timer(start)
my_update_screen(new_img, screen, rect, file, num_imgs, ns)
normal_cursor()
if gl.START_FULLSCREEN:
command_fullscreen(screen, new_img, file, num_imgs, rect)
my_update_screen(new_img, screen, rect, file, num_imgs, ns)
# start with menu open
screen_width = screen.get_width()
screen_height = screen.get_height()
new_img_width = new_img.get_width()
new_img_height = new_img.get_height()
(refresh_img, screen, file, num_imgs, new_img, img,\
new_img_width, new_img_height, rect) = command_main_menu(refresh_img, screen,\
file, len(gl.files), rect, new_img, img, new_img_width, new_img_height, ns)
minus1 = minus2 = 0
# main loop
while 1:
event = pygame.event.poll()
pygame.time.wait(1) # so pygame doesn't use 100% CPU
cursor = pygame.mouse.get_pos()
last_rect = Rect(rect)
screen_width = screen.get_width()
screen_height = screen.get_height()
new_img_width = new_img.get_width()
new_img_height = new_img.get_height()
# drag image code:
if gl.HAND_TOOL:
if left_click(event): # calculate drag coordinates:
if gl.IMG_BORDER:
border_fix(screen) # erase the current border
grab_hand_cursor()
minus1 = cursor[0] - rect[0]
minus2 = cursor[1] - rect[1]
gl.DO_DRAG = 1
if event.type == MOUSEMOTION and gl.DO_DRAG: # move the image when dragged:
grab_hand_cursor()
rect[0] = cursor[0] - minus1
rect[1] = cursor[1] - minus2
screen.fill(gl.IMGV_COLOR, last_rect)
screen.blit(new_img, rect)
update(rect.union(last_rect))
if event.type == MOUSEBUTTONUP: # released mouse button, redisplay status bars:
drag_hand_cursor()
my_update_screen(new_img, screen, rect, file, num_imgs, ns)
gl.DO_DRAG = 0
if event.type == VIDEORESIZE:#
print 'resize imgv.py'#
screen = pygame.display.set_mode(event.dict['size'], RESIZABLE)
rect = get_center(screen, new_img)
my_update_screen(new_img, screen, rect, file, num_imgs, ns)
if event.type == KEYDOWN:
gl.HAND_TOOL = 0
if event.key not in (K_DOWN, K_UP, K_RIGHT, K_LEFT):
normal_cursor() # stop displaying hand tool
(screen, rect, new_img, img, refresh_img, file, num_imgs,\
screen_width, screen_height, new_img_width, new_img_height, last_rect) =\
handle_keyboard(event, screen, rect, new_img, img, refresh_img, file, len(gl.files),\
screen_width, screen_height, new_img_width, new_img_height, last_rect, ns)
if event.type == KEYUP:
stop_auto_repeat()
check_quit(event)
if event.type == MOUSEBUTTONDOWN: # open main menu:
if right_click(event):
gl.HAND_TOOL = 0
(refresh_img, screen, file, num_imgs, new_img, img,\
new_img_width, new_img_height, rect) = command_main_menu(refresh_img, screen,\
file, num_imgs, rect, new_img, img, new_img_width, new_img_height, ns)
if (gl.KEEP_MENU_OPEN == "1" and gl.COUNT_CLICKS == 1) or gl.JUST_RESIZED: # Re-open the purposely closed window that frees up RAM
gl.COUNT_CLICKS = 0
gl.JUST_RESIZED = 0
(refresh_img, screen, file, num_imgs, new_img, img,\
new_img_width, new_img_height, rect) = command_main_menu(refresh_img, screen,\
file, num_imgs, rect, new_img, img, new_img_width, new_img_height, ns)
start_auto_repeat(rect, last_rect, new_img, screen, file, len(gl.files), screen_width, screen_height, event)
clean_screen()
def show_dirs(screen, num_imgs, file):
wait_cursor()
if platform == 'win32':
try:
os.chdir(gl.DRIVE + ":")
except:
pass # Probably an OSError from not having a cd in the drive
slash, get_curdir = os.sep, os.getcwd()
fg_color = gl.SILVER
font_size = 10
font = pygame.font.Font(gl.FONT_NAME, font_size)
font.set_bold(1) # very important
line = 55
name_max = 16 # dir name max
menu_items = []
if get_curdir[-1] == slash:
curdir = get_curdir
else:
curdir = get_curdir + slash
paint_screen(screen, gl.BLACK)
screen_height = screen.get_height()
if not gl.BEEN_THERE_DONE_THAT:
show_message(screen, "You can type in a directory number or shortcut (L/T/A/D/C/V/S/Q) instead of clicking: _", "bottom", 11)
else:
show_message(screen, "Directory number or shortcut: _", "bottom", 11)
curdir_msg = check_truncate(screen.get_width(), curdir)
all_files = os.listdir('.')
all_files.sort()
n_dirs = len([d for d in all_files if os.path.isdir(d)])
all_files = [f for f in all_files if not os.path.isdir(f)]
n_files = len(all_files)
all_images = get_imgs(os.getcwd(), 0)
n_images = len(all_images)
get_movies = lambda x, y: [i.upper().endswith(y) for i in all_images].count(True)
n_movies = get_movies(all_images, ".MPG") + get_movies(all_images, ".MPEG")
dirs_text = "Directories"
files_text = "Files"
images_text = "Images"
movies_text = "MPEGs"
if n_dirs == 1: dirs_text = "Directory" # unplural
if n_files == 1: files_text = "File"
if n_images == 1: images_text = "Image"
if n_movies == 1: movies_text = "MPEG"
files_msg = "[%d %s. %d %s. %d %s. %d %s]" % (n_dirs, dirs_text, n_files, files_text, n_images - n_movies, images_text, n_movies, movies_text)
curdir_msg_wpos = (screen.get_width() / 2 - font.size(curdir_msg)[0] / 2) - font.size(files_msg)[0] / 2 + 10
files_msg_wpos = screen.get_width() / 2 + font.size(curdir_msg)[0] / 2 - font.size(files_msg)[0] / 2 + 20
if not gl.REFRESH_IMG_COUNT and gl.CACHE_DIR_OK:
dirs = gl.CACHE_DIRS
gl.REFRESH_IMG_COUNT = 1
set_caption(curdir)
show_message(screen, curdir_msg, (curdir_msg_wpos, 4), 10, ("bold"))
show_message(screen, files_msg, (files_msg_wpos, 4), 10)
else:
show_message(screen, curdir_msg, (curdir_msg_wpos, 4), 10, ("bold"))
show_message(screen, files_msg, (files_msg_wpos, 4), 10)
set_caption(curdir)
dirs = os.listdir(curdir)
dirs.sort()
dirs = strip_dirs(dirs)
# ensure the root dir and last dir items go at top
dirs.insert(0, "..")
dirs.insert(0, slash)
gl.CACHE_DIRS = dirs
ren_load_rect = imgv_button(screen, " (L)oad ", 0, 18, "topleft")
ren_load_subdirs_rect = imgv_button(screen, " Subdirs (T)oo ", 62, 18, "topleft")
if platform == 'win32':
ren_drive_rect = imgv_button(screen, " Change (D)rive ", 271, 18, "topleft")
dirpl_rect = imgv_button(screen, " (A)dd To Playlist ", 160, 18, "topleft")
untag_all_rect = imgv_button(screen, " (C)lear Tags ", 380, 18, "topleft")
view_tagged_rect = imgv_button(screen, " (V)iew Tags ", 472, 18, "topleft")
filter_rect = imgv_button(screen, " (S)earch ", 559, 18, "topleft")
col = 10
show_message(screen, "Right-Click directories to tag multiple directories to load. Ctrl+Left-Click to untag.", (10, 40), 10)
if gl.MULT_DIRS != []:
show_message(screen, "[Dirs tagged: %s]" % len(gl.MULT_DIRS), (440, 40), 10, "bold")
if gl.FILTER_COMMAND != {}:
show_message(screen, "[Filter: on]", (560, 40), 10, "bold")
# add numbers to directory names
if gl.ADDED_DIR_NUMS == 0 and dirs[0] != '*':
gl.ADDED_DIR_NUMS = 1
for i, d in enumerate(dirs):
dirs[i] = '*' + str(i) + gl.DIRNUMSEP + d
for d in dirs:
d = d[1:] # strip out the '*' marker
if d[3:] == slash:
if gl.DIRNUM_COLORS:
dmsg = d
else:
ren = font.render(d, 1, gl.MSG_COLOR)
else:
if len(d) > name_max:
if gl.DIRNUM_COLORS:
dmsg = truncate_name(d, name_max)
else:
ren = font.render(truncate_name(d, name_max), 1, gl.MSG_COLOR)
else:
if gl.DIRNUM_COLORS:
dmsg = d + slash
else:
ren = font.render(d + slash, 1, gl.MSG_COLOR)
# print directory names on screen, wrapping if necessary
font_height = font.size(' '.join(d.split(' ')[1:]))[1]
if (line + font_height) >= (screen_height - 10):
line = 55 # reset to beginning of screen
col = col + (name_max + 130) # go to next column
if gl.DIRNUM_COLORS:
before_color = gl.MSG_COLOR
if gl.MSG_COLOR == gl.SILVER:
gl.MSG_COLOR = (142, 142, 142)
else:
gl.MSG_COLOR = gl.SILVER
if before_color == gl.WHITE:
ren_rect = show_message(screen, dmsg, (col, line), font_size, ("bold"), (len(dmsg[:dmsg.index(gl.DIRNUMSEP) + 1]), gl.SILVER))
else:
ren_rect = show_message(screen, dmsg, (col, line), font_size, ("bold"), (len(dmsg[:dmsg.index(gl.DIRNUMSEP) + 1]), before_color))
else:
ren_rect = ren.get_rect()
ren_rect[0] = col
ren_rect[1] = line
screen.blit(ren, ren_rect)
update(ren_rect)
line = line + 12
menu_items.append((ren_rect, d))
if gl.DIRNUM_COLORS:
gl.MSG_COLOR = before_color
normal_cursor()
pygame.event.set_blocked(MOUSEMOTION)
while 1:
event = pygame.event.poll()
pygame.time.wait(1)
cursor = pygame.mouse.get_pos()
(esc_rect, close_font) = close_button(screen)
hover_fx(screen, curdir, menu_items, cursor)
if platform == 'win32':#
hover_cursor(cursor, [ren_load_rect, ren_load_subdirs_rect, ren_drive_rect, dirpl_rect, untag_all_rect, view_tagged_rect, filter_rect, esc_rect] + [x[0] for x in menu_items])
hover_button(ren_load_rect, cursor, screen, " (L)oad ", 0, 18, "topleft")
hover_button(ren_load_subdirs_rect, cursor, screen, " Subdirs (T)oo ", 62, 18, "topleft")
if platform == 'win32':
hover_button(ren_drive_rect, cursor, screen, " Change (D)rive ", 271, 18, "topleft")
hover_button(dirpl_rect, cursor, screen, " (A)dd To Playlist ", 160, 18, "topleft")
hover_button(untag_all_rect, cursor, screen, " (C)lear Tags ", 380, 18, "topleft")
hover_button(view_tagged_rect, cursor, screen, " (V)iew Tags ", 472, 18, "topleft")
hover_button(filter_rect, cursor, screen, " (S)earch ", 559, 18, "topleft")
if left_click(event):
for item in menu_items:
if item[0].collidepoint(cursor):
if pygame.mouse.get_pressed()[0] and (pygame.key.get_pressed()[K_LCTRL] or\
pygame.key.get_pressed()[K_RCTRL]):
try: # untag directory
gl.MULT_DIRS.remove(os.getcwd() + slash + ' '.join(item[1].split(' ')[1:]))
show_message(screen, " " * 30, (440, 40), 10, ("bold"))
show_message(screen, "[Dirs tagged: %s]" % len(gl.MULT_DIRS), (440, 40), 10, "bold")
except:
pass
else: # (normal mode) change to a single directory and load its images
(num_imgs, file) = do_change_dir(screen, num_imgs, file, item[1])
return (num_imgs, file)
if right_click(event):
for item in menu_items:
if item[0].collidepoint(cursor):
# tag directory
if os.getcwd()[-1] != slash:
gl.MULT_DIRS.append(os.getcwd() + slash + ' '.join(item[1].split(' ')[1:]))
else:
gl.MULT_DIRS.append(os.getcwd() + ' '.join(item[1].split(' ')[1:]))
show_message(screen, " " * 30, (440, 40), 10, ("bold"))
show_message(screen, "[Dirs tagged: %s]" % len(gl.MULT_DIRS), (440, 40), 10, "bold")
# allow number keys to be used to change directories
dirnum = None
if event.type == KEYDOWN and event.key in (K_1, K_2, K_3, K_4, K_5, K_6, K_7, K_8, K_9, K_0, K_KP1, K_KP2, K_KP3, K_KP4, K_KP5, K_KP6, K_KP7, K_KP8, K_KP9, K_KP0):
dirnum = get_dirnum(screen, event.key)
if dirnum != 'backspaced':
for item in menu_items:
if item[1].startswith(str(dirnum)):
(num_imgs, file) = do_change_dir(screen, num_imgs, file, item[1])
return (num_imgs, file)
break
if hit_key(event, K_RETURN) or hit_key(event, K_SPACE) or hit_key(event, K_l):
(num_imgs, file) = do_load_dir()
break
if hit_key(event, K_t): # load subdirs on keypress 't'
(num_imgs, file) = do_subdirs_too()
break
if hit_key(event, K_s): # search
command_get_filter_info(screen)
(num_imgs, file) = show_dirs(screen, num_imgs, file)
break
if hit_key(event, K_d): # change drives
if platform == 'win32':
gl.WAS_IN_CHANGE_DRIVES = 1
(num_imgs, file) = do_change_drive(screen, num_imgs, file)
gl.WAS_IN_CHANGE_DRIVES = 0
break
if hit_key(event, K_c): # clear tag list
do_untag(screen)
if hit_key(event, K_v): # view tagged dirs
do_view_tagged(screen, num_imgs, file)
break # break main loop to display properly
if hit_key(event, K_a): # add curdir to playlist
command_add_to_play_list(screen, curdir)
return (num_imgs, file)
if hit_key(event, K_ESCAPE):
gl.ESCAPED = 1
gl.ADDED_DIR_NUMS = 0
break
if left_click(event):
if esc_rect.collidepoint(cursor):
gl.ESCAPED = 1
gl.ADDED_DIR_NUMS = 0
break
if ren_load_rect.collidepoint(cursor): # load current dir
(num_imgs, file) = do_load_dir()
break
if ren_load_subdirs_rect.collidepoint(cursor): # load subdirs too
(num_imgs, file) = do_subdirs_too()
break
if platform == 'win32':
if ren_drive_rect.collidepoint(cursor):
gl.WAS_IN_CHANGE_DRIVES = 1
(num_imgs, file) = do_change_drive(screen, num_imgs, file)
gl.WAS_IN_CHANGE_DRIVES = 0
break
if dirpl_rect.collidepoint(cursor):
command_add_to_play_list(screen, curdir)
return (num_imgs, file)
if untag_all_rect.collidepoint(cursor):
do_untag(screen)
if view_tagged_rect.collidepoint(cursor):
do_view_tagged(screen, num_imgs, file)
break # break main loop to display properly
if filter_rect.collidepoint(cursor):
command_get_filter_info(screen)
(num_imgs, file) = show_dirs(screen, num_imgs, file)
break
check_quit(event)
gl.CACHE_DIR_OK = 1
return (num_imgs, file)
