#include <stdio.h>
#include <stdlib.h>

#define XSTART 180
#define YSTART 100
#define WIDTH 2048
#define HEIGHT 1216
#define CWIDTH 1600
#define CHEIGHT 700

void printargs(int argc, char * argv[]);
void inputppm(char * filename, char *** red, char *** green, char *** blue);
void outputppm(char * outfilename, char **red, char **green, char **blue);
void outputcroppedppm(char * outfilename, char **red, char **green, char **blue);

void main(int argc, char * argv[])
{
	char * infilename;
	FILE * infile;

	char ** red;
	char ** green;
	char ** blue;

	int i,j;

	printargs(argc, argv);

	if (argc < 2)
	{
		printf("Usage: gigo filename\n");
		exit(0);
	}
	
	infilename = argv[1];	

	if ((infile = fopen(infilename, "rb")) == NULL)
	{
		printf("Couldn't find %s\n", infilename);
		exit(0);
	}

	printf("Mallocing red[].\n");
	red = (char **) malloc (WIDTH * sizeof (char *));
	for (i=0; i<WIDTH; i++)
	{
		if (!(i %100)) 
			printf("Mallocing red[%d].\n", i);
		red[i] = (char *) malloc (HEIGHT * sizeof (char));
	}	

	printf("Mallocing green[].\n");
	green = (char **) malloc (WIDTH * sizeof (char *));
	for (i=0; i<WIDTH; i++)
	{
		if (!(i %100)) 
			printf("Mallocing green[%d].\n", i);
		green[i] = (char *) malloc (HEIGHT * sizeof (char));
	}	

	printf("Mallocing blue[].\n");
	blue = (char **) malloc (WIDTH * sizeof (char *));
	for (i=0; i<WIDTH; i++)
	{
		if (!(i %100)) 
			printf("Mallocing blue[%d].\n", i);
		blue[i] = (char *) malloc (HEIGHT * sizeof (char));
	}	

	inputppm(infilename, &red, &green, &blue);

	if (infilename[0] == 's')  
	     infilename[0] = 'S';
	else infilename[0] = 's';

	outputcroppedppm(infilename, red, green, blue);
}

void inputppm(char * filename, char *** red, char *** green, char *** blue)
{
	FILE * infile;
	int w, h, i, j;

	infile = fopen(filename, "rb");

	fscanf(infile, "P6\n%d %d\n255\n", &w, &h);  

	if (w != 2048 || h != 1216)
	{
		printf("Oops! w and h are %d, %d\n", w, h);
		exit(0);
	}

        for (j=0; j<HEIGHT; j++) 
        {
            if (!(j %100)) 
                printf("Reading line %d\n", j); 
            for (i=0; i<WIDTH; i++) 
            {
                fscanf(infile, "%c", &((*red)[i][j])); 
                fscanf(infile, "%c", &((*green)[i][j])); 
                fscanf(infile, "%c", &((*blue)[i][j])); 
            }
        }

	fclose(infile);
}

void outputppm(char * outfilename, char **red, char **green, char **blue)
{
	FILE * outfile;
	int i, j;
	
	if ((outfile = fopen(outfilename, "wb")) == NULL)
	{
		printf("Couldn't open %s\n", outfilename);
		exit(0);
	}

	fprintf(outfile, "P6\n%d %d\n255\n", WIDTH, HEIGHT);

	for (j=0; j<HEIGHT; j++)
	{
	    if (!(j %100)) 
		printf("Writing line %d\n", j);
	    for (i=0; i<WIDTH; i++)
	    {
		fprintf(outfile, "%c", red[i][j]);
		fprintf(outfile, "%c", green[i][j]);
		fprintf(outfile, "%c", blue[i][j]);
	    }
	}

	fclose(outfile);
}

void outputcroppedppm(char * outfilename, char **red, char **green, char **blue)
{
	FILE * outfile;
	int i, j;
	
	if ((outfile = fopen(outfilename, "wb")) == NULL)
	{
		printf("Couldn't open %s\n", outfilename);
		exit(0);
	}

	fprintf(outfile, "P6\n%d %d\n255\n", CWIDTH, CHEIGHT);

	for (j=YSTART; j<YSTART+CHEIGHT; j++)
	{
	    if (!(j %100)) 
		printf("Writing line %d\n", j);
	    for (i=XSTART; i<XSTART+CWIDTH; i++)
	    {
		if (blue[i][j] > 1.2 *   red[i][j]  ||
		    blue[i][j] > 1.2 * green[i][j] )
		{
		    fprintf(outfile, "%c", 0);
		    fprintf(outfile, "%c", 0);
		    fprintf(outfile, "%c", 0);
		}	
		else 
		{
		    fprintf(outfile, "%c", red[i][j]);
		    fprintf(outfile, "%c", green[i][j]);
		    fprintf(outfile, "%c", blue[i][j]);
		}	
	    }
	}

	fclose(outfile);
}

void printargs(int argc, char * argv[])
{
	int i;

	for (i=0; i<argc; i++)
	{
		printf("%d: %s\n", i, argv[i]);
	}
}