#include "naut/bitfield.h"
#include "test.h"

int main(void) {
    naut_bitfield bf;
    /* deliberately not a multiple of 64 to exercise tail masking */
    CHECK(naut_bitfield_init(&bf, 1000) == NAUT_OK);
    CHECK_EQ(bf.nwords, 16);
    CHECK_EQ(naut_bitfield_count(&bf), 0);
    CHECK(!naut_bitfield_all_set(&bf));

    naut_bitfield_set(&bf, 0);
    naut_bitfield_set(&bf, 63);
    naut_bitfield_set(&bf, 64);
    naut_bitfield_set(&bf, 999);
    CHECK(naut_bitfield_test(&bf, 0));
    CHECK(naut_bitfield_test(&bf, 999));
    CHECK(!naut_bitfield_test(&bf, 1));
    CHECK_EQ(naut_bitfield_count(&bf), 4);

    CHECK_EQ(naut_bitfield_find_set(&bf, 0), 0);
    CHECK_EQ(naut_bitfield_find_set(&bf, 1), 63);
    CHECK_EQ(naut_bitfield_find_set(&bf, 65), 999);
    CHECK_EQ(naut_bitfield_find_set(&bf, 1000), SIZE_MAX);

    CHECK_EQ(naut_bitfield_find_zero(&bf, 0), 1);
    CHECK_EQ(naut_bitfield_find_zero(&bf, 63), 65);

    naut_bitfield_clear(&bf, 63);
    CHECK(!naut_bitfield_test(&bf, 63));
    CHECK_EQ(naut_bitfield_count(&bf), 3);

    /* all-set respects nbits, not nwords*64 */
    naut_bitfield_set_all(&bf);
    CHECK_EQ(naut_bitfield_count(&bf), 1000);
    CHECK(naut_bitfield_all_set(&bf));
    CHECK_EQ(naut_bitfield_find_zero(&bf, 0), SIZE_MAX);

    /* wire round-trip (BEP-3 MSB-first) */
    naut_bitfield_clear_all(&bf);
    naut_bitfield_set(&bf, 0);    /* -> byte 0, bit 0x80 */
    naut_bitfield_set(&bf, 7);    /* -> byte 0, bit 0x01 */
    naut_bitfield_set(&bf, 8);    /* -> byte 1, bit 0x80 */
    uint8_t wire[125];            /* ceil(1000/8) */
    naut_bitfield_to_wire(&bf, wire, sizeof(wire));
    CHECK_EQ(wire[0], 0x81);
    CHECK_EQ(wire[1], 0x80);

    naut_bitfield bf2;
    naut_bitfield_init(&bf2, 1000);
    naut_bitfield_from_wire(&bf2, wire, sizeof(wire));
    CHECK(naut_bitfield_test(&bf2, 0));
    CHECK(naut_bitfield_test(&bf2, 7));
    CHECK(naut_bitfield_test(&bf2, 8));
    CHECK_EQ(naut_bitfield_count(&bf2), 3);

    naut_bitfield_free(&bf);
    naut_bitfield_free(&bf2);
    TEST_MAIN_END();
}