Fix memory leaks.

[simgrid.git] / src / instr / instr_paje_containers.hpp

diff --git a/src/instr/instr_paje_containers.hpp b/src/instr/instr_paje_containers.hpp
index 688abeb..50501ff 100644 (file)
--- a/src/instr/instr_paje_containers.hpp
+++ b/src/instr/instr_paje_containers.hpp
@@ -7,10 +7,14 @@
 #define INSTR_PAJE_CONTAINERS_HPP
 
 #include "src/instr/instr_private.hpp"
+#include <string>
 
 namespace simgrid {
 namespace instr {
 class Type;
+class LinkType;
+class StateType;
+class VariableType;
 
 class Container {
   long long int id_;
@@ -33,6 +37,10 @@ public:
   void logCreation();
   void logDestruction();
 
+  StateType* getState(std::string name);
+  LinkType* getLink(std::string name);
+  VariableType* getVariable(std::string name);
+
   static Container* getRootContainer();
 };