#!/opt/bin/perl

use Gtk2 -init;
use Gtk2::Gdk::Keysyms;

use Gtk2::CV::ImageWindow;
use Gtk2::CV::Schnauzer;

$VERSION = 0.1;

my $viewer;

package cluster;

use Glib::Object::Subclass Gtk2::Window;

use Gtk2::SimpleList;

sub INIT_INSTANCE {
   my ($self) = @_;
   
   $self->set_default_size (600, 400);

   my $vbox = new Gtk2::VBox;
   $self->add ($vbox);

   my $box = new Gtk2::HBox;
   $vbox->add ($box);
   $box->add (my $sw = new Gtk2::ScrolledWindow);
   $sw->add (
      $self->{list} = new Gtk2::SimpleList
         "#"    => "int",
         "Name" => "text",
   );

   $box->add ($self->{schnauzer} = new Gtk2::CV::Schnauzer);

   $self->{list}->get_column(0)->set_sort_column_id(0);
   $self->{list}->get_column(1)->set_sort_column_id(1);

   $self->{list}->signal_connect (cursor_changed => sub {
      my $row = scalar +($_[0]->get_selection->get_selected_rows)[0]->get_indices;

      my $k = $_[0]{data}[$row][1];
      $k = $self->{cluster}{$k};

      $self->{schnauzer}->set_files (
         [sort map "$self->{path}/$_", @{$k->[1]}],
      );

      1;

   });
}

sub analyse {
   my ($self, $path) = @_;

   opendir my $dir, $path
      or die "$path: $!";

   $self->{path} = $path;

   my @files = map {
                     my $str = $_;
                     $str =~ s/[\-_ ]+/ /g;
                     $str =~ s/\.[^\.]+//g;
                     [$str, $_]
                   }
                   grep !/\.(sfv|crc|par|par2)$/, readdir $dir;

   my %cluster;

   my @regexps = (
      [qr<\d+$>, qr<^(.+?)\ *(\d+)$>],
      [qr<\ \d+$>, qr<^(.+?)(\d+)\ +(\d+)$>],
   );

   for my $info (@regexps) {
      my $re = $info->[0];
      for (@files) {
         if ($_->[0] =~ $re && $_->[0] =~ $info->[1]) {
            push @{ ($cluster{$1} ||= [ $info, []])->[1] }, $_->[1];
         }
      }
   }

   $self->{cluster} = \%cluster;

   while (my ($k, $v) = each %cluster) {
      my $n = scalar @{$v->[1]};
      push @{$self->{list}{data}}, [$n, $k] if $n > 2;
   }
}

package main;

sub new_schnauzer {
   my $w = new Gtk2::Window;
   $w->add (my $s = new Gtk2::CV::Schnauzer);

   $s->signal_connect (activate => sub {
      $viewer->load_image ($_[1]);
   });
   $s->signal_connect (key_press_event => \&std_keys);
   $w->signal_connect (delete_event => sub { $_[0]->destroy });
   $s->signal_connect (destroy => sub { %{$_[0]} = () });

   $s->set_dir ("/root/pix");
   $w->show_all;

   $w;
}

sub std_keys {
   my $key = $_[1]->keyval;

   my $ctrl = grep $_ eq "control-mask", @{$_[1]->state};

   if ($key == $Gtk2::Gdk::Keysyms{q}) {
      main_quit Gtk2;
   } elsif ($ctrl && $key == $Gtk2::Gdk::Keysyms{v}) {
      new_schnauzer;
   } else {
      return 0;
   }

   1;
}

$viewer = new Gtk2::CV::ImageWindow;
$viewer->signal_connect (key_press_event => \&std_keys);
$viewer->signal_connect (delete_event => sub { main_quit Gtk2 });
$viewer->show_all;

my $cluster = new cluster;

#$cluster->analyse ("/fs/samsung/store-anime-done");
#$cluster->show_all;

#$viewer->set (path => "/fs/samsung/store-anime-done/yuumi_kazuaki_-_love_to_hajieki_to_sayonara_to_-_150.jpg");
#$viewer->set (path => "/root/pix/kvvz.jpg");

main Gtk2;