// memsize.h

#ifndef memsize_H
#define memsize_H

// Return the total memory (in bytes) associated with
// various data structures.

#include <string>
#include <vector>
#include <set>
#include <map>

// String.
size_t memsize_string(std::string);

// Vector.
// Contained objects size is evaluated with sizeof.
template<class T>
size_t memsize_vector(std::vector<T> arg) {
  return sizeof(arg) + arg.capacity()*sizeof(T);
}

// Set.
// Contained objects size is evaluated with sizeof.
template<class T>
size_t memsize_set(std::set<T> arg) {
  struct Node {
    bool color;
    Node* parent;
    Node* left;
    Node* right;
  };
  return sizeof(arg) + arg.size()*(sizeof(T) + sizeof(Node));
}

// Multiset.
// Contained objects size is evaluated with sizeof.
template<class T>
size_t memsize_multiset(std::multiset<T> arg) {
  struct Node {
    bool color;
    Node* parent;
    Node* left;
    Node* right;
  };
  return sizeof(arg) + arg.size()*(sizeof(T) + sizeof(Node));
}

// Map.
// Contained objects size is evaluated with sizeof.
template<class T, class U>
size_t memsize_map(std::map<T,U> arg) {
  struct Node {
    bool color;
    Node* parent;
    Node* left;
    Node* right;
  };
  return sizeof(arg) +
         arg.size()*(sizeof(std::pair<T,U>) + sizeof(Node));
}

// Vector of strings.
size_t memsize_vector_string(std::vector<std::string> arg);

// Map of string-strings.
size_t memsize_map_string(std::map<std::string,std::string> arg);

#endif