/* modified version of cementi.c...includes ability to apply fileters */

#include <math.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/time.h>
#include <fcntl.h>
#include <assert.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/wait.h>
#include "cement.h"
#include "pnmutils.h"
#include "file.h"
#include "powLookup.h"

/* Global Variables */

extern int errno;
extern char **environ;

/* Function prototypes needed in executable and in library */
static int my_system (char *command);
static inline double createValue(double a, unsigned char b, double weight);

static int verbose=0;

#ifndef OBJECT_COMPILE
/* Functions needed only in executable */
void usage(char * string);
void parse_commandline(int argc, char ** argv,
                       char**input_filenameA,
                       char**input_filenameB,
                       double *R,
                       double *G,
                       double *B);

void mystatus(char *action, char *filename, double R,double G, double B, double complete);
#endif

int grey(char *filename)
{
  int counter = 0, no_pixels;
  int bytes_per_pixel;
  double temp_red, temp_green, temp_blue, *red, *green, *blue;
  File *source_fd;
  FILE *dest_fd;
  char string[MAX_STRING_LENGTH];
  struct image_params source_header = {NULL, 0, 0, 0, 0, 0, 0};

  source_fd = fileopen(filename, "r");
  if (source_fd == NULL) {
    fprintf(stderr, "grey: unable to open %s\n", filename);
    return -1;
  }

  sprintf(string, "greyed.%s", filename);
  dest_fd = fopen(string, "w");
  if (dest_fd == NULL) {
    fprintf(stderr, "grey: unable to create %s\n", string);
    return -1;
  }

  get_image_params(source_fd, &source_header);
  if (source_header.type == 'A') {
    bytes_per_pixel = 3;
    sprintf(string, "P%c\n%d %d\n%d\n%d\n%f\n", source_header.type, source_header.width,
	    source_header.height, source_header.max_val, 1, 4.7);
    fwrite(string, sizeof(unsigned char), strlen(string), dest_fd);
  }

  printf("grey: reading in source pixels...");
  no_pixels = source_header.width * source_header.height;
  red = (double *) malloc(sizeof(double) * no_pixels);
  green = (double *) malloc(sizeof(double) * no_pixels);
  blue = (double *) malloc(sizeof(double) * no_pixels);

  counter = 0;
  while (counter < no_pixels) {
    fileread(&(red[counter]), sizeof(double), 1, source_fd);
    fileread(&(green[counter]), sizeof(double), 1, source_fd);
    fileread(&(blue[counter]), sizeof(double), 1, source_fd); 
    counter++;
  }
  printf("done\n");

  /* loop to grey the image data */
  
  counter = 0;
  while (counter < no_pixels) {
    temp_red = 0.299 * red[counter] +
               0.587 * green[counter] +
               0.114 * blue[counter];
    temp_green = temp_red;
    temp_blue = temp_red;

    fwrite(&temp_red, sizeof(double), 1, dest_fd);
    fwrite(&temp_green, sizeof(double), 1, dest_fd);
    fwrite(&temp_blue, sizeof(double), 1, dest_fd);

    counter++;
  }
  
  fileclose(source_fd);
  fclose(dest_fd);

  /* moving the temporary image file to the destination file */

  sprintf(string, "rm %s; mv greyed.%s %s", filename, filename, filename);
  if (my_system(string) == -1)
  {
    printf("Error calling %s\n", string);
    exit(EXIT_FAILURE);
  }
  
  return 0;
}

