#ifndef MEMPROF_MIBMAP_H_
#define MEMPROF_MIBMAP_H_

#include "memprof_meminfoblock.h"
#include "sanitizer_common/sanitizer_addrhashmap.h"
#include "sanitizer_common/sanitizer_mutex.h"

namespace __memprof {

struct LockedMemInfoBlock {
  __sanitizer::StaticSpinMutex mutex;
  MemInfoBlock mib;
};

// The MIB map stores a mapping from stack ids to MemInfoBlocks.
typedef __sanitizer::AddrHashMap<LockedMemInfoBlock *, 200003> MIBMapTy;

// Insert a new MemInfoBlock or merge with an existing block identified by the
// stack id.
void InsertOrMerge(const uptr Id, const MemInfoBlock &Block, MIBMapTy &Map);

} // namespace __memprof

#endif // MEMPROF_MIBMAP_H_