personalLib.c

#include "personalLib.h"

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


/*
 * This function prints the help menu
 * @Param: /
 * @returns void
 */
void helpmenu(){
    printf("-i met daarachter de inputfile\n ");

    printf("voorbeeld\n");
    printf("./splitchannels -i meme.bmp\n");
}

/*
 * This function will read an input file and create 3 files, each contain only the red, green and blue colors of the respective pixels. Also a 4th file with the colors inverted
 * char pointer used to store the input bmp file name
 * @return void
 */
void code(char* inputFileName){
    FILE *f = fopen(inputFileName, "rb");
    if (f == NULL) {
        printf("something went wrong while opening file");
        exit(1);
    }
    int imageSize = 0;
    imageSize = getSizeBmp(f);

    int arraySize = imageSize/3;
    pixel* RGB = malloc(arraySize * sizeof(*RGB));
    unsigned char *bmpheader = malloc(54);
    readInputBmpfile(RGB, imageSize, bmpheader, f);

    outputBMP(RGB, bmpheader, imageSize,inputFileName);

    fclose(f);
    free(inputFileName);
    free(bmpheader);
    free(RGB);
}

/*
 *This function gets the size of the bmp image
 * @param: File pointer f of input bmp file
 * @return: returns the image size as int imageSize
 */
int getSizeBmp(FILE *f){
    unsigned char bmpheader[54];
    fread(bmpheader, sizeof(unsigned char ), 54, f);
    unsigned int breedte = bmpheader[18] | bmpheader[19] << 8 | bmpheader[20] << 16 | bmpheader[21] << 24;
    unsigned int hoogte = bmpheader[22] | bmpheader[23] << 8 | bmpheader[24] << 16 | bmpheader[25] << 24;
    int imageSize = breedte * hoogte * 3;

    return imageSize;
}

/*
 * This function read the pixels of the input bmp file and stores them in the array of struct RGB and stores header information into pointer bmpheader
 * @param: pointer array of struct pixel
 * @param: int imagesize = size of input image
 * @param: unsigned char pointer bmpheader: header information of input bmp file
 * @param: File pointer f of input bmp file
 * @return: void
 */
void readInputBmpfile(pixel* RGB, int imageSize, unsigned char* bmpheader, FILE* f){
    rewind(f);
    unsigned char tempRGB[imageSize];
    fread(bmpheader, sizeof(unsigned char ), 54, f);
    fread(tempRGB, sizeof(unsigned char), imageSize, f);
    int  x = 0;
    int arraySize = imageSize/3;
    for(int i = 0; i < arraySize; i++){

        RGB[i].blue = tempRGB[x];
        RGB[i].green = tempRGB[x+1];
        RGB[i].red = tempRGB[x+2];
        x = x + 3;
    }


}

/*
 * This function creates 3 files, each contain only the red, green and blue colors of the respective pixels. Also a 4th file with the colors inverted
 * @param: pointer array of struct pixel
 * @param: unsigned char pointer bmpheader: header information of input bmp file
 * @param: int imagesize = size of input image
 * @param: char pointer used to store the input bmp file name
 * @return: void
 */
void outputBMP(pixel* RGB, unsigned char* bmpheader, int imageSize, char* inputFileName){
    removeBMPfromString(inputFileName);
    char strRed[] = "_red.bmp";
    char strGreen[] = "_green.bmp";
    char strBlue[] = "_blue.bmp";
    char strInverted[] = "_inverted.bmp";

    unsigned char* RGBout = malloc(imageSize * sizeof(*RGBout));
    char* outputName;

    //rood
    outputRGBRed(RGBout, RGB, imageSize);
    outputName = outputBMPFileName(inputFileName, strRed); //voor rood
    createOutput(bmpheader , outputName, RGBout, imageSize ); // voor rood

    //groen
    outputRGBGreen(RGBout, RGB, imageSize);
    outputName = outputBMPFileName(inputFileName, strGreen); //voor groen
    createOutput(bmpheader, outputName, RGBout, imageSize);//voor groen

    //blauw
    outputRGBBlue(RGBout, RGB, imageSize);
    outputName = outputBMPFileName(inputFileName, strBlue);
    createOutput(bmpheader, outputName, RGBout, imageSize);

    //geinverteerd
    outputRGBinverted(RGBout, RGB, imageSize);
    outputName = outputBMPFileName(inputFileName, strInverted);
    createOutput(bmpheader, outputName, RGBout, imageSize);

    free(outputName);
    free(RGBout);
}

