from cpp_boggle import (
    Boggler33,
    Boggler34,
    Boggler44,
    Boggler55,
    BucketBoggler33,
    BucketBoggler34,
    BucketBoggler44,
    OrderlyTreeBuilder22,
    OrderlyTreeBuilder33,
    OrderlyTreeBuilder34,
    OrderlyTreeBuilder44,
)

from boggle.ibuckets import PyBucketBoggler22, PyBucketBoggler23

Bogglers = {(3, 3): Boggler33, (3, 4): Boggler34, (4, 4): Boggler44, (5, 5): Boggler55}

BucketBogglers = {
    (2, 2): PyBucketBoggler22,
    (2, 3): PyBucketBoggler23,
    (3, 3): BucketBoggler33,
    (3, 4): BucketBoggler34,
    (4, 4): BucketBoggler44,
}

OrderlyTreeBuilders = {
    (2, 2): OrderlyTreeBuilder22,
    (3, 3): OrderlyTreeBuilder33,
    (3, 4): OrderlyTreeBuilder34,
    (4, 4): OrderlyTreeBuilder44,
}


# Matches PyBoggler constructor
def cpp_boggler(t, dims):
    return Bogglers[dims](t)


def cpp_bucket_boggler(t, dims):
    return BucketBogglers[dims](t)


def cpp_orderly_tree_builder(t, dims):
    return OrderlyTreeBuilders[dims](t)


LEN_TO_DIMS = {
    4: (2, 2),
    6: (2, 3),
    9: (3, 3),
    12: (3, 4),
    16: (4, 4),
    25: (5, 5),
}