#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;
};