00001 /* _______ __ 00002 / ___/ / ___ __ _ / / ___ 00003 / /__/ _ \/ _ \/ V \/ _ \/ _ \ 00004 \___/_//_/\___/_/_/_/_.__/\___/ 00005 */ 00006 // CHOMBO Copyright (c) 2000-2004, The Regents of the University of 00007 // California, through Lawrence Berkeley National Laboratory (subject to 00008 // receipt of any required approvals from U.S. Dept. of Energy). All 00009 // rights reserved. 00010 // 00011 // Redistribution and use in source and binary forms, with or without 00012 // modification, are permitted provided that the following conditions are met: 00013 // 00014 // (1) Redistributions of source code must retain the above copyright 00015 // notice, this list of conditions and the following disclaimer. 00016 // (2) Redistributions in binary form must reproduce the above copyright 00017 // notice, this list of conditions and the following disclaimer in the 00018 // documentation and/or other materials provided with the distribution. 00019 // (3) Neither the name of Lawrence Berkeley National Laboratory, U.S. 00020 // Dept. of Energy nor the names of its contributors may be used to endorse 00021 // or promote products derived from this software without specific prior 00022 // written permission. 00023 // 00024 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 00025 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 00026 // TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A 00027 // PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER 00028 // OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 00029 // EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 00030 // PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 00031 // PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 00032 // LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 00033 // NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 00034 // SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 00035 // 00036 // You are under no obligation whatsoever to provide any bug fixes, 00037 // patches, or upgrades to the features, functionality or performance of 00038 // the source code ("Enhancements") to anyone; however, if you choose to 00039 // make your Enhancements available either publicly, or directly to 00040 // Lawrence Berkeley National Laboratory, without imposing a separate 00041 // written license agreement for such Enhancements, then you hereby grant 00042 // the following license: a non-exclusive, royalty-free perpetual license 00043 // to install, use, modify, prepare derivative works, incorporate into 00044 // other computer software, distribute, and sublicense such Enhancements or 00045 // derivative works thereof, in binary and source code form. 00046 // 00047 // TRADEMARKS. Product and company names mentioned herein may be the 00048 // trademarks of their respective owners. Any rights not expressly granted 00049 // herein are reserved. 00050 // 00051 00052 #ifndef _MEMTRACK_H_ 00053 #define _MEMTRACK_H_ 00054 00055 #ifdef ENABLE_MEMORY_TRACKING 00056 00057 #include <iostream> 00058 using std::ostream; 00059 00061 void ReportUnfreedMemory(ostream& a_os); 00062 00064 void ReportMemoryUsage(ostream& a_os); 00065 00067 void UnfreedMemory(); 00068 00070 void MemoryUsage(); 00071 00073 void memTrackingOn(); 00074 00076 void memtrackingOff(); 00077 00078 void overallMemoryUsage(long long& a_currentTotal, 00079 long long& a_peak); 00080 00081 void memtrackStamp(Real& a_current, 00082 Real& a_peak); 00083 00085 00089 class Memtrack 00090 { 00091 public: 00093 00100 static void ReportUnfreedMemory(ostream& a_os) 00101 { 00102 ::ReportUnfreedMemory(a_os); 00103 } 00104 00106 static void UnfreedMemory() 00107 { 00108 ::UnfreedMemory(); 00109 } 00110 00111 static void memTrackingOn() 00112 { 00113 ::memTrackingOn(); 00114 } 00115 00116 static void memtrackingOff() 00117 { 00118 ::memtrackingOff(); 00119 } 00120 00121 static void overallMemoryUsage(long long& a_currentTotal, 00122 long long& a_peak) 00123 { 00124 ::overallMemoryUsage(a_currentTotal, a_peak); 00125 } 00126 }; 00127 00128 // =========== end user functions.=================================== 00129 00130 void AddTrack(void* a_addr, 00131 size_t a_asize, 00132 const char* a_fname, 00133 int a_lnum, 00134 bool a_malloc); 00135 00136 void RemoveTrack(void* a_addr, 00137 bool a_malloc); 00138 00139 // void* operator new (size_t a_size, 00140 // char const* a_file, 00141 // int a_line); 00142 00143 // void operator delete (void *a_p) throw(); 00144 00145 // void* operator new[] (size_t a_size, 00146 // char const* a_file, 00147 // int a_line); 00148 00149 // void operator delete[] (void *a_p) throw(); 00150 00151 // these are ordered in this way so that we don't double track malloc 00152 // functions. 00153 void* mallocp(size_t a_size, 00154 const char* a_file, 00155 int a_line); 00156 00157 void* reallocp(void* a_p, 00158 size_t a_size, 00159 const char* a_file, 00160 int a_line); 00161 00162 void* callocp(size_t a_nelem, 00163 size_t a_elsize, 00164 const char* a_file, 00165 int a_line); 00166 00167 void freep(void* a_p); 00168 00169 #define calloc(a_a,a_b) callocp (a_a, a_b, __FILE__, __LINE__) 00170 #define realloc(a_a, a_b) reallocp(a_a, a_b, __FILE__, __LINE__) 00171 #define malloc(a_a) mallocp (a_a, __FILE__, __LINE__) 00172 #define free(a_a) freep (a_a) 00173 00174 #endif 00175 00176 #endif