int blur(char *filename, int power)
{
  int counter = 0, no_pixels;
  int bytes_per_pixel, no_colours = 0;
  int x, y, dx, dy;
  double temp_red, temp_green, temp_blue, *red, *green, *blue;
  File *source_fd;
  FILE *dest_fd;
  char string[MAX_STRING_LENGTH];
  struct image_params source_header = {NULL, 0, 0, 0, 0, 0, 0};

  source_fd = fileopen(filename, "r");
  if (source_fd == NULL) {
    fprintf(stderr, "blur: unable to open %s\n", filename);
    return -1;
  }

  sprintf(string, "blurred.%s", filename);
  dest_fd = fopen(string, "w");
  if (dest_fd == NULL) {
    fprintf(stderr, "blur: unable to create %s\n", string);
    return -1;
  }

  get_image_params(source_fd, &source_header);
  if (source_header.type == 'A') {
    bytes_per_pixel = 3;
    sprintf(string, "P%c\n%d %d\n%d\n%d\n%f\n", source_header.type, source_header.width,
	    source_header.height, source_header.max_val, 1, 4.7);
    fwrite(string, sizeof(unsigned char), strlen(string), dest_fd);
  }

  printf("blur: reading in source pixels...");
  no_pixels = source_header.width * source_header.height;
  red = (double *) malloc(sizeof(double) * no_pixels);
  green = (double *) malloc(sizeof(double) * no_pixels);
  blue = (double *) malloc(sizeof(double) * no_pixels);

  counter = 0;
  while (counter < no_pixels) {
    fileread(&(red[counter]), sizeof(double), 1, source_fd);
    fileread(&(green[counter]), sizeof(double), 1, source_fd);
    fileread(&(blue[counter]), sizeof(double), 1, source_fd); 
    counter++;
  }
  printf("done\n");

  /* loop to blur the image data */
  for (y = 0; y < source_header.height; y++) {
    for (x = 0; x < source_header.width; x++) {
      temp_red = 0;
      temp_green = 0;
      temp_blue = 0;
      no_colours = 0;
      for (dy = -power + y; dy <= power + y; dy++) {
        for (dx = -power + x; dx <= power + x; dx++) {
          if ((dx >= 0) && (dx < source_header.width) && (dy >= 0) && (dy < source_header.height)) {
            temp_red =  temp_red + red[dx + dy * source_header.width];
            temp_green =  temp_green + green[dx + dy * source_header.width];
            temp_blue =  temp_blue + blue[dx + dy * source_header.width];
            no_colours++;
          }
        }
      }
      temp_red = temp_red / no_colours;
      temp_green = temp_green / no_colours;
      temp_blue = temp_blue / no_colours;

      fwrite(&temp_red, sizeof(double), 1, dest_fd);
      fwrite(&temp_green, sizeof(double), 1, dest_fd);
      fwrite(&temp_blue, sizeof(double), 1, dest_fd);
    }
  }

  fileclose(source_fd);
  fclose(dest_fd);

  /* moving the temporary image file to the destination file */

  sprintf(string, "rm %s; mv blurred.%s %s", filename, filename, filename);
  if (my_system(string) == -1)
  {
    printf("Error calling %s\n", string);
    exit(EXIT_FAILURE);
  }
  
  return 0;
}

/* Functions */
#ifndef CEMENT_REMOVE
int cementi(char *input_filenameA,
            char *input_filenameB,
            double R,
            double G,
            double B,
            char* command,
            void (*status)(char *action, char *filename, double R,double G, double B, double complete),
	    int argc,
	    char **argv
           )
#else
int cemento(char *input_filenameA,
            char *input_filenameB,
            double R,
            double G,
            double B,
            char* command,
            void (*status)(char *action, char *filename, double R,double G, double B, double complete),
	    int argc,
	    char **argv
           )
