src/Utils/ProfilingInfo.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2003-2008 - TortoiseSVN
   4
   5 // This program is free software; you can redistribute it and/or
   6 // modify it under the terms of the GNU General Public License
   7 // as published by the Free Software Foundation; either version 2
   8 // of the License, or (at your option) any later version.
   9
  10 // This program is distributed in the hope that it will be useful,
  11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 // GNU General Public License for more details.
  14
  15 // You should have received a copy of the GNU General Public License
  16 // along with this program; if not, write to the Free Software Foundation,
  17 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18 //
  19
  20 #include "stdafx.h"
  21
  22 #ifndef __INTRIN_H_
  23 #include <intrin.h>
  24 #endif
  25
  26 #include "ProfilingInfo.h"
  27
  28 #ifdef _DEBUG
  29
  30 //////////////////////////////////////////////////////////////////////
  31 /// construction / destruction
  32 //////////////////////////////////////////////////////////////////////
  33
  34 CRecordProfileEvent::CRecordProfileEvent (CProfilingRecord* aRecord)
  35     : record (aRecord)
  36     , start (__rdtsc())
  37 {
  38 }
  39
  40 CRecordProfileEvent::~CRecordProfileEvent()
  41 {
  42         if (record)
  43                 record->Add (__rdtsc() - start);
  44 }
  45
  46 #endif
  47
  48 //////////////////////////////////////////////////////////////////////
  49 // construction / destruction
  50 //////////////////////////////////////////////////////////////////////
  51
  52 CProfilingRecord::CProfilingRecord ( const char* name
  53                                    , const char* file
  54                                    , int line)
  55     : name (name)
  56     , file (file)
  57     , line (line)
  58     , count (0)
  59     , sum (0)
  60     , minValue (ULLONG_MAX)
  61     , maxValue (0)
  62 {
  63 }
  64
  65 //////////////////////////////////////////////////////////////////////
  66 // record values
  67 //////////////////////////////////////////////////////////////////////
  68
  69 void CProfilingRecord::Add (unsigned __int64 value)
  70 {
  71     ++count;
  72     sum += value;
  73
  74     if (value < minValue)
  75         minValue = value;
  76     if (value > maxValue)
  77         maxValue = value;
  78 }
  79
  80 //////////////////////////////////////////////////////////////////////
  81 // modification
  82 //////////////////////////////////////////////////////////////////////
  83
  84 void CProfilingRecord::Reset()
  85 {
  86     count = 0;
  87     sum = 0;
  88
  89     minValue = LLONG_MAX;
  90     maxValue = 0;
  91 }
  92
  93 //////////////////////////////////////////////////////////////////////
  94 // construction / destruction
  95 //////////////////////////////////////////////////////////////////////
  96
  97 CProfilingInfo::CProfilingInfo()
  98 {
  99 }
 100
 101 CProfilingInfo::~CProfilingInfo(void)
 102 {
 103         if (records.size() > 0)
 104         {
 105                 // write profile to file
 106
 107                 TCHAR buffer [MAX_PATH];
 108                 if (GetModuleFileNameEx (GetCurrentProcess(), NULL, buffer, MAX_PATH) > 0)
 109                         try
 110                         {
 111                                 std::wstring fileName (buffer);
 112                                 fileName += L".profile";
 113
 114                                 std::string report = GetInstance()->GetReport();
 115
 116                                 CFile file (fileName.c_str(), CFile::modeCreate | CFile::modeWrite );
 117                                 file.Write (report.c_str(), (UINT)report.size());
 118                         }
 119                         catch (...)
 120                         {
 121                                 // ignore all file errors etc.
 122                         }
 123
 124
 125                 // free data
 126
 127                 for (size_t i = 0; i < records.size(); ++i)
 128                         delete records[i];
 129         }
 130 }
 131
 132 //////////////////////////////////////////////////////////////////////
 133 // access to default instance
 134 //////////////////////////////////////////////////////////////////////
 135
 136 CProfilingInfo* CProfilingInfo::GetInstance()
 137 {
 138     static CProfilingInfo instance;
 139     return &instance;
 140 }
 141
 142 //////////////////////////////////////////////////////////////////////
 143 // create a report
 144 //////////////////////////////////////////////////////////////////////
 145
 146 static std::string IntToStr (unsigned __int64 value)
 147 {
 148         char buffer[100];
 149         _ui64toa_s (value, buffer, 100, 10);
 150
 151         std::string result = buffer;
 152         for (size_t i = 3; i < result.length(); i += 4)
 153                 result.insert (result.length() - i, 1, ',');
 154
 155         return result;
 156 };
 157
 158 std::string CProfilingInfo::GetReport() const
 159 {
 160     enum { LINE_LENGTH = 500 };
 161
 162     char lineBuffer [LINE_LENGTH];
 163     const char * const format ="%10s%17s%17s%17s%6s %s\t%s\n";
 164
 165     std::string result;
 166     result.reserve (LINE_LENGTH * records.size());
 167     sprintf_s ( lineBuffer, format
 168               , "count", "sum", "min", "max"
 169               , "line", "name", "file");
 170     result += lineBuffer;
 171
 172     for ( TRecords::const_iterator iter = records.begin(), end = records.end()
 173         ; iter != end
 174         ; ++iter)
 175     {
 176         unsigned __int64 minValue = (*iter)->GetMinValue();
 177         if (minValue == ULLONG_MAX)
 178             minValue = 0;
 179
 180         sprintf_s ( lineBuffer, format
 181
 182                   , IntToStr ((*iter)->GetCount()).c_str()
 183                   , IntToStr ((*iter)->GetSum()).c_str()
 184                   , IntToStr (minValue).c_str()
 185                   , IntToStr ((*iter)->GetMaxValue()).c_str()
 186
 187                   , IntToStr ((*iter)->GetLine()).c_str()
 188                   , (*iter)->GetName()
 189                   , (*iter)->GetFile());
 190
 191         result += lineBuffer;
 192     }
 193
 194     return result;
 195 }
 196
 197 //////////////////////////////////////////////////////////////////////
 198 // add a new record
 199 //////////////////////////////////////////////////////////////////////
 200
 201 CProfilingRecord* CProfilingInfo::Create ( const char* name
 202                                          , const char* file
 203                                          , int line)
 204 {
 205     CProfilingRecord* record = new CProfilingRecord (name, file, line);
 206     records.push_back (record);
 207
 208     return record;
 209 }
 210