git.zarb.org Git - booh/blob - bin/booh-classifier

   1 #! /usr/bin/ruby
   2 #
   3 #                         *  BOOH  *
   4 #
   5 # A.k.a 'Best web-album Of the world, Or your money back, Humerus'.
   6 #
   7 # The acronyn sucks, however this is a tribute to Dragon Ball by
   8 # Akira Toriyama, where the last enemy beaten by heroes of Dragon
   9 # Ball is named "Boo". But there was already a free software project
  10 # called Boo, so this one will be it "Booh". Or whatever.
  11 #
  12 #
  13 # Copyright (c) 2004-2010 Guillaume Cottenceau <http://zarb.org/~gc/resource/gc_mail.png>
  14 #
  15 # This software may be freely redistributed under the terms of the GNU
  16 # public license version 2.
  17 #
  18 # You should have received a copy of the GNU General Public License
  19 # along with this program; if not, write to the Free Software
  20 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  21
  22 require 'getoptlong'
  23 require 'tempfile'
  24
  25 require 'gtk2'
  26 require 'booh/libadds'
  27
  28 require 'gettext'
  29 include GetText
  30 bindtextdomain("booh")
  31
  32 require 'booh/rexml/document'
  33 include REXML
  34
  35 require 'booh/booh-lib'
  36 include Booh
  37 require 'booh/UndoHandler'
  38
  39
  40 #- options
  41 $options = [
  42     [ '--help',          '-h', GetoptLong::NO_ARGUMENT, _("Get help message") ],
  43     [ '--sort-by-exif-date', '-s', GetoptLong::NO_ARGUMENT, _("Sort entries by EXIF date") ],
  44     [ '--verbose-level', '-v', GetoptLong::REQUIRED_ARGUMENT, _("Set max verbosity level (0: errors, 1: warnings, 2: important messages, 3: other messages)") ],
  45     [ '--version',       '-V', GetoptLong::NO_ARGUMENT, _("Print version and exit") ],
  46 ]
  47
  48 $preloader_allowed = false
  49
  50 def usage
  51     puts _("Usage: %s [OPTION]...") % File.basename($0)
  52     $options.each { |ary|
  53         printf " %3s, %-15s %s\n", ary[1], ary[0], ary[3]
  54     }
  55 end
  56
  57 def handle_options
  58     parser = GetoptLong.new
  59     parser.set_options(*$options.collect { |ary| ary[0..2] })
  60     begin
  61         parser.each_option do |name, arg|
  62             case name
  63             when '--help'
  64                 usage
  65                 exit(0)
  66
  67             when '--sort-by-exif-date'
  68                 $sort_by_exif_date = true
  69
  70             when '--verbose-level'
  71                 $verbose_level = arg.to_i
  72
  73             when '--version'
  74                 puts _("Booh version %s
  75
  76 Copyright (c) 2005-2010 Guillaume Cottenceau.
  77 This is free software; see the source for copying conditions.  There is NO
  78 warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.") % $VERSION
  79
  80                 exit(0)
  81
  82             end
  83         end
  84     rescue
  85         puts $!
  86         usage
  87         exit(1)
  88     end
  89 end
  90
  91 def total_memory
  92     meminfo = IO.readlines('/proc/meminfo').join
  93     meminfo =~ /MemTotal:.*?(\d+)/ or return -1
  94     memory = $1.to_i
  95     meminfo =~ /SwapTotal:.*?(\d+)/ or return -1
  96     return memory + $1.to_i
  97 end
  98
  99 def startup_memfree
 100     if $startup_memfree.nil?
 101         meminfo = IO.readlines('/proc/meminfo').join
 102         meminfo =~ /MemFree:.*?(\d+)/ or return -1
 103         memfree = $1
 104         meminfo =~ /Buffers:.*?(\d+)/ and buffers = $1
 105         meminfo =~ /Cached:.*?(\d+)/ and cached = $1
 106         $startup_memfree = memfree.to_i + buffers.to_i + cached.to_i
 107     end
 108     return $startup_memfree
 109 end
 110
 111 def set_cache_memory_use_figure
 112
 113     if $config['cache-memory-use'] =~ /memfree_(\d+)/
 114         $config['cache-memory-use-figure'] = startup_memfree*$1.to_f/100
 115     else
 116         $config['cache-memory-use-figure'] = $config['cache-memory-use'].to_i
 117     end
 118     #- cannot fork if process is > 0.5 total memory
 119     if $config['cache-memory-use-figure'] > total_memory * 0.4
 120         $config['cache-memory-use-figure'] = total_memory * 0.4
 121         msg 2, _("Cache memory used: %s kB (reduced because cannot exceed 50%% of total memory)") % $config['cache-memory-use-figure']
 122     else
 123         msg 2, _("Cache memory used: %s kB") % $config['cache-memory-use-figure']
 124     end
 125 end
 126
 127 def read_config
 128     $config = {}
 129     $config_file = File.expand_path('~/.booh-classifier-rc')
 130     if File.readable?($config_file)
 131         $xmldoc = REXML::Document.new(File.new($config_file))
 132         $xmldoc.root.elements.each { |element|
 133             txt = element.get_text
 134             if txt
 135                 if txt.value =~ /~~~/
 136                     $config[element.name] = txt.value.split(/~~~/)
 137                 else
 138                     $config[element.name] = txt.value
 139                 end
 140             elsif element.elements.size == 0
 141                 $config[element.name] = ''
 142             else
 143                 $config[element.name] = {}
 144                 element.each { |chld|
 145                     txt = chld.get_text
 146                     $config[element.name][chld.name] = txt ? txt.value : nil
 147                 }
 148             end
 149         }
 150     end
 151     $config['video-viewer'] ||= '/usr/bin/mplayer %f || /usr/bin/vlc %f'
 152     $config['browser'] ||= "/usr/bin/mozilla-firefox -remote 'openURL(%f,new-window)' || /usr/bin/mozilla-firefox %f || /usr/bin/firefox -remote 'openURL(%f,new-window)' || /usr/bin/firefox %f"
 153     $config['preload-distance'] ||= '5'
 154     $config['cache-memory-use'] ||= 'memfree_80%'
 155     $config['rotate-set-exif'] ||= 'true'
 156     $config['thumbnails-height'] ||= '64'
 157     set_cache_memory_use_figure
 158 end
 159
 160 def check_config
 161     missing = %w(mplayer).delete_if { |prg| system("which #{prg} >/dev/null 2>/dev/null") }
 162     if missing != []
 163         show_popup($main_window, utf8(_("The following program(s) are needed to handle videos: '%s'. Videos will be ignored.") % missing.join(', ')), { :pos_centered => true })
 164     end
 165
 166     if !system("which exif >/dev/null 2>/dev/null")
 167         show_popup($main_window, utf8(_("The program 'exif' is needed to view EXIF data. Please install it.")), { :pos_centered => true })
 168     end
 169     viewer_binary = $config['video-viewer'].split.first
 170     if viewer_binary && ! File.executable?(viewer_binary)
 171         show_popup($main_window, utf8(_("The configured video viewer seems to be unavailable.
 172 You should fix this in Edit/Preferences so that you can view videos.
 173
 174 Problem was: '%s' is not an executable file.
 175 Hint: don't forget to specify the full path to the executable,
 176 e.g. '/usr/bin/mplayer' is correct but 'mplayer' only is not.") % viewer_binary), { :pos_centered => true, :not_transient => true })
 177     end
 178     check_browser
 179 end
 180
 181 def write_config
 182     ios = File.open($config_file, "w")
 183     $xmldoc = Document.new "<booh-classifier-rc version='#{$VERSION}'/>"
 184     $xmldoc << XMLDecl.new(XMLDecl::DEFAULT_VERSION, $CURRENT_CHARSET)
 185     $config.each_pair { |key, value|
 186         elem = $xmldoc.root.add_element key
 187         if value.is_a? Hash
 188             $config[key].each_pair { |subkey, subvalue|
 189                 subelem = elem.add_element subkey
 190                 subelem.add_text subvalue.to_s
 191             }
 192         elsif value.is_a? Array
 193             elem.add_text value.join('~~~')
 194         else
 195             if !value
 196                 elem.remove
 197             else
 198                 elem.add_text value.to_s
 199             end
 200         end
 201     }
 202     $xmldoc.write(ios)
 203     ios.close
 204 end
 205
 206 def save_undo(name, closure, *params)
 207     UndoHandler.save_undo(name, closure, [ *params ])
 208     $undo_mb.sensitive = true
 209     $redo_mb.sensitive = false
 210 end
 211
 212 def get_mem
 213     IO.readlines('/proc/self/status').join =~ /VmSize.*?(\d+)\s*kB/
 214     msg 3, "VmSize: #{$1}"
 215     return $1.to_i
 216 end
 217
 218 def show_mem(*txt)
 219     txt.length > 0 and print txt[0]
 220     msg 2, "RSS: #{get_mem}"
 221 end
 222
 223 class Gdk::Color
 224     def darker
 225         color = dup
 226         color.red = [ color.red - 10000, 0 ].max
 227         color.green = [ color.green - 10000, 0 ].max
 228         color.blue = [ color.blue - 10000, 0 ].max
 229         return color
 230     end
 231     def lighter
 232         color = dup
 233         color.red = [ color.red + 10000, 65535 ].min
 234         color.green = [ color.green + 10000, 65535 ].min
 235         color.blue = [ color.blue + 10000, 65535 ].min
 236         return color
 237     end
 238 end
 239
 240 $color_red = Gdk::Color.new(65535, 0, 0)
 241 $colors = [ Gdk::Color.new(0, 65535, 0),
 242             Gdk::Color.new(0, 0, 65535),
 243             Gdk::Color.new(65535, 65535, 0),
 244             Gdk::Color.new(0, 65535, 65535),
 245             Gdk::Color.new(65535, 0, 65535) ]
 246
 247 class Label
 248     attr_accessor :color, :name, :button, :counter
 249     def initialize(name)
 250         @name = name
 251     end
 252 end
 253
 254 class InterruptedLoading < Exception
 255     #- not a StandardError, not catched by a simple rescue
 256 end
 257
 258 def show_pixbufs_present
 259     if 3 <= $verbose_level
 260         out = 'Full pixbufs ['
 261         for entry in $allentries
 262             out += entry.pixbuf_full_present? ? 'F' : '.'
 263         end
 264         msg 3, out + ']'
 265     end
 266 end
 267
 268 class Entry
 269     @@max_width = nil
 270     def Entry.thumbnails_height
 271         return $config['thumbnails-height'].to_i
 272     end
 273
 274     attr_accessor :path, :guipath, :type, :angle, :button, :image, :alignment, :removed, :labeled, :loader
 275
 276     def initialize(path, type, guipath)
 277         @path = path
 278         @type = type
 279         @guipath = guipath
 280         if @@max_width.nil?
 281             @@max_width = $main_window.root_window.size[0] - $labels_vbox.allocation.width - ( $videoborder_pixbuf.width + MainView.borders_thickness) * 2
 282         end
 283     end
 284
 285     def pixbuf_full_present?
 286         return ! @pixbuf_full.nil?
 287     end
 288     def free_pixbuf_full
 289         if @pixbuf_full.nil?
 290             return false
 291         else
 292             msg 3, ">>> free_pixbuf_full #{path}"
 293             @pixbuf_full = nil
 294             return true
 295         end
 296     end
 297
 298     def pixbuf_main
 299         Gtk.main_iteration while Gtk.events_pending?
 300         width, height = $mainview.window.size
 301         width = MainView.get_usable_width(width)
 302         height = MainView.get_usable_height(height)
 303         if @pixbuf_main.nil? || width != @width || height != @height
 304             @width = width
 305             @height = height
 306             load_into_pixbuf_full  #- make sure it is loaded
 307             if @pixbuf_full.nil?
 308                 return
 309             end
 310             if @pixbuf_full.width.to_f / @pixbuf_full.height > width.to_f / height
 311                 resized_height = @pixbuf_full.height * (width.to_f/@pixbuf_full.width)
 312                 if @pixbuf_full.width > width || @pixbuf_full.height > resized_height
 313                     @pixbuf_main = @pixbuf_full.scale(width, resized_height, Gdk::Pixbuf::INTERP_BILINEAR)
 314                 else
 315                     @pixbuf_main = @pixbuf_full
 316                 end
 317             else
 318                 resized_width = @pixbuf_full.width * (height.to_f/@pixbuf_full.height)
 319                 if @pixbuf_full.width > resized_width || @pixbuf_full.height > height
 320                     @pixbuf_main = @pixbuf_full.scale(resized_width, height, Gdk::Pixbuf::INTERP_BILINEAR)
 321                 else
 322                     @pixbuf_main = @pixbuf_full
 323                 end
 324             end
 325         end
 326         return @pixbuf_main
 327     end
 328     def pixbuf_main_present?
 329         return ! @pixbuf_main.nil?
 330     end
 331     def free_pixbuf_main
 332         if @pixbuf_main.nil?
 333             return false
 334         else
 335             msg 3, ">>> free_pixbuf_main #{path}"
 336             @pixbuf_main = nil
 337             return true
 338         end
 339     end
 340
 341     def pixbuf_thumbnail
 342         Gtk.main_iteration while Gtk.events_pending?
 343         if @pixbuf_thumbnail.nil?
 344             if @pixbuf_main
 345                 msg 3, ">>> pixbuf_thumbnail from main #{path}"
 346                 @pixbuf_thumbnail = @pixbuf_main.scale(@pixbuf_main.width * (Entry.thumbnails_height.to_f/@pixbuf_main.height), Entry.thumbnails_height, Gdk::Pixbuf::INTERP_BILINEAR)
 347             else
 348                 msg 3, ">>> pixbuf_thumbnail from file #{path}"
 349                 @pixbuf_thumbnail = load_into_pixbuf_at_size { |w, h|
 350                     if @angle == 0
 351                         if h > Entry.thumbnails_height
 352                             [ w * Entry.thumbnails_height.to_f/h, Entry.thumbnails_height ]
 353                         else
 354                             [ w, h ]
 355                         end
 356                     else
 357                         if w > Entry.thumbnails_height
 358                             [ Entry.thumbnails_height, h * Entry.thumbnails_height.to_f/w ]
 359                         else
 360                             [ w, h ]
 361                         end
 362                     end
 363                 }
 364             end
 365         end
 366         return @pixbuf_thumbnail
 367     end
 368     def free_pixbuf_thumbnail
 369         if @pixbuf_thumbnail.nil?
 370             return false
 371         else
 372             msg 3, ">>> free_pixbuf_thumbnail #{path}"
 373             @pixbuf_thumbnail = nil
 374             return true
 375         end
 376     end
 377
 378     def outline_color
 379         if removed
 380             return $color_red
 381         elsif labeled
 382             return labeled.color
 383         else
 384             return nil
 385         end
 386     end
 387
 388     def show_bg
 389         if outline_color.nil?
 390             button.modify_bg(Gtk::StateType::NORMAL, nil)
 391             button.modify_bg(Gtk::StateType::PRELIGHT, nil)
 392             button.modify_bg(Gtk::StateType::ACTIVE, nil)
 393         else
 394             button.modify_bg(Gtk::StateType::NORMAL, outline_color)
 395             button.modify_bg(Gtk::StateType::PRELIGHT, outline_color.lighter)
 396             button.modify_bg(Gtk::StateType::ACTIVE, outline_color)
 397         end
 398     end
 399
 400     def get_beautified_name
 401         if type == 'image'
 402             size = get_image_size(path)
 403             return _("%s (%sx%s, %s KB)") % [@guipath.gsub(/\.[^.]+$/, ''),
 404                                              size[:x],
 405                                              size[:y],
 406                                              commify(file_size(path)/1024)]
 407         else
 408             return _("%s (video - %s KB)") % [@guipath.gsub(/\.[^.]+$/, ''),
 409                                              commify(file_size(path)/1024)]
 410         end
 411     end
 412
 413     def cancel_loader
 414         if ! @loader.nil?
 415             #- avoid unneeded memory allocation
 416             @loader.signal_handler_disconnect(@area_prepared_cb)
 417             begin
 418                 @loader.close
 419             rescue
 420                 #- ignore loader errors, at that point they are fairly normal, we're canceling a partial load
 421             end
 422             @loader = nil
 423         end
 424     end
 425
 426     private
 427     def load_into_pixbuf_full
 428         if @pixbuf_full.nil?
 429             msg 3, ">>> load_into_pixbuf_full #{path}"
 430             @pixbuf_full = load_into_pixbuf_at_size { |w, h|
 431                 if @angle == 0
 432                     if w > @@max_width
 433                         #- save memory and speedup (+35%) loading
 434                         [ w * (factor = @@max_width.to_f/w), h * factor ]
 435                     else
 436                         [ w, h ]
 437                     end
 438                 else
 439                     if h > @@max_width
 440                         [ w * (factor = @@max_width.to_f/h), h * factor ]
 441                     else
 442                         [ w, h ]
 443                     end
 444                 end
 445             }
 446             show_pixbufs_present
 447         end
 448     end
 449
 450     def load_into_pixbuf_at_size(&specify_size)
 451         if @type == 'video'
 452             if @video_image_path.nil?
 453                 orig_base = File.basename(path)
 454                 tmpdir = gen_video_thumbnail(path, false, 0)
 455                 if tmpdir.nil?
 456                     return
 457                 end
 458                 @video_image_path = "#{tmpdir}/00000001.jpg"
 459             end
 460             image_path = @video_image_path
 461         else
 462             image_path = @path
 463         end
 464         if @angle.nil?
 465             if @type == 'image'
 466                 @angle = guess_rotate(image_path)
 467             else
 468                 @angle = 0
 469             end
 470         end
 471         begin
 472             #- use a pixbuf loader and check Gtk.events_pending? on each chunk, to keep the UI responsive even
 473             #- if loaded pictures are several MBs large
 474             if @loader.nil?
 475                 @loader = Gdk::PixbufLoader.new
 476                 @loader.signal_connect('size-prepared') { |l, w, h|
 477                     @loader.set_size(*specify_size.call(w, h))
 478                 }
 479                 @area_prepared_cb = @loader.signal_connect('area-prepared') { @loaded_pixbuf = @loader.pixbuf }
 480                 @loader_offset = 0
 481             end
 482             msg 3, "calling load_not_freezing_ui on #{image_path}, offset #{@loader_offset}"
 483             @loader_offset = @loader.load_not_freezing_ui(image_path, @loader_offset)
 484             if @loader_offset > 0
 485                 #- interrupted
 486                 raise InterruptedLoading
 487             end
 488             @loader = nil
 489             if @loaded_pixbuf.nil?
 490                 raise "Loaded pixbuf nil - #{path} #{image_path}"
 491             end
 492         rescue
 493             msg 0, "Cannot load #{image_path}: #{$!}"
 494             return
 495         ensure
 496             if @video_image_path && @loader.nil?
 497                 File.delete(@video_image_path)
 498                 Dir.rmdir(File.dirname(@video_image_path))
 499                 @video_image_path = nil
 500             end
 501         end
 502         if @loaded_pixbuf
 503             if @angle != 0
 504                 msg 3, ">>> load_into_pixbuf_full #{image_path} => rotate #{@angle}"
 505                 @loaded_pixbuf = rotate_pixbuf(@loaded_pixbuf, @angle)
 506             end
 507         end
 508         retval = @loaded_pixbuf
 509         @loaded_pixbuf = nil
 510         return retval
 511     end
 512
 513     def to_s
 514         @path
 515     end
 516 end
 517
 518 $allentries = []
 519
 520 def gc
 521     start = Time.now
 522     GC.start
 523     msg 3, "GC in #{Time.now - start} s"
 524 end
 525
 526 def free_cache_if_needed
 527     i = $allentries.index($mainview.get_shown_entry)
 528     return if i.nil?
 529     if get_mem > $config['cache-memory-use-figure']
 530         msg 3, "too much RSS, triggering GC"
 531         gc
 532     end
 533     if get_mem < $config['cache-memory-use-figure']
 534         return
 535     end
 536     msg 3, "too much RSS, freeing some cache"
 537     start = Time.now
 538     freed = 0
 539     ($allentries.size - 1).downto($config['preload-distance'].to_i + 1) { |j|
 540         index = i + j
 541         if i + j < $allentries.size
 542             $allentries[i + j].free_pixbuf_full
 543             if $allentries[i + j].free_pixbuf_main
 544                 freed += 1
 545             end
 546         end
 547         if i - j >= 0
 548             $allentries[i - j].free_pixbuf_full
 549             if $allentries[i - j].free_pixbuf_main
 550                 freed += 1
 551             end
 552         end
 553         if freed >= 10
 554             gc
 555             if get_mem < $config['cache-memory-use-figure'] * 3 / 4
 556                 msg 3, "RSS down enough - freeing done in #{Time.now - start} s"
 557                 show_pixbufs_present
 558                 return
 559             end
 560             freed = 0
 561         end
 562     }
 563     msg 3, "freeing done in #{Time.now - start} s"
 564     show_pixbufs_present
 565 end
 566
 567 def run_preloader_real
 568     msg 3, "*** >> main preloading triggered..."
 569     if $mainview.get_shown_entry
 570         free_cache_if_needed
 571         if $config['preload-distance'].to_i == 0
 572             return true
 573         end
 574         index = $allentries.index($mainview.get_shown_entry)
 575         index_right = index
 576         index_left = index
 577         loaded_right = 0
 578         loaded_left = 0
 579         right_done = false
 580         left_done = false
 581         loaded = []
 582         while ! right_done || ! left_done
 583             if ! right_done
 584                 index_right += 1
 585                 while index_right < $allentries.size && ! visible($allentries[index_right])
 586                     index_right += 1
 587                 end
 588                 if index_right == $allentries.size
 589                     right_done = true
 590                 else
 591                     if ! $allentries[index_right].pixbuf_main_present?
 592                         msg 3, "preloading #{$allentries[index_right].path}"
 593                         begin
 594                             $allentries[index_right].pixbuf_main
 595                         rescue InterruptedLoading
 596                             msg 3, "*** >>>> interrupted, rerun"
 597                             return false
 598                         end
 599                     end
 600                     loaded << index_right
 601                     loaded_right += 1
 602                     if loaded_right == $config['preload-distance'].to_i
 603                         right_done = true
 604                     end
 605                 end
 606             end
 607
 608             if ! left_done
 609                 index_left -= 1
 610                 while index_left >= 0 && ! visible($allentries[index_left])
 611                     index_left -= 1
 612                 end
 613                 if index_left == -1
 614                     left_done = true
 615                 else
 616                     if ! $allentries[index_left].pixbuf_main_present?
 617                         msg 3, "preloading #{$allentries[index_left].path}"
 618                         begin
 619                             $allentries[index_left].pixbuf_main
 620                         rescue InterruptedLoading
 621                             msg 3, "*** >>>> interrupted, rerun"
 622                             return false
 623                         end
 624                     end
 625                     loaded << index_left
 626                     loaded_left += 1
 627                     if loaded_left == $config['preload-distance'].to_i
 628                         left_done = true
 629                     end
 630                 end
 631             end
 632
 633             #- in case just loaded another directory
 634             if $preloader_force_exit
 635                 $preloader_force_exit = false
 636                 return true
 637             end
 638             #- in case moved fast
 639             if index != $allentries.index($mainview.get_shown_entry)
 640                 msg 3, "*** >>>> moved already, rerun"
 641                 return false
 642             end
 643         end
 644     end
 645     msg 3, "*** << main preloading finished"
 646     return true
 647 end
 648
 649 def run_preloader
 650     if ! $preloader_allowed
 651         msg 3, "*** preloader not yet allowed"
 652         return
 653     end
 654
 655     if $preloader_running
 656         msg 3, "preloader already running"
 657         return
 658     end
 659     msg 3, "run preloader"
 660     $preloader_running = true
 661     Gtk.idle_add {
 662         msg 3, "begin preloader from timeout "
 663         if run_preloader_real
 664             $preloader_running = false
 665             false
 666         else
 667             true
 668         end
 669     }
 670 end
 671
 672 class MainView < Gtk::DrawingArea
 673
 674     @@borders_thickness = 5
 675     @@borders_length = 25
 676     @@redraw_pending = nil
 677
 678     def MainView.borders_thickness
 679         return @@borders_thickness
 680     end
 681
 682     def MainView.get_usable_width(available_width)
 683         return available_width - ($videoborder_pixbuf.width + @@borders_thickness) * 2
 684     end
 685
 686     def MainView.get_usable_height(available_height)
 687         return available_height - @@borders_thickness * 2
 688     end
 689
 690     def initialize
 691         super()
 692         signal_connect('expose-event') { draw }
 693         signal_connect('configure-event') { update_shown }
 694     end
 695
 696     def try_show_entry(entry)
 697         if entry && entry.button
 698             if entry.button.has_focus?
 699                 redraw
 700             else
 701                 entry.button.grab_focus
 702             end
 703         end
 704     end
 705
 706     def set_shown_entry(entry)
 707         t1 = Time.now
 708         if entry && entry == @entry
 709             return
 710         end
 711         if entry && ! entry.button
 712             #- not loaded yet
 713             return
 714         end
 715         @entry = entry
 716         @entry and msg 3, "*** set entry to #{@entry.path}"
 717         redraw
 718         msg 3, "entry shown in: #{Time.now - t1} s"
 719     end
 720
 721     def get_shown_entry
 722         return @entry
 723     end
 724
 725     def show_next_entry(entry)
 726         index = $allentries.index(entry)
 727         if index < $allentries.size - 1
 728             index += 1
 729         end
 730         while index < $allentries.size - 1 && $allentries[index] && $allentries[index].button && ! $allentries[index].button.visible?
 731             index += 1
 732         end
 733         while $allentries[index] && $allentries[index].button && ! $allentries[index].button.visible? && index > 0
 734             index -= 1
 735         end
 736         if index < $allentries.size && $allentries[index] && $allentries[index].button && $allentries[index].button.visible?
 737             try_show_entry($allentries[index])
 738             return
 739         end
 740         #- find a fallback before
 741         while index < $allentries.size && index > 0 && $allentries[index] && (! $allentries[index].button || ! $allentries[index].button.visible?)
 742             index -= 1
 743         end
 744         if index < $allentries.size && index > 0 && $allentries[index] && $allentries[index].button && $allentries[index].button.visible?
 745             try_show_entry($allentries[index])
 746         end
 747     end
 748
 749     def redraw
 750         if @@redraw_pending
 751             msg 3, "redraw already pending"
 752             return
 753         end
 754         msg 3, "redraw"
 755         @@redraw_pending = Gtk.idle_add {
 756             msg 3, "begin redraw from timeout "
 757             begin
 758                 msg 3, "try redraw from timeout"
 759                 redraw_real
 760                 @@redraw_pending = nil
 761                 run_preloader
 762                 false
 763             rescue InterruptedLoading
 764                 msg 3, "interrupted, will retry"
 765                 true
 766             end
 767         }
 768     end
 769
 770     def redraw_real
 771         @entry and sb_msg(_("Selected %s") % @entry.get_beautified_name)
 772         if ! update_shown
 773             return
 774         end
 775         w, h = window.size
 776         window.begin_paint(Gdk::Rectangle.new(0, 0, w, h))
 777         window.clear
 778         draw
 779         window.end_paint
 780     end
 781
 782     def update_shown
 783         if @entry
 784             msg 3, "################################################ trying to show #{@entry.path}"
 785             pixbuf = @entry.pixbuf_main
 786             if pixbuf
 787                 @pixbuf = pixbuf
 788                 width, height = window.size
 789                 @xpos = (width - @pixbuf.width)/2
 790                 @ypos = (height - @pixbuf.height)/2
 791                 return true
 792             else
 793                 return false
 794             end
 795         else
 796             @pixbuf = nil
 797             return true
 798         end
 799     end
 800
 801     def draw
 802         if @pixbuf
 803             window.draw_pixbuf(nil, @pixbuf, 0, 0, @xpos, @ypos, -1, -1, Gdk::RGB::DITHER_NONE, -1, -1)
 804             if @entry && @entry.type == 'video'
 805                 window.draw_borders($videoborder_pixbuf, @xpos - $videoborder_pixbuf.width, @xpos + @pixbuf.width, @ypos, @ypos + @pixbuf.height)
 806             end
 807             if @entry && ! @entry.outline_color.nil?
 808                 gc = Gdk::GC.new(window)
 809                 colormap.alloc_color(@entry.outline_color, false, true)
 810                 gc.set_foreground(@entry.outline_color)
 811                 if @entry.type == 'video'
 812                     xleft = @xpos - $videoborder_pixbuf.width
 813                     xright = @xpos + @pixbuf.width + $videoborder_pixbuf.width
 814                 else
 815                     xleft = @xpos
 816                     xright = @xpos + @pixbuf.width
 817                 end
 818                 window.draw_polygon(gc, true, [[xleft - @@borders_thickness, @ypos - @@borders_thickness],
 819                                                [xright + @@borders_thickness, @ypos - @@borders_thickness],
 820                                                [xright + @@borders_thickness, @ypos + @pixbuf.height + @@borders_thickness],
 821                                                [xleft - @@borders_thickness, @ypos + @pixbuf.height + @@borders_thickness],
 822                                                [xleft - @@borders_thickness, @ypos - 1],
 823                                                [xleft - 1, @ypos - 1],
 824                                                [xleft - 1, @ypos + @pixbuf.height + 1],
 825                                                [xright + 1, @ypos + @pixbuf.height + 1],
 826                                                [xright + 1, @ypos - 1],
 827                                                [xleft - @@borders_thickness, @ypos - 1]])
 828             end
 829         end
 830     end
 831 end
 832
 833 def autoscroll_if_needed(button, center)
 834     xpos_left = button.allocation.x
 835     xpos_right = button.allocation.x + button.allocation.width
 836     hadj = $imagesline_sw.hadjustment
 837     current_minx_visible = hadj.value
 838     current_maxx_visible = hadj.value + hadj.page_size
 839     if ! center
 840         if xpos_left < current_minx_visible
 841             #- autoscroll left
 842             newval = hadj.value - (current_minx_visible - xpos_left)
 843             hadj.value = newval
 844         elsif xpos_right > current_maxx_visible
 845             #- autoscroll right
 846             newval = hadj.value + (xpos_right - current_maxx_visible)
 847             if newval > hadj.upper - hadj.page_size
 848                 newval = hadj.upper - hadj.page_size
 849             end
 850             hadj.value = newval
 851         end
 852     else
 853         hadj.value = clamp((xpos_left + xpos_right) / 2 - hadj.page_size / 2, 0, hadj.upper - hadj.page_size)
 854     end
 855 end
 856
 857 def show_popup(parent, msg, *options)
 858     dialog = Gtk::Dialog.new
 859     if options[0]
 860         options = options[0]
 861     else
 862         options = {}
 863     end
 864     if options[:title]
 865         dialog.title = options[:title]
 866     else
 867         dialog.title = utf8(_("Booh message"))
 868     end
 869     lbl = Gtk::Label.new
 870     if options[:nomarkup]
 871         lbl.text = msg
 872     else
 873         lbl.markup = msg
 874     end
 875     if options[:centered]
 876         lbl.set_justify(Gtk::Justification::CENTER)
 877     end
 878     if options[:selectable]
 879         lbl.selectable = true
 880     end
 881     if options[:scrolled]
 882         sw = Gtk::ScrolledWindow.new(nil, nil)
 883         sw.set_policy(Gtk::POLICY_AUTOMATIC, Gtk::POLICY_AUTOMATIC)
 884         sw.add_with_viewport(lbl)
 885         dialog.vbox.add(sw)
 886         dialog.set_default_size(500, 600)
 887     else
 888         dialog.vbox.add(lbl)
 889         dialog.set_default_size(200, 120)
 890     end
 891     if options[:bottomwidget]
 892         dialog.vbox.add(options[:bottomwidget])
 893     end
 894     if options[:okcancel]
 895         cancel = dialog.add_button(Gtk::Stock::CANCEL, Gtk::Dialog::RESPONSE_CANCEL)
 896         if ! options[:bottomwidget]
 897             cancel.grab_focus
 898         end
 899         dialog.add_button(Gtk::Stock::OK, Gtk::Dialog::RESPONSE_OK)
 900     else
 901         ok = dialog.add_button(Gtk::Stock::OK, Gtk::Dialog::RESPONSE_OK).grab_focus
 902         if ! options[:bottomwidget]
 903             ok.grab_focus
 904         end
 905     end
 906
 907     if options[:pos_centered]
 908         dialog.window_position = Gtk::Window::POS_CENTER
 909     else
 910         dialog.window_position = Gtk::Window::POS_MOUSE
 911     end
 912
 913     if options[:linkurl]
 914         linkbut = Gtk::Button.new('')
 915         linkbut.child.markup = "<span foreground=\"#00000000FFFF\" underline=\"single\">#{options[0][:linkurl]}</span>"
 916         linkbut.signal_connect('clicked') {
 917             open_url(options[0][:linkurl] + '/index.html')
 918             dialog.response(Gtk::Dialog::RESPONSE_OK)
 919             set_mousecursor_normal
 920         }
 921         linkbut.relief = Gtk::RELIEF_NONE
 922         linkbut.signal_connect('enter-notify-event') { set_mousecursor(Gdk::Cursor::HAND2, linkbut); false }
 923         linkbut.signal_connect('leave-notify-event') { set_mousecursor(nil, linkbut); false }
 924         dialog.vbox.add(Gtk::Alignment.new(0.5, 0.5, 0, 0).add(linkbut))
 925     end
 926
 927     dialog.show_all
 928
 929     if options[:stuff_connector]
 930         options[:stuff_connector].call({ :dialog => dialog })
 931     end
 932
 933     if !options[:not_transient]
 934         dialog.transient_for = parent
 935         dialog.run { |response|
 936             if options[:data_getter]
 937                 options[:data_getter].call
 938             end
 939             dialog.destroy
 940             if options[:okcancel]
 941                 return response == Gtk::Dialog::RESPONSE_OK
 942             end
 943         }
 944     else
 945         dialog.signal_connect('response') { dialog.destroy }
 946     end
 947 end
 948
 949 def view_entry(entry)
 950     if entry.type == 'image'
 951         show_popup($main_window,
 952                    utf8(`exif -m '#{entry.path}'`),
 953                    { :title => utf8(_("EXIF data of %s") % File.basename(entry.path)), :nomarkup => true, :scrolled => true, :not_transient => true })
 954     else
 955         cmd = from_utf8($config['video-viewer']).gsub('%f', "'#{entry.path}'") + ' &'
 956         msg 2, cmd
 957         system(cmd)
 958     end
 959 end
 960
 961 def update_counters
 962     value = 0
 963     $allentries.each { |entry|
 964         if ! entry.removed && entry.labeled.nil?
 965             value += 1
 966         end
 967     }
 968     $unlabelled_counter.set_markup('<tt>' + value.to_s + '</tt>')
 969     value = 0
 970     $allentries.each { |entry|
 971         if entry.removed
 972             value += 1
 973         end
 974     }
 975     $toremove_counter.set_markup('<tt>' + value.to_s + '</tt>')
 976     $labels.values.each { |label|
 977         value = 0
 978         $allentries.each { |entry|
 979             if entry.labeled == label
 980                 value += 1
 981             end
 982         }
 983         label.counter.set_markup('<tt>' + value.to_s + '</tt>')
 984     }
 985 end
 986
 987 def thumbnail_keypressed(entry, event)
 988     if event.state & Gdk::Window::MOD1_MASK != 0
 989         #- ALT pressed: Alt-Left and Alft-Right rotate
 990         if event.keyval == Gdk::Keyval::GDK_Left || event.keyval == Gdk::Keyval::GDK_Right
 991             if event.keyval == Gdk::Keyval::GDK_Left
 992                 entry.angle = (entry.angle - 90) % 360
 993             else
 994                 entry.angle = (entry.angle + 90) % 360
 995             end
 996             entry.free_pixbuf_full
 997             entry.free_pixbuf_main
 998             entry.free_pixbuf_thumbnail
 999             show_pixbufs_present
1000             $mainview.redraw
1001             entry.image.pixbuf = entry.pixbuf_thumbnail
1002             if $config['rotate-set-exif'] == 'true' && entry.type == 'image'
1003                 Exif.set_orientation(entry.path, angle_to_exif_orientation(entry.angle))
1004             end
1005         end
1006
1007     elsif event.state & Gdk::Window::CONTROL_MASK != 0
1008         #- CONTROL pressed: Ctrl-z and Ctrl-r for undo/redo, Ctrl-space for recentre
1009         if event.keyval == Gdk::Keyval::GDK_z
1010             perform_undo
1011         end
1012         if event.keyval == Gdk::Keyval::GDK_r
1013             perform_redo
1014         end
1015         if event.keyval == Gdk::Keyval::GDK_space
1016             shown = $mainview.get_shown_entry
1017             shown and autoscroll_if_needed(shown.button, true)
1018         end
1019
1020     else
1021         removed_before = entry.removed
1022         label_before = entry.labeled
1023
1024         if event.keyval == Gdk::Keyval::GDK_Delete
1025             if ! FileTest.writable?(entry.path)
1026                 show_popup($main_window, utf8(_("Notice: no write access to '%s', permission will be denied at execute step.") % entry.path))
1027             end
1028             entry.removed = true
1029             entry.labeled = nil
1030             entry.show_bg
1031             update_visibility(entry)
1032             update_counters
1033             $mainview.show_next_entry(entry)
1034
1035             save_undo(_("set for removal"),
1036                       proc {
1037                           entry.removed = removed_before
1038                           entry.labeled = label_before
1039                           entry.show_bg
1040                           update_visibility(entry)
1041                           update_counters
1042                           if entry.button.visible?
1043                               $mainview.try_show_entry(entry)
1044                           end
1045                           proc {
1046                               entry.removed = true
1047                               entry.labeled = nil
1048                               entry.show_bg
1049                               update_visibility(entry)
1050                               update_counters
1051                               if entry.button.visible?
1052                                   $mainview.try_show_entry(entry)
1053                               end
1054                           }
1055                       })
1056
1057         elsif event.keyval == Gdk::Keyval::GDK_space
1058             if entry.labeled
1059                 msg = _("Cleared label")
1060             elsif entry.removed
1061                 msg = _("Cleared set for removal")
1062             end
1063             entry.removed = false
1064             entry.labeled = nil
1065             entry.show_bg
1066             update_counters
1067             $mainview.show_next_entry(entry)
1068
1069             save_undo(msg,
1070                       proc {
1071                           entry.removed = removed_before
1072                           entry.labeled = label_before
1073                           entry.show_bg
1074                           update_counters
1075                           $mainview.try_show_entry(entry)
1076                           proc {
1077                               entry.removed = false
1078                               entry.labeled = nil
1079                               entry.show_bg
1080                               update_counters
1081                               $mainview.try_show_entry(entry)
1082                           }
1083                       })
1084
1085         elsif event.keyval == Gdk::Keyval::GDK_Return
1086             view_entry(entry)
1087
1088         elsif event.keyval == Gdk::Keyval::GDK_Home
1089             index = 0
1090             while $allentries[index] && $allentries[index].button && !visible($allentries[index])
1091                 index += 1
1092             end
1093             if $allentries[index] && $allentries[index].button
1094                 $allentries[index].button.grab_focus
1095             end
1096
1097         elsif event.keyval == Gdk::Keyval::GDK_End
1098             index = $allentries.size - 1
1099             while $allentries[index] && ! $allentries[index].button
1100                 #- not yet loaded
1101                 index -= 1
1102             end
1103             while $allentries[index] && $allentries[index].button && !visible($allentries[index])
1104                 index -= 1
1105             end
1106             if $allentries[index] && $allentries[index].button
1107                 $allentries[index].button.grab_focus
1108             end
1109
1110         else
1111             char = [ Gdk::Keyval.to_unicode(event.keyval) ].pack("C*")
1112             if char =~ /^[a-zA-z0-9]$/
1113                 label = $labels[char]
1114
1115                 if label.nil?
1116                     vb = Gtk::VBox.new(false, 0)
1117                     vb.pack_start(labelentry = Gtk::Entry.new.set_text(char), false, false)
1118                     vb.pack_start(Gtk::Alignment.new(0.5, 0.5, 0, 0).add(bt = Gtk::ColorButton.new))
1119                     color = bt.color = Gdk::Color.new(16384 + rand(49151), 16384 + rand(49151), 16384 + rand(49151))
1120                     bt.signal_connect('color-set') { color = bt.color }
1121                     text = nil
1122                     labelentry.signal_connect('changed') {  #- cannot add a new label with first letter of an existing label
1123                         while $labels.has_key?(labelentry.text[0,1])
1124                             labelentry.text = labelentry.text.sub(/./, '')
1125                         end
1126                     }
1127                     if show_popup($main_window,
1128                                   utf8(_("You typed the text character '%s', which is not associated with a label.\nType in the full name of the label below to create a new one.")) % char,
1129                                   { :okcancel => true, :bottomwidget => vb, :data_getter => proc { text = labelentry.text },
1130                                     :stuff_connector => proc { |stuff| labelentry.select_region(0, 0)
1131                                                                        labelentry.position = -1
1132                                                                        labelentry.signal_connect('activate') { stuff[:dialog].response(Gtk::Dialog::RESPONSE_OK) } } } )
1133                         if text.length > 0
1134                             char = text[0,1]  #- in case it changed
1135                             label = Label.new(text)
1136                             label.color = color
1137                             $labels[char] = label
1138                             $ordered_labels << label
1139                             lbl = Gtk::Label.new.set_markup('<b>(' + char + ')</b>' + text[1..-1]).set_justify(Gtk::Justification::CENTER)
1140                             $labels_vbox.pack_start(Gtk::HBox.new(false, 5).pack_start(label.button = Gtk::CheckButton.new.add(evt = Gtk::EventBox.new.add(lbl))).
1141                                                                             pack_start(Gtk::Label.new, true, true).
1142                                                                             pack_start(label.counter = Gtk::Label.new.set_markup('<tt>0</tt>'), false, false).show_all)
1143                             label.button.active = true
1144                             label.button.signal_connect('toggled') { update_all_visibilities }
1145                             evt.modify_bg(Gtk::StateType::NORMAL, label.color)
1146                             evt.modify_bg(Gtk::StateType::PRELIGHT, label.color.lighter.lighter)
1147                             evt.modify_bg(Gtk::StateType::ACTIVE, label.color.lighter)
1148                         end
1149                     end
1150                 end
1151
1152                 if label
1153                     entry.removed = false
1154                     entry.labeled = label
1155                     entry.show_bg
1156                     update_visibility(entry)
1157                     update_counters
1158                     $mainview.show_next_entry(entry)
1159
1160                     save_undo(_("set label"),
1161                               proc {
1162                                   entry.removed = removed_before
1163                                   entry.labeled = label_before
1164                                   entry.show_bg
1165                                   update_visibility(entry)
1166                                   update_counters
1167                                   if entry.button.visible?
1168                                       $mainview.try_show_entry(entry)
1169                                   end
1170                                   proc {
1171                                       entry.removed = false
1172                                       entry.labeled = label
1173                                       entry.show_bg
1174                                       update_visibility(entry)
1175                                       update_counters
1176                                       if entry.button.visible?
1177                                           $mainview.try_show_entry(entry)
1178                                       end
1179                                   }
1180                               })
1181                 end
1182             end
1183         end
1184     end
1185 end
1186
1187 def sb_msg(msg)
1188     $statusbar.pop(0)
1189     if msg
1190         $statusbar.push(0, utf8(msg))
1191     end
1192 end
1193
1194 def show_entry(entry, i, tips)
1195     #- scope entry
1196     #msg 3, "showing entry #{entry}"
1197     entry.image = Gtk::Image.new(entry.pixbuf_thumbnail)
1198     if entry.type == 'video'
1199         entry.button = Gtk::Button.new.add(Gtk::HBox.new.pack_start(da1 = Gtk::DrawingArea.new.set_size_request($videoborder_pixbuf.width, -1), false, false).
1200                                            pack_start(entry.image).
1201                                            pack_start(da2 = Gtk::DrawingArea.new.set_size_request($videoborder_pixbuf.width, -1), false, false))
1202         da1.signal_connect('realize') { da1.window.set_back_pixmap($videoborder_pixmap, false) }
1203         da2.signal_connect('realize') { da2.window.set_back_pixmap($videoborder_pixmap, false) }
1204     else
1205         entry.button = Gtk::Button.new.add(entry.image)
1206     end
1207     tips.set_tip(entry.button, entry.get_beautified_name, nil)
1208     $imagesline.pack_start(entry.alignment = Gtk::Alignment.new(0.5, 1, 0, 0).add(entry.button).show_all, false, false)
1209     entry.button.signal_connect('clicked') {
1210         shown = $mainview.get_shown_entry
1211         if shown != entry
1212             shown and shown.alignment.set(0.5, 1, 0, 0)
1213             entry.alignment.set(0.5, 0, 0, 0)
1214             autoscroll_if_needed(entry.button, false)
1215             $mainview.set_shown_entry(entry)
1216         end
1217     }
1218     entry.button.signal_connect('button-press-event') { |w, event|
1219         if entry.type == 'video' && event.event_type == Gdk::Event::BUTTON2_PRESS
1220             video_view(entry)
1221         end
1222     }
1223     entry.button.signal_connect('focus-in-event') { entry.button.clicked }
1224     entry.button.signal_connect('key-press-event') { |w, e| thumbnail_keypressed(entry, e) }
1225     if i == 0
1226         entry.button.grab_focus
1227     end
1228     update_visibility(entry)
1229 end
1230
1231 def show_entries(allentries)
1232     update_counters
1233     sb_msg(_("Loading images..."))
1234     $loading_progressbar.fraction = 0
1235     $loading_progressbar.text = utf8(_("Loading... %d%") % 0)
1236     $loading_progressbar.show
1237     t1 = Time.now
1238     total_loaded_files = 0
1239     total_loaded_size = 0
1240     i = 0
1241     tips = Gtk::Tooltips.new
1242     while i < allentries.size
1243         begin
1244             entry = allentries[i]
1245             if i == 0
1246                 loaded_pixbuf = entry.pixbuf_main
1247             else
1248                 loaded_pixbuf = entry.pixbuf_thumbnail
1249             end
1250         rescue InterruptedLoading
1251             redo
1252         end
1253
1254         if $allentries != allentries
1255             #- loaded another directory while this one was not yet finished
1256             msg 3, "allentries differ, stopping this deprecated load"
1257             return
1258         end
1259
1260         if loaded_pixbuf
1261             show_entry(entry, i, tips)
1262             if $allentries != allentries
1263                 #- loaded another directory while this one was not yet finished
1264                 msg 3, "allentries differ, stopping this deprecated load"
1265                 return
1266             end
1267
1268             total_loaded_size += file_size(entry.path)
1269             total_loaded_files += 1
1270             i += 1
1271             if i > $config['preload-distance'].to_i && i <= $config['preload-distance'].to_i * 2
1272                 #- when we're at preload distance, begin preloading to preload distance
1273                 begin
1274                     allentries[i - $config['preload-distance'].to_i].pixbuf_main
1275                 rescue InterruptedLoading
1276                 end
1277             end
1278             if i == $config['preload-distance'].to_i * 2 + 1
1279                 #- when we're after double preload distance, activate normal preloading
1280                 $preloader_allowed = true
1281             end
1282
1283         else
1284             allentries.delete_at(i)
1285         end
1286         $loading_progressbar.fraction = i.to_f / allentries.size
1287         $loading_progressbar.text = utf8(_("Loading... %d%") % (100 * $loading_progressbar.fraction))
1288         if $quit
1289             return
1290         end
1291         if i % 25 == 0
1292             gc
1293         end
1294     end
1295     $preloader_allowed = true
1296     if i <= $config['preload-distance'].to_i * 2
1297         #- not yet preloaded correctly
1298         run_preloader
1299     end
1300     sb_msg(_("%d images of total %s kB loaded in %3.2f seconds.") % [ total_loaded_files, commify(total_loaded_size / 1024), Time.now - t1 ])
1301     $loading_progressbar.hide
1302     $execute.sensitive = true
1303 end
1304
1305 def reset_all
1306     reset_labels
1307     reset_thumbnails
1308     $mainview.set_shown_entry(nil)
1309     sb_msg(nil)
1310     $preloader_allowed = false
1311     $execute.sensitive = false
1312 end
1313
1314 def open_dir(*paths)
1315     #- remove visual stuff, so that user will see something is happening
1316     reset_all
1317     sb_msg(_("Scanning source directory... %s") % "")
1318     Gtk.main_iteration while Gtk.events_pending?
1319
1320     for path in paths
1321         path = File.expand_path(path.sub(%r|/$|, ''))
1322         $workingdir = path
1323         entries = []
1324         if File.directory?(path)
1325             examined_dirs = `find '#{path}' -type d -follow`.split.sort.collect { |v| v.chomp }
1326             #- validate first
1327             examined_dirs.each { |dir|
1328                 if dir =~ /'/
1329                     return utf8(_("Source directory or sub-directories can't contain a single-quote character, sorry: %s") % dir)
1330                 end
1331                 begin
1332                     Dir.entries(dir).each { |file|
1333                         if file =~ /'/ && type = entry2type(file) && type == 'video'
1334                             return utf8(_("Videos can't contain a single quote character ('), sorry: %s") % "#{dir}/#{file}")
1335                         end
1336                     }
1337                 rescue
1338                     puts "Failed to open directory #{dir}: #{$!}"
1339                 end
1340             }
1341
1342             #- scan for populate second
1343             examined_dirs.each { |dir|
1344                 if File.basename(dir) =~ /^\./
1345                     msg 1, _("Ignoring directory %s, begins with a dot (indicating a hidden directory)") % dir
1346                     next
1347                 end
1348                 begin
1349                     entries += Dir.entries(dir).collect { |file| File.join(dir, file) }
1350                 rescue
1351                     #- already puts'ed 10 lines upper
1352                 end
1353                 sb_msg(_("Scanning source directory... %s") % (_("%d entries found") % entries.size))
1354                 Gtk.main_iteration while Gtk.events_pending?
1355             }
1356
1357         else
1358             entries << path
1359         end
1360
1361         if $sort_by_exif_date
1362             dates = {}
1363             entries.each { |file|
1364                 date_time = Exif.datetimeoriginal(file)
1365                 if ! date_time.nil?
1366                     dates[file] = date_time
1367                 end
1368             }
1369             entries = smartsort(entries, dates)
1370         else
1371             entries.sort!
1372         end
1373         entries.each { |file|
1374             type = entry2type(file)
1375             if type
1376                 if File.directory?(path)
1377                     $allentries << Entry.new(file, type, file[path.length + 1 .. -1])
1378                 else
1379                     $allentries << Entry.new(file, type, file)
1380                 end
1381             end
1382         }
1383     end
1384     return nil
1385 end
1386
1387 def open_dir_popup
1388     fc = Gtk::FileChooserDialog.new(utf8(_("Specify the directory to work with")),
1389                                     nil,
1390                                     Gtk::FileChooser::ACTION_SELECT_FOLDER,
1391                                     nil,
1392                                     [Gtk::Stock::OPEN, Gtk::Dialog::RESPONSE_ACCEPT], [Gtk::Stock::CANCEL, Gtk::Dialog::RESPONSE_CANCEL])
1393     fc.transient_for = $main_window
1394     if $workingdir
1395         fc.current_folder = $workingdir
1396     end
1397     ok = false
1398     load = false
1399     while !ok
1400         if fc.run == Gtk::Dialog::RESPONSE_ACCEPT
1401             msg = open_dir(fc.filename)
1402             if msg
1403                 show_popup(fc, msg)
1404                 ok = false
1405             else
1406                 ok = true
1407                 load = true
1408             end
1409         else
1410             ok = true
1411         end
1412     end
1413     fc.destroy
1414     if load
1415         show_entries($allentries)
1416     end
1417 end
1418
1419 def try_quit(*options)
1420     if ! $allentries.detect { |e| e.removed || e.labeled } || show_popup($main_window,
1421                                                                          utf8(_("You have not executed the classification. Are you sure you want to quit?")),
1422                                                                          { :okcancel => true })
1423         Gtk.main_quit
1424         $quit = true
1425         return false
1426     else
1427         return true
1428     end
1429 end
1430
1431 def execute
1432     dialog = Gtk::Dialog.new
1433     dialog.title = utf8(_("Booh message"))
1434
1435     vb1 = Gtk::VBox.new(false, 5)
1436     label = Gtk::Label.new.set_markup(utf8(_("You're about to <b>execute</b> actions on the marked images.\nPlease confirm below the actions. You cannot undo this operation!")))
1437     vb1.pack_start(label, false, false)
1438
1439     lastpath = $workingdir
1440
1441     table = Gtk::Table.new(0, 0, false)
1442     table.set_row_spacings(5)
1443     table.set_column_spacings(5)
1444     table.attach(Gtk::Label.new.set_markup(utf8(_("<b>Label name:</b>"))).set_justify(Gtk::Justification::CENTER), 0, 1, 0, 1, Gtk::FILL, Gtk::FILL, 5, 0)
1445     table.attach(Gtk::Label.new.set_markup(utf8(_("<b>Amount of pictures:</b>"))).set_justify(Gtk::Justification::CENTER), 1, 2, 0, 1, Gtk::FILL, Gtk::FILL, 5, 0)
1446     table.attach(Gtk::Label.new.set_markup(utf8(_("<b>Pictures examples:</b>"))).set_justify(Gtk::Justification::CENTER), 2, 3, 0, 1, Gtk::FILL, Gtk::FILL, 5, 0)
1447     table.attach(Gtk::Label.new.set_markup(utf8(_("<b>Action to perform:</b>"))).set_justify(Gtk::Justification::CENTER), 3, 4, 0, 1, Gtk::FILL, Gtk::FILL, 5, 0)
1448     add_row = proc { |row, name, color, truthproc, normal|
1449         table.attach(Gtk::Alignment.new(0, 0.5, 1, 0).add(Gtk::EventBox.new.add(Gtk::Label.new.set_markup(name)).modify_bg(Gtk::StateType::NORMAL, color)),
1450                      0, 1, row, row + 1, Gtk::FILL, Gtk::FILL, 5, 5)
1451         counter = 0
1452         examples = Gtk::HBox.new(false, 5)
1453         $allentries.each { |entry|
1454             if truthproc.call(entry)
1455                 counter += 1
1456                 if counter < 4
1457                     thumbnail = Gtk::Image.new(entry.pixbuf_thumbnail)
1458                     if entry.type == 'video'
1459                         thumbnail = Gtk::HBox.new.pack_start(da1 = Gtk::DrawingArea.new.set_size_request($videoborder_pixbuf.width, -1), false, false).
1460                                                   pack_start(thumbnail).
1461                                                   pack_start(da2 = Gtk::DrawingArea.new.set_size_request($videoborder_pixbuf.width, -1), false, false)
1462                         da1.signal_connect('realize') { da1.window.set_back_pixmap($videoborder_pixmap, false) }
1463                         da2.signal_connect('realize') { da2.window.set_back_pixmap($videoborder_pixmap, false) }
1464                     end
1465                     examples.pack_start(thumbnail, false, false)
1466                 elsif counter == 4
1467                     examples.pack_start(Gtk::Label.new.set_markup("<b>...</b>"), false, false)
1468                 end
1469             end
1470         }
1471         table.attach(Gtk::Label.new(counter.to_s).set_justify(Gtk::Justification::CENTER), 1, 2, row, row + 1, 0, 0, 5, 5)
1472         table.attach(examples, 2, 3, row, row + 1, Gtk::FILL, Gtk::FILL, 5, 5)
1473
1474         if counter == 0
1475             return {}
1476         end
1477
1478         combostore = Gtk::ListStore.new(Gdk::Pixbuf, String)
1479         iter = combostore.append
1480         if normal
1481             iter[0] = $main_window.render_icon(Gtk::Stock::PASTE, Gtk::IconSize::MENU)
1482             iter[1] = utf8(_("Copy to:"))
1483             iter = combostore.append
1484             iter[0] = $main_window.render_icon(Gtk::Stock::GO_FORWARD, Gtk::IconSize::MENU)
1485             iter[1] = utf8(_("Move to:"))
1486         else
1487             iter[0] = $main_window.render_icon(Gtk::Stock::DELETE, Gtk::IconSize::MENU)
1488             iter[1] = utf8(_("Permanently remove"))
1489         end
1490         iter = combostore.append
1491         iter[0] = $main_window.render_icon(Gtk::Stock::MEDIA_STOP, Gtk::IconSize::MENU)
1492         iter[1] = utf8(_("Do nothing"))
1493         combo = Gtk::ComboBox.new(combostore)
1494         combo.active = 0
1495         renderer = Gtk::CellRendererPixbuf.new
1496         combo.pack_start(renderer, false)
1497         combo.set_attributes(renderer, :pixbuf => 0)
1498         renderer = Gtk::CellRendererText.new
1499         combo.pack_start(renderer, true)
1500         combo.set_attributes(renderer, :text => 1)
1501
1502         if normal
1503             pathbutton = Gtk::Button.new.add(pathlabel = Gtk::Label.new.set_markup(utf8(_("<i>(unset)</i>"))))
1504             pathbutton.signal_connect('clicked') {
1505                 fc = Gtk::FileChooserDialog.new(utf8(_("Specify the directory where to move the pictures to")),
1506                                                 nil,
1507                                                 Gtk::FileChooser::ACTION_SELECT_FOLDER,
1508                                                 nil,
1509                                                 [Gtk::Stock::OPEN, Gtk::Dialog::RESPONSE_ACCEPT], [Gtk::Stock::CANCEL, Gtk::Dialog::RESPONSE_CANCEL])
1510                 fc.transient_for = dialog
1511                 if lastpath
1512                     fc.current_folder = lastpath
1513                 end
1514                 if fc.run == Gtk::Dialog::RESPONSE_ACCEPT
1515                     pathlabel.text = fc.filename
1516                     pathlabel.set_alignment(0, 0.5)
1517                 end
1518                 lastpath = fc.filename
1519                 fc.destroy
1520             }
1521             combo.signal_connect('changed') {
1522                 pathbutton.sensitive = combo.active <= 1
1523             }
1524             vb = Gtk::VBox.new(false, 5)
1525             vb.pack_start(combo, false, false)
1526             vb.pack_start(pathbutton, false, false)
1527             table.attach(Gtk::Alignment.new(0, 0.5, 1, 0).add(vb), 3, 4, row, row + 1, Gtk::FILL, Gtk::FILL, 5, 5)
1528             { :combo => combo, :pathlabel => pathlabel }
1529         else
1530             table.attach(Gtk::Alignment.new(0, 0.5, 1, 0).add(combo), 3, 4, row, row + 1, Gtk::FILL, Gtk::FILL, 5, 5)
1531             { :combo => combo }
1532         end
1533     }
1534     stuff = {}
1535     stuff['toremove'] = add_row.call(1, utf8(_("<i>to remove</i>")), $color_red, proc { |entry| entry.removed }, false)
1536     $ordered_labels.each_with_index { |label, row| stuff[label] = add_row.call(row + 2, label.name, label.color, proc { |entry| entry.labeled == label }, true) }
1537     vb1.pack_start(sw = Gtk::ScrolledWindow.new(nil, nil).add_with_viewport(table).set_policy(Gtk::POLICY_AUTOMATIC, Gtk::POLICY_AUTOMATIC), true, true)
1538
1539     toremove_amount = $allentries.find_all { |entry| entry.removed }.size
1540     toremove_size = commify($allentries.find_all { |entry| entry.removed }.collect { |entry| file_size(entry.path) }.sum / 1024)
1541     check_removal = Gtk::CheckButton.new(utf8(_("I have noticed I am about to permanently remove the %d above mentioned pictures (total %s kB).") % [ toremove_amount, toremove_size ]))
1542     if toremove_amount > 0
1543         vb1.pack_start(check_removal, false, false)
1544         stuff['toremove'][:combo].signal_connect('changed') { |widget|
1545             check_removal.sensitive = widget.active == 0
1546         }
1547     end
1548
1549     dialog.vbox.add(vb1)
1550
1551     dialog.set_default_size(800, 600)
1552     dialog.add_button(Gtk::Stock::CANCEL, Gtk::Dialog::RESPONSE_CANCEL)
1553     dialog.add_button(Gtk::Stock::OK, Gtk::Dialog::RESPONSE_OK)
1554     dialog.window_position = Gtk::Window::POS_MOUSE
1555     dialog.transient_for = $main_window
1556
1557     dialog.show_all
1558
1559     while true
1560         dialog.run { |response|
1561             if response == Gtk::Dialog::RESPONSE_OK
1562                 problem = false
1563                 if toremove_amount > 0 && stuff['toremove'][:combo].active == 0
1564                     if ! check_removal.active?
1565                         show_popup(dialog, utf8(_("You have not confirmed that you noticed the permanent removal of the pictures marked for deletion.")))
1566                         problem = true
1567                         break
1568                     end
1569                     $allentries.each { |entry|
1570                         if entry.removed
1571                             if ! FileTest.writable?(entry.path)
1572                                 show_popup(dialog, utf8(_("Sorry, permission denied to remove '%s'.") % [ entry.path ]))
1573                                 problem = true
1574                                 break
1575                             end
1576                         end
1577                     }
1578                 end
1579                 label2entries = {}
1580                 $labels.values.each { |label| label2entries[label] = [] }
1581                 $allentries.each { |entry| entry.labeled and label2entries[entry.labeled] << entry }
1582                 stuff.keys.each { |key|
1583                     if key.is_a?(Label) && stuff[key][:combo] && stuff[key][:combo].active <= 1
1584                         destination = stuff[key][:pathlabel].text
1585                         if destination[0] != ?/
1586                             show_popup(dialog, utf8(_("You have not selected a directory where to move/copy %s.") % key.name))
1587                             problem = true
1588                             break
1589                         end
1590                         begin
1591                             Dir.mkdir(destination)
1592                         rescue Errno::EEXIST
1593                         end
1594                         begin
1595                             st = File.stat(destination)
1596                         rescue
1597                             show_popup(dialog, utf8(_("Directory %s, where to move/copy %s, is not valid or not createable.") % [destination, key.name]))
1598                             problem = true
1599                             break
1600                         end
1601                         if ! st.directory? || ! st.writable?
1602                             show_popup(dialog, utf8(_("Directory %s, where to move/copy %s, is not valid or not writable.") % [destination, key.name]))
1603                             problem = true
1604                             break
1605                         end
1606                         label2entries[key].each { |entry|
1607                             begin
1608                                 File.stat(File.join(destination, File.basename(entry.path)))
1609                                 show_popup(dialog, utf8(_("Sorry, a file '%s' already exists in directory '%s'.") % [ File.basename(entry.path), destination ]))
1610                                 problem = true
1611                                 break
1612                             rescue
1613                             end
1614                         }
1615                         if stuff[key][:combo].active == 1
1616                             label2entries[key].each { |entry|
1617                                 if ! FileTest.writable?(entry.path)
1618                                     show_popup(dialog, utf8(_("Sorry, permission denied to move '%s'.") % [ entry.path ]))
1619                                     problem = true
1620                                     break
1621                                 end
1622                             }
1623                         end
1624                         if problem
1625                             break
1626                         end
1627                     end
1628                 }
1629                 if ! problem
1630                     begin
1631                         moved = 0
1632                         copied = 0
1633                         stuff.keys.each { |key|
1634                             if key.is_a?(Label) && stuff[key][:combo] && stuff[key][:combo].active <= 1
1635                                 destination = stuff[key][:pathlabel].text
1636                                 label2entries[key].each { |entry|
1637                                     if stuff[key][:combo].active == 0
1638                                         system("cp -dp '#{entry.path}' '#{destination}'") or raise "failed to copy '#{entry.path}'"
1639                                         copied += 1
1640                                     elsif stuff[key][:combo].active == 1
1641                                         system("mv '#{entry.path}' '#{destination}'") or raise "failed to move '#{entry.path}'"
1642                                         moved += 1
1643                                     end
1644                                 }
1645                             end
1646                         }
1647                         removed = 0
1648                         if stuff['toremove'][:combo] && stuff['toremove'][:combo].active == 0
1649                             $allentries.each { |entry|
1650                                 if entry.removed
1651                                     File.delete(entry.path)
1652                                     removed += 1
1653                                 end
1654                             }
1655                         end
1656                     rescue
1657                         msg 1, "woops: #{$!}"
1658                         show_popup(dialog, utf8(_("Unexpected error: '%s'.") % $!))
1659                     end
1660                     show_popup(dialog, utf8(_("Successfully moved %d files, copied %d file, and removed %d files.") % [ moved, copied, removed ]))
1661                     dialog.destroy
1662                     reset_all
1663                     return
1664                 end
1665
1666             else
1667                 dialog.destroy
1668                 return
1669             end
1670         }
1671     end
1672 end
1673
1674 def visible(entry)
1675     if ! entry
1676         #- just "executed"
1677         return
1678     end
1679     if ! entry.button
1680         #- not yet loaded
1681         return
1682     end
1683     if entry.labeled
1684         if entry.labeled.button.active?
1685             return true
1686         else
1687             return false
1688         end
1689     elsif entry.removed
1690         if $toremove_button.active?
1691             return true
1692         else
1693             return false
1694         end
1695     else
1696         if $unlabelled_button.active?
1697             return true
1698         else
1699             return false
1700         end
1701     end
1702 end
1703
1704 def update_visibility(entry)
1705     v = visible(entry)
1706     if v.nil?
1707         return
1708     end
1709     if v
1710         entry.button.show
1711     else
1712         entry.button.hide
1713     end
1714 end
1715
1716 def update_all_visibilities_aux
1717     $allentries.each { |entry|
1718         update_visibility(entry)
1719     }
1720     shown = $mainview.get_shown_entry
1721     shown or return
1722     while shown.button && ! shown.button.visible? && shown != $allentries.last
1723         shown = $allentries[$allentries.index(shown) + 1]
1724     end
1725     if shown.button && shown.button.visible?
1726         shown.button.grab_focus
1727         return
1728     end
1729     $allentries.reverse.each { |entry|
1730         if entry.button && entry.button.visible?
1731             entry.button.grab_focus
1732             return
1733         end
1734     }
1735 end
1736
1737 def update_all_visibilities
1738     update_all_visibilities_aux
1739     Gtk.main_iteration while Gtk.events_pending?
1740     shown = $mainview.get_shown_entry
1741     shown and autoscroll_if_needed(shown.button, false)
1742 end
1743
1744
1745 def preferences
1746     dialog = Gtk::Dialog.new(utf8(_("Edit preferences")),
1747                              $main_window,
1748                              Gtk::Dialog::MODAL | Gtk::Dialog::DESTROY_WITH_PARENT,
1749                              [Gtk::Stock::OK, Gtk::Dialog::RESPONSE_OK],
1750                              [Gtk::Stock::CANCEL, Gtk::Dialog::RESPONSE_CANCEL])
1751
1752     tooltips = Gtk::Tooltips.new
1753
1754     table_y = 0
1755
1756     dialog.vbox.add(tbl = Gtk::Table.new(0, 0, false))
1757     tbl.attach(Gtk::Alignment.new(1, 0.5, 0, 0).add(Gtk::Label.new.set_markup(utf8(_("Command for watching videos: ")))),
1758                0, 1, table_y, table_y + 1, Gtk::FILL, Gtk::SHRINK, 2, 2)
1759     tbl.attach(Gtk::Alignment.new(0, 0.5, 1, 0).add(video_viewer_entry = Gtk::Entry.new.set_text($config['video-viewer']).set_size_request(250, -1)),
1760                1, 2, table_y, table_y + 1, Gtk::FILL, Gtk::SHRINK, 2, 2)
1761     tooltips.set_tip(video_viewer_entry, utf8(_("Use %f to specify the filename;\nfor example: /usr/bin/mplayer %f")), nil)
1762
1763     table_y += 1
1764     tbl.attach(Gtk::Alignment.new(1, 0.5, 0, 0).add(Gtk::Label.new.set_markup(utf8(_("Browser's command: ")))),
1765                0, 1, table_y, table_y + 1, Gtk::FILL, Gtk::SHRINK, 2, 2)
1766     tbl.attach(Gtk::Alignment.new(0, 0.5, 1, 0).add(browser_entry = Gtk::Entry.new.set_text($config['browser'])),
1767                1, 2, table_y, table_y + 1, Gtk::FILL, Gtk::SHRINK, 2, 2)
1768     tooltips.set_tip(browser_entry, utf8(_("Use %f to specify the filename;\nfor example: /usr/bin/mozilla-firefox -remote 'openURL(%f,new-window)' || /usr/bin/mozilla-firefox %f")), nil)
1769
1770     table_y += 1
1771     tbl.attach(Gtk::Alignment.new(1, 0.5, 0, 0).add(Gtk::Label.new.set_markup(utf8(_("Thumbnails height: ")))),
1772                0, 1, table_y, table_y + 1, Gtk::FILL, Gtk::SHRINK, 2, 2)
1773     tbl.attach(Gtk::Alignment.new(0, 0.5, 1, 0).add(thumbnails_height = Gtk::SpinButton.new(32, 256, 16).set_value($config['thumbnails-height'].to_i)),
1774                1, 2, table_y, table_y + 1, Gtk::FILL, Gtk::SHRINK, 2, 2)
1775     tooltips.set_tip(thumbnails_height, utf8(_("The desired height of the thumbnails in the thumbnails line of the bottom")), nil)
1776
1777     table_y += 1
1778     tbl.attach(Gtk::Alignment.new(1, 0.5, 0, 0).add(Gtk::Label.new.set_markup(utf8(_("Preloading distance: ")))),
1779                0, 1, table_y, table_y + 1, Gtk::FILL, Gtk::SHRINK, 2, 2)
1780     tbl.attach(Gtk::Alignment.new(0, 0.5, 1, 0).add(preload_distance = Gtk::SpinButton.new(0, 50, 1).set_value($config['preload-distance'].to_i)),
1781                1, 2, table_y, table_y + 1, Gtk::FILL, Gtk::SHRINK, 2, 2)
1782     tooltips.set_tip(preload_distance, utf8(_("Amount of pictures preloaded left and right to the currently shown")), nil)
1783
1784     table_y += 1
1785     tbl.attach(Gtk::Alignment.new(1, 0.5, 0, 0).add(Gtk::Label.new.set_markup(utf8(_("Cache memory use: ")))),
1786                0, 1, table_y, table_y + 1, Gtk::FILL, Gtk::SHRINK, 2, 2)
1787     tbl.attach(Gtk::Alignment.new(0, 0.5, 1, 0).add(cache_vbox = Gtk::VBox.new(false, 0)),
1788                1, 2, table_y, table_y + 1, Gtk::FILL, Gtk::SHRINK, 2, 2)
1789     cache_vbox.pack_start(Gtk::HBox.new(false, 0).pack_start(cache_memfree_radio = Gtk::RadioButton.new(''), false, false).
1790                                                   pack_start(cache_memfree_spin = Gtk::SpinButton.new(0, 100, 10), false, false).
1791                                                   pack_start(cache_memfree_label = Gtk::Label.new(utf8(_("% of free memory"))), false, false), false, false)
1792     cache_memfree_spin.signal_connect('value-changed') { cache_memfree_radio.active = true }
1793     tooltips.set_tip(cache_memfree_spin, utf8(_("Percentage of free memory (+ buffers/cache) measured at startup")), nil)
1794     cache_vbox.pack_start(Gtk::HBox.new(false, 0).pack_start(cache_specify_radio = Gtk::RadioButton.new(cache_memfree_radio, ''), false, false).
1795                                                   pack_start(cache_specify_spin = Gtk::SpinButton.new(0, 4000, 50), false, false).
1796                                                   pack_start(cache_specify_label = Gtk::Label.new(utf8(_("MB"))).set_sensitive(false), false, false), false, false)
1797     cache_specify_spin.signal_connect('value-changed') { cache_specify_radio.active = true }
1798     cache_memfree_radio.signal_connect('toggled') {
1799         if cache_memfree_radio.active?
1800             cache_memfree_label.sensitive = true
1801             cache_specify_label.sensitive = false
1802         else
1803             cache_specify_label.sensitive = true
1804             cache_memfree_label.sensitive = false
1805         end
1806     }
1807     tooltips.set_tip(cache_specify_spin, utf8(_("Amount of memory in megabytes")), nil)
1808     if $config['cache-memory-use'] =~ /memfree_(\d+)/
1809         cache_memfree_spin.value = $1.to_i
1810     else
1811         cache_specify_spin.value = $config['cache-memory-use'].to_i / 1024
1812     end
1813
1814     table_y += 1
1815     tbl.attach(update_exif_orientation_check = Gtk::CheckButton.new(utf8(_("Update file's EXIF orientation when rotating a picture"))),
1816                0, 2, table_y, table_y + 1, Gtk::FILL, Gtk::SHRINK, 2, 2)
1817     tooltips.set_tip(update_exif_orientation_check, utf8(_("When rotating a picture (Alt-Right/Left), also update EXIF orientation in the file itself")), nil)
1818     update_exif_orientation_check.active = $config['rotate-set-exif'] == 'true'
1819
1820     dialog.vbox.show_all
1821     dialog.run { |response|
1822         if response == Gtk::Dialog::RESPONSE_OK
1823             $config['video-viewer'] = from_utf8(video_viewer_entry.text)
1824             $config['browser'] = from_utf8(browser_entry.text)
1825             $config['thumbnails-height'] = thumbnails_height.value
1826             $config['preload-distance'] = preload_distance.value
1827             $config['rotate-set-exif'] = update_exif_orientation_check.active?.to_s
1828             if cache_memfree_radio.active?
1829                 $config['cache-memory-use'] = "memfree_#{cache_memfree_spin.value}%"
1830             else
1831                 $config['cache-memory-use'] = cache_specify_spin.value.to_i * 1024
1832             end
1833             set_cache_memory_use_figure
1834         end
1835     }
1836     dialog.destroy
1837 end
1838
1839 def perform_undo
1840     if $undo_mb.sensitive?
1841         $redo_mb.sensitive = true
1842         if not more_undoes = UndoHandler.undo($statusbar)
1843             $undo_mb.sensitive = false
1844         end
1845     end
1846 end
1847
1848 def perform_redo
1849     if $redo_mb.sensitive?
1850         $undo_mb.sensitive = true
1851         if not more_redoes = UndoHandler.redo($statusbar)
1852             $redo_mb.sensitive = false
1853         end
1854     end
1855 end
1856
1857 def create_menubar
1858     #- menu
1859     mb = Gtk::MenuBar.new
1860
1861     filemenu = Gtk::MenuItem.new(utf8(_("_File")))
1862     filesubmenu = Gtk::Menu.new
1863     filesubmenu.append(open      = Gtk::ImageMenuItem.new(Gtk::Stock::OPEN))
1864     filesubmenu.append(            Gtk::SeparatorMenuItem.new)
1865     filesubmenu.append($execute  = Gtk::ImageMenuItem.new(Gtk::Stock::EXECUTE).set_sensitive(false))
1866     filesubmenu.append(            Gtk::SeparatorMenuItem.new)
1867     filesubmenu.append(quit      = Gtk::ImageMenuItem.new(Gtk::Stock::QUIT))
1868     filemenu.set_submenu(filesubmenu)
1869     mb.append(filemenu)
1870
1871     open.signal_connect('activate') { open_dir_popup }
1872     $execute.signal_connect('activate') { execute }
1873     quit.signal_connect('activate') { try_quit }
1874
1875     editmenu = Gtk::MenuItem.new(utf8(_("_Edit")))
1876     editsubmenu = Gtk::Menu.new
1877     editsubmenu.append($undo_mb    = Gtk::ImageMenuItem.new(Gtk::Stock::UNDO).set_sensitive(false))
1878     editsubmenu.append($redo_mb    = Gtk::ImageMenuItem.new(Gtk::Stock::REDO).set_sensitive(false))
1879     editsubmenu.append(              Gtk::SeparatorMenuItem.new)
1880     editsubmenu.append(prefs       = Gtk::ImageMenuItem.new(Gtk::Stock::PREFERENCES))
1881     editmenu.set_submenu(editsubmenu)
1882     mb.append(editmenu)
1883
1884     $undo_mb.signal_connect('activate') { perform_undo }
1885     $redo_mb.signal_connect('activate') { perform_redo }
1886     prefs.signal_connect('activate') { preferences }
1887
1888     helpmenu = Gtk::MenuItem.new(utf8(_("_Help")))
1889     helpsubmenu = Gtk::Menu.new
1890     helpsubmenu.append(howto = Gtk::ImageMenuItem.new(Gtk::Stock::HELP))
1891     helpsubmenu.append(speed = Gtk::ImageMenuItem.new(utf8(_("Speedup: key shortcuts"))))
1892     speed.image = Gtk::Image.new("#{$FPATH}/images/stock-info-16.png")
1893     helpsubmenu.append(tutos = Gtk::ImageMenuItem.new(utf8(_("Online tutorials (opens a web-browser)"))))
1894     tutos.image = Gtk::Image.new("#{$FPATH}/images/stock-web-16.png")
1895     helpsubmenu.append(Gtk::SeparatorMenuItem.new)
1896     helpsubmenu.append(about = Gtk::ImageMenuItem.new(Gtk::Stock::ABOUT))
1897     helpmenu.set_submenu(helpsubmenu)
1898     mb.append(helpmenu)
1899
1900     howto.signal_connect('activate') {
1901         show_popup($main_window, utf8(_("<span size='large' weight='bold'>Help</span>
1902
1903 1. Open a directory with <span foreground='darkblue'>File/Open</span>; the classifier will scan it (including subdirectories) and
1904 show thumbnails for all photos and videos at the bottom.
1905
1906 2. You can then navigate through images with the <span foreground='darkblue'>Left/Right</span> keyboard keys, or by <span foreground='darkblue'>clicking</span>
1907 on thumbnails.
1908
1909 3. You may associate a <span foreground='darkblue'>label</span> to each thumbnail. Either hit the <span foreground='darkblue'>Delete</span> key to associate
1910 the built-in <i>to remove</i> label, or hit any alphabetical key to associate a label you define.
1911 The first time you hit a key without any label associated, a popup will ask for the full
1912 name of this label, and what color you want. To clear the current label, hit the <span foreground='darkblue'>Space</span> key.
1913
1914 4. To help you better view what thumbnails are associated to your labels, you may <span foreground='darkblue'>hide</span>
1915 some of them by unchecking the labels checkboxes on the left.
1916
1917 5. Once you're finished reviewing all thumbnails, use <span foreground='darkblue'>File/Execute</span> to execute the desired
1918 actions according to associated labels. You can permanently remove (or not) images with
1919 the <i>to remove</i> label, and copy or move images with the labels you defined.
1920 ")), { :pos_centered => true, :not_transient => true })
1921     }
1922     speed.signal_connect('activate') {
1923         show_popup($main_window, utf8(_("<span size='large' weight='bold'>Key shortcuts</span>
1924
1925 <span foreground='darkblue'>Left/Right</span>: move left and right in images
1926 <span foreground='darkblue'>Enter</span>: 'view' current image: for images, display EXIF data; for videos, play it
1927 <span foreground='darkblue'>Alt-Left/Right</span>: rotate current image clockwise/counter-clockwise
1928 <span foreground='darkblue'>Delete</span>: assign the 'to remove' label on current image
1929 <span foreground='darkblue'>Space</span>: clear any label on current image
1930 <span foreground='darkblue'>Control-z</span>: undo
1931 <span foreground='darkblue'>Control-r</span>: redo
1932 <span foreground='darkblue'>Control-Space</span>: recenter thumbnails on current item
1933
1934 Any alphabetical key will assign (or popup for) the associated label on current image.
1935 ")), { :pos_centered => true, :not_transient => true })
1936     }
1937     tutos.signal_connect('activate') { open_url('http://booh.org/tutorial') }
1938     about.signal_connect('activate') { call_about }
1939
1940
1941     #- no toolbar, to save height
1942
1943     return mb
1944 end
1945
1946 def reset_labels
1947     for child in $labels_vbox.children
1948         $labels_vbox.remove(child)
1949     end
1950     $labels_vbox.pack_start(Gtk::Label.new(utf8(_("Labels list:"))).set_justify(Gtk::Justification::CENTER), false, false).show_all
1951     $labels = {}
1952     $ordered_labels = []
1953     lbl = Gtk::Label.new.set_markup(utf8(_("<i>unlabelled</i>")))
1954     $labels_vbox.pack_start(Gtk::HBox.new(false, 5).pack_start($unlabelled_button = Gtk::CheckButton.new.add(Gtk::EventBox.new.add(lbl)), false, false).
1955                                                     pack_start(Gtk::Label.new, true, true).  #- I suck
1956                                                     pack_start($unlabelled_counter = Gtk::Label.new.set_markup('<tt>0</tt>'), false, false).show_all)
1957     $unlabelled_button.active = true
1958     $unlabelled_button.signal_connect('toggled') { update_all_visibilities }
1959     lbl = Gtk::Label.new.set_markup(utf8(_("<i>to remove</i>")))
1960     $labels_vbox.pack_start(Gtk::HBox.new(false, 5).pack_start($toremove_button = Gtk::CheckButton.new.add(evt = Gtk::EventBox.new.add(lbl)), false, false).
1961                                                     pack_start(Gtk::Label.new, true, true).
1962                                                     pack_start($toremove_counter = Gtk::Label.new.set_markup('<tt>0</tt>'), false, false).show_all)
1963     $toremove_button.active = true
1964     $toremove_button.signal_connect('toggled') { update_all_visibilities }
1965     evt.modify_bg(Gtk::StateType::NORMAL, $color_red)
1966     evt.modify_bg(Gtk::StateType::PRELIGHT, $color_red.lighter.lighter)
1967     evt.modify_bg(Gtk::StateType::ACTIVE, $color_red.lighter)
1968 end
1969
1970 def cleanup_loaders
1971     $allentries.each { |e|
1972         e.cancel_loader
1973     }
1974 end
1975
1976 def reset_thumbnails
1977     cleanup_loaders
1978     $allentries = []
1979     if $preloader_running
1980         $preloader_force_exit = true
1981     end
1982     for child in $imagesline.children
1983         $imagesline.remove(child)
1984     end
1985     set_imagesline_size_request
1986 end
1987
1988 def set_imagesline_size_request
1989     $imagesline.set_size_request(-1, Gtk::Button.new.size_request[1] + Entry.thumbnails_height + Entry.thumbnails_height/4)
1990 end
1991
1992 def create_main_window
1993
1994     $videoborder_pixbuf = Gdk::Pixbuf.new("#{$FPATH}/images/video_border.png")
1995     $videoborder_pixmap, = $videoborder_pixbuf.render_pixmap_and_mask(0)
1996
1997     mb = create_menubar
1998
1999     main_vbox = Gtk::VBox.new(false, 0)
2000     main_vbox.pack_start(mb, false, false)
2001     mainview_hbox = Gtk::HBox.new
2002     mainview_hbox.pack_start(Gtk::Alignment.new(0.5, 0, 1, 1).add(left_vbox = Gtk::VBox.new(false, 5)), false, true)
2003     left_vbox.pack_start(($labels_vbox = Gtk::VBox.new(false, 5)), false, true)
2004     left_vbox.pack_end($loading_progressbar = Gtk::ProgressBar.new.set_text(utf8(_("Loading... %d%") % 0)), false, true)
2005     mainview_hbox.pack_start($mainview = MainView.new, true, true)
2006     main_vbox.pack_start(mainview_hbox, true, true)
2007     $imagesline_sw = Gtk::ScrolledWindow.new(nil, nil)
2008     $imagesline_sw.set_policy(Gtk::POLICY_ALWAYS, Gtk::POLICY_NEVER)
2009     $imagesline_sw.add_with_viewport($imagesline = Gtk::HBox.new(false, 0).show)
2010     main_vbox.pack_start($imagesline_sw, false, false)
2011     main_vbox.pack_end($statusbar = Gtk::Statusbar.new, false, false)
2012
2013     set_imagesline_size_request
2014
2015     $main_window = create_window
2016     $main_window.add(main_vbox)
2017     $main_window.signal_connect('delete-event') {
2018         try_quit
2019     }
2020
2021     #- read/save size and position of window
2022     if $config['pos-x'] && $config['pos-y']
2023         $main_window.move($config['pos-x'].to_i, $config['pos-y'].to_i)
2024     else
2025         $main_window.window_position = Gtk::Window::POS_CENTER
2026     end
2027     msg 3, "size: #{$config['width']}x#{$config['height']}"
2028     $main_window.set_default_size(($config['width'] || 800).to_i, ($config['height'] || 600).to_i)
2029     $main_window.signal_connect('configure-event') {
2030         msg 3, "configure: pos: #{$main_window.window.root_origin.inspect} size: #{$main_window.window.size.inspect}"
2031         x, y = $main_window.window.root_origin
2032         width, height = $main_window.window.size
2033         $config['pos-x'] = x
2034         $config['pos-y'] = y
2035         $config['width'] = width
2036         $config['height'] = height
2037         false
2038     }
2039
2040     $main_window.show_all
2041     $loading_progressbar.hide
2042 end
2043
2044
2045 handle_options
2046 read_config
2047 Gtk.init
2048
2049
2050 create_main_window
2051 check_config
2052
2053 if ARGV[0]
2054     if msg = open_dir(*ARGV)
2055         puts msg
2056     else
2057         Gtk.idle_add {
2058             show_entries($allentries)
2059             false
2060         }
2061     end
2062 end
2063 Gtk.main
2064
2065 cleanup_loaders
2066
2067 write_config