mymalloc.h

Go to the documentation of this file.

/*******************************************************************************
 * \copyright   This file is part of the GADGET4 N-body/SPH code developed
 * \copyright   by Volker Springel. Copyright (C) 2014-2020 by Volker Springel
 * \copyright   (vspringel@mpa-garching.mpg.de) and all contributing authors.
 *******************************************************************************/
 
#ifndef MYMALLOC_H
#define MYMALLOC_H
 
#include <stdio.h>
 
#define CACHELINESIZE 64
 
#define MAXBLOCKS 5000
#define MAXCHARS 40
 
#define LOC __FUNCTION__, __FILE__, __LINE__
#define MMM(x, y) (x, #x, y, __FUNCTION__, __FILE__, __LINE__)
#define DDD(x) (x, __FUNCTION__, __FILE__, __LINE__)
 
#define mymalloc(x, y) mymalloc_movable_fullinfo(NULL, x, y, __FUNCTION__, __FILE__, __LINE__, 0, 0, NULL)
#define mymalloc_clear(x, y) mymalloc_movable_fullinfo(NULL, x, y, __FUNCTION__, __FILE__, __LINE__, 0, 1, NULL)
#define mymalloc_g(x, y) mymalloc_movable_fullinfo(NULL, x, y, __FUNCTION__, __FILE__, __LINE__, 0, 0, callorigin)
 
#define mymalloc_movable(x, y, z) mymalloc_movable_fullinfo(x, y, z, __FUNCTION__, __FILE__, __LINE__, 1, 0, NULL)
#define mymalloc_movable_clear(x, y, z) mymalloc_movable_fullinfo(x, y, z, __FUNCTION__, __FILE__, __LINE__, 1, 1, NULL)
#define mymalloc_movable_g(x, y, z) mymalloc_movable_fullinfo(x, y, z, __FUNCTION__, __FILE__, __LINE__, 1, 0, callorigin)
 
#define myfree(x) myfree_movable_fullinfo(x, __FUNCTION__, __FILE__, __LINE__, 0)
#define myfree_movable(x) myfree_movable_fullinfo(x, __FUNCTION__, __FILE__, __LINE__, 1)
 
#define myrealloc(x, y) myrealloc_movable_fullinfo(x, y, __FUNCTION__, __FILE__, __LINE__, 0)
#define myrealloc_movable(x, y) myrealloc_movable_fullinfo(x, y, __FUNCTION__, __FILE__, __LINE__, 1)
 
#include "../mpi_utils/setcomm.h"
 
class memory : public setcomm
{
 public:
  memory() : setcomm("delayed init") {}
 
  size_t AllocatedBytes;
  size_t FreeBytes;
  char *Base; 
  void mymalloc_init(int maxmemsize, enum restart_options restartflag);
 
  void *mymalloc_movable_fullinfo(void *ptr, const char *varname, size_t n, const char *func, const char *file, int line,
                                  int movable_flag, int clear_flag, char *originflag);
 
  void *myrealloc_movable_fullinfo(void *p, size_t n, const char *func, const char *file, int line, int movable_flag);
 
  void myfree_movable_fullinfo(void *p, const char *func, const char *file, int line, int movable_flag);
 
  void *myfree_query_last_block(void);
 
  size_t roundup_to_multiple_of_cacheline_size(size_t n);
 
  void report_detailed_memory_usage_of_largest_task(void);
 
  void check_maxmemsize_setting(int maxmemsize);
 
  int myMPI_Win_allocate_shared(MPI_Aint size, int disp_unit, MPI_Info info, MPI_Comm comm, void *baseptr, MPI_Win *win);
  int myMPI_Win_shared_query(MPI_Win win, int rank, MPI_Aint *size, int *disp_unit, void *baseptr);
 
  inline double getAllocatedBytesInMB(void) { return AllocatedBytes * TO_MBYTE_FAC; }
 
  template <typename T>
  inline T *alloc(T *&ptr, const char *varname, size_t n, const char *func, const char *file, int linenr)
  {
    return static_cast<T *>(mymalloc_movable_fullinfo(&ptr, varname, n * sizeof(T), func, file, linenr, 0, 0, NULL));
  }
 
  template <typename T>
  inline T *alloc_movable(T *&ptr, const char *varname, size_t n, const char *func, const char *file, int linenr)
  {
    return static_cast<T *>(mymalloc_movable_fullinfo(&ptr, varname, n * sizeof(T), func, file, linenr, 1, 0, NULL));
  }
 
  template <typename T>
  inline T *realloc(T *&ptr, const char *varname, size_t n, const char *func, const char *file, int linenr)
  {
    return static_cast<T *>(myrealloc_movable_fullinfo(&ptr, n * sizeof(T), func, file, linenr, 0));
  }
 
  template <typename T>
  inline T *realloc_movable(T *&ptr, const char *varname, size_t n, const char *func, const char *file, int linenr)
  {
    return static_cast<T *>(myrealloc_movable_fullinfo(&ptr, n * sizeof(T), func, file, linenr, 1));
  }
 
  void dealloc(void *ptr, const char *func, const char *file, int linenr) { myfree_movable_fullinfo(ptr, func, file, linenr, 0); }
 
  void dealloc_movable(void *ptr, const char *func, const char *file, int linenr)
  {
    myfree_movable_fullinfo(ptr, func, file, linenr, 1);
  }
 
  void dump_memory_table(void);
 
 private:
  size_t AllocatedBytesGeneric;
 
  size_t HighMarkBytes;
  size_t HighMarkBytesWithoutGeneric;
 
  double OldGlobHighMarkMB;
  double OldGlobHighMarkMBWithoutGeneric;
 
  FILE *FdMemory; 
  size_t TotBytes; 
  int Nblocks;     
  char **Table;         
  size_t *BlockSize;    
  char *MovableFlag;    
  char *GenericFlag;    
  char ***BasePointers; 
  char *VarName;        
  char *FunctionName;   
  char *ParentFileName; 
  char *FileName;       
  int *LineNumber;      
  char *HighMarkTabBuf; 
  char *HighMarkTabBufWithoutGeneric; 
  enum restart_options RestartFlag;
 
  int dump_memory_table_buffer(char *p);
 
  void report_memory_usage(int rank, char *tabbuf);
};
 
extern memory Mem;
 
#endif