bears.c_languages.codeclone_detection.ClangASTPrintBear

Complexity

Total Complexity

7

Size/Duplication

Total Lines	53
Duplicated Lines	0 %

2 Methods

Rating	Name	Duplication	Size	Complexity
A	run()	0	12	1
B	print_node()	0	39	6

from coalib.bears.LocalBear import LocalBear
from coalib.bearlib.parsing.clang.cindex import Index, TranslationUnit


class ClangASTPrintBear(LocalBear):
    def print_node(self, cursor, filename, before="", spec_before=""):
        '''
        Prints this node and all child nodes recursively in the style of:

        -node
        |-child
        `-another child
         |-child of child
         `-last child of child

        :param cursor:      The node to print. (Clang cursor.)
        :param before:      What to print before the node.
        :param spec_before: What to print before this node but to replace with
                            spaces for child nodes.
        '''
        file = cursor.location.file

        if file is not None and file.name.decode() == filename:
            self.debug(
                before + spec_before + "-" + str(cursor.displayname.decode()),
                str(cursor.kind),
                "Lines",
                str(cursor.extent.start.line) + "-" +
                str(cursor.extent.end.line),
                "(" + " ".join(str(token.spelling.decode())
                               for token in cursor.get_tokens()) + ")")

        children = list(cursor.get_children())

        if len(children) > 0:
            for child in children[:-1]:
                self.print_node(child,
                                filename,
                                before + len(spec_before)*" " + "|")

            self.print_node(children[-1],
                            filename,
                            before + len(spec_before)*" ",
                            "`")

    def run(self,
            filename,
            file):
        """
        This bear is meant for debugging purposes relating to clang. It just
        prints out the whole AST for a file to the DEBUG channel.
        """
        root = Index.create().parse(
            filename,
            options=TranslationUnit.PARSE_DETAILED_PROCESSING_RECORD).cursor

        self.print_node(root, filename)