X-Git-Url: https://bilbo.iut-bm.univ-fcomte.fr/and/gitweb/loba.git/blobdiff_plain/72cb21403f86fbb162c8e4555f27234d337b6b6a..a561a2e124710e98e19bafea2b87f577779b68b5:/messages.h

diff --git a/messages.h b/messages.h
index 8de464a..1906fdf 100644
--- a/messages.h
+++ b/messages.h
@@ -9,19 +9,22 @@
 
 class message {
 public:
-    enum message_type { INFO, CREDIT, LOAD, CTRL_CLOSE, DATA_CLOSE };
+    enum message_type { INFO, LOAD, CTRL_CLOSE, DATA_CLOSE };
 
-    message(message_type t, double a): type(t), amount(a) { }
+    message(message_type t, double a, double c = 0.0);
 
     message_type get_type() const       { return type;   }
     double get_amount() const           { return amount; }
-    double get_size() const;
+    double get_credit() const           { return credit; }
+    double get_size() const             { return size;   }
 
     std::string to_string();
 
 private:
     message_type type;
     double amount;
+    double credit;
+    double size;
 };
 
 class message_queue {