/*
 * Copyright 2005-2017 ECMWF.
 *
 * This software is licensed under the terms of the Apache Licence Version 2.0
 * which can be obtained at http://www.apache.org/licenses/LICENSE-2.0.
 *
 * In applying this licence, ECMWF does not waive the privileges and immunities granted to it by
 * virtue of its status as an intergovernmental organisation nor does it submit to any jurisdiction.
 */

#include "grib_api_internal.h"

/*
   This is used by make_class.pl

   START_CLASS_DEF
   CLASS      = accessor
   SUPER      = grib_accessor_class_data_simple_packing
   IMPLEMENTS = init
   IMPLEMENTS = pack_double
   IMPLEMENTS = unpack_double
   IMPLEMENTS = value_count
   MEMBERS=const char* half_byte
   MEMBERS=const char* packingType
   MEMBERS=const char* ieee_packing
   MEMBERS=const char* precision
   MEMBERS=const char* widthOfFirstOrderValues
   MEMBERS=const char* N1
   MEMBERS=const char* N2
   MEMBERS=const char* numberOfGroups
   MEMBERS=const char* numberOfSecondOrderPackedValues
   MEMBERS=const char* extraValues
   MEMBERS=const char* pl
   MEMBERS=const char* Ni
   MEMBERS=const char* Nj
   MEMBERS=const char* jPointsAreConsecutive
   MEMBERS=const char* bitmap
   MEMBERS=const char* groupWidths

   END_CLASS_DEF

 */

/* START_CLASS_IMP */

/*

Don't edit anything between START_CLASS_IMP and END_CLASS_IMP
Instead edit values between START_CLASS_DEF and END_CLASS_DEF
or edit "accessor.class" and rerun ./make_class.pl

*/

static int pack_double(grib_accessor*, const double* val,size_t *len);
static int unpack_double(grib_accessor*, double* val,size_t *len);
static int value_count(grib_accessor*,long*);
static void init(grib_accessor*,const long, grib_arguments* );
static void init_class(grib_accessor_class*);

typedef struct grib_accessor_data_g1second_order_row_by_row_packing {
    grib_accessor          att;
/* Members defined in gen */
/* Members defined in values */
	int  carg;
	const char* seclen;
	const char* offsetdata;
	const char* offsetsection;
	int dirty;
/* Members defined in data_simple_packing */
	int edition;
	const char*  units_factor;
	const char*  units_bias;
	const char*  changing_precision;
	const char*  number_of_values;
	const char*  bits_per_value;
	const char*  reference_value;
	const char*  binary_scale_factor;
	const char*  decimal_scale_factor;
	const char*  optimize_scaling_factor;
/* Members defined in data_g1second_order_row_by_row_packing */
	const char* half_byte;
	const char* packingType;
	const char* ieee_packing;
	const char* precision;
	const char* widthOfFirstOrderValues;
	const char* N1;
	const char* N2;
	const char* numberOfGroups;
	const char* numberOfSecondOrderPackedValues;
	const char* extraValues;
	const char* pl;
	const char* Ni;
	const char* Nj;
	const char* jPointsAreConsecutive;
	const char* bitmap;
	const char* groupWidths;
} grib_accessor_data_g1second_order_row_by_row_packing;

extern grib_accessor_class* grib_accessor_class_data_simple_packing;

static grib_accessor_class _grib_accessor_class_data_g1second_order_row_by_row_packing = {
    &grib_accessor_class_data_simple_packing,                      /* super                     */
    "data_g1second_order_row_by_row_packing",                      /* name                      */
    sizeof(grib_accessor_data_g1second_order_row_by_row_packing),  /* size                      */
    0,                           /* inited */
    &init_class,                 /* init_class */
    &init,                       /* init                      */
    0,                  /* post_init                      */
    0,                    /* free mem                       */
    0,                       /* describes himself         */
    0,                /* get length of section     */
    0,              /* get length of string      */
    &value_count,                /* get number of values      */
    0,                 /* get number of bytes      */
    0,                /* get offset to bytes           */
    0,            /* get native type               */
    0,                /* get sub_section                */
    0,               /* grib_pack procedures long      */
    0,                 /* grib_pack procedures long      */
    0,                  /* grib_pack procedures long      */
    0,                /* grib_unpack procedures long    */
    &pack_double,                /* grib_pack procedures double    */
    &unpack_double,              /* grib_unpack procedures double  */
    0,                /* grib_pack procedures string    */
    0,              /* grib_unpack procedures string  */
    0,          /* grib_pack array procedures string    */
    0,        /* grib_unpack array procedures string  */
    0,                 /* grib_pack procedures bytes     */
    0,               /* grib_unpack procedures bytes   */
    0,            /* pack_expression */
    0,              /* notify_change   */
    0,                /* update_size   */
    0,            /* preferred_size   */
    0,                    /* resize   */
    0,      /* nearest_smaller_value */
    0,                       /* next accessor    */
    0,                    /* compare vs. another accessor   */
    0,     /* unpack only ith value          */
    0,     /* unpack a subarray         */
    0,              		/* clear          */
    0,               		/* clone accessor          */
};


