#!/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 move_pics {
   my ($self, $spec, $specdir) = @_;
   return unless defined $ARGV[1];

   # dirname: 2nd program argument + category + name from analyze (in the list)
   my $dirname = $ARGV[1] . '/'.$self->{dirlbl}->get_text;

   # only add the special when ctrl is pressed
   if ((not defined $specdir) and $spec) {
      my @s = $self->{list}->get_selected_indices;
      unless (defined $s[0]) {
         print "nothing selected!\n";
         return;
      }
      my $d = $self->{list}->{data}->[$s[0]]->[1];
      unless (defined $d) {
         print "no such name on idx $s[0]\n";
      }
      mkdir $dirname;
      $dirname .= '/'.$d;
   } elsif (defined $specdir) {
      $dirname .= '/'.$specdir;
   }
   print "dirname: '$dirname'\n";
   mkdir $dirname;
   $self->{schnauzer}->movesel ($dirname);
}

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

   my $box = new Gtk2::HBox;
   $self->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->{schnauzer}->{draw}->signal_connect (key_press_event => sub {
   #   my $key = $_[1]->keyval;
   #   my $state = $_[1]->state;
   #
   #  my $ctrl = grep $_ eq "control-mask", @{$_[1]->state};
   #   if ($ctrl && $key == $Gtk2::Gdk::Keysyms{c}) {
   #      $self->move_pics (0, $self->{direntry}->get_text())
   #         if ($self->{direntry}->get_text() ne "")
   #   } else {
   #      return 0;
   #   }
   #   return 1;
   #});

   $self->{schnauzer}->signal_connect (activate => sub { $viewer->load_image ($_[1]) });

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

   $self->{list}->signal_connect (key_press_event => sub {
      my $key = $_[1]->keyval;
      my $state = $_[1]->state;

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

      if ($key == $Gtk2::Gdk::Keysyms{t}) {
         $self->cycle_move_dir_idx;
      } elsif ($key == $Gtk2::Gdk::Keysyms{m}) {
         $self->move_pics ($ctrl);
      } elsif ($key == $Gtk2::Gdk::Keysyms{c}) {
         $self->move_pics (0, $self->{direntry}->get_text())
            if ($self->{direntry}->get_text() ne "")
      } elsif ($key == $Gtk2::Gdk::Keysyms{Up}) {
         return 0;
      } elsif ($key == $Gtk2::Gdk::Keysyms{Down}) {
         return 0;
      } else {
         $self->{schnauzer}->handle_key ($key, $state);
      }
      return 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_paths (
         [map "$self->{path}/$_", @{$k->[1]}],
      );

      1;

   });
}

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

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

   $self->{path} = $path;

   my @files = map {
                     $_ = Glib::filename_to_unicode $_;
                     my $str = $_;
                     $str =~ s/[\-_ ]+/ /g;
                     $str =~ s/\.[^\.]+$//g;
                     $str =~ s/(?<=[0-9 ])(?:fc|bc|front|back|cover)$//;
                     [$str, $_]
                   }
                   grep !/\.(sfv|crc|par|par2)$/, readdir $dir;

   my %cluster;

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

   my %chk_files;
   $chk_files{$_->[1]} = 1 for @files;
   for my $info (@regexps) {
      my $re = $info->[0];
      for (@files) {
         if ($_->[0] =~ $re && $_->[0] =~ $info->[1]) {
            push @{ ($cluster{$1} ||= [ $info, []])->[1] }, $_->[1];
            delete $chk_files{$_->[1]};
           # print "PUSH $1 : $_->[1] $_->[0] $re $info->[1] \n";
         }
      }
   }
   #push @{$cluster{"THE REST"}->[1]}, $_ for keys %chk_files;

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

   my @rest;
   while (my ($k, $v) = each %cluster) {
      my $n = scalar @{$v->[1]};
      push @{$self->{list}{data}}, [$n, $k] if $n > 2;
      if ($n <= 2) { 
         push @rest, $k 
      }
   }
   $cluster{"THE REST"} = [ [], [] ];
   for my $r (@rest) {
      push @{$cluster{"THE REST"}->[1]}, $_ for @{$cluster{$r}->[1]};
      delete $cluster{$_};
   }
   push @{$cluster{"THE REST"}->[1]}, $_ for keys %chk_files;
   push @{$self->{list}{data}}, [(scalar @{$cluster{"THE REST"}->[1]}), "THE REST"];
}

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);

   $s->set_dir (".");
   $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 ($ARGV[0]);
$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;