/*
 * This function removes .bmp from the name of the input bmp file
 * @param: char pointer used to store the input bmp file name
 * @return: void
 */
void removeBMPfromString(char* inputFileName){
    int a = strlen(inputFileName);

    inputFileName[a-1] = '\0';
    inputFileName[a-2] = '\0';
    inputFileName[a-3] = '\0';
    inputFileName[a-4] = '\0';

}

/*
 * This function puts _color.bmp behind the name of the outputFile example: inputFileName = meme.bmp the .bmp is removed in removeBMPfromString this function makes it for color red meme_red.bmp
 * @param: char pointer used to store the input bmp file name
 * @param: char str[] = contains what has to come behind inputFileName ex: meme_red.bmp   str[] contains _red.bmp
 * @return: char* outputname = outputname of file
 */
char* outputBMPFileName(char* inputFileName, char str[]){

    int sizeName = strlen(inputFileName) + strlen(str);

    char* outputName = (char*) malloc(sizeName);
    strcpy(outputName, inputFileName);
    strcat(outputName, str);
    return outputName;

}

/*
 * This function creates the outputFile and writes the outputfile
 * @param: unsigned char pointer bmpheader: header information of input bmp file
 * @param: char* outputName= name of outputfile
 * @param: unsigned char pointer RGBout= contains the pixels of the outputFile
 * @param: int imagesize = size of input image
 * @return: void
 */
void createOutput(unsigned char* bmpheader, char* outputName, unsigned char* RGBout, int imageSize){

    FILE *fp;
    fp = fopen(outputName, "wb");
    if(fp == NULL){
        printf("file does not create\n");
        exit(0);
    }
    printf("file created succesfully\n");
    fwrite(bmpheader, 1, 54, fp);
    fwrite(RGBout, sizeof(unsigned char), imageSize, fp);
    printf("output  finished\n");
    fclose(fp);


}

/*
 * This function creates the output rgb for red
 * @param: unsigned char pointer RGBRed = contains the output rgb
 * @param: pointer array of struct pixel
 * @param: int imagesize = size of input image
 * @return: void
 */
void outputRGBRed(unsigned char* RGBRed, pixel* RGB, int imageSize){
    int x = 0;
    int j = imageSize / 3;
    for(int i = 0; i < j; i++ ){
        RGBRed[x] = 0;
        RGBRed[x+1] = 0;
        RGBRed[x+2] = 0;
        RGBRed[x+2] = RGB[i].red;
        x = x +3;
    }

}

/*
 * This function creates the output rgb for green
 * @param: unsigned char pointer RGBGreen = contains the output rgb
 * @param: pointer array of struct pixel
 * @param: int imageSize = size of input image
 * @return: void
 */
void outputRGBGreen(unsigned char* RGBGreen, pixel* RGB, int imageSize){
    int x = 0;
    int j = imageSize / 3;
    for(int i = 0; i < j; i++ ){
        RGBGreen[x] = 0;
        RGBGreen[x+1] = 0;
        RGBGreen[x+2] = 0;
        RGBGreen[x+1] = RGB[i].green;
        x = x +3;
    }

}

/*
 * This function creates the output rgb for blue
 * @param: unsigned char pointer RGBBlue = contains the output rgb
 * @param: pointer array of struct pixel
 * @param int imageSize = size of input image
 * @return: void
 */
void outputRGBBlue(unsigned char* RGBBlue, pixel* RGB, int imageSize){
    int x = 0;
    int j = imageSize / 3;
    for(int i = 0; i < j; i++){
        RGBBlue[x] = 0;
        RGBBlue[x+1] = 0;
        RGBBlue[x+2] = 0;
        RGBBlue[x] = RGB[i].blue;
        x = x + 3;
    }
}

/*
 * This function creates the output rgb for inverted
 * @param: unsigned char pointer RGBinverted = contains the output rgb
 * @param: pointer array of struct pixel
 * @param: int imageSize = size of input image
 * @return: void
 */
void outputRGBinverted(unsigned char* RGBinverted, pixel* RGB, int imageSize){
    int x = 0;
    int j = imageSize / 3;
    for(int i = 0; i < j; i++ ){
        RGBinverted[x] = 0;
        RGBinverted[x+1] = 0;
        RGBinverted[x+2] = 0;
        RGBinverted[x] = 255 - RGB[i].blue;
        RGBinverted[x+1] = 255 - RGB[i].green;
        RGBinverted[x+2] = 255 - RGB[i].red;
        x = x + 3;
    }
}