Add missing include.

[loba.git] / messages.cpp

diff --git a/messages.cpp b/messages.cpp
index fd19f6b979657881f082d09e7bc0ea8a480a3ac2..2d5de949b0ddd72a05d2661ae81b0234018bdde7 100644 (file)
--- a/messages.cpp
+++ b/messages.cpp
@@ -28,10 +28,26 @@ message::message(message_type t, double a, double c)
 
 std::string message::to_string()
 {
-    static const char* str[DATA_CLOSE + 1] = { "CTRL", "DATA",
-                                               "CTRL_CLOSE", "DATA_CLOSE" };
     std::ostringstream oss;
-    oss << str[type] << ": " << amount;
+    switch (type) {
+    case CTRL:
+        oss << "CTRL: " << amount << " (info)";
+        if (opt::bookkeeping)
+            oss << "; " << credit << " (credit)";
+        break;
+    case DATA:
+        oss << "DATA: " << amount << " (load)";
+        break;
+    case CTRL_CLOSE:
+        oss << "CTRL_CLOSE";
+        break;
+    case DATA_CLOSE:
+        oss << "DATA_CLOSE";
+        break;
+    default:
+        oss << "UNKNOWN MESSAGE TYPE: " << type;
+        break;
+    }
     return oss.str();
 }