45f6215127172784b078a3a3e9358ce683633f9f
[dja/scandal.git] / scan2pages.sh
1 #!/bin/bash
2
3 function usage {
4 cat >&2 << __EOF__ 
5 Usage: $0 [-vfpsmut] [-d depth] pdffile outdir
6 Convert pdffile - a pdf of scanned facing pages - to a set of images in outdir,
7 each with only one page.
8 OPTIONS:
9         -v: Be verbose.
10         -d depth: Use given depth. Default is 1. Unpaper can only handle up to 8.
11         -s: skip masking/trimming. Overrides -m.
12         -f: Forceably redo everything.
13         -p: Forceably redo pdf conversion. Implies options below, equivalent to -f.
14         -m: Forceably redo masking/trimming and other preprocessing. Impiles options below.
15         -u: Forceably redo unpaper processing. Implies option below. IGNORED
16         -t: Forceably redo final trimming and cleaning. IGNORED
17 __EOF__
18 }
19
20 ## Setup and utilities ##
21 # Mac OS X doesn't have seq. It has jot instead.
22 Linux_seq="seq"
23 Darwin_seq="jot -"
24 if [[ $(uname) == "Darwin" ]]; then my_seq=$Darwin_seq;
25 else my_seq=$Linux_seq;
26 fi;
27
28
29 ## Process flags ##
30 forcepdf=
31 forcemask=
32 forceunpaper=
33 forceclean=
34 convertflags=
35 unpaperflags=
36 depthflags="-depth 1"
37 skipmask=
38 while getopts 'vd:sfpmuc' OPTION
39 do
40         case $OPTION in
41         v)  convertflags="$convertflags -verbose"
42                 unpaperflags="$unpaperflags -v --time"
43                 ;;
44         d)      depthflags="-depth $OPTARG"
45                 ;;
46         s)      skipmask=1;
47                 ;;
48         f)      forcepdf=1; forcemask=1; forceunpaper=1; forceclean=1
49                 ;;
50         p)      forcepdf=1; forcemask=1; forceunpaper=1; forceclean=1
51                 ;;
52         m)      forcemask=1; forceunpaper=1; forceclean=1
53                 ;;
54         u)      forceunpaper=1; forceclean=1
55                 ;;
56         t)      forceclean=1
57                 ;;
58 #       b)      bflag=1
59 #               bval="$OPTARG"
60 #               ;;
61         ?)      usage
62                 exit 2
63                 ;;
64         esac
65 done
66 shift $(($OPTIND - 1))
67
68 # check we have an input and output!
69 if [[ $# != 2 ]]; then
70         echo "Wrong number of parameters (2 required, $# given: [$@])" >&2
71         usage
72         exit 2
73 fi
74
75 file=$1
76 dir=$2
77
78 filedir=$(dirname $1)
79 base=$(basename $1 .pdf)
80
81 # make the output dir
82 mkdir -p $dir
83
84 # figure out the number of pages
85 dscname=$dir/${base}.dsc
86 pdf2dsc $file $dscname || exit 1
87 pages=$(awk '$1 ~ "%%Pages" {print $2}' $dscname)
88 echo "Got $pages page(s)."
89 rm $dscname
90
91 # process pages 1 by 1 to avoid convert gobbling all the memory
92 for pg in `$my_seq 1 $pages`; do
93         echo "Processing page $pg."
94         
95         pgn=$(printf '%03d' $pg)
96         
97         # convert from pdf
98         origpnm=$dir/pg-${pgn}.pnm
99         if [[ ! $([ -e $origpnm ]) || $forcepdf ]]; then
100                 convert $convertflags $depthflags -density 300 $file[$(expr $pg - 1)] \
101                         $origpnm || exit 1
102         fi;
103         
104         # preprocess it!
105         preppnm=$dir/pg-pp-${pgn}.pnm
106         if [[ ! $skipmask && ( ! $([ -e $preppnm ]) || $forcemask ) ]]; then
107                 # create mask: 
108                 # ... downscale, blur,
109                 convert $convertflags -resize 25% -depth 8 -blur 10 $origpnm $dir/pg-mask-${pgn}.pnm ||exit 1
110
111                 # ... get crop co-ords. They're off by ~2 as I don't know how to
112                 # properly correct for the border.
113                 cropcords=$(convert -border 1x1 -bordercolor '#000' -resize 400% -trim -fuzz 90% -format "%wx%h%O" $dir/pg-mask-${pgn}.pnm info: || exit 1)
114                 
115                 # ... crop and despeckle? the final pre-prepared image
116                 convert $convertflags -crop $cropcords $origpnm $preppnm || exit 1
117         elif [[ $skipmask ]]; then
118                 cp $origpnm $preppnm
119         fi;
120         
121         #unpaper it
122         #names go a bit funny here
123         #also, ignore flags starting here
124         unppnm=$dir/upg-${pgn}-%01d.pnm
125         unpaper $unpaperflags --layout double --overwrite --no-blackfilter -ni 10 -op 2 $preppnm $unppnm || exit 1
126
127         # final convert and clean
128         convert $convertflags $dir/upg-${pgn}-?.pnm $dir/upg-${pgn}-%01d.tiff || exit 1
129         
130         
131 done
132
133 mkdir $dir/pages
134 mv $dir/upg-*.tiff $dir/pages
135         

UCC git Repository :: git.ucc.asn.au