X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/9960ea202c197830987b497fd1d2619c887ea9eb..6fb292d7e99020dc305b845e0a2284a9d811a4b2:/src/instr/instr_paje_types.hpp?ds=sidebyside

diff --git a/src/instr/instr_paje_types.hpp b/src/instr/instr_paje_types.hpp
index b07bc6575a..b1f8befcd2 100644
--- a/src/instr/instr_paje_types.hpp
+++ b/src/instr/instr_paje_types.hpp
@@ -1,4 +1,4 @@
-/* Copyright (c) 2010-2017. The SimGrid Team. All rights reserved.          */
+/* Copyright (c) 2010-2020. The SimGrid Team. All rights reserved.          */
 
 /* This program is free software; you can redistribute it and/or modify it
  * under the terms of the license (GNU LGPL) which comes with this package. */
@@ -7,6 +7,7 @@
 #define INSTR_PAJE_TYPES_HPP
 
 #include "src/instr/instr_private.hpp"
+#include <memory>
 #include <string>
 #include <vector>
 
@@ -20,82 +21,104 @@ class Type {
   std::string name_;
   std::string color_;
   Type* father_;
-
-public:
-  std::map<std::string, Type*> children_;
+  std::map<std::string, std::unique_ptr<Type>> children_;
   Container* issuer_ = nullptr;
 
-  Type(std::string name, std::string alias, std::string color, Type* father);
-  virtual ~Type();
-
-  std::string getName() { return name_; }
-  const char* getCname() { return name_.c_str(); }
-  long long int getId() { return id_; }
-  bool isColored() { return not color_.empty(); }
+protected:
+  Container* get_issuer() const { return issuer_; }
 
-  Type* byName(std::string name);
-  ContainerType* getOrCreateContainerType(std::string name);
-  EventType* getOrCreateEventType(std::string name);
-  LinkType* getOrCreateLinkType(std::string name, Type* source, Type* dest);
-  StateType* getOrCreateStateType(std::string name);
-  VariableType* getOrCreateVariableType(std::string name, std::string color);
-
-  void setCallingContainer(Container* container) { issuer_ = container; }
-
-  void logDefinition(e_event_type event_type);
-  void logDefinition(Type* source, Type* dest);
+public:
+  static xbt::signal<void(Type&, e_event_type event_type)> on_creation;
+
+  Type(e_event_type event_type, const std::string& name, const std::string& alias, const std::string& color,
+       Type* father);
+  virtual ~Type() = default;
+
+  long long int get_id() const { return id_; }
+  const std::string& get_name() const { return name_; }
+  const char* get_cname() const { return name_.c_str(); }
+  const std::string& get_color() const { return color_; }
+  Type* get_father() const { return father_; }
+  const std::map<std::string, std::unique_ptr<Type>>& get_children() const { return children_; }
+  bool is_colored() const { return not color_.empty(); }
+
+  Type* by_name(const std::string& name);
+  LinkType* by_name_or_create(const std::string& name, Type* source, Type* dest);
+  VariableType* by_name_or_create(const std::string& name, const std::string& color);
+
+  template <class T> T* by_name_or_create(const std::string& name)
+  {
+    auto cont = children_.find(name);
+    return cont == children_.end() ? new T(name, this) : static_cast<T*>(cont->second.get());
+  }
+
+  Type* set_calling_container(Container* container)
+  {
+    issuer_ = container;
+    return this;
+  }
 };
 
 class ContainerType : public Type {
 public:
-  explicit ContainerType(std::string name) : Type(name, name, "", nullptr){};
-  ContainerType(std::string name, Type* father);
+  explicit ContainerType(const std::string& name) : Type(PAJE_DefineContainerType, name, name, "", nullptr){};
+  ContainerType(const std::string& name, Type* father) : Type(PAJE_DefineContainerType, name, name, "", father){};
 };
 
 class VariableType : public Type {
   std::vector<VariableEvent*> events_;
 public:
-  VariableType(std::string name, std::string color, Type* father);
-  ~VariableType();
-  void setEvent(double timestamp, double value);
-  void addEvent(double timestamp, double value);
-  void subEvent(double timestamp, double value);
+  VariableType(const std::string& name, const std::string& color, Type* father)
+      : Type(PAJE_DefineVariableType, name, name, color, father)
+  {
+  }
+  void instr_event(double now, double delta, const char* resource, double value);
+  void set_event(double timestamp, double value);
+  void add_event(double timestamp, double value);
+  void sub_event(double timestamp, double value);
 };
 
 class ValueType : public Type {
 public:
-  std::map<std::string, EntityValue*> values_;
-  ValueType(std::string name, std::string alias, Type* father) : Type(name, alias, "", father){};
-  ValueType(std::string name, Type* father) : Type(name, name, "", father){};
-  virtual ~ValueType();
-  void addEntityValue(std::string name, std::string color);
-  void addEntityValue(std::string name);
-  EntityValue* getEntityValue(std::string name);
+  std::map<std::string, EntityValue> values_;
+  ValueType(e_event_type event_type, const std::string& name, const std::string& alias, Type* father)
+      : Type(event_type, name, alias, "", father){};
+  ValueType(e_event_type event_type, const std::string& name, Type* father)
+      : Type(event_type, name, name, "", father){};
+  ~ValueType() override = default;
+  void add_entity_value(const std::string& name, const std::string& color);
+  void add_entity_value(const std::string& name);
+  EntityValue* get_entity_value(const std::string& name);
 };
 
 class LinkType : public ValueType {
 public:
-  LinkType(std::string name, std::string alias, Type* father);
-  void startEvent(Container* startContainer, std::string value, std::string key);
-  void startEvent(Container* startContainer, std::string value, std::string key, int size);
-  void endEvent(Container* endContainer, std::string value, std::string key);
+  static xbt::signal<void(LinkType&, Type&, Type&)> on_creation;
+  LinkType(const std::string& name, Type* source, Type* dest, const std::string& alias, Type* father)
+      : ValueType(PAJE_DefineLinkType, name, alias, father)
+  {
+    on_creation(*this, *source, *dest);
+  }
+  void start_event(Container* startContainer, const std::string& value, const std::string& key);
+  void start_event(Container* startContainer, const std::string& value, const std::string& key, int size);
+  void end_event(Container* endContainer, const std::string& value, const std::string& key);
 };
 
 class EventType : public ValueType {
 public:
-  EventType(std::string name, Type* father);
+  EventType(const std::string& name, Type* father) : ValueType(PAJE_DefineEventType, name, father) {}
 };
 
 class StateType : public ValueType {
   std::vector<StateEvent*> events_;
 public:
-  StateType(std::string name, Type* father);
-  ~StateType();
-  void setEvent(std::string value_name);
-  void pushEvent(std::string value_name);
-  void pushEvent(std::string value_name, TIData* extra);
-  void popEvent();
+  StateType(const std::string& name, Type* father) : ValueType(PAJE_DefineStateType, name, father) {}
+  void set_event(const std::string& value_name);
+  void push_event(const std::string& value_name);
+  void push_event(const std::string& value_name, TIData* extra);
+  void pop_event();
+  void pop_event(TIData* extra);
 };
-}
-}
+} // namespace instr
+} // namespace simgrid
 #endif