/* pgmtops.c - read a portable graymap and produce a PostScript file
**
** Copyright (C) 1989 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.
*/

#include <stdio.h>
#ifdef	SYSV
#include <string.h>
#define index strchr
#else	SYSV
#include <strings.h>
#endif	SYSV
#include "pgm.h"

#define max(a,b) ((a) > (b) ? (a) : (b))

main( argc, argv )
int argc;
char *argv[];
    {
    FILE *ifd;
    register gray *grayrow, *gP;
    int argn, rleflag, rows, cols, format, bps, padright, row, col;
    gray maxval;
    float scale;
    char name[100], *cp;
    int maxvaltobps();
    char *usage = "[-rle] [-scale <x>] [pgmfile]";

    pm_progname = argv[0];

    argn = 1;
    rleflag = 0;
    scale = 1.0;

    /* Check for flags. */
    while ( argn < argc && argv[argn][0] == '-' )
	{
	if ( strncmp(argv[argn],"-rle",max(strlen(argv[argn]),2)) == 0 ||
	     strncmp(argv[argn],"-runlength",max(strlen(argv[argn]),2)) == 0 )
	    rleflag = 1;
	else if ( strncmp(argv[argn],"-scale",max(strlen(argv[argn]),2)) == 0 )
	    {
	    argn++;
	    if ( argn == argc || sscanf( argv[argn], "%f", &scale ) != 1 )
		pm_usage( usage );
	    }
	else
	    pm_usage( usage );
	argn++;
	}

    if ( argn < argc )
	{
	ifd = pm_openr( argv[argn] );
	strcpy( name, argv[argn] );
	if ( strcmp( name, "-" ) == 0 )
	    strcpy( name, "noname" );

	if ( ( cp = index( name, '.' ) ) != 0 )
	    *cp = '\0';
	argn++;
	}
    else
	{
	ifd = stdin;
	strcpy( name, "noname" );
	}

    if ( argn != argc )
	pm_usage( usage );

    pgm_readpgminit( ifd, &cols, &rows, &maxval, &format );
    grayrow = pgm_allocrow( cols );

    /* Figure out bps. */
    bps = maxvaltobps( maxval );
    
    /* Compute padding to round cols * bps up to the nearest multiple of 8. */
    padright = ( ( cols * bps + 7 ) / 8 ) * 8 - cols * bps;

    if ( rleflag )
	rleputinit( name, cols, rows, bps, scale );
    else
	putinit( name, cols, rows, bps, scale );
    for ( row = 0; row < rows; row++ )
	{
	pgm_readpgmrow( ifd, grayrow, cols, maxval, format );
        for ( col = 0, gP = grayrow; col < cols; col++, gP++ )
	    if ( rleflag )
		rleputgray( *gP );
	    else
		putgray( *gP );
	for ( col = 0; col < padright; col++ )
	    if ( rleflag )
		rleputgray( maxval );
	    else
		putgray( maxval );
        }

    pm_close( ifd );

    if ( rleflag )
	rleputrest( );
    else
	putrest( );

    exit( 0 );
    }

int
maxvaltobps( maxval )
gray maxval;
    {
    switch ( maxval )
	{
	case 1:
	return 1;

	case 3:
	return 2;

	case 15:
	return 4;

	case 255:
	return 8;

#ifdef notdef
	case 7:
	return 3;

	case 31:
	return 5;

	case 63:
	return 6;

	case 127:
	return 7;
#endif notdef

	default:
	pm_error( "maxval of %d is not supported", maxval, 0,0,0,0 );
	/* NOTREACHED */
	}
    }


int bitspersample, item, bitsperitem, bitshift, itemsperline, items;
int rleitem, rlebitsperitem, rlebitshift;
int repeat, itembuf[128], count, repeatitem, repeatcount;
#define HSBUFSIZ 256

putinit( name, cols, rows, bps, scale )
char *name;
int cols, rows, bps;
float scale;
    {
    int scols, srows, llx, lly;

    scols = cols * 0.96 + 0.5;	/*   0.96 is the multiple of   */
    srows = rows * 0.96 + 0.5;	/* 72/300 that is closest to 1 */
    llx = 300 - ( scols / 2 );
    lly = 400 - ( srows / 2 );

    printf( "%%!PS-Adobe-2.0 EPSF-2.0\n" );
    printf( "%%%%Creator: pgmtops\n" );
    printf( "%%%%Title: %s.ps\n", name );
    printf( "%%%%Pages: 1\n" );
    printf(
	"%%%%BoundingBox: %d %d %d %d\n", llx, lly, llx + scols, lly + srows );
    printf( "%%%%EndComments\n" );
    printf( "%%%%EndProlog\n" );
    printf( "%%%%Page 1 1\n" );
    printf( "/picstr %d string def\n", HSBUFSIZ );
    printf( "gsave\n" );
    printf( "%d %d translate\n", llx, lly );
    printf( "%g %g scale\n", scale, scale );
    printf( "%d %d scale\n", scols, srows );
    printf( "%d %d %d\n", cols, rows, bps );
    printf( "[ %d 0 0 -%d 0 %d ]\n", cols, rows, rows );
    printf( "{ currentfile picstr readhexstring pop }\n" );
    printf( "image\n" );

    bitspersample = bps;
    itemsperline = items = 0;
    item = 0;
    bitsperitem = 0;
    bitshift = 8 - bitspersample;
    }

putitem( )
    {
    if ( itemsperline == 30 )
	{
	putchar( '\n' );
	itemsperline = 0;
	}
    printf( "%02x", item );
    itemsperline++;
    items++;
    item = 0;
    bitsperitem = 0;
    bitshift = 8 - bitspersample;
    }

