#include "p25p1_check_ldu.h"

#include "Hamming.hpp"
#include "ReedSolomon.hpp"


//#define __CHECK_HDU_DEBUG__


int check_and_fix_hamming_10_12_13(char* hex, char* parity)
{
    static Hamming_10_6_3_TableImpl hamming;

    return hamming.decode(hex, parity);
}

int check_and_fix_redsolomon_24_12_13(char* data, char* parity)
{
    static DSDReedSolomon_24_12_13 reed_solomon_24_12_13;

#ifdef __CHECK_HDU_DEBUG__
    char original[12][6];
    for (int i = 0; i < 12; i++) {
        for (int j = 0; j < 6; j++) {
            original[i][j] = data[i*6+j];
        }
    }
#endif

    int irrecoverable_error = reed_solomon_24_12_13.decode(data, parity);

#ifdef __CHECK_HDU_DEBUG__
    printf("Results for Reed-Solomon code (24,12,13)\n\n");
    if (irrecoverable_error == 0) {
        printf("  i  original fixed\n");
        for (int i = 0; i < 12; i++) {
            printf("%3d  [", i);
            for (int j = 0; j < 6; j++) {
                printf("%c", (original[i][j] == 1)? 'X' : ' ');
            }
            printf("] [");
            for (int j = 0; j < 6; j++) {
                printf("%c", (data[i*6+j] == 1)? 'X' : ' ');
            }
            printf("]\n");
        }
    } else {
        printf("Irrecoverable errors found\n");
        printf("  i  original fixed\n");
        for (int i = 0; i < 12; i++) {
            printf("%3d  [", i);
            for (int j = 0; j < 6; j++) {
                printf("%c", (original[i][j] == 1)? 'X' : ' ');
            }
            printf("]\n");
        }
    }
    printf("\n");
#endif

    return irrecoverable_error;
}

int check_and_fix_redsolomon_24_16_9(char* data, char* parity)
{
    static DSDReedSolomon_24_16_9 reed_solomon_24_16_9;

#ifdef __CHECK_HDU_DEBUG__
    char original[16][6];
    for (int i = 0; i < 16; i++) {
        for (int j = 0; j < 6; j++) {
            original[i][j] = data[i*6+j];
        }
    }
#endif

    int irrecoverable_error = reed_solomon_24_16_9.decode(data, parity);

#ifdef __CHECK_HDU_DEBUG__
    printf("Results for Reed-Solomon code (24,16,9)\n\n");
    if (irrecoverable_error == 0) {
        printf("  i  original fixed\n");
        for (int i = 0; i < 16; i++) {
            printf("%3d  [", i);
            for (int j = 0; j < 6; j++) {
                printf("%c", (original[i][j] == 1)? 'X' : ' ');
            }
            printf("] [");
            for (int j = 0; j < 6; j++) {
                printf("%c", (data[i*6+j] == 1)? 'X' : ' ');
            }
            printf("]\n");
        }
    } else {
        printf("Irrecoverable errors found\n");
        printf("  i  original fixed\n");
        for (int i = 0; i < 16; i++) {
            printf("%3d  [", i);
            for (int j = 0; j < 6; j++) {
                printf("%c", (original[i][j] == 1)? 'X' : ' ');
            }
            printf("]\n");
        }
    }
    printf("\n");
#endif

    return irrecoverable_error;
}