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)