#!/usr/local/bin/perl

# thumb 0.22
# created by: anders wisch
# modified by: matt hine



use Image::Magick;

use POSIX qw( ceil );



# usage information.

$usage = "usage: thumb [-hrti] [directory]\n";


# help information.

$help  = "";
$help = join '', $help, "usage: thumb [-hrti] [directory]\n";
$help = join '', $help, "\n";
$help = join '', $help, "  -h (display this help message)\n";
$help = join '', $help, "  -r (recurse through directories, starting with . -- currently broken)\n";
$help = join '', $help, "  -i (generate indexes only)\n";
$help = join '', $help, "  -t (generate thumbnails only)\n";
$help = join '', $help, "\n";
$help = join '', $help, "\n";
$help = join '', $help, "in template.html for each thumbnail directory/page:\n";
$help = join '', $help, "\n";
$help = join '', $help, "<!thumb\n";
$help = join '', $help, "  indexname=[string] (may be omitted; default is 'index.html'\n";
$help = join '', $help, "  suffix=[string] (may be omitted; default is '_')\n";
$help = join '', $help, "  target=[string|\"javascript\"] (may be omitted; default is 'window')\n";
$help = join '', $help, "      (\"javascript\" opens target=_blank if browser doesn't do javascript)\n";
$help = join '', $help, "\n";
$help = join '', $help, "  thumbnailwidth=[number] (may be omitted)\n";
$help = join '', $help, "  thumbnailheight=[number] (may be omitted)\n";
$help = join '', $help, "  pixelarea=[number] (may be omitted; default is 10000)\n";
$help = join '', $help, "  thumbnailscale=[number] (may be omitted)\n";
$help = join '', $help, "\n";
$help = join '', $help, "  cellpadding=[number] (may be omitted; default is 10)\n";
$help = join '', $help, "  thumbnailborder=[number] (may be omitted; default is 5)\n";
$help = join '', $help, "  columns=[number] (may be omitted; default is 5)\n";
$help = join '', $help, "  tablebgcolor=[hex color] (may be omitted; default is no color)\n";
$help = join '', $help, ">\n";
$help = join '', $help, "\n";


# flags are assigned default values.

$indexonly = 0;

$thumbnailsonly = 0;

$recursive = 0;



# default root directory (wherever the script was called).

$root = '.';



# parse the command line arguments.

die $usage if ( @ARGV > 3 );

foreach (@ARGV)

