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