#include <stdio.h>
#include "math.h"
#include "pHash.h"

using namespace cimg_library;

void sort_names(char **names, int L1) {
    for (int i = 0; i < L1; i++) {
        int min = i;
        for (int j = i + 1; j < L1; j++) {
            if (strcmp(names[j], names[min]) <= 0) min = j;
        }
        if (i != min) {
            char *swap = names[i];
            names[i] = names[min];
            names[min] = swap;
        }
    }
}

int main(int argc, char **argv) {
    if (argc < 3) {
        printf(" not enough input args\n");
        printf(" %s dir1 dir2\n", argv[0]);
        return -1;
    }

    const char *dirname1 = argv[1];
    const char *dirname2 = argv[2];

    int alpha = 2;
    int level = 2;

    int nbfiles1;
    char **files1 = ph_readfilenames(dirname1, nbfiles1);
    sort_names(files1, nbfiles1);

    int nbfiles2;
    char **files2 = ph_readfilenames(dirname2, nbfiles2);
    sort_names(files2, nbfiles2);

    if (nbfiles1 != nbfiles2) {
        printf("number files in both directories not equal\n");
        return -2;
    }
    printf("nb files %d\n", nbfiles1);

    uint8_t **hash1 = (uint8_t **)malloc(nbfiles1 * sizeof(uint8_t *));
    uint8_t *hash2 = NULL;
    int count = 0;
    int hashlen1 = 0, hashlen2 = 0;
    double dist = 0;
    printf("intra distances\n");
    printf("***************\n");
    for (int i = 0; i < nbfiles1; i++) {
        printf("file1[%d]: %s\n", i, files1[i]);
        hash1[count] = ph_mh_imagehash(files1[i], hashlen1, alpha, level);
        if (hash1 == NULL) continue;

        printf("file2[%d]: %s\n", i, files2[i]);
        hash2 = ph_mh_imagehash(files2[i], hashlen2, alpha, level);
        if (hash2 == NULL) continue;

        dist = ph_hammingdistance2(hash1[count], hashlen1, hash2, hashlen2);
        printf("distance = %f\n", dist);
        printf("-------------\n");
        free(hash2);
        count++;
    }
    printf("\n\n");
    printf("--hit any key--\n");
    getchar();
    printf("inter distances\n");
    printf("****************\n");
    for (int i = 0; i < count; i++) {
        for (int j = i + 1; j < count; j++) {
            dist = ph_hammingdistance2(hash1[i], hashlen1, hash1[j], hashlen1);
            printf(" files1[%d] files2[%d] dist = %f\n", i, j, dist);
            printf("----------------\n");
        }
    }
    printf("done\n");

    return 0;
}