#ifndef CUBE_HASH_2WAY_H__
#define CUBE_HASH_2WAY_H__ 1

#include <stdint.h>
#include "simd-utils.h"

#if defined(__AVX2__)

#if defined(__AVX512F__) && defined(__AVX512VL__) && defined(__AVX512DQ__) && defined(__AVX512BW__)

struct _cube_4way_context
{
    __m512i h[8];
    int hashlen;
    int rounds;
    int blocksize;
    int pos; 
} __attribute__ ((aligned (128)));

typedef struct _cube_4way_context cube_4way_context;

int cube_4way_init( cube_4way_context* sp, int hashbitlen, int rounds,
                       int blockbytes );
int cube_4way_update( cube_4way_context *sp, const void *data, size_t size );
int cube_4way_close( cube_4way_context *sp, void *output );
int cube_4way_update_close( cube_4way_context *sp, void *output,
                            const void *data, size_t size );
int cube_4way_full( cube_4way_context *sp, void *output, int hashbitlen,
                    const void *data, size_t size );

int cube_4x256_full( cube_4way_context *sp, void *output, int hashbitlen,
                     const void *data, size_t size );

#define cube512_4way_init( sp ) cube_4way_update( sp, 512 )
#define cube512_4way_update cube_4way_update
#define cube512_4way_update_close cube_4way_update
#define cube512_4way_close cube_4way_update
#define cube512_4way_full( sp, output, data, size ) \
           cube_4way_full( sp, output, 512, data, size )
#define cube512_4x256_full( sp, output, data, size ) \
           cube_4x256_full( sp, output, 512, data, size )

#define cube256_4way_init( sp ) cube_4way_update( sp, 256 )
#define cube256_4way_update cube_4way_update
#define cube256_4way_update_close cube_4way_update
#define cube256_4way_close cube_4way_update
#define cube256_4way_full( sp, output, data, size ) \
           cube_4way_full( sp, output, 256, data, size )
#define cube256_4x256_full( sp, output, data, size ) \
           cube_4x256_full( sp, output, 256, data, size )

#endif

// 2x128, 2 way parallel SSE2

struct _cube_2way_context
{
    __m256i h[8];
    int hashlen;           // __m128i
    int rounds;
    int blocksize;         // __m128i
    int pos;               // number of __m128i read into x from current block
} __attribute__ ((aligned (128)));

typedef struct _cube_2way_context cube_2way_context;

int cube_2way_init( cube_2way_context* sp, int hashbitlen, int rounds,
                       int blockbytes );
int cube_2way_update( cube_2way_context *sp, const void *data, size_t size );
int cube_2way_close( cube_2way_context *sp, void *output );
int cube_2way_update_close( cube_2way_context *sp, void *output,
                            const void *data, size_t size );
int cube_2way_full( cube_2way_context *sp, void *output, int hashbitlen,
                    const void *data, size_t size );


#endif
#endif