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

#define BUFFER_SIZE 100
#define FILE_HEADER_STRING "__info2_image_file_format__"

// TODO Implementieren Sie geeignete Hilfsfunktionen für das Lesen der Bildserie aus einer Datei

// TODO Vervollständigen Sie die Funktion readImages unter Benutzung Ihrer Hilfsfunktionen
GrayScaleImageSeries *readImages(const char *path)
{
    FILE *file = fopen(path, "rb");
    if (file == NULL)
    {
        return NULL;
    }

    // Überprüfe den Header
    char fileTag[strlen(FILE_HEADER_STRING)];
    fread(fileTag, sizeof(fileTag[0]), strlen(FILE_HEADER_STRING), file);

    if (strcmp(fileTag, FILE_HEADER_STRING) != 0)
    {
        fclose(file);
        return NULL;
    }

    // Lese die Metadaten: Anzahl der Bilder, Breite und Höhe
    unsigned short numberOfImages, width, height;
    fread(&numberOfImages, sizeof(numberOfImages), 1, file);
    fread(&width, sizeof(width), 1, file);
    fread(&height, sizeof(height), 1, file);

    GrayScaleImageSeries *series = (GrayScaleImageSeries *)malloc(sizeof(GrayScaleImageSeries));
    if (series == NULL)
    {
        fclose(file);
        return NULL;
    }

    series->count = numberOfImages;
    series->images = (GrayScaleImage *)malloc(numberOfImages * sizeof(GrayScaleImage));
    series->labels = (unsigned char *)malloc(numberOfImages * sizeof(unsigned char));

    if (series->images == NULL || series->labels == NULL)
    {
        free(series);
        fclose(file);
        return NULL;
    }

    for (int i = 0; i < numberOfImages; i++)
    {
        series->images[i].width = width;
        series->images[i].height = height;
        series->images[i].buffer = (GrayScalePixelType *)malloc(width * height * sizeof(GrayScalePixelType));

        if (series->images[i].buffer == NULL)
        {
            // Fehlerbehandlung: Speicher freigeben, wenn malloc fehlschlägt
            for (int j = 0; j < i; j++)
            {
                free(series->images[j].buffer);
            }
            free(series->images);
            free(series->labels);
            free(series);
            fclose(file);
            return NULL;
        }

        // Lese die Pixel-Daten und das Label
        fread(series->images[i].buffer, sizeof(GrayScalePixelType), width * height, file);
        fread(&series->labels[i], sizeof(unsigned char), 1, file);
    }

    fclose(file);
    return series;
}

// TODO Vervollständigen Sie die Funktion clearSeries, welche eine Bildserie vollständig aus dem Speicher freigibt
void clearSeries(GrayScaleImageSeries *series)
{
    if (series == NULL)
        return;

    for (int i = 0; i < series->count; i++)
    {
        free(series->images[i].buffer); // Speicher für das Bild freigeben
    }

    free(series->images); // Speicher für die Bild-Array freigeben
    free(series->labels); // Speicher für die Labels freigeben
    free(series);
}