1 files changed, 37 insertions, 0 deletions
diff --git a/src/gpgpu-sim/traffic_breakdown.h b/src/gpgpu-sim/traffic_breakdown.h
new file mode 100644
index 0000000..c9b8df5
--- /dev/null
+++ b/src/gpgpu-sim/traffic_breakdown.h
@@ -0,0 +1,37 @@
+#pragma once 
+
+#include <stdio.h>
+#include <map>
+#include <string> 
+
+// Breakdown traffic through the network according to category
+class traffic_breakdown
+{
+public: 
+   traffic_breakdown(const std::string &network_name) 
+   : m_network_name(network_name) 
+   { }
+
+   // print the stats 
+   void print(FILE* fout); 
+
+   // record the amount and type of traffic introduced by this mem_fetch object 
+   void record_traffic(class mem_fetch * mf, unsigned int size); 
+
+protected:
+
+   std::string m_network_name; 
+
+   /// helper functions to identify the type of traffic sent 
+   std::string classify_memfetch(class mem_fetch * mf); 
+
+   /// helper functions to identify the size of traffic sent 
+   unsigned int packet_size(class mem_fetch * mf); 
+
+   typedef std::string mf_packet_type;  // use string so that it remains extensible 
+   typedef unsigned int mf_packet_size; 
+   typedef std::map < mf_packet_size, unsigned int > traffic_class_t; 
+   typedef std::map < mf_packet_type, traffic_class_t > traffic_stat_t; 
+
+   traffic_stat_t m_stats; 
+};