#endif
{
    struct image_params a_params={NULL,0,0,0,0,0.0};
    struct image_params b_params={NULL,0,0,0,0,0.0};
    File * a, * b, * dest;
    double p=EXPONENT;  /* exponent for pow */
    double        a_pixel[ARRAY_SIZE];
    unsigned char b_pixel[ARRAY_SIZE];
    double out_pixel[ARRAY_SIZE];
    int i,j;
    int channels_per_pixel;
    int counter = 0, position;
    int filtered_before = 0;
//    double R, G, B;  /* weights for cement */

    /* Timing measurments */
    struct timeval timeValStart;
    struct timeval timeValEnd;
    struct timezone timeZone;
    int minutes;
    double seconds=0.0;
#ifdef GET_TIME_MEASUREMENTS
    struct timeval runTimeStart;
    struct timeval runTimeStop;
    double fileReadTime=0.0;
    double fileWriteTime=0.0;
    double computeTime=0.0;
#endif
    
    unsigned int pixels=0;
    unsigned int numberOfPixels;

    char string[MAX_STRING_LENGTH];
    char temp_command[MAX_STRING_LENGTH];
    unsigned int image_ptr=0;

    if(p!=powLookupExponent)
    {
        printf("Exonents must match\n");
        exit(EXIT_FAILURE);
    }

    a_params.filename=input_filenameA;
    b_params.filename=input_filenameB;
    
    /* applying the filters, if requested */

    counter = argc - 6;
    position = 1;
    filtered_before = 0;
    while (counter > 0) {
        if (strcmp(argv[5 + position], "b") == 0) {
            if (filtered_before == 0) {
	      sprintf(temp_command, "./cementinit %s 1 1 1 -o temp.plm", b_params.filename);
	      system(temp_command);
	    }
	    printf("applying the blur filter...\n");
	    blur("temp.plm", atoi(argv[6 + position]));
	    position = position + 2;
	    counter = counter - 2;
	    filtered_before = 1;
	} else if (strcmp(argv[5 + position], "g") == 0) {
            if (filtered_before == 0) {
              sprintf(temp_command, "./cementinit %s 1 1 1 -o temp.plm", b_params.filename);
	      system(temp_command);
            }
	    printf("applying the grey filter...\n");
            grey("temp.plm");
	    position++;
	    counter--;
            filtered_before = 1;
	}
    }

    if (filtered_before == 1) {
      system("./plm2pnm temp.plm -o temp.ppm");
      b_params.filename = "temp.ppm";
    }

//    parse_commandline(argc, argv, &a_params, &b_params, &R, &G, &B);

    if ((a=fileopen(a_params.filename, "r"))==NULL)
    {
        fprintf(stderr, "Unable to open %s.\n", a_params.filename);
        exit(EXIT_FAILURE);
    }

    if ((b=fileopen(b_params.filename, "r"))==NULL)
    {
        fprintf(stderr, "Unable to open %s.\n", b_params.filename);
        exit(EXIT_FAILURE);
    }

    sprintf(string,"tmp.%s",a_params.filename);
    if ((dest=fileopen(string, "w"))==NULL)
    {
        fprintf(stderr, "Unable to open %s.\n", string);
        exit(EXIT_FAILURE);
    }

    get_image_type(a, &a_params);

    if (a_params.type=='A')
    {
        channels_per_pixel=3;
        sprintf(string, "P%c\n", a_params.type);
        filewrite(string,sizeof(unsigned char),strlen(string),dest);
        
        get_image_params_preserve_comments(a,&a_params,dest);
    }
    else
    {
        fprintf(stderr, "Portable Lightspace Map (plm) must be type A and it is type %c\n",a_params.type);
        exit(EXIT_FAILURE);
    }


    if(isJpegFile(b)==0)
    {
        get_image_params(b, &b_params);
    }
    else
    {
        b_params.type='6';
        b_params.max_val=255;

        if(read_JPEG_file (b_params.filename)==0)
        {
            printf("ERROR: File %s does not exist\n",b_params.filename);
            exit(0);
        }
        //        write_JPEG_file("test.jpg",95);
        //        exit(0);

        image_ptr=0;
        b_params.width=image_width;
        b_params.height=image_height;
    }



    if ((a_params.width!=b_params.width) || (a_params.height!=b_params.height)
        || (a_params.max_val!=b_params.max_val) || (a_params.exponent!=(float)EXPONENT))
    {
        printf("%f %f\n",a_params.exponent,EXPONENT);
        printf("%d %d\n",a_params.width,b_params.width);
        printf("%d %d\n",a_params.height,b_params.height);
        printf("%d %d\n",a_params.max_val,b_params.max_val);
        fprintf(stderr, "Lightspace must have the same dimensions, maximum "
                "values, and exponent.\n");
        exit(EXIT_FAILURE);
    }

    sprintf(string, "# %s %s %s %f %f %f\n",command,input_filenameA,input_filenameB,R,
            G,B);
    filewrite(string,sizeof(unsigned char),strlen(string),dest);
    
#ifndef CEMENT_REMOVE
    sprintf(string, "%d %d\n%d\n%d\n%f\n",a_params.width,
            a_params.height,a_params.max_val, ++a_params.numberOfImages,EXPONENT);
#else
    sprintf(string, "%d %d\n%d\n%d\n%f\n",a_params.width,
            a_params.height, a_params.max_val,--a_params.numberOfImages,EXPONENT);
#endif
    filewrite(string,sizeof(unsigned char),strlen(string),dest);

    
#ifdef SILENT_RUN

#ifndef CEMENT_REMOVE
//    printf("Add %s %d %d %d\r",b_params.filename,(int)R,(int)G,(int)B);
#else
//    printf("Del %s %d %d %d\r",b_params.filename,(int)R,(int)G,(int)B);
#endif
#else

#ifndef CEMENT_REMOVE
    printf("Adding file: \"%s\", ",b_params.filename);
#else
    printf("Removing file: \"%s\", ",b_params.filename);
#endif
    printf("P%c, %dx%d, ",b_params.type,b_params.width,b_params.height);

    printf("R=%7.3f ",R);
    printf("G=%7.3f ",G);
    printf("B=%7.3f to ",B);

    printf("lightspace \"%s\", ",a_params.filename);
    printf("P%c\n",a_params.type);
#endif
    
    /* Start Timer */
    if(gettimeofday(&timeValStart,&timeZone)==-1)
    {
        fprintf(stderr,"Error getting the time\n");
        exit(0);
    }

    numberOfPixels=a_params.width*a_params.height;

    if(((float)numberOfPixels/NUMBER_PIXELS_PER_READ)!=(int)((float)numberOfPixels/NUMBER_PIXELS_PER_READ))
    {
        if(verbose)
            printf("\nERROR: The number of pixels %d is not evenly divisible by %d. ",numberOfPixels,NUMBER_PIXELS_PER_READ);

        numberOfPixels=(int)((float)numberOfPixels/NUMBER_PIXELS_PER_READ)*NUMBER_PIXELS_PER_READ;
        if(verbose)
            printf("We will go up to %d pixels.\n",numberOfPixels);
    }

    for(i=0;i<numberOfPixels;i+=NUMBER_PIXELS_PER_READ)
    {
        if(i%(3*NUMBER_PIXELS_PER_READ)==0)
        {
#ifndef CEMENT_REMOVE
            if(status!=NULL)
                status("Add",b_params.filename,R,G,B,(double)100*i/numberOfPixels);
#else
            if(status!=NULL)
                status("Del",b_params.filename,R,G,B,(double)100*i/numberOfPixels);
#endif
        }
        
#ifdef GET_TIME_MEASUREMENTS
        gettimeofday(&runTimeStart,&timeZone);
#endif

        fileread(a_pixel, sizeof(double),        ARRAY_SIZE, a);

        if(isJpegFile(b)==0)
            fileread(b_pixel, sizeof(unsigned char), ARRAY_SIZE, b);
        else
        {
            memcpy(b_pixel,&image_buffer[image_ptr],ARRAY_SIZE);
            image_ptr+=ARRAY_SIZE;
        }
#ifdef GET_TIME_MEASUREMENTS
        gettimeofday(&runTimeStop,&timeZone);

        seconds=(double)runTimeStop.tv_sec+(double)runTimeStop.tv_usec/1000000;
        seconds-=(double)runTimeStart.tv_sec+(double)runTimeStart.tv_usec/1000000;
        fileReadTime+=seconds;

        gettimeofday(&runTimeStart,&timeZone);
#endif
        
        for(j=0;j<ARRAY_SIZE;j+=3)
        {
            out_pixel[j  ]=createValue(a_pixel[j  ],b_pixel[j  ],R);
            out_pixel[j+1]=createValue(a_pixel[j+1],b_pixel[j+1],G);
            out_pixel[j+2]=createValue(a_pixel[j+2],b_pixel[j+2],B);
        }

#ifdef GET_TIME_MEASUREMENTS
        gettimeofday(&runTimeStop,&timeZone);

        seconds=(double)runTimeStop.tv_sec+(double)runTimeStop.tv_usec/1000000;
        seconds-=(double)runTimeStart.tv_sec+(double)runTimeStart.tv_usec/1000000;

        computeTime+=seconds;

        gettimeofday(&runTimeStart,&timeZone);
#endif
        
        filewrite(out_pixel, sizeof(double), ARRAY_SIZE, dest);

#ifdef GET_TIME_MEASUREMENTS
        gettimeofday(&runTimeStop,&timeZone);

        seconds=(double)runTimeStop.tv_sec+(double)runTimeStop.tv_usec/1000000;
        seconds-=(double)runTimeStart.tv_sec+(double)runTimeStart.tv_usec/1000000;
        fileWriteTime+=seconds;
#endif
    }

    if(numberOfPixels!=a_params.width*a_params.height)
    {
        if(verbose)
            printf("Cleanup on pixels %d to %d\n",numberOfPixels,a_params.width*a_params.height);

        for(i=numberOfPixels;i<a_params.width*a_params.height;i++)
        {

            fileread(a_pixel, sizeof(double),CHANNELS_PER_PIXEL, a);

            if(isJpegFile(a)==0)
                fileread(b_pixel, sizeof(unsigned char), CHANNELS_PER_PIXEL, b);
            else
            {
                memcpy(b_pixel,&image_buffer[image_ptr],CHANNELS_PER_PIXEL);
                image_ptr+=CHANNELS_PER_PIXEL;
            }

            out_pixel[0]=createValue(a_pixel[0],b_pixel[0],R);
            out_pixel[1]=createValue(a_pixel[1],b_pixel[1],G);
            out_pixel[2]=createValue(a_pixel[2],b_pixel[2],B);
            filewrite(out_pixel, sizeof(double), CHANNELS_PER_PIXEL, dest);
            pixels++;

        }
    }

    /* End timer */
    if(gettimeofday(&timeValEnd,&timeZone)==-1)
    {
        printf("Error getting the time\n");
        exit(0);
    }

    pixels=i;

    while (!fileeof(a))
    {
        fileread(a_pixel, sizeof(double),CHANNELS_PER_PIXEL, a);

        if(isJpegFile(a)==0)
            fileread(b_pixel, sizeof(unsigned char), CHANNELS_PER_PIXEL, b);
        else
        {
            memcpy(b_pixel,&image_buffer[image_ptr],CHANNELS_PER_PIXEL);
            image_ptr+=CHANNELS_PER_PIXEL;
        }

        if(fileeof(a))
            break;
        
        out_pixel[0]=createValue(a_pixel[0],b_pixel[0],R);
        out_pixel[1]=createValue(a_pixel[1],b_pixel[1],G);
        out_pixel[2]=createValue(a_pixel[2],b_pixel[2],B);
        printf("EP A: %f %f %f\t",a_pixel[0],a_pixel[1],a_pixel[2]);
        printf("B: %d %d %d\t",b_pixel[0],b_pixel[1],b_pixel[2]);
        printf("OUT: %f %f %f\n",out_pixel[0],out_pixel[1],out_pixel[2]);
        filewrite(out_pixel, sizeof(double), CHANNELS_PER_PIXEL, dest);
        pixels++;
    }
    
    fileclose(a);
    fileclose(b);
    fileclose(dest);

    seconds=(double)timeValEnd.tv_sec+(double)timeValEnd.tv_usec/1000000;
    seconds-=(double)timeValStart.tv_sec+(double)timeValStart.tv_usec/1000000;

#ifdef GET_TIME_MEASUREMENTS
    printf("%f s doing File Read %f%%\n",fileReadTime,100*fileReadTime/seconds);
    printf("%f s doing File Writes %f%%\n",fileWriteTime,100*fileWriteTime/seconds);
    printf("%f s doing Computation %f%%\n",computeTime,100*computeTime/seconds);
    printf("%f s doing Timing and looping\n",seconds-fileReadTime-fileWriteTime-computeTime);
#endif
    
    if(seconds>60)
    {
        minutes=seconds/60;
        seconds=seconds-minutes*60;
    }
    else
        minutes=0;

    if(verbose || GET_RUN_TIME)
    {
#ifndef SILENT_RUN
        if(minutes==1)
            printf("Run Time is about %d minute %f seconds.\n",minutes,seconds);
        else
            printf("Run Time is about %d minutes %f seconds.\n",minutes,seconds);

#else


#ifndef CEMENT_REMOVE
//        printf("Add %s %d %d %d %4.2f(s)\n",b_params.filename,(int)R,(int)G,(int)B,minutes*60+seconds);
            if(status!=NULL)
                status("Add",b_params.filename,R,G,B,(double)100);
#else
//        printf("Del %s %d %d %d %4.2f(s)\n",b_params.filename,(int)R,(int)G,(int)B,minutes*60+seconds);
            if(status!=NULL)
                status("Del",b_params.filename,R,G,B,(double)100);
#endif

#endif
    }

    if(isJpegFile(b)==1)
        free(image_buffer);

    sprintf(string,"rm %s; mv tmp.%s  %s",a_params.filename,a_params.filename,a_params.filename);
    if(verbose)
        printf("%s\n",string);

    if(my_system (string)==-1)
    {
        printf("Error calling %s\n",string);
        exit(EXIT_FAILURE);
    }

    return(EXIT_SUCCESS);
}


