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