{

	if ( /^-/ )

	{

		for( split //, $' )

		{

			if ( /h/ ) { die $help; }

			elsif ( /i/ ) { $indexonly = 1; next; }

			elsif ( /t/ ) { $thumbnailsonly = 1; next; }

			elsif ( /r/ ) { $recursive = 1; next; }
			
			else { die $usage; }

		}

	}

	else

	{

		$root = $_;

		die $usage if ( not -e $root );

	}

}

die "-i and -t may not be simultaneously selected.\n" if ( $indexonly and $thumbnailsonly );



# call our process method on the root directory.

process( $root );



sub process

{

	my $path = shift;



	my $template;



	my $indexname = 'index.html';

	my $suffix = '_';
	
	my $viewsuffix = '-';

	my $target = '_parent';

	my $thumbnailwidth;

	my $thumbnailheight;

	my $pixelarea;

	my $thumbnailscale;

	my $cellpadding = 10;

	my $thumbnailborder = 5;

	my $columns = 5;

	

	# read in the template.

	print "reading in $path/template.html\n";

	open TEMPLATE, "$path/template.html" or die "template.html not found in $path.\n";

	my $template = join '', <TEMPLATE>;

	close TEMPLATE;

	

	# get parameters from the template.

	die "thumb call not found in $path/template.html\n" if ( not $template =~ /<!thumb\s(.*?)>/si );

	my @parameters = split /\s+/s, $1;

	foreach ( @parameters )

	{

		if ( /^indexname=/ ) { $indexname = $'; next; }

		if ( /^suffix=/ ) { $suffix = $'; next; }
		
		if ( /^viewsuffix=/ ) { $suffix = $'; next; }

		if ( /^target=/ ) { $target = $'; next; }

		if ( /^thumbnailwidth=/ ) { $thumbnailwidth = $'; next; }

		if ( /^thumbnailheight=/ ) { $thumbnailheight = $'; next; }

		if ( /^pixelarea=/ ) { $pixelarea = $'; next; }

		if ( /^thumbnailscale=/ ) { $thumbnailscale = $'; next; }

		if ( /^cellpadding=/ ) { $cellpadding = $'; next; }

		if ( /^thumbnailborder=/ ) { $thumbnailborder = $'; next; }

		if ( /^tablebgcolor=/ ) { $tablebgcolor = $'; next; }

		if ( /^columns=/ ) { $columns = $'; next; }

	}

	# read the files in the current directory.

	opendir ROOT, $path;

	my @files = readdir ROOT;

	closedir ROOT;

	

	# process the files.

	foreach ( @files )

	{

		next if ( /^\.|\.\.$/ );

		my $fullname = "$path/$_";

		if ( -d $fullname and $recursive )

		{

			print "\nprocessing $fullname\n";

			process( $fullname );

			print "finished processing $fullname\n\n";

		}

		elsif ( /(.*?)\.jpg$/i and not /$suffix\.jpg$/i and not $indexonly )

		{

			generate_thumbnail( $fullname, $indexname, $suffix, $pixelarea, $thumbnailscale, $thumbnailwidth, $thumbnailheight );

			print "generated new $path/$1$suffix.jpg\n";

		}

	}

	

	# render index.

	if ( not $thumbnailsonly )

	{

		my @images = grep !/$suffix.jpg$/i, grep /.jpg$/i, @files;

		my $imagewidth, $imageheight;

		my $table;

		

		$table = "<!-- if javascript is specified -->\n<script language=\"JavaScript\">\n  counter=0;\n  function window_open(filename, x, y) {\n    windowops =

eval(\"'toolbar=no,location=no,directories=no,status=no,menubar=no,scrollbars=no,resizable=no,copyhistory=yes,width=\" + x + \",height=\" + y + \"'\");\n    counter = counter + 1;\n   

    window.open(filename, 'New' + counter, windowops);\n    return;\n  }\n</script>\n<!-- endif -->\n" if ( $target eq 'javascript' );

		

		if ($tablebgcolor=='' || $tablebgcolor=='none' || $tablebgcolor=='#') {
			$table = join '', $table, "<table border=0 cellspacing=0 cellpadding=$cellpadding>\n"; }
		else {
			$table = join '', $table, "<table border=0 bgcolor=$tablebgcolor cellspacing=0 cellpadding=$cellpadding>\n"; }

		my $k = 0;

		for ( 1..ceil( @images / $columns ) )

		{

			$table = join '', $table, "<tr>\n";

			for ( 1..$columns )

			{

				if ( $images[$k] =~ /(.*?)\.jpg$/i )

				{
				
					$realimg = $images[$k];

					( $imagewidth, $imageheight ) = get_dimensions( "$path/$images[$k]" );

					$table = join '', $table, "<td align=center valign=center>\n";

					$table = join '', $table, ( ( $target eq 'javascript' ) ? "<a href=\"$realimg\" onMouseOver=\"window.status='$1'; return true\" onclick=\"window_open('$realimg', $imagewidth+20, $imageheight+20);return false\" target=_blank >"

						: "<a href=\"$realimg\" target=\"$target\">" );

					$table = join '', $table, "<img src=\"$1$suffix.jpg\" border=$thumbnailborder></a>\n</td>\n";

				}

				$k++;

			}

			$table = join '', $table, "</tr>\n";

		}

		$table = join '', $table, "</table>\n";

		

		$template =~ s/<!thumb\s(.*?)>/$table/si;

		

		open INDEX, ">$path/$indexname";

		print INDEX $template;

		close INDEX;

		print "generated new $path/$indexname\n";

	}

}



sub get_dimensions

{

	my $fullname = shift;

	my $image, $x;

	$image = Image::Magick->new;

	$x = $image->Read( $fullname );

	warn "$x" if "$x";

	my( $width, $height ) = $image->Get( 'width', 'height' );

	undef $image;

	return ( $width, $height );

}



sub generate_thumbnail

{

	my $image = shift;

	my $indexname = shift;

	my $suffix = shift;

	my $pixelarea = shift;

	my $thumbnailscale = shift;

	my $thumbnailwidth = shift;

	my $thumbnailheight = shift;

	my $thumbnail, $x;

	

	$thumbnail = Image::Magick->new;

	

	$x = $thumbnail->Read( "$image" );

	warn "$x" if "$x";

	

	my( $imagewidth, $imageheight ) = $thumbnail->Get( 'width', 'height' );

	$ratio = $imagewidth / $imageheight;

	

	if ( defined( $pixelarea ) )

	{

		$thumbnailheight = int( sqrt( $pixelarea / $ratio ) );

		$thumbnailwidth = $thumbnailheight * $ratio;

	}

        elsif ( defined( $thumbnailscale ) )

        {

		$thumbnailheight = $imageheight * $thumbnailscale;

		$thumbnailwidth = $imagewidth * $thumbnailscale;

	}

	elsif ( defined( $thumbnailwidth ) )

	{

		$thumbnailheight = $thumbnailwidth / $ratio if ( not defined( $thumbnailheight ) );

	}

	elsif ( defined( $thumbnailheight ) )

	{

		$thumbnailwidth = $thumbnailheight * $ratio if ( not defined( $thumbnailwidth ) );

	}

	else

	{

		$thumbnailheight = int( sqrt( 10000 / $ratio ) );

		$thumbnailwidth = $thumbnailheight * $ratio;

	}

	

	$x = $thumbnail->Scale( width=>$thumbnailwidth, height=>$thumbnailheight );

	warn "$x" if "$x";

	

	$image =~ s/\.jpg$/$suffix\.jpg/i;

	$x = $thumbnail->Write( $image );

	warn "$x" if "$x";

	

	undef $thumbnail, $x;

	

	$new_thumbnails++;

}