import os
import fnmatch
import re

from .core.graph import Graph, GraphModification


def findFiles(folder, patterns):
    rules = [re.compile(fnmatch.translate(pattern), re.IGNORECASE) for pattern in patterns]
    outFiles = []
    for name in os.listdir(folder):
        for rule in rules:
            if rule.match(name):
                filepath = os.path.join(folder, name)
                outFiles.append(filepath)
                break
    return outFiles


def photogrammetryPipeline(output='', inputFolder='', inputImages=[], inputViewpoints=[]):
    # type: () -> Graph
    graph = Graph('pipeline')
    with GraphModification(graph):
        cameraInit = graph.addNewNode('CameraInit')
        if inputFolder:
            images = findFiles(inputFolder, ['*.jpg', '*.png'])
            cameraInit.viewpoints.extend([{'image': image} for image in images])
        if inputImages:
            cameraInit.viewpoints.extend([{'image': image} for image in inputImages])
        if inputViewpoints:
            cameraInit.viewpoints.extend(inputViewpoints)
        featureExtraction = graph.addNewNode('FeatureExtraction',
                                             input=cameraInit.outputSfm)
        imageMatching = graph.addNewNode('ImageMatching',
                                             input=featureExtraction.input,
                                             featuresDirectory=featureExtraction.output,
                                             )
        featureMatching = graph.addNewNode('FeatureMatching',
                                           input=imageMatching.input,
                                           featuresDirectory=imageMatching.featuresDirectory,
                                           imagePairsList=imageMatching.output)
        structureFromMotion = graph.addNewNode('StructureFromMotion',
                                               input=featureMatching.input,
                                               featuresDirectory=featureMatching.featuresDirectory,
                                               matchesDirectory=featureMatching.output)
        prepareDenseScene = graph.addNewNode('PrepareDenseScene',
                                             input=structureFromMotion.output)
        cameraConnection = graph.addNewNode('CameraConnection',
                                    ini=prepareDenseScene.ini)
        depthMap = graph.addNewNode('DepthMap',
                                    ini=cameraConnection.ini)
        depthMapFilter = graph.addNewNode('DepthMapFilter',
                                          ini=depthMap.ini)
        meshing = graph.addNewNode('Meshing',
                                   ini=depthMapFilter.ini)
        texturing = graph.addNewNode('Texturing',
                                     ini=meshing.ini)
        publish = graph.addNewNode('Publish',
                                   inputFiles=[texturing.outputMesh, texturing.outputMaterial, texturing.outputTextures],
                                   output=output)
    return graph