source: trunk/gsdl/bin/script/pdftoimg.pl@ 10493

#!/usr/bin/perl -w


###########################################################################
#
# pdftoimg.pl -- convert PDF documents to various types of Image format
#
# A component of the Greenstone digital library software
# from the New Zealand Digital Library Project at the 
# University of Waikato, New Zealand.
#
# Copyright (C) 2001 New Zealand Digital Library Project
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
#
###########################################################################
# pdftoimg.pl is a wrapper for running convert utility which converts
# PDF documents to various types of image (e.g. PNG, GIF, JPEG format, 
# Then use PagedImgPlug to deal with the the images

BEGIN {
    die "GSDLHOME not set\n" unless defined $ENV{'GSDLHOME'};
    unshift (@INC, "$ENV{'GSDLHOME'}/perllib");
}

use parsargv;
use util;
use Cwd;
use File::Basename
;
sub print_usage {
    # note - we don't actually ever use most of these options...
    print STDERR  
    ("pdftoimg.pl wrapper for pdftoimg.\n",
     "Usage: pdftoimg [options] <PDF-file> <output>>\n",
     "Options:\n",
     "\t-convert_to\toutput image type for the PDF\n"
     );
    exit (1);
}

sub main {
    my (@ARGV) = @_;
    my ($convert_to);
    
    # read command-line arguments so that
    # you can change the command in this script
    if (!parsargv::parse(\@ARGV,
             'convert_to/.*/^', \$convert_to,
             )) {
        print_usage();
    }
    
    # Make sure the input file exists and can be opened for reading
    if (scalar(@ARGV) != 2) {
    print_usage();
    }
    
    my $input_filename = $ARGV[0];
    my $output_filestem = $ARGV[1];
    
    # test that the directories exist to create the output file, or
    # we should exit immediately. (File:: is included by util.pm)
    &util::mk_dir($output_filestem) if (!-e $output_filestem);
        
    my @dir = split (/(\/|\\)/, $input_filename);
    my $input_basename = pop(@dir);
    $input_basename =~ s/\.pdf//i;
    my $dir = join ("", @dir);

    if (!-r $input_filename) {
    print STDERR "Error: unable to open $input_filename for reading\n";
    exit(1);
    }
    # don't include path on windows (to avoid having to play about
    # with quoting when GSDLHOME might contain spaces) but assume
    # that the PATH is set up correctly.
    $cmd = "convert";
    #if ($ENV{'GSDLOS'} =~ /^windows$/);
    #Convert utility will convert the PDF to GIF Animation
    if ($convert_to eq "gif") {
    if ($ENV{'GSDLOS'} =~ /^windows$/){
        $cmd .= " ".$input_filename." ".$output_filestem."\\".$input_basename."-%02d.".$convert_to;
    } else {
        $cmd .= " ".$input_filename." ".$output_filestem."/".$input_basename."-%02d.".$convert_to;
    }
    } else {
    if ($ENV{'GSDLOS'} =~ /^windows$/){
        $cmd .= " ".$input_filename." ".$output_filestem."\\".$input_basename.".".$convert_to;
    } else {
        $cmd .= " ".$input_filename." ".$output_filestem."/".$input_basename.".".$convert_to;
    }
    }   
 
    # system() returns -1 if it can't run, otherwise it's $cmds ret val.
    # note we return 0 if the file is "encrypted"
    $!=0;
    if (system($cmd)!=0) {
    print STDERR "Convert error for $input_filename $!\n";
    # leave these for gsConvert.pl...
    #&util::rm("$output_filestem.text") if (-e "$output_filestem.text");
    #&util::rm("$output_filestem.err") if (-e "$output_filestem.err");
    return 1;
    } else {
    # command execute successfully
    create_itemfile($output_filestem, $input_basename, $convert_to);
    }
    return 0;
}

sub create_itemfile
{
    my ($output_dir, $convert_basename, $convert_to) = @_;
    opendir(DIR, $output_dir) || die "can't opendir $output_dir: $!";  
    my $item_file = $output_dir."/".$convert_basename.".item";
    open(FILE,">$item_file");
    
    print FILE "<PagedDocument>\n";

    my $page_num = "";
    @dir_files = grep {-f "$output_dir/$_"} readdir(DIR);

    # Sort files in the directory by page_num    
    sub page_num {
    my ($dir) = @_;
    my $pagenum = "";
    if ($ENV{'GSDLOS'} =~ /^windows$/){
       ($pagenum) =($dir =~ m/^.*\.(\d+)$/i);   
    } else {
       ($pagenum) =($dir =~ m/^.*-(\d+)\.(.*)$/i);
    }
    $pagenum = $pagenum || 1;
    return $pagenum;
    }

    # sort the files in the directory in the order of page_num rather than lexically. 
    @dir_files = sort { page_num($a) <=> page_num($b) } @dir_files;

    foreach my $file (@dir_files){
        if ($ENV{'GSDLOS'} =~ /^windows$/ && $convert_to ne "gif"){
      ($page_num) =($file =~ m/^.*\.(\d+)$/i); 
    } else {
        ($page_num) =($file =~ m/^.*-(\d+)\.(.*)/i);   
    }
    # as the converter will convert the document to image files start from page 0
    $page_num =$page_num + 1 if defined $page_num;
    $page_num = 1 unless defined $page_num;
    if ($file !~ /\.item/i){
        print FILE "   <Page pagenum=\"$page_num\" imgfile=\"$file\" txtfile=\"\"/>\n";
    } 
    }

    print FILE "</PagedDocument>\n";
    closedir DIR;
    return "";
}

# indicate our error status, 0 = success
exit (&main(@ARGV));