grib_accessor_class* grib_accessor_class_data_g1second_order_row_by_row_packing = &_grib_accessor_class_data_g1second_order_row_by_row_packing;


static void init_class(grib_accessor_class* c)
{
	c->dump	=	(*(c->super))->dump;
	c->next_offset	=	(*(c->super))->next_offset;
	c->string_length	=	(*(c->super))->string_length;
	c->byte_count	=	(*(c->super))->byte_count;
	c->byte_offset	=	(*(c->super))->byte_offset;
	c->get_native_type	=	(*(c->super))->get_native_type;
	c->sub_section	=	(*(c->super))->sub_section;
	c->pack_missing	=	(*(c->super))->pack_missing;
	c->is_missing	=	(*(c->super))->is_missing;
	c->pack_long	=	(*(c->super))->pack_long;
	c->unpack_long	=	(*(c->super))->unpack_long;
	c->pack_string	=	(*(c->super))->pack_string;
	c->unpack_string	=	(*(c->super))->unpack_string;
	c->pack_string_array	=	(*(c->super))->pack_string_array;
	c->unpack_string_array	=	(*(c->super))->unpack_string_array;
	c->pack_bytes	=	(*(c->super))->pack_bytes;
	c->unpack_bytes	=	(*(c->super))->unpack_bytes;
	c->pack_expression	=	(*(c->super))->pack_expression;
	c->notify_change	=	(*(c->super))->notify_change;
	c->update_size	=	(*(c->super))->update_size;
	c->preferred_size	=	(*(c->super))->preferred_size;
	c->resize	=	(*(c->super))->resize;
	c->nearest_smaller_value	=	(*(c->super))->nearest_smaller_value;
	c->next	=	(*(c->super))->next;
	c->compare	=	(*(c->super))->compare;
	c->unpack_double_element	=	(*(c->super))->unpack_double_element;
	c->unpack_double_subarray	=	(*(c->super))->unpack_double_subarray;
	c->clear	=	(*(c->super))->clear;
	c->make_clone	=	(*(c->super))->make_clone;
}

/* END_CLASS_IMP */

static void init(grib_accessor* a,const long v, grib_arguments* args)
{
    grib_accessor_data_g1second_order_row_by_row_packing *self =(grib_accessor_data_g1second_order_row_by_row_packing*)a;
    grib_handle* gh = grib_handle_of_accessor(a);

    self->half_byte    = grib_arguments_get_name(gh,args,self->carg++);
    self->packingType    = grib_arguments_get_name(gh,args,self->carg++);
    self->ieee_packing    = grib_arguments_get_name(gh,args,self->carg++);
    self->precision    = grib_arguments_get_name(gh,args,self->carg++);
    self->widthOfFirstOrderValues    = grib_arguments_get_name(gh,args,self->carg++);
    self->N1 = grib_arguments_get_name(gh,args,self->carg++);
    self->N2 = grib_arguments_get_name(gh,args,self->carg++);
    self->numberOfGroups = grib_arguments_get_name(gh,args,self->carg++);
    self->numberOfSecondOrderPackedValues = grib_arguments_get_name(gh,args,self->carg++);
    self->extraValues = grib_arguments_get_name(gh,args,self->carg++);
    self->Ni = grib_arguments_get_name(gh,args,self->carg++);
    self->Nj = grib_arguments_get_name(gh,args,self->carg++);
    self->pl = grib_arguments_get_name(gh,args,self->carg++);
    self->jPointsAreConsecutive = grib_arguments_get_name(gh,args,self->carg++);
    self->groupWidths = grib_arguments_get_name(gh,args,self->carg++);
    self->bitmap = grib_arguments_get_name(gh,args,self->carg++);
    self->edition=1;
    a->flags |= GRIB_ACCESSOR_FLAG_DATA;
}

