David's final changes: more profiler features, fixes.

[ipdf/code.git] / src / profiler.h

diff --git a/src/profiler.h b/src/profiler.h
index f69f06a..ffffae9 100644 (file)
--- a/src/profiler.h
+++ b/src/profiler.h
@@ -1,6 +1,7 @@
 #ifndef _PROFILER_H
 #define _PROFILER_H
 
+#include <stdint.h>
 #include <map>
 #include <string>
 #include <stack>
@@ -20,6 +21,8 @@ namespace IPDF
                void EndFrame();
 
                void Enable(bool enabled) { m_enabled = enabled; }
+
+               void AddCounter(std::string name, uint64_t amt);
        private:
                struct ProfileZone
                {
@@ -32,6 +35,7 @@ namespace IPDF
                };
 
                std::map<std::string, ProfileZone> m_zones;
+               std::map<std::string, uint64_t> m_counters;
                std::stack<std::string> m_zone_stack;
                bool m_enabled;
        };