//
//  h64e.c
//  H64E-2
//
//  Created by dianshi on 3/13/20.
//  Copyright © 2020 dianshi. All rights reserved.
//

#include "h64e.h"

int h64e_init(H64E_t *s)
{
    printf("init structure\n");
    s->fmt.column_size       = 16;
    s->fmt.offset_addr       = 0;
    s->fmt.flag_offset       = 0;
    s->fmt.flag_no_group     = 0;
    s->fmt.flag_hex          = 1;
    s->fmt.group             = H64E_G_BYTE;
    s->fmt.flag_output_types = 0;
    s->fmt.output_type       = H64E_O_NONE;
    
    s->sin = NULL;
    s->sout = NULL;
    return 0;
}

int h64e_check_param( H64E_t *fmt )
{
    printf("Check parametrs set\n");
    return 0;
}

int h64e_set_input(H64E_t *s, H64E_stream_in *sin)
{
    return 0;
}

int h64e_set_output(H64E_t *s, H64E_stream_out *sout)
{
    return 0;
}

int h64e_convert(H64E_t *s)
{
    return 0;
}

int h64e_destroy(H64E_t *s)
{
    return 0;
}

int h64e_fmt_init( H64E_format *fs)
{
    memset(fs,0,sizeof(H64E_format));
    
    return 0;
}

/*
 * out_size - allways give enought data to buffer otherwise it will partially write data, or flush data out more often
 */
int h64e_fmt_byte(H64E_format *fs, uint8_t *in_data, int32_t in_size, uint8_t *out_data, int32_t out_size)
{
    //printf("h64e_fmt_byte in %d out %d\n",in_size,out_size);
    int ret=0;
    int cur_size=0;
    int i, j=0;
    /*
     offset - max size 64bit = 0x(2bytes)+0000_0000_0000_0000(16bytes)+ ":"1(bytes) + whitespace(1byte)                                      = 19bytes
     hex values - 00(2bytes)*32+whitespace(1byte)*32+space_between_columns(1byte)+asciichar(1byte)*32+sapce_betweeen_ascii_columns(1byte)*32 = 1121
     new line - 1byte
     total = 19 + 1121 + 1 = 1141
     */
    const int SZ=2048; //offset values in hex , spaces, number of hex values, second ascii characters
    uint8_t buf[SZ+1];
    int buf_sz=0;
    int trail_size=0;
    
    for (i=0; i<in_size; i++)
    {
        //group byte output string
        if (((fs->group_fmt == H64E_G_BYTE)||(fs->group_fmt == H64E_G_NONE)) && (fs->output_fmt == H64E_O_STRING))
        {
            if (isprint(in_data[i]))
            {
                buf_sz = snprintf(&buf[0], SZ, "%c",(unsigned char)in_data[i]);
            } else {
                buf_sz = snprintf(&buf[0], SZ, ".",(unsigned char)in_data[i]);
            }
        } else {
        //group byte output
            buf_sz = snprintf(&buf[0], SZ, "%02x",in_data[i]);
        }
        //count converted chars per line
        fs->column_pos += 1;
        fs->total_output += 1;
        //add space if configured
        if (fs->f_space)
        {
            if (buf_sz<SZ)
            {
                buf[buf_sz] = ' ';
                buf_sz += 1;
            }
        }

        if (fs->column_pos == fs->column_size)
        {
            if (fs->f_new_line) {
                //set triger to newline
                fs->t_new_line = 1;
                fs->column_pos = 0;
                //set new line to buffer
                if (buf_sz+1<SZ)
                {
                    buf[buf_sz] = '\n';
                    buf_sz += 1;
                }
            }
        }
        //printf("D: %s",buf);
        if (trail_size+buf_sz<out_size)
        {
            for(j=0;j<buf_sz;j++)
            {
                out_data[trail_size+j] = buf[j];
            }
            ret = buf_sz;
            
        } else {
           //not enought output to handle data
            ret=-1;
        }
    }
#if 0
    printf("D:[");
    for (i=0;i<ret;i++)
    {
        printf("%c",(unsigned char)out_data[i]);
    }
    printf("]\n");
#endif

    return ret;
}

/*
 * out_size - allways give enought data to buffer otherwise it will partially write data, or flush data out more often
 */
