from meshroom.core.graph import Graph
import pytest

import logging
logger = logging.getLogger('test')

valid3DExtensionFiles = [(f'test.{ext}', True) for ext in ('obj', 'stl', 'fbx', 'gltf', 'abc', 'ply')]
invalid3DExtensionFiles = [(f'test.{ext}', False) for ext in ('', 'exe', 'jpg', 'png', 'py')]

valid2DSemantics= [(semantic, True) for semantic in ('image', 'imageList', 'sequence')]
invalid2DSemantics = [(semantic, False) for semantic in ('3d', '', 'multiline', 'color/hue')]


def test_attribute_retrieve_linked_input_and_output_attributes():
    """
    Check that an attribute can retrieve the linked input and output attributes
    """

    # n0 -- n1 -- n2
    #   \          \
    #    ---------- n3

    g = Graph('')
    n0 = g.addNewNode('Ls', input='')
    n1 = g.addNewNode('Ls', input=n0.output)
    n2 = g.addNewNode('Ls', input=n1.output)
    n3 = g.addNewNode('AppendFiles', input=n1.output, input2=n2.output)

    # check that the attribute can retrieve its linked input attributes

    assert n0.output.hasOutputConnections
    assert not n3.output.hasOutputConnections

    assert len(n0.input.getLinkedInAttributes()) == 0
    assert len(n1.input.getLinkedInAttributes()) == 1
    assert n1.input.getLinkedInAttributes()[0] == n0.output
    
    assert len(n1.output.getLinkedOutAttributes()) == 2
    
    assert n1.output.getLinkedOutAttributes()[0] == n2.input
    assert n1.output.getLinkedOutAttributes()[1] == n3.input

    n0.graph = None

    # Bounding cases
    assert not n0.output.hasOutputConnections
    assert len(n0.input.getLinkedInAttributes()) == 0
    assert len(n0.output.getLinkedOutAttributes()) == 0

@pytest.mark.parametrize("givenFile,expected", valid3DExtensionFiles + invalid3DExtensionFiles)
def test_attribute_is3D_file_extensions(givenFile, expected):
    """
    Check what makes an attribute a valid 3d media
    """

    g = Graph('')
    n0 = g.addNewNode('Ls', input='')

    # Given
    assert not n0.input.is3D

    # When
    n0.input.value = givenFile

    # Then
    assert n0.input.is3D == expected


def test_attribute_i3D_by_description_semantic():
    """ """

    # Given
    g = Graph('')
    n0 = g.addNewNode('Ls', input='')  

    assert not n0.output.is3D

    # When
    n0.output.desc._semantic = "3d"

    # Then
    assert n0.output.is3D

@pytest.mark.parametrize("givenSemantic,expected", valid2DSemantics + invalid2DSemantics)
def test_attribute_is2D_file_semantic(givenSemantic, expected):
    """
    Check what makes an attribute a valid 2d media
    """

    g = Graph('')
    n0 = g.addNewNode('Ls', input='')

    # Given
    n0.input.desc._semantic = ""
    assert not n0.input.is2D

    # When
    n0.input.desc._semantic = givenSemantic

    # Then
    assert n0.input.is2D == expected