Added diagram type enum

src/class_diagram/model/diagram.cc

@@ -36,6 +36,11 @@ const std::vector<type_safe::object_ref<const enum_>> diagram::enums() const
     return enums_;
 }
 
+common::model::diagram_t diagram::type() const
+{
+    return common::model::diagram_t::kClass;
+}
+
 bool diagram::has_class(const class_ &c) const
 {
     return std::any_of(classes_.cbegin(), classes_.cend(),

src/class_diagram/model/diagram.h

@@ -41,6 +41,8 @@ public:
     diagram &operator=(const diagram &) = delete;
     diagram &operator=(diagram &&) = default;
 
+    common::model::diagram_t type() const override;
+
     const std::vector<type_safe::object_ref<const class_>> classes() const;
 
     const std::vector<type_safe::object_ref<const enum_>> enums() const;

src/common/model/diagram.h

@@ -34,6 +34,8 @@ public:
     diagram();
     virtual ~diagram();
 
+    virtual diagram_t type() const = 0;
+
     diagram(const diagram &) = delete;
     diagram(diagram &&);
     diagram &operator=(const diagram &) = delete;

src/common/model/enums.h

@@ -21,6 +21,8 @@
 
 namespace clanguml::common::model {
 
+enum class diagram_t { kClass, kSequence, kPackage };
+
 enum class access_t { kPublic, kProtected, kPrivate };
 
 enum class relationship_t {

src/package_diagram/model/diagram.cc

@@ -23,6 +23,11 @@
 
 namespace clanguml::package_diagram::model {
 
+common::model::diagram_t diagram::type() const
+{
+    return common::model::diagram_t::kPackage;
+}
+
 std::string diagram::to_alias(const std::string &full_name) const
 {
     LOG_DBG("Looking for alias for {}", full_name);

src/package_diagram/model/diagram.h

@@ -38,6 +38,8 @@ public:
     diagram &operator=(const diagram &) = delete;
     diagram &operator=(diagram &&) = default;
 
+    common::model::diagram_t type() const override;
+
     std::string to_alias(const std::string &full_name) const;
 };
 }

src/sequence_diagram/model/diagram.cc

@@ -18,17 +18,16 @@
 
 #include "diagram.h"
 
-#include <clang-c/CXCompilationDatabase.h>
-#include <clang-c/Index.h>
-#include <spdlog/spdlog.h>
-
 #include <functional>
 #include <memory>
-#include <string>
-#include <vector>
 
 namespace clanguml::sequence_diagram::model {
 
+common::model::diagram_t diagram::type() const
+{
+    return common::model::diagram_t::kSequence;
+}
+
 std::string diagram::to_alias(const std::string &full_name) const
 {
     return full_name;

src/sequence_diagram/model/diagram.h

@@ -34,6 +34,8 @@ public:
     diagram &operator=(const diagram &) = delete;
     diagram &operator=(diagram &&) = default;
 
+    common::model::diagram_t type() const override;
+
     std::string to_alias(const std::string &full_name) const;
 
     bool started{false};