from .base import SymmetricSpectrum, GraphData, x

data = {
    3: GraphData(
        vertex_count=6,
        edge_count=6,
        distance_classes=[1, 2, 2, 1],
        spectrum=SymmetricSpectrum([(1, 2), (2, 1)]),
    ),
    4: GraphData(
        vertex_count=24,
        edge_count=36,
        distance_classes=[1, 3, 5, 6, 5, 3, 1],
        spectrum=SymmetricSpectrum(
            [
                (1, 3),
                (3, 1),
                (x**2 - 3, 2),
                (x**2 - 2 * x - 1, 3),
                (x**2 + 2 * x - 1, 3),
            ]
        ),
    ),
    5: GraphData(
        vertex_count=120,
        edge_count=240,
        distance_classes=[1, 4, 9, 15, 20, 22, 20, 15, 9, 4, 1],
        spectrum=SymmetricSpectrum(
            [
                (0, 12),
                (1, 6),
                (4, 1),
                (x**2 - 5, 6),
                (x**2 + 5 * x + 5, 4),
                (x**2 + 3 * x + 1, 4),
                (x**2 + 2 * x - 1, 5),
                (x**3 + 2 * x**2 - 5 * x - 4, 5),
            ]
        ),
    ),
    6: GraphData(
        vertex_count=720,
        edge_count=1800,
        distance_classes=[1, 5, 14, 29, 49, 71, 90, 101, 101, 90, 71, 49, 29, 14, 5, 1],
        spectrum=SymmetricSpectrum(
            [
                (0, 20),
                (1, 25),
                (2, 15),
                (3, 5),
                (4, 5),
                (5, 1),
                (x**2 - 3, 20),
            ],
            not_shown_count=558,
        ),
    ),
    7: GraphData(
        vertex_count=5040,
        edge_count=15120,
        distance_classes=[
            1,
            6,
            20,
            49,
            98,
            169,
            259,
            359,
            455,
            531,
            573,
            573,
            531,
            455,
            359,
            259,
            169,
            98,
            49,
            20,
            6,
            1,
        ],
        spectrum=SymmetricSpectrum(
            [
                (0, 35),
                (1, 20),
                (2, 45),
                (6, 1),
            ],
            not_shown_count=4873,
        ),
    ),
    8: GraphData(
        vertex_count=40320, spectrum=SymmetricSpectrum([], not_shown_count=40320)
    ),
}

# TODO (maybe): verify these quantities with outputs from Haskell