static inline double createValue(double a, unsigned char b, double weight)
{
#ifndef CEMENT_REMOVE
    return(a+weight*powLookupTable[b]);
#else
    return(a-weight*powLookupTable[b]);
#endif
}

static int my_system (char *command)
{
    int pid, status;

    if (command == 0)
        return 1;
    pid = fork();
    if (pid == -1)
        return -1;
    if (pid == 0)
    {
        char *argv[4];
        argv[0] = "sh";
        argv[1] = "-c";
        argv[2] = command;
        argv[3] = 0;
        execve("/bin/sh", argv, environ);
        exit(127);
    }
    do {
        if (waitpid(pid, &status, 0) == -1) {
            if (errno != EINTR)
                return -1;
        } else
            return status;
    } while(1);
}

#ifndef OBJECT_COMPILE
int main(int argc, char ** argv)
{
    char * input_filenameA=NULL;
    char * input_filenameB=NULL;
    double R,G,B;

    parse_commandline(argc, argv, &input_filenameA,&input_filenameB,&R,&G,&B);

#ifndef CEMENT_REMOVE
    return(cementi(input_filenameA, input_filenameB,R,G,B,argv[0],mystatus, argc, argv));
#else
    return(cemento(input_filenameA, input_filenameB,R,G,B,argv[0],mystatus, argc, argv));
#endif
}