putgray( g )
gray g;
    {
    if ( bitsperitem == 8 )
	putitem( );
    item += g << bitshift;
    bitsperitem += bitspersample;
    bitshift -= bitspersample;
    }

putrest( )
    {
    if ( bitsperitem > 0 )
	putitem( );
    while ( items % HSBUFSIZ != 0 )
	putitem( );
    printf( "\n" );
    printf( "grestore\n" );
    printf( "showpage\n" );
    printf( "%%%%Trailer\n" );
    }

rleputinit( name, cols, rows, bps, scale )
char *name;
int cols, rows, bps;
float scale;
    {
    int scols, srows, llx, lly;

    scols = cols * 0.96 + 0.5;	/*   0.96 is the multiple of   */
    srows = rows * 0.96 + 0.5;	/* 72/300 that is closest to 1 */
    llx = 300 - ( scols / 2 );
    lly = 400 - ( srows / 2 );

    printf( "%%!PS-Adobe-2.0 EPSF-2.0\n" );
    printf( "%%%%Creator: pgmtops\n" );
    printf( "%%%%Title: %s.ps\n", name );
    printf( "%%%%Pages: 1\n" );
    printf(
	"%%%%BoundingBox: %d %d %d %d\n", llx, lly, llx + scols, lly + srows );
    printf( "%%%%EndComments\n" );
    printf( "%%%%EndProlog\n" );
    printf( "/rlestr1 1 string def\n" );
    printf( "/rlestr 128 string def\n" );
    printf( "/readrlestring {\n" );
    printf( "  currentfile rlestr1 readhexstring pop  0 get\n" );
    printf( "  dup 127 le {\n" );
    printf( "    currentfile rlestr 0  4 3 roll  1 add  getinterval\n" );
    printf( "    readhexstring  pop\n" );
    printf( "  } {\n" );
    printf( "    256 exch sub  dup\n" );
    printf( "    currentfile rlestr1 readhexstring pop  0 get\n" );
    printf( "    exch 0 exch 1 exch 1 sub { rlestr exch 2 index put } for\n" );
    printf( "    pop  rlestr exch 0 exch getinterval\n" );
    printf( "  } ifelse\n" );
    printf( "} bind def\n" );
    printf( "%%%%EndProlog\n" );
    printf( "%%%%Page 1 1\n" );
    printf( "gsave\n" );
    printf( "%d %d translate\n", llx, lly );
    printf( "%g %g scale\n", scale, scale );
    printf( "%d %d scale\n", scols, srows );
    printf( "%d %d %d\n", cols, rows, bps );
    printf( "[ %d 0 0 -%d 0 %d ]\n", cols, rows, rows );
    printf( "{ readrlestring }\n" );
    printf( "image\n" );

    bitspersample = bps;
    itemsperline = items = 0;
    rleitem = 0;
    rlebitsperitem = 0;
    rlebitshift = 8 - bitspersample;
    repeat = 1;
    count = 0;
    }

rleputbuffer( )
    {
    int i;

    if ( repeat )
	{
	item = 256 - count;
	putitem( );
	item = repeatitem;
	putitem( );
	}
    else
	{
	item = count - 1;
	putitem( );
	for ( i = 0; i < count; i++ )
	    {
	    item = itembuf[i];
	    putitem( );
	    }
	}
    repeat = 1;
    count = 0;
    }

rleputitem( )
    {
    int i;

    if ( count == 128 )
	rleputbuffer( );

    if ( repeat && count == 0 )
	{ /* Still initializing a repeat buf. */
	itembuf[count] = repeatitem = rleitem;
	count++;
	}
    else if ( repeat )
	{ /* Repeating - watch for end of run. */
	if ( rleitem == repeatitem )
	    { /* Run continues. */
	    itembuf[count] = rleitem;
	    count++;
	    }
	else
	    { /* Run ended - is it long enough to dump? */
	    if ( count > 2 )
		{ /* Yes, dump a repeat-mode buffer and start a new one. */
		rleputbuffer( );
		itembuf[count] = repeatitem = rleitem;
		count++;
		}
	    else
		{ /* Not long enough - convert to non-repeat mode. */
		repeat = 0;
		itembuf[count] = repeatitem = rleitem;
		count++;
		repeatcount = 1;
		}
	    }
	}
    else
	{ /* Not repeating - watch for a run worth repeating. */
	if ( rleitem == repeatitem )
	    { /* Possible run continues. */
	    repeatcount++;
	    if ( repeatcount > 3 )
		{ /* Long enough - dump non-repeat part and start repeat. */
		count = count - ( repeatcount - 1 );
		rleputbuffer( );
		count = repeatcount;
		for ( i = 0; i < count; i++ )
		    itembuf[i] = rleitem;
		}
	    else
		{ /* Not long enough yet - continue as non-repeat buf. */
		itembuf[count] = rleitem;
		count++;
		}
	    }
	else
	    { /* Broken run. */
	    itembuf[count] = repeatitem = rleitem;
	    count++;
	    repeatcount = 1;
	    }
	}

    rleitem = 0;
    rlebitsperitem = 0;
    rlebitshift = 8 - bitsperitem;
    }

rleputgray( g )
gray g;
    {
    if ( rlebitsperitem == 8 )
	{
	rleputitem( );
	}
    rleitem += g << rlebitshift;
    rlebitsperitem += bitsperitem;
    rlebitshift -= bitsperitem;
    }

rleputrest( )
    {
    if ( rlebitsperitem > 0 )
	rleputitem( );
    if ( count > 0 )
	rleputbuffer( );
    printf( "\n" );
    printf( "grestore\n" );
    printf( "showpage\n" );
    printf( "%%%%Trailer\n" );
    }