blob: da689700953b46f01a8ea0dc1605c7ed4becd8f3 [file] [log] [blame]
Fred Drake8c081a12001-10-12 20:57:55 +00001/*
2 * This is the High Performance Python Profiler portion of HotShot.
3 */
4
Tim Peters885d4572001-11-28 20:27:42 +00005#include "Python.h"
6#include "compile.h"
7#include "eval.h"
8#include "frameobject.h"
9#include "structmember.h"
Fred Drake8c081a12001-10-12 20:57:55 +000010
Fred Drake8c081a12001-10-12 20:57:55 +000011/*
12 * Which timer to use should be made more configurable, but that should not
Tim Petersfeab23f2001-10-13 00:11:10 +000013 * be difficult. This will do for now.
Fred Drake8c081a12001-10-12 20:57:55 +000014 */
Martin v. Löwis6238d2b2002-06-30 15:26:10 +000015#ifdef MS_WINDOWS
Fred Drake8c081a12001-10-12 20:57:55 +000016#include <windows.h>
Tim Peters1566a172001-10-12 22:08:39 +000017#include <direct.h> /* for getcwd() */
Tim Peters7d99ff22001-10-13 07:37:52 +000018typedef __int64 hs_time;
19#define GETTIMEOFDAY(P_HS_TIME) \
20 { LARGE_INTEGER _temp; \
21 QueryPerformanceCounter(&_temp); \
22 *(P_HS_TIME) = _temp.QuadPart; }
23
Tim Petersfeab23f2001-10-13 00:11:10 +000024
Fred Drake8c081a12001-10-12 20:57:55 +000025#else
26#ifndef HAVE_GETTIMEOFDAY
27#error "This module requires gettimeofday() on non-Windows platforms!"
28#endif
Andrew MacIntyre7bf68332002-03-03 02:59:16 +000029#if defined(macintosh) || (defined(PYOS_OS2) && defined(PYCC_GCC))
Jack Jansen963659a2001-10-23 22:26:16 +000030#include <sys/time.h>
31#else
Fred Drake8c081a12001-10-12 20:57:55 +000032#include <sys/resource.h>
33#include <sys/times.h>
Jack Jansen963659a2001-10-23 22:26:16 +000034#endif
Fred Drake8c081a12001-10-12 20:57:55 +000035typedef struct timeval hs_time;
36#endif
37
38#if !defined(__cplusplus) && !defined(inline)
39#ifdef __GNUC__
40#define inline __inline
41#endif
42#endif
43
44#ifndef inline
45#define inline
46#endif
47
48#define BUFFERSIZE 10240
49
Jack Jansen963659a2001-10-23 22:26:16 +000050#ifdef macintosh
51#define PATH_MAX 254
52#endif
53
Andrew MacIntyre7bf68332002-03-03 02:59:16 +000054#if defined(PYOS_OS2) && defined(PYCC_GCC)
55#define PATH_MAX 260
56#endif
57
Martin v. Löwis8eb92a02002-09-19 08:03:21 +000058#if defined(__sgi) && _COMPILER_VERSION>700 && !defined(PATH_MAX)
59/* fix PATH_MAX not being defined with MIPSPro 7.x
60 if mode is ANSI C (default) */
61#define PATH_MAX 1024
62#endif
63
Tim Peters1566a172001-10-12 22:08:39 +000064#ifndef PATH_MAX
65# ifdef MAX_PATH
66# define PATH_MAX MAX_PATH
Martin v. Löwis21ee4092002-09-30 16:19:48 +000067# elif defined (_POSIX_PATH_MAX)
68# define PATH_MAX _POSIX_PATH_MAX
Tim Peters1566a172001-10-12 22:08:39 +000069# else
70# error "Need a defn. for PATH_MAX in _hotshot.c"
71# endif
72#endif
73
Fred Drake8c081a12001-10-12 20:57:55 +000074typedef struct {
75 PyObject_HEAD
76 PyObject *filemap;
77 PyObject *logfilename;
78 int index;
79 unsigned char buffer[BUFFERSIZE];
80 FILE *logfp;
81 int lineevents;
82 int linetimings;
Fred Drake30d1c752001-10-15 22:11:02 +000083 int frametimings;
Fred Drake8c081a12001-10-12 20:57:55 +000084 /* size_t filled; */
85 int active;
86 int next_fileno;
Fred Drake8c081a12001-10-12 20:57:55 +000087 hs_time prev_timeofday;
88} ProfilerObject;
89
90typedef struct {
91 PyObject_HEAD
Fred Drake4c2e1af2001-10-29 20:45:57 +000092 PyObject *info;
Fred Drake8c081a12001-10-12 20:57:55 +000093 FILE *logfp;
Fred Drake8c081a12001-10-12 20:57:55 +000094 int linetimings;
Fred Drake30d1c752001-10-15 22:11:02 +000095 int frametimings;
Fred Drake8c081a12001-10-12 20:57:55 +000096} LogReaderObject;
97
98static PyObject * ProfilerError = NULL;
99
100
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000101#ifndef MS_WINDOWS
Fred Drake8c081a12001-10-12 20:57:55 +0000102#ifdef GETTIMEOFDAY_NO_TZ
103#define GETTIMEOFDAY(ptv) gettimeofday((ptv))
104#else
105#define GETTIMEOFDAY(ptv) gettimeofday((ptv), (struct timezone *)NULL)
106#endif
107#endif
108
109
110/* The log reader... */
111
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000112PyDoc_STRVAR(logreader_close__doc__,
Tim Petersfeab23f2001-10-13 00:11:10 +0000113"close()\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000114"Close the log file, preventing additional records from being read.");
Fred Drake8c081a12001-10-12 20:57:55 +0000115
116static PyObject *
117logreader_close(LogReaderObject *self, PyObject *args)
118{
Fred Drake666bf522002-07-18 19:11:44 +0000119 if (self->logfp != NULL) {
120 fclose(self->logfp);
121 self->logfp = NULL;
Fred Drake8c081a12001-10-12 20:57:55 +0000122 }
Fred Drake666bf522002-07-18 19:11:44 +0000123 Py_INCREF(Py_None);
124
125 return Py_None;
126}
127
128PyDoc_STRVAR(logreader_fileno__doc__,
129"fileno() -> file descriptor\n"
130"Returns the file descriptor for the log file, if open.\n"
131"Raises ValueError if the log file is closed.");
132
133static PyObject *
134logreader_fileno(LogReaderObject *self)
135{
136 if (self->logfp == NULL) {
137 PyErr_SetString(PyExc_ValueError,
138 "logreader's file object already closed");
139 return NULL;
140 }
141 return PyInt_FromLong(fileno(self->logfp));
Fred Drake8c081a12001-10-12 20:57:55 +0000142}
143
Fred Drake8c081a12001-10-12 20:57:55 +0000144
145/* Log File Format
146 * ---------------
147 *
148 * The log file consists of a sequence of variable-length records.
149 * Each record is identified with a record type identifier in two
150 * bits of the first byte. The two bits are the "least significant"
151 * bits of the byte.
152 *
153 * Low bits: Opcode: Meaning:
154 * 0x00 ENTER enter a frame
155 * 0x01 EXIT exit a frame
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000156 * 0x02 LINENO execution moved onto a different line
Fred Drake8c081a12001-10-12 20:57:55 +0000157 * 0x03 OTHER more bits are needed to deecode
158 *
159 * If the type is OTHER, the record is not packed so tightly, and the
160 * remaining bits are used to disambiguate the record type. These
161 * records are not used as frequently so compaction is not an issue.
162 * Each of the first three record types has a highly tailored
163 * structure that allows it to be packed tightly.
164 *
165 * The OTHER records have the following identifiers:
166 *
167 * First byte: Opcode: Meaning:
168 * 0x13 ADD_INFO define a key/value pair
169 * 0x23 DEFINE_FILE define an int->filename mapping
170 * 0x33 LINE_TIMES indicates if LINENO events have tdeltas
Fred Drake30d1c752001-10-15 22:11:02 +0000171 * 0x43 DEFINE_FUNC define a (fileno,lineno)->funcname mapping
172 * 0x53 FRAME_TIMES indicates if ENTER/EXIT events have tdeltas
Fred Drake8c081a12001-10-12 20:57:55 +0000173 *
174 * Packed Integers
175 *
176 * "Packed integers" are non-negative integer values encoded as a
177 * sequence of bytes. Each byte is encoded such that the most
178 * significant bit is set if the next byte is also part of the
179 * integer. Each byte provides bits to the least-significant end of
180 * the result; the accumulated value must be shifted up to place the
181 * new bits into the result.
182 *
183 * "Modified packed integers" are packed integers where only a portion
184 * of the first byte is used. In the rest of the specification, these
185 * are referred to as "MPI(n,name)", where "n" is the number of bits
186 * discarded from the least-signicant positions of the byte, and
187 * "name" is a name being given to those "discarded" bits, since they
188 * are a field themselves.
189 *
190 * ENTER records:
191 *
192 * MPI(2,type) fileno -- type is 0x00
Fred Drake8c081a12001-10-12 20:57:55 +0000193 * PI lineno
Fred Drake30d1c752001-10-15 22:11:02 +0000194 * PI tdelta -- iff frame times are enabled
Fred Drake8c081a12001-10-12 20:57:55 +0000195 *
196 * EXIT records
197 *
Fred Drake30d1c752001-10-15 22:11:02 +0000198 * MPI(2,type) tdelta -- type is 0x01; tdelta will be 0
199 * if frame times are disabled
Fred Drake8c081a12001-10-12 20:57:55 +0000200 *
201 * LINENO records
202 *
203 * MPI(2,type) lineno -- type is 0x02
204 * PI tdelta -- iff LINENO includes it
205 *
206 * ADD_INFO records
207 *
Fred Drake30d1c752001-10-15 22:11:02 +0000208 * BYTE type -- always 0x13
Fred Drake8c081a12001-10-12 20:57:55 +0000209 * PI len1 -- length of first string
210 * BYTE string1[len1] -- len1 bytes of string data
211 * PI len2 -- length of second string
212 * BYTE string2[len2] -- len2 bytes of string data
213 *
214 * DEFINE_FILE records
215 *
Fred Drake30d1c752001-10-15 22:11:02 +0000216 * BYTE type -- always 0x23
Fred Drake8c081a12001-10-12 20:57:55 +0000217 * PI fileno
218 * PI len -- length of filename
219 * BYTE filename[len] -- len bytes of string data
220 *
Fred Drake30d1c752001-10-15 22:11:02 +0000221 * DEFINE_FUNC records
222 *
223 * BYTE type -- always 0x43
224 * PI fileno
225 * PI lineno
226 * PI len -- length of funcname
227 * BYTE funcname[len] -- len bytes of string data
228 *
Fred Drake8c081a12001-10-12 20:57:55 +0000229 * LINE_TIMES records
Fred Drake30d1c752001-10-15 22:11:02 +0000230 *
231 * This record can be used only before the start of ENTER/EXIT/LINENO
232 * records. If have_tdelta is true, LINENO records will include the
233 * tdelta field, otherwise it will be omitted. If this record is not
234 * given, LINENO records will not contain the tdelta field.
235 *
236 * BYTE type -- always 0x33
Fred Drake8c081a12001-10-12 20:57:55 +0000237 * BYTE have_tdelta -- 0 if LINENO does *not* have
238 * timing information
Fred Drake30d1c752001-10-15 22:11:02 +0000239 * FRAME_TIMES records
240 *
241 * This record can be used only before the start of ENTER/EXIT/LINENO
242 * records. If have_tdelta is true, ENTER and EXIT records will
243 * include the tdelta field, otherwise it will be omitted. If this
244 * record is not given, ENTER and EXIT records will contain the tdelta
245 * field.
246 *
247 * BYTE type -- always 0x53
248 * BYTE have_tdelta -- 0 if ENTER/EXIT do *not* have
249 * timing information
Fred Drake8c081a12001-10-12 20:57:55 +0000250 */
251
252#define WHAT_ENTER 0x00
253#define WHAT_EXIT 0x01
254#define WHAT_LINENO 0x02
255#define WHAT_OTHER 0x03 /* only used in decoding */
256#define WHAT_ADD_INFO 0x13
257#define WHAT_DEFINE_FILE 0x23
258#define WHAT_LINE_TIMES 0x33
Fred Drake30d1c752001-10-15 22:11:02 +0000259#define WHAT_DEFINE_FUNC 0x43
260#define WHAT_FRAME_TIMES 0x53
Fred Drake8c081a12001-10-12 20:57:55 +0000261
262#define ERR_NONE 0
263#define ERR_EOF -1
264#define ERR_EXCEPTION -2
Fred Drake4c2e1af2001-10-29 20:45:57 +0000265#define ERR_BAD_RECTYPE -3
Fred Drake8c081a12001-10-12 20:57:55 +0000266
267#define PISIZE (sizeof(int) + 1)
268#define MPISIZE (PISIZE + 1)
269
270/* Maximum size of "normal" events -- nothing that contains string data */
271#define MAXEVENTSIZE (MPISIZE + PISIZE*2)
272
273
274/* Unpack a packed integer; if "discard" is non-zero, unpack a modified
275 * packed integer with "discard" discarded bits.
276 */
277static int
278unpack_packed_int(LogReaderObject *self, int *pvalue, int discard)
279{
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000280 int c;
Fred Drake8c081a12001-10-12 20:57:55 +0000281 int accum = 0;
282 int bits = 0;
Fred Drake8c081a12001-10-12 20:57:55 +0000283 int cont;
284
285 do {
Fred Drake8c081a12001-10-12 20:57:55 +0000286 /* read byte */
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000287 if ((c = fgetc(self->logfp)) == EOF)
288 return ERR_EOF;
289 accum |= ((c & 0x7F) >> discard) << bits;
Fred Drake8c081a12001-10-12 20:57:55 +0000290 bits += (7 - discard);
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000291 cont = c & 0x80;
Fred Drake8c081a12001-10-12 20:57:55 +0000292 discard = 0;
Fred Drake8c081a12001-10-12 20:57:55 +0000293 } while (cont);
294
Fred Drake8c081a12001-10-12 20:57:55 +0000295 *pvalue = accum;
296
297 return 0;
298}
299
300/* Unpack a string, which is encoded as a packed integer giving the
301 * length of the string, followed by the string data.
302 */
303static int
304unpack_string(LogReaderObject *self, PyObject **pvalue)
305{
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000306 int i;
Fred Drake8c081a12001-10-12 20:57:55 +0000307 int len;
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000308 int err;
Guido van Rossum0b624f62002-07-20 00:38:01 +0000309 int ch;
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000310 char *buf;
311
312 if ((err = unpack_packed_int(self, &len, 0)))
313 return err;
Fred Drake8c081a12001-10-12 20:57:55 +0000314
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000315 buf = malloc(len);
316 for (i=0; i < len; i++) {
Guido van Rossum0b624f62002-07-20 00:38:01 +0000317 ch = fgetc(self->logfp);
318 buf[i] = ch;
319 if (ch == EOF) {
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000320 free(buf);
321 return ERR_EOF;
Fred Drake8c081a12001-10-12 20:57:55 +0000322 }
323 }
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000324 *pvalue = PyString_FromStringAndSize(buf, len);
325 free(buf);
326 if (*pvalue == NULL) {
327 return ERR_EXCEPTION;
328 }
329 return 0;
Fred Drake8c081a12001-10-12 20:57:55 +0000330}
331
332
Fred Drake4c2e1af2001-10-29 20:45:57 +0000333static int
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000334unpack_add_info(LogReaderObject *self)
Fred Drake4c2e1af2001-10-29 20:45:57 +0000335{
336 PyObject *key;
337 PyObject *value = NULL;
338 int err;
339
Fred Drake4c2e1af2001-10-29 20:45:57 +0000340 err = unpack_string(self, &key);
341 if (!err) {
342 err = unpack_string(self, &value);
343 if (err)
344 Py_DECREF(key);
345 else {
346 PyObject *list = PyDict_GetItem(self->info, key);
347 if (list == NULL) {
348 list = PyList_New(0);
349 if (list == NULL) {
350 err = ERR_EXCEPTION;
351 goto finally;
352 }
353 if (PyDict_SetItem(self->info, key, list)) {
354 err = ERR_EXCEPTION;
355 goto finally;
356 }
357 }
358 if (PyList_Append(list, value))
359 err = ERR_EXCEPTION;
360 }
361 }
362 finally:
363 Py_XDECREF(key);
364 Py_XDECREF(value);
365 return err;
366}
367
368
369static void
Fred Drake666bf522002-07-18 19:11:44 +0000370eof_error(LogReaderObject *self)
Fred Drake4c2e1af2001-10-29 20:45:57 +0000371{
Fred Drake666bf522002-07-18 19:11:44 +0000372 fclose(self->logfp);
373 self->logfp = NULL;
Fred Drake4c2e1af2001-10-29 20:45:57 +0000374 PyErr_SetString(PyExc_EOFError,
375 "end of file with incomplete profile record");
376}
377
Fred Drake8c081a12001-10-12 20:57:55 +0000378static PyObject *
379logreader_tp_iternext(LogReaderObject *self)
380{
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000381 int c;
382 int what;
Fred Drake8c081a12001-10-12 20:57:55 +0000383 int err = ERR_NONE;
384 int lineno = -1;
385 int fileno = -1;
386 int tdelta = -1;
387 PyObject *s1 = NULL, *s2 = NULL;
388 PyObject *result = NULL;
389#if 0
390 unsigned char b0, b1;
391#endif
392
393 if (self->logfp == NULL) {
394 PyErr_SetString(ProfilerError,
395 "cannot iterate over closed LogReader object");
396 return NULL;
397 }
Fred Drake4c2e1af2001-10-29 20:45:57 +0000398
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000399restart:
400 /* decode the record type */
Fred Drake666bf522002-07-18 19:11:44 +0000401 if ((c = fgetc(self->logfp)) == EOF) {
402 fclose(self->logfp);
403 self->logfp = NULL;
Fred Drake8c081a12001-10-12 20:57:55 +0000404 return NULL;
Fred Drake666bf522002-07-18 19:11:44 +0000405 }
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000406 what = c & WHAT_OTHER;
407 if (what == WHAT_OTHER)
408 what = c; /* need all the bits for type */
409 else
410 ungetc(c, self->logfp); /* type byte includes packed int */
Fred Drake8c081a12001-10-12 20:57:55 +0000411
Fred Drake8c081a12001-10-12 20:57:55 +0000412 switch (what) {
413 case WHAT_ENTER:
414 err = unpack_packed_int(self, &fileno, 2);
415 if (!err) {
Fred Drake30d1c752001-10-15 22:11:02 +0000416 err = unpack_packed_int(self, &lineno, 0);
417 if (self->frametimings && !err)
418 err = unpack_packed_int(self, &tdelta, 0);
Fred Drake8c081a12001-10-12 20:57:55 +0000419 }
420 break;
421 case WHAT_EXIT:
422 err = unpack_packed_int(self, &tdelta, 2);
423 break;
424 case WHAT_LINENO:
425 err = unpack_packed_int(self, &lineno, 2);
426 if (self->linetimings && !err)
427 err = unpack_packed_int(self, &tdelta, 0);
428 break;
429 case WHAT_ADD_INFO:
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000430 err = unpack_add_info(self);
Fred Drake8c081a12001-10-12 20:57:55 +0000431 break;
432 case WHAT_DEFINE_FILE:
433 err = unpack_packed_int(self, &fileno, 0);
434 if (!err) {
435 err = unpack_string(self, &s1);
436 if (!err) {
437 Py_INCREF(Py_None);
438 s2 = Py_None;
439 }
440 }
441 break;
Fred Drake30d1c752001-10-15 22:11:02 +0000442 case WHAT_DEFINE_FUNC:
443 err = unpack_packed_int(self, &fileno, 0);
444 if (!err) {
445 err = unpack_packed_int(self, &lineno, 0);
446 if (!err)
447 err = unpack_string(self, &s1);
448 }
449 break;
Fred Drake8c081a12001-10-12 20:57:55 +0000450 case WHAT_LINE_TIMES:
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000451 if ((c = fgetc(self->logfp)) == EOF)
Fred Drake8c081a12001-10-12 20:57:55 +0000452 err = ERR_EOF;
453 else {
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000454 self->linetimings = c ? 1 : 0;
455 goto restart;
456 }
Fred Drake4c2e1af2001-10-29 20:45:57 +0000457 break;
Fred Drake30d1c752001-10-15 22:11:02 +0000458 case WHAT_FRAME_TIMES:
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000459 if ((c = fgetc(self->logfp)) == EOF)
Fred Drake30d1c752001-10-15 22:11:02 +0000460 err = ERR_EOF;
461 else {
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000462 self->frametimings = c ? 1 : 0;
463 goto restart;
464 }
Fred Drake4c2e1af2001-10-29 20:45:57 +0000465 break;
Fred Drake8c081a12001-10-12 20:57:55 +0000466 default:
Fred Drake4c2e1af2001-10-29 20:45:57 +0000467 err = ERR_BAD_RECTYPE;
Fred Drake8c081a12001-10-12 20:57:55 +0000468 }
Fred Drake4c2e1af2001-10-29 20:45:57 +0000469 if (err == ERR_BAD_RECTYPE) {
470 PyErr_SetString(PyExc_ValueError,
471 "unknown record type in log file");
472 }
473 else if (err == ERR_EOF) {
Fred Drake666bf522002-07-18 19:11:44 +0000474 eof_error(self);
Fred Drake8c081a12001-10-12 20:57:55 +0000475 }
476 else if (!err) {
477 result = PyTuple_New(4);
478 PyTuple_SET_ITEM(result, 0, PyInt_FromLong(what));
479 PyTuple_SET_ITEM(result, 2, PyInt_FromLong(fileno));
Fred Drake30d1c752001-10-15 22:11:02 +0000480 if (s1 == NULL)
Fred Drake8c081a12001-10-12 20:57:55 +0000481 PyTuple_SET_ITEM(result, 1, PyInt_FromLong(tdelta));
Fred Drake30d1c752001-10-15 22:11:02 +0000482 else
Fred Drake8c081a12001-10-12 20:57:55 +0000483 PyTuple_SET_ITEM(result, 1, s1);
Fred Drake30d1c752001-10-15 22:11:02 +0000484 if (s2 == NULL)
485 PyTuple_SET_ITEM(result, 3, PyInt_FromLong(lineno));
486 else
Fred Drake8c081a12001-10-12 20:57:55 +0000487 PyTuple_SET_ITEM(result, 3, s2);
Fred Drake8c081a12001-10-12 20:57:55 +0000488 }
489 /* The only other case is err == ERR_EXCEPTION, in which case the
490 * exception is already set.
491 */
492#if 0
493 b0 = self->buffer[self->index];
494 b1 = self->buffer[self->index + 1];
495 if (b0 & 1) {
496 /* This is a line-number event. */
497 what = PyTrace_LINE;
498 lineno = ((b0 & ~1) << 7) + b1;
499 self->index += 2;
500 }
501 else {
502 what = (b0 & 0x0E) >> 1;
503 tdelta = ((b0 & 0xF0) << 4) + b1;
504 if (what == PyTrace_CALL) {
505 /* we know there's a 2-byte file ID & 2-byte line number */
506 fileno = ((self->buffer[self->index + 2] << 8)
507 + self->buffer[self->index + 3]);
508 lineno = ((self->buffer[self->index + 4] << 8)
509 + self->buffer[self->index + 5]);
510 self->index += 6;
511 }
512 else
513 self->index += 2;
514 }
515#endif
516 return result;
517}
518
519static void
520logreader_dealloc(LogReaderObject *self)
521{
522 if (self->logfp != NULL) {
523 fclose(self->logfp);
524 self->logfp = NULL;
525 }
526 PyObject_Del(self);
527}
528
529static PyObject *
530logreader_sq_item(LogReaderObject *self, int index)
531{
532 PyObject *result = logreader_tp_iternext(self);
533 if (result == NULL && !PyErr_Occurred()) {
534 PyErr_SetString(PyExc_IndexError, "no more events in log");
535 return NULL;
536 }
537 return result;
538}
539
Fred Drake62c1e3c2001-12-04 21:40:53 +0000540static void
541do_stop(ProfilerObject *self);
Fred Drake8c081a12001-10-12 20:57:55 +0000542
543static int
544flush_data(ProfilerObject *self)
545{
546 /* Need to dump data to the log file... */
547 size_t written = fwrite(self->buffer, 1, self->index, self->logfp);
Tim Peters1566a172001-10-12 22:08:39 +0000548 if (written == (size_t)self->index)
Fred Drake8c081a12001-10-12 20:57:55 +0000549 self->index = 0;
550 else {
551 memmove(self->buffer, &self->buffer[written],
552 self->index - written);
553 self->index -= written;
554 if (written == 0) {
555 char *s = PyString_AsString(self->logfilename);
556 PyErr_SetFromErrnoWithFilename(PyExc_IOError, s);
Fred Drake62c1e3c2001-12-04 21:40:53 +0000557 do_stop(self);
Fred Drake8c081a12001-10-12 20:57:55 +0000558 return -1;
559 }
560 }
561 if (written > 0) {
562 if (fflush(self->logfp)) {
563 char *s = PyString_AsString(self->logfilename);
564 PyErr_SetFromErrnoWithFilename(PyExc_IOError, s);
Fred Drake62c1e3c2001-12-04 21:40:53 +0000565 do_stop(self);
Fred Drake8c081a12001-10-12 20:57:55 +0000566 return -1;
567 }
568 }
569 return 0;
570}
571
Fred Drake62c1e3c2001-12-04 21:40:53 +0000572static inline int
Fred Drake8c081a12001-10-12 20:57:55 +0000573pack_packed_int(ProfilerObject *self, int value)
574{
575 unsigned char partial;
576
577 do {
578 partial = value & 0x7F;
579 value >>= 7;
580 if (value)
581 partial |= 0x80;
582 self->buffer[self->index] = partial;
583 self->index++;
584 } while (value);
Fred Drake62c1e3c2001-12-04 21:40:53 +0000585 return 0;
Fred Drake8c081a12001-10-12 20:57:55 +0000586}
587
588/* Encode a modified packed integer, with a subfield of modsize bits
589 * containing the value "subfield". The value of subfield is not
590 * checked to ensure it actually fits in modsize bits.
591 */
Fred Drake62c1e3c2001-12-04 21:40:53 +0000592static inline int
Fred Drake8c081a12001-10-12 20:57:55 +0000593pack_modified_packed_int(ProfilerObject *self, int value,
594 int modsize, int subfield)
595{
596 const int maxvalues[] = {-1, 1, 3, 7, 15, 31, 63, 127};
597
598 int bits = 7 - modsize;
599 int partial = value & maxvalues[bits];
600 unsigned char b = subfield | (partial << modsize);
601
602 if (partial != value) {
603 b |= 0x80;
604 self->buffer[self->index] = b;
605 self->index++;
Fred Drake62c1e3c2001-12-04 21:40:53 +0000606 return pack_packed_int(self, value >> bits);
Fred Drake8c081a12001-10-12 20:57:55 +0000607 }
Fred Drake62c1e3c2001-12-04 21:40:53 +0000608 self->buffer[self->index] = b;
609 self->index++;
610 return 0;
Fred Drake8c081a12001-10-12 20:57:55 +0000611}
612
Fred Drake62c1e3c2001-12-04 21:40:53 +0000613static int
Fred Drake30d1c752001-10-15 22:11:02 +0000614pack_string(ProfilerObject *self, const char *s, int len)
Fred Drake8c081a12001-10-12 20:57:55 +0000615{
Fred Drake62c1e3c2001-12-04 21:40:53 +0000616 if (len + PISIZE + self->index >= BUFFERSIZE) {
617 if (flush_data(self) < 0)
618 return -1;
619 }
620 if (pack_packed_int(self, len) < 0)
621 return -1;
Fred Drake8c081a12001-10-12 20:57:55 +0000622 memcpy(self->buffer + self->index, s, len);
623 self->index += len;
Fred Drake62c1e3c2001-12-04 21:40:53 +0000624 return 0;
Fred Drake8c081a12001-10-12 20:57:55 +0000625}
626
Fred Drake62c1e3c2001-12-04 21:40:53 +0000627static int
Fred Drake8c081a12001-10-12 20:57:55 +0000628pack_add_info(ProfilerObject *self, const char *s1, const char *s2)
629{
630 int len1 = strlen(s1);
631 int len2 = strlen(s2);
632
Fred Drake62c1e3c2001-12-04 21:40:53 +0000633 if (len1 + len2 + PISIZE*2 + 1 + self->index >= BUFFERSIZE) {
634 if (flush_data(self) < 0)
635 return -1;
636 }
Fred Drake8c081a12001-10-12 20:57:55 +0000637 self->buffer[self->index] = WHAT_ADD_INFO;
638 self->index++;
Fred Drake62c1e3c2001-12-04 21:40:53 +0000639 if (pack_string(self, s1, len1) < 0)
640 return -1;
641 return pack_string(self, s2, len2);
Fred Drake8c081a12001-10-12 20:57:55 +0000642}
643
Fred Drake62c1e3c2001-12-04 21:40:53 +0000644static int
Fred Drake8c081a12001-10-12 20:57:55 +0000645pack_define_file(ProfilerObject *self, int fileno, const char *filename)
646{
647 int len = strlen(filename);
648
Fred Drake62c1e3c2001-12-04 21:40:53 +0000649 if (len + PISIZE*2 + 1 + self->index >= BUFFERSIZE) {
650 if (flush_data(self) < 0)
651 return -1;
652 }
Fred Drake8c081a12001-10-12 20:57:55 +0000653 self->buffer[self->index] = WHAT_DEFINE_FILE;
654 self->index++;
Fred Drake62c1e3c2001-12-04 21:40:53 +0000655 if (pack_packed_int(self, fileno) < 0)
656 return -1;
657 return pack_string(self, filename, len);
Fred Drake30d1c752001-10-15 22:11:02 +0000658}
659
Fred Drake62c1e3c2001-12-04 21:40:53 +0000660static int
Fred Drake30d1c752001-10-15 22:11:02 +0000661pack_define_func(ProfilerObject *self, int fileno, int lineno,
662 const char *funcname)
663{
664 int len = strlen(funcname);
665
Fred Drake62c1e3c2001-12-04 21:40:53 +0000666 if (len + PISIZE*3 + 1 + self->index >= BUFFERSIZE) {
667 if (flush_data(self) < 0)
668 return -1;
669 }
Fred Drake30d1c752001-10-15 22:11:02 +0000670 self->buffer[self->index] = WHAT_DEFINE_FUNC;
671 self->index++;
Fred Drake62c1e3c2001-12-04 21:40:53 +0000672 if (pack_packed_int(self, fileno) < 0)
673 return -1;
674 if (pack_packed_int(self, lineno) < 0)
675 return -1;
676 return pack_string(self, funcname, len);
Fred Drake8c081a12001-10-12 20:57:55 +0000677}
678
Fred Drake62c1e3c2001-12-04 21:40:53 +0000679static int
Fred Drake8c081a12001-10-12 20:57:55 +0000680pack_line_times(ProfilerObject *self)
681{
Fred Drake62c1e3c2001-12-04 21:40:53 +0000682 if (2 + self->index >= BUFFERSIZE) {
683 if (flush_data(self) < 0)
684 return -1;
685 }
Fred Drake8c081a12001-10-12 20:57:55 +0000686 self->buffer[self->index] = WHAT_LINE_TIMES;
687 self->buffer[self->index + 1] = self->linetimings ? 1 : 0;
688 self->index += 2;
Fred Drake62c1e3c2001-12-04 21:40:53 +0000689 return 0;
Fred Drake8c081a12001-10-12 20:57:55 +0000690}
691
Fred Drake62c1e3c2001-12-04 21:40:53 +0000692static int
Fred Drake30d1c752001-10-15 22:11:02 +0000693pack_frame_times(ProfilerObject *self)
694{
Fred Drake62c1e3c2001-12-04 21:40:53 +0000695 if (2 + self->index >= BUFFERSIZE) {
696 if (flush_data(self) < 0)
697 return -1;
698 }
Fred Drake30d1c752001-10-15 22:11:02 +0000699 self->buffer[self->index] = WHAT_FRAME_TIMES;
700 self->buffer[self->index + 1] = self->frametimings ? 1 : 0;
701 self->index += 2;
Fred Drake62c1e3c2001-12-04 21:40:53 +0000702 return 0;
Fred Drake30d1c752001-10-15 22:11:02 +0000703}
704
Fred Drake62c1e3c2001-12-04 21:40:53 +0000705static inline int
Fred Drake8c081a12001-10-12 20:57:55 +0000706pack_enter(ProfilerObject *self, int fileno, int tdelta, int lineno)
707{
Fred Drake62c1e3c2001-12-04 21:40:53 +0000708 if (MPISIZE + PISIZE*2 + self->index >= BUFFERSIZE) {
709 if (flush_data(self) < 0)
710 return -1;
711 }
Fred Drake8c081a12001-10-12 20:57:55 +0000712 pack_modified_packed_int(self, fileno, 2, WHAT_ENTER);
Fred Drake8c081a12001-10-12 20:57:55 +0000713 pack_packed_int(self, lineno);
Fred Drake30d1c752001-10-15 22:11:02 +0000714 if (self->frametimings)
Fred Drake62c1e3c2001-12-04 21:40:53 +0000715 return pack_packed_int(self, tdelta);
716 else
717 return 0;
Fred Drake8c081a12001-10-12 20:57:55 +0000718}
719
Fred Drake62c1e3c2001-12-04 21:40:53 +0000720static inline int
Fred Drake8c081a12001-10-12 20:57:55 +0000721pack_exit(ProfilerObject *self, int tdelta)
722{
Fred Drake62c1e3c2001-12-04 21:40:53 +0000723 if (MPISIZE + self->index >= BUFFERSIZE) {
724 if (flush_data(self) < 0)
725 return -1;
Fred Drake30d1c752001-10-15 22:11:02 +0000726 }
Fred Drake62c1e3c2001-12-04 21:40:53 +0000727 if (self->frametimings)
728 return pack_modified_packed_int(self, tdelta, 2, WHAT_EXIT);
729 self->buffer[self->index] = WHAT_EXIT;
730 self->index++;
731 return 0;
Fred Drake8c081a12001-10-12 20:57:55 +0000732}
733
Fred Drake62c1e3c2001-12-04 21:40:53 +0000734static inline int
Fred Drake8c081a12001-10-12 20:57:55 +0000735pack_lineno(ProfilerObject *self, int lineno)
736{
Fred Drake62c1e3c2001-12-04 21:40:53 +0000737 if (MPISIZE + self->index >= BUFFERSIZE) {
738 if (flush_data(self) < 0)
739 return -1;
740 }
741 return pack_modified_packed_int(self, lineno, 2, WHAT_LINENO);
Fred Drake8c081a12001-10-12 20:57:55 +0000742}
743
Fred Drake62c1e3c2001-12-04 21:40:53 +0000744static inline int
Fred Drake8c081a12001-10-12 20:57:55 +0000745pack_lineno_tdelta(ProfilerObject *self, int lineno, int tdelta)
746{
Fred Drake62c1e3c2001-12-04 21:40:53 +0000747 if (MPISIZE + PISIZE + self->index >= BUFFERSIZE) {
748 if (flush_data(self) < 0)
749 return 0;
750 }
751 if (pack_modified_packed_int(self, lineno, 2, WHAT_LINENO) < 0)
752 return -1;
753 return pack_packed_int(self, tdelta);
Fred Drake8c081a12001-10-12 20:57:55 +0000754}
755
756static inline int
757get_fileno(ProfilerObject *self, PyCodeObject *fcode)
758{
Fred Drake30d1c752001-10-15 22:11:02 +0000759 /* This is only used for ENTER events. */
760
761 PyObject *obj;
762 PyObject *dict;
Fred Drake8c081a12001-10-12 20:57:55 +0000763 int fileno;
764
Fred Drake30d1c752001-10-15 22:11:02 +0000765 obj = PyDict_GetItem(self->filemap, fcode->co_filename);
766 if (obj == NULL) {
Fred Drake8c081a12001-10-12 20:57:55 +0000767 /* first sighting of this file */
Fred Drake30d1c752001-10-15 22:11:02 +0000768 dict = PyDict_New();
769 if (dict == NULL) {
Fred Drake8c081a12001-10-12 20:57:55 +0000770 return -1;
771 }
Fred Drake30d1c752001-10-15 22:11:02 +0000772 fileno = self->next_fileno;
773 obj = Py_BuildValue("iN", fileno, dict);
774 if (obj == NULL) {
775 return -1;
776 }
777 if (PyDict_SetItem(self->filemap, fcode->co_filename, obj)) {
778 Py_DECREF(obj);
Fred Drake8c081a12001-10-12 20:57:55 +0000779 return -1;
780 }
781 self->next_fileno++;
Fred Drake30d1c752001-10-15 22:11:02 +0000782 Py_DECREF(obj);
Fred Drake62c1e3c2001-12-04 21:40:53 +0000783 if (pack_define_file(self, fileno,
784 PyString_AS_STRING(fcode->co_filename)) < 0)
785 return -1;
Fred Drake8c081a12001-10-12 20:57:55 +0000786 }
787 else {
788 /* already know this ID */
Fred Drake30d1c752001-10-15 22:11:02 +0000789 fileno = PyInt_AS_LONG(PyTuple_GET_ITEM(obj, 0));
790 dict = PyTuple_GET_ITEM(obj, 1);
791 }
792 /* make sure we save a function name for this (fileno, lineno) */
793 obj = PyInt_FromLong(fcode->co_firstlineno);
794 if (obj == NULL) {
795 /* We just won't have it saved; too bad. */
796 PyErr_Clear();
797 }
798 else {
799 PyObject *name = PyDict_GetItem(dict, obj);
800 if (name == NULL) {
Fred Drake62c1e3c2001-12-04 21:40:53 +0000801 if (pack_define_func(self, fileno, fcode->co_firstlineno,
802 PyString_AS_STRING(fcode->co_name)) < 0)
803 return -1;
Fred Drake30d1c752001-10-15 22:11:02 +0000804 if (PyDict_SetItem(dict, obj, fcode->co_name))
805 return -1;
806 }
Fred Drake8c081a12001-10-12 20:57:55 +0000807 }
808 return fileno;
809}
810
811static inline int
812get_tdelta(ProfilerObject *self)
813{
814 int tdelta;
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000815#ifdef MS_WINDOWS
Fred Drake8c081a12001-10-12 20:57:55 +0000816 hs_time tv;
Tim Peters7d99ff22001-10-13 07:37:52 +0000817 hs_time diff;
Fred Drake8c081a12001-10-12 20:57:55 +0000818
Tim Peters7d99ff22001-10-13 07:37:52 +0000819 GETTIMEOFDAY(&tv);
820 diff = tv - self->prev_timeofday;
821 tdelta = (int)diff;
Fred Drake8c081a12001-10-12 20:57:55 +0000822#else
823 struct timeval tv;
824
825 GETTIMEOFDAY(&tv);
826
827 if (tv.tv_sec == self->prev_timeofday.tv_sec)
828 tdelta = tv.tv_usec - self->prev_timeofday.tv_usec;
829 else
830 tdelta = ((tv.tv_sec - self->prev_timeofday.tv_sec) * 1000000
831 + tv.tv_usec);
832#endif
833 self->prev_timeofday = tv;
834 return tdelta;
835}
836
837
838/* The workhorse: the profiler callback function. */
839
840static int
841profiler_callback(ProfilerObject *self, PyFrameObject *frame, int what,
842 PyObject *arg)
843{
Fred Drake30d1c752001-10-15 22:11:02 +0000844 int tdelta = -1;
Fred Drake8c081a12001-10-12 20:57:55 +0000845 int fileno;
846
Fred Drake30d1c752001-10-15 22:11:02 +0000847 if (self->frametimings)
848 tdelta = get_tdelta(self);
Fred Drake8c081a12001-10-12 20:57:55 +0000849 switch (what) {
850 case PyTrace_CALL:
851 fileno = get_fileno(self, frame->f_code);
852 if (fileno < 0)
853 return -1;
Fred Drake62c1e3c2001-12-04 21:40:53 +0000854 if (pack_enter(self, fileno, tdelta,
855 frame->f_code->co_firstlineno) < 0)
856 return -1;
Fred Drake8c081a12001-10-12 20:57:55 +0000857 break;
858 case PyTrace_RETURN:
Fred Drake62c1e3c2001-12-04 21:40:53 +0000859 if (pack_exit(self, tdelta) < 0)
860 return -1;
Fred Drake8c081a12001-10-12 20:57:55 +0000861 break;
862 default:
863 /* should never get here */
864 break;
865 }
866 return 0;
867}
868
869
870/* Alternate callback when we want PyTrace_LINE events */
871
872static int
873tracer_callback(ProfilerObject *self, PyFrameObject *frame, int what,
874 PyObject *arg)
875{
876 int fileno;
877
Fred Drake8c081a12001-10-12 20:57:55 +0000878 switch (what) {
879 case PyTrace_CALL:
880 fileno = get_fileno(self, frame->f_code);
881 if (fileno < 0)
882 return -1;
Fred Drake62c1e3c2001-12-04 21:40:53 +0000883 return pack_enter(self, fileno,
884 self->frametimings ? get_tdelta(self) : -1,
885 frame->f_code->co_firstlineno);
886
Fred Drake8c081a12001-10-12 20:57:55 +0000887 case PyTrace_RETURN:
Fred Drake62c1e3c2001-12-04 21:40:53 +0000888 return pack_exit(self, get_tdelta(self));
889
Tim Peters1566a172001-10-12 22:08:39 +0000890 case PyTrace_LINE:
Fred Drake8c081a12001-10-12 20:57:55 +0000891 if (self->linetimings)
Michael W. Hudson806d1c82002-09-11 17:09:45 +0000892 return pack_lineno_tdelta(self, frame->f_lineno,
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000893 get_tdelta(self));
Fred Drake8c081a12001-10-12 20:57:55 +0000894 else
Michael W. Hudson02ff6a92002-09-11 15:36:32 +0000895 return pack_lineno(self, frame->f_lineno);
Fred Drake62c1e3c2001-12-04 21:40:53 +0000896
Fred Drake8c081a12001-10-12 20:57:55 +0000897 default:
898 /* ignore PyTrace_EXCEPTION */
899 break;
900 }
901 return 0;
902}
903
904
905/* A couple of useful helper functions. */
906
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000907#ifdef MS_WINDOWS
Tim Petersfeab23f2001-10-13 00:11:10 +0000908static LARGE_INTEGER frequency = {0, 0};
Fred Drake8c081a12001-10-12 20:57:55 +0000909#endif
910
911static unsigned long timeofday_diff = 0;
912static unsigned long rusage_diff = 0;
913
914static void
915calibrate(void)
916{
917 hs_time tv1, tv2;
918
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000919#ifdef MS_WINDOWS
Tim Peters7d99ff22001-10-13 07:37:52 +0000920 hs_time diff;
Fred Drake8c081a12001-10-12 20:57:55 +0000921 QueryPerformanceFrequency(&frequency);
922#endif
923
924 GETTIMEOFDAY(&tv1);
925 while (1) {
926 GETTIMEOFDAY(&tv2);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000927#ifdef MS_WINDOWS
Tim Peters7d99ff22001-10-13 07:37:52 +0000928 diff = tv2 - tv1;
929 if (diff != 0) {
930 timeofday_diff = (unsigned long)diff;
Fred Drake8c081a12001-10-12 20:57:55 +0000931 break;
932 }
933#else
934 if (tv1.tv_sec != tv2.tv_sec || tv1.tv_usec != tv2.tv_usec) {
935 if (tv1.tv_sec == tv2.tv_sec)
936 timeofday_diff = tv2.tv_usec - tv1.tv_usec;
937 else
938 timeofday_diff = (1000000 - tv1.tv_usec) + tv2.tv_usec;
939 break;
940 }
941#endif
942 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000943#if defined(MS_WINDOWS) || defined(macintosh) || defined(PYOS_OS2) || \
944 defined(__VMS)
Fred Drake8c081a12001-10-12 20:57:55 +0000945 rusage_diff = -1;
946#else
947 {
948 struct rusage ru1, ru2;
949
950 getrusage(RUSAGE_SELF, &ru1);
951 while (1) {
952 getrusage(RUSAGE_SELF, &ru2);
953 if (ru1.ru_utime.tv_sec != ru2.ru_utime.tv_sec) {
954 rusage_diff = ((1000000 - ru1.ru_utime.tv_usec)
955 + ru2.ru_utime.tv_usec);
956 break;
957 }
958 else if (ru1.ru_utime.tv_usec != ru2.ru_utime.tv_usec) {
959 rusage_diff = ru2.ru_utime.tv_usec - ru1.ru_utime.tv_usec;
960 break;
961 }
962 else if (ru1.ru_stime.tv_sec != ru2.ru_stime.tv_sec) {
963 rusage_diff = ((1000000 - ru1.ru_stime.tv_usec)
964 + ru2.ru_stime.tv_usec);
965 break;
966 }
967 else if (ru1.ru_stime.tv_usec != ru2.ru_stime.tv_usec) {
968 rusage_diff = ru2.ru_stime.tv_usec - ru1.ru_stime.tv_usec;
969 break;
970 }
971 }
972 }
973#endif
974}
975
976static void
977do_start(ProfilerObject *self)
978{
979 self->active = 1;
980 GETTIMEOFDAY(&self->prev_timeofday);
981 if (self->lineevents)
982 PyEval_SetTrace((Py_tracefunc) tracer_callback, (PyObject *)self);
983 else
984 PyEval_SetProfile((Py_tracefunc) profiler_callback, (PyObject *)self);
985}
986
987static void
988do_stop(ProfilerObject *self)
989{
990 if (self->active) {
991 self->active = 0;
992 if (self->lineevents)
993 PyEval_SetTrace(NULL, NULL);
994 else
995 PyEval_SetProfile(NULL, NULL);
996 }
997 if (self->index > 0) {
998 /* Best effort to dump out any remaining data. */
999 flush_data(self);
1000 }
1001}
1002
1003static int
1004is_available(ProfilerObject *self)
1005{
1006 if (self->active) {
1007 PyErr_SetString(ProfilerError, "profiler already active");
1008 return 0;
1009 }
1010 if (self->logfp == NULL) {
1011 PyErr_SetString(ProfilerError, "profiler already closed");
1012 return 0;
1013 }
1014 return 1;
1015}
1016
1017
1018/* Profiler object interface methods. */
1019
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001020PyDoc_STRVAR(addinfo__doc__,
Fred Drake4c2e1af2001-10-29 20:45:57 +00001021"addinfo(key, value)\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001022"Insert an ADD_INFO record into the log.");
Fred Drake4c2e1af2001-10-29 20:45:57 +00001023
1024static PyObject *
1025profiler_addinfo(ProfilerObject *self, PyObject *args)
1026{
1027 PyObject *result = NULL;
1028 char *key, *value;
1029
1030 if (PyArg_ParseTuple(args, "ss:addinfo", &key, &value)) {
1031 if (self->logfp == NULL)
1032 PyErr_SetString(ProfilerError, "profiler already closed");
1033 else {
Fred Drake62c1e3c2001-12-04 21:40:53 +00001034 if (pack_add_info(self, key, value) == 0) {
1035 result = Py_None;
1036 Py_INCREF(result);
1037 }
Fred Drake4c2e1af2001-10-29 20:45:57 +00001038 }
1039 }
1040 return result;
1041}
1042
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001043PyDoc_STRVAR(close__doc__,
Tim Petersfeab23f2001-10-13 00:11:10 +00001044"close()\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001045"Shut down this profiler and close the log files, even if its active.");
Fred Drake8c081a12001-10-12 20:57:55 +00001046
1047static PyObject *
Fred Drake666bf522002-07-18 19:11:44 +00001048profiler_close(ProfilerObject *self)
Fred Drake8c081a12001-10-12 20:57:55 +00001049{
Fred Drake666bf522002-07-18 19:11:44 +00001050 do_stop(self);
1051 if (self->logfp != NULL) {
1052 fclose(self->logfp);
1053 self->logfp = NULL;
Fred Drake8c081a12001-10-12 20:57:55 +00001054 }
Fred Drake666bf522002-07-18 19:11:44 +00001055 Py_INCREF(Py_None);
1056 return Py_None;
1057}
1058
1059#define fileno__doc__ logreader_fileno__doc__
1060
1061static PyObject *
1062profiler_fileno(ProfilerObject *self)
1063{
1064 if (self->logfp == NULL) {
1065 PyErr_SetString(PyExc_ValueError,
1066 "profiler's file object already closed");
1067 return NULL;
1068 }
1069 return PyInt_FromLong(fileno(self->logfp));
Fred Drake8c081a12001-10-12 20:57:55 +00001070}
1071
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001072PyDoc_STRVAR(runcall__doc__,
Tim Petersfeab23f2001-10-13 00:11:10 +00001073"runcall(callable[, args[, kw]]) -> callable()\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001074"Profile a specific function call, returning the result of that call.");
Fred Drake8c081a12001-10-12 20:57:55 +00001075
1076static PyObject *
1077profiler_runcall(ProfilerObject *self, PyObject *args)
1078{
1079 PyObject *result = NULL;
1080 PyObject *callargs = NULL;
1081 PyObject *callkw = NULL;
1082 PyObject *callable;
1083
1084 if (PyArg_ParseTuple(args, "O|OO:runcall",
1085 &callable, &callargs, &callkw)) {
1086 if (is_available(self)) {
1087 do_start(self);
1088 result = PyEval_CallObjectWithKeywords(callable, callargs, callkw);
1089 do_stop(self);
1090 }
1091 }
1092 return result;
1093}
1094
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001095PyDoc_STRVAR(runcode__doc__,
Tim Petersfeab23f2001-10-13 00:11:10 +00001096"runcode(code, globals[, locals])\n"
1097"Execute a code object while collecting profile data. If locals is\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001098"omitted, globals is used for the locals as well.");
Fred Drake8c081a12001-10-12 20:57:55 +00001099
1100static PyObject *
1101profiler_runcode(ProfilerObject *self, PyObject *args)
1102{
1103 PyObject *result = NULL;
1104 PyCodeObject *code;
1105 PyObject *globals;
1106 PyObject *locals = NULL;
1107
1108 if (PyArg_ParseTuple(args, "O!O!|O:runcode",
1109 &PyCode_Type, &code,
1110 &PyDict_Type, &globals,
1111 &locals)) {
1112 if (is_available(self)) {
1113 if (locals == NULL || locals == Py_None)
1114 locals = globals;
1115 else if (!PyDict_Check(locals)) {
1116 PyErr_SetString(PyExc_TypeError,
1117 "locals must be a dictionary or None");
1118 return NULL;
1119 }
1120 do_start(self);
1121 result = PyEval_EvalCode(code, globals, locals);
1122 do_stop(self);
1123#if 0
1124 if (!PyErr_Occurred()) {
1125 result = Py_None;
1126 Py_INCREF(result);
1127 }
1128#endif
1129 }
1130 }
1131 return result;
1132}
1133
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001134PyDoc_STRVAR(start__doc__,
Tim Petersfeab23f2001-10-13 00:11:10 +00001135"start()\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001136"Install this profiler for the current thread.");
Fred Drake8c081a12001-10-12 20:57:55 +00001137
1138static PyObject *
1139profiler_start(ProfilerObject *self, PyObject *args)
1140{
1141 PyObject *result = NULL;
1142
Fred Drake666bf522002-07-18 19:11:44 +00001143 if (is_available(self)) {
1144 do_start(self);
1145 result = Py_None;
1146 Py_INCREF(result);
Fred Drake8c081a12001-10-12 20:57:55 +00001147 }
1148 return result;
1149}
1150
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001151PyDoc_STRVAR(stop__doc__,
Tim Petersfeab23f2001-10-13 00:11:10 +00001152"stop()\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001153"Remove this profiler from the current thread.");
Fred Drake8c081a12001-10-12 20:57:55 +00001154
1155static PyObject *
1156profiler_stop(ProfilerObject *self, PyObject *args)
1157{
1158 PyObject *result = NULL;
1159
Fred Drake666bf522002-07-18 19:11:44 +00001160 if (!self->active)
1161 PyErr_SetString(ProfilerError, "profiler not active");
1162 else {
1163 do_stop(self);
1164 result = Py_None;
1165 Py_INCREF(result);
Fred Drake8c081a12001-10-12 20:57:55 +00001166 }
1167 return result;
1168}
1169
1170
1171/* Python API support. */
1172
1173static void
1174profiler_dealloc(ProfilerObject *self)
1175{
1176 do_stop(self);
1177 if (self->logfp != NULL)
1178 fclose(self->logfp);
1179 Py_XDECREF(self->filemap);
1180 Py_XDECREF(self->logfilename);
1181 PyObject_Del((PyObject *)self);
1182}
1183
Fred Drake8c081a12001-10-12 20:57:55 +00001184static PyMethodDef profiler_methods[] = {
Fred Drake4c2e1af2001-10-29 20:45:57 +00001185 {"addinfo", (PyCFunction)profiler_addinfo, METH_VARARGS, addinfo__doc__},
Fred Drake666bf522002-07-18 19:11:44 +00001186 {"close", (PyCFunction)profiler_close, METH_NOARGS, close__doc__},
1187 {"fileno", (PyCFunction)profiler_fileno, METH_NOARGS, fileno__doc__},
Fred Drake8c081a12001-10-12 20:57:55 +00001188 {"runcall", (PyCFunction)profiler_runcall, METH_VARARGS, runcall__doc__},
1189 {"runcode", (PyCFunction)profiler_runcode, METH_VARARGS, runcode__doc__},
Fred Drake666bf522002-07-18 19:11:44 +00001190 {"start", (PyCFunction)profiler_start, METH_NOARGS, start__doc__},
1191 {"stop", (PyCFunction)profiler_stop, METH_NOARGS, stop__doc__},
Fred Drake8c081a12001-10-12 20:57:55 +00001192 {NULL, NULL}
1193};
1194
Guido van Rossum9cb64b92002-07-17 16:15:35 +00001195static PyMemberDef profiler_members[] = {
Fred Drake30d1c752001-10-15 22:11:02 +00001196 {"frametimings", T_LONG, offsetof(ProfilerObject, linetimings), READONLY},
1197 {"lineevents", T_LONG, offsetof(ProfilerObject, lineevents), READONLY},
1198 {"linetimings", T_LONG, offsetof(ProfilerObject, linetimings), READONLY},
Fred Drake8c081a12001-10-12 20:57:55 +00001199 {NULL}
1200};
1201
1202static PyObject *
Guido van Rossum9cb64b92002-07-17 16:15:35 +00001203profiler_get_closed(ProfilerObject *self, void *closure)
Fred Drake8c081a12001-10-12 20:57:55 +00001204{
Guido van Rossum9cb64b92002-07-17 16:15:35 +00001205 PyObject *result = (self->logfp == NULL) ? Py_True : Py_False;
1206 Py_INCREF(result);
Fred Drake8c081a12001-10-12 20:57:55 +00001207 return result;
1208}
1209
Guido van Rossum9cb64b92002-07-17 16:15:35 +00001210static PyGetSetDef profiler_getsets[] = {
Fred Draked1eb8b62002-07-17 18:54:20 +00001211 {"closed", (getter)profiler_get_closed, NULL,
Fred Drake5c3ed3d2002-07-17 19:38:05 +00001212 PyDoc_STR("True if the profiler's output file has already been closed.")},
Guido van Rossum9cb64b92002-07-17 16:15:35 +00001213 {NULL}
1214};
1215
Fred Drake8c081a12001-10-12 20:57:55 +00001216
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001217PyDoc_STRVAR(profiler_object__doc__,
Tim Petersfeab23f2001-10-13 00:11:10 +00001218"High-performance profiler object.\n"
1219"\n"
1220"Methods:\n"
1221"\n"
Fred Drake30d1c752001-10-15 22:11:02 +00001222"close(): Stop the profiler and close the log files.\n"
Fred Drake666bf522002-07-18 19:11:44 +00001223"fileno(): Returns the file descriptor of the log file.\n"
Fred Drake30d1c752001-10-15 22:11:02 +00001224"runcall(): Run a single function call with profiling enabled.\n"
1225"runcode(): Execute a code object with profiling enabled.\n"
1226"start(): Install the profiler and return.\n"
1227"stop(): Remove the profiler.\n"
Tim Petersfeab23f2001-10-13 00:11:10 +00001228"\n"
1229"Attributes (read-only):\n"
1230"\n"
Fred Drake30d1c752001-10-15 22:11:02 +00001231"closed: True if the profiler has already been closed.\n"
1232"frametimings: True if ENTER/EXIT events collect timing information.\n"
Michael W. Hudsondd32a912002-08-15 14:59:02 +00001233"lineevents: True if line events are reported to the profiler.\n"
1234"linetimings: True if line events collect timing information.");
Fred Drake8c081a12001-10-12 20:57:55 +00001235
1236static PyTypeObject ProfilerType = {
1237 PyObject_HEAD_INIT(NULL)
1238 0, /* ob_size */
Fred Drake4c2e1af2001-10-29 20:45:57 +00001239 "_hotshot.ProfilerType", /* tp_name */
Fred Drake8c081a12001-10-12 20:57:55 +00001240 (int) sizeof(ProfilerObject), /* tp_basicsize */
1241 0, /* tp_itemsize */
1242 (destructor)profiler_dealloc, /* tp_dealloc */
1243 0, /* tp_print */
Guido van Rossum9cb64b92002-07-17 16:15:35 +00001244 0, /* tp_getattr */
Fred Drake8c081a12001-10-12 20:57:55 +00001245 0, /* tp_setattr */
1246 0, /* tp_compare */
1247 0, /* tp_repr */
1248 0, /* tp_as_number */
1249 0, /* tp_as_sequence */
1250 0, /* tp_as_mapping */
1251 0, /* tp_hash */
1252 0, /* tp_call */
1253 0, /* tp_str */
Jason Tishlerfb8595d2003-01-06 12:41:26 +00001254 PyObject_GenericGetAttr, /* tp_getattro */
Fred Drake8c081a12001-10-12 20:57:55 +00001255 0, /* tp_setattro */
1256 0, /* tp_as_buffer */
Fred Drake4c2e1af2001-10-29 20:45:57 +00001257 Py_TPFLAGS_DEFAULT, /* tp_flags */
Fred Drake8c081a12001-10-12 20:57:55 +00001258 profiler_object__doc__, /* tp_doc */
Guido van Rossum9cb64b92002-07-17 16:15:35 +00001259 0, /* tp_traverse */
1260 0, /* tp_clear */
1261 0, /* tp_richcompare */
1262 0, /* tp_weaklistoffset */
1263 0, /* tp_iter */
1264 0, /* tp_iternext */
1265 profiler_methods, /* tp_methods */
1266 profiler_members, /* tp_members */
1267 profiler_getsets, /* tp_getset */
1268 0, /* tp_base */
1269 0, /* tp_dict */
1270 0, /* tp_descr_get */
1271 0, /* tp_descr_set */
Fred Drake8c081a12001-10-12 20:57:55 +00001272};
1273
1274
1275static PyMethodDef logreader_methods[] = {
Fred Drake666bf522002-07-18 19:11:44 +00001276 {"close", (PyCFunction)logreader_close, METH_NOARGS,
Fred Drake8c081a12001-10-12 20:57:55 +00001277 logreader_close__doc__},
Fred Drake666bf522002-07-18 19:11:44 +00001278 {"fileno", (PyCFunction)logreader_fileno, METH_NOARGS,
1279 logreader_fileno__doc__},
Fred Drake8c081a12001-10-12 20:57:55 +00001280 {NULL, NULL}
1281};
1282
Guido van Rossum9cb64b92002-07-17 16:15:35 +00001283static PyMemberDef logreader_members[] = {
Fred Drake5c3ed3d2002-07-17 19:38:05 +00001284 {"info", T_OBJECT, offsetof(LogReaderObject, info), RO,
1285 PyDoc_STR("Dictionary mapping informational keys to lists of values.")},
Guido van Rossum9cb64b92002-07-17 16:15:35 +00001286 {NULL}
1287};
Fred Drake8c081a12001-10-12 20:57:55 +00001288
1289
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001290PyDoc_STRVAR(logreader__doc__,
1291"logreader(filename) --> log-iterator\n\
1292Create a log-reader for the timing information file.");
Fred Drake8c081a12001-10-12 20:57:55 +00001293
1294static PySequenceMethods logreader_as_sequence = {
1295 0, /* sq_length */
1296 0, /* sq_concat */
1297 0, /* sq_repeat */
1298 (intargfunc)logreader_sq_item, /* sq_item */
1299 0, /* sq_slice */
1300 0, /* sq_ass_item */
1301 0, /* sq_ass_slice */
1302 0, /* sq_contains */
1303 0, /* sq_inplace_concat */
1304 0, /* sq_inplace_repeat */
1305};
1306
Fred Drake666bf522002-07-18 19:11:44 +00001307static PyObject *
1308logreader_get_closed(LogReaderObject *self, void *closure)
1309{
1310 PyObject *result = (self->logfp == NULL) ? Py_True : Py_False;
1311 Py_INCREF(result);
1312 return result;
1313}
1314
1315static PyGetSetDef logreader_getsets[] = {
1316 {"closed", (getter)logreader_get_closed, NULL,
1317 PyDoc_STR("True if the logreader's input file has already been closed.")},
1318 {NULL}
1319};
1320
Fred Drake8c081a12001-10-12 20:57:55 +00001321static PyTypeObject LogReaderType = {
1322 PyObject_HEAD_INIT(NULL)
1323 0, /* ob_size */
Fred Drake4c2e1af2001-10-29 20:45:57 +00001324 "_hotshot.LogReaderType", /* tp_name */
Fred Drake8c081a12001-10-12 20:57:55 +00001325 (int) sizeof(LogReaderObject), /* tp_basicsize */
1326 0, /* tp_itemsize */
1327 (destructor)logreader_dealloc, /* tp_dealloc */
1328 0, /* tp_print */
Guido van Rossum9cb64b92002-07-17 16:15:35 +00001329 0, /* tp_getattr */
Fred Drake8c081a12001-10-12 20:57:55 +00001330 0, /* tp_setattr */
1331 0, /* tp_compare */
1332 0, /* tp_repr */
1333 0, /* tp_as_number */
1334 &logreader_as_sequence, /* tp_as_sequence */
1335 0, /* tp_as_mapping */
1336 0, /* tp_hash */
1337 0, /* tp_call */
1338 0, /* tp_str */
Jason Tishlerfb8595d2003-01-06 12:41:26 +00001339 PyObject_GenericGetAttr, /* tp_getattro */
Fred Drake8c081a12001-10-12 20:57:55 +00001340 0, /* tp_setattro */
1341 0, /* tp_as_buffer */
Fred Drake4c2e1af2001-10-29 20:45:57 +00001342 Py_TPFLAGS_DEFAULT, /* tp_flags */
Fred Drake8c081a12001-10-12 20:57:55 +00001343 logreader__doc__, /* tp_doc */
Fred Drake8c081a12001-10-12 20:57:55 +00001344 0, /* tp_traverse */
1345 0, /* tp_clear */
1346 0, /* tp_richcompare */
1347 0, /* tp_weaklistoffset */
Raymond Hettinger1da1dbf2003-03-17 19:46:11 +00001348 PyObject_SelfIter, /* tp_iter */
Fred Drake8c081a12001-10-12 20:57:55 +00001349 (iternextfunc)logreader_tp_iternext,/* tp_iternext */
Guido van Rossum9cb64b92002-07-17 16:15:35 +00001350 logreader_methods, /* tp_methods */
1351 logreader_members, /* tp_members */
Fred Drake666bf522002-07-18 19:11:44 +00001352 logreader_getsets, /* tp_getset */
Guido van Rossum9cb64b92002-07-17 16:15:35 +00001353 0, /* tp_base */
1354 0, /* tp_dict */
1355 0, /* tp_descr_get */
1356 0, /* tp_descr_set */
Fred Drake8c081a12001-10-12 20:57:55 +00001357};
1358
1359static PyObject *
1360hotshot_logreader(PyObject *unused, PyObject *args)
1361{
1362 LogReaderObject *self = NULL;
1363 char *filename;
Neil Schemenauer8b6b4912002-05-29 18:19:14 +00001364 int c;
1365 int err = 0;
Fred Drake8c081a12001-10-12 20:57:55 +00001366
1367 if (PyArg_ParseTuple(args, "s:logreader", &filename)) {
1368 self = PyObject_New(LogReaderObject, &LogReaderType);
1369 if (self != NULL) {
Fred Drake30d1c752001-10-15 22:11:02 +00001370 self->frametimings = 1;
Fred Drake8c081a12001-10-12 20:57:55 +00001371 self->linetimings = 0;
Fred Drake4c2e1af2001-10-29 20:45:57 +00001372 self->info = NULL;
Fred Drake8c081a12001-10-12 20:57:55 +00001373 self->logfp = fopen(filename, "rb");
1374 if (self->logfp == NULL) {
1375 PyErr_SetFromErrnoWithFilename(PyExc_IOError, filename);
1376 Py_DECREF(self);
1377 self = NULL;
Fred Drake4c2e1af2001-10-29 20:45:57 +00001378 goto finally;
1379 }
1380 self->info = PyDict_New();
1381 if (self->info == NULL) {
1382 Py_DECREF(self);
1383 goto finally;
1384 }
Neil Schemenauer8b6b4912002-05-29 18:19:14 +00001385 /* read initial info */
1386 for (;;) {
1387 if ((c = fgetc(self->logfp)) == EOF) {
Fred Drake666bf522002-07-18 19:11:44 +00001388 eof_error(self);
Neil Schemenauer8b6b4912002-05-29 18:19:14 +00001389 break;
1390 }
1391 if (c != WHAT_ADD_INFO) {
1392 ungetc(c, self->logfp);
1393 break;
1394 }
1395 err = unpack_add_info(self);
Fred Drake4c2e1af2001-10-29 20:45:57 +00001396 if (err) {
1397 if (err == ERR_EOF)
Fred Drake666bf522002-07-18 19:11:44 +00001398 eof_error(self);
Fred Drake4c2e1af2001-10-29 20:45:57 +00001399 else
1400 PyErr_SetString(PyExc_RuntimeError,
1401 "unexpected error");
1402 break;
1403 }
Fred Drake8c081a12001-10-12 20:57:55 +00001404 }
1405 }
1406 }
Fred Drake4c2e1af2001-10-29 20:45:57 +00001407 finally:
Fred Drake8c081a12001-10-12 20:57:55 +00001408 return (PyObject *) self;
1409}
1410
1411
1412/* Return a Python string that represents the version number without the
1413 * extra cruft added by revision control, even if the right options were
1414 * given to the "cvs export" command to make it not include the extra
1415 * cruft.
1416 */
1417static char *
1418get_version_string(void)
1419{
1420 static char *rcsid = "$Revision$";
1421 char *rev = rcsid;
1422 char *buffer;
1423 int i = 0;
1424
Neal Norwitz3afb2d22002-03-20 21:32:07 +00001425 while (*rev && !isdigit((int)*rev))
Fred Drake8c081a12001-10-12 20:57:55 +00001426 ++rev;
1427 while (rev[i] != ' ' && rev[i] != '\0')
1428 ++i;
1429 buffer = malloc(i + 1);
1430 if (buffer != NULL) {
1431 memmove(buffer, rev, i);
1432 buffer[i] = '\0';
1433 }
1434 return buffer;
1435}
1436
1437/* Write out a RFC 822-style header with various useful bits of
1438 * information to make the output easier to manage.
1439 */
1440static int
1441write_header(ProfilerObject *self)
1442{
1443 char *buffer;
1444 char cwdbuffer[PATH_MAX];
1445 PyObject *temp;
1446 int i, len;
1447
1448 buffer = get_version_string();
1449 if (buffer == NULL) {
1450 PyErr_NoMemory();
1451 return -1;
1452 }
Fred Drake4c2e1af2001-10-29 20:45:57 +00001453 pack_add_info(self, "hotshot-version", buffer);
1454 pack_add_info(self, "requested-frame-timings",
1455 (self->frametimings ? "yes" : "no"));
1456 pack_add_info(self, "requested-line-events",
Fred Drake8c081a12001-10-12 20:57:55 +00001457 (self->lineevents ? "yes" : "no"));
Fred Drake4c2e1af2001-10-29 20:45:57 +00001458 pack_add_info(self, "requested-line-timings",
1459 (self->linetimings ? "yes" : "no"));
1460 pack_add_info(self, "platform", Py_GetPlatform());
1461 pack_add_info(self, "executable", Py_GetProgramFullPath());
Fred Drakef12a68c2001-11-09 15:59:36 +00001462 free(buffer);
Fred Drake8c081a12001-10-12 20:57:55 +00001463 buffer = (char *) Py_GetVersion();
1464 if (buffer == NULL)
1465 PyErr_Clear();
1466 else
Fred Drake4c2e1af2001-10-29 20:45:57 +00001467 pack_add_info(self, "executable-version", buffer);
Fred Drake8c081a12001-10-12 20:57:55 +00001468
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001469#ifdef MS_WINDOWS
Tim Peters885d4572001-11-28 20:27:42 +00001470 PyOS_snprintf(cwdbuffer, sizeof(cwdbuffer), "%I64d", frequency.QuadPart);
Fred Drake4c2e1af2001-10-29 20:45:57 +00001471 pack_add_info(self, "reported-performance-frequency", cwdbuffer);
Fred Drake8c081a12001-10-12 20:57:55 +00001472#else
Tim Peters885d4572001-11-28 20:27:42 +00001473 PyOS_snprintf(cwdbuffer, sizeof(cwdbuffer), "%lu", rusage_diff);
Fred Drake4c2e1af2001-10-29 20:45:57 +00001474 pack_add_info(self, "observed-interval-getrusage", cwdbuffer);
Tim Peters885d4572001-11-28 20:27:42 +00001475 PyOS_snprintf(cwdbuffer, sizeof(cwdbuffer), "%lu", timeofday_diff);
Fred Drake4c2e1af2001-10-29 20:45:57 +00001476 pack_add_info(self, "observed-interval-gettimeofday", cwdbuffer);
Fred Drake8c081a12001-10-12 20:57:55 +00001477#endif
Fred Drake8c081a12001-10-12 20:57:55 +00001478
Fred Drake4c2e1af2001-10-29 20:45:57 +00001479 pack_add_info(self, "current-directory",
Fred Drake8c081a12001-10-12 20:57:55 +00001480 getcwd(cwdbuffer, sizeof cwdbuffer));
1481
1482 temp = PySys_GetObject("path");
1483 len = PyList_GET_SIZE(temp);
1484 for (i = 0; i < len; ++i) {
1485 PyObject *item = PyList_GET_ITEM(temp, i);
1486 buffer = PyString_AsString(item);
Fred Draked1eb8b62002-07-17 18:54:20 +00001487 if (buffer == NULL) {
1488 pack_add_info(self, "sys-path-entry", "<non-string-path-entry>");
1489 PyErr_Clear();
1490 }
1491 else {
1492 pack_add_info(self, "sys-path-entry", buffer);
1493 }
Fred Drake8c081a12001-10-12 20:57:55 +00001494 }
Fred Drake4c2e1af2001-10-29 20:45:57 +00001495 pack_frame_times(self);
1496 pack_line_times(self);
1497
Fred Drake8c081a12001-10-12 20:57:55 +00001498 return 0;
1499}
1500
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001501PyDoc_STRVAR(profiler__doc__,
1502"profiler(logfilename[, lineevents[, linetimes]]) -> profiler\n\
1503Create a new profiler object.");
Fred Drake8c081a12001-10-12 20:57:55 +00001504
1505static PyObject *
1506hotshot_profiler(PyObject *unused, PyObject *args)
1507{
1508 char *logfilename;
1509 ProfilerObject *self = NULL;
1510 int lineevents = 0;
1511 int linetimings = 1;
1512
1513 if (PyArg_ParseTuple(args, "s|ii:profiler", &logfilename,
1514 &lineevents, &linetimings)) {
1515 self = PyObject_New(ProfilerObject, &ProfilerType);
1516 if (self == NULL)
1517 return NULL;
Fred Drake30d1c752001-10-15 22:11:02 +00001518 self->frametimings = 1;
Fred Drake8c081a12001-10-12 20:57:55 +00001519 self->lineevents = lineevents ? 1 : 0;
1520 self->linetimings = (lineevents && linetimings) ? 1 : 0;
1521 self->index = 0;
1522 self->active = 0;
1523 self->next_fileno = 0;
Tim Peters1566a172001-10-12 22:08:39 +00001524 self->logfp = NULL;
Fred Drake8c081a12001-10-12 20:57:55 +00001525 self->logfilename = PyTuple_GET_ITEM(args, 0);
1526 Py_INCREF(self->logfilename);
1527 self->filemap = PyDict_New();
1528 if (self->filemap == NULL) {
1529 Py_DECREF(self);
1530 return NULL;
1531 }
1532 self->logfp = fopen(logfilename, "wb");
1533 if (self->logfp == NULL) {
1534 Py_DECREF(self);
1535 PyErr_SetFromErrnoWithFilename(PyExc_IOError, logfilename);
1536 return NULL;
1537 }
1538 if (timeofday_diff == 0) {
1539 /* Run this several times since sometimes the first
1540 * doesn't give the lowest values, and we're really trying
1541 * to determine the lowest.
1542 */
1543 calibrate();
1544 calibrate();
1545 calibrate();
1546 }
1547 if (write_header(self))
1548 /* some error occurred, exception has been set */
1549 self = NULL;
1550 }
1551 return (PyObject *) self;
1552}
1553
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001554PyDoc_STRVAR(coverage__doc__,
1555"coverage(logfilename) -> profiler\n\
Fred Drake30d1c752001-10-15 22:11:02 +00001556Returns a profiler that doesn't collect any timing information, which is\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001557useful in building a coverage analysis tool.");
Fred Drake30d1c752001-10-15 22:11:02 +00001558
1559static PyObject *
1560hotshot_coverage(PyObject *unused, PyObject *args)
1561{
1562 char *logfilename;
1563 PyObject *result = NULL;
1564
1565 if (PyArg_ParseTuple(args, "s:coverage", &logfilename)) {
1566 result = hotshot_profiler(unused, args);
1567 if (result != NULL) {
1568 ProfilerObject *self = (ProfilerObject *) result;
1569 self->frametimings = 0;
1570 self->linetimings = 0;
1571 self->lineevents = 1;
1572 }
1573 }
1574 return result;
1575}
1576
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001577PyDoc_VAR(resolution__doc__) =
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001578#ifdef MS_WINDOWS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001579PyDoc_STR(
Tim Petersfeab23f2001-10-13 00:11:10 +00001580"resolution() -> (performance-counter-ticks, update-frequency)\n"
1581"Return the resolution of the timer provided by the QueryPerformanceCounter()\n"
1582"function. The first value is the smallest observed change, and the second\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001583"is the result of QueryPerformanceFrequency()."
1584)
Fred Drake8c081a12001-10-12 20:57:55 +00001585#else
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001586PyDoc_STR(
Tim Petersfeab23f2001-10-13 00:11:10 +00001587"resolution() -> (gettimeofday-usecs, getrusage-usecs)\n"
1588"Return the resolution of the timers provided by the gettimeofday() and\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001589"getrusage() system calls, or -1 if the call is not supported."
1590)
Fred Drake8c081a12001-10-12 20:57:55 +00001591#endif
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001592;
Fred Drake8c081a12001-10-12 20:57:55 +00001593
1594static PyObject *
1595hotshot_resolution(PyObject *unused, PyObject *args)
1596{
1597 PyObject *result = NULL;
1598
1599 if (PyArg_ParseTuple(args, ":resolution")) {
1600 if (timeofday_diff == 0) {
1601 calibrate();
1602 calibrate();
1603 calibrate();
1604 }
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001605#ifdef MS_WINDOWS
Fred Drake8c081a12001-10-12 20:57:55 +00001606 result = Py_BuildValue("ii", timeofday_diff, frequency.LowPart);
1607#else
1608 result = Py_BuildValue("ii", timeofday_diff, rusage_diff);
1609#endif
1610 }
1611 return result;
1612}
1613
1614
1615static PyMethodDef functions[] = {
Fred Drake30d1c752001-10-15 22:11:02 +00001616 {"coverage", hotshot_coverage, METH_VARARGS, coverage__doc__},
Fred Drake8c081a12001-10-12 20:57:55 +00001617 {"profiler", hotshot_profiler, METH_VARARGS, profiler__doc__},
1618 {"logreader", hotshot_logreader, METH_VARARGS, logreader__doc__},
1619 {"resolution", hotshot_resolution, METH_VARARGS, resolution__doc__},
1620 {NULL, NULL}
1621};
1622
1623
1624void
1625init_hotshot(void)
1626{
1627 PyObject *module;
1628
1629 LogReaderType.ob_type = &PyType_Type;
1630 ProfilerType.ob_type = &PyType_Type;
1631 module = Py_InitModule("_hotshot", functions);
1632 if (module != NULL) {
1633 char *s = get_version_string();
1634
1635 PyModule_AddStringConstant(module, "__version__", s);
1636 free(s);
1637 Py_INCREF(&LogReaderType);
1638 PyModule_AddObject(module, "LogReaderType",
1639 (PyObject *)&LogReaderType);
1640 Py_INCREF(&ProfilerType);
1641 PyModule_AddObject(module, "ProfilerType",
1642 (PyObject *)&ProfilerType);
1643
1644 if (ProfilerError == NULL)
1645 ProfilerError = PyErr_NewException("hotshot.ProfilerError",
1646 NULL, NULL);
1647 if (ProfilerError != NULL) {
1648 Py_INCREF(ProfilerError);
1649 PyModule_AddObject(module, "ProfilerError", ProfilerError);
1650 }
1651 PyModule_AddIntConstant(module, "WHAT_ENTER", WHAT_ENTER);
1652 PyModule_AddIntConstant(module, "WHAT_EXIT", WHAT_EXIT);
1653 PyModule_AddIntConstant(module, "WHAT_LINENO", WHAT_LINENO);
1654 PyModule_AddIntConstant(module, "WHAT_OTHER", WHAT_OTHER);
1655 PyModule_AddIntConstant(module, "WHAT_ADD_INFO", WHAT_ADD_INFO);
1656 PyModule_AddIntConstant(module, "WHAT_DEFINE_FILE", WHAT_DEFINE_FILE);
Fred Drake30d1c752001-10-15 22:11:02 +00001657 PyModule_AddIntConstant(module, "WHAT_DEFINE_FUNC", WHAT_DEFINE_FUNC);
Fred Drake8c081a12001-10-12 20:57:55 +00001658 PyModule_AddIntConstant(module, "WHAT_LINE_TIMES", WHAT_LINE_TIMES);
1659 }
1660}