void mystatus(char *action, char *filename, double R,double G, double B, double complete)
{
    printf("%s %s %d %d %d %d%%\r",action,filename,(int)R,(int)G,(int)B,(int)complete);
    fflush(stdout);
    if((int)complete==100)
      printf("\n");
}



void usage(char * string)
{
    fprintf(stderr, "Use: %s [Lightspace file] [Imagespace file] <R> <G> <B>\n",string);
    fprintf(stderr, " [Lightspace file] is Portable Lightspace Map (plm) file image to be added to\n");
    fprintf(stderr, " [Imagespace file] is ppm file of image to add to the lightspace.\n");
    fprintf(stderr, " <R>=Float for Red weight.\n");
    fprintf(stderr, " <G>=Float for Green weight.\n");
    fprintf(stderr, " <B>=Float for Blue weight.\n");
    exit(EXIT_SUCCESS);
}

void parse_commandline(int argc, char ** argv,
                       char**input_filenameA,
                       char**input_filenameB,
                       double *R,
                       double *G,
                       double *B)
{
    if (argc<6)
        usage(argv[0]);

    *input_filenameA=(char*)strdup(argv[1]);
    *input_filenameB=(char*)strdup(argv[2]);

    sscanf(argv[3],"%lf",R); /* sscanf does not recognize %g, must be %lf */
    sscanf(argv[4],"%lf",G); /* sscanf does not recognize %g, must be %lf */
    sscanf(argv[5],"%lf",B); /* sscanf does not recognize %g, must be %lf */
}
#endif