// =============================================================== //
//                                                                 //
//   File      : MG_checkfield.cxx                                 //
//   Purpose   :                                                   //
//                                                                 //
//   Institute of Microbiology (Technical University Munich)       //
//   http://www.arb-home.de/                                       //
//                                                                 //
// =============================================================== //

#include "merge.hxx"

#include <item_sel_list.h>

#include <aw_awar.hxx>
#include <aw_root.hxx>
#include <aw_msg.hxx>
#include <arb_progress.h>

#include <arbdbt.h>
#include <arb_strbuf.h>
#include <arb_global_defs.h>

#include <cctype>

#define AWAR_CHECK AWAR_MERGE_TMP "chk/"

#define AWAR_COMPARE_FIELD AWAR_CHECK "compare"
#define AWAR_REPORT_FIELD  AWAR_CHECK "report"
#define AWAR_TOUPPER       AWAR_CHECK "ToUpper"
#define AWAR_EXCLUDE       AWAR_CHECK "exclude"
#define AWAR_CORRECT       AWAR_CHECK "correct"
#define AWAR_ETAG          AWAR_CHECK "tag"


static int gbs_cmp_strings(char *str1, char *str2, int *tab) { // returns 0 if strings are equal
    char *s1, *s2;
    int c1, c2;
    s1 = str1;
    s2 = str2;
    int count = 10;
    do {
        do { c1 = *(s1++); } while (tab[c1] < 0);
        do { c2 = *(s2++); } while (tab[c2] < 0);
        if (tab[c1] != tab[c2]) {   // difference found
            return 1;
        }
        count --;
    } while (count && c1 && c2);
    return 0;
}


static char *MG_diff_strings(char *str1, char * &str2, char *exclude, bool ToUpper, bool correct,
                             char **res1, char **res2, bool& corrrected) {

    char  buffer1[256];
    char  buffer2[256];
    char *dest1 = buffer1;
    char *dest2 = buffer2;
    char *s1, *s2;
    int   c1, c2;
    int   count = 3;
    int   tab[256];
    int   i;

    s1 = str1;
    s2 = str2;
    *dest1 = 0;
    *dest2 = 0;
    tab[0] = 0;
    char gapchar = '#';
    if (strlen(exclude)) gapchar = exclude[0];
    else                 exclude = NULp;

    for (i=1; i<256; i++) { // LOOP_VECTORIZED[!<6.0,!>=8.0]
        tab[i] = i;
        if (exclude && strchr(exclude, i)) {
            tab[i] = -1;
            continue;
        }
        if (ToUpper && i >= 'a' && i <= 'z') {
            tab[i] = i-'a'+'A';
        }
    }

    do {
        do { c1 = *(s1++); } while (tab[c1] < 0);
        do { c2 = *(s2++); } while (tab[c2] < 0);
        if (tab[c1] != tab[c2]) {   // difference found
            if (correct) {
                // check substitution
                {
                    int c = s2[-1];
                    s2[-1] = s1[-1];
                    if (toupper(c1) == toupper(c2) || !gbs_cmp_strings(s1, s2, &tab[0])) {
                        corrrected = true;
                        continue;
                    }
                    s2[-1]  = c;
                }

                // check insertion in s2
                if (!gbs_cmp_strings(s1-1, s2, &tab[0])) {
                    s2[-1] = gapchar;
                    do { c2 = *(s2++); } while (tab[c2] < 0); // eat s2
                    corrrected = true;
                    continue;
                }
                // check deletion in s2
                if (!gbs_cmp_strings(s1, s2-1, &tab[0])) {
                    int toins = c1;
                    char *toinspos = s2-1;
                    if (toinspos > str2) toinspos--;
                    if (tab[(unsigned char)toinspos[0]]> 0) { // real insertion
                        GBS_strstruct buf(strlen(str2+10));
                        int           pos = s2-str2-1;

                        buf.ncat(str2, pos);
                        buf.put(toins);
                        buf.cat(str2+pos);

                        freeset(str2, buf.release());

                        s2         = str2+pos+1;
                        corrrected = true;

                        continue;
                    }
                    int side=1; // 0 = left   1= right
                    if (tab[(unsigned char)s1[0]]<0) side = 0;
                    if (! side) {
                        while (toinspos > str2 &&
                                tab[(unsigned char)toinspos[-1]] < 0) toinspos--;
                    }
                    toinspos[0] = toins;
                    corrrected = true;
                    do { c1 = *(s1++); } while (tab[c1] < 0); // eat s1
                    continue;
                }

                // one correction rejected -> don't try further
                corrrected = false;
                correct    = false;
            }
            if (count >= 0) {
                sprintf(dest1, "%ti ", s1-str1-1);
                sprintf(dest2, "%ti ", s2-str2-1);
                dest1 += strlen(dest1);
                dest2 += strlen(dest2);
            }
            count --;
        }
    } while (c1 && c2);

    if (c1 || c2) {
        sprintf(dest1, "... %ti ", s1-str1-1);
        sprintf(dest2, "... %ti ", s2-str2-1);
        dest1 += strlen(dest1);
        dest2 += strlen(dest2);
    }
    if (count<0) {
        sprintf(dest1, "and %i more", 1-count);
        sprintf(dest2, "and %i more", 1-count);
        dest1 += strlen(dest1);
        dest2 += strlen(dest2);
    }
    if (strlen(buffer1)) {
        *res1 = ARB_strdup(buffer1);
        *res2 = ARB_strdup(buffer2);
    }
    else {
        *res1 = NULp;
        *res2 = NULp;
    }
    return NULp;
}

