#!/bin/sh
#
# ppmmargin - add a margin to a PNM image
#
# Copyright (C) 1991 by Jef Poskanzer.
#
# Permission to use, copy, modify, and distribute this software and its
# documentation for any purpose and without fee is hereby granted, provided
# that the above copyright notice appear in all copies and that both that
# copyright notice and this permission notice appear in supporting
# documentation.  This software is provided "as is" without express or
# implied warranty.

tempdir=$(mktemp -d "${TMPDIR:-/tmp}/netpbm.XXXXXXXX") ||
    { echo "Could not create temporary file. Exiting." 1>&2; exit 1; }
trap 'rm -rf $tempdir' 0 1 3 15

tmp1=$tempdir/pnmm1
tmp2=$tempdir/pnmm2
tmp3=$tempdir/pnmm3
tmp4=$tempdir/pnmm4

color="-gofigure"
plainopt=""

# Parse args.
while true ; do
    case "$1" in
        -version|--version )
        pnmpad --version; exit $?;
        ;;
        -p|-pl|-pla|-plai|-plain )
        plainopt="-plain"
        shift
        ;;
        -w|-wh|-whi|-whit|-white )
        color="-white"
        shift
        ;;
        -b|-bl|-bla|-blac|-black )
        color="-black"
        shift
        ;;
        -c|-co|-col|-colo|-color )
        shift
        if [ ! ${1-""} ] ; then
            echo "usage: $0 [-white|-black|-color <colorspec>] <size> [pnmfile]" 1>&2
            exit 1
        fi
        color="$1"
        shift
        ;;
        -* )
        echo "usage: $0 [-white|-black|-color <colorspec>] <size> [pnmfile]" 1>&2
        exit 1
        ;;
        * )
        break
        ;;
    esac
done

if [ ! ${1-""} ] ; then
    echo "usage: $0 [-white|-black|-color <colorspec>] <size> [pnmfile]" 1>&2
    exit 1
fi
size="$1"
shift

case $size in
    ''|*[!0-9]*)
        echo "Size argument '$size' is not a whole number"
        exit 1
        ;;
esac

if [ ${2-""} ] ; then
    echo "usage: $0 [-white|-black|-color <colorspec>] <size> [pnmfile]" 1>&2
    exit 1
fi

# Capture input file in a tmp file, in case it's a pipe.
# TODO: This code does not consider the case when the input file is specified
# and there is also input coming in from a pipe.

cat $@ > $tmp1

if [ $size -eq 0 ] ; then
    # Zero margin; just copy input to output
    pamtopnm $plainopt $tmp1;
else
    # If -white or -black, write output with pnmpad and exit.
    # Otherwise construct spacer files.

    case "$color" in
        -gofigure )
        pnmcut 0 0 1 1 $tmp1 | pnmtile $size 1 > $tmp2
        ;;
        -white | -black )
        pnmpad $plainopt $color \
            -left=$size -right=$size -top=$size -bottom=$size $tmp1
        rm -rf "$tempdir"
        exit
        ;;
        * )
        ppmmake $color $size 1 > $tmp2
        ;;
    esac
    pamflip -rotate90 $tmp2 > $tmp3

    # Cat things together.
    pnmcat -lr $tmp2 $tmp1 $tmp2 > $tmp4
    pnmcat -tb $plainopt $tmp3 $tmp4 $tmp3
fi

rm -rf "$tempdir"
