#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <cvl/cvl.h>
#define READONLY 0
#define PICSIZE 1024
#define BLOCKSIZE 1024	/* Default disk sector size	*/
/*
 *        NAME:  normalize.c
 *        DESCRIPTION:  This program pads the comments of a CVL-formatted
 *                      picture file so that the picture data will begin
 *                      on a disk sector boundry
 *        COMPILATION:  cc -s -O -o normalize normalize.c -lcvl
 *                      (This program uses errprnt.)
 *        MODIFICATION HISTORY:  Written by Joseph Sanjour -- CVL
 *                                       on May 20, 1985
 */

char *argv0;		/* for errprnt			*/
struct cvl_p header;
int tmp_fd;		/* file descriptor for tmp	*/
int to_pad, i;
int blocksize = BLOCKSIZE;
char *bufin, *bufout; 	/* buffers for copying pic.	*/
char *tmp = "/tmp/chcmnt.?????";	/* file name for tmp-file	*/
int *picbuf;	/* buffer for picture	*/

main(argc, argv)
    int argc;
    char **argv;
{ 

    char **p;		/* pointer to call line arguments 	*/
    int n = argc;       /* used for processing options 		*/
    register int len; 	/* length of input header 		*/
    int newlcom = 0;	/* length of comments after cleaning	*/
    int gotchar = 0;	/* boolean flag for cleaning comments	*/
    int verbose = 0;	/* boolean flag for verbose option	*/

	argv0 = argv[0];

/*   set option flags 	*/
	for (p = argv; n-- > 0; p++)
		if(p[0][0] == '-')
		{	if ((p[0][1] >= '0') && (p[0][1] <= '9'))
				blocksize = atoi(&p[0][1]);
			else if (p[0][1] == 'v')
				verbose = 1;
			else errprnt("usage:  normalize [-blocksize] [-v] [<picturefile]");	}

/*   read input header  */
	if((len = ihead(0,&header)) < 0)
 		errprnt("cannot read input header or file is not a picture");
	if (verbose)
		fprintf(stderr,"\n%d characters in comment before cleanup.\n",header.cv_lcom);

/*   create tmpfile     */
 	pidfn();        /*  adds pid to tmp file name */
	if ((tmp_fd = creat(tmp, 0600)) < 0 )
    		errprnt("cannot create %s",tmp);

/*   Read comments and clean out previous normalization, if any	*/
	bufin = (bufout = (char *)malloc(header.cv_lcom + blocksize));
	if (read(0,bufin,header.cv_lcom) != header.cv_lcom)
    		errprnt("error in reading input picture\'s comments");
	for (i = 0;(i < header.cv_lcom); i++)
		if ((bufin[i] != '\0') && (bufin[i] != 10))
			if ((bufin[i] == '\n') && gotchar)
			{	bufout[newlcom++] = '\n';
				gotchar = 0;	}
			else
			{	bufout[newlcom++] = bufin[i];
				gotchar = 1;	}
	if (verbose)
		fprintf(stderr,"%d characters in comment after cleanup.\n",newlcom);

/*   Calculate number of characters to add to the comment field to
     make header(32) plus comments equal to a multiple of blocksize	*/
	to_pad = blocksize - ((newlcom + 32) % blocksize);
	for (i = 0; i <= to_pad; i++)
		if ((i % 80) == 0)
			bufout[i + newlcom] = '\n';
		else
			bufout[i + newlcom] = '\0';
	bufout[i + newlcom] = '\n';
	header.cv_lcom = newlcom + to_pad;
	write(tmp_fd,bufout,header.cv_lcom);
	if (verbose)
		fprintf(stderr,"%d characters in comment after padding.\n",header.cv_lcom);

/*  copy header to output */
 	if (ohead(1, &header) == -1) {
		die();
		errprnt("error writing picture header"); }

/*   copy comments      */
	if ((tmp_fd = open(tmp,READONLY)) == -1) {
		die();
 		errprnt("cannot reopen %s for reading",tmp); }

	if (cpcmnt(tmp_fd,1,header.cv_lcom) == -1) {
		die();
		errprnt("cannot copy comments from %s to output",tmp); }

/*   copy picture       */
	picbuf = (int *)malloc(header.cv_rows * header.cv_bp * 
	(header.cv_bp != 1 ? header.cv_cols : (header.cv_cols + 7) & ~7) / 8.0);

	while (len = readp (0,picbuf,PICSIZE))
	{	if (len == -1) {
			die();
			errprnt("read error copying picture"); }
		if (write(1,picbuf,len) != len) {
			die();
			errprnt("write error copying picture"); }
     	}
die();
return(0);
}                   /*   end   main  */
  
pidfn()  /* attach pid to file name (stolen from lpr.c) */
{
	register i,j,c;
	int p;

	p = getpid();
	i = 0;
	while(tmp[i] != '?')
 		i++;
	i += 4;
	for(j=0; j<5; j++) {
		c = (p%10) + '0';
		p /= 10;
		tmp[i] = c;
		i--;
	}
} /* pidfn */

die()
{ close(tmp_fd);
  unlink(tmp);
}