long mg_count_queried(GBDATA *gb_main) {
    long queried  = 0;
    for (GBDATA *gb_spec = GBT_first_species(gb_main);
         gb_spec;
         gb_spec = GBT_next_species(gb_spec))
    {
        if (IS_QUERIED_SPECIES(gb_spec)) queried++;
    }
    return queried;
}

static void mg_check_field_cb(AW_window *aww) {
    AW_root  *root         = aww->get_root();
    GB_ERROR  error        = NULp;
    char     *compareField = root->awar(AWAR_COMPARE_FIELD)->read_string();
    char     *exclude      = root->awar(AWAR_EXCLUDE)->read_string();
    bool      ToUpper      = root->awar(AWAR_TOUPPER)->read_int();
    bool      correct      = root->awar(AWAR_CORRECT)->read_int();
    char     *tag          = root->awar(AWAR_ETAG)->read_string();
    int       correctCount = 0;

    if (strcmp(compareField, NO_FIELD_SELECTED) == 0) {
        error = "Please select a field to compare";
    }

    if (!error) error = GB_begin_transaction(GLOBAL_gb_src);
    if (!error) error = GB_begin_transaction(GLOBAL_gb_dst);

    const char *reportField = NULp;
    if (!error) {
        reportField = prepare_and_get_selected_itemfield(root, AWAR_REPORT_FIELD, GLOBAL_gb_src, SPECIES_get_selector());
        if (!reportField) {
            error = GB_await_error();
        }
        else {
            const char *otherdb_reportField = prepare_and_get_selected_itemfield(root, AWAR_REPORT_FIELD, GLOBAL_gb_dst, SPECIES_get_selector());
            if (!otherdb_reportField) {
                error = GB_await_error();
            }
            else {
                mg_assert(strcmp(reportField, otherdb_reportField) == 0);
            }
        }
    }

    if (!error) {
        GBDATA *gb_src_species_data = GBT_get_species_data(GLOBAL_gb_src);
        GBDATA *gb_dst_species_data = GBT_get_species_data(GLOBAL_gb_dst);

        GBDATA *gb_src_species;
        GBDATA *gb_dst_species;

        // First step: count selected species
        arb_progress progress("Checking fields", mg_count_queried(GLOBAL_gb_src));

        // Delete all 'report' fields in target database
        for (gb_dst_species = GBT_first_species_rel_species_data(gb_dst_species_data);
             gb_dst_species && !error;
             gb_dst_species = GBT_next_species(gb_dst_species))
        {
            GBDATA *gbd    = GB_search(gb_dst_species, reportField, GB_FIND);
            if (gbd) error = GB_delete(gbd);
        }

        bool seenQueried    = false;
        for (gb_src_species = GBT_first_species_rel_species_data(gb_src_species_data);
             gb_src_species && !error;
             gb_src_species = GBT_next_species(gb_src_species))
        {
            { // Delete all 'report' fields in source database
                GBDATA *gbd    = GB_search(gb_src_species, reportField, GB_FIND);
                if (gbd) error = GB_delete(gbd);
            }

            if (!error) {
                if (IS_QUERIED_SPECIES(gb_src_species)) {
                    seenQueried = true;
                    const char *src_name = GBT_get_name_or_description(gb_src_species);
                    gb_dst_species       = GB_find_string(gb_dst_species_data, "name", src_name, GB_IGNORE_CASE, SEARCH_GRANDCHILD);
                    if (!gb_dst_species) {
                        aw_message(GBS_global_string("WARNING: Species %s not found in target DB", src_name));
                    }
                    else {
                        gb_dst_species = GB_get_father(gb_dst_species);

                        GBDATA *gb_src_field = GB_search(gb_src_species, compareField, GB_FIND);
                        GBDATA *gb_dst_field = GB_search(gb_dst_species, compareField, GB_FIND);

                        char *src_val = gb_src_field ? GB_read_as_tagged_string(gb_src_field, tag) : NULp;
                        char *dst_val = gb_dst_field ? GB_read_as_tagged_string(gb_dst_field, tag) : NULp;

                        if (src_val || dst_val) {
                            char *src_positions = NULp;
                            char *dst_positions = NULp;

                            if (src_val && dst_val) {
                                bool corrected = false;
                                MG_diff_strings(src_val, dst_val, exclude, ToUpper, correct, &src_positions, &dst_positions, corrected);
                                if (corrected) {
                                    error = GB_write_autoconv_string(gb_dst_field, dst_val);
                                    if (!error) {
                                        GB_write_flag(gb_dst_species, 1);
                                        correctCount++;
                                    }
                                }
                            }
                            else {
                                src_positions = GBS_global_string_copy("field missing in %s DB", src_val ? "other" : "this");
                                dst_positions = GBS_global_string_copy("field missing in %s DB", dst_val ? "other" : "this");
                            }

                            if (src_positions && !error) {
                                error             = GBT_write_string(gb_dst_species, reportField, dst_positions);
                                if (!error) error = GBT_write_string(gb_src_species, reportField, src_positions);
                            }

                            free(dst_positions);
                            free(src_positions);
                        }

                        free(dst_val);
                        free(src_val);
                    }
                    progress.inc_and_check_user_abort(error);
                }
            }
        }

        if (!seenQueried && !error) {
            error = "Empty hitlist in source database (nothing to do)";
        }
    }

    error = GB_end_transaction(GLOBAL_gb_src, error);
    error = GB_end_transaction(GLOBAL_gb_dst, error);

    if (error) {
        aw_message(error);
    }
    else if (correctCount) {
        aw_message(GBS_global_string("Corrected field content of %i species\n"
                                     "(species %s been marked in target database).",
                                     correctCount, correctCount == 1 ? "has" : "have"));
    }

    free(tag);
    free(exclude);
    free(compareField);
}


AW_window *create_mg_check_fields_window(AW_root *aw_root) {
    aw_root->awar_string(AWAR_COMPARE_FIELD, NO_FIELD_SELECTED);
    aw_root->awar_string(AWAR_REPORT_FIELD, "tmp", AW_ROOT_DEFAULT);
    aw_root->awar_string(AWAR_EXCLUDE, ".-", AW_ROOT_DEFAULT);
    aw_root->awar_string(AWAR_ETAG, "");
    aw_root->awar_int(AWAR_TOUPPER);
    aw_root->awar_int(AWAR_CORRECT);

    AW_window_simple *aws = new AW_window_simple;
    aws->init(aw_root, "MERGE_COMPARE_FIELD", "COMPARE DATABASE FIELD");
    aws->load_xfig("merge/seqcheck.fig");

    aws->callback(AW_POPDOWN);
    aws->create_button("CLOSE", "CLOSE", "C");

    aws->at("help");
    aws->callback(makeHelpCallback("checkfield.hlp"));
    aws->create_button("HELP", "HELP", "H");


    aws->at("exclude");
    aws->create_input_field(AWAR_EXCLUDE);

    aws->at("toupper");
    aws->create_toggle(AWAR_TOUPPER);

    aws->at("correct");
    aws->create_toggle(AWAR_CORRECT);

    aws->at("tag");
    aws->create_input_field(AWAR_ETAG, 6);

    create_itemfield_selection_button(aws, FieldSelDef(AWAR_COMPARE_FIELD, GLOBAL_gb_dst, SPECIES_get_selector(), FIELD_FILTER_STRING_READABLE), "source");
    create_itemfield_selection_button(aws, FieldSelDef(AWAR_REPORT_FIELD,  GLOBAL_gb_dst, SPECIES_get_selector(), FIELD_FILTER_STRING_WRITEABLE, "report-field", SF_ALLOW_NEW), "dest");

    aws->at("go");
    aws->highlight();
    aws->callback(mg_check_field_cb);
    aws->create_button("GO", "GO");

    return aws;
}