WIP

src/sequence_diagram/visitor/translation_unit_visitor.cc

@@ -71,6 +71,17 @@ bool translation_unit_visitor::VisitFunctionDecl(
     return true;
 }
 
+bool translation_unit_visitor::VisitFunctionTemplateDecl(
+    clang::FunctionTemplateDecl *function_declaration)
+{
+    if (!function_declaration->isCXXClassMember())
+        current_class_decl_ = nullptr;
+
+    current_function_template_decl_ = function_declaration;
+
+    return true;
+}
+
 bool translation_unit_visitor::VisitCallExpr(clang::CallExpr *expr)
 {
     using clanguml::common::model::message_t;
@@ -109,11 +120,20 @@ bool translation_unit_visitor::VisitCallExpr(clang::CallExpr *expr)
         m.from = current_class_decl_->getQualifiedNameAsString();
         m.from_usr = current_method_decl_->getID();
     }
-    else {
+    else if (current_function_decl_ != nullptr) {
         // Handle call expression within free function
         m.from = current_function_decl_->getQualifiedNameAsString() + "()";
         m.from_usr = current_function_decl_->getID();
     }
+    else {
+        m.from = current_function_template_decl_->getQualifiedNameAsString();
+        std::vector<std::string> params;
+        for (const auto &template_parameter :
+            *current_function_template_decl_->getTemplateParameters()) {
+            params.push_back(template_parameter->getNameAsString());
+        }
+        m.from += fmt::format("<{}>", fmt::join(params, ","));
+    }
 
     const auto &current_ast_context = current_class_decl_
         ? current_class_decl_->getASTContext()
@@ -153,6 +173,9 @@ bool translation_unit_visitor::VisitCallExpr(clang::CallExpr *expr)
         if (!callee_decl)
             return true;
 
+        if (callee_decl->isTemplateDecl())
+            LOG_DBG("Call to template function!!!!");
+
         const auto *callee_function = callee_decl->getAsFunction();
 
         if (!callee_function)

src/sequence_diagram/visitor/translation_unit_visitor.h

@@ -41,6 +41,9 @@ public:
 
     virtual bool VisitFunctionDecl(clang::FunctionDecl *function_declaration);
 
+    bool VisitFunctionTemplateDecl(
+        clang::FunctionTemplateDecl *function_declaration);
+
     clanguml::sequence_diagram::model::diagram &diagram();
 
     const clanguml::config::sequence_diagram &config() const;
@@ -59,6 +62,8 @@ private:
     clang::CXXRecordDecl *current_class_decl_;
     clang::CXXMethodDecl *current_method_decl_;
     clang::FunctionDecl *current_function_decl_;
+    clang::FunctionTemplateDecl *current_function_template_decl_;
+
 };
 
 }

tests/t20003/.clang-uml

@@ -0,0 +1,14 @@
+compilation_database_dir: ..
+output_directory: puml
+diagrams:
+  t20003_sequence:
+    type: sequence
+    glob:
+      - ../../tests/t20003/t20003.cc
+    include:
+      namespaces:
+        - clanguml::t20003
+    using_namespace:
+      - clanguml::t20003
+    start_from:
+      - function: "clanguml::t20003::m1<T>()"

tests/t20003/t20003.cc

@@ -0,0 +1,16 @@
+#include <algorithm>
+#include <numeric>
+#include <vector>
+
+namespace clanguml {
+namespace t20003 {
+
+template <typename T> void m4(T p) { }
+
+template <typename T> void m3(T p) { m4<T>(); }
+
+template <typename T> void m2(T p) { m3<T>(p); }
+
+template <typename T> void m1(T p) { m2<T>(p); }
+}
+}

tests/t20003/test_case.h

@@ -0,0 +1,42 @@
+/**
+ * tests/t20003/test_case.cc
+ *
+ * Copyright (c) 2021-2022 Bartek Kryza <bkryza@gmail.com>
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+TEST_CASE("t20003", "[test-case][sequence]")
+{
+    auto [config, db] = load_config("t20003");
+
+    auto diagram = config.diagrams["t20003_sequence"];
+
+    REQUIRE(diagram->name == "t20003_sequence");
+
+    auto model = generate_sequence_diagram(*db, diagram);
+
+    REQUIRE(model->name() == "t20003_sequence");
+
+    auto puml = generate_sequence_puml(diagram, *model);
+
+    REQUIRE_THAT(puml, StartsWith("@startuml"));
+    REQUIRE_THAT(puml, EndsWith("@enduml\n"));
+
+    REQUIRE_THAT(puml, HasFunctionCall("m1<T>", "m2<T>"));
+    REQUIRE_THAT(puml, HasFunctionCall("m2<T>", "m3<T>"));
+    REQUIRE_THAT(puml, HasFunctionCall("m3<T>", "m4<T>"));
+
+    save_puml(
+        "./" + config.output_directory() + "/" + diagram->name + ".puml", puml);
+}

tests/test_cases.cc

@@ -249,6 +249,7 @@ using namespace clanguml::test::matchers;
 ///
 #include "t20001/test_case.h"
 #include "t20002/test_case.h"
+#include "t20003/test_case.h"
 
 ///
 /// Package diagram tests