Refactored sequence diagram model classes

src/common/model/enums.cc

@@ -74,6 +74,26 @@ std::string to_string(message_t r)
         return "call";
     case message_t::kReturn:
         return "return";
+    case message_t::kIf:
+        return "if";
+    case message_t::kElse:
+        return "else";
+    case message_t::kElseIf:
+        return "else if";
+    case message_t::kIfEnd:
+        return "end if";
+    case message_t::kWhile:
+        return "while";
+    case message_t::kWhileEnd:
+        return "end while";
+    case message_t::kDo:
+        return "do";
+    case message_t::kDoEnd:
+        return "end do";
+    case message_t::kFor:
+        return "for";
+    case message_t::kForEnd:
+        return "end for";
     default:
         assert(false);
         return "";

src/common/model/enums.h

@@ -51,14 +51,15 @@ enum class message_t {
     kDo,
     kDoEnd,
     kFor,
-    kForEnd
+    kForEnd,
+    kNone
 };
 
 std::string to_string(relationship_t r);
 
 std::string to_string(access_t r);
 
-std::string to_string(message_t r);
+std::string to_string(message_t m);
 
 std::string to_string(diagram_t r);
 

src/common/visitor/translation_unit_visitor.h

@@ -42,8 +42,6 @@ public:
 
     clang::SourceManager &source_manager() const;
 
-    void finalize();
-
 protected:
     void set_source_location(const clang::Decl &decl,
         clanguml::common::model::source_location &element);