#ifndef __NIST_H__
#define __NIST_H__

/*define data alignment for different C compilers*/
#if defined(__GNUC__)
#define DATA_ALIGN(x) x __attribute__((aligned(16)))
#else
#define DATA_ALIGN(x) __declspec(align(16)) x
#endif

#include "compat.h"
#include "algo/sha/sha3-defs.h"
/*
 * NIST API Specific types.
 */

//typedef unsigned char BitSequence;

//#ifdef HAS_64
 // typedef u64 DataLength;
//#else
 // typedef unsigned long DataLength;
//#endif

// can't find u32 or fft-t
#include <stdint.h>
typedef uint32_t u32;
typedef int fft_t;


//typedef enum { SUCCESS = 0, FAIL = 1, BAD_HASHBITLEN = 2} HashReturn;

typedef struct {
  unsigned int hashbitlen;
  unsigned int blocksize;
  unsigned int n_feistels;

#ifdef HAS_64
  u64 count;
#else
  u32 count_low;
  u32 count_high;
#endif

  DATA_ALIGN(u32 A[32]);
  u32 *B;
  u32 *C;
  u32 *D;
  DATA_ALIGN(unsigned char buffer[128]);
  
} hashState_sd;

/* 
 * NIST API
 */

HashReturn init_sd(hashState_sd *state, int hashbitlen);

HashReturn update_sd(hashState_sd *state, const BitSequence *data, DataLength databitlen);

HashReturn final_sd(hashState_sd *state, BitSequence *hashval);

HashReturn update_final_sd( hashState_sd *state, BitSequence *hashval,
                            const BitSequence *data, DataLength databitlen );

//HashReturn Hash(int hashbitlen, const BitSequence *data, DataLength databitlen,
//                BitSequence *hashval);

/* 
 * Internal API
 */

int SupportedLength(int hashbitlen);
int RequiredAlignment(void);
void SIMD_Compress(hashState_sd * state, const unsigned char *M, int final);

void fft128_natural(fft_t *a, unsigned char *x);
void fft256_natural(fft_t *a, unsigned char *x);

#endif