static int value_count(grib_accessor* a,long* count)
{
    grib_accessor_data_g1second_order_row_by_row_packing *self =(grib_accessor_data_g1second_order_row_by_row_packing*)a;
    grib_handle* gh = grib_handle_of_accessor(a);
    long n=0,i=0;
    long numberOfRows=0;
    long jPointsAreConsecutive=0;
    long Ni=0,Nj=0;
    int bitmapPresent=0;
    size_t plSize=0;
    long* pl=0;
    int ret=0;
    grib_context* c=a->context;

    if (self->bitmap) bitmapPresent=1;
    if((ret=grib_get_long_internal(gh,self->jPointsAreConsecutive,&jPointsAreConsecutive)) != GRIB_SUCCESS)
        return ret;
    if((ret=grib_get_long_internal(gh,self->Ni,&Ni)) != GRIB_SUCCESS)
        return ret;
    if((ret=grib_get_long_internal(gh,self->Nj,&Nj)) != GRIB_SUCCESS)
        return ret;
    if (jPointsAreConsecutive) {
        numberOfRows=Ni;
    } else {
        numberOfRows=Nj;
    }

    plSize=0;
    ret=grib_get_size(gh,self->pl,&plSize);
    if (ret==GRIB_SUCCESS) {
        pl=(long*)grib_context_malloc_clear(a->context,sizeof(long)*plSize);
        if((ret=grib_get_long_array(gh,self->pl,pl,&plSize)) != GRIB_SUCCESS)
            return ret;
    }
    ret=0;

    n=0;
    if (bitmapPresent) {
        long *bitmap,*pbitmap;
        size_t numberOfPoints = 0;

        if (plSize) {
            for (i=0;i<numberOfRows;i++) numberOfPoints+=pl[i];
            grib_context_free(c,pl);
        } else {
            numberOfPoints=Ni*Nj;
        }
        bitmap=(long*)grib_context_malloc_clear(a->context,sizeof(long)*numberOfPoints);
        pbitmap=bitmap;
        grib_get_long_array(gh,self->bitmap,bitmap,&numberOfPoints);
        for (i=0;i<numberOfPoints;i++) n+=*(bitmap++);

        grib_context_free(a->context,pbitmap);
    } else {
        if (plSize) {
            for (i=0;i<numberOfRows;i++) n+=pl[i];
            grib_context_free(c,pl);
        } else {
            n=Ni*Nj;
        }
    }

    *count=n;
    return ret;
}

