#if !defined(ECHO_HASH_4WAY_H__)
#define ECHO_HASH_4WAY_H__ 1

#ifdef __cplusplus
extern "C" {
#endif

#if defined(__VAES__)

#include "simd-utils.h"

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

typedef struct {
  __m512i state[4][4];
  __m512i buffer[4 * 192 / 16]; // 4x128 interleaved 192 bytes
  __m512i k;
  __m512i hashsize;
  __m512i const1536;

  unsigned int uRounds;
  unsigned int uHashSize;
  unsigned int uBlockLength;
  unsigned int uBufferBytes;
  unsigned int processed_bits;

} echo_4way_context __attribute__((aligned(64)));
#define echo512_4way_context echo_4way_context

int echo_4way_init(echo_4way_context *state, int hashbitlen);
#define echo512_4way_init(state) echo_4way_init(state, 512)
#define echo256_4way_init(state) echo_4way_init(state, 256)

int echo_4way_update(echo_4way_context *state, const void *data,
                     unsigned int databitlen);
#define echo512_4way_update echo_4way_update

// int echo_4way_close( echo_4way_context *state, void *hashval );
// #define echo512_4way_close echo_4way_close

int echo_4way_update_close(echo_4way_context *state, void *hashval,
                           const void *data, int databitlen);
#define echo512_4way_update_close echo_4way_update_close

int echo_4way_full(echo_4way_context *ctx, void *hashval, int nHashSize,
                   const void *data, int datalen);
#define echo512_4way_full(state, hashval, data, datalen)                       \
  echo_4way_full(state, hashval, 512, data, datalen)
#define echo256_4way_full(state, hashval, data, datalen)                       \
  echo_4way_full(state, hashval, 256, data, datalen)

#endif // AVX512

typedef struct {
  __m256i state[4][4];
  __m256i buffer[4 * 192 / 16]; // 4x128 interleaved 192 bytes
  __m256i k;
  __m256i hashsize;
  __m256i const1536;

  unsigned int uRounds;
  unsigned int uHashSize;
  unsigned int uBlockLength;
  unsigned int uBufferBytes;
  unsigned int processed_bits;

} echo_2way_context __attribute__((aligned(64)));
#define echo512_2way_context echo_2way_context

int echo_2way_init(echo_2way_context *state, int hashbitlen);
#define echo512_2way_init(state) echo_2way_init(state, 512)
#define echo256_2way_init(state) echo_2way_init(state, 256)

int echo_2way_update(echo_2way_context *state, const void *data,
                     unsigned int databitlen);
#define echo512_2way_update echo_2way_update

int echo_2way_update_close(echo_2way_context *state, void *hashval,
                           const void *data, int databitlen);
#define echo512_2way_update_close echo_2way_update_close

int echo_2way_full(echo_2way_context *ctx, void *hashval, int nHashSize,
                   const void *data, int datalen);
#define echo512_2way_full(state, hashval, data, datalen)                       \
  echo_2way_full(state, hashval, 512, data, datalen)
#define echo256_2way_full(state, hashval, data, datalen)                       \
  echo_2way_full(state, hashval, 256, data, datalen)

#endif // VAES

#ifdef __cplusplus
}
#endif

#endif // ECHO_HASH_4WAY_H__