int h64e_fmt_byte_align16(H64E_format *fs, uint8_t *in_data, int32_t in_size, uint8_t *out_data, int32_t out_size)
{
    //printf("h64e_fmt_byte in %d out %d\n",in_size,out_size);
    int ret=0;
    int cur_size=0;
    int i, j=0;
    /*
     offset - max size 64bit = 0x(2bytes)+0000_0000_0000_0000(16bytes)+ ":"1(bytes) + whitespace(1byte)                                      = 19bytes
     hex values - 00(2bytes)*32+whitespace(1byte)*32+space_between_columns(1byte)+asciichar(1byte)*32+sapce_betweeen_ascii_columns(1byte)*32 = 1121
     new line - 1byte
     total = 19 + 1121 + 1 = 1141
     */
    const int OUT_SZ=1024; //offset values in hex , spaces, number of hex values, second ascii characters
    const int BUF_SZ=64;
    uint8_t buf_offset[BUF_SZ+1];
    uint8_t buf1[BUF_SZ+1];
    uint8_t buf2[BUF_SZ+1];
    uint8_t buf3[BUF_SZ+1];
    uint8_t out_part1[OUT_SZ+1];
    uint8_t out_part2[OUT_SZ+1];
    uint8_t out_part3[OUT_SZ+1];
    int out_part1_sz=0;
    int out_part2_sz=0;
    int out_part3_sz=0;
    int buf_offset_sz=0;
    int buf1_sz=0;
    int buf2_sz=0;
    int buf3_sz=0;
    int loop_size=0;
    int trail_size=0;
    
    //set offset to output
    if (fs->f_offset)
    {
        //printf("Print offset\n");
        int offset = fs->start_offset + fs->total_output;
        buf_offset_sz = snprintf((char *)&buf_offset[0], BUF_SZ, "%08x: ",offset);
    }
    
    if (in_size%fs->column_size != 0)
    {
        loop_size = (in_size/fs->column_size+1)*fs->column_size;
    } else {
        loop_size = in_size;
    }
    
    
    //size shout be allways matching (size mod columnsize == 0)
    //non matching size should be used at the end, on last line of output
    //printf("Loop_size %d\n",loop_size);
    for (i=0; i<loop_size; i++)
    {
        if (i<in_size)
        {
            
            
            
            //group byte output
            if (fs->f_hex)
            {
                
                buf1_sz = snprintf((char *)&buf1[0], BUF_SZ, "%02x",in_data[i]);
            }
            
            //ascii output
            if (fs->f_ascii)
            {
                if (isprint(in_data[i]))
                {
                    buf2_sz = snprintf((char *)&buf2[0], BUF_SZ, "%c",(unsigned char)in_data[i]);
                } else {
                    buf2_sz = snprintf((char *)&buf2[0], BUF_SZ, ".");
                }
            }
            
            
            if (fs->f_space)
            {
                //printf("Space enabled\n");
                if (fs->f_ascii)
                {
                    if (buf2_sz<BUF_SZ)
                    {
                        buf2[buf2_sz] = ' ';
                        buf2_sz += 1;
                    }
                }
                if (fs->f_hex)
                {
                    if (buf1_sz<BUF_SZ)
                    {
                        buf1[buf1_sz] = ' ';
                        buf1_sz += 1;
                    }
                }
            }
            
            /*
            if (fs->f_output_types)
            {
                if (fs->output_fmt == H64E_O_INT8)
                {
                    int ii = in_data[i];
                    buf3_sz = snprintf((char *)&buf3[0], BUF_SZ, "%4d ", (int8_t)ii);
                } else if (fs->output_fmt == H64E_O_UINT8) {
                    unsigned int uu = in_data[i];
                    buf3_sz = snprintf((char *)&buf3[0], BUF_SZ, "%4u ", uu);
                }
            }
            */
            
        } else {
            printf("Empty spaces\n");
            //group byte output string
            if (((fs->group_fmt == H64E_G_BYTE)||(fs->group_fmt == H64E_G_NONE)) && (fs->output_fmt == H64E_O_STRING))
            {
                buf2_sz = snprintf((char *)&buf2[0], BUF_SZ, " ");
            } else {
                //group byte output
                buf2_sz = snprintf((char *)&buf2[0], BUF_SZ, "  ");
            }
            if (fs->f_space)
            {
                if (buf2_sz<BUF_SZ)
                {
                    buf2[buf2_sz] = ' ';
                    buf2_sz += 1;
                }
                if (buf3_sz<BUF_SZ)
                {
                    buf3[buf3_sz] = ' ';
                    buf3_sz += 1;
                }
            }
        }
        //count converted chars per line
        fs->column_pos += 1;
        fs->total_output += 1;

        if (fs->column_pos == fs->column_size)
        {
            if (fs->f_new_line) {
                //set triger to newline
                fs->t_new_line = 1;
                fs->column_pos = 0;
                //set new line to hex buffer
                //printf("%d %d %d\n",fs->f_ascii,fs->f_hex,fs->f_output_types);
                if ((fs->f_ascii==1) && (fs->f_output_types==0) )
                {
                    if (buf2_sz+1<BUF_SZ)
                    {
                        buf2[buf2_sz] = '\n';
                        buf2_sz += 1;
                    }
                }
                
                if ((fs->f_hex==1) && (fs->f_ascii==0) && (fs->f_output_types==0) )
                {
                    if (buf1_sz+1<BUF_SZ)
                    {
                        buf1[buf1_sz] = '\n';
                        buf1_sz += 1;
                    }
                }
                
                /*
                if (fs->f_output_types)
                {
                    if (buf2_sz+1<BUF_SZ)
                    {
                        buf2[buf2_sz] = '\n';
                        buf2_sz += 1;
                    }
                } else {
                    if (buf1_sz+1<BUF_SZ)
                    {
                        buf1[buf1_sz] = '\n';
                        buf1_sz += 1;
                    }
                }
                */
            }
        }

        //copy data to buffer after each iteration
        for (j=0; j<buf1_sz; j++)
        {
            out_part1[out_part1_sz+j] = buf1[j];
        }
        out_part1_sz += buf1_sz;
        buf1_sz = 0;
        
        for (j=0; j<buf2_sz; j++)
        {
            out_part2[out_part2_sz+j] = buf2[j];
        }
        out_part2_sz += buf2_sz;
        buf2_sz = 0;
        
        for (j=0; j<buf3_sz; j++)
        {
            out_part3[out_part3_sz+j] = buf3[j];
        }
        out_part3_sz += buf3_sz;
        buf3_sz = 0;
    }
    
    //printf("buf_offset_sz %d\n",buf_offset_sz);
    //printf("out_part1_sz %d\n",out_part1_sz);
    //printf("out_part2_sz %d\n",out_part2_sz);
    //printf("out_part3_sz %d\n",out_part3_sz);
    
    //concat all data from 3 buffers
    for (i=0;i<buf_offset_sz;i++)
    {
        if (i<out_size)
        {
            out_data[i] = buf_offset[i];
            trail_size += 1;
        }
    }
    
    for (i=0,j=trail_size;i<out_part1_sz;i++)
    {
        if (j+i<out_size)
        {
            out_data[j+i] = out_part1[i];
            trail_size += 1;
        }
    }
    
    //if space between sapce disabled then put one space between ascii&hex output
    //printf("%d %d %d\n",fs->f_ascii,fs->f_hex,fs->f_space);
    if ((fs->f_ascii==1) && (fs->f_hex==1) && (fs->f_space==0))
    {
        if (trail_size<out_size)
        {
            out_data[trail_size] = ' ';
            trail_size += 1;
        }
    }
    
    for (i=0,j=trail_size;i<out_part2_sz;i++)
    {
        if (j+i<out_size)
        {
            out_data[j+i] = out_part2[i];
            trail_size += 1;
        }
    }
    
    /*
    for (i=0;i<out_part3_sz;i++)
    {
        if (i<out_size)
        {
            out_data[j+i] = out_part3[i];
            trail_size += 1;
        }
    }
    */
    
    ret = trail_size;
    //printf("ret %d\n",ret);
    
#if 0
    printf("D:[");
    for (i=0;i<ret;i++)
    {
        printf("%c",(unsigned char)out_data[i]);
    }
    printf("]\n");
#endif

    return ret;
}

int h64e_fmt_word(  H64E_format *fs, uint8_t *in_data, int32_t in_size, uint8_t *out_data, int32_t out_size)
{
    return 0;
}

int h64e_fmt_dword(  H64E_format *fs, uint8_t *in_data, int32_t in_size, uint8_t *out_data, int32_t out_size)
{
    return 0;
}

int h64e_fmt_qword(  H64E_format *fs, uint8_t *in_data, int32_t in_size, uint8_t *out_data, int32_t out_size)
{
    return 0;
}
int h64e_fmt_string(  H64E_format *fs, uint8_t *in_data, int32_t in_size, uint8_t *out_data, int32_t out_size)
{
    return 0;
}

int h64e_fmt_finish(H64E_format *fs)
{
    
    
    return 0;
}