hexagon_spiral

Complexity

Total Complexity

15

Size/Duplication

Total Lines	62
Duplicated Lines	0 %

Importance

Changes

0

4 Functions

Rating	Name	Duplication	Size	Complexity
A	determine_level()	0	10	4
B	convert_to_position()	0	18	8
A	hex_spiral()	0	5	2
A	calc_distance()	0	5	1

__author__ = 'KenMercusLai'


def determine_level(number):
    if number == 1:
        return 0, 0
    level = 1
    start_number = 2
    while 1:
        if start_number <= number <= start_number + level * 6 - 1:
            return level, number - start_number
        start_number += level * 6
        level += 1


def convert_to_position(level, shift):
    if level == 0:
        return 0, 0
    position_map_of_numbers = []
    for i in range(level):
        position_map_of_numbers.append((-level + 1 + i, 1 + i))
    for i in range(1, level + 1):
        position_map_of_numbers.append((i, level))
    for i in range(level - 1, 0, -1):
        position_map_of_numbers.append((level, i))
    for i in range(level):
        position_map_of_numbers.append((level - i, 0 - i))
    for i in range(level):
        position_map_of_numbers.append((0 - i, -level))
    for i in range(level):
        position_map_of_numbers.append((-level, -level + i))
    position_map_of_numbers.append((-level, 0))
    return position_map_of_numbers[shift]


def calc_distance(num0, num1):
    return max(
        abs(num1[1] - num0[1]),
        abs(num1[0] - num0[0]),
        abs((num1[0] - num1[1]) * -1 - (num0[0] - num0[1]) * -1),
    )


def hex_spiral(first, second):
    return calc_distance(
        *map(
            lambda x: convert_to_position(x[0], x[1]),
            [determine_level(first), determine_level(second)],
        )
    )


# These "asserts" using only for self-checking and not necessary for
# auto-testing
if __name__ == '__main__':  # pragma: no cover
    assert hex_spiral(2, 9) == 1, "First"
    assert hex_spiral(9, 2) == 1, "Reverse First"
    assert hex_spiral(6, 19) == 2, "Second, short way"
    assert hex_spiral(5, 11) == 3, "Third"
    assert hex_spiral(13, 15) == 2, "Fourth, One row"
    assert hex_spiral(11, 17) == 4, "Fifth, One more test"