#ifndef TEST_UTIL_H
#define TEST_UTIL_H

#include <stdint.h>
#include "ntru.h"

void rand_poly(NtruIntPoly *a, uint16_t N, uint16_t modulus);

uint8_t equals_one(NtruIntPoly *a);

uint8_t equals_poly(NtruIntPoly *a, NtruIntPoly *b);

uint8_t equals_poly_mod(NtruIntPoly *a, NtruIntPoly *b, uint16_t modulus);

uint8_t equals_key_pair(NtruEncKeyPair *kp1, NtruEncKeyPair *kp2);

uint8_t equals_arr(uint8_t *arr1, uint8_t *arr2, uint16_t len);

uint8_t equals_params(NtruEncParams *params1, NtruEncParams *params2);

/* Generates a random NtruIntPoly with coefficients between 0 and 2^pow2q-1 */
uint8_t rand_poly_pow2(uint16_t N, uint16_t pow2q, NtruIntPoly *poly, NtruRandContext *rand_ctx);

/**
 * @brief Ternary to general integer polynomial
 *
 * Converts a NtruTernPoly to an equivalent NtruIntPoly.
 * 
 * @param a a ternary polynomial
 * @param b output parameter; a pointer to store the new polynomial
 */
void ntru_tern_to_int(NtruTernPoly *a, NtruIntPoly *b);

#ifndef NTRU_AVOID_HAMMING_WT_PATENT
/**
 * @brief Product-form to general polynomial
 *
 * Converts a NtruProdPoly to an equivalent NtruIntPoly.
 *
 * @param a a product-form polynomial
 * @param b output parameter; a pointer to store the new polynomial
 * @param modulus the modulus; must be a power of two
 */
void ntru_prod_to_int(NtruProdPoly *a, NtruIntPoly *b, uint16_t modulus);
#endif   /* NTRU_AVOID_HAMMING_WT_PATENT */

/**
 * @brief Private polynomial to general polynomial
 *
 * Converts a NtruPrivPoly (i.e. a NtruTernPoly or NtruProdPoly) to an
 * equivalent NtruIntPoly.
 *
 * @param a a "private" polynomial
 * @param b output parameter; a pointer to store the new polynomial
 * @param modulus the modulus; must be a power of two
 */
void ntru_priv_to_int(NtruPrivPoly *a, NtruIntPoly *b, uint16_t modulus);

/**
 * @brief string to uint8_t array
 *
 * Converts a char array to a uint8_t array. If char is longer than uint8_t,
 * only the least significant 8 bits of each element are copied.
 *
 * @param in the NtruEncrypt parameters to use
 * @param out pointer to write the key pair to (output parameter)
 */
void str_to_uint8(char *in, uint8_t *out);

void print_result(char *test_name, uint8_t valid);

#endif