// ConsoleApplication6.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include "pgm.h"
#include <memory>

void getPGMfile (char filename[], PGMImage *img)
{
    FILE *in_file;
    char ch;
    int row, col, type;
    int ch_int;

    in_file = fopen(filename, "r");
    if (in_file == NULL)
    {
        fprintf(stderr, "Error: Unable to open file %s\n\n", filename);
        exit(8);
    }

    printf("\nReading image file: %s\n", filename);

    /*determine pgm image type (only type three can be used)*/
    ch = getc(in_file);
    if(ch != 'P')
    {
        printf("ERROR(1): Not valid pgm/ppm file type\n");
        exit(1);
    }
    ch = getc(in_file);
    /*convert the one digit integer currently represented as a character to
    an integer(48 == '0')*/
    type = ch - 48;
    if((type != 2) && (type != 3) && (type != 5) && (type != 6))
    {
        printf("ERROR(2): Not valid pgm/ppm file type\n");
        exit(1);
    }

    while(getc(in_file) != '\n');             /* skip to end of line*/

    while (getc(in_file) == '#')              /* skip comment lines */
    {
        while (getc(in_file) != '\n');          /* skip to end of comment line */
    }

    /*there seems to be a difference between color and b/w.  This line is needed
    by b/w but doesn't effect color reading...*/
    fseek(in_file, -1, SEEK_CUR);             /* backup one character*/

    fscanf(in_file,"%d", &((*img).width));
    fscanf(in_file,"%d", &((*img).height));
    fscanf(in_file,"%d", &((*img).maxVal));

    printf("\n width  = %d",(*img).width);
    printf("\n height = %d",(*img).height);
    printf("\n maxVal = %d",(*img).maxVal);
    printf("\n");

    if (((*img).width  > MAX) || ((*img).height  > MAX))
    {
        printf("\n\n***ERROR - image too big for current image structure***\n\n");
        exit(1);
    }

    if(type == 2) /*uncompressed ascii file (B/W)*/
    {
        for (row=(*img).height-1; row >=0; row--)
            for (col=0; col< (*img).width; col++)
            {
                fscanf(in_file,"%d", &ch_int);
                (*img).data[row][col].red = ch_int;
                (*img).data[row][col].green = ch_int;
                (*img).data[row][col].blue = ch_int;
            }
    }
    else if(type == 3) /*uncompressed ascii file (color)*/
    {
        for (row=(*img).height-1; row >=0; row--)
            for (col=0; col< (*img).width; col++)
            {

                fscanf(in_file,"%d", &ch_int);
                ((*img).data[row][col].red) = (unsigned char)ch_int;

                fscanf(in_file,"%d", &ch_int);
                ((*img).data[row][col].green) = (unsigned char)ch_int;

                fscanf(in_file,"%d", &ch_int);
                ((*img).data[row][col].blue) = (unsigned char)ch_int;
            }
    }
    else if(type == 5) /*compressed file (B/W)*/
        /*note: this type remains untested at this time...*/
    {
        while(getc(in_file) != '\n'); /*skip to end of line*/

        for (row=(*img).height-1; row >=0; row--)
            for (col=0; col< (*img).width; col++)
            {
                ch = getc(in_file);
                (*img).data[row][col].red = ch;
                (*img).data[row][col].green = ch;
                (*img).data[row][col].blue = ch;
            }
    }

    else if(type == 6) /*compressed file (color)*/
    {
        while(getc(in_file) != '\n'); /*skip to end of line*/

        for (row=(*img).height-1; row >=0; row--)
            for (col=0; col< (*img).width; col++)
            {
                (*img).data[row][col].red = getc(in_file);
                (*img).data[row][col].green = getc(in_file);
                (*img).data[row][col].blue = getc(in_file);
            }
    }

    fclose(in_file);
    printf("\nDone reading file.\n");
}


void save(PGMImage *img)
{
    int i, j, nr, nc, k;
    int red, green, blue;
    FILE *iop;

    nr = img->height;
    nc = img->width;

    iop = fopen("image1.pgm", "w");
    fprintf(iop, "P6\n");
    fprintf(iop, "%d %d\n", nc, nr);
    fprintf(iop, "255\n");

    k = 1;
    for(i = nr - 1; i  >= 0; i--)
    {
        for(j = 0; j <  nc; j++)
        {
            red = img->data[i][j].red;
            green = img->data[i][j].green;
            blue = img->data[i][j].blue;
            if(red <  0)
            {
                printf("IMG_WRITE: Found value %d at row %d col %d\n", red, i, j);
                printf("           Setting red to zero\n");
                red = 0;
            }
            if(green <  0)
            {
                printf("IMG_WRITE: Found value %d at row %d col %d\n", green,i, j);
                printf("           Setting green to zero\n");
                green = 0;
            }
            if(blue <  0)
            {
                printf("IMG_WRITE: Found value %d at row %d col %d\n", blue, i, j);
                printf("           Setting green to zero\n");
                blue = 0;
            }
            if(red  > 255)
            {
                printf("IMG_WRITE: Found value %d at row %d col %d\n", red, i, j);
                printf("           Setting red to 255\n");
                red = 255;
            }
            if(green  > 255)
            {
                printf("IMG_WRITE: Found value %d at row %d col %d\n", green,i, j);
                printf("           Setting green to 255\n");
                green = 255;
            }
            if(blue  > 255)
            {
                printf("IMG_WRITE: Found value %d at row %d col %d\n", blue, i, j);
                printf("           Setting blue to 255\n");
                blue = 255;
            }

            putc(red, iop);
            putc(green, iop);
            putc(blue, iop);
        }
    }
    fprintf(iop, "\n");
    fclose(iop);
}

void read_grid_from_file( int** grid, const size_t row, const size_t column, FILE* inf ) {
    size_t x, y;
    for( x = 0; x < column; ++x ) {
        for( y = 0; y < row; ++y ) {
            fscanf( inf, "%d", &grid[x][y] );
            printf( "%d ", grid[x][y] );
        }
        printf( "\n" );
    }
}

int main(int argc, char* argv[])
{
    PGMImage img;
    getPGMfile("progalap.pgm", &img);

    int x, y;
    for(x = 0; x < img.width; x++)
    {
        for(y = 0; y < img.height; x++)
        {
            RGB_INT temp = {0};
            memcpy(&temp, &img.data[x][y], sizeof(RGB_INT));
            int index = img.height - y;
            img.data[x][y] = img.data[index][y];
            img.data[index][y] = temp;
        }
    }

    save(&img);
    return 0;

}