/*-
 * Copyright (c)2003 Citrus Project,
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include "dcmtk/config/osconfig.h"
#include "citrus_mapper_zone.h"

#ifdef HAVE_SYS_QUEUE_H
#include <sys/queue.h>
#else
#include "dcmtk/oficonv/queue.h"
#endif

#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "citrus_bcs.h"
#include "citrus_types.h"
#include "citrus_module.h"
#include "citrus_region.h"
#include "citrus_memstream.h"
#include "citrus_mmap.h"
#include "citrus_hash.h"
#include "citrus_mapper.h"

/* ---------------------------------------------------------------------- */

_CITRUS_MAPPER_DECLS(mapper_zone);
_CITRUS_MAPPER_DEF_OPS(mapper_zone);


/* ---------------------------------------------------------------------- */

struct _zone {
    uint32_t     z_begin;
    uint32_t     z_end;
};

struct _citrus_mapper_zone {
    struct _zone     mz_col;
    struct _zone     mz_row;
    int32_t      mz_col_offset;
    int32_t      mz_row_offset;
    int      mz_col_bits;
};

struct _parse_state {
    enum { S_BEGIN, S_OFFSET }  ps_state;
    union {
        uint32_t    u_imm;
        int32_t     s_imm;
        struct _zone    zone;
    } u;
#define ps_u_imm    u.u_imm
#define ps_s_imm    u.s_imm
#define ps_zone     u.zone
    int ps_top;
};

int
_citrus_mapper_zone_mapper_getops(struct _citrus_mapper_ops *ops)
{

    memcpy(ops, &_citrus_mapper_zone_mapper_ops,
           sizeof(_citrus_mapper_zone_mapper_ops));

    return (0);
}

#define BUFSIZE 20
#define T_ERR   0x100
#define T_IMM   0x101

static int
get_imm(struct _citrus_memory_stream *ms, struct _parse_state *ps)
{
    int c, i, sign = 0;
    char buf[BUFSIZE + 1];
    char *p;

    for (i = 0; i < BUFSIZE; i++) {
retry:
        c = _citrus_memory_stream_peek(ms);
        if (i == 0) {
            if (sign == 0 && (c == '+' || c == '-')) {
                sign = c;
                _citrus_memory_stream_getc(ms);
                goto retry;
            } else if (!_citrus_bcs_isdigit((uint8_t)c))
                break;
        } else if (!_citrus_bcs_isxdigit((uint8_t)c))
            if (!(i == 1 && c == 'x'))
                break;
        buf[i] = (char) _citrus_memory_stream_getc(ms);
    }
    buf[i] = '\0';
    ps->ps_u_imm = strtoul(buf, &p, 0);
    if ((p - buf) != i)
        return (T_ERR);
    if (sign == '-')
        ps->ps_u_imm = (unsigned long)-(long)ps->ps_u_imm;
    return (T_IMM);
}

static int
get_tok(struct _citrus_memory_stream *ms, struct _parse_state *ps)
{
    int c;

loop:
    c = _citrus_memory_stream_peek(ms);
    if (c == 0x00)
        return (EOF);
    if (_citrus_bcs_isspace((uint8_t)c)) {
        _citrus_memory_stream_getc(ms);
        goto loop;
    }

    switch (ps->ps_state) {
    case S_BEGIN:
        switch (c) {
        case ':':
        case '-':
        case '/':
            _citrus_memory_stream_getc(ms);
            return (c);
        case '0':
        case '1':
        case '2':
        case '3':
        case '4':
        case '5':
        case '6':
        case '7':
        case '8':
        case '9':
            return (get_imm(ms, ps));
        }
        break;
    case S_OFFSET:
        switch (c) {
        case '/':
            _citrus_memory_stream_getc(ms);
            return (c);
        case '+':
        case '-':
        case '0':
        case '1':
        case '2':
        case '3':
        case '4':
        case '5':
        case '6':
        case '7':
        case '8':
        case '9':
            return (get_imm(ms, ps));
        }
        break;
    }
    return (T_ERR);
}

static int
parse_zone(struct _citrus_memory_stream *ms, struct _parse_state *ps, struct _zone *z)
{

    if (get_tok(ms, ps) != T_IMM)
        return (-1);
    z->z_begin = ps->ps_u_imm;
    if (get_tok(ms, ps) != '-')
        return (-1);
    if (get_tok(ms, ps) != T_IMM)
        return (-1);
    z->z_end = ps->ps_u_imm;

    if (z->z_begin > z->z_end)
        return (-1);

    return (0);
}

static int
check_rowcol(struct _zone *z, int32_t ofs, uint32_t maxval)
{
    uint32_t remain;

    if (maxval != 0 && z->z_end >= maxval)
        return (-1);

    if (ofs > 0) {
        if (maxval == 0)
            /* this should 0x100000000 - z->z_end */
            remain = (z->z_end == 0) ? 0xFFFFFFFF :
                0xFFFFFFFF - z->z_end + 1;
        else
            remain = maxval - z->z_end;
        if ((uint32_t)ofs > remain)
            return (-1);
    } else if (ofs < 0) {
        if (z->z_begin < (uint32_t)-ofs)
            return (-1);
    }

    return (0);
}

