*** empty log message ***
[booh] / 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-2008 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
44     [ '--verbose-level', '-v', GetoptLong::REQUIRED_ARGUMENT, _("Set max verbosity level (0: errors, 1: warnings, 2: important messages, 3: other messages)") ],
45     [ '--sort-by-exif-date', '-s', GetoptLong::NO_ARGUMENT, _("Sort entries by EXIF date") ],
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 '--verbose-level'
68                 $verbose_level = arg.to_i
69
70             when '--sort-by-exif-date'
71                 $sort_by_exif_date = true
72
73             end
74         end
75     rescue
76         puts $!
77         usage
78         exit(1)
79     end
80 end
81
82 def startup_memfree
83     if $startup_memfree.nil?
84         meminfo = IO.readlines('/proc/meminfo').join
85         meminfo =~ /MemFree:.*?(\d+)/ or return -1
86         memfree = $1
87         meminfo =~ /Buffers:.*?(\d+)/ and buffers = $1
88         meminfo =~ /Cached:.*?(\d+)/ and cached = $1
89         $startup_memfree = memfree.to_i + buffers.to_i + cached.to_i
90     end
91     return $startup_memfree
92 end
93
94 def set_cache_memory_use_figure
95     
96     if $config['cache-memory-use'] =~ /memfree_(\d+)/
97         $config['cache-memory-use-figure'] = startup_memfree*$1.to_f/100
98     else
99         $config['cache-memory-use-figure'] = $config['cache-memory-use'].to_i
100     end
101     msg 2, _("Cache memory used: %s kB") % $config['cache-memory-use-figure']
102 end
103
104 def read_config
105     $config = {}
106     $config_file = File.expand_path('~/.booh-classifier-rc')
107     if File.readable?($config_file)
108         $xmldoc = REXML::Document.new(File.new($config_file))
109         $xmldoc.root.elements.each { |element|
110             txt = element.get_text
111             if txt
112                 if txt.value =~ /~~~/
113                     $config[element.name] = txt.value.split(/~~~/)
114                 else
115                     $config[element.name] = txt.value
116                 end
117             elsif element.elements.size == 0
118                 $config[element.name] = ''
119             else
120                 $config[element.name] = {}
121                 element.each { |chld|
122                     txt = chld.get_text
123                     $config[element.name][chld.name] = txt ? txt.value : nil
124                 }
125             end
126         }
127     end
128     $config['video-viewer'] ||= '/usr/bin/mplayer %f'
129     $config['browser'] ||= "/usr/bin/mozilla-firefox -remote 'openURL(%f,new-window)' || /usr/bin/mozilla-firefox %f"
130     $config['preload-distance'] ||= '5'
131     $config['cache-memory-use'] ||= 'memfree_80%'
132     $config['rotate-set-exif'] ||= 'true'
133     set_cache_memory_use_figure
134 end
135
136 def check_config
137     missing = %w(mplayer).delete_if { |prg| system("which #{prg} >/dev/null 2>/dev/null") }
138     if missing != []
139         show_popup($main_window, utf8(_("The following program(s) are needed to handle videos: '%s'. Videos will be ignored.") % missing.join(', ')), { :pos_centered => true })
140     end
141
142     if !system("which exif >/dev/null 2>/dev/null")
143         show_popup($main_window, utf8(_("The program 'exif' is needed to view EXIF data. Please install it.")), { :pos_centered => true })
144     end
145     viewer_binary = $config['video-viewer'].split.first
146     if viewer_binary && ! File.executable?(viewer_binary)
147         show_popup($main_window, utf8(_("The configured video viewer seems to be unavailable.
148 You should fix this in Edit/Preferences so that you can view videos.
149
150 Problem was: '%s' is not an executable file.
151 Hint: don't forget to specify the full path to the executable,
152 e.g. '/usr/bin/mplayer' is correct but 'mplayer' only is not.") % viewer_binary), { :pos_centered => true, :not_transient => true })
153     end
154     browser_binary = $config['browser'].split.first
155     if browser_binary && ! File.executable?(browser_binary)
156         show_popup($main_window, utf8(_("The configured browser seems to be unavailable.
157 You should fix this in Edit/Preferences so that you can open URLs.
158
159 Problem was: '%s' is not an executable file.") % browser_binary), { :pos_centered => true, :not_transient => true })
160     end
161 end
162
163 def write_config
164     ios = File.open($config_file, "w")
165     $xmldoc = Document.new "<booh-classifier-rc version='#{$VERSION}'/>"
166     $xmldoc << XMLDecl.new(XMLDecl::DEFAULT_VERSION, $CURRENT_CHARSET)
167     $config.each_pair { |key, value|
168         elem = $xmldoc.root.add_element key
169         if value.is_a? Hash
170             $config[key].each_pair { |subkey, subvalue|
171                 subelem = elem.add_element subkey
172                 subelem.add_text subvalue.to_s
173             }
174         elsif value.is_a? Array
175             elem.add_text value.join('~~~')
176         else
177             if !value
178                 elem.remove
179             else
180                 elem.add_text value.to_s
181             end
182         end
183     }
184     $xmldoc.write(ios, 0)
185     ios.close
186 end
187
188 def save_undo(name, closure, *params)
189     UndoHandler.save_undo(name, closure, [ *params ])
190     $undo_mb.sensitive = true
191     $redo_mb.sensitive = false
192 end
193
194 def get_mem
195     IO.readlines('/proc/self/status').join =~ /VmRSS.*?(\d+)\s*kB/
196     msg 3, "RSS: #{$1}"
197     return $1.to_i
198 end
199
200 def show_mem(*txt)
201     txt.length > 0 and print txt[0]
202     msg 2, "RSS: #{get_mem}"
203 end
204
205 class Gdk::Color
206     def darker
207         color = dup
208         color.red = [ color.red - 10000, 0 ].max
209         color.green = [ color.green - 10000, 0 ].max
210         color.blue = [ color.blue - 10000, 0 ].max
211         return color
212     end
213     def lighter
214         color = dup
215         color.red = [ color.red + 10000, 65535 ].min
216         color.green = [ color.green + 10000, 65535 ].min
217         color.blue = [ color.blue + 10000, 65535 ].min
218         return color
219     end
220 end
221
222 $color_red = Gdk::Color.new(65535, 0, 0)
223 $colors = [ Gdk::Color.new(0, 65535, 0),
224             Gdk::Color.new(0, 0, 65535),
225             Gdk::Color.new(65535, 65535, 0),
226             Gdk::Color.new(0, 65535, 65535),
227             Gdk::Color.new(65535, 0, 65535) ]
228
229 class Label
230     attr_accessor :color, :name, :button
231     def initialize(name)
232         @name = name
233     end
234 end
235
236 class Entry
237     @@thumbnails_height = 64
238     @@max_width = nil
239     def Entry.thumbnails_height
240         return @@thumbnails_height
241     end
242
243     attr_accessor :path, :type, :angle, :button, :image, :alignment, :removed, :labeled
244
245     def initialize(path, type)
246         @path = path
247         @type = type
248         if @@max_width.nil?
249             @@max_width = $main_window.root_window.size[0] - $labels_vbox.allocation.width - ( $videoborder_pixbuf.width + MainView.borders_thickness) * 2
250         end
251     end
252
253     def pixbuf_full
254         if @pixbuf_full.nil?
255             msg 3, ">>> pixbuf_full #{path}"
256             load_into_pixbuf_full
257         end
258         return @pixbuf_full
259     end
260     def free_pixbuf_full
261         if @pixbuf_full.nil?
262             return false
263         else
264             msg 3, ">>> free_pixbuf_full #{path}"
265             @pixbuf_full = nil
266             return true
267         end
268     end
269     def pixbuf_main
270         width, height = $mainview.window.size 
271         width = MainView.get_usable_width(width)
272         height = MainView.get_usable_height(height)
273         if @pixbuf_main.nil? || width != @width || height != @height
274             msg 3, ">>> pixbuf_main #{path}"
275             @width = width
276             @height = height
277             load_into_pixbuf_full  #- make sure it is loaded
278             if @pixbuf_full.nil?
279                 return
280             end
281             if @pixbuf_full.width.to_f / @pixbuf_full.height > width.to_f / height
282                 resized_height = @pixbuf_full.height * (width.to_f/@pixbuf_full.width)
283                 if @pixbuf_full.width > width || @pixbuf_full.height > resized_height
284                     @pixbuf_main = @pixbuf_full.scale(width, resized_height, Gdk::Pixbuf::INTERP_BILINEAR)
285                 else
286                     @pixbuf_main = @pixbuf_full
287                 end
288             else
289                 resized_width = @pixbuf_full.width * (height.to_f/@pixbuf_full.height)
290                 if @pixbuf_full.width > resized_width || @pixbuf_full.height > height
291                     @pixbuf_main = @pixbuf_full.scale(resized_width, height, Gdk::Pixbuf::INTERP_BILINEAR)
292                 else
293                     @pixbuf_main = @pixbuf_full
294                 end
295             end
296         end
297         return @pixbuf_main
298     end
299     def free_pixbuf_main
300         if @pixbuf_main.nil?
301             return false
302         else
303             msg 3, ">>> free_pixbuf_main #{path}"
304             @pixbuf_main = nil
305             return true
306         end
307     end
308     def pixbuf_thumbnail
309         if @pixbuf_thumbnail.nil?
310             if @pixbuf_main
311                 msg 3, ">>> pixbuf_thumbnail from main #{path}"
312                 @pixbuf_thumbnail = @pixbuf_main.scale(@pixbuf_main.width * (@@thumbnails_height.to_f/@pixbuf_main.height), @@thumbnails_height, Gdk::Pixbuf::INTERP_BILINEAR)
313             else
314                 msg 3, ">>> pixbuf_thumbnail from file #{path}"
315                 @pixbuf_thumbnail = load_into_pixbuf_at_size { |w, h|
316                     if @angle == 0
317                         if h > @@thumbnails_height
318                             [ w * @@thumbnails_height.to_f/h, @@thumbnails_height ]
319                         else
320                             [ w, h ]
321                         end
322                     else
323                         if w > @@thumbnails_height
324                             [ @@thumbnails_height, h * @@thumbnails_height.to_f/w ]
325                         else
326                             [ w, h ]
327                         end
328                     end
329                 }
330             end
331         end
332         return @pixbuf_thumbnail
333     end
334     def free_pixbuf_thumbnail
335         if @pixbuf_thumbnail.nil?
336             return false
337         else
338             msg 3, ">>> free_pixbuf_thumbnail #{path}"
339             @pixbuf_thumbnail = nil
340             return true
341         end
342     end
343
344     def outline_color
345         if removed
346             return $color_red
347         elsif labeled
348             return labeled.color
349         else
350             return nil
351         end
352     end
353
354     def show_bg
355         if outline_color.nil?
356             button.modify_bg(Gtk::StateType::NORMAL, nil)
357             button.modify_bg(Gtk::StateType::PRELIGHT, nil)
358             button.modify_bg(Gtk::StateType::ACTIVE, nil)
359         else
360             button.modify_bg(Gtk::StateType::NORMAL, outline_color)
361             button.modify_bg(Gtk::StateType::PRELIGHT, outline_color.lighter)
362             button.modify_bg(Gtk::StateType::ACTIVE, outline_color)
363         end
364     end
365
366     def get_beautified_name
367         if type == 'image'
368             size = get_image_size(path)
369             return _("%s (%sx%s, %s KB)") % [File.basename(@path).gsub(/\.[^.]+$/, ''),
370                                              size[:x],
371                                              size[:y],
372                                              commify(file_size(path)/1024)]
373         else
374             return _("%s (video - %s KB)") % [File.basename(@path).gsub(/\.[^.]+$/, ''),
375                                              commify(file_size(path)/1024)]
376         end
377     end
378
379     private
380     def cleanup_dir(dir)
381         Dir.entries(dir).each { |file| file != '.' && file != '..' and File.delete(File.join(dir, file)) }
382         Dir.delete(dir)
383     end
384
385     def load_into_pixbuf_full
386         if @pixbuf_full.nil?
387             msg 3, ">>> load_into_pixbuf_full #{path}"
388             @pixbuf_full = load_into_pixbuf_at_size { |w, h|
389                 if @angle == 0
390                     if w > @@max_width
391                         #- save memory and speedup (+35%) loading 
392                         [ w * (factor = @@max_width.to_f/w), h * factor ]
393                     else
394                         [ w, h ]
395                     end
396                 else
397                     if h > @@max_width
398                         [ w * (factor = @@max_width.to_f/h), h * factor ]
399                     else
400                         [ w, h ]
401                     end
402                 end
403             }
404         end
405     end
406
407     def load_into_pixbuf_at_size(&specify_size)
408         pixbuf = nil
409         if @type == 'video'
410             tmp = Tempfile.new("boohclassifiertemp")
411             tmp.close!
412             Dir.mkdir(dest_dir = tmp.path)
413             orig_base = File.basename(path)
414             tmpdir = gen_video_thumbnail(path, false, 0)
415             if tmpdir.nil?
416                 return
417             end
418             image_path = "#{tmpdir}/00000001.jpg"
419         else
420             image_path = @path
421         end
422         if @angle.nil?
423             if @type == 'image'
424                 @angle = guess_rotate(image_path)
425             else
426                 @angle = 0
427             end
428         end
429         begin
430             #- use a pixbuf loader and trigger Gtk.main_iteration on each chunk if needed, to keep the UI responsive even
431             #- if loaded pictures are several MBs large
432             loader = Gdk::PixbufLoader.new
433             loader.signal_connect('size-prepared') { |l, w, h|
434                 r = specify_size.call(w, h)
435                 msg 3, "specified sizes: #{r[0]} #{r[1]}"
436                 loader.set_size(*specify_size.call(w, h))
437             }
438             loader.signal_connect('area-prepared') { pixbuf = loader.pixbuf }
439             file = File.new(image_path)
440             while (chunk = file.read(4096)) != nil
441                 loader.write(chunk)
442                 Gtk.main_iteration while Gtk.events_pending?
443             end
444             file.close
445             loader.close
446             if pixbuf.nil?
447                 raise "Loaded pixbuf nil - #{path} #{image_path}"
448             end
449         rescue Gdk::PixbufError
450             msg 0, "Cannot load #{image_path}: #{$!}"
451             return
452         ensure
453             if @type == 'video'
454                 File.delete(image_path)
455                 Dir.rmdir(tmpdir)
456             end
457         end
458         if pixbuf
459             if @angle != 0
460                 msg 3, ">>> load_into_pixbuf_full #{image_path} => rotate #{@angle}"
461                 pixbuf = rotate_pixbuf(pixbuf, @angle)
462             end
463         end
464         if @type == 'video'
465             cleanup_dir(dest_dir)
466         end
467         return pixbuf
468     end
469
470     def to_s
471         @path
472     end
473 end
474
475 $allentries = []
476
477 def run_preloader_real
478     msg 3, "*** >> main preloading triggered..."
479     if $preloader_running
480         msg 3, "*** >>>>>> already running, return <<<<<<<<"
481         return
482     end
483     $preloader_running = true
484     if $mainview.get_shown_entry
485         mem = get_mem
486         if mem > $config['cache-memory-use-figure']
487             msg 3, "too much RSS, stopping preloading, triggering GC"
488             $preloader_running = false
489             GC.start
490             msg 3, "GC finished"
491             return
492         end
493         index = $allentries.index($mainview.get_shown_entry)
494         for j in 1 .. $config['preload-distance'].to_i
495             i = index + j
496             if i < $allentries.size
497                 $allentries[i].pixbuf_main
498             end
499             #- in case just loaded another directory
500             if $preloader_force_exit
501                 $preloader_running = false
502                 $preloader_force_exit = false
503                 return
504             end
505             i = index - j
506             if i >= 0
507                 $allentries[i].pixbuf_main
508             end
509             #- in case just loaded another directory
510             if $preloader_force_exit
511                 $preloader_running = false
512                 $preloader_force_exit = false
513                 return
514             end
515         end
516         check_memory_free_cache_if_needed
517     end
518     $preloader_running = false
519     msg 3, "*** << main preloading finished"
520     #- if we're already on a different image, rerun the preloader
521     if index != $allentries.index($mainview.get_shown_entry)
522         run_preloader_real
523     end
524 end
525
526 def run_preloader
527     if ! $preloader_allowed
528         msg 3, "*** preloader not yet allowed"
529         return
530     end
531     Gtk.timeout_add(10) {
532         run_preloader_real
533         false
534     }
535 end
536
537 class MainView < Gtk::DrawingArea
538
539     @@borders_thickness = 5
540     @@borders_length = 25
541
542     def MainView.borders_thickness
543         return @@borders_thickness
544     end
545
546     def MainView.get_usable_width(available_width)
547         return available_width - ($videoborder_pixbuf.width + @@borders_thickness) * 2
548     end
549
550     def MainView.get_usable_height(available_height)
551         return available_height - @@borders_thickness * 2
552     end
553     
554     def initialize
555         super()
556         signal_connect('expose-event') { draw }
557         signal_connect('configure-event') { update_shown }
558         @preloader_running = false
559     end
560
561     def set_shown_entry(entry)
562         t1 = Time.now
563         if ! entry.nil? && entry == @entry
564             return
565         end
566         if ! entry.nil?
567             if ! entry.button
568                 #- not loaded yet
569                 return
570             else
571                 entry.button.grab_focus
572             end
573         end
574         @entry = entry
575         redraw        
576         run_preloader
577         msg 3, "entry shown in: #{Time.now - t1} s"
578     end
579
580     def get_shown_entry
581         return @entry
582     end
583
584     def show_next_entry(entry)
585         index = $allentries.index(entry) + 1
586         if index < $allentries.size
587             set_shown_entry($allentries[index])
588         end
589     end
590
591     def redraw
592         update_shown
593         w, h = window.size
594         window.begin_paint(Gdk::Rectangle.new(0, 0, w, h))
595         window.clear
596         draw
597         window.end_paint
598     end
599
600     def update_shown
601         if @entry
602             @pixbuf = @entry.pixbuf_main
603             width, height = window.size 
604             @xpos = (width - @pixbuf.width)/2
605             @ypos = (height - @pixbuf.height)/2
606         else
607             @pixbuf = nil
608         end
609     end
610
611     def draw
612         if @pixbuf
613             window.draw_pixbuf(nil, @pixbuf, 0, 0, @xpos, @ypos, -1, -1, Gdk::RGB::DITHER_NONE, -1, -1)
614             if @entry && @entry.type == 'video'
615                 window.draw_borders($videoborder_pixbuf, @xpos - $videoborder_pixbuf.width, @xpos + @pixbuf.width, @ypos, @ypos + @pixbuf.height)
616             end
617             if ! @entry.outline_color.nil?
618                 gc = Gdk::GC.new(window)
619                 colormap.alloc_color(@entry.outline_color, false, true)
620                 gc.set_foreground(@entry.outline_color)
621                 if @entry && @entry.type == 'video'
622                     xleft = @xpos - $videoborder_pixbuf.width
623                     xright = @xpos + @pixbuf.width + $videoborder_pixbuf.width
624                 else
625                     xleft = @xpos
626                     xright = @xpos + @pixbuf.width
627                 end
628                 window.draw_polygon(gc, true, [[xleft - @@borders_thickness, @ypos - @@borders_thickness],
629                                                [xright + @@borders_thickness, @ypos - @@borders_thickness],
630                                                [xright + @@borders_thickness, @ypos + @pixbuf.height + @@borders_thickness],
631                                                [xleft - @@borders_thickness, @ypos + @pixbuf.height + @@borders_thickness],
632                                                [xleft - @@borders_thickness, @ypos - 1],
633                                                [xleft - 1, @ypos - 1],
634                                                [xleft - 1, @ypos + @pixbuf.height + 1],
635                                                [xright + 1, @ypos + @pixbuf.height + 1],
636                                                [xright + 1, @ypos - 1],
637                                                [xleft - @@borders_thickness, @ypos - 1]])
638             end
639         end
640     end
641 end
642
643 def check_memory_free_cache_if_needed
644     i = $allentries.index($mainview.get_shown_entry)
645     return if i.nil?
646     if get_mem < $config['cache-memory-use-figure'] * 2 / 3
647         return
648     end
649     msg 3, "too much RSS, triggering GC"
650     GC.start
651     msg 3, "GC finished"
652     ($allentries.size - 1).downto(1) { |j|
653         if get_mem < $config['cache-memory-use-figure'] / 2
654             break
655         end
656         index = i + j
657         msg 3, "too much RSS, freeing full size of #{i+j} and #{i-j}..."
658         if i + j < $allentries.size
659             $allentries[i+j].free_pixbuf_full
660         end
661         if i - j > 0
662             $allentries[i-j].free_pixbuf_full
663         end
664     }
665 end
666
667 def autoscroll_if_needed(button)
668     xpos_left = button.allocation.x
669     xpos_right = button.allocation.x + button.allocation.width
670     hadj = $imagesline_sw.hadjustment
671     current_minx_visible = hadj.value
672     current_maxx_visible = hadj.value + hadj.page_size
673     if xpos_left < current_minx_visible
674         #- autoscroll left
675         newval = hadj.value - (current_minx_visible - xpos_left)
676         hadj.value = newval
677         button.queue_draw  #- TOREMOVE: the visual focus is displayed incorrectly
678     elsif xpos_right > current_maxx_visible
679         #- autoscroll right
680         newval = hadj.value + (xpos_right - current_maxx_visible)
681         if newval > hadj.upper - hadj.page_size
682             newval = hadj.upper - hadj.page_size
683         end
684         hadj.value = newval
685         button.queue_draw  #- TOREMOVE: the visual focus is displayed incorrectly
686     end
687 end
688
689 def show_popup(parent, msg, *options)
690     dialog = Gtk::Dialog.new
691     if options[0]
692         options = options[0]
693     else
694         options = {}
695     end
696     if options[:title]
697         dialog.title = options[:title]
698     else
699         dialog.title = utf8(_("Booh message"))
700     end
701     lbl = Gtk::Label.new
702     if options[:nomarkup]
703         lbl.text = msg
704     else
705         lbl.markup = msg
706     end
707     if options[:centered]
708         lbl.set_justify(Gtk::Justification::CENTER)
709     end
710     if options[:selectable]
711         lbl.selectable = true
712     end
713     if options[:topwidget]
714         dialog.vbox.add(options[0][:topwidget])
715     end
716     if options[:scrolled]
717         sw = Gtk::ScrolledWindow.new(nil, nil)
718         sw.set_policy(Gtk::POLICY_AUTOMATIC, Gtk::POLICY_AUTOMATIC)
719         sw.add_with_viewport(lbl)
720         dialog.vbox.add(sw)
721         dialog.set_default_size(500, 600)
722     else
723         dialog.vbox.add(lbl)
724         dialog.set_default_size(200, 120)
725     end
726     if options[:bottomwidget]
727         dialog.vbox.add(options[:bottomwidget])
728     end
729     if options[:okcancel]
730         dialog.add_button(Gtk::Stock::CANCEL, Gtk::Dialog::RESPONSE_CANCEL)
731     end
732     dialog.add_button(Gtk::Stock::OK, Gtk::Dialog::RESPONSE_OK)
733
734     if options[:pos_centered]
735         dialog.window_position = Gtk::Window::POS_CENTER
736     else
737         dialog.window_position = Gtk::Window::POS_MOUSE
738     end
739
740     if options[:linkurl]
741         linkbut = Gtk::Button.new('')
742         linkbut.child.markup = "<span foreground=\"#00000000FFFF\" underline=\"single\">#{options[0][:linkurl]}</span>"
743         linkbut.signal_connect('clicked') {
744             open_url(options[0][:linkurl] + '/index.html')
745             dialog.response(Gtk::Dialog::RESPONSE_OK)
746             set_mousecursor_normal
747         }
748         linkbut.relief = Gtk::RELIEF_NONE
749         linkbut.signal_connect('enter-notify-event') { set_mousecursor(Gdk::Cursor::HAND2, linkbut); false }
750         linkbut.signal_connect('leave-notify-event') { set_mousecursor(nil, linkbut); false }
751         dialog.vbox.add(Gtk::Alignment.new(0.5, 0.5, 0, 0).add(linkbut))
752     end
753
754     dialog.show_all
755
756     if options[:stuff_connector]
757         options[:stuff_connector].call({ :dialog => dialog })
758     end
759                                         
760     if !options[:not_transient]
761         dialog.transient_for = parent
762         dialog.run { |response|
763             if options[:data_getter]
764                 options[:data_getter].call
765             end
766             dialog.destroy
767             if options[:okcancel]
768                 return response == Gtk::Dialog::RESPONSE_OK
769             end
770         }
771     else
772         dialog.signal_connect('response') { dialog.destroy }
773     end
774 end
775
776 def view_entry(entry)
777     if entry.type == 'image'
778         show_popup($main_window,
779                    utf8(`exif -m '#{entry.path}'`),
780                    { :title => utf8(_("EXIF data of %s") % File.basename(entry.path)), :nomarkup => true, :scrolled => true, :not_transient => true })
781     else
782         cmd = from_utf8($config['video-viewer']).gsub('%f', "'#{entry.path}'") + ' &'
783         msg 2, cmd
784         system(cmd)
785     end
786 end
787
788 def thumbnail_keypressed(entry, event)
789     if event.state & Gdk::Window::MOD1_MASK != 0
790         #- ALT pressed: Alt-Left and Alft-Right rotate
791         if event.keyval == Gdk::Keyval::GDK_Left || event.keyval == Gdk::Keyval::GDK_Right
792             if event.keyval == Gdk::Keyval::GDK_Left
793                 entry.angle = (entry.angle - 90) % 360
794             else
795                 entry.angle = (entry.angle + 90) % 360
796             end
797             entry.free_pixbuf_full
798             entry.free_pixbuf_main
799             entry.free_pixbuf_thumbnail
800             $mainview.redraw
801             entry.image.pixbuf = entry.pixbuf_thumbnail
802             if $config['rotate-set-exif'] == 'true' && entry.type == 'image'
803                 Exif.set_orientation(entry.path, angle_to_exif_orientation(entry.angle))
804             end
805         end
806
807     elsif event.state & Gdk::Window::CONTROL_MASK != 0
808         #- CONTROL pressed: Ctrl-z and Ctrl-r for undo/redo
809         if event.keyval == Gdk::Keyval::GDK_z
810             perform_undo
811         end
812         if event.keyval == Gdk::Keyval::GDK_r
813             perform_redo
814         end
815
816     else
817         removed_before = entry.removed
818         label_before = entry.labeled
819
820         if event.keyval == Gdk::Keyval::GDK_Delete
821             entry.removed = true
822             entry.labeled = nil
823             entry.show_bg
824             $mainview.show_next_entry(entry)
825             update_visibility(entry)
826
827             save_undo(_("set for removal"),
828                       proc {
829                           entry.removed = removed_before
830                           entry.labeled = label_before
831                           entry.show_bg
832                           update_visibility(entry)
833                           if entry.button.visible?
834                               $mainview.set_shown_entry(entry)
835                           end
836                           proc {
837                               entry.removed = true
838                               entry.labeled = nil
839                               entry.show_bg
840                               update_visibility(entry)
841                               if entry.button.visible?
842                                   $mainview.set_shown_entry(entry)
843                               end
844                           }
845                       })
846
847         elsif event.keyval == Gdk::Keyval::GDK_space
848             if entry.labeled
849                 msg = _("Cleared label")
850             elsif entry.removed
851                 msg = _("Cleared set for removal")
852             end
853             entry.removed = false
854             entry.labeled = nil
855             entry.show_bg
856             $mainview.show_next_entry(entry)
857
858             save_undo(msg,
859                       proc {
860                           entry.removed = removed_before
861                           entry.labeled = label_before
862                           entry.show_bg
863                           $mainview.set_shown_entry(entry)
864                           proc {
865                               entry.removed = false
866                               entry.labeled = nil
867                               entry.show_bg
868                               $mainview.set_shown_entry(entry)
869                           }
870                       })
871
872         elsif event.keyval == Gdk::Keyval::GDK_Return
873             view_entry(entry)
874
875         else
876             char = [ Gdk::Keyval.to_unicode(event.keyval) ].pack("C*")
877             if char =~ /^[a-zA-z0-9]$/
878                 label = $labels[char]
879                 
880                 if label.nil?
881                     vb = Gtk::VBox.new(false, 0)
882                     vb.pack_start(entry = Gtk::Entry.new.set_text(char), false, false)
883                     vb.pack_start(Gtk::Alignment.new(0.5, 0.5, 0, 0).add(bt = Gtk::ColorButton.new))
884                     color = bt.color = Gdk::Color.new(16384 + rand(49151), 16384 + rand(49151), 16384 + rand(49151))
885                     bt.signal_connect('color-set') { color = bt.color }
886                     text = nil
887                     entry.signal_connect('changed') {  #- cannot add a new label with first letter of an existing label
888                         while $labels.has_key?(entry.text[0,1])
889                             entry.text = entry.text.sub(/./, '')
890                         end
891                     }
892                     if show_popup($main_window,
893                                   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,
894                                   { :okcancel => true, :bottomwidget => vb, :data_getter => proc { text = entry.text },
895                                     :stuff_connector => proc { |stuff| entry.select_region(0, 0)
896                                                                        entry.position = -1
897                                                                        entry.signal_connect('activate') { stuff[:dialog].response(Gtk::Dialog::RESPONSE_OK) } } } )
898                         if text.length > 0
899                             char = text[0,1]  #- in case it changed
900                             label = Label.new(text)
901                             label.color = color
902                             $labels[char] = label
903                             $ordered_labels << label
904                             lbl = Gtk::Label.new.set_markup('<b>(' + char + ')</b>' + text[1..-1]).set_justify(Gtk::Justification::CENTER)
905                             $labels_vbox.pack_start(label.button = Gtk::CheckButton.new.add(evt = Gtk::EventBox.new.add(lbl)).show_all)
906                             label.button.active = true
907                             label.button.signal_connect('toggled') { update_all_visibilities }
908                             evt.modify_bg(Gtk::StateType::NORMAL, label.color)
909                             evt.modify_bg(Gtk::StateType::PRELIGHT, label.color.lighter.lighter)
910                             evt.modify_bg(Gtk::StateType::ACTIVE, label.color.lighter)
911                         end
912                     end
913
914                 else
915                     entry.removed = false
916                     entry.labeled = label
917                     entry.show_bg
918                     $mainview.show_next_entry(entry)
919                     update_visibility(entry)
920
921                     save_undo(_("set label"),
922                               proc {
923                                   entry.removed = removed_before
924                                   entry.labeled = label_before
925                                   entry.show_bg
926                                   update_visibility(entry)
927                                   if entry.button.visible?
928                                       $mainview.set_shown_entry(entry)
929                                   end
930                                   proc {
931                                       entry.removed = false
932                                       entry.labeled = label
933                                       entry.show_bg
934                                       update_visibility(entry)
935                                       if entry.button.visible?
936                                           $mainview.set_shown_entry(entry)
937                                       end
938                                   }
939                               })
940                 end
941             end
942         end
943     end
944 end
945
946 def sb_msg(msg)
947     $statusbar.pop(0)
948     if msg
949         $statusbar.push(0, utf8(msg))
950     end
951 end
952
953 def show_entry(entry, i)
954     #- scope entry
955     msg 3, "showing entry #{entry}"
956     entry.image = Gtk::Image.new(entry.pixbuf_thumbnail)
957     if entry.type == 'video'
958         entry.button = Gtk::Button.new.add(Gtk::HBox.new.pack_start(da1 = Gtk::DrawingArea.new.set_size_request($videoborder_pixbuf.width, -1), false, false).
959                                            pack_start(entry.image).
960                                            pack_start(da2 = Gtk::DrawingArea.new.set_size_request($videoborder_pixbuf.width, -1), false, false))
961         da1.signal_connect('realize') { da1.window.set_back_pixmap($videoborder_pixmap, false) }
962         da2.signal_connect('realize') { da2.window.set_back_pixmap($videoborder_pixmap, false) }
963     else
964         entry.button = Gtk::Button.new.add(entry.image)
965     end
966     Gtk::Tooltips.new.set_tip(entry.button, entry.get_beautified_name, nil)
967     $imagesline.pack_start(entry.alignment = Gtk::Alignment.new(0.5, 1, 0, 0).add(entry.button).show_all, false, false)
968     entry.button.signal_connect('clicked') {
969         if (last_shown = $mainview.get_shown_entry) != entry
970             entry.alignment.set(0.5, 0, 0, 0)
971             last_shown and last_shown.alignment.set(0.5, 1, 0, 0)
972             $mainview.set_shown_entry(entry)
973             sb_msg(_("Selected %s") % entry.get_beautified_name)
974         end
975     }
976     entry.button.signal_connect('button-press-event') { |w, event|
977         if entry.type == 'video' && event.event_type == Gdk::Event::BUTTON2_PRESS
978             video_view(entry)
979         end
980     }
981     entry.button.signal_connect('focus-in-event') { entry.button.clicked; autoscroll_if_needed(entry.button) }
982     entry.button.signal_connect('key-press-event') { |w, e| thumbnail_keypressed(entry, e) }
983     if i == 0
984         entry.button.grab_focus
985     end
986     update_visibility(entry)
987     Gtk.main_iteration while Gtk.events_pending?
988 end
989
990 def show_entries(allentries)
991     sb_msg(_("Loading images..."))
992     $loading_progressbar.fraction = 0
993     $loading_progressbar.text = utf8(_("Loading... %d%") % 0)
994     $loading_progressbar.show
995     t1 = Time.now
996     total_loaded_files = 0
997     total_loaded_size = 0
998     i = 0
999     while i < allentries.size
1000 #        printf "%d %s\n", i, __LINE__
1001         entry = allentries[i]
1002         if i == 0
1003             loaded_pixbuf = entry.pixbuf_main
1004         else
1005             loaded_pixbuf = entry.pixbuf_thumbnail
1006         end
1007         if $allentries != allentries
1008             #- loaded another directory while this one was not yet finished
1009             msg 3, "allentries differ, stopping this deprecated load"
1010             return
1011         end
1012
1013         if loaded_pixbuf
1014             show_entry(entry, i)
1015             if $allentries != allentries
1016                 #- loaded another directory while this one was not yet finished
1017                 msg 3, "allentries differ, stopping this deprecated load"
1018                 return
1019             end
1020
1021             total_loaded_size += file_size(entry.path)
1022             if i % 4 == 0
1023                 check_memory_free_cache_if_needed
1024             end
1025             total_loaded_files += 1
1026             i += 1
1027             if i > $config['preload-distance'].to_i && i <= $config['preload-distance'].to_i * 2
1028                 #- when we're at preload distance, beging preloading to preload distance
1029                 allentries[i - $config['preload-distance'].to_i].pixbuf_main
1030             end
1031             if i == $config['preload-distance'].to_i * 2 + 1
1032                 #- when we're after double preload distance, activate normal preloading
1033                 $preloader_allowed = true
1034             end
1035             
1036         else
1037             allentries.delete_at(i)
1038         end
1039         $loading_progressbar.fraction = i.to_f / allentries.size
1040         $loading_progressbar.text = utf8(_("Loading... %d%") % (100 * $loading_progressbar.fraction))
1041     end
1042     if i <= $config['preload-distance'].to_i * 2
1043         #- not yet preloaded correctly
1044         $preloader_allowed = true
1045         run_preloader
1046     end
1047     sb_msg(_("%d images of total %s kB loaded in %3.2f seconds.") % [ total_loaded_files, commify(total_loaded_size / 1024), Time.now - t1 ])
1048     $loading_progressbar.hide
1049     $execute.sensitive = true
1050 end
1051
1052 def reset_all
1053     reset_labels
1054     reset_thumbnails
1055     $mainview.set_shown_entry(nil)
1056     sb_msg(nil)
1057     $preloader_allowed = false
1058 end
1059
1060 def open_dir(path)
1061     #- remove visual stuff, so that user will see something is happening
1062     reset_all
1063     sb_msg(_("Scanning source directory..."))
1064     Gtk.main_iteration while Gtk.events_pending?
1065
1066     path = File.expand_path(path.sub(%r|/$|, ''))
1067     examined_dirs = `find '#{path}' -type d -follow`.sort.collect { |v| v.chomp }
1068     #- validate first
1069     examined_dirs.each { |dir|
1070         if dir =~ /'/
1071             return utf8(_("Source directory or sub-directories can't contain a single-quote character, sorry: %s") % dir)
1072         end
1073         Dir.entries(dir).each { |file|
1074             if file =~ /'/ && type = entry2type(file) && type == 'video'
1075                 return utf8(_("Videos can't contain a single quote character ('), sorry: %s") % "#{dir}/#{file}")
1076             end
1077         }
1078     }
1079
1080     #- scan for populate second
1081     examined_dirs.each { |dir|
1082         if File.basename(dir) =~ /^\./
1083             msg 1, _("Ignoring directory %s, begins with a dot (indicating a hidden directory)") % dir
1084             next
1085         end
1086         entries = Dir.entries(dir)
1087         if $sort_by_exif_date
1088             dates = {}
1089             entries.each { |file|
1090                 date_time = Exif.datetimeoriginal(File.join(dir, file))
1091                 if ! date_time.nil?
1092                     dates[file] = date_time
1093                 end
1094             }
1095             entries = smartsort(entries, dates)
1096         else
1097             entries.sort!
1098         end
1099         entries.each { |file|
1100             type = entry2type(file)
1101             if type
1102                 $allentries << Entry.new(File.join(dir, file), type)
1103             end
1104         }
1105     }
1106     $workingdir = path
1107     return nil
1108 end
1109
1110 def open_dir_popup
1111     fc = Gtk::FileChooserDialog.new(utf8(_("Specify the directory to work with")),
1112                                     nil,
1113                                     Gtk::FileChooser::ACTION_SELECT_FOLDER,
1114                                     nil,
1115                                     [Gtk::Stock::OPEN, Gtk::Dialog::RESPONSE_ACCEPT], [Gtk::Stock::CANCEL, Gtk::Dialog::RESPONSE_CANCEL])
1116     fc.transient_for = $main_window
1117     if $workingdir
1118         fc.current_folder = $workingdir
1119     end
1120     ok = false
1121     load = false
1122     while !ok
1123         if fc.run == Gtk::Dialog::RESPONSE_ACCEPT
1124             msg = open_dir(fc.filename)
1125             if msg
1126                 show_popup(fc, msg)
1127                 ok = false
1128             else
1129                 ok = true
1130                 load = true
1131             end
1132         else
1133             ok = true
1134         end
1135     end
1136     fc.destroy
1137     if load
1138         show_entries($allentries)
1139     end
1140 end
1141
1142 def try_quit(*options)
1143     Gtk.main_quit
1144 end
1145
1146 def execute
1147     dialog = Gtk::Dialog.new
1148     dialog.title = utf8(_("Booh message"))
1149
1150     vb1 = Gtk::VBox.new(false, 5)
1151     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!")))
1152     vb1.pack_start(label, false, false)
1153
1154     lastpath = $workingdir
1155
1156     table = Gtk::Table.new(0, 0, false)
1157     table.set_row_spacings(5)
1158     table.set_column_spacings(5)
1159     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)
1160     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)
1161     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)
1162     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)
1163     add_row = proc { |row, name, color, truthproc, normal|
1164         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)),
1165                      0, 1, row, row + 1, Gtk::FILL, Gtk::FILL, 5, 5)
1166         counter = 0
1167         examples = Gtk::HBox.new(false, 5)
1168         $allentries.each { |entry|
1169             if truthproc.call(entry)
1170                 counter += 1
1171                 if counter < 4
1172                     thumbnail = Gtk::Image.new(entry.pixbuf_thumbnail)
1173                     if entry.type == 'video'
1174                         thumbnail = Gtk::HBox.new.pack_start(da1 = Gtk::DrawingArea.new.set_size_request($videoborder_pixbuf.width, -1), false, false).
1175                                                   pack_start(thumbnail).
1176                                                   pack_start(da2 = Gtk::DrawingArea.new.set_size_request($videoborder_pixbuf.width, -1), false, false)
1177                         da1.signal_connect('realize') { da1.window.set_back_pixmap($videoborder_pixmap, false) }
1178                         da2.signal_connect('realize') { da2.window.set_back_pixmap($videoborder_pixmap, false) }
1179                     end
1180                     examples.pack_start(thumbnail, false, false)
1181                 elsif counter == 4
1182                     examples.pack_start(Gtk::Label.new.set_markup("<b>...</b>"), false, false)
1183                 end
1184             end
1185         }
1186         table.attach(Gtk::Label.new(counter.to_s).set_justify(Gtk::Justification::CENTER), 1, 2, row, row + 1, 0, 0, 5, 5)
1187         table.attach(examples, 2, 3, row, row + 1, Gtk::FILL, Gtk::FILL, 5, 5)
1188
1189         combostore = Gtk::ListStore.new(Gdk::Pixbuf, String)
1190         iter = combostore.append
1191         if normal
1192             iter[0] = $main_window.render_icon(Gtk::Stock::GO_FORWARD, Gtk::IconSize::MENU)
1193             iter[1] = utf8(_("Move to:"))
1194             iter = combostore.append
1195             iter[0] = $main_window.render_icon(Gtk::Stock::PASTE, Gtk::IconSize::MENU)
1196             iter[1] = utf8(_("Copy to:"))
1197         else
1198             iter[0] = $main_window.render_icon(Gtk::Stock::DELETE, Gtk::IconSize::MENU)
1199             iter[1] = utf8(_("Permanently remove"))
1200         end
1201         iter = combostore.append
1202         iter[0] = $main_window.render_icon(Gtk::Stock::MEDIA_STOP, Gtk::IconSize::MENU)
1203         iter[1] = utf8(_("Do nothing"))
1204         combo = Gtk::ComboBox.new(combostore)
1205         combo.active = 0
1206         renderer = Gtk::CellRendererPixbuf.new
1207         combo.pack_start(renderer, false)
1208         combo.set_attributes(renderer, :pixbuf => 0)
1209         renderer = Gtk::CellRendererText.new
1210         combo.pack_start(renderer, true)
1211         combo.set_attributes(renderer, :text => 1)
1212
1213         if normal
1214             pathbutton = Gtk::Button.new.add(pathlabel = Gtk::Label.new.set_markup(utf8(_("<i>(unset)</i>"))))
1215             pathbutton.signal_connect('clicked') {
1216                 fc = Gtk::FileChooserDialog.new(utf8(_("Specify the directory where to move the pictures to")),
1217                                                 nil,
1218                                                 Gtk::FileChooser::ACTION_SELECT_FOLDER,
1219                                                 nil,
1220                                                 [Gtk::Stock::OPEN, Gtk::Dialog::RESPONSE_ACCEPT], [Gtk::Stock::CANCEL, Gtk::Dialog::RESPONSE_CANCEL])
1221                 fc.transient_for = dialog
1222                 fc.current_folder = lastpath
1223                 if fc.run == Gtk::Dialog::RESPONSE_ACCEPT
1224                     pathlabel.text = fc.filename
1225                     pathlabel.set_alignment(0, 0.5)
1226                 end
1227                 lastpath = fc.filename
1228                 fc.destroy
1229             }
1230             combo.signal_connect('changed') {
1231                 pathbutton.sensitive = combo.active <= 1
1232             }
1233             vb = Gtk::VBox.new(false, 5)
1234             vb.pack_start(combo, false, false)
1235             vb.pack_start(pathbutton, false, false)
1236             table.attach(Gtk::Alignment.new(0, 0.5, 1, 0).add(vb), 3, 4, row, row + 1, Gtk::FILL, Gtk::FILL, 5, 5)
1237             { :combo => combo, :pathlabel => pathlabel }
1238         else
1239             table.attach(Gtk::Alignment.new(0, 0.5, 1, 0).add(combo), 3, 4, row, row + 1, Gtk::FILL, Gtk::FILL, 5, 5)
1240             { :combo => combo }
1241         end
1242     }
1243     stuff = {}
1244     stuff['toremove'] = add_row.call(1, utf8(_("<i>to remove</i>")), $color_red, proc { |entry| entry.removed }, false)
1245     $ordered_labels.each_with_index { |label, row| stuff[label] = add_row.call(row + 2, label.name, label.color, proc { |entry| entry.labeled == label }, true) }
1246     vb1.pack_start(sw = Gtk::ScrolledWindow.new(nil, nil).add_with_viewport(table).set_policy(Gtk::POLICY_AUTOMATIC, Gtk::POLICY_AUTOMATIC), true, true)
1247
1248     toremove_amount = $allentries.find_all { |entry| entry.removed }.size
1249     toremove_size = commify($allentries.find_all { |entry| entry.removed }.collect { |entry| file_size(entry.path) }.sum / 1024)
1250     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 ]))
1251     if toremove_amount > 0
1252         vb1.pack_start(check_removal, false, false)
1253         stuff['toremove'][:combo].signal_connect('changed') { |widget|
1254             check_removal.sensitive = widget.active == 0
1255         }
1256     end
1257
1258     dialog.vbox.add(vb1)
1259
1260     dialog.set_default_size(800, 600)
1261     dialog.add_button(Gtk::Stock::CANCEL, Gtk::Dialog::RESPONSE_CANCEL)
1262     dialog.add_button(Gtk::Stock::OK, Gtk::Dialog::RESPONSE_OK)
1263     dialog.window_position = Gtk::Window::POS_MOUSE
1264     dialog.transient_for = $main_window
1265
1266     dialog.show_all
1267
1268     while true
1269         dialog.run { |response|        
1270             if response == Gtk::Dialog::RESPONSE_OK
1271                 if toremove_amount > 0 && ! check_removal.active? && stuff['toremove'][:combo].active == 0
1272                     show_popup(dialog, utf8(_("You have not confirmed that you noticed the permanent removal of the pictures marked for deletion.")))
1273                     break
1274                 end
1275                 problem = false
1276                 label2entries = {}
1277                 $labels.values.each { |label| label2entries[label] = [] }
1278                 $allentries.each { |entry| entry.labeled and label2entries[entry.labeled] << entry }
1279                 stuff.keys.each { |key|
1280                     if key.is_a?(Label) && stuff[key][:combo].active <= 1
1281                         destination = stuff[key][:pathlabel].text
1282                         if destination[0] != ?/
1283                             show_popup(dialog, utf8(_("You have not selected a directory where to move/copy %s.") % key.name))
1284                             problem = true
1285                             break
1286                         end
1287                         begin
1288                             Dir.mkdir(destination)
1289                         rescue Errno::EEXIST
1290                         end
1291                         begin
1292                             st = File.stat(destination)
1293                         rescue
1294                             show_popup(dialog, utf8(_("Directory %s, where to move/copy %s, is not valid or not createable.") % [destination, key.name]))
1295                             problem = true
1296                             break
1297                         end
1298                         if ! st.directory? || ! st.writable?
1299                             show_popup(dialog, utf8(_("Directory %s, where to move/copy %s, is not valid or not writable.") % [destination, key.name]))
1300                             problem = true
1301                             break
1302                         end
1303                         label2entries[key].each { |entry|
1304                             begin
1305                                 File.stat(File.join(destination, File.basename(entry.path)))
1306                                 show_popup(dialog, utf8(_("Sorry, a file '%s' already exists in directory '%s'.") % [ File.basename(entry.path), destination ]))
1307                                 problem = true
1308                                 break
1309                             rescue
1310                             end
1311                         }
1312                         if problem
1313                             break
1314                         end
1315                     end
1316                 }
1317                 if ! problem
1318                     begin
1319                         moved = 0
1320                         copied = 0
1321                         stuff.keys.each { |key|
1322                             if key.is_a?(Label) && stuff[key][:combo].active <= 1
1323                                 destination = stuff[key][:pathlabel].text
1324                                 label2entries[key].each { |entry|
1325                                     if stuff[key][:combo].active == 0
1326                                         File.rename(entry.path, File.join(destination, File.basename(entry.path)))
1327                                         moved += 1
1328                                     elsif stuff[key][:combo].active == 1
1329                                         system("cp -dp '#{entry.path}' '#{destination}'")
1330                                         copied += 1
1331                                     end
1332                                 }
1333                             end
1334                         }
1335                         removed = 0
1336                         if stuff['toremove'][:combo].active == 0
1337                             $allentries.each { |entry|
1338                                 if entry.removed
1339                                     File.delete(entry.path)
1340                                     removed += 1
1341                                 end
1342                             }
1343                         end
1344                     rescue
1345                         msg 1, "woops: #{$!}"
1346                         show_popup(dialog, utf8(_("Unexpected system call error: '%s'.") % $!))
1347                     end
1348                     show_popup(dialog, utf8(_("Successfully moved %d files, copied %d file, and removed %d files.") % [ moved, copied, removed ]))
1349                     dialog.destroy
1350                     reset_all
1351                     return
1352                 end
1353
1354             else
1355                 dialog.destroy
1356                 return
1357             end
1358         }
1359     end
1360 end
1361
1362 def update_visibility(entry)
1363     if ! entry.button
1364         #- not yet loaded
1365         return
1366     end
1367     if entry.labeled
1368         if entry.labeled.button.active?
1369             entry.button.show
1370         else
1371             entry.button.hide
1372         end
1373     elsif entry.removed
1374         if $toremove_button.active?
1375             entry.button.show
1376         else
1377             entry.button.hide
1378         end
1379     else
1380         if $unlabelled_button.active?
1381             entry.button.show
1382         else
1383             entry.button.hide
1384         end
1385     end
1386 end
1387         
1388 def update_all_visibilities
1389     $allentries.each { |entry|
1390         update_visibility(entry)
1391     }
1392     shown = $mainview.get_shown_entry
1393     if shown.nil?
1394         return
1395     end
1396     while shown.button && ! shown.button.visible? && shown != $allentries.last
1397         shown = $allentries[$allentries.index(shown) + 1]
1398     end 
1399     if shown.button && shown.button.visible?
1400         shown.button.grab_focus
1401         return
1402     end
1403     $allentries.reverse.each { |entry|
1404         if entry.button && entry.button.visible?
1405             entry.button.grab_focus
1406             return
1407         end
1408     }
1409 end
1410
1411 def preferences
1412     dialog = Gtk::Dialog.new(utf8(_("Edit preferences")),
1413                              $main_window,
1414                              Gtk::Dialog::MODAL | Gtk::Dialog::DESTROY_WITH_PARENT,
1415                              [Gtk::Stock::OK, Gtk::Dialog::RESPONSE_OK],
1416                              [Gtk::Stock::CANCEL, Gtk::Dialog::RESPONSE_CANCEL])
1417
1418     tooltips = Gtk::Tooltips.new
1419     
1420     dialog.vbox.add(tbl = Gtk::Table.new(0, 0, false))
1421     tbl.attach(Gtk::Alignment.new(1, 0.5, 0, 0).add(Gtk::Label.new.set_markup(utf8(_("Command for watching videos: ")))),
1422                0, 1, 0, 1, Gtk::FILL, Gtk::SHRINK, 2, 2)
1423     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)),
1424                1, 2, 0, 1, Gtk::FILL, Gtk::SHRINK, 2, 2)
1425     tooltips.set_tip(video_viewer_entry, utf8(_("Use %f to specify the filename;\nfor example: /usr/bin/mplayer %f")), nil)
1426
1427     tbl.attach(Gtk::Alignment.new(1, 0.5, 0, 0).add(Gtk::Label.new.set_markup(utf8(_("Browser's command: ")))),
1428                0, 1, 1, 2, Gtk::FILL, Gtk::SHRINK, 2, 2)
1429     tbl.attach(Gtk::Alignment.new(0, 0.5, 1, 0).add(browser_entry = Gtk::Entry.new.set_text($config['browser'])),
1430                1, 2, 1, 2, Gtk::FILL, Gtk::SHRINK, 2, 2)
1431     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)
1432
1433     tbl.attach(Gtk::Alignment.new(1, 0.5, 0, 0).add(Gtk::Label.new.set_markup(utf8(_("Preloading distance: ")))),
1434                0, 1, 2, 3, Gtk::FILL, Gtk::SHRINK, 2, 2)
1435     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)),
1436                1, 2, 2, 3, Gtk::FILL, Gtk::SHRINK, 2, 2)
1437     tooltips.set_tip(preload_distance, utf8(_("Amount of pictures preloaded left and right to the currently shown")), nil)
1438
1439     tbl.attach(Gtk::Alignment.new(1, 0.5, 0, 0).add(Gtk::Label.new.set_markup(utf8(_("Cache memory use: ")))),
1440                0, 1, 3, 4, Gtk::FILL, Gtk::SHRINK, 2, 2)
1441     tbl.attach(Gtk::Alignment.new(0, 0.5, 1, 0).add(cache_vbox = Gtk::VBox.new(false, 0)),
1442                1, 2, 3, 4, Gtk::FILL, Gtk::SHRINK, 2, 2)
1443     cache_vbox.pack_start(Gtk::HBox.new(false, 0).pack_start(cache_memfree_radio = Gtk::RadioButton.new(''), false, false).
1444                                                   pack_start(cache_memfree_spin = Gtk::SpinButton.new(0, 100, 10), false, false).
1445                                                   pack_start(cache_memfree_label = Gtk::Label.new(utf8(_("% of free memory"))), false, false), false, false)
1446     cache_memfree_spin.signal_connect('value-changed') { cache_memfree_radio.active = true }
1447     tooltips.set_tip(cache_memfree_spin, utf8(_("Percentage of free memory (+ buffers/cache) measured at startup")), nil)
1448     cache_vbox.pack_start(Gtk::HBox.new(false, 0).pack_start(cache_specify_radio = Gtk::RadioButton.new(cache_memfree_radio, ''), false, false).
1449                                                   pack_start(cache_specify_spin = Gtk::SpinButton.new(0, 4000, 50), false, false).
1450                                                   pack_start(cache_specify_label = Gtk::Label.new(utf8(_("MB"))).set_sensitive(false), false, false), false, false)
1451     cache_specify_spin.signal_connect('value-changed') { cache_specify_radio.active = true }
1452     cache_memfree_radio.signal_connect('toggled') {
1453         if cache_memfree_radio.active?
1454             cache_memfree_label.sensitive = true
1455             cache_specify_label.sensitive = false
1456         else
1457             cache_specify_label.sensitive = true
1458             cache_memfree_label.sensitive = false
1459         end
1460     }
1461     tooltips.set_tip(cache_specify_spin, utf8(_("Amount of memory in megabytes")), nil)
1462     if $config['cache-memory-use'] =~ /memfree_(\d+)/
1463         cache_memfree_spin.value = $1.to_i
1464     else
1465         cache_specify_spin.value = $config['cache-memory-use'].to_i
1466     end
1467
1468     tbl.attach(update_exif_orientation_check = Gtk::CheckButton.new(utf8(_("Update file's EXIF orientation when rotating a picture"))),
1469                0, 2, 4, 5, Gtk::FILL, Gtk::SHRINK, 2, 2)
1470     tooltips.set_tip(update_exif_orientation_check, utf8(_("When rotating a picture (Alt-Right/Left), also update EXIF orientation in the file itself")), nil)
1471     update_exif_orientation_check.active = $config['rotate-set-exif'] == 'true'
1472
1473     dialog.vbox.show_all
1474     dialog.run { |response|
1475         if response == Gtk::Dialog::RESPONSE_OK
1476             $config['video-viewer'] = from_utf8(video_viewer_entry.text)
1477             $config['browser'] = from_utf8(browser_entry.text)
1478             $config['preload-distance'] = preload_distance.value
1479             $config['rotate-set-exif'] = update_exif_orientation_check.active?.to_s
1480             if cache_memfree_radio.active?
1481                 $config['cache-memory-use'] = "memfree_#{cache_memfree_spin.value}%"
1482             else
1483                 $config['cache-memory-use'] = cache_specify_spin.value
1484             end
1485             set_cache_memory_use_figure
1486         end
1487     }
1488     dialog.destroy
1489 end
1490
1491 def perform_undo
1492     if $undo_mb.sensitive?
1493         $redo_mb.sensitive = true
1494         if not more_undoes = UndoHandler.undo($statusbar)
1495             $undo_mb.sensitive = false
1496         end
1497     end
1498 end
1499
1500 def perform_redo
1501     if $redo_mb.sensitive?
1502         $undo_mb.sensitive = true
1503         if not more_redoes = UndoHandler.redo($statusbar)
1504             $redo_mb.sensitive = false
1505         end
1506     end
1507 end
1508
1509 def create_menubar    
1510     #- menu
1511     mb = Gtk::MenuBar.new
1512
1513     filemenu = Gtk::MenuItem.new(utf8(_("_File")))
1514     filesubmenu = Gtk::Menu.new
1515     filesubmenu.append(open      = Gtk::ImageMenuItem.new(Gtk::Stock::OPEN))
1516     filesubmenu.append(            Gtk::SeparatorMenuItem.new)
1517     filesubmenu.append($execute  = Gtk::ImageMenuItem.new(Gtk::Stock::EXECUTE).set_sensitive(false))
1518     filesubmenu.append(            Gtk::SeparatorMenuItem.new)
1519     filesubmenu.append(quit      = Gtk::ImageMenuItem.new(Gtk::Stock::QUIT))
1520     filemenu.set_submenu(filesubmenu)
1521     mb.append(filemenu)
1522
1523     open.signal_connect('activate') { open_dir_popup }
1524     $execute.signal_connect('activate') { execute }
1525     quit.signal_connect('activate') { try_quit }
1526
1527     editmenu = Gtk::MenuItem.new(utf8(_("_Edit")))
1528     editsubmenu = Gtk::Menu.new
1529     editsubmenu.append($undo_mb    = Gtk::ImageMenuItem.new(Gtk::Stock::UNDO).set_sensitive(false))
1530     editsubmenu.append($redo_mb    = Gtk::ImageMenuItem.new(Gtk::Stock::REDO).set_sensitive(false))
1531     editsubmenu.append(              Gtk::SeparatorMenuItem.new)
1532     editsubmenu.append(prefs       = Gtk::ImageMenuItem.new(Gtk::Stock::PREFERENCES))
1533     editmenu.set_submenu(editsubmenu)
1534     mb.append(editmenu)
1535
1536     $undo_mb.signal_connect('activate') { perform_undo }
1537     $redo_mb.signal_connect('activate') { perform_redo }
1538     prefs.signal_connect('activate') { preferences }
1539     
1540     helpmenu = Gtk::MenuItem.new(utf8(_("_Help")))
1541     helpsubmenu = Gtk::Menu.new
1542     helpsubmenu.append(howto = Gtk::ImageMenuItem.new(Gtk::Stock::HELP))
1543     helpsubmenu.append(speed = Gtk::ImageMenuItem.new(utf8(_("Speedup: key shortcuts"))))
1544     speed.image = Gtk::Image.new("#{$FPATH}/images/stock-info-16.png")
1545     helpsubmenu.append(tutos = Gtk::ImageMenuItem.new(utf8(_("Online tutorials (opens a web-browser)"))))
1546     tutos.image = Gtk::Image.new("#{$FPATH}/images/stock-web-16.png")
1547     helpsubmenu.append(Gtk::SeparatorMenuItem.new)
1548     helpsubmenu.append(about = Gtk::ImageMenuItem.new(Gtk::Stock::ABOUT))
1549     helpmenu.set_submenu(helpsubmenu)
1550     mb.append(helpmenu)
1551
1552     howto.signal_connect('activate') {
1553         show_popup($main_window, utf8(_("<span size='large' weight='bold'>Help</span>
1554
1555 1. Open a directory with <span foreground='darkblue'>File/Open</span>; the classifier will scan it (including subdirectories) and
1556 show thumbnails for all images and videos at the bottom.
1557
1558 2. You can then navigate through images with the <span foreground='darkblue'>Left/Right</span> keyboard keys, or by <span foreground='darkblue'>clicking</span>
1559 on thumbnails.
1560
1561 3. You may associate a <span foreground='darkblue'>label</span> to each thumbnail. Either hit the <span foreground='darkblue'>Delete</span> key to associate
1562 the built-in <i>to remove</i> label, or hit any alphabetical key to associate a label you define.
1563 The first time you hit a key without any label associated, a popup will ask for the full
1564 name of this label, and what color you want. To clear the current label, hit the <span foreground='darkblue'>Space</span> key.
1565
1566 4. To help you better view what thumbnails are associated to your labels, you may <span foreground='darkblue'>hide</span>
1567 some of them by unchecking the labels checkboxes on the left.
1568
1569 5. Once you're finished reviewing all thumbnails, use <span foreground='darkblue'>File/Execute</span> to execute the desired
1570 actions according to associated labels. You can permanently remove (or not) images with
1571 the <i>to remove</i> label, and copy or move images with the labels you defined.
1572 ")), { :pos_centered => true, :not_transient => true })
1573     }
1574     speed.signal_connect('activate') {
1575         show_popup($main_window, utf8(_("<span size='large' weight='bold'>Key shortcuts</span>
1576
1577 <span foreground='darkblue'>Left/Right</span>: move left and right in images
1578 <span foreground='darkblue'>Enter</span>: 'view' current image: for images, display EXIF data; for videos, play it
1579 <span foreground='darkblue'>Alt-Left/Right</span>: rotate current image clockwise/counter-clockwise
1580 <span foreground='darkblue'>Delete</span>: assign the 'to remove' label on current image
1581 <span foreground='darkblue'>Space</span>: clear any label on current image
1582 <span foreground='darkblue'>Control-z</span>: undo
1583 <span foreground='darkblue'>Control-r</span>: redo
1584
1585 Any alphabetical key will assign (or popup for) the associated label on current image.
1586 ")), { :pos_centered => true, :not_transient => true })
1587     }
1588     tutos.signal_connect('activate') { open_url('http://booh.org/tutorial.html') }
1589     about.signal_connect('activate') { call_about }
1590
1591
1592     #- no toolbar, to save height
1593
1594     return mb
1595 end
1596
1597 def reset_labels
1598     for child in $labels_vbox.children
1599         $labels_vbox.remove(child)
1600     end
1601     $labels_vbox.pack_start(Gtk::Label.new(utf8(_("Labels list:"))).set_justify(Gtk::Justification::CENTER), false, false).show_all
1602     $labels = {}
1603     $ordered_labels = []
1604     lbl = Gtk::Label.new.set_markup(utf8(_("<i>unlabelled</i>")))
1605     $labels_vbox.pack_start($unlabelled_button = Gtk::CheckButton.new.add(Gtk::EventBox.new.add(lbl)).show_all)
1606     $unlabelled_button.active = true
1607     $unlabelled_button.signal_connect('toggled') { update_all_visibilities }
1608     lbl = Gtk::Label.new.set_markup(utf8(_("<i>to remove</i>")))
1609     $labels_vbox.pack_start($toremove_button = Gtk::CheckButton.new.add(evt = Gtk::EventBox.new.add(lbl)).show_all)
1610     $toremove_button.active = true
1611     $toremove_button.signal_connect('toggled') { update_all_visibilities }
1612     evt.modify_bg(Gtk::StateType::NORMAL, $color_red)
1613     evt.modify_bg(Gtk::StateType::PRELIGHT, $color_red.lighter.lighter)
1614     evt.modify_bg(Gtk::StateType::ACTIVE, $color_red.lighter)
1615 end
1616
1617 def reset_thumbnails
1618     $allentries = []
1619     if $preloader_running
1620         $preloader_force_exit = true
1621     end
1622     for child in $imagesline.children
1623         $imagesline.remove(child)
1624     end
1625 end
1626
1627 def create_main_window
1628
1629     $videoborder_pixbuf = Gdk::Pixbuf.new("#{$FPATH}/images/video_border.png")
1630     $videoborder_pixmap, = $videoborder_pixbuf.render_pixmap_and_mask(0)
1631
1632     mb = create_menubar
1633
1634     main_vbox = Gtk::VBox.new(false, 0)
1635     main_vbox.pack_start(mb, false, false)
1636     mainview_hbox = Gtk::HBox.new
1637     mainview_hbox.pack_start(Gtk::Alignment.new(0.5, 0, 1, 1).add(left_vbox = Gtk::VBox.new(false, 5)), false, true)
1638     left_vbox.pack_start(($labels_vbox = Gtk::VBox.new(false, 5)), false, true)
1639     left_vbox.pack_end($loading_progressbar = Gtk::ProgressBar.new.set_text(utf8(_("Loading... %d%") % 0)), false, true)
1640     mainview_hbox.pack_start($mainview = MainView.new, true, true)
1641     main_vbox.pack_start(mainview_hbox, true, true)
1642     $imagesline_sw = Gtk::ScrolledWindow.new(nil, nil)
1643     $imagesline_sw.set_policy(Gtk::POLICY_ALWAYS, Gtk::POLICY_NEVER)
1644     $imagesline_sw.add_with_viewport($imagesline = Gtk::HBox.new(false, 0).show)
1645     main_vbox.pack_start($imagesline_sw, false, false)
1646     main_vbox.pack_end($statusbar = Gtk::Statusbar.new, false, false)
1647
1648     $imagesline.set_size_request(-1, Gtk::Button.new.size_request[1] + Entry.thumbnails_height + 15)
1649
1650     $main_window = Gtk::Window.new
1651     $main_window.add(main_vbox)
1652     $main_window.signal_connect('delete-event') {
1653         try_quit({ :disallow_cancel => true })
1654     }
1655
1656     #- read/save size and position of window
1657     if $config['pos-x'] && $config['pos-y']
1658         $main_window.move($config['pos-x'].to_i, $config['pos-y'].to_i)
1659     else
1660         $main_window.window_position = Gtk::Window::POS_CENTER
1661     end
1662     msg 3, "size: #{$config['width']}x#{$config['height']}"
1663     $main_window.set_default_size(($config['width'] || 700).to_i, ($config['height'] || 600).to_i)
1664     $main_window.signal_connect('configure-event') {
1665         msg 3, "configure: pos: #{$main_window.window.root_origin.inspect} size: #{$main_window.window.size.inspect}"
1666         x, y = $main_window.window.root_origin
1667         width, height = $main_window.window.size
1668         $config['pos-x'] = x
1669         $config['pos-y'] = y
1670         $config['width'] = width
1671         $config['height'] = height
1672         false
1673     }
1674
1675     $main_window.show_all
1676     $loading_progressbar.hide
1677 end
1678
1679
1680 handle_options
1681 read_config
1682 Gtk.init
1683
1684
1685 create_main_window
1686 check_config
1687
1688 if ARGV[0]
1689     if msg = open_dir(ARGV[0])
1690         puts msg
1691     else
1692         show_entries($allentries)
1693     end
1694 end
1695 Gtk.main
1696
1697 write_config