/* Sample Solution for 2D convolution for 24-354 General Robotics HW 4.

Modified by: Frederick Layton, fcl

   By Mary Berna 1-29-2002
      mberna
   Based on lab2.c by Ben Stolt
                      bstolt
*/


typedef struct pgm {
  int w;
  int h;
  int max;
  int* pData;
} pgm;

#include <string.h>
#include <stdio.h>
#include <stdlib.h>
/* 
   This LoadPGM function is an updated version that will deal with 
   comments in file headers and do at least some basic checking if the 
   file can be opened or is the correct format. It does assume the file 
   is actually a valid length once the header has been read. 
*/

void LoadPGM( char* filename, pgm* pPgm ) {
  FILE* ifp;
  int word;
  int nRead = 0;
  char readChars[256];

  //open the file, check if successful
  ifp = fopen( filename, "r" );
  if (!ifp) {
    printf("Error: Unable to open file %s.\n\n", filename);
    exit(1);
  }

  //read headers from file
  printf ("Reading PGM file: %s...\n", filename);
  fscanf (ifp, "%s", readChars);

  if (strcmp(readChars, "P2") == 0) {
    //valid file type
    //get a word from the file
    fscanf (ifp, "%s", readChars);
    while (readChars[0] == '#') {
      //if a comment, get the rest of the line and a new word
      fgets (readChars, 255, ifp);
      fscanf (ifp, "%s", readChars);
    }
    
    //ok, comments are gone
    //get width, height, color depth
    sscanf (readChars, "%d", &pPgm->w);
    fscanf (ifp, "%d", &pPgm->h);
    fscanf (ifp, "%d", &pPgm->max);

    // allocate some memory, note that on the HandyBoard you want to 
    // use constant memory and NOT use calloc/malloc
    pPgm->pData = (int*)malloc(sizeof(int) * pPgm->w * pPgm->h);
    
    // now read in the image data itself    
    for (nRead = 0; nRead < pPgm->w * pPgm->h; nRead++) {
      fscanf(ifp, "%d" ,&word);
      pPgm->pData[nRead] = word;
      // printf("nRead = %d %d\n",nRead,pPgm->pData[nRead]);
    }

    printf ("Loaded PGM. Size: %dx%d, Greyscale: %d \n", 
	    pPgm->w, pPgm->h, pPgm->max + 1);
  }
  else {
    printf ("Error: Read file type %s. Format unsupported.\n\n", readChars);
    exit(1);
  }
  fclose(ifp);
}

int find_max(int x, int y)
{
  if (x > y)
    return x;
  return y;
}

int find_min(int x, int y)
{
  if (x < y)
    return(x);
  return(y);
}

/* Simply saves the PGM file to a file whose name is in filename */

void WritePGM( char* filename, pgm* pPgm ) {
  FILE* ofp;
  int temp;
  int i,j;

  strcat(filename, ".out.pgm");
  ofp = fopen(filename, "w");
  if (!ofp) {
    printf("Error: Unable to open file %s.\n\n", filename);
    exit(1);
  }

  printf ("Writing ouput PGM: %s\n\n", filename);

  //write the header
  fprintf( ofp, "P2\n" );
  fprintf( ofp, "%d %d\n", pPgm->w, pPgm->h );
  fprintf( ofp, "%d\n", pPgm->max );

  //write the image data
  for (i = 0; i < pPgm->h;i++) {
    for (j = 0; j < pPgm->w; j++)
    {
      temp = find_max(0, *(pPgm->pData + i * pPgm->w + j));
      temp = find_min(temp, pPgm->max);
      fprintf(ofp,"%d ", temp);
    }
    fprintf(ofp,"\n");
  }
  fclose(ofp);	
}


/* Possibly useful in debugging...prints the data portion of the pgm
 * to the screen.
 */
void PrintPGM(pgm * my_pgm)
{
  int count;
  for (count = 0; count < my_pgm->w*my_pgm->h; count++)
  {
    if (count % my_pgm->w == 0)
      printf("\n");
    printf("%d ", (my_pgm->pData)[count]);
  }
}

/* Done for you...don't forget that you need to flip the mask
 * in convolution.
 */
void FlipMask(pgm * mask)
{
  int * mBuff;
  int * temp;
  int i;

  mBuff = mask->pData;
  temp = (int *) malloc((sizeof(int)) * (mask->w) * (mask->h));

  for (i = 0; i < (mask->h * mask->w); i++)
  {
    temp[i] = (mask->pData)[(mask->w) * (mask->h) - i -1];
  }

  memcpy(mBuff, temp, (sizeof(int)) * (mask->w) * (mask->h));
  free(temp);
}

/* Perform the convolution */
pgm * Convolve(pgm * image, pgm * mask)
{
  int i,j,m,n, size, current, sum;
  int low_val;
  int high_val;
  float scalar;
 
  pgm * result;

  result = (pgm *) malloc(sizeof(pgm));
  result->w = image->w;
  result->h = image->h;
  result->max = 255;
  result->pData = (int *) malloc(sizeof(int)*(image->w) * (image->h));
  FlipMask(mask);
  
  low_val = 0;
  high_val = 0;
	
  size = image->w * image->h;
  
  for(i = 0; i < size; i ++)
  	result->pData[i] = 255;

   for(i = mask->h/2; i < image->h - mask->h/2; i++){
   		for(j = mask->w/2; j < image->w - mask->w/2; j++){
   		
   			current = i*image->w + j;
   			sum = 0;
   			
   			for(m = 0; m < mask->h; m++){
   				for(n = 0; n < mask->w; n++){	
   				
	   				sum += (image->pData[(i + mask->h/2 - m)*image->w + (j + mask->w/2 - n)]) * mask->pData[m*mask->w + n];

   				}
   			}
   		
   		result->pData[current] = sum;	
   			//find values for contrasting
   			if(sum < low_val)
   				low_val = sum;
   			
   			if(sum > high_val)
   				high_val = sum;
   				
   		}
   	
   	
   } 
   

//contrast!
   scalar = 255/(float)(high_val - low_val);
   	
   for(i = mask->h/2; i < image->h - mask->h/2; i++){
   	for(j = mask->w/2; j < image->w - mask->w/2; j++){
		current = i*image->w + j;
		result->pData[current] = (int)((float)(result->pData[current] - low_val) * scalar);
   	}
   }
   


  return(result);
}


int main(int argc, char * argv[]) {
  pgm* conv_img;
  pgm* mask;
  pgm* image;
  char filename[256];
  char maskname[256];

  //check if an image filename was given, if not, ask for one
  if (argc > 1) {
    strcpy(filename, argv[1]);
    printf("Image filename is %s\n", filename);
  }
  else {
    printf ("Enter image filename: ");
    scanf ("%s", filename);
  }

  //check if a mask filename was given, if not, ask for one
  if (argc > 2) {
    strcpy(maskname, argv[2]);
    printf("Mask filename is %s\n", maskname);
  }
  else {
    printf ("Enter mask filename:  ");
    scanf ("%s", maskname);
  }

  //allocate memory for the pgm struct
  image = (pgm *) malloc (sizeof(pgm));
  mask  = (pgm *) malloc (sizeof(pgm));


  //read the file
  LoadPGM(filename, image);
  LoadPGM(maskname, mask);
  //Perform the convolution
  conv_img = Convolve(image, mask); 

  //finally write the new file
  WritePGM(filename, conv_img);

  free(image->pData);
  free(image);
  free(mask->pData);  
  free(mask);
  free(conv_img->pData);
  free(conv_img);

 
  //all done
  printf("Done. Have a nice day...\n\n");
  
  return 0;
}