Cleaned up a little with variable names
[charm.git] / src / ck-perf / trace-counter.h
1 /*****************************************************************************
2  * $Source$
3  * $Author$
4  * $Date$
5  * $Revision$
6  *****************************************************************************/
7
8 /**
9  * \addtogroup CkPerf
10 */
11 /*@{*/
12
13 #ifndef __trace_counter_h__
14 #define __trace_counter_h__
15
16 #include <stdio.h>
17 #include <errno.h>
18 #include "trace.h"
19 #include "ck.h"
20 #include "trace-common.h"
21 #include "conv-mach.h"
22 #ifdef CMK_ORIGIN2000
23 #include <sys/hwperftypes.h>
24 #endif
25
26 #define MAX_ENTRIES 500
27
28 //! track statistics for all entry points
29 class StatTable {
30   public:
31     StatTable(char** name, char** desc, int argc);
32     ~StatTable();
33     //! one entry is called for 'time' seconds, value is counter reading 
34     void setEp(int epidx, int stat, long long value, double time);
35     //! write three lines for each stat:
36     //!   1. number of calls for each entry
37     //!   2. average count for each entry
38     //!   3. total time in us spent for each entry
39     void write(FILE* fp);
40     void clear();
41     inline int numStats() { return numStats_; }
42
43   private:
44     //! struct to maintain statistics
45     struct Statistics {
46       char*  name;                    // name of stat being tracked
47       char*  desc;                    // description of stat being tracked
48       UInt   numCalled[MAX_ENTRIES];  // total number times called
49       UInt   totCount[MAX_ENTRIES];   // track average of value
50       double totTime[MAX_ENTRIES];    // total time associated with this counter
51
52       Statistics(): name(NULL) { }
53     };
54
55     Statistics* stats_;             // track stats for each entry point
56     int         numStats_;          // size of statistics being tracked
57     // bool        error_;             // will be set to true if error writing info
58 };
59
60 // counter log pool
61 class CountLogPool {
62   public:
63     CountLogPool(char* pgm, char** name, char** desc, int argc);
64     ~CountLogPool();
65     void write(void) ;
66     void writeSts(void);
67     void setEp(int epidx, long long count1, long long count2, double time);
68     void clearEps() { stats_.clear(); }
69
70   private:
71     FILE*     fp_;
72     StatTable stats_;
73 };
74
75 //! For each processor, TraceCounter calculates mean, stdev, etc of 
76 //! CPU performance counters for each entry point.
77 class TraceCounter : public Trace {
78   public:
79     TraceCounter();
80     ~TraceCounter();
81     void userEvent(int e) { }
82     void creation(envelope *e, int num=1) { }
83     void beginExecute(envelope *e);
84     void beginExecute(int event, int msgType, int ep, int srcPe, int mlen=0);
85     void endExecute(void);
86     void beginIdle(void) { }
87     void endIdle(void) { }
88     void beginPack(void);
89     void endPack(void);
90     void beginUnpack(void);
91     void endUnpack(void);
92     void beginCharmInit(void) { }
93     void endCharmInit(void) { }
94     void enqueue(envelope *e) { }
95     void dequeue(envelope *e) { }
96     void beginComputation(void);
97     void endComputation(void) { }
98
99     void traceInit(char **argv);
100     int traceRegisterUserEvent(const char*) { return 0; }
101     void traceClearEps();
102     void traceWriteSts();
103     void traceClose();
104     void traceBegin() { }
105     void traceEnd() { }
106
107     //! CounterArg is a linked list of strings that allows
108     //! processing of command line args
109     struct CounterArg {
110       int         code;
111       char*       arg;
112       char*       desc;
113       CounterArg* next;
114
115       CounterArg(): code(-1), arg(NULL), desc(NULL), next(NULL) { }
116       CounterArg(int c, char* a, char* d):
117         code(c), arg(a), desc(d), next(NULL) { }
118       void setValues(int _code, char* _arg, char* _desc) {
119         code = _code;  arg = _arg;  desc = _desc;
120       }
121     };
122
123   private:
124     int         execEP_;       // id currently executing entry point
125     double      startEP_;      // start time of currently executing ep
126     double      startPack_;    // start time of pack operation
127     double      startUnpack_;  // start time of unpack operation
128     CounterArg* firstArg_;     // pointer to start of linked list of args
129     CounterArg* lastArg_;      // pointer to end of linked list of args
130     int         argStrSize_;   // size of maximum arg string (formatted output)
131
132     int         counter1_;     // 1st counter to track
133     int         counter2_;     // 2nd counter to track
134
135     #ifdef CMK_ORIGIN2000
136     enum Status { IDLE, WORKING };
137     Status      status_;       // to prevent errors
138     int         genStart_;     // track value of start_counters
139     int         fileDesc_;     // file descriptor for accessing hw counters
140     hwperf_profevctrarg_t evctrArgs_;  // tells what type of counter to profile
141     hwperf_cntr_t         cnts_;       // reads the counters values
142     // starts hw profiling for these counters
143     // returns -1 if error, generation number if no error
144     int startCounter(int counter1, int counter2);
145     // gets hw profile values for these counters
146     // returns -1 if error, generation number if no error
147     int readCounters
148     (
149       int        counter1,
150       long long* value1,
151       int        counter2,
152       long long* value2
153     );
154     #endif // CMK_ORIGIN2000
155
156     //! add the argument parameters to the linked list of args choices
157     void registerArg(CounterArg* arg);
158     //! see if the arg (str or code) matches any in the linked list of choices
159     //! and sets arg->code to the SGI code
160     //! return true if arg matches, false otherwise
161     bool matchArg(CounterArg* arg);
162     //! print out all arguments in the linked-list of choices
163     void printHelp();
164 };
165
166 #endif  // __trace_counter_h__
167
168 /*@}*/
169
170