diff --git a/.clang-uml b/.clang-uml
index 53fed4a6..ec0124c5 100644
--- a/.clang-uml
+++ b/.clang-uml
@@ -15,6 +15,8 @@ generate_links:
   tooltip: "{% if existsIn(element, \"comment\") and existsIn(element.comment, \"brief\") %}{{ abbrv(trim(replace(element.comment.brief.0, \"\\n+\", \" \")), 256) }}{% else %}{{ element.name }}{% endif %}"
 diagrams:
   # Class diagrams
+  class_translation_unit_visitor:
+    include!: uml/class/class_translation_unit_visitor.yml
   architecture_visitors_class:
     include!: uml/class/architecture_visitors_class.yml
   config_class:
diff --git a/uml/class/class_translation_unit_visitor.yml b/uml/class/class_translation_unit_visitor.yml
new file mode 100644
index 00000000..2180a56b
--- /dev/null
+++ b/uml/class/class_translation_unit_visitor.yml
@@ -0,0 +1,26 @@
+type: class
+title: Class diagram TU visitor
+include_relations_also_as_members: false
+generate_method_arguments: none
+generate_packages: false
+glob:
+  - src/common/visitor/*.cc
+  - src/class_diagram/visitor/*.cc
+include:
+  namespaces:
+    - clanguml
+    - clang
+  context:
+    - clanguml::class_diagram::visitor::translation_unit_visitor
+    - match:
+        radius: 2
+        pattern: "clang::RecursiveASTVisitor<clanguml::class_diagram::visitor::translation_unit_visitor>"
+exclude:
+  access:
+    - public
+    - protected
+    - private
+  relationships:
+    - dependency
+  element_types:
+    - enum
\ No newline at end of file