static int
parse_var(struct _citrus_mapper_zone *mz, struct _citrus_memory_stream *ms)
{
    struct _parse_state ps;
    uint32_t colmax, rowmax;
    int isrc, ret;

    ps.ps_state = S_BEGIN;

    if (parse_zone(ms, &ps, &mz->mz_col))
        return (-1);

    ret = get_tok(ms, &ps);
    if (ret == '/') {
        /* rowzone / colzone / bits */
        isrc = 1;
        mz->mz_row = mz->mz_col;

        if (parse_zone(ms, &ps, &mz->mz_col))
            return (-1);
        if (get_tok(ms, &ps) != '/')
            return (-1);
        if (get_tok(ms, &ps) != T_IMM)
            return (-1);
        mz->mz_col_bits = ps.ps_u_imm;
        if (mz->mz_col_bits < 0 || mz->mz_col_bits > 32)
            return (-1);
        ret = get_tok(ms, &ps);
    } else {
        /* colzone */
        isrc = 0;
        mz->mz_col_bits = 32;
        mz->mz_row.z_begin = mz->mz_row.z_end = 0;
    }
    if (ret == ':') {
        /* offset */
        ps.ps_state = S_OFFSET;
        if (get_tok(ms, &ps) != T_IMM)
            return (-1);
        mz->mz_col_offset = ps.ps_s_imm;
        if (isrc) {
            /* row/col */
            mz->mz_row_offset = mz->mz_col_offset;
            if (get_tok(ms, &ps) != '/')
                return (-1);
            if (get_tok(ms, &ps) != T_IMM)
                return (-1);
            mz->mz_col_offset = ps.ps_s_imm;
        } else
            mz->mz_row_offset = 0;
        ret = get_tok(ms, &ps);
    }
    if (ret != EOF)
        return (-1);

    /* sanity check */
    colmax = (mz->mz_col_bits == 32) ? 0 : 1 << mz->mz_col_bits;
    rowmax = (mz->mz_col_bits == 0) ? 0 : 1 << (32-mz->mz_col_bits);
    if (check_rowcol(&mz->mz_col, mz->mz_col_offset, colmax))
        return (-1);
    if (check_rowcol(&mz->mz_row, mz->mz_row_offset, rowmax))
        return (-1);

    return (0);
}

static int
/*ARGSUSED*/
_citrus_mapper_zone_mapper_init(struct _citrus_mapper_area * ma ,
    struct _citrus_csmapper * cm, const char * dir ,
    const void * var, size_t lenvar,
    struct _citrus_mapper_traits * mt, size_t lenmt)
{
    struct _citrus_mapper_zone *mz;
    struct _citrus_memory_stream ms;
    struct _citrus_region r;
    (void) ma;
    (void) dir;

    if (lenmt < sizeof(*mt))
        return (EINVAL);

    mz = malloc(sizeof(*mz));
    if (mz == NULL)
        return (errno);

    mz->mz_col.z_begin = mz->mz_col.z_end = 0;
    mz->mz_row.z_begin = mz->mz_row.z_end = 0;
    mz->mz_col_bits = 0;
    mz->mz_row_offset = 0;
    mz->mz_col_offset = 0;

    _citrus_region_init(&r, CITRUS_DECONST(void *, var), lenvar);
    _citrus_memory_stream_bind(&ms, &r);
    if (parse_var(mz, &ms)) {
        free(mz);
        return (EINVAL);
    }
    cm->cm_closure = mz;
    mt->mt_src_max = mt->mt_dst_max = 1;    /* 1:1 converter */
    mt->mt_state_size = 0;          /* stateless */

    return (0);
}

static void
/*ARGSUSED*/
_citrus_mapper_zone_mapper_uninit(struct _citrus_csmapper *cm )
{
    if (cm && cm->cm_closure)
        free(cm->cm_closure);
}

static int
/*ARGSUSED*/
_citrus_mapper_zone_mapper_convert(struct _citrus_csmapper * cm,
    _citrus_index_t * dst, _citrus_index_t src,
    void * ps )
{
    struct _citrus_mapper_zone *mz = cm->cm_closure;
    uint32_t col, row;
    (void) ps;

    if (mz->mz_col_bits == 32) {
        col = src;
        row = 0;
        if (col < mz->mz_col.z_begin || col > mz->mz_col.z_end)
            return (_CITRUS_MAPPER_CONVERT_NONIDENTICAL);
        if (mz->mz_col_offset > 0)
            col += (uint32_t)mz->mz_col_offset;
        else
            col -= (uint32_t)-mz->mz_col_offset;
        *dst = col;
    } else {
        col = src & (((uint32_t)1 << mz->mz_col_bits) - 1);
        row = src >> mz->mz_col_bits;
        if (row < mz->mz_row.z_begin || row > mz->mz_row.z_end ||
            col < mz->mz_col.z_begin || col > mz->mz_col.z_end)
            return (_CITRUS_MAPPER_CONVERT_NONIDENTICAL);
        if (mz->mz_col_offset > 0)
            col += (uint32_t)mz->mz_col_offset;
        else
            col -= (uint32_t)-mz->mz_col_offset;
        if (mz->mz_row_offset > 0)
            row += (uint32_t)mz->mz_row_offset;
        else
            row -= (uint32_t)-mz->mz_row_offset;
        *dst = col | (row << mz->mz_col_bits);
    }
    return (_CITRUS_MAPPER_CONVERT_SUCCESS);
}

static void
/*ARGSUSED*/
_citrus_mapper_zone_mapper_init_state(void)
{

}
