summaryrefslogtreecommitdiff
path: root/src/base/ThreadProfiler.h
blob: a581271070f532a1d8637a3965957c9591edb68b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
//
//  libavg - Media Playback Engine. 
//  Copyright (C) 2003-2014 Ulrich von Zadow
//
//  This library is free software; you can redistribute it and/or
//  modify it under the terms of the GNU Lesser General Public
//  License as published by the Free Software Foundation; either
//  version 2 of the License, or (at your option) any later version.
//
//  This library is distributed in the hope that it will be useful,
//  but WITHOUT ANY WARRANTY; without even the implied warranty of
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
//  Lesser General Public License for more details.
//
//  You should have received a copy of the GNU Lesser General Public
//  License along with this library; if not, write to the Free Software
//  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
//
//  Current versions can be found at www.libavg.de
//

#ifndef _ThreadProfiler_H_ 
#define _ThreadProfiler_H_

#include "../api.h"
#include "ILogSink.h"

#include <boost/thread.hpp>
#include <boost/shared_ptr.hpp>
#include <boost/thread/tss.hpp>

#include <vector>
#include <map>
#if defined(_WIN32) || defined(_LIBCPP_VERSION)
#include <unordered_map>
#else
#include <tr1/unordered_map>
#endif
namespace avg {

class ThreadProfiler;
typedef boost::shared_ptr<ThreadProfiler> ThreadProfilerPtr;
class ProfilingZone;
typedef boost::shared_ptr<ProfilingZone> ProfilingZonePtr;
class ProfilingZoneID;

class AVG_API ThreadProfiler
{
public:
    static ThreadProfiler* get();
    static void kill();
    ThreadProfiler();
    virtual ~ThreadProfiler();
    void setLogCategory(category_t category);
 
    void start();
    void restart();
    void startZone(const ProfilingZoneID& zoneID);
    void stopZone(const ProfilingZoneID& zoneID);
    void dumpStatistics();
    void reset();
    int getNumZones();

    const std::string& getName() const;
    void setName(const std::string& sName);

private:
    ProfilingZonePtr addZone(const ProfilingZoneID& zoneID);
    std::string m_sName;

#if defined(_WIN32) || defined(_LIBCPP_VERSION)
    typedef std::unordered_map<const ProfilingZoneID*, ProfilingZonePtr> ZoneMap;
#else
    typedef std::tr1::unordered_map<const ProfilingZoneID*, ProfilingZonePtr> ZoneMap;
#endif
    typedef std::vector<ProfilingZonePtr> ZoneVector;
    ZoneMap m_ZoneMap;
    ZoneVector m_ActiveZones;
    ZoneVector m_Zones;
    bool m_bRunning;
    category_t m_LogCategory;

    static boost::thread_specific_ptr<ThreadProfiler*> s_pInstance;
};

}

#endif