static int unpack_double(grib_accessor* a, double* values, size_t *len)
{
    grib_accessor_data_g1second_order_row_by_row_packing* self =  (grib_accessor_data_g1second_order_row_by_row_packing*)a;
    grib_handle* gh = grib_handle_of_accessor(a);
    int ret=0;
    long numberOfGroups,numberOfSecondOrderPackedValues;
    long* groupWidths=0;
    long* firstOrderValues=0;
    long* X=0;
    long numberOfRows,numberOfColumns;
    long *numbersPerRow;
    long pos=0;
    long widthOfFirstOrderValues=0;
    long jPointsAreConsecutive;
    unsigned char* buf = (unsigned char*)gh->buffer->data;
    long k,i,j,n,Ni,Nj;
    double reference_value;
    long binary_scale_factor;
    long decimal_scale_factor;
    double s,d;
    size_t groupWidthsSize=0;
    int bitmapPresent=0;
    size_t plSize=0;
    long* pl=0;

    buf += grib_byte_offset(a);

    if((ret=grib_get_long_internal(gh,self->numberOfGroups,&numberOfGroups)) != GRIB_SUCCESS)
        return ret;

    if((ret=grib_get_long_internal(gh,self->jPointsAreConsecutive,&jPointsAreConsecutive)) != GRIB_SUCCESS)
        return ret;

    if (self->bitmap) bitmapPresent=1;
    ret=grib_get_size(gh,self->pl,&plSize);
    if (ret==GRIB_SUCCESS) {
        pl=(long*)grib_context_malloc_clear(a->context,sizeof(long)*plSize);
        if((ret=grib_get_long_array(gh,self->pl,pl,&plSize)) != GRIB_SUCCESS)
            return ret;
    }

    if((ret=grib_get_long_internal(gh,self->Ni,&Ni)) != GRIB_SUCCESS)
        return ret;
    if((ret=grib_get_long_internal(gh,self->Nj,&Nj)) != GRIB_SUCCESS)
        return ret;
    if (jPointsAreConsecutive) {
        numberOfRows=Ni;
        numberOfColumns=Nj;
    } else {
        numberOfRows=Nj;
        numberOfColumns=Ni;
    }

    numbersPerRow=(long*)grib_context_malloc_clear(a->context,sizeof(long)*numberOfRows);
    if (bitmapPresent) {
        long *bitmap,*pbitmap;
        size_t numberOfPoints=Ni*Nj;

        if (plSize) {
            numberOfPoints=0;
            for (i=0;i<numberOfRows;i++) numberOfPoints+=pl[i];
        }
        bitmap=(long*)grib_context_malloc_clear(a->context,sizeof(long)*numberOfPoints);
        pbitmap=bitmap;
        grib_get_long_array(gh,self->bitmap,bitmap,&numberOfPoints);
        if (plSize) {
            for (i=0;i<numberOfRows;i++) {
                for (j=0;j<pl[i];j++) {
                    numbersPerRow[i]+=*(bitmap++);
                }
            }
        } else {
            for (i=0;i<numberOfRows;i++) {
                numbersPerRow[i]=0;
                for (j=0;j<Ni;j++) {
                    numbersPerRow[i]+=*(bitmap++);
                }
            }
        }

        grib_context_free(a->context,pbitmap);
    } else {
        if (plSize) {
            for (i=0;i<numberOfRows;i++) numbersPerRow[i]=pl[i];
        } else {
            for (i=0;i<numberOfRows;i++)
                numbersPerRow[i]=numberOfColumns;
        }
    }

    if((ret=grib_get_long_internal(gh,self->widthOfFirstOrderValues,&widthOfFirstOrderValues)) != GRIB_SUCCESS)
        return ret;

    if((ret=grib_get_long_internal(gh,self->binary_scale_factor,&binary_scale_factor)) != GRIB_SUCCESS)
        return ret;

    if((ret=grib_get_long_internal(gh,self->decimal_scale_factor,&decimal_scale_factor)) != GRIB_SUCCESS)
        return ret;

    if((ret=grib_get_double_internal(gh,self->reference_value,&reference_value)) != GRIB_SUCCESS)
        return ret;

    if((ret=grib_get_long_internal(gh,self->numberOfSecondOrderPackedValues,
            &numberOfSecondOrderPackedValues)) != GRIB_SUCCESS)
        return ret;

    groupWidths=(long*)grib_context_malloc_clear(a->context,sizeof(long)*numberOfGroups);
    groupWidthsSize=numberOfGroups;
    if((ret=grib_get_long_array_internal(gh,self->groupWidths, groupWidths,&groupWidthsSize)) != GRIB_SUCCESS)
        return ret;

    firstOrderValues=(long*)grib_context_malloc_clear(a->context,sizeof(long)*numberOfGroups);
    grib_decode_long_array(buf,&pos,widthOfFirstOrderValues,numberOfGroups,firstOrderValues);
    pos = 8 * ( (pos + 7 ) / 8);

    n=0;
    for (i=0; i<numberOfGroups; i++) n+=numbersPerRow[i];

    X=(long*)grib_context_malloc_clear(a->context,sizeof(long)*n);
    n=0;
    k=0;
    for (i=0; i<numberOfGroups; i++) {
        if (groupWidths[i]>0) {
            for (j=0;j<numbersPerRow[k];j++) {
                X[n]=grib_decode_unsigned_long(buf,&pos,groupWidths[i]);
                X[n]+=firstOrderValues[i];
                n++;
            }
        } else {
            for (j=0;j<numbersPerRow[k];j++) {
                X[n]=firstOrderValues[i];
                n++;
            }
        }
        k++;
    }

    s = grib_power(binary_scale_factor,2);
    d = grib_power(-decimal_scale_factor,10) ;
    for (i=0; i<n; i++) {
        values[i] = (double) (((X[i]*s)+reference_value)*d);
    }
    grib_context_free(a->context,firstOrderValues);
    grib_context_free(a->context,X);
    grib_context_free(a->context,groupWidths);
    if (plSize) grib_context_free(a->context,pl);

    return ret;
}

static int pack_double(grib_accessor* a, const double* cval, size_t *len)
{
    grib_handle* gh = grib_handle_of_accessor(a);
    char type[]="grid_second_order";
    size_t size=strlen(type);

    grib_set_string(gh,"packingType",type,&size);

    return grib_set_double_array(gh,"values",cval,*len);
}
