//
//  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.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 buf1[BUF_SZ+1];
    uint8_t buf2[BUF_SZ+1];
    uint8_t out_part1[OUT_SZ+1];
    uint8_t out_part2[OUT_SZ+1];
    int out_part1_sz=0;
    int out_part2_sz=0;
    int buf1_sz=0;
    int buf2_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;
        buf1_sz = snprintf((char *)&buf1[0], BUF_SZ, "%08x: ",offset);
        //no error check
        memcpy(out_part1,buf1,buf1_sz);
        out_part1_sz += buf1_sz;
    }
    
    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 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]))
                {
                    buf1_sz = snprintf((char *)&buf1[0], BUF_SZ, "%c",(unsigned char)in_data[i]);
                } else {
                    buf1_sz = snprintf((char *)&buf1[0], BUF_SZ, ".");
                }
            } else {
            //group byte output
                buf1_sz = snprintf((char *)&buf1[0], BUF_SZ, "%02x",in_data[i]);
            }
            if (fs->f_space)
            {
                //printf("Space enabled\n");
                if (buf1_sz<BUF_SZ)
                {
                    buf1[buf1_sz] = ' ';
                    buf1_sz += 1;
                }
            }
            if (fs->f_output_types)
            {
                if (fs->output_fmt == H64E_O_INT8)
                {
                    buf2_sz = snprintf((char *)&buf2[0], BUF_SZ, "%4d ", (int8_t)in_data[i]);
                } else if (fs->output_fmt == H64E_O_UINT8) {
                    buf2_sz = snprintf((char *)&buf2[0], BUF_SZ, "%4u ", (uint8_t)in_data[i]);
                }
            }
            
        } 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))
            {
                buf1_sz = snprintf((char *)&buf1[0], BUF_SZ, " ");
            } else {
                //group byte output
                buf1_sz = snprintf((char *)&buf1[0], BUF_SZ, "  ");
            }
            if (fs->f_space)
            {
                if (buf1_sz<BUF_SZ)
                {
                    buf1[buf1_sz] = ' ';
                    buf1_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 buffer
                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;
        
        for (j=0; j<buf2_sz; j++)
        {
            out_part2[out_part2_sz+j] = buf2[j];
        }
        out_part2_sz += buf2_sz;
    }
    
    //printf("out_part1_sz %d\n",out_part1_sz);
    //printf("out_part2_sz %d\n",out_part2_sz);
    
    //concat all data from 2 buffers
    for (i=0;i<out_part1_sz;i++)
    {
        if (i<out_size)
        {
            out_data[i] = out_part1[i];
            trail_size += 1;
        }
    }
    
    for (i=0;i<out_part2_sz;i++)
    {
        if (out_part1_sz+i<out_size)
        {
            out_data[i] = out_part2[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;
}