Skip friends outside of included scope

2021-03-11 18:20:02 +01:00
parent 3bdac248ba
commit bf64b75bdd
2 changed files with 13 additions and 0 deletions
--- a/src/uml/class_diagram_visitor.h
+++ b/src/uml/class_diagram_visitor.h
@@ -215,8 +215,15 @@ static enum CXChildVisitResult friend_class_visitor(
    enum CXChildVisitResult ret = CXChildVisit_Break;
    switch (cursor.kind()) {
        case CXCursor_TypeRef: {
+            if (!ctx->ctx->config.should_include(
+                    cursor.referenced().fully_qualified())) {
+                ret = CXChildVisit_Continue;
+                break;
+            }
+
            spdlog::info("Adding friend declaration: {}, {}", cursor,
                cursor.referenced());
+
            class_relationship r;
            r.type = relationship_t::kFriendship;
            r.label = "<<friend>>";