use strict;
use warnings;
use File::Find;
use File::stat;
use Tk;

my @wanted_count;   #sizes of desired file types
my $other_count;    #size of every other file type

my $canvas_w = 300; #width of display
my $canvas_h = 300; #height of display
my $radius = 100;   #radius of pie chart
my $map = 15;       #size of key map image
my @color = qw(#FF0000 #00FF00 #0000FF #FF00FF #00FFFF #FFFF00);

if ($#ARGV < 1){
  print "Usage:$0 <initial dir> <list of extentions>\n";
  print "If want to see *.mp3 files, then make extention mp3\n";
  print "So far, this program can handle up to 5 extentions\n";
}

foreach (@ARGV){
  push(@wanted_count, 0); #init each file type
}


my $mw = MainWindow->new(-title=>"File Monitor",
                         -width=>$canvas_w+180,
                         -height=>$canvas_h+10);
my $canvas = $mw->Canvas(-width=>$canvas_w,
                         -height=>$canvas_h,
                         -relief=>"sunken",
                         -borderwidth=>2)->place(-x=>2,-y=>2);

finddepth(\&wanted, shift @ARGV);
my $total = sum($other_count, @wanted_count);
my $last = 0;
for (my $i=0; $i<$#wanted_count; $i++){
  lookup($last, $i, $ARGV[$i], $wanted_count[$i]);
  $last += $wanted_count[$i]/$total*360;
}
lookup($last, $#ARGV+1, "Other", $other_count);

MainLoop;

sub lookup{
  my ($start, $i, $caption, $value) = @_;
  $canvas->createArc($canvas_w/2-$radius,$canvas_h/2-$radius,$canvas_w/2+$radius,$canvas_h/2+$radius,
                     -start=>$start,
                     -extent=>$value/$total*359.9,
                     -fill=>$color[$i]);
  my $img = $mw->Photo('img' . $i,
		       -width=>$map,
		       -height=>$map);
  $img->put($color[$i], -to=>0,0,$map,$map);
  $mw->Label(-image=>'img' . $i,
	     -relief=>"flat")->place(-x=>$canvas_w+10, -y=>$i*25);
  $mw->Label(-font=>"Arial 8",
	     -text=>$caption . value($value))->place(-x=>$canvas_w+10+1.25*$map, -y=>$i*25);
}

sub value {
  my $out = " (";
  my $order = 0;
  while ($_[0] > 1024){
    $_[0] /= 1024;
    $order++;
  }
  $out = sprintf("%s %.3f ",$out,$_[0]);
  $out .= do {
    (!--$order) ? "k" :
      (!--$order) ? "M" :
	(!--$order) ? "G" : "";
  };
  return $out . "B)";
}

sub sum{
  my ($c,@d) = @_;
  foreach(@d){
    $c += $_;
  }
  return $c;
}


sub wanted{
  if (-d $File::Find::name){ return; }
  if (!-f $File::Find::name){
    print "$File::Find::name does not exist\n";
    return;
  }
  for(my $i=0;$i<=$#ARGV;$i++){
    if (/\.$ARGV[$i]$/){
      $wanted_count[$i] += stat($File::Find::name)->size;
      return;
    }
  }
  $other_count += stat($File::Find::name)->size;
}