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