#!/bin/env python2

import re
from sys import argv
from os import walk


EXCLUDES = ["implementation", "testbench"]


def _vhdltree(level, filepath, pattern, vhd_files):
    included_entities = find_entities(filepath, pattern)
    if included_entities:
        for entity, name in included_entities.iteritems():
            path = vhd_files.get(name.lower())
            if path:
                print "    "*level + entity + " : " + path
                _vhdltree(level+1, path, pattern, vhd_files)
            else:
                print "    "*level + entity + " : Not Found"


def find_entities(filepath, pattern):
    included_entities = {}
    with open(filepath) as f:
        for l in f:
            m = pattern.match(l)
            if m:
                included_entities[m.group(1)] = m.group(2).split(".")[-1]
    return included_entities


def find_vhd(proot):
    vhd_files = {}
    for (dirpath, dirnames, filenames) in walk(proot):
        i, excluded = 0, False
        while i < len(EXCLUDES) and not excluded:
            excluded = EXCLUDES[i] in dirpath.lower()
            i += 1
        if not excluded:
            for n in filenames:
                l = n.split(".")
                ext = l[-1]
                if ext.lower() == "vhd":
                    basename = l[-2]
                    vhd_files[basename.lower()] = dirpath + '/' + n
    return vhd_files


def vhdltree(filepath, proot):
    p = re.compile("\s*([^\s:]+)\s*:\s*entity\s*([^\s]+)\s*", re.IGNORECASE)
    vhd_files = find_vhd(proot)
    _vhdltree(0, filepath, p, vhd_files)

if __name__ == "__main__":
    vhdltree(argv[1], argv[2])

#!/usr/bin/env python2

import re
from sys import argv
from os import walk


EXCLUDES = ["implementation", "testbench"]


def _vhdltree(level, filepath, pattern, vhd_files):
    included_entities = find_entities(filepath, pattern)
    if included_entities:
        for entity, name in included_entities.iteritems():
            path = vhd_files.get(name.lower())
            if path:
                print "    "*level + entity + " : " + path
                _vhdltree(level+1, path, pattern, vhd_files)
            else:
                print "    "*level + entity + " : Not Found"


def find_entities(filepath, pattern):
    included_entities = {}
    with open(filepath) as f:
        for l in f:
            m = pattern.match(l)
            if m:
                included_entities[m.group(1)] = m.group(2).split(".")[-1]
    return included_entities


def find_vhd(proot):
    vhd_files = {}
    for (dirpath, dirnames, filenames) in walk(proot):
        i, excluded = 0, False
        while i < len(EXCLUDES) and not excluded:
            excluded = EXCLUDES[i] in dirpath.lower()
            i += 1
        if not excluded:
            for n in filenames:
                l = n.split(".")
                ext = l[-1]
                if ext.lower() == "vhd":
                    basename = l[-2]
                    vhd_files[basename.lower()] = dirpath + '/' + n
    return vhd_files


def vhdltree(filepath, proot):
    p = re.compile("\s*([^\s:]+)\s*:\s*entity\s*([^\s]+)\s*", re.IGNORECASE)
    vhd_files = find_vhd(proot)
    _vhdltree(0, filepath, p, vhd_files)

if __name__ == "__main__":
    vhdltree(argv[1], argv[2])

Remove legacy env location