#include <stdio.h>

#include "dictre.h"

extern struct Word words[];

extern int nwords;

extern struct Def defs[];

extern int ndefs;

static char num_to_ia5char(int n)

{

    /* From RFC 1421 */

    if (n >= 0 && n <= 25)

        return 'A' + n;

    else if (n >= 26 && n <= 51)

        return 'a' + (n - 26);

    else if (n >= 52 && n <= 61)

        return '0' + (n - 52);

    else if (n == 62)

        return '+';

    else if (n == 63)

        return '-';

    else

        return '!'; /* Error */

}

static int num_to_ia5(char *dest, int n)

{

    char tmp[20];

    int i, max;

    for(i =0; i <= 10; ++i)

    {

        tmp[i] = num_to_ia5char(n % 64);

        if (n < 64)

            break;

        n /= 64;

    }

    max = i;

    /* reverse the number */

    for (i=0; i<=max; ++i)

        dest[i] = tmp[max-i];

    /* Ending '\0' */

    dest[max+1] = '\0';

    return max;

}

static int write_dictionary_data(FILE *fdefs)

{

    int i;

    int offset = 0;

    for (i=0; i < ndefs; ++i)

    {

        fwrite(defs[i].d, defs[i].length, 1, fdefs);

        defs[i].offset = offset;

        offset += defs[i].length;

    }

}

static void write_dictionary_index(FILE *index)

{

    int i;

    for (i=0; i < nwords; ++i)

    {

        char offset_str[50];

        char length_str[50];

        num_to_ia5(offset_str, defs[words[i].def].offset);

        num_to_ia5(length_str, defs[words[i].def].length);

        fprintf(index, "%s\t%s\t%s\n",

                words[i].w, offset_str, length_str);

    }

}

void write_dictionary(const char *name)

{

    FILE *i, *d;

    char tmpname[500];

    strcpy(tmpname, name);

    strcat(tmpname, ".dict");

    d = fopen(tmpname, "wb");

    write_dictionary_data(d);

    fclose(d);

    strcpy(tmpname, name);

    strcat(tmpname, ".index");

    i = fopen(tmpname, "wb");

    write_dictionary_index(